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(strptime(date, "%a, %d %b %Y %T %z", &t));
  51         item_date = mktime(&t);
  52         assert(item_date != -1);
  53         if (item_date <= feed_date) {
  54                 printf("No new feeds assumed.\n");
  55                 fail(0);
  56         } else if (item_date > feed_date_new) {
  57                 feed_date_new = item_date;
  58         }
  59 }
  60
  61 void set_item_link(const char* link)
  62 {
  63         strncpy(item_link, link, TEXT_BUFFER_SIZE);
  64 }
  65
  66 void set_item_desc(const char* desc)
  67 {
  68         strncpy(item_desc, desc, TEXT_BUFFER_SIZE);
  69 }
  70
  71 void set_item_title(const char* title)
  72 {
  73         strncpy(item_title, title, TEXT_BUFFER_SIZE);
  74         sanitize(item_title);
  75 }
  76
  77 const char* get_item_title()
  78 {
  79         return item_title;
  80 }
  81
  82 void feed_open(const char* title)
  83 {
  84         struct stat st;
  85
  86         strncpy(feed_title, title, TEXT_BUFFER_SIZE);
  87         sanitize(feed_title);
  88
  89         if (!stat(feed_title, &st)) feed_date = st.st_mtime;
  90         mkdir(feed_title, 0755);
  91         assert(!chdir(feed_title));
  92         feed_is_open = true;
  93 }
  94
  95
  96
  97 void feed_flush()
  98 {
  99         struct timeval times[2];
 100         times[0].tv_sec  = feed_date_new;
 101         times[0].tv_usec = 0;
 102         times[1].tv_sec  = feed_date_new;
 103         times[1].tv_usec = 0;
 104
 105         if (!feed_is_open) return;
 106
 107         assert(!chdir(".."));
 108         assert(!utimes(feed_title, times));
 109         feed_is_open = false;
 110 }
 111
 112 void item_flush()
 113 {
 114         struct timeval times[2];
 115         FILE* f;
 116         char buf[FILE_NAME_LENGTH + 1] = { 0 };
 117         assert(feed_is_open);
 118
 119         if (item_title[0]) {
 120                 strncpy(buf, item_title, FILE_NAME_LENGTH);
 121         } else {
 122                 assert(item_desc[0]);
 123                 strncpy(buf, item_desc, FILE_NAME_LENGTH);
 124         }
 125
 126         assert(f = fopen(buf, "w"));
 127         if (item_title[0]) {
 128                 fprintf(f, "%s\n\n", item_title);
 129         }
 130         if (item_desc[0]) {
 131                 fprintf(f, "%s\n\n", item_desc);
 132         }
 133         if (item_link[0]) {
 134                 fprintf(f, "Link: <a href=\"%s\">%s</a>\n", item_link, item_link);
 135         }
 136         fclose(f);
 137
 138         /* FIXME: item date might not have been set. */
 139         times[0].tv_sec  = item_date;
 140         times[0].tv_usec = 0;
 141         times[1].tv_sec  = item_date;
 142         times[1].tv_usec = 0;
 143
 144         assert(!utimes(buf, times));
 145 }