conky/src/rss.c

/*
 * rss.c
 * RSS stuff (prss version)
 *
 * prss.c and prss.h written by Sisu (Mikko Sysikaski)
 * new rss.c written by hifi (Toni Spets)
 */

#include <stdio.h>
#include <string.h>
#include <time.h>
#include <assert.h>
#include "prss.h"
#include <curl/curl.h>
#include <curl/types.h>
#include <curl/easy.h>

#define MAX_FEEDS 16

struct MemoryStruct {
	char *memory;
	size_t size;
};

typedef struct feed_ {
	char* uri;
	int last_update;
	PRSS* data;
} feed;

int num_feeds = 0;
feed feeds[MAX_FEEDS];

size_t
WriteMemoryCallback(void *ptr, size_t size, size_t nmemb, void *data)
{
	size_t realsize = size * nmemb;
	struct MemoryStruct *mem = (struct MemoryStruct *)data;

	mem->memory = (char *)realloc(mem->memory, mem->size + realsize + 1);
	if (mem->memory) {
		memcpy(&(mem->memory[mem->size]), ptr, realsize);
		mem->size += realsize;
		mem->memory[mem->size] = 0;
	}
	return realsize;
}


int rss_delay(int *wait, int delay)
{
	time_t now = time(NULL);

	// make it minutes
	if(delay < 1) delay = 1;
	delay *= 60;

	if(!*wait) {
		*wait = now + delay;
		return 1;
	}

	if(now >= *wait + delay) {
		*wait = now + delay;
		return 1;
	}

	return 0;
}

PRSS*
get_rss_info(char *uri, int delay)
{
	CURL *curl = NULL;
	CURLcode res;
	// curl temps
	struct MemoryStruct chunk;
	chunk.memory = NULL;
	chunk.size = 0;

	// pointers to struct
	feed *curfeed = NULL;
	PRSS *curdata = NULL;
	int *last_update = 0;

	int i;

	// first seek for the uri in list
	if(num_feeds > 0) {
		for(i = 0; i < num_feeds; i++) {
			if(feeds[i].uri != NULL)
				if(!strcmp(feeds[i].uri, uri)) {
					curfeed = &feeds[i];
					break;
				}
		}
	}

	if(!curfeed) { // new feed
		if(num_feeds == MAX_FEEDS-1) return NULL;
		curfeed = &feeds[num_feeds];
		curfeed->uri = (char *)malloc(sizeof(char) * strlen(uri)+1);
		strncpy(curfeed->uri, uri, strlen(uri)+1);
		num_feeds++;
	}

	last_update = &curfeed->last_update;
	curdata = curfeed->data;

	if(!rss_delay(last_update, delay))
		return curdata; // wait for delay to pass

	if(curdata != NULL)
		prss_free(curdata); // clean up old data

	curl = curl_easy_init();
	if(curl) {
		curl_easy_setopt(curl, CURLOPT_URL, uri);
		curl_easy_setopt(curl, CURLOPT_NOPROGRESS, 1);
		curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, WriteMemoryCallback);
		curl_easy_setopt(curl, CURLOPT_WRITEDATA, (void *)&chunk);
		curl_easy_setopt(curl, CURLOPT_USERAGENT, "conky-rss/1.0");

		res = curl_easy_perform(curl);
		if(chunk.size) {
			curdata = prss_parse_data(chunk.memory);
			free(chunk.memory);
		}

		curfeed->data = curdata;

		curl_easy_cleanup(curl);
	}

	return curdata;
}