[Libirc-commits] SF.net SVN: libirc: [353] trunk/libirc/examples/stupidBot/src/stupidBot.cpp

[<Jef...@us...>]

Revision: 353
          http://libirc.svn.sourceforge.net/libirc/?rev=353&view=rev
Author:   JeffM2501
Date:     2007-11-28 14:35:25 -0800 (Wed, 28 Nov 2007)

Log Message:
-----------
start an rss feed reporter for CIA feeds.

Modified Paths:
--------------
    trunk/libirc/examples/stupidBot/src/stupidBot.cpp

Modified: trunk/libirc/examples/stupidBot/src/stupidBot.cpp
===================================================================
--- trunk/libirc/examples/stupidBot/src/stupidBot.cpp	2007-11-28 21:12:49 UTC (rev 352)
+++ trunk/libirc/examples/stupidBot/src/stupidBot.cpp	2007-11-28 22:35:25 UTC (rev 353)
@@ -21,6 +21,70 @@
 #include <string>
 #include <map>
 
+#include "URLManager.h"
+
+#ifdef _WIN32
+#include <windows.h>
+#else
+
+#endif
+
+double getCurrentSeconds ( void )
+{
+#ifdef _WIN32
+	return GetTickCount() * 1000.0;
+#else
+	return (double)time(NULL);
+#endif
+}
+
+class CIARSSFeedAnnouncer : public URLJobHandler
+{
+public:
+	CIARSSFeedAnnouncer( const std::string &chan, const std::string &project );
+
+	void set ( const std::string &chan, const std::string &proj );
+
+	virtual bool write ( int jobID, unsigned char* data, unsigned int size );
+	virtual bool done ( int jobID, unsigned char* data, unsigned int size );	// return true to have the URLManager delete the handler
+
+	void update ( IRCClient &client );
+
+protected:
+	void sendFeed ( IRCClient &client );
+	void getFeed ( void );
+
+	typedef struct  
+	{
+		// raw feed data
+		std::string decription;
+		std::string title;
+		std::string guID;
+		std::string pubDate;
+
+		std::string author;
+		std::string project;
+		std::string revistion;
+		std::string path;
+		std::string comment;
+	}CIAFeedRecord;
+
+	double lastUpdateTime;
+	double updateFrequency;
+
+	bool working;
+
+	std::string channel;
+	std::string project;
+	std::string url;
+	std::string filePath;
+
+	std::string feedData;
+
+	std::string lastPubDate;
+	std::string lastGUID;
+};
+
 typedef struct 
 {
 	std::string name;
@@ -68,6 +132,7 @@
 	channelEcho		chanEchos;
 	channelCIAEcho	CIAEchos;
 	bool			ignorePMs;
+	std::string		rssStoragePath;
 }trStupidBotInfo;
 
 trStupidBotInfo	theBotInfo;
@@ -119,6 +184,8 @@
 {
 	theBotInfo.config = file;
 	theBotInfo.ignorePMs = false;
+	theBotInfo.rssStoragePath = "./";
+
 	std::string config;
 
 	// read the file data
@@ -282,6 +349,8 @@
 			}
 			else if ( command == "ignorepm" )
 				theBotInfo.ignorePMs = true;
+			else if ( command == "ciarsspath")
+				theBotInfo.rssStoragePath = params[0];
 		}
 		itr++;
 	}
@@ -443,6 +512,12 @@
 		fprintf(fp,"%s",lineEnd.c_str());
 	}
 
+	if (theBotInfo.rssStoragePath.size() )
+	{
+		fprintf(fp,"ciarsspath:\"%s\"",theBotInfo.rssStoragePath.c_str());
+		fprintf(fp,"%s",lineEnd.c_str());
+	}
+
 	fclose(fp);
 }
 
@@ -1603,4 +1678,128 @@
 	installBotCommand(new kickCommand);
 }
 
+//------------CIARSSFeedAnnouncer---------------
 
+CIARSSFeedAnnouncer::CIARSSFeedAnnouncer( const std::string &chan, const std::string &project )
+{
+	set(chan,project);
+}
+
+void CIARSSFeedAnnouncer::set ( const std::string &chan, const std::string &proj )
+{
+	working = false;
+	updateFrequency = 5.0 * 60.0;
+	lastUpdateTime = updateFrequency * -2.0;
+
+	project = proj;
+	channel = chan;
+	url = "http://cia.vc/stats/project/" + project + "/.rss?ver=2&medium=plaintex";
+	feedData = "";
+
+	filePath = theBotInfo.rssStoragePath + project + ".rss";
+	FILE *fp = fopen(filePath.c_str(),"rt");
+	if (fp)
+	{
+		char temp[512] = {0};
+		fscanf(fp,"%s",temp);
+		lastPubDate = temp;
+		temp[0] = 0;
+		fscanf(fp,"%s",temp);
+		lastGUID = temp;
+		fclose(fp);
+	}
+}
+
+bool CIARSSFeedAnnouncer::write ( int jobID, unsigned char* data, unsigned int size )
+{
+	char *p = (char*)malloc(size+1);
+	if (p)
+	{
+		memcpy(p,data,size);
+		p[size] =0;
+		feedData += p;
+		free(p);
+	}
+	return false;
+}
+
+bool CIARSSFeedAnnouncer::done ( int jobID, unsigned char* data, unsigned int size )
+{
+	lastUpdateTime = getCurrentSeconds();
+
+	char *p = (char*)malloc(size+1);
+	if (p)
+	{
+		memcpy(p,data,size);
+		p[size] =0;
+		feedData += p;
+		free(p);
+	}
+
+	working = false;
+	return false;
+}// return true to have the URLManager delete the handler
+
+void CIARSSFeedAnnouncer::update ( IRCClient &client )
+{
+	if (!working)
+	{
+		if (feedData.size())
+		{
+			sendFeed (client);
+		}
+		else
+		{
+			double now = getCurrentSeconds();
+			if ( lastUpdateTime + updateFrequency  < now )
+				getFeed();
+		}
+	}
+}
+
+void CIARSSFeedAnnouncer::sendFeed ( IRCClient &client )
+{
+	std::vector<CIAFeedRecord> feedRecords;
+	std::vector<std::string> feedStrings;
+
+	feedStrings = string_util::tokenize(feedData,std::string("<item>"));
+
+	if ( feedStrings.size() > 1)
+	{
+		// parse the crap out of each line
+
+		for ( int i = 1; i < (int)feedStrings.size(); i++ )
+		{
+
+			std::string line = feedStrings[i];
+			line = string_util::replace_all(line,std::string("</pubDate>"),std::string(""));
+			line = string_util::replace_all(line,std::string("</guid>"),std::string(""));
+			line = string_util::replace_all(line,std::string("</link>"),std::string(""));
+			line = string_util::replace_all(line,std::string("</description>"),std::string(""));
+			line = string_util::replace_all(line,std::string("</title>"),std::string(""));
+			line = string_util::replace_all(line,std::string("</item>"),std::string(""));
+		}
+	}
+
+	// write out the last pub date so we can get it in case we get shutdown.
+	FILE *fp = fopen(filePath.c_str(),"wt");
+	if (fp)
+	{
+		fprintf(fp,"%s\n",lastPubDate.c_str());
+		fprintf(fp,"%s\n",lastGUID.c_str());
+		fclose(fp);
+	}
+
+	feedData = "";
+}
+
+void CIARSSFeedAnnouncer::getFeed ( void )
+{
+	if (working)
+		return;
+
+	feedData = "";
+	working = true;
+	URLManager::Instance().addJob ( url, this );
+}
+


This was sent by the SourceForge.net collaborative development platform, the world's largest Open Source development site.