Changeset

249:50c4348c2494

mod_twitter: Initial commit.
author dersd <xdersd@gmail.com>
date Sat, 28 Aug 2010 22:19:54 +0400
parents 248:db1b4c6089b6
children 250:67f03b7786af
files mod_twitter/mod_twitter.lua
diffstat 1 files changed, 456 insertions(+), 0 deletions(-) [+]
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/mod_twitter/mod_twitter.lua	Sat Aug 28 22:19:54 2010 +0400
@@ -0,0 +1,456 @@
+-- for Prosody
+-- via dersd
+
+if module:get_host_type() ~= "component" then
+	error(module.name.." should be loaded as a component, check out http://prosody.im/doc/components", 0);
+end
+
+local jid_split = require "util.jid".split;
+local st = require "util.stanza";
+local componentmanager = require "core.componentmanager";
+local datamanager = require "util.datamanager";
+local timer = require "util.timer";
+local http = require "net.http";
+local json = require "json";
+local base64 = require "util.encodings".base64;
+
+local component_host = module:get_host();
+local component_name = module.name;
+local data_cache = {};
+
+function print_r(obj)
+	return require("util.serialization").serialize(obj);
+end
+
+function send_stanza(stanza)
+	if stanza ~= nil then
+		core_route_stanza(prosody.hosts[component_host], stanza)
+	end
+end
+
+function dmsg(jid, msg)
+	module:log("debug", msg or "nil");
+	if jid ~= nil then
+		send_stanza(st.message({to=jid, from=component_host, type='chat'}):tag("body"):text(msg or "nil"):up());
+	end
+end
+
+function substring(string, start_string, ending_string)
+	local s_value_start, s_value_finish = nil, nil;
+	if start_string ~= nil then
+		_, s_value_start = string:find(start_string);
+		if s_value_start == nil then
+			-- error
+			return nil;
+		end
+	else
+		return nil;
+	end
+	if ending_string ~= nil then
+		_, s_value_finish = string:find(ending_string, s_value_start+1);
+		if s_value_finish == nil then
+			-- error
+			return nil;
+		end
+	else
+		s_value_finish = string:len()+1;
+	end
+	return string:sub(s_value_start+1, s_value_finish-1);
+end
+
+local http_timeout = 30;
+local http_queue = setmetatable({}, { __mode = "k" }); -- auto-cleaning nil elements
+data_cache['prosody_os'] = prosody.platform;
+data_cache['prosody_version'] = prosody.version;
+local http_headers = {
+	["User-Agent"] = "Prosody ("..data_cache['prosody_version'].."; "..data_cache['prosody_os']..")" --"ELinks (0.4pre5; Linux 2.4.27 i686; 80x25)",
+};
+
+function http_action_callback(response, code, request, xcallback)
+	if http_queue == nil or http_queue[request] == nil then return; end
+	local id = http_queue[request];
+	http_queue[request] = nil;
+	if xcallback == nil then
+		dmsg(nil, "http_action_callback reports that xcallback is nil");
+	else
+		xcallback(id, response, request);
+	end
+	return true;
+end
+
+function http_add_action(tid, url, method, post, fcallback)
+	local request = http.request(url, { headers = http_headers or {}, body = http.formencode(post or {}), method = method or "GET" }, function(response, code, request) http_action_callback(response, code, request, fcallback) end);
+	http_queue[request] = tid;
+	timer.add_task(http_timeout, function() http.destroy_request(request); end);
+	return true;
+end
+
+local users = setmetatable({}, {__mode="k"});
+local user = {};
+user.__index = user;
+user.dosync = false;
+user.valid = false;
+user.data = {};
+
+function user:login()
+	userdata = datamanager.load(self.jid, component_host, "data");
+	if userdata ~= nil then
+		self.data = userdata;
+		if self.data['_twitter_sess'] ~= nil then
+			http_headers['Cookie'] = "_twitter_sess="..self.data['_twitter_sess']..";";
+		end
+		send_stanza(st.presence({to=self.jid, from=component_host}));
+		self:twitterAction("VerifyCredentials");
+		if self.data.dosync == 1 then
+			self.dosync = true;
+			timer.add_task(self.data.refreshrate, function() return users[self.jid]:sync(); end)
+		end
+	else
+		send_stanza(st.message({to=self.jid, from=component_host, type='chat'}):tag("body"):text("You are not signed in."));
+	end
+end
+
+function user:logout()
+	datamanager.store(self.jid, component_host, "data", self.data);
+	self.dosync = false;
+	send_stanza(st.presence({to=self.jid, from=component_host, type='unavailable'}));
+end
+
+function user:sync()
+	if self.dosync then
+		table.foreach(self.data.synclines, function(ind, line) self:twitterAction(line.name, {sinceid=line.sinceid}) end);
+		return self.data.refreshrate;
+	end
+end
+
+function user:signin()
+	if datamanager.load(self.jid, component_host, "data") == nil then
+		datamanager.store(self.jid, component_host, "data", {login=self.data.login, password=self.data.password, refreshrate=60, dosync=1, synclines={{name='HomeTimeline', sinceid=0}}, syncstatus=0})
+		send_stanza(st.presence{to=self.jid, from=component_host, type='subscribe'});
+		send_stanza(st.presence{to=self.jid, from=component_host, type='subscribed'});
+	end
+end
+
+function user:signout()
+	if datamanager.load(self.jid, component_host, "data") ~= nil then
+		datamanager.store(self.jid, component_host, "data", nil);
+		send_stanza(st.presence({to=self.jid, from=component_host, type='unavailable'}));
+		send_stanza(st.presence({to=self.jid, from=component_host, type='unsubscribe'}));
+		send_stanza(st.presence({to=self.jid, from=component_host, type='unsubscribed'}));
+	end
+end
+
+local twitterApiUrl = "http://api.twitter.com";
+local twitterApiVersion = "1";
+local twitterApiDataType = "json";
+local twitterActionUrl = function(action) return twitterApiUrl.."/"..twitterApiVersion.."/"..action.."."..twitterApiDataType end;
+local twitterActionMap = {
+	PublicTimeline = {
+		url = twitterActionUrl("statuses/public_timeline"),
+		method = "GET",
+		needauth = false,
+	},
+	HomeTimeline = {
+		url = twitterActionUrl("statuses/home_timeline"),
+		method = "GET",
+		needauth = true,
+	},
+	FriendsTimeline = {
+		url = twitterActionUrl("statuses/friends_timeline"),
+		method = "GET",
+		needauth = true,
+	},
+	UserTimeline = {
+		url = twitterActionUrl("statuses/friends_timeline"),
+		method = "GET",
+		needauth = true,
+	},
+	VerifyCredentials = {
+		url = twitterActionUrl("account/verify_credentials"),
+		method = "GET",
+		needauth = true,
+	},
+	UpdateStatus = {
+		url = twitterActionUrl("statuses/update"),
+		method = "POST",
+		needauth = true,
+	},
+	Retweet = {
+		url = twitterActionUrl("statuses/retweet/%tweetid"),
+		method = "POST",
+		needauth = true,
+	}
+}
+
+function user:twitterAction(line, params)
+	local action = twitterActionMap[line];
+	if action then
+		local url = action.url;
+		local post = {};
+		--if action.needauth and not self.valid and line ~= "VerifyCredentials" then
+		--	return
+		--end
+		if action.needauth then
+			http_headers['Authorization'] = "Basic "..base64.encode(self.data.login..":"..self.data.password);
+			--url = string.gsub(url, "http\:\/\/", string.format("http://%s:%s@", self.data.login, self.data.password));
+		end
+		if params and type(params) == "table" then
+			post = params;
+		end
+		if action.method == "GET" and post ~= {} then
+			url = url.."?"..http.formencode(post);
+		end
+		http_add_action(line, url, action.method, post, function(...) self:twitterActionResult(...) end);
+	else
+		send_stanza(st.message({to=self.jid, from=component_host, type='chat'}):tag("body"):text("Wrong twitter action!"):up());
+	end
+end
+
+local twitterActionResultMap = {
+	PublicTimeline = {exec=function(jid, response)
+		--send_stanza(st.message({to=jid, from=component_host, type='chat'}):tag("body"):text(print_r(response)):up());
+		return
+	end},
+	HomeTimeline = {exec=function(jid, response)
+		--send_stanza(st.message({to=jid, from=component_host, type='chat'}):tag("body"):text(print_r(response)):up());
+		return
+	end},
+	FriendsTimeline = {function(jid, response)
+		return
+	end},
+	UserTimeline = {exec=function(jid, response)
+		return
+	end},
+	VerifyCredentials = {exec=function(jid, response)
+		if response ~= nil and response.id ~= nil then
+			users[jid].valid = true;
+			users[jid].id = response.id;
+		end
+		return
+	end},
+	UpdateStatus = {exec=function(jid, response)
+		return
+	end},
+	Retweet = {exec=function(jid, response)
+		return
+	end}
+}
+
+function user:twitterActionResult(id, response, request)
+	if request ~= nil and request.responseheaders['set-cookie'] ~= nil and request.responseheaders['location'] ~= nil then
+		--self.data['_twitter_sess'] = substring(request.responseheaders['set-cookie'], "_twitter_sess=", ";");
+		--http_add_action(id, request.responseheaders['location'], "GET", {}, function(...) self:twitterActionResult(...) end);
+		return true;
+	end
+	local result, tmp_json = pcall(function() json.decode(response or "{}") end);
+	if result and id ~= nil then
+		twitterActionResultMap[id]:exec(self.jid, tmp_json);
+	end
+	return true;
+end
+
+function iq_success(event)
+	local origin, stanza = event.origin, event.stanza;
+	local reply = data_cache.success;
+	if reply == nil then
+		reply = st.iq({type='result', from=stanza.attr.to or component_host});
+		data_cache.success = reply;
+	end
+	reply.attr.id = stanza.attr.id;
+	reply.attr.to = stanza.attr.from;
+	origin.send(reply);
+	return true;
+end
+
+function iq_disco_info(event)
+	local origin, stanza = event.origin, event.stanza;
+	local from = {};
+	from.node, from.host, from.resource = jid_split(stanza.attr.from);
+	local bjid = from.node.."@"..from.host;
+	local reply = data_cache.disco_info;
+	if reply == nil then
+		reply = st.iq({type='result', from=stanza.attr.to or component_host}):query("http://jabber.org/protocol/disco#info")
+				:tag("identity", {category='gateway', type='chat', name=component_name}):up();
+		reply = reply:tag("feature", {var="urn:xmpp:receipts"}):up();
+        reply = reply:tag("feature", {var="http://jabber.org/protocol/commands"}):up();
+        reply = reply:tag("feature", {var="jabber:iq:register"}):up();
+ 		--reply = reply:tag("feature", {var="jabber:iq:time"}):up();
+		--reply = reply:tag("feature", {var="jabber:iq:version"}):up();
+        --reply = reply:tag("feature", {var="http://jabber.org/protocol/stats"}):up();
+		data_cache.disco_info = reply;
+	end
+	reply.attr.id = stanza.attr.id;
+	reply.attr.to = stanza.attr.from;
+	origin.send(reply);
+	return true;
+end
+
+function iq_disco_items(event)
+	local origin, stanza = event.origin, event.stanza;
+	local reply = data_cache.disco_items;
+ 	if reply == nil then
+		reply = st.iq({type='result', from=stanza.attr.to or component_host}):query("http://jabber.org/protocol/disco#items");
+		data_cache.disco_items = reply;
+	end
+	reply.attr.id = stanza.attr.id;
+	reply.attr.to = stanza.attr.from;
+	origin.send(reply);
+	return true;
+end
+
+function iq_register(event)
+	local origin, stanza = event.origin, event.stanza;
+	if stanza.attr.type == "get" then
+		local reply = data_cache.registration_form;
+		if reply == nil then
+			reply = st.iq({type='result', from=stanza.attr.to or component_host})
+					:tag("query", { xmlns="jabber:iq:register" })
+					:tag("instructions"):text("Enter your twitter data"):up()
+					:tag("username"):up()
+					:tag("password"):up();
+			data_cache.registration_form = reply
+		end
+		reply.attr.id = stanza.attr.id;
+		reply.attr.to = stanza.attr.from;
+		origin.send(reply);
+	elseif stanza.attr.type == "set" then
+		local from = {};
+		from.node, from.host, from.resource = jid_split(stanza.attr.from);
+		local bjid = from.node.."@"..from.host;
+		local username, password = "", "";
+		local reply;
+		for _, tag in ipairs(stanza.tags[1].tags) do
+			if tag.name == "remove" then
+				users[bjid]:signout();
+				iq_success(event);
+				return true;
+			end
+			if tag.name == "username" then
+				username = tag[1];
+			end
+			if tag.name == "password" then
+				password = tag[1];
+			end
+		end
+		if username ~= nil and password ~= nil then
+			users[bjid] = setmetatable({}, user);
+			users[bjid].jid = bjid;
+			users[bjid].data.login = username;
+			users[bjid].data.password = password;
+			users[bjid]:signin();
+			users[bjid]:login();
+		end
+		iq_success(event);
+		return true;
+	end
+end
+
+function presence_stanza_handler(event)
+	local origin, stanza = event.origin, event.stanza;
+	local to = {};
+	local from = {};
+	local pres = {};
+	to.node, to.host, to.resource = jid_split(stanza.attr.to);
+	from.node, from.host, from.resource = jid_split(stanza.attr.from);
+	pres.type = stanza.attr.type;
+	for _, tag in ipairs(stanza.tags) do pres[tag.name] = tag[1]; end
+	local from_bjid = nil;
+	if from.node ~= nil and from.host ~= nil then
+		from_bjid = from.node.."@"..from.host;
+	elseif from.host ~= nil then
+		from_bjid = from.host;
+	end
+	if pres.type == nil then
+		if users[from_bjid] ~= nil then
+			-- Status change
+			if pres['status'] ~= nil and users[from_bjid]['data']['sync_status'] then
+				users[from_bjid]:twitterAction("UpdateStatus", {status=pres['status']});
+			end
+		else
+			-- User login request
+			users[from_bjid] = setmetatable({}, user);
+			users[from_bjid].jid = from_bjid;
+			users[from_bjid]:login();
+		end
+		origin.send(st.presence({to=from_bjid, from=component_host}));
+	elseif pres.type == 'subscribe' and users[from_bjid] ~= nil then
+		origin.send(st.presence{to=from_bjid, from=component_host, type='subscribed'});
+	elseif pres.type == 'unsubscribed' and users[from_bjid] ~= nil then
+		users[from_bjid]:logout();
+		users[from_bjid]:signout();
+		users[from_bjid] = nil;
+	elseif pres.type == 'unavailable' and users[from_bjid] ~= nil then
+		users[from_bjid]:logout();
+		users[from_bjid] = nil;
+	end
+	return true;
+end
+
+function confirm_message_delivery(event)
+	local reply = st.message({id=event.stanza.attr.id, to=event.stanza.attr.from, from=event.stanza.attr.to or component_host}):tag("received", {xmlns = "urn:xmpp:receipts"});
+	origin.send(reply);
+	return true;
+end
+
+function message_stanza_handler(event)
+	local origin, stanza = event.origin, event.stanza;
+	local to = {};
+	local from = {};
+	local msg = {};
+	to.node, to.host, to.resource = jid_split(stanza.attr.to);
+	from.node, from.host, from.resource = jid_split(stanza.attr.from);
+	local bjid = nil;
+	if from.node ~= nil and from.host ~= nil then
+		from_bjid = from.node.."@"..from.host;
+	elseif from.host ~= nil then
+		from_bjid = from.host;
+	end
+	local to_bjid = nil;
+	if to.node ~= nil and to.host ~= nil then
+		to_bjid = to.node.."@"..to.host;
+	elseif to.host ~= nil then
+		to_bjid = to.host;
+	end
+	for _, tag in ipairs(stanza.tags) do
+		msg[tag.name] = tag[1];
+		if tag.attr.xmlns == "urn:xmpp:receipts" then
+			confirm_message_delivery({origin=origin, stanza=stanza});
+		end
+		-- can handle more xmlns
+	end
+	-- Now parse the message
+	if stanza.attr.to == component_host then
+		if msg.body == "!myinfo" then
+			if users[from_bjid] ~= nil then
+				origin.send(st.message({to=stanza.attr.from, from=component_host, type='chat'}):tag("body"):text(print_r(users[from_bjid])):up());
+			end
+		end
+		-- Other messages go to twitter
+		user:twitterAction("UpdateStatus", {status=msg.body});
+	else
+		-- Message to uid@host/resource
+	end
+	return true;
+end
+
+module:hook("presence/host", presence_stanza_handler);
+module:hook("message/host", message_stanza_handler);
+
+module:hook("iq/host/jabber:iq:register:query", iq_register);
+module:hook("iq/host/http://jabber.org/protocol/disco#info:query", iq_disco_info);
+module:hook("iq/host/http://jabber.org/protocol/disco#items:query", iq_disco_items);
+module:hook("iq/host", function(data)
+	-- IQ to a local host recieved
+	local origin, stanza = data.origin, data.stanza;
+	if stanza.attr.type == "get" or stanza.attr.type == "set" then
+		return module:fire_event("iq/host/"..stanza.tags[1].attr.xmlns..":"..stanza.tags[1].name, data);
+	else
+		module:fire_event("iq/host/"..stanza.attr.id, data);
+		return true;
+	end
+end);
+
+module.unload = function()
+	componentmanager.deregister_component(component_host);
+end
+component = componentmanager.register_component(component_host, function() return; end);