diff -r 000000000000 -r 7d84f4403d67 init.lua
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/init.lua	Mon Dec 21 19:27:08 2009 +0000
@@ -0,0 +1,144 @@
+local verse = require "verse";
+local st = require "util.stanza";
+
+module("riddim", package.seeall);
+plugins = {};
+
+local riddim_mt = {};
+riddim_mt.__index = riddim_mt;
+
+function new(stream, config)
+	if not stream then
+		error("riddim.new(): Verse stream required as first parameter", 2);
+	end
+	return setmetatable({ stream = stream, config = config or {} }, riddim_mt);
+end
+
+-- self.conn is ready for stanzas
+function riddim_mt:start()
+	self:add_plugin("groupchat");
+	self:add_plugin("commands");
+	self:add_plugin("ping");
+	self:event("started");
+	self.stream:hook("stanza", function (stanza)
+			local body = stanza:get_child("body");
+			local event = {
+				sender = { jid = stanza.attr.from };
+				body = (body and body:get_text()) or nil;
+				stanza = stanza;
+			};
+			if stanza.name == "message" then
+				local replied;
+				local bot = self;
+				function event:reply(reply)
+					if replied then return false; end
+					replied = true;
+					return bot:send_message(stanza.attr.from, reply);
+				end
+			end
+			local ret;
+			if stanza.name == "iq" and (stanza.attr.type == "get" or stanza.attr.type == "set") then
+				local xmlns = stanza.tags[1] and stanza.tags[1].attr.xmlns;
+				if xmlns then
+					event.xmlns = xmlns;
+					print(event.stanza)
+					ret = self:event("iq/"..xmlns, event);
+					if not ret then
+						ret = self:event(stanza.name, event);
+					end
+				end
+			else
+				ret = self:event(stanza.name, event);
+			end
+			
+			if ret and type(ret) == "table" and ret.name then
+				self:send(ret);
+			end
+			return ret;
+		end, 1);
+end
+
+function riddim_mt:send(s)
+	return self.stream:send(tostring(s));
+end
+
+function riddim_mt:event(name, ...)
+	return self.stream:event("bot/"..name, ...);
+end
+	
+function riddim_mt:hook(name, ...)
+	return self.stream:hook("bot/"..name, ...);
+end
+
+function riddim_mt:send_message(to, text, formatted_text)
+	self:send(st.message({ to = to, type = "chat" }):tag("body"):text(text));
+end
+
+function riddim_mt:add_plugin(name)
+	require("riddim.plugins."..name);
+	return riddim.plugins[name](self);
+end
+	
+-- Built-in bot starter
+if not (... and package.loaded[...] ~= nil) then
+	require "verse.client";
+	
+	-- Config loading
+	local chunk, err = loadfile("config.lua");
+	if not chunk then
+		print("File or syntax error:", err);
+		return 1;
+	end
+
+	local config = {};
+	setfenv(chunk, setmetatable(config, {__index = _G}));
+	local ok, err = pcall(chunk);
+	if not ok then
+		print("Error while processing config:", err);
+		return 1;
+	end
+	setmetatable(config, nil);
+
+	if not config.jid then
+		io.write("Enter the bot's JID: ");
+		config.jid = io.read("*l");
+	end
+	
+	if not config.password then
+		io.write("Enter the password for "..config.jid..": ");
+		config.password = io.read("*l");
+	end
+	
+	-- Create the stream object and bot object
+	local c = verse.new();
+	local b = riddim.new(c, config);
+	
+	if config.debug then
+		c:hook("incoming-raw", print);
+	end
+	
+	for _, plugin in ipairs(config.plugins or {"ping"}) do
+		c:add_plugin(plugin);
+	end
+	
+	b:hook("started", function ()
+		b:send(verse.presence());
+		for k, v in pairs(autojoin or {}) do
+			if type(k) == "number" then
+				b:join_room(v);
+			elseif type(k) == "string" then
+				if type(v) == "string" then
+					b:join_room(k, v);
+				end
+			end
+		end
+	end);
+	
+	c:hook("binding-success", function () b:start(); end)
+	
+	c:connect_client(config.jid, config.password);
+	
+	verse.loop();
+end
+
+return _M;