Working multifeed RSS - needs testing
[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 #define MAX_FEEDS 16
19
20 struct MemoryStruct {
21         char *memory;
22         size_t size;
23 };
24
25 typedef struct feed_ {
26         char* uri;
27         int last_update;
28         PRSS* data;
29 } feed;
30
31 int num_feeds = 0;
32 feed feeds[MAX_FEEDS];
33
34 size_t
35 WriteMemoryCallback(void *ptr, size_t size, size_t nmemb, void *data)
36 {
37         size_t realsize = size * nmemb;
38         struct MemoryStruct *mem = (struct MemoryStruct *)data;
39
40         mem->memory = (char *)realloc(mem->memory, mem->size + realsize + 1);
41         if (mem->memory) {
42                 memcpy(&(mem->memory[mem->size]), ptr, realsize);
43                 mem->size += realsize;
44                 mem->memory[mem->size] = 0;
45         }
46         return realsize;
47 }
48
49
50 int rss_delay(int *wait, int delay)
51 {
52         time_t now = time(NULL);
53
54         // make it minutes
55         if(delay < 1) delay = 1;
56         delay *= 60;
57
58         if(!*wait) {
59                 *wait = now + delay;
60                 return 1;
61         }
62
63         if(now >= *wait + delay) {
64                 *wait = now + delay;
65                 return 1;
66         }
67
68         return 0;
69 }
70
71 PRSS*
72 get_rss_info(char *uri, int delay)
73 {
74         CURL *curl = NULL;
75         CURLcode res;
76         // curl temps
77         struct MemoryStruct chunk;
78         chunk.memory = NULL;
79         chunk.size = 0;
80
81         // pointers to struct
82         feed *curfeed = NULL;
83         PRSS *curdata = NULL;
84         int *last_update = 0;
85
86         int i;
87
88         // first seek for the uri in list
89         if(num_feeds > 0) {
90                 for(i = 0; i < num_feeds; i++) {
91                         if(feeds[i].uri != NULL)
92                                 if(!strcmp(feeds[i].uri, uri)) {
93                                         curfeed = &feeds[i];
94                                         break;
95                                 }
96                 }
97         }
98
99         if(!curfeed) { // new feed
100                 if(num_feeds == MAX_FEEDS-1) return NULL;
101                 curfeed = &feeds[num_feeds];
102                 curfeed->uri = (char *)malloc(sizeof(char) * strlen(uri)+1);
103                 strncpy(curfeed->uri, uri, strlen(uri)+1);
104                 num_feeds++;
105         }
106
107         last_update = &curfeed->last_update;
108         curdata = curfeed->data;
109
110         if(!rss_delay(last_update, delay))
111                 return curdata; // wait for delay to pass
112
113         if(curdata != NULL)
114                 prss_free(curdata); // clean up old data
115
116         curl = curl_easy_init();
117         if(curl) {
118                 curl_easy_setopt(curl, CURLOPT_URL, uri);
119                 curl_easy_setopt(curl, CURLOPT_NOPROGRESS, 1);
120                 curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, WriteMemoryCallback);
121                 curl_easy_setopt(curl, CURLOPT_WRITEDATA, (void *)&chunk);
122                 curl_easy_setopt(curl, CURLOPT_USERAGENT, "conky-rss/1.0");
123
124                 res = curl_easy_perform(curl);
125                 if(chunk.size) {
126                         curdata = prss_parse_data(chunk.memory);
127                         free(chunk.memory);
128                 }
129
130                 curfeed->data = curdata;
131
132                 curl_easy_cleanup(curl);
133         }
134
135         return curdata;
136 }