xmoo-hg: luacomponent.cpp

// g++ -Wall -O2 -pipe -march=i686 "%f" /usr/local/lib/libgloox.so /usr/lib/liblua5.1.so
#include </usr/local/include/gloox/gloox.h>
#include </usr/local/include/gloox/client.h>
#include </usr/local/include/gloox/clientbase.h>
#include </usr/local/include/gloox/messagehandler.h>
#include </usr/local/include/gloox/tag.h>
#include </usr/local/include/gloox/stanza.h>
#include </usr/local/include/gloox/connectionlistener.h>
#include </usr/local/include/gloox/presencehandler.h>
#include </usr/local/include/gloox/component.h>
 
extern "C" {
#include <lua.h>
#include <lualib.h>
#include <lauxlib.h>
}
 
using namespace gloox;
 
lua_State *L = NULL;
lua_State *calcL = NULL;
char BotName[26];
class Bot* theBot = NULL;
 
int lcJoinMUC(lua_State* L);
int lcSleep(lua_State* L);
int lcMemUsage(lua_State* L);
int lcSendPresence(lua_State* L);
int lcSendMessage(lua_State* L);
 
class Bot : public MessageHandler, ConnectionListener , PresenceHandler, LogHandler
{
 public:
   Bot()
   {
   	::theBot = this;
   	lua_getglobal(L, "BotPass");
	if(!lua_isstring(L, -1))
	{
		printf("Please, in the body of the script, set BotJID, BotPass, and BotName!\n");
		return;
	}
      // JID jid( lua_tostring(L, -3) );
      j = new Component ( "jabber:component:accept", "localhost", "xmoo", "secret", 5347);
 
      j->registerConnectionListener( this );
      j->registerMessageHandler( this );
      j->registerPresenceHandler( this );
      j->logInstance().registerLogHandler( LogLevelDebug, LogAreaAll, this );
      // j->setInitialPriority( 5 );
      // j->setAutoPresence( true );
      printf("Connecting...\n");
      j->connect();
   }
 
   virtual void handleMessage( Stanza *stanza, MessageSession *session )
   {
		if(stanza->findChildWithAttrib("xmlns","jabber:x:delay"))
		{
			printf("History message, ignoring.\n");
			return;
		}
		
		printf("Message.\n");
			    /* push functions and arguments */
		lua_getglobal(L, "onMessage");  /* function to be called */
		if(!lua_isfunction(L, -1))
		{
			printf("Unhandled message... please add an onMessage function to the script\n");
			return;
		}
		lua_newtable(L);
		lua_pushstring(L, stanza->body().c_str());
		lua_setfield(L,-2, "body");
		lua_pushstring(L, stanza->from().full().c_str());
		lua_setfield(L,-2, "fromFull");
		lua_pushstring(L, stanza->from().bare().c_str());
		lua_setfield(L,-2, "fromBare");
		lua_pushstring(L, stanza->to().full().c_str());
		lua_setfield(L,-2, "to");
		lua_pushnumber(L, stanza->subtype());
		lua_setfield(L,-2, "type");
	
		/* do the call (1 argument, 1 result) */
		if (lua_pcall(L, 1, 1, 0) != 0)
		{
			printf("Error running function `onMessage': %s",lua_tostring(L, -1));
			return;
		}
		if(lua_istable(L, -1))
		{
			Tag *t = new Tag( "message" );
			
			lua_getfield(L, -1, "type");
			t->addAttribute( "type", luaL_optlstring(L, -1, "chat", NULL) );
			lua_pop(L,1);
			lua_getfield(L, -1, "to");
			t->addAttribute( "to", luaL_optlstring(L, -1, stanza->from().bare().c_str(), NULL )); // 
			lua_pop(L,1);
			lua_getfield(L, -1, "from");
			t->addAttribute( "from", luaL_optlstring(L, -1, j->jid().full().c_str(), NULL ));
			lua_pop(L,1);
			lua_getfield(L, -1, "body");
			t->addChild( new Tag( "body", luaL_optlstring(L, -1, "", NULL) ) );
			lua_pop(L,1);
			j->send( t ); // Send it!
		}
	
		
	}
   
	virtual void onConnect()
	{
		// do something when the connection is established
	printf("Connecting...");
	if(L)
	{
		lua_pushinteger(L, StanzaMessageChat);
		lua_setglobal(L, "MSG_CHAT");
		lua_pushinteger(L, StanzaMessageGroupchat);
		lua_setglobal(L, "MSG_GROUPCHAT");
		lua_pushinteger(L, StanzaMessageNormal);
		lua_setglobal(L, "MSG_NORMAL");
		lua_pushinteger(L, StanzaMessageError);
		lua_setglobal(L, "MSG_ERROR");
		
		lua_pushinteger(L, StanzaPresenceError);
		lua_setglobal(L, "PRE_ERROR");
		lua_pushinteger(L, StanzaPresenceAvailable);
		lua_setglobal(L, "PRE_AVAILABLE");
		lua_pushinteger(L, StanzaPresenceUnavailable);
		lua_setglobal(L, "PRE_UNAVAILABLE");
 
		lua_register(L, "JoinMUC", lcJoinMUC);
		lua_register(L, "Sleep", lcSleep);
		lua_register(L, "MemUsage", lcMemUsage);
		lua_register(L, "SendPresence", lcSendPresence);
		lua_register(L, "SendMessage", lcSendMessage);
	}
	else
		printf("Lua not loaded?!\n");
	
	 printf("Connected.\n");
	 
	 		lua_getglobal(L, "onConnect");  /* function to be called */
		if(!lua_isfunction(L, -1))
		{
			printf("No onConnect() function found.\n");
			return;
		}
		
		/*Tag *t = new Tag( "presence" );
		t->addAttribute( "to", "test@conference.jabber.org/Betty" );
		t->addChild(new Tag( "status", "Betty!" ) );
		j->send( t ); */
	
		/* do the call (0 arguments, 0 results) */
		if (lua_pcall(L, 0, 0, 0) != 0)
			return; // error(L, "error running function `f': %s",lua_tostring(L, -1));
     
	}
	
	virtual void handlePresence( Stanza * stanza)
	{
				printf("Presence.\n");
			    /* push functions and arguments */
		lua_getglobal(L, "onPresence");  /* function to be called */
		if(!lua_isfunction(L, -1))
			return;
			
		lua_newtable(L);
		lua_pushstring(L, stanza->body().c_str());
		lua_setfield(L,-2, "body");
		lua_pushstring(L, stanza->from().full().c_str());
		lua_setfield(L,-2, "fromFull");
		lua_pushstring(L, stanza->from().bare().c_str());
		lua_setfield(L,-2, "fromBare");
		lua_pushstring(L, stanza->to().full().c_str());
		lua_setfield(L,-2, "to");
		lua_pushnumber(L, stanza->subtype());
		lua_setfield(L,-2, "ntype");
		lua_pushstring(L, stanza->findAttribute("type").c_str());
		lua_setfield(L,-2, "type");
		
		
	
		/* do the call (1 arguments, 0 results) */
		if (lua_pcall(L, 1, 0, 0) != 0)
		{
			printf("Error running function `onPresence': %s",lua_tostring(L, -1));
			return;
		}
		return;
	}
 
	virtual bool onTLSConnect( const CertInfo& info )
	{
		// decide whether you trust the certificate, examine the CertInfo structure
		// It's not like a bot is bothered with this :)
		return true; // if you trust it, otherwise return false
	}
	
	virtual void onDisconnect(gloox::ConnectionError e)
	{
		std::string msg;
		if (e == ConnStreamError)
		{
			
		}
		switch(e)
		{
			case ConnStreamError:
				msg = j->streamErrorText();
				break;
			case ConnIoError:
				msg = "An I/O error occured";
				break;
			default:
				msg = "Unknown error";
		}
		printf("Disconnected: %s\n", msg.c_str());
	}
	
	virtual void handleLog( LogLevel level, LogArea area, const std::string & message)
	{
		printf("Log: %s\n", message.c_str());
	}	
	
 public:
   Component* j;
};
 
 
int main( int argc, char* argv[] )
{
	L = lua_open();   /* opens Lua */
	if(L == NULL)
	{
		printf("Unable to initialize Lua.\n");
		exit(1);
	}
	else
	{
		
	}
	luaL_openlibs(L);
	if(luaL_dofile(L, "bot.lua"))
	{
		printf("Unable to load/run the bot file\n\t%s\n", lua_tostring(L, -1));
		exit(1);
	}
	else
		printf("File loaded successfully.\n");
	
	Bot b;
}
 
int lcJoinMUC(lua_State* L)
{
	printf("Joining MUC...\n");
	if(!lua_isstring(L,1))
	{
		printf("Ooops, not passed a string :(\n");
		return 0;
	}
	Tag *t = new Tag( "presence" );
	t->addAttribute( "to", lua_tostring(L, 1) );
	if (lua_isstring(L, 2))
		t->addAttribute( "type", lua_tostring(L, 2) );
	
	// t->addChild("status", "Betty!");
	t->addChild(new Tag( "status", "" ) );
	printf(t->xml().c_str() );
	theBot->j->send( t );
	// 
	printf("Conference (%s) joined.\n",lua_tostring(L, 1));
	return 0;
}
 
int lcLeaveMUC(lua_State* L)
{
	if(!lua_isstring(L,1))
	{
		printf("Ooops, not passed a string :(\n");
		return 0;
	}
	Tag *t = new Tag( "presence" );
	t->addAttribute( "to", lua_tostring(L, 1) );
	t->addAttribute( "type", "unavailable" );
	t->addChild(new Tag( "status", "" ) );
	theBot->j->send( t );
	printf("Conference (%s) joined.\n",lua_tostring(L, 1));
	return 0;
}
 
int lcSleep(lua_State* L)
{
	sleep((int)lua_tonumber(L, 1));
	return 0;
}	
 
int lcMemUsage(lua_State* L)
{
	lua_pushinteger(L, lua_gc(L, LUA_GCCOUNT, 0));
	return 1;
}
 
// function SendPresence(to, type, text[, from]);
int lcSendPresence(lua_State* L)
{
	if(!lua_isstring(L,1))
	{
		printf("Ooops, not passed a string :(\n");
		// lua_error(L, 
		return 0;
	}
	Tag *t = new Tag( "presence" );
	t->addAttribute( "to", lua_tostring(L, 1) );
	if (lua_isstring(L, 2))
		t->addAttribute( "type", lua_tostring(L, 2) );
	else
		t->addAttribute( "type", "available" );
	
	if (lua_isstring(L, 3))
		t->addChild(new Tag( "status", lua_tostring(L, 3) ) );	
	else
		t->addChild(new Tag( "status", "" ) );
	
	if (lua_isstring(L, 4))
		t->addAttribute( "from", lua_tostring(L, 4) );
	
	t->addChild(new Tag("x", "xmlns", "http://jabber.org/protocol/muc#user"));
			
	theBot->j->send( t );
	return 0;
}
 
int lcSendMessage(lua_State* L)
{
	if(!lua_isstring(L,1))
	{
		printf("Ooops, not passed a string :(\n");
		// lua_error(L, 
		return 0;
	}
	Tag *t = new Tag( "message" );
	t->addAttribute( "to", lua_tostring(L, 1) );
	
	if (lua_isstring(L, 2))
		t->addAttribute( "from", lua_tostring(L, 2) );
	
	if (lua_isstring(L, 3))
		t->addChild(new Tag( "body", lua_tostring(L, 3) ) );	
	else
		t->addChild(new Tag( "body", "" ) );
	
	if (lua_isstring(L, 4))
		t->addAttribute( "type", lua_tostring(L, 4) );
	else
		t->addAttribute( "type", "normal" );
	
			
	theBot->j->send( t );
	return 0;
}

1	// g++ -Wall -O2 -pipe -march=i686 "%f" /usr/local/lib/libgloox.so /usr/lib/liblua5.1.so
2	#include </usr/local/include/gloox/gloox.h>
3	#include </usr/local/include/gloox/client.h>
4	#include </usr/local/include/gloox/clientbase.h>
5	#include </usr/local/include/gloox/messagehandler.h>
6	#include </usr/local/include/gloox/tag.h>
7	#include </usr/local/include/gloox/stanza.h>
8	#include </usr/local/include/gloox/connectionlistener.h>
9	#include </usr/local/include/gloox/presencehandler.h>
10	#include </usr/local/include/gloox/component.h>
11
12	extern "C" {
13	#include <lua.h>
14	#include <lualib.h>
15	#include <lauxlib.h>
16	}
17
18	using namespace gloox;
19
20	lua_State *L = NULL;
21	lua_State *calcL = NULL;
22	char BotName[26];
23	class Bot* theBot = NULL;
24
25	int lcJoinMUC(lua_State* L);
26	int lcSleep(lua_State* L);
27	int lcMemUsage(lua_State* L);
28	int lcSendPresence(lua_State* L);
29	int lcSendMessage(lua_State* L);
30
31	class Bot : public MessageHandler, ConnectionListener , PresenceHandler, LogHandler
32	{
33	public:
34	Bot()
35	{
36	::theBot = this;
37	lua_getglobal(L, "BotPass");
38	if(!lua_isstring(L, -1))
39	{
40	printf("Please, in the body of the script, set BotJID, BotPass, and BotName!\n");
41	return;
42	}
43	// JID jid( lua_tostring(L, -3) );
44	j = new Component ( "jabber:component:accept", "localhost", "xmoo", "secret", 5347);
45
46	j->registerConnectionListener( this );
47	j->registerMessageHandler( this );
48	j->registerPresenceHandler( this );
49	j->logInstance().registerLogHandler( LogLevelDebug, LogAreaAll, this );
50	// j->setInitialPriority( 5 );
51	// j->setAutoPresence( true );
52	printf("Connecting...\n");
53	j->connect();
54	}
55
56	virtual void handleMessage( Stanza stanza, MessageSession session )
57	{
58	if(stanza->findChildWithAttrib("xmlns","jabber:x:delay"))
59	{
60	printf("History message, ignoring.\n");
61	return;
62	}
63
64	printf("Message.\n");
65	/* push functions and arguments */
66	lua_getglobal(L, "onMessage"); /* function to be called */
67	if(!lua_isfunction(L, -1))
68	{
69	printf("Unhandled message... please add an onMessage function to the script\n");
70	return;
71	}
72	lua_newtable(L);
73	lua_pushstring(L, stanza->body().c_str());
74	lua_setfield(L,-2, "body");
75	lua_pushstring(L, stanza->from().full().c_str());
76	lua_setfield(L,-2, "fromFull");
77	lua_pushstring(L, stanza->from().bare().c_str());
78	lua_setfield(L,-2, "fromBare");
79	lua_pushstring(L, stanza->to().full().c_str());
80	lua_setfield(L,-2, "to");
81	lua_pushnumber(L, stanza->subtype());
82	lua_setfield(L,-2, "type");
83
84	/* do the call (1 argument, 1 result) */
85	if (lua_pcall(L, 1, 1, 0) != 0)
86	{
87	printf("Error running function `onMessage': %s",lua_tostring(L, -1));
88	return;
89	}
90	if(lua_istable(L, -1))
91	{
92	Tag *t = new Tag( "message" );
93
94	lua_getfield(L, -1, "type");
95	t->addAttribute( "type", luaL_optlstring(L, -1, "chat", NULL) );
96	lua_pop(L,1);
97	lua_getfield(L, -1, "to");
98	t->addAttribute( "to", luaL_optlstring(L, -1, stanza->from().bare().c_str(), NULL )); //
99	lua_pop(L,1);
100	lua_getfield(L, -1, "from");
101	t->addAttribute( "from", luaL_optlstring(L, -1, j->jid().full().c_str(), NULL ));
102	lua_pop(L,1);
103	lua_getfield(L, -1, "body");
104	t->addChild( new Tag( "body", luaL_optlstring(L, -1, "", NULL) ) );
105	lua_pop(L,1);
106	j->send( t ); // Send it!
107	}
108
109
110	}
111
112	virtual void onConnect()
113	{
114	// do something when the connection is established
115	printf("Connecting...");
116	if(L)
117	{
118	lua_pushinteger(L, StanzaMessageChat);
119	lua_setglobal(L, "MSG_CHAT");
120	lua_pushinteger(L, StanzaMessageGroupchat);
121	lua_setglobal(L, "MSG_GROUPCHAT");
122	lua_pushinteger(L, StanzaMessageNormal);
123	lua_setglobal(L, "MSG_NORMAL");
124	lua_pushinteger(L, StanzaMessageError);
125	lua_setglobal(L, "MSG_ERROR");
126
127	lua_pushinteger(L, StanzaPresenceError);
128	lua_setglobal(L, "PRE_ERROR");
129	lua_pushinteger(L, StanzaPresenceAvailable);
130	lua_setglobal(L, "PRE_AVAILABLE");
131	lua_pushinteger(L, StanzaPresenceUnavailable);
132	lua_setglobal(L, "PRE_UNAVAILABLE");
133
134	lua_register(L, "JoinMUC", lcJoinMUC);
135	lua_register(L, "Sleep", lcSleep);
136	lua_register(L, "MemUsage", lcMemUsage);
137	lua_register(L, "SendPresence", lcSendPresence);
138	lua_register(L, "SendMessage", lcSendMessage);
139	}
140	else
141	printf("Lua not loaded?!\n");
142
143	printf("Connected.\n");
144
145	lua_getglobal(L, "onConnect"); /* function to be called */
146	if(!lua_isfunction(L, -1))
147	{
148	printf("No onConnect() function found.\n");
149	return;
150	}
151
152	/Tag t = new Tag( "presence" );
153	t->addAttribute( "to", "test@conference.jabber.org/Betty" );
154	t->addChild(new Tag( "status", "Betty!" ) );
155	j->send( t ); */
156
157	/* do the call (0 arguments, 0 results) */
158	if (lua_pcall(L, 0, 0, 0) != 0)
159	return; // error(L, "error running function `f': %s",lua_tostring(L, -1));
160
161	}
162
163	virtual void handlePresence( Stanza * stanza)
164	{
165	printf("Presence.\n");
166	/* push functions and arguments */
167	lua_getglobal(L, "onPresence"); /* function to be called */
168	if(!lua_isfunction(L, -1))
169	return;
170
171	lua_newtable(L);
172	lua_pushstring(L, stanza->body().c_str());
173	lua_setfield(L,-2, "body");
174	lua_pushstring(L, stanza->from().full().c_str());
175	lua_setfield(L,-2, "fromFull");
176	lua_pushstring(L, stanza->from().bare().c_str());
177	lua_setfield(L,-2, "fromBare");
178	lua_pushstring(L, stanza->to().full().c_str());
179	lua_setfield(L,-2, "to");
180	lua_pushnumber(L, stanza->subtype());
181	lua_setfield(L,-2, "ntype");
182	lua_pushstring(L, stanza->findAttribute("type").c_str());
183	lua_setfield(L,-2, "type");
184
185
186
187	/* do the call (1 arguments, 0 results) */
188	if (lua_pcall(L, 1, 0, 0) != 0)
189	{
190	printf("Error running function `onPresence': %s",lua_tostring(L, -1));
191	return;
192	}
193	return;
194	}
195
196	virtual bool onTLSConnect( const CertInfo& info )
197	{
198	// decide whether you trust the certificate, examine the CertInfo structure
199	// It's not like a bot is bothered with this :)
200	return true; // if you trust it, otherwise return false
201	}
202
203	virtual void onDisconnect(gloox::ConnectionError e)
204	{
205	std::string msg;
206	if (e == ConnStreamError)
207	{
208
209	}
210	switch(e)
211	{
212	case ConnStreamError:
213	msg = j->streamErrorText();
214	break;
215	case ConnIoError:
216	msg = "An I/O error occured";
217	break;
218	default:
219	msg = "Unknown error";
220	}
221	printf("Disconnected: %s\n", msg.c_str());
222	}
223
224	virtual void handleLog( LogLevel level, LogArea area, const std::string & message)
225	{
226	printf("Log: %s\n", message.c_str());
227	}
228
229	public:
230	Component* j;
231	};
232
233
234	int main( int argc, char* argv[] )
235	{
236	L = lua_open(); /* opens Lua */
237	if(L == NULL)
238	{
239	printf("Unable to initialize Lua.\n");
240	exit(1);
241	}
242	else
243	{
244
245	}
246	luaL_openlibs(L);
247	if(luaL_dofile(L, "bot.lua"))
248	{
249	printf("Unable to load/run the bot file\n\t%s\n", lua_tostring(L, -1));
250	exit(1);
251	}
252	else
253	printf("File loaded successfully.\n");
254
255	Bot b;
256	}
257
258	int lcJoinMUC(lua_State* L)
259	{
260	printf("Joining MUC...\n");
261	if(!lua_isstring(L,1))
262	{
263	printf("Ooops, not passed a string :(\n");
264	return 0;
265	}
266	Tag *t = new Tag( "presence" );
267	t->addAttribute( "to", lua_tostring(L, 1) );
268	if (lua_isstring(L, 2))
269	t->addAttribute( "type", lua_tostring(L, 2) );
270
271	// t->addChild("status", "Betty!");
272	t->addChild(new Tag( "status", "" ) );
273	printf(t->xml().c_str() );
274	theBot->j->send( t );
275	//
276	printf("Conference (%s) joined.\n",lua_tostring(L, 1));
277	return 0;
278	}
279
280	int lcLeaveMUC(lua_State* L)
281	{
282	if(!lua_isstring(L,1))
283	{
284	printf("Ooops, not passed a string :(\n");
285	return 0;
286	}
287	Tag *t = new Tag( "presence" );
288	t->addAttribute( "to", lua_tostring(L, 1) );
289	t->addAttribute( "type", "unavailable" );
290	t->addChild(new Tag( "status", "" ) );
291	theBot->j->send( t );
292	printf("Conference (%s) joined.\n",lua_tostring(L, 1));
293	return 0;
294	}
295
296	int lcSleep(lua_State* L)
297	{
298	sleep((int)lua_tonumber(L, 1));
299	return 0;
300	}
301
302	int lcMemUsage(lua_State* L)
303	{
304	lua_pushinteger(L, lua_gc(L, LUA_GCCOUNT, 0));
305	return 1;
306	}
307
308	// function SendPresence(to, type, text[, from]);
309	int lcSendPresence(lua_State* L)
310	{
311	if(!lua_isstring(L,1))
312	{
313	printf("Ooops, not passed a string :(\n");
314	// lua_error(L,
315	return 0;
316	}
317	Tag *t = new Tag( "presence" );
318	t->addAttribute( "to", lua_tostring(L, 1) );
319	if (lua_isstring(L, 2))
320	t->addAttribute( "type", lua_tostring(L, 2) );
321	else
322	t->addAttribute( "type", "available" );
323
324	if (lua_isstring(L, 3))
325	t->addChild(new Tag( "status", lua_tostring(L, 3) ) );
326	else
327	t->addChild(new Tag( "status", "" ) );
328
329	if (lua_isstring(L, 4))
330	t->addAttribute( "from", lua_tostring(L, 4) );
331
332	t->addChild(new Tag("x", "xmlns", "http://jabber.org/protocol/muc#user"));
333
334	theBot->j->send( t );
335	return 0;
336	}
337
338	int lcSendMessage(lua_State* L)
339	{
340	if(!lua_isstring(L,1))
341	{
342	printf("Ooops, not passed a string :(\n");
343	// lua_error(L,
344	return 0;
345	}
346	Tag *t = new Tag( "message" );
347	t->addAttribute( "to", lua_tostring(L, 1) );
348
349	if (lua_isstring(L, 2))
350	t->addAttribute( "from", lua_tostring(L, 2) );
351
352	if (lua_isstring(L, 3))
353	t->addChild(new Tag( "body", lua_tostring(L, 3) ) );
354	else
355	t->addChild(new Tag( "body", "" ) );
356
357	if (lua_isstring(L, 4))
358	t->addAttribute( "type", lua_tostring(L, 4) );
359	else
360	t->addAttribute( "type", "normal" );
361
362
363	theBot->j->send( t );
364	return 0;
365	}