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 bool   feed_is_open = false;
  35
  36 char   item_title[TEXT_BUFFER_SIZE] = { 0 };
  37 char   item_link [TEXT_BUFFER_SIZE] = { 0 };
  38 char   item_desc [TEXT_BUFFER_SIZE] = { 0 };
  39 time_t item_date = 0;
  40
  41 void sanitize(char* str)
  42 {
  43         char* evil;
  44         while ((evil = strchr(str, '/'))) *evil = '\\';
  45 }
  46
  47 void set_item_date(const char* date)
  48 {
  49         struct tm t;
  50         assert(item_title[0]); /* Has to be set first. */
  51         assert(strptime(date, "%a, %d %b %Y %T %z", &t));
  52         item_date = mktime(&t);
  53         assert(item_date != -1);
  54         if (item_date <= feed_date) {
  55                 printf("No new feeds assumed.\n");
  56                 fail(0);
  57         } else if (item_date > feed_date_new) {
  58                 feed_date_new = item_date;
  59         }
  60 }
  61
  62 void set_item_link(const char* link)
  63 {
  64         assert(item_title[0]); /* Has to be set first. */
  65         strncpy(item_link, link, TEXT_BUFFER_SIZE);
  66 }
  67
  68 void set_item_desc(const char* desc)
  69 {
  70         strncpy(item_desc, desc, TEXT_BUFFER_SIZE);
  71 }
  72
  73 void set_item_title(const char* title)
  74 {
  75         strncpy(item_title, title, TEXT_BUFFER_SIZE - 1);
  76
  77         assert(item_desc[0] == 0); /* Has to be set first. */
  78         if (strlen(title) >= TEXT_BUFFER_SIZE) {
  79                 item_title[TEXT_BUFFER_SIZE - 4] = '.';
  80                 item_title[TEXT_BUFFER_SIZE - 3] = '.';
  81                 item_title[TEXT_BUFFER_SIZE - 2] = '.';
  82         }
  83
  84         sanitize(item_title);
  85 }
  86
  87 const char* get_item_title()
  88 {
  89         return item_title;
  90 }
  91
  92 void feed_open(const char* title)
  93 {
  94         struct stat st;
  95
  96         strncpy(feed_title, title, TEXT_BUFFER_SIZE);
  97         sanitize(feed_title);
  98
  99         if (!stat(feed_title, &st)) feed_date = st.st_mtime;
 100         mkdir(feed_title, 0755);
 101         assert(!chdir(feed_title));
 102         feed_is_open = true;
 103 }
 104
 105
 106
 107 void feed_flush()
 108 {
 109         struct timeval times[2];
 110         times[0].tv_sec  = feed_date_new;
 111         times[0].tv_usec = 0;
 112         times[1].tv_sec  = feed_date_new;
 113         times[1].tv_usec = 0;
 114
 115         if (!feed_is_open) return;
 116
 117         assert(!chdir(".."));
 118         assert(!utimes(feed_title, times));
 119         feed_is_open = false;
 120 }
 121
 122 void item_flush()
 123 {
 124         struct timeval times[2];
 125         FILE* f;
 126         char buf[FILE_NAME_LENGTH + 1] = { 0 };
 127         assert(feed_is_open);
 128
 129         if (item_title[0]) {
 130                 strncpy(buf, item_title, FILE_NAME_LENGTH);
 131                 if (strlen(item_title) > FILE_NAME_LENGTH) {
 132                         buf[FILE_NAME_LENGTH - 3] = '.';
 133                         buf[FILE_NAME_LENGTH - 2] = '.';
 134                         buf[FILE_NAME_LENGTH - 1] = '.';
 135                 }
 136         } else {
 137                 assert(item_desc[0]);
 138                 strncpy(buf, item_desc, FILE_NAME_LENGTH);
 139         }
 140
 141         assert(f = fopen(buf, "w"));
 142         if (item_title[0]) {
 143                 fprintf(f, "%s\n\n", item_title);
 144         }
 145         if (item_desc[0]) {
 146                 fprintf(f, "%s\n\n", item_desc);
 147         }
 148         if (item_link[0]) {
 149                 fprintf(f, "Link: <a href=\"%s\">%s</a>\n", item_link, item_link);
 150         }
 151         fclose(f);
 152
 153         /* FIXME: item date might not have been set. */
 154         times[0].tv_sec  = item_date;
 155         times[0].tv_usec = 0;
 156         times[1].tv_sec  = item_date;
 157         times[1].tv_usec = 0;
 158
 159         assert(!utimes(buf, times));
 160
 161         item_title[0] = 0;
 162         item_desc[0] = 0;
 163         item_link[0] = 0;
 164 }