bot.lua

  1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
--[[ Lua XMPP MOO Component by Matthew Wild (MattJ)					]]

BotName = "Component";
BotPass = "";

function onPresence(msg)
	success, message = pcall(handlePresence, msg);
	if not success then print("\nERROR: "..message.."\n"); end
end

function onConnect()
end

function onMessage(msg)
	success, message = pcall(handleMessage, msg);
	if not success then print("\nERROR: "..message.."\n"); end
end


function handlePresence(msg)
	local room, server, nick, id, person;
	room,server,nick = msg.to:match("^(.+)@(.+)/(.+)$");
	if not (room and nick) then print("\nERROR: Invalid room or nick!\n"); return; end -- We need to send an error
	-- print("Type: "..msg.type.." type of type: "..type(msg.type),"\n");
	if msg.type == "available" or msg.type == "" and not people[msg.fromFull] then
		print("\nRecieved Available presence to join a room\n");
		people[msg.fromFull] = { ["nick"] = nick, ["jid"] = msg.fromFull, ["presence"] = msg.type, room = rooms[room], occjid = msg.to, _parent = occupant, group = {}, _contents = { ["in"] = createcontainertable(); } }
		MoveToRoom(people[msg.fromFull], room);
		-- rooms[room].people[nick] = people[msg.fromFull];
		--for id, person in pairs(rooms[room].people) do
		--	SendPresence(msg.fromFull, person.presence, "", room.."@"..server.."/"..person.nick);
		--end
	elseif msg.type == "unavailable" then
		RemoveFromRoom(rooms[room], msg.fromFull);
		people[msg.fromFull] = nil;
	end
	print("Returning\n");
	SendPresence(msg.fromFull, msg.type, "", msg.to);
end



function handleMessage(msg)
	local action, object, room, server, person;
	person = people[msg.fromFull]
	if msg.type == MSG_ERROR then print("ERROR MESSAGE from "..msg.fromFull..": ", msg.body); return nil; end
	if not person then print("No such person as "..msg.fromFull, tostring(people[msg.fromFull]).."\n"); 
	for k,v in pairs(people) do print(string.format("people[\"%s\"]", k),", "); end return nil; end
	room, server = msg.to:match("^(.+)@(.+)$");
	if not (room and server) then print("ERROR! Malformed JID? "..msg.to.."\n"); return nil; end --TODO: Reply with error
	server = server:lower();
	if not rooms[room] then return; end -- TODO: Reply with error
	
	person:_processCommand(msg.body);

	return nil;
end

function BroadcastPresence(people, fromnick, ptype, text)
	local nick, t;
	if type(people) == "string" then if not rooms[people] then print("No room called '"..people.."'","\n"); return; else print("\nLooking up room: "..people, "\n"); people = rooms[people].people; end end
	if not people then print("No such room. \n"); return; end
	for nick, t in pairs(people) do
		if t and t.jid then 
			SendPresence(t.jid, ptype or "available", text or "", GetJIDParts(t.occjid).node.."@"..GetJIDParts(t.occjid).server.."/"..fromnick);
		else print("\nOrphaned occupant: "..nick, "\n");
		end
	end
end

function BroadcastMessage(people_, from, text, mtype, sender_sees)
	local nick, t
	for nick, t in pairs(people_) do
		print("\nMessage sent to: "..t.jid);
		if t.nick ~= from or sender_sees ~= false then 
			SendMessage(t.jid, GetJIDParts(t.occjid).node.."@"..GetJIDParts(t.occjid).server.."/"..from, text or "", mtype or "groupchat");
		end
	end
	print("\nDone\n");
end

function GetOccupantNick(jid)
	local pjid, p
	for pjid, p in pairs(people) do
		if CompareJIDs(pjid, jid) then return p.nick; end
	end
	return "Somebody";
end

function CompareJIDs(j1, j2)
	if not (j1 and j2) then return false; end
	print("j2: "..type(j2));
	j1node, j1server, j1res = j1:match("^(.+)@([%a%.]+)/?(.*)$");
	j2node, j2server, j2res = j2:match("^(.+)@([%a%.]+)/?(.*)$");
	if j1node == j2node and j1server:lower() == j2server:lower() and (j1res == j2res or j1res == "" or j2res == "") then return true; end
	return false;
end

function MoveToRoom(person, room)
	if not ( person and room) then return; end
		
	if type(room) ~= "table" then
		if not rooms[room] then rooms[room] = CreateObject(classes.obj_room); rooms[room]:_create{ room = room, server = GetJIDParts(person.occjid).server}; end
		 room = rooms[room];
	end
	
	if person.room ~= room then
		if person.room then
			-- Tell everyone in room that user is leaving
			--BroadcastPresence(person.room.people, person.occjid, "unavailable");
			for nick, t in pairs(person.room.people) do
				-- print(string.format("\nPRESENCE: to: %s, from: %s, type: %s\n", t.jid, ));
				if nick ~= person.nick then
					SendPresence(t.jid, "unavailable", "", GetJIDParts(t.occjid).node.."@"..GetJIDParts(t.occjid).server.."/"..person.nick);
				end
			end
			-- Tell user that everyone else in the room has gone (clear room roster in their client)
			for nick, t in pairs(person.room.people) do
				if person.nick ~= nick then
					SendPresence(person.jid, "unavailable", "", GetJIDParts(person.occjid).node.."@"..person.room._properties.server.."/"..t.nick);
				end
			end
			person.room.people[person.nick] = nil; -- Remove from room
		end
	end
	person.room = room;
	room.people[person.nick] = person;
	--Tell user about people in the new room
	for nick, t in pairs(person.room.people) do
		-- print(string.format("\nPRESENCE: to: %s, from: %s, type: %s\n", t.jid, ));
		if nick ~= person.nick then
			SendPresence(person.jid, "available", "", GetJIDParts(person.occjid).node.."@"..GetJIDParts(person.occjid).server.."/"..nick);
		end
	end
	--Tell people in new room that user has joined
	-- BroadcastPresence(person.room.people, person.nick, "available");
	for nick, t in pairs(person.room.people) do
		-- print(string.format("\nPRESENCE: to: %s, from: %s, type: %s\n", t.jid, ));
		if nick ~= person.nick then
			SendPresence(t.jid, "available", "", GetJIDParts(t.occjid).node.."@"..GetJIDParts(t.occjid).server.."/"..person.nick);
		end
	end
	SendPresence(person.jid, "available", "", person.occjid);
	room:_enter{ room = person.room, person = person, nick = person.nick };
end

function RemoveFromRoom(room, jid)
	if not ( room and jid) then return; end
	for id, t in pairs(room.people) do
		if CompareJIDs(t.jid, from) then room.people[id] = nil; end
	end
	-- Tell everyone in room that user is leaving
	for nick, t in pairs(room.people) do
			SendPresence(t.jid, "unavailable", "", GetJIDParts(t.occjid).node.."@"..GetJIDParts(t.occjid).server.."/"..people[jid].nick);
	end
end

function FSay(room, fromnick, text)
	BroadcastMessage(room.people, fromnick , text);
end

function PSay(room, tonick, fromnick, text)
	--print(string.format("Room JID: %s, To: %s, From: %s\n", room.jid, tonick, fromnick));
	for k,v in pairs(room.people) do print(string.format("people[\"%s\"]", k),", "); end
	print("\n");
	-- room.jid..(((fromnick and fromnick:len() > 0) and "/"..fromnick) or "")
	SendMessage(room.people[tonick].jid, room.people[tonick].occjid, text or "", (fromnick and fromnick:len() > 0 and "groupchat") or "groupchat");
end

dofile("common.lua");

rooms = rooms or { };
people = people or setmetatable({ default = {} }, { __newindex = function (t, k, v) if not rawget(v, "_parent") then rawset(v, "_parent", rawget(t, "default")); end return rawset(t, k, setmetatable(v, { __index = occupant })); end }); -- This is our inheritance code;