isbear@23: 
isbear@23: /* Copyright 2009 Myhailo Danylenko
isbear@23: 
isbear@23: This file is part of lua-lm.
isbear@23: 
isbear@23: lua-lm is free software: you can redistribute it and/or modify
isbear@23: it under the terms of the GNU General Public License as published by
isbear@23: the Free Software Foundation, either version 2 of the License, or
isbear@23: (at your option) any later version.
isbear@23: 
isbear@23: This program is distributed in the hope that it will be useful,
isbear@23: but WITHOUT ANY WARRANTY; without even the implied warranty of
isbear@23: MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
isbear@23: GNU General Public License for more details.
isbear@23: 
isbear@23: You should have received a copy of the GNU General Public License
isbear@23: along with this program.  If not, see <http://www.gnu.org/licenses/>. */
isbear@0: 
isbear@0: #include <lua.h>
isbear@0: #include <lauxlib.h>
isbear@0: #include <glib.h>			// GDestroyNotify, GMainContext
isbear@0: #include <loudmouth/loudmouth.h>
isbear@0: 
isbear@6: #include "config.h"
isbear@0: #include "util.h"
isbear@0: #include "lm_types.h"
isbear@0: #include "lm_message.h"
isbear@0: #include "lm_message_handler.h"
isbear@0: 
isbear@0: /// lm.connection
isbear@0: /// Central module, representing connection to the server.
isbear@0: /// You should create a new connection object, then open it (establish
isbear@0: /// connection), then authenticate to the server.
isbear@0: 
isbear@0: /// connection state
isbear@0: /// Stirng, representing current connection state.
isbear@4: /// G:
isbear@11: const string2enum_t state_lm_connection[] = {
isbear@0: 	{ "closed",         LM_CONNECTION_STATE_CLOSED         },
isbear@0: 	{ "opening",        LM_CONNECTION_STATE_OPENING        },
isbear@0: 	{ "open",           LM_CONNECTION_STATE_OPEN           },
isbear@0: 	{ "authenticating", LM_CONNECTION_STATE_AUTHENTICATING },
isbear@0: 	{ "authenticated",  LM_CONNECTION_STATE_AUTHENTICATED  },
isbear@0: 	{ NULL,             0                                  }, // XXX
isbear@0: };
isbear@0: 
isbear@0: /// handler priority
isbear@0: /// String, according to which handler will be placed into one
isbear@0: /// of three handler groups.
isbear@4: /// G:
isbear@11: const string2enum_t priority_lm_handler[] = {
isbear@0: 	{ "last",   LM_HANDLER_PRIORITY_LAST   },
isbear@0: 	{ "normal", LM_HANDLER_PRIORITY_NORMAL },
isbear@0: 	{ "first",  LM_HANDLER_PRIORITY_FIRST  },
isbear@0: 	{ NULL,     0                          }, // XXX
isbear@0: };
isbear@0: 
isbear@0: /// disconnect reason
isbear@0: /// String, indicating the reason of disconnection occured.
isbear@4: /// G:
isbear@4: static const string2enum_t llm_disconnect_reason[] = {
isbear@0: 	{ "ok",                LM_DISCONNECT_REASON_OK                },
isbear@0: 	{ "ping time out",     LM_DISCONNECT_REASON_PING_TIME_OUT     },
isbear@0: 	{ "hup",               LM_DISCONNECT_REASON_HUP               },
isbear@0: 	{ "error",             LM_DISCONNECT_REASON_ERROR             },
isbear@0: 	{ "resource conflict", LM_DISCONNECT_REASON_RESOURCE_CONFLICT },
isbear@0: 	{ "invalid xml",       LM_DISCONNECT_REASON_INVALID_XML       },
isbear@0: 	{ "unknown",           LM_DISCONNECT_REASON_UNKNOWN           },
isbear@0: 	{ "ping_time_out",     LM_DISCONNECT_REASON_PING_TIME_OUT     },
isbear@0: 	{ "resource_conflict", LM_DISCONNECT_REASON_RESOURCE_CONFLICT },
isbear@0: 	{ "invalid_xml",       LM_DISCONNECT_REASON_INVALID_XML       },
isbear@0: 	{ NULL,                0                                      }, // XXX
isbear@0: };
isbear@0: 
isbear@0: /// lm.connection.new
isbear@0: /// Creates a new connection (closed).
isbear@0: /// A: string (server name), lightuserdata (C glib main context object, optional)
isbear@0: /// R: lm connection object
isbear@11: static int new_lm_connection (lua_State *L)
isbear@0: {
isbear@0: 	const char *server = luaL_checkstring (L, 1);
isbear@0: 	LmConnection *connection;
isbear@6: 	if (lua_gettop (L) < 2)
isbear@0: 		connection = lm_connection_new (server);
isbear@6: 	else {
isbear@0: 		luaL_argcheck (L, lua_islightuserdata (L, 2), 2, "glib main context lightuserdata expected");
isbear@0: 		connection = lm_connection_new_with_context (server, (GMainContext *) lua_touserdata (L, 2));
isbear@0: 	}
isbear@11: 	bless_lm_connection (L, connection);
isbear@0: 	lm_connection_unref (connection);
isbear@38: 	D ("Connection %p created", connection);
isbear@0: 	return 1;
isbear@0: }
isbear@0: 
isbear@0: /// lm.connection.bless
isbear@0: /// Blesses given pointer to lm connection object.
isbear@0: /// Note: it adds a reference to connection.
isbear@0: /// A: lightuserdata (C lm connection object)
isbear@0: /// R: lm connection object
isbear@11: static int bless_lua_lm_connection (lua_State *L)
isbear@0: {
isbear@0: 	luaL_argcheck (L, lua_islightuserdata (L, 1), 1, "loudmouth connection lightuserdata expected");
isbear@11: 	bless_lm_connection (L, (LmConnection *) lua_touserdata (L, 1));
isbear@0: 	return 1;
isbear@0: }
isbear@0: 
isbear@0: /// connection callback function
isbear@0: /// User function, that will be called on connection establishment operation end,
isbear@0: /// eg. successful/unsuccessful opening or authentication.
isbear@0: /// A: lm connection object, boolean (success)
isbear@11: static void callback_lm_connection (LmConnection *connection, int success, llm_callback_t *cb)
isbear@0: {
isbear@0: 	lua_rawgeti (cb->L, LUA_REGISTRYINDEX, cb->reference);
isbear@11: 	bless_lm_connection (cb->L, connection);
isbear@0: 	// XXX lm_connection_unref (connection);
isbear@0: 	lua_pushboolean (cb->L, success);
isbear@0: 	if (lua_pcall (cb->L, 2, 0, 0)) {
isbear@9: 		W ("Connection callback error: %s", lua_tostring (cb->L, -1));
isbear@0: 		lua_pop (cb->L, 1);
isbear@0: 	}
isbear@0: }
isbear@0: 
isbear@0: /// connection:open
isbear@0: /// Opens connection to the server and then calls callback function.
isbear@0: /// A: connection callback function
isbear@6: /// R: lm connection object or nil, string (error message)
isbear@11: static int open_lm_connection (lua_State *L)
isbear@0: {
isbear@0: 	llm_connection_t *object = luaL_checklm_connection (L, 1);
isbear@0: 	llm_callback_t *cb;
isbear@30: 	GError *err = NULL;
isbear@0: 	luaL_argcheck (L, lua_isfunction (L, 2), 2, "function expected");
isbear@0: 	
isbear@0: 	cb = luaL_malloc (L, sizeof (llm_callback_t));
isbear@0: 	cb->reference = luaL_ref (L, LUA_REGISTRYINDEX);
isbear@0: 	cb->L         = L;
isbear@0: 
isbear@11: 	if (lm_connection_open (object->connection, (LmResultFunction) callback_lm_connection,
isbear@6: 				 cb, (GDestroyNotify) llm_callback_destroy, &err))
isbear@6: 		return 1;
isbear@6: 	else {
isbear@6: 		lua_pushnil (L);
isbear@6: 		lua_pushstring (L, err->message);
isbear@30: 		I ("Connection opening failed: %s", err -> message);
isbear@30: 		g_error_free (err);
isbear@6: 		return 2;
isbear@6: 	}
isbear@0: }
isbear@0: 
isbear@0: /// connection:authenticate
isbear@0: /// Tries to authenticate against opened connection, then calls callback function.
isbear@0: /// A: string (username), string (password), string (resource), connection callback function
isbear@6: /// R: lm connection object or nil, string (error message)
isbear@11: static int authenticate_lm_connection (lua_State *L)
isbear@0: {
isbear@45: 	llm_connection_t *object   = luaL_checklm_connection (L, 1);
isbear@45: 	const char       *username = luaL_checkstring (L, 2);
isbear@45: 	const char       *password = luaL_checkstring (L, 3);
isbear@45: 	const char       *resource = luaL_checkstring (L, 4);
isbear@45: 	GError           *err      = NULL;
isbear@45: 	llm_callback_t   *cb;
isbear@0: 	luaL_argcheck (L, lua_isfunction (L, 5), 5, "function expected");
isbear@0: 
isbear@0: 	cb = luaL_malloc (L, sizeof (llm_callback_t));
isbear@0: 	cb->reference = luaL_ref (L, LUA_REGISTRYINDEX);
isbear@0: 	cb->L         = L;
isbear@0: 
isbear@6: 	if (lm_connection_authenticate (object->connection, username, password, resource,
isbear@11: 						(LmResultFunction) callback_lm_connection, cb,
isbear@6: 						(GDestroyNotify) llm_callback_destroy, &err)) {
isbear@6: 		lua_pop (L, 3);
isbear@6: 		return 1;
isbear@6: 	} else {
isbear@6: 		lua_pushnil (L);
isbear@6: 		lua_pushstring (L, err->message);
isbear@9: 		I ("Connection authentication failed: %s", err->message);
isbear@30: 		g_error_free (err);
isbear@6: 		return 2;
isbear@6: 	}
isbear@0: }
isbear@0: 
isbear@0: /// connection:port
isbear@0: /// Gets or sets server port to connect.
isbear@0: /// A: integer (optional)
isbear@6: /// R: integer (when called with no args) or lm connection object
isbear@11: static int port_lm_connection (lua_State *L)
isbear@0: {
isbear@0: 	llm_connection_t *object = luaL_checklm_connection (L, 1);
isbear@0: 	if (lua_gettop (L) > 1) { // Set
isbear@63: 		lm_connection_set_port (object->connection, luaL_checkinteger (L, 2));
isbear@6: 		lua_pop (L, 1);
isbear@0: 	} else { // Get
isbear@0: 		lua_pushnumber (L, lm_connection_get_port (object->connection));
isbear@0: 	}
isbear@6: 	return 1;
isbear@0: }
isbear@0: 
isbear@0: /// connection:server
isbear@0: /// Gets or sets server to connect to.
isbear@0: /// A: string (optional, server name)
isbear@6: /// R: string (when called with no args) or lm connection object
isbear@11: static int server_lm_connection (lua_State *L)
isbear@0: {
isbear@0: 	llm_connection_t *object = luaL_checklm_connection (L, 1);
isbear@0: 	if (lua_gettop (L) > 1) { // Set
isbear@0: 		lm_connection_set_server (object->connection, luaL_checkstring (L, 2));
isbear@6: 		lua_pop (L, 1);
isbear@0: 	} else { // Get
isbear@0: 		lua_pushstring (L, lm_connection_get_server (object->connection));
isbear@0: 	}
isbear@6: 	return 1;
isbear@0: }
isbear@0: 
isbear@0: /// connection:jid
isbear@0: /// Gets or sets jid for connection.
isbear@0: /// A: string (optional)
isbear@6: /// R: string (when called with no args) or lm connection object
isbear@11: static int jid_lm_connection (lua_State *L)
isbear@0: {
isbear@0: 	llm_connection_t *object = luaL_checklm_connection (L, 1);
isbear@0: 	if (lua_gettop (L) > 1) { // Set
isbear@0: 		lm_connection_set_jid (object->connection, luaL_checkstring (L, 2));
isbear@6: 		lua_pop (L, 1);
isbear@0: 	} else { // Get
isbear@0: 		lua_pushstring (L, lm_connection_get_jid (object->connection));
isbear@0: 	}
isbear@6: 	return 1;
isbear@0: }
isbear@0: 
isbear@0: /// connection:keep_alive_rate
isbear@0: /// Gets or sets keep alive packets rate for connection.
isbear@0: /// Note, that on some platforms there is no get function even in
isbear@0: /// loudmouth versions, that should have it according to documentation.
isbear@0: /// integer (optional, seconds)
isbear@6: /// integer (when called with no args) or lm connection object or nil, string (error message, when get function is not available in loudmouth)
isbear@11: static int keep_alive_rate_lm_connection (lua_State *L)
isbear@0: {
isbear@0: 	llm_connection_t *object = luaL_checklm_connection (L, 1);
isbear@0: 	if (lua_gettop (L) > 1) { // Set
isbear@63: 		lm_connection_set_keep_alive_rate (object->connection, luaL_checkinteger (L, 2));
isbear@6: 		lua_pop (L, 1);
isbear@6: 		return 1;
isbear@0: 	} else { // Get
isbear@0: #ifdef HAVE_LM_CONNECTION_GET_KEEP_ALIVE_RATE
isbear@0: 		lua_pushnumber (L, lm_connection_get_keep_alive_rate (object->connection));
isbear@6: 		return 1;
isbear@0: #else
isbear@9: 		W ("Sorry, your loudmouth have no get_keep_alive_rate ()");
isbear@6: 		lua_pushnil (L);
isbear@19: 		lua_pushliteral (L, "Sorry, your loudmouth have no get_keep_alive_rate ()");
isbear@6: 		return 2;
isbear@0: #endif
isbear@0: 	}
isbear@0: }
isbear@0: 
isbear@0: /// connection:proxy
isbear@0: /// Gets or sets proxy server for connection.
isbear@0: /// A: lm proxy object (optional)
isbear@6: /// R: lm proxy object or nil (when called with no args) or lm connection object
isbear@11: static int proxy_lm_connection (lua_State *L)
isbear@0: {
isbear@0: 	llm_connection_t *object = luaL_checklm_connection (L, 1);
isbear@0: 	if (lua_gettop (L) > 1) { // Set
isbear@0: 		llm_proxy_t *proxy = luaL_checklm_proxy (L, 2);
isbear@0: 		lm_connection_set_proxy (object->connection, proxy->proxy);
isbear@6: 		lua_pop (L, 1);
isbear@0: 	} else { // Get
isbear@0: 		LmProxy *proxy = lm_connection_get_proxy (object->connection);
isbear@0: 		lua_pop (L, 1);
isbear@0: 		if (proxy) {
isbear@11: 			bless_lm_proxy (L, proxy);
isbear@0: 			// XXX lm_proxy_unref (proxy);
isbear@0: 		} else
isbear@0: 			lua_pushnil (L);
isbear@0: 	}
isbear@6: 	return 1;
isbear@0: }
isbear@0: 
isbear@0: /// connection:ssl
isbear@0: /// Gets or sets ssl object for connection.
isbear@0: /// A: lm ssl object (optional)
isbear@6: /// R: lm ssl object or nil (when called with no args) or lm connection object
isbear@11: static int ssl_lm_connection (lua_State *L)
isbear@0: {
isbear@0: 	llm_connection_t *object = luaL_checklm_connection (L, 1);
isbear@0: 	if (lua_gettop (L) > 1) { // Set
isbear@0: 		llm_ssl_t *ssl = luaL_checklm_ssl (L, 2);
isbear@0: 		lm_connection_set_ssl (object->connection, ssl->ssl);
isbear@6: 		lua_pop (L, 1);
isbear@0: 	} else { // Get
isbear@0: 		LmSSL *ssl = lm_connection_get_ssl (object->connection);
isbear@0: 		lua_pop (L, 1);
isbear@0: 		if (ssl) {
isbear@11: 			bless_lm_ssl (L, ssl);
isbear@0: 			// XXX lm_ssl_unref (ssl);
isbear@0: 		} else
isbear@0: 			lua_pushnil (L);
isbear@0: 	}
isbear@6: 	return 1;
isbear@0: }
isbear@0: 
isbear@0: /// connection:close
isbear@0: /// Close connection.
isbear@6: /// R: lm connection object or nil, string (error message)
isbear@11: static int close_lm_connection (lua_State *L)
isbear@0: {
isbear@0: 	llm_connection_t *object = luaL_checklm_connection (L, 1);
isbear@6: 	GError *err = NULL;
isbear@6: 	if (lm_connection_close (object->connection, &err))
isbear@6: 		return 1;
isbear@6: 	else {
isbear@9: 		I ("Connection close failed: %s", err->message);
isbear@6: 		lua_pushnil (L);
isbear@6: 		lua_pushstring (L, err->message);
isbear@30: 		g_error_free (err);
isbear@6: 		return 2;
isbear@6: 	}
isbear@0: }
isbear@0: 
isbear@0: /// connection:status
isbear@0: /// Returns string, describing connection state.
isbear@0: /// R: connection state
isbear@11: static int status_lm_connection (lua_State *L)
isbear@0: {
isbear@0: 	llm_connection_t *connection = luaL_checklm_connection (L, 1);
isbear@11: 	luaL_pushenum (L, lm_connection_get_state (connection->connection), state_lm_connection);
isbear@0: 	return 1;
isbear@0: }
isbear@0: 
isbear@0: /// connection:send
isbear@10: /// Sends message. If specified, handler function will be called upon receiving of response to this message.
isbear@10: /// Handler function can be either a message handler object or just a message handler function.
isbear@10: /// Message can be raw xml string. However, you cannot use it with handler function.
isbear@10: /// In short:
isbear@10: /// * connection, errmsg = connection:send ( "raw xml" )
isbear@10: /// * connection, errmsg = connection:send ( message )
isbear@10: /// * connection, errmsg = connection:send ( message, function ( connection, message ) end )
isbear@10: /// * connection, errmsg = connection:send ( message, handler )
isbear@10: /// If connection is nil, errmsg contains error message.
isbear@10: /// A: lm message object/string, message handler callback function/lm message handler object (optional)
isbear@6: /// R: lm connection object or nil, string (error message)
isbear@11: static int send_lm_connection (lua_State *L)
isbear@0: {
isbear@0: 	llm_connection_t *object = luaL_checklm_connection (L, 1);
isbear@0: 	int status;
isbear@6: 	GError *err = NULL;
isbear@10: 	if (lua_gettop (L) < 3) { // Send
isbear@10: 		if (lua_type (L, 2) == LUA_TSTRING)
isbear@10: 			status = lm_connection_send_raw (object->connection, lua_tostring (L, 2), &err);
isbear@10: 		else {
isbear@10: 			llm_message_t *message = luaL_checklm_message (L, 2);
isbear@10: 			status = lm_connection_send (object->connection, message->message, &err);
isbear@10: 		}
isbear@10: 	} else if (lua_isfunction (L, 3)) { // Send w/reply, func
isbear@10: 		llm_message_t    *message = luaL_checklm_message (L, 2);
isbear@10: 		llm_callback_t   *cb      = luaL_malloc (L, sizeof (llm_callback_t));
isbear@0: 		LmMessageHandler *handler;
isbear@0: 
isbear@0: 		cb->reference = luaL_ref (L, LUA_REGISTRYINDEX);
isbear@10: 		cb->L         = L;
isbear@0: 		handler = lm_message_handler_new (
isbear@11: 					(LmHandleMessageFunction)callback_lm_handler,
isbear@0: 					cb, (GDestroyNotify)llm_callback_destroy);
isbear@0: 		status = lm_connection_send_with_reply (object->connection,
isbear@6: 							message->message, handler, &err);
isbear@0: 		lm_message_handler_unref (handler);
isbear@0: 	} else { // Send w/reply, handler
isbear@10: 		llm_message_t *message = luaL_checklm_message (L, 2);
isbear@11: 		llm_handler_t *handler = luaL_checklm_handler (L, 3);
isbear@0: 		status = lm_connection_send_with_reply (object->connection, message->message,
isbear@11: 							handler->handler, &err);
isbear@0: 		lua_pop (L, 1);
isbear@0: 	};
isbear@6: 	lua_pop (L, 1);
isbear@6: 	if (status)
isbear@6: 		return 1;
isbear@6: 	else {
isbear@9: 		I ("Message sending failed: %s", err->message);
isbear@6: 		lua_pushnil (L);
isbear@6: 		lua_pushstring (L, err->message);
isbear@30: 		g_error_free (err);
isbear@6: 		return 2;
isbear@6: 	}
isbear@0: }
isbear@0: 
isbear@0: /// connection:handler
isbear@0: /// Registers or unregisters handler function for a given type of messages.
isbear@0: /// To unregister handler, omit the priority argument.
isbear@30: /// To unregister reply handler, omit message type argument too.
isbear@0: /// Handler function can be specified as plain function or message handler object.
isbear@30: /// A: message handler callback function or lm message handler object, message type (optional), handler priority (optional)
isbear@6: /// R: lm connection object
isbear@11: static int handler_lm_connection (lua_State *L)
isbear@0: {
isbear@0: 	llm_connection_t *object = luaL_checklm_connection (L, 1);
isbear@30: 	int top  = lua_gettop (L);
isbear@30: 	if (top > 2) { // Have message type
isbear@30: 		int type = luaL_checkenum (L, 3, type_lm_message);
isbear@0: 
isbear@30: 		if (lua_gettop (L) > 3) { // Register
isbear@30: 			int priority = luaL_checkenum (L, 4, priority_lm_handler);
isbear@0: 
isbear@30: 			if (lua_isfunction (L, 2)) { // Function
isbear@30: 				LmMessageHandler *handler;
isbear@30: 				llm_callback_t *cb = luaL_malloc (L, sizeof (llm_callback_t));
isbear@30: 				lua_pushvalue (L, 2);
isbear@30: 				cb->reference = luaL_ref (L, LUA_REGISTRYINDEX);
isbear@30: 				cb->L         = L;
isbear@0: 
isbear@30: 				handler = lm_message_handler_new (
isbear@30: 						(LmHandleMessageFunction)callback_lm_handler,
isbear@30: 						cb, (GDestroyNotify)llm_callback_destroy);
isbear@30: 				lm_connection_register_message_handler (object->connection,
isbear@30: 									handler, type, priority);
isbear@30: 				lm_message_handler_unref (handler);
isbear@30: 			} else { // Object
isbear@30: 				llm_handler_t *handler = luaL_checklm_handler (L, 2);
isbear@30: 				lm_connection_register_message_handler (object->connection,
isbear@30: 									handler->handler,
isbear@30: 									type, priority);
isbear@30: 			}
isbear@30: 			lua_pop (L, 1);
isbear@30: 		} else { // Unregister message handler
isbear@11: 			llm_handler_t *handler = luaL_checklm_handler (L, 2);
isbear@30: 			lm_connection_unregister_message_handler (object->connection,
isbear@30: 								  handler->handler,
isbear@30: 								  type);
isbear@0: 		}
isbear@0: 		lua_pop (L, 1);
isbear@30: 	} else { // Unregister reply handler
isbear@31: #ifdef HAVE_LM_CONNECTION_UNREGISTER_REPLY_HANDLER
isbear@11: 		llm_handler_t *handler = luaL_checklm_handler (L, 2);
isbear@30: 		lm_connection_unregister_reply_handler (object -> connection,
isbear@30: 		                                        handler -> handler);
isbear@30: #else
isbear@31: 		W ("Sorry, your loudmouth have no unregister_reply_handler ()");
isbear@30: 		lua_pushnil (L);
isbear@31: 		lua_pushliteral (L, "Sorry, your loudmouth have no unregister_reply_handler ()");
isbear@30: 		return 2;
isbear@30: #endif
isbear@0: 	}
isbear@30: 	lua_pop (L, 1);
isbear@6: 	return 1;
isbear@0: }
isbear@0: 
isbear@0: /// disconnect callback function
isbear@0: /// Function, that will be called when disconnection occurs.
isbear@0: /// A: lm connection object, disconnect reason
isbear@0: void llm_disconnect_callback (LmConnection *connection, LmDisconnectReason reason, llm_callback_t *cb)
isbear@0: {
isbear@0: 	lua_rawgeti (cb->L, LUA_REGISTRYINDEX, cb->reference);
isbear@11: 	bless_lm_connection (cb->L, connection);
isbear@0: 	// XXX lm_connection_unref (connection);
isbear@0: 	luaL_pushenum (cb->L, reason, llm_disconnect_reason);
isbear@0: 	if (lua_pcall (cb->L, 2, 0, 0)) {
isbear@9: 		W ("Disconnect callback error: %s", lua_tostring (cb->L, -1));
isbear@0: 		lua_pop (cb->L, 1);
isbear@0: 	}
isbear@0: }
isbear@0: 
isbear@0: /// connection:ondisconnect
isbear@0: /// Sets callback, that will be called on connection disconnect.
isbear@0: /// A: disconnect callback function
isbear@6: /// R: lm connection object
isbear@11: static int ondisconnect_lm_connection (lua_State *L)
isbear@0: {
isbear@0: 	llm_connection_t *object = luaL_checklm_connection (L, 1);
isbear@0: 	llm_callback_t *cb;
isbear@0: 	luaL_argcheck (L, lua_isfunction (L, 2), 2, "function expected");
isbear@0: 
isbear@0: 	cb = luaL_malloc (L, sizeof (llm_callback_t));
isbear@0: 	cb->reference  = luaL_ref (L, LUA_REGISTRYINDEX);
isbear@0: 	cb->L          = L;
isbear@0: 
isbear@0: 	lm_connection_set_disconnect_function (object->connection,
isbear@0: 					       (LmDisconnectFunction)llm_disconnect_callback,
isbear@0: 					       cb, (GDestroyNotify)llm_callback_destroy);
isbear@6: 	return 1;
isbear@0: }
isbear@0: 
isbear@0: /// connection:open_wait
isbear@0: /// Synchronous open call, that will block until connection will be opened.
isbear@6: /// R: lm connection object or nil, string (error message)
isbear@11: static int open_wait_lm_connection (lua_State *L)
isbear@0: {
isbear@0: 	llm_connection_t *object = luaL_checklm_connection (L, 1);
isbear@6: 	GError *err = NULL;
isbear@6: 	if (lm_connection_open_and_block (object->connection, &err))
isbear@6: 		return 1;
isbear@6: 	else {
isbear@9: 		I ("Synchronous connection opening failed: %s", err->message);
isbear@6: 		lua_pushnil (L);
isbear@6: 		lua_pushstring (L, err->message);
isbear@30: 		g_error_free (err);
isbear@6: 		return 2;
isbear@6: 	}
isbear@0: }
isbear@0: 
isbear@0: /// connection:authenticate_wait
isbear@0: /// Synchronous authentication call, that will wait until the end of authentication.
isbear@0: /// A: string (username), string (password), string (resource)
isbear@6: /// R: lm connection object or nil, string (error message)
isbear@11: static int authenticate_wait_lm_connection (lua_State *L)
isbear@0: {
isbear@0: 	llm_connection_t *object = luaL_checklm_connection (L, 1);
isbear@0: 	const char *username = luaL_checkstring (L, 2);
isbear@0: 	const char *password = luaL_checkstring (L, 3);
isbear@0: 	const char *resource = luaL_checkstring (L, 4);
isbear@6: 	GError *err = NULL;
isbear@6: 	if (lm_connection_authenticate_and_block (object->connection, username,
isbear@6: 								password, resource, &err))
isbear@6: 		return 1;
isbear@6: 	else {
isbear@9: 		I ("Synchronous authentication failed: %s", err->message);
isbear@6: 		lua_pushnil (L);
isbear@6: 		lua_pushstring (L, err->message);
isbear@30: 		g_error_free (err);
isbear@6: 		return 2;
isbear@6: 	}
isbear@0: }
isbear@0: 
isbear@0: /// connection:send_wait
isbear@0: /// Synchronous call, that will send message and wait for reply to it.
isbear@0: /// A: lm message object
isbear@6: /// R: lm message object or nil, string (error message)
isbear@11: static int send_wait_lm_connection (lua_State *L)
isbear@0: {
isbear@0: 	llm_connection_t *object = luaL_checklm_connection (L, 1);
isbear@0: 	llm_message_t *message = luaL_checklm_message (L, 2);
isbear@6: 	GError *err = NULL;
isbear@0: 	LmMessage *new = lm_connection_send_with_reply_and_block (object->connection,
isbear@6: 									message->message, &err);
isbear@6: 	if (!new) {
isbear@9: 		I ("Synchronous message sending failed: %s", err->message);
isbear@0: 		lua_pushnil (L);
isbear@6: 		lua_pushstring (L, err->message);
isbear@30: 		g_error_free (err);
isbear@6: 		return 2;
isbear@6: 	} else {
isbear@11: 		bless_lm_message (L, new);
isbear@0: 		lm_message_unref (new); // XXX
isbear@6: 		return 1;
isbear@0: 	}
isbear@0: }
isbear@0: 
isbear@0: /// connection:pointer
isbear@0: /// Returns pointer to underlying C loudmouth structure.
isbear@0: /// R: lightuserdata
isbear@11: static int pointer_lm_connection (lua_State *L)
isbear@0: {
isbear@0: 	llm_connection_t *object = luaL_checklm_connection (L, 1);
isbear@0: 	lua_pushlightuserdata (L, object->connection);
isbear@0: 	lua_remove (L, -2);
isbear@0: 	return 1;
isbear@0: }
isbear@0: 
isbear@11: static int gc_lm_connection (lua_State *L)
isbear@0: {
isbear@0: 	llm_connection_t *object = luaL_checklm_connection (L, 1);
isbear@38: 	D ("Connection %p gc called", object);
isbear@0: 	lm_connection_unref (object->connection);
isbear@0: 	return 0;
isbear@0: }
isbear@0: 
isbear@11: static const luaL_Reg reg_f_lm_connection[] = {
isbear@11: 	{ "new",   new_lm_connection       },
isbear@11: 	{ "bless", bless_lua_lm_connection },
isbear@12: 	{ NULL,    NULL                    },
isbear@0: };
isbear@0: 
isbear@11: static const luaL_Reg reg_m_lm_connection[] = {
isbear@12: 	{ "open",              open_lm_connection              },
isbear@12: 	{ "close",             close_lm_connection             },
isbear@12: 	{ "authenticate",      authenticate_lm_connection      },
isbear@13: 	{ "status",            status_lm_connection            },
isbear@12: 	{ "port",              port_lm_connection              },
isbear@12: 	{ "server",            server_lm_connection            },
isbear@12: 	{ "jid",               jid_lm_connection               },
isbear@12: 	{ "keep_alive_rate",   keep_alive_rate_lm_connection   },
isbear@12: 	{ "state",             status_lm_connection            },
isbear@12: 	{ "proxy",             proxy_lm_connection             },
isbear@12: 	{ "ssl",               ssl_lm_connection               },
isbear@12: 	{ "send",              send_lm_connection              },
isbear@12: 	{ "handler",           handler_lm_connection           },
isbear@12: 	{ "ondisconnect",      ondisconnect_lm_connection      },
isbear@12: 	{ "open_wait",         open_wait_lm_connection         },
isbear@12: 	{ "authenticate_wait", authenticate_wait_lm_connection },
isbear@12: 	{ "send_wait",         send_wait_lm_connection         },
isbear@12: 	{ "pointer",           pointer_lm_connection           },
isbear@12: 	{ "__gc",              gc_lm_connection                },
isbear@12: 	{ NULL,                NULL                            },
isbear@0: };
isbear@0: 
isbear@0: int luaopen_lm_connection (lua_State *L)
isbear@0: {
isbear@0: 	luaL_newmetatable (L, "loudmouth.connection");
isbear@19: 	lua_pushvalue (L, -1);
isbear@19: 	lua_setfield (L, -2, "__index");
isbear@54: 	luaL_setfuncs (L, reg_m_lm_connection, 0);
isbear@0: 	lua_pop (L, 1);
isbear@16: 	lua_newtable (L); // XXX we can specify here exact amount of fields
isbear@54: 	luaL_setfuncs (L, reg_f_lm_connection, 0);
isbear@0: 	return 1;
isbear@0: }
isbear@0: