RSS updates: more flexible output and updated prss code
[monky] / src / rss.c
1 /*
2  * rss.c
3  * RSS stuff (prss version)
4  *
5  * prss.c and prss.h written by Sisu (Mikko Sysikaski)
6  * new rss.c written by hifi (Toni Spets)
7  */
8
9 #include <stdio.h>
10 #include <string.h>
11 #include <time.h>
12 #include <assert.h>
13 #include "prss.h"
14 #include <curl/curl.h>
15 #include <curl/types.h>
16 #include <curl/easy.h>
17
18 #include "conky.h"
19
20 #define MAX_FEEDS 16
21
22 struct MemoryStruct {
23         char *memory;
24         size_t size;
25 };
26
27 typedef struct feed_ {
28         char* uri;
29         int last_update;
30         PRSS* data;
31 } feed;
32
33 int num_feeds = 0;
34 feed feeds[MAX_FEEDS];
35
36 size_t
37 WriteMemoryCallback(void *ptr, size_t size, size_t nmemb, void *data)
38 {
39         size_t realsize = size * nmemb;
40         struct MemoryStruct *mem = (struct MemoryStruct *)data;
41
42         mem->memory = (char *)realloc(mem->memory, mem->size + realsize + 1);
43         if (mem->memory) {
44                 memcpy(&(mem->memory[mem->size]), ptr, realsize);
45                 mem->size += realsize;
46                 mem->memory[mem->size] = 0;
47         }
48         return realsize;
49 }
50
51
52 int rss_delay(int *wait, int delay)
53 {
54         time_t now = time(NULL);
55
56         // make it minutes
57         if(delay < 1) delay = 1;
58         delay *= 60;
59
60         if(!*wait) {
61                 *wait = now + delay;
62                 return 1;
63         }
64
65         if(now >= *wait + delay) {
66                 *wait = now + delay;
67                 return 1;
68         }
69
70         return 0;
71 }
72
73 PRSS*
74 get_rss_info(char *uri, int delay)
75 {
76         CURL *curl = NULL;
77         CURLcode res;
78         // curl temps
79         struct MemoryStruct chunk;
80         chunk.memory = NULL;
81         chunk.size = 0;
82
83         // pointers to struct
84         feed *curfeed = NULL;
85         PRSS *curdata = NULL;
86         int *last_update = 0;
87
88         int i;
89
90         // first seek for the uri in list
91         if(num_feeds > 0) {
92                 for(i = 0; i < num_feeds; i++) {
93                         if(feeds[i].uri != NULL)
94                                 if(!strcmp(feeds[i].uri, uri)) {
95                                         curfeed = &feeds[i];
96                                         break;
97                                 }
98                 }
99         }
100
101         if(!curfeed) { // new feed
102                 if(num_feeds == MAX_FEEDS-1) return NULL;
103                 curfeed = &feeds[num_feeds];
104                 curfeed->uri = (char *)malloc(sizeof(char) * strlen(uri)+1);
105                 strncpy(curfeed->uri, uri, strlen(uri)+1);
106                 num_feeds++;
107         }
108
109         last_update = &curfeed->last_update;
110         curdata = curfeed->data;
111
112         if(!rss_delay(last_update, delay))
113                 return curdata; // wait for delay to pass
114
115         if(curdata != NULL)
116                 prss_free(curdata); // clean up old data
117
118         curl = curl_easy_init();
119         if(curl) {
120                 curl_easy_setopt(curl, CURLOPT_URL, uri);
121                 curl_easy_setopt(curl, CURLOPT_NOPROGRESS, 1);
122                 curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, WriteMemoryCallback);
123                 curl_easy_setopt(curl, CURLOPT_WRITEDATA, (void *)&chunk);
124                 curl_easy_setopt(curl, CURLOPT_USERAGENT, "conky-rss/1.0");
125
126                 res = curl_easy_perform(curl);
127                 if(chunk.size) {
128                         curdata = prss_parse_data(chunk.memory);
129                         free(chunk.memory);
130                 } else
131                         ERR("No data from server");
132
133                 curfeed->data = curdata;
134
135                 curl_easy_cleanup(curl);
136         }
137
138         return curdata;
139 }