src/fs.c

   1 /* agg - the news aggregator
   2  * Copyright (C) 2011 Andreas Waidler <arandes@programmers.at>
   3  *
   4  * Permission to use, copy, modify, and/or distribute this software for any
   5  * purpose with or without fee is hereby granted, provided that the above
   6  * copyright notice and this permission notice appear in all copies.
   7  *
   8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
   9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  15  */
  16 #define _XOPEN_SOURCE
  17 #include <sys/time.h>
  18 #include <sys/stat.h>
  19 #include <unistd.h>
  20 #include <string.h>
  21 #include <assert.h>
  22 #include <errno.h>
  23 #include <stdio.h>
  24 #include <fcntl.h>
  25 #include <time.h>
  26 #include "config.h"
  27 #include "bool.h"
  28 #include "fail.h"
  29 #include "fs.h"
  30
  31 char   feed_title[TEXT_BUFFER_SIZE] = { 0 };
  32 time_t feed_date = 0;
  33 time_t feed_date_new = 0;
  34
  35 char   item_title[TEXT_BUFFER_SIZE] = { 0 };
  36 char   item_link [TEXT_BUFFER_SIZE] = { 0 };
  37 char   item_desc [TEXT_BUFFER_SIZE] = { 0 };
  38 time_t item_date = 0;
  39
  40 void fs_buffer(char* dest, const char* src, size_t size)
  41 {
  42         assert(dest);
  43         assert(src);
  44         assert(size);
  45
  46         strncpy(dest, src, size);
  47
  48         /* Destination is not null-terminated iff
  49          * strlen(src) >= size. */
  50         if (dest[size - 1] != '\0') {
  51                 dest[size - 4] = '.';
  52                 dest[size - 3] = '.';
  53                 dest[size - 2] = '.';
  54                 dest[size - 1] = '\0';
  55         }
  56
  57         assert(strlen(dest) < size);
  58 }
  59
  60 void sanitize(char* str)
  61 {
  62         char* evil;
  63         while ((evil = strchr(str, '/'))) *evil = '\\';
  64 }
  65
  66 void set_item_date(const char* date)
  67 {
  68         struct tm t;
  69         assert(strptime(date, "%a, %d %b %Y %T %z", &t));
  70         item_date = mktime(&t);
  71         assert(item_date != -1);
  72 }
  73
  74 void set_item_link(const char* link)
  75 {
  76         fs_buffer(item_link, link, TEXT_BUFFER_SIZE);
  77 }
  78
  79 void set_item_desc(const char* desc)
  80 {
  81         fs_buffer(item_desc, desc, TEXT_BUFFER_SIZE);
  82 }
  83
  84 void set_item_title(const char* title)
  85 {
  86         fs_buffer(item_title, title, TEXT_BUFFER_SIZE);
  87 }
  88
  89 const char* get_item_title()
  90 {
  91         return item_title;
  92 }
  93
  94 void feed_open(const char* title)
  95 {
  96         struct stat st;
  97
  98         if (feed_title[0]) {
  99                 /* Feed contained title twice. The
 100                  * directory may just have been created.
 101                  * Deletion may lead to loss of previous
 102                  * mtime. We don't care; that feed is
 103                  * fucked up anyways. */
 104                 chdir("..");
 105                 rmdir(feed_title);
 106                 fail(111);
 107         }
 108
 109         fs_buffer(feed_title, title, TEXT_BUFFER_SIZE);
 110         sanitize(feed_title);
 111
 112         if (!stat(feed_title, &st)) feed_date = st.st_mtime;
 113         mkdir(feed_title, 0755);
 114         assert(!chdir(feed_title));
 115 }
 116
 117
 118
 119 void feed_flush()
 120 {
 121         struct timeval times[2];
 122         times[0].tv_sec  = feed_date_new;
 123         times[0].tv_usec = 0;
 124         times[1].tv_sec  = feed_date_new;
 125         times[1].tv_usec = 0;
 126
 127         if (!feed_title[0]) return;
 128
 129         assert(!chdir(".."));
 130         assert(!utimes(feed_title, times));
 131         feed_title[0] = '\0';
 132 }
 133
 134
 135 void fs_set(const char* name, const char* value)
 136 {
 137         FILE* f;
 138
 139         assert(name);
 140         assert(value);
 141
 142         /* Only write if attribute has a value. */
 143         if (value[0]) {
 144                 assert(f = fopen(name, "w"));
 145                 fprintf(f, "%s", value);
 146                 fclose(f);
 147         }
 148 }
 149
 150 void item_flush()
 151 {
 152         struct timeval times[2];
 153         char buf[FILE_NAME_LENGTH + 1] = { 0 };
 154         const char* name_src;
 155         assert(feed_title[0]);
 156
 157         /* snip: from set_item_date */
 158         /* assert(item_title[0]); */
 159         if (item_date > 0 && item_date <= feed_date) {
 160                 printf("No new feeds assumed.\n");
 161                 fail(0);
 162         } else if (item_date > feed_date_new) {
 163                 feed_date_new = item_date;
 164         }
 165         /* snap */
 166
 167         name_src = item_title[0] ? item_title : item_desc;
 168         assert(name_src[0]);
 169         fs_buffer(buf, name_src, FILE_NAME_LENGTH + 1);
 170         sanitize(buf);
 171
 172         assert(mkdir(buf, 0755) == 0);
 173         assert(chdir(buf) == 0);
 174         fs_set("title", item_title);
 175         fs_set("desc",  item_desc);
 176         fs_set("link", item_link);
 177         assert(chdir("..") == 0);
 178
 179         /* FIXME: item date might not have been set. */
 180
 181
 182         times[0].tv_sec  = item_date;
 183         times[0].tv_usec = 0;
 184         times[1].tv_sec  = item_date;
 185         times[1].tv_usec = 0;
 186
 187         assert(!utimes(buf, times));
 188
 189         item_title[0] = 0;
 190         item_desc[0] = 0;
 191         item_link[0] = 0;
 192 }