#include <lua.h>
#include <lauxlib.h>
#include <glib.h>			// GDestroyNotify, GMainContext
#include <loudmouth/loudmouth.h>

#include "util.h"
#include "lm_types.h"
#include "lm_message.h"
#include "lm_message_handler.h"

/// lm.connection
/// Central module, representing connection to the server.
/// You should create a new connection object, then open it (establish
/// connection), then authenticate to the server.

/// connection state
/// Stirng, representing current connection state.
/// G:
const string2enum_t llm_connection_state[] = {
	{ "closed",         LM_CONNECTION_STATE_CLOSED         },
	{ "opening",        LM_CONNECTION_STATE_OPENING        },
	{ "open",           LM_CONNECTION_STATE_OPEN           },
	{ "authenticating", LM_CONNECTION_STATE_AUTHENTICATING },
	{ "authenticated",  LM_CONNECTION_STATE_AUTHENTICATED  },
	{ NULL,             0                                  }, // XXX
};

/// handler priority
/// String, according to which handler will be placed into one
/// of three handler groups.
/// G:
const string2enum_t llm_handler_priority[] = {
	{ "last",   LM_HANDLER_PRIORITY_LAST   },
	{ "normal", LM_HANDLER_PRIORITY_NORMAL },
	{ "first",  LM_HANDLER_PRIORITY_FIRST  },
	{ NULL,     0                          }, // XXX
};

/// disconnect reason
/// String, indicating the reason of disconnection occured.
/// G:
static const string2enum_t llm_disconnect_reason[] = {
	{ "ok",                LM_DISCONNECT_REASON_OK                },
	{ "ping time out",     LM_DISCONNECT_REASON_PING_TIME_OUT     },
	{ "hup",               LM_DISCONNECT_REASON_HUP               },
	{ "error",             LM_DISCONNECT_REASON_ERROR             },
	{ "resource conflict", LM_DISCONNECT_REASON_RESOURCE_CONFLICT },
	{ "invalid xml",       LM_DISCONNECT_REASON_INVALID_XML       },
	{ "unknown",           LM_DISCONNECT_REASON_UNKNOWN           },
	{ "ping_time_out",     LM_DISCONNECT_REASON_PING_TIME_OUT     },
	{ "resource_conflict", LM_DISCONNECT_REASON_RESOURCE_CONFLICT },
	{ "invalid_xml",       LM_DISCONNECT_REASON_INVALID_XML       },
	{ NULL,                0                                      }, // XXX
};

/// lm.connection.new
/// Creates a new connection (closed).
/// A: string (server name), lightuserdata (C glib main context object, optional)
/// R: lm connection object
static int llm_connection_new (lua_State *L)
{
	const char *server = luaL_checkstring (L, 1);
	LmConnection *connection;
	if (lua_gettop (L) < 2) {
		connection = lm_connection_new (server);
		lua_pop (L, 1);
	} else {
		luaL_argcheck (L, lua_islightuserdata (L, 2), 2, "glib main context lightuserdata expected");
		connection = lm_connection_new_with_context (server, (GMainContext *) lua_touserdata (L, 2));
		lua_pop (L, 2);
	}
	llm_connection_bless (L, connection);
	lm_connection_unref (connection);
	return 1;
}

/// lm.connection.bless
/// Blesses given pointer to lm connection object.
/// Note: it adds a reference to connection.
/// A: lightuserdata (C lm connection object)
/// R: lm connection object
static int llm_connection_bless_lua (lua_State *L)
{
	luaL_argcheck (L, lua_islightuserdata (L, 1), 1, "loudmouth connection lightuserdata expected");
	llm_connection_bless (L, (LmConnection *) lua_touserdata (L, 1));
	lua_remove (L, -2);
	return 1;
}

/// connection callback function
/// User function, that will be called on connection establishment operation end,
/// eg. successful/unsuccessful opening or authentication.
/// A: lm connection object, boolean (success)
static void llm_connection_callback (LmConnection *connection, int success, llm_callback_t *cb)
{
	lua_rawgeti (cb->L, LUA_REGISTRYINDEX, cb->reference);
	llm_connection_bless (cb->L, connection);
	// XXX lm_connection_unref (connection);
	lua_pushboolean (cb->L, success);
	if (lua_pcall (cb->L, 2, 0, 0)) {
		// XXX lua_error (cb->L);
		lua_pop (cb->L, 1);
		return;
	}
}

/// connection:open
/// Opens connection to the server and then calls callback function.
/// A: connection callback function
/// R: boolean (success)
static int llm_connection_open (lua_State *L)
{
	llm_connection_t *object = luaL_checklm_connection (L, 1);
	llm_callback_t *cb;
	luaL_argcheck (L, lua_isfunction (L, 2), 2, "function expected");
	
	cb = luaL_malloc (L, sizeof (llm_callback_t));
	cb->reference = luaL_ref (L, LUA_REGISTRYINDEX);
	cb->L         = L;

	lua_pushboolean (L, lm_connection_open (object->connection,
						(LmResultFunction)llm_connection_callback, cb,
						(GDestroyNotify)llm_callback_destroy, NULL));
	lua_remove (L, -2);
	return 1;
}

/// connection:authenticate
/// Tries to authenticate against opened connection, then calls callback function.
/// A: string (username), string (password), string (resource), connection callback function
/// R: boolean (success)
static int llm_connection_authenticate (lua_State *L)
{
	llm_connection_t *object = luaL_checklm_connection (L, 1);
	const char *username = luaL_checkstring (L, 2);
	const char *password = luaL_checkstring (L, 3);
	const char *resource = luaL_checkstring (L, 4);
	llm_callback_t *cb;
	int status;
	luaL_argcheck (L, lua_isfunction (L, 5), 5, "function expected");

	cb = luaL_malloc (L, sizeof (llm_callback_t));
	cb->reference = luaL_ref (L, LUA_REGISTRYINDEX);
	cb->L         = L;

	status = lm_connection_authenticate (object->connection, username, password, resource,
						(LmResultFunction)llm_connection_callback, cb,
						(GDestroyNotify)llm_callback_destroy, NULL);
	lua_pop (L, 4);
	lua_pushboolean (L, status);
	return 1;
}

/// connection:port
/// Gets or sets server port to connect.
/// A: integer (optional)
/// R: integer (when called with no args)
static int llm_connection_port (lua_State *L)
{
	llm_connection_t *object = luaL_checklm_connection (L, 1);
	if (lua_gettop (L) > 1) { // Set
		lm_connection_set_port (object->connection, luaL_checkint (L, 2));
		lua_pop (L, 2);
		return 0;
	} else { // Get
		lua_pushnumber (L, lm_connection_get_port (object->connection));
		lua_remove (L, -2);
		return 1;
	}
}

/// connection:server
/// Gets or sets server to connect to.
/// A: string (optional, server name)
/// R: string (when called with no args)
static int llm_connection_server (lua_State *L)
{
	llm_connection_t *object = luaL_checklm_connection (L, 1);
	if (lua_gettop (L) > 1) { // Set
		lm_connection_set_server (object->connection, luaL_checkstring (L, 2));
		lua_pop (L, 2);
		return 0;
	} else { // Get
		lua_pushstring (L, lm_connection_get_server (object->connection));
		lua_remove (L, -2);
		return 1;
	}
}

/// connection:jid
/// Gets or sets jid for connection.
/// A: string (optional)
/// R: string (when called with no args)
static int llm_connection_jid (lua_State *L)
{
	llm_connection_t *object = luaL_checklm_connection (L, 1);
	if (lua_gettop (L) > 1) { // Set
		lm_connection_set_jid (object->connection, luaL_checkstring (L, 2));
		lua_pop (L, 2);
		return 0;
	} else { // Get
		lua_pushstring (L, lm_connection_get_jid (object->connection));
		lua_remove (L, -2);
		return 1;
	}
}

/// connection:keep_alive_rate
/// Gets or sets keep alive packets rate for connection.
/// Note, that on some platforms there is no get function even in
/// loudmouth versions, that should have it according to documentation.
/// integer (optional, seconds)
/// integer (when called with no args)
static int llm_connection_keep_alive_rate (lua_State *L)
{
	llm_connection_t *object = luaL_checklm_connection (L, 1);
	if (lua_gettop (L) > 1) { // Set
		lm_connection_set_keep_alive_rate (object->connection, luaL_checkint (L, 2));
		lua_pop (L, 2);
		return 0;
	} else { // Get
#ifdef HAVE_LM_CONNECTION_GET_KEEP_ALIVE_RATE
		lua_pushnumber (L, lm_connection_get_keep_alive_rate (object->connection));
		lua_remove (L, -2);
#else
		lua_pop (L, 1);
		lua_pushstring (L, "Sorry, your loudmouth have no get_keep_alive_rate");
		lua_error (L);
#endif
		return 1;
	}
}

/// connection:proxy
/// Gets or sets proxy server for connection.
/// A: lm proxy object (optional)
/// R: lm proxy object or nil (when called with no args)
static int llm_connection_proxy (lua_State *L)
{
	llm_connection_t *object = luaL_checklm_connection (L, 1);
	if (lua_gettop (L) > 1) { // Set
		llm_proxy_t *proxy = luaL_checklm_proxy (L, 2);
		lm_connection_set_proxy (object->connection, proxy->proxy);
		lua_pop (L, 2);
		return 0;
	} else { // Get
		LmProxy *proxy = lm_connection_get_proxy (object->connection);
		lua_pop (L, 1);
		if (proxy) {
			llm_proxy_bless (L, proxy);
			// XXX lm_proxy_unref (proxy);
		} else
			lua_pushnil (L);
		return 1;
	}
}

/// connection:ssl
/// Gets or sets ssl object for connection.
/// A: lm ssl object (optional)
/// R: lm ssl object or nil (when called with no args)
static int llm_connection_ssl (lua_State *L)
{
	llm_connection_t *object = luaL_checklm_connection (L, 1);
	if (lua_gettop (L) > 1) { // Set
		llm_ssl_t *ssl = luaL_checklm_ssl (L, 2);
		lm_connection_set_ssl (object->connection, ssl->ssl);
		lua_pop (L, 2);
		return 0;
	} else { // Get
		LmSSL *ssl = lm_connection_get_ssl (object->connection);
		lua_pop (L, 1);
		if (ssl) {
			llm_ssl_bless (L, ssl);
			// XXX lm_ssl_unref (ssl);
		} else
			lua_pushnil (L);
		return 1;
	}
}

/// connection:close
/// Close connection.
/// R: boolean (success)
static int llm_connection_close (lua_State *L)
{
	llm_connection_t *object = luaL_checklm_connection (L, 1);
	lua_pushboolean (L, lm_connection_close (object->connection, NULL));
	lua_remove (L, -2);
	return 1;
}

/// connection:status
/// Returns string, describing connection state.
/// R: connection state
static int llm_connection_status (lua_State *L)
{
	llm_connection_t *connection = luaL_checklm_connection (L, 1);
	luaL_pushenum (L, lm_connection_get_state (connection->connection), llm_connection_state);
	lua_remove (L, -2);
	return 1;
}

/// connection:send
/// Sends message (object). If specified, handler function will be called upon
/// receiving of response to that message.
/// Handler function can be either message handler object or just a function.
/// A: lm message object, message handler callback function or lm message handler object (optional)
/// R: boolean (success)
static int llm_connection_send (lua_State *L)
{
	llm_connection_t *object = luaL_checklm_connection (L, 1);
	llm_message_t *message = luaL_checklm_message (L, 2);
	int status;
	if (lua_gettop (L) < 3) // Send
		status = lm_connection_send (object->connection, message->message, NULL);
	else if (lua_isfunction (L, 3)) { // Send w/reply, func
		llm_callback_t *cb = luaL_malloc (L, sizeof (llm_callback_t));
		LmMessageHandler *handler;

		cb->reference = luaL_ref (L, LUA_REGISTRYINDEX);
		cb->L = L;
		handler = lm_message_handler_new (
					(LmHandleMessageFunction)llm_message_handler_callback,
					cb, (GDestroyNotify)llm_callback_destroy);
		status = lm_connection_send_with_reply (object->connection,
							message->message, handler, NULL);
		lm_message_handler_unref (handler);
	} else { // Send w/reply, handler
		llm_message_handler_t *handler = luaL_checklm_message_handler (L, 3);
		status = lm_connection_send_with_reply (object->connection, message->message,
							handler->message_handler, NULL);
		lua_pop (L, 1);
	};
	lua_pop (L, 2);
	lua_pushboolean (L, status);
	return 1;
}

/// connection:send_raw
/// Sends arbitrary string to opened connection.
/// A: string
/// R: boolean (status)
static int llm_connection_send_raw (lua_State *L)
{
	llm_connection_t *object = luaL_checklm_connection (L, 1);
	const char *string = luaL_checkstring (L, 2);
	int status = lm_connection_send_raw (object->connection, string, NULL);
	lua_pop (L, 2);
	lua_pushboolean (L, status);
	return 1;
}

/// connection:handler
/// Registers or unregisters handler function for a given type of messages.
/// To unregister handler, omit the priority argument.
/// Handler function can be specified as plain function or message handler object.
/// Though, you can unregister only a message handler object.
/// A: message handler callback function or lm message handler object, message type, handler priority (optional)
static int llm_connection_handler (lua_State *L)
{
	llm_connection_t *object = luaL_checklm_connection (L, 1);
	int type = luaL_checkenum (L, 3, llm_message_type);

	if (lua_gettop (L) > 3) { // Register
		int priority = luaL_checkenum (L, 4, llm_handler_priority);

		if (lua_isfunction (L, 2)) { // Function
			LmMessageHandler *handler;
			llm_callback_t *cb = luaL_malloc (L, sizeof (llm_callback_t));
			lua_pushvalue (L, 2);
			cb->reference = luaL_ref (L, LUA_REGISTRYINDEX);
			cb->L         = L;

			handler = lm_message_handler_new (
					(LmHandleMessageFunction)llm_message_handler_callback,
					cb, (GDestroyNotify)llm_callback_destroy);
			lm_connection_register_message_handler (object->connection,
								handler, type, priority);
			lm_message_handler_unref (handler);
		} else { // Object
			llm_message_handler_t *handler = luaL_checklm_message_handler (L, 2);
			lm_connection_register_message_handler (object->connection,
								handler->message_handler,
								type, priority);
		}
		lua_pop (L, 1);
	} else { // Unregister
		llm_message_handler_t *handler = luaL_checklm_message_handler (L, 2);
		lm_connection_unregister_message_handler (object->connection,
							  handler->message_handler,
							  type);
	}
	lua_pop (L, 2);
	return 0;
}

/// disconnect callback function
/// Function, that will be called when disconnection occurs.
/// A: lm connection object, disconnect reason
void llm_disconnect_callback (LmConnection *connection, LmDisconnectReason reason, llm_callback_t *cb)
{
	lua_rawgeti (cb->L, LUA_REGISTRYINDEX, cb->reference);
	llm_connection_bless (cb->L, connection);
	// XXX lm_connection_unref (connection);
	luaL_pushenum (cb->L, reason, llm_disconnect_reason);
	if (lua_pcall (cb->L, 2, 0, 0)) {
		// XXX lua_error (cb->L);
		lua_pop (cb->L, 1);
		return;
	}
}

/// connection:ondisconnect
/// Sets callback, that will be called on connection disconnect.
/// A: disconnect callback function
static int llm_connection_ondisconnect (lua_State *L)
{
	llm_connection_t *object = luaL_checklm_connection (L, 1);
	llm_callback_t *cb;
	luaL_argcheck (L, lua_isfunction (L, 2), 2, "function expected");

	cb = luaL_malloc (L, sizeof (llm_callback_t));
	cb->reference  = luaL_ref (L, LUA_REGISTRYINDEX);
	cb->L          = L;

	lm_connection_set_disconnect_function (object->connection,
					       (LmDisconnectFunction)llm_disconnect_callback,
					       cb, (GDestroyNotify)llm_callback_destroy);
	lua_pop (L, 1);
	return 0;
}

/// connection:open_wait
/// Synchronous open call, that will block until connection will be opened.
/// R: boolean (success)
static int llm_connection_open_wait (lua_State *L)
{
	llm_connection_t *object = luaL_checklm_connection (L, 1);
	lua_pushboolean (L, lm_connection_open_and_block (object->connection, NULL));
	lua_remove (L, -2);
	return 1;
}

/// connection:authenticate_wait
/// Synchronous authentication call, that will wait until the end of authentication.
/// A: string (username), string (password), string (resource)
/// R: boolean (success)
static int llm_connection_authenticate_wait (lua_State *L)
{
	llm_connection_t *object = luaL_checklm_connection (L, 1);
	const char *username = luaL_checkstring (L, 2);
	const char *password = luaL_checkstring (L, 3);
	const char *resource = luaL_checkstring (L, 4);
	int status = lm_connection_authenticate_and_block (object->connection, username,
								password, resource, NULL);
	lua_pop (L, 4);
	lua_pushboolean (L, status);
	return 1;
}

/// connection:send_wait
/// Synchronous call, that will send message and wait for reply to it.
/// A: lm message object
/// R: lm message object or nil
static int llm_connection_send_wait (lua_State *L)
{
	llm_connection_t *object = luaL_checklm_connection (L, 1);
	llm_message_t *message = luaL_checklm_message (L, 2);
	LmMessage *new = lm_connection_send_with_reply_and_block (object->connection,
									message->message, NULL);
	lua_pop (L, 2);
	if (!new)
		lua_pushnil (L);
	else {
		llm_message_bless (L, new);
		lm_message_unref (new); // XXX
	}
	return 1;
}

/// connection:pointer
/// Returns pointer to underlying C loudmouth structure.
/// R: lightuserdata
static int llm_connection_pointer (lua_State *L)
{
	llm_connection_t *object = luaL_checklm_connection (L, 1);
	lua_pushlightuserdata (L, object->connection);
	lua_remove (L, -2);
	return 1;
}

static int llm_connection_gc (lua_State *L)
{
	llm_connection_t *object = luaL_checklm_connection (L, 1);
	lm_connection_unref (object->connection);
	lua_pop (L, 1);
	return 0;
}

static const luaL_Reg llm_connection_reg_f[] = {
	{ "new",   llm_connection_new       },
	{ "bless", llm_connection_bless_lua },
	{ NULL,    NULL                     },
};

static const luaL_Reg llm_connection_reg_m[] = {
	{ "open",                llm_connection_open                },
	{ "close",               llm_connection_close               },
	{ "authenticate",        llm_connection_authenticate        },
	{ "port",                llm_connection_port                },
	{ "server",              llm_connection_server              },
	{ "jid",                 llm_connection_jid                 },
	{ "keep_alive_rate",     llm_connection_keep_alive_rate     },
	{ "state",               llm_connection_status              },
	{ "proxy",               llm_connection_proxy               },
	{ "ssl",                 llm_connection_ssl                 },
	{ "send",                llm_connection_send                },
	{ "send_raw",            llm_connection_send_raw            },
	{ "handler",             llm_connection_handler             },
	{ "ondisconnect",        llm_connection_ondisconnect        },
	{ "open_wait",           llm_connection_open_wait           },
	{ "authenticate_wait",   llm_connection_authenticate_wait   },
	{ "send_wait",           llm_connection_send_wait           },
	{ "pointer",             llm_connection_pointer             },
	{ "__gc",                llm_connection_gc                  },
	{ NULL,                  NULL                               },
};

int luaopen_lm_connection (lua_State *L)
{
	luaL_newmetatable (L, "loudmouth.connection");
	lua_pushstring (L, "__index");
	lua_pushvalue (L, -2);
	lua_settable (L, -3);
	luaL_register (L, NULL, llm_connection_reg_m);
	lua_pop (L, 1);
	luaL_register (L, "lm.connection", llm_connection_reg_f);
	return 1;
}