#include <glib.h>

#include <gmodule.h>

#include <lua.h>

#include <lauxlib.h>

#include <lualib.h>

#include <stdio.h>

#include <stdlib.h>    // getenv

#include <string.h>    // strcmp

#include "config.h"

#include "util.h"

#include "logprint.h"    // scr_LogPrint

#include "screen.h"      // scr_Beep, scr_WriteIncomingMessage

#include "hbuf.h"        // HBB_PREFIX_INFO

#include "commands.h"    // process_command, cmd_add, cmd_del

#include "xmpp.h"        // xmpp_getstatus, xmpp_getstatusmsg, lconnection

#include "xmpp_helper.h" // xmpp_add_feature, xmpp_del_feature

#include "roster.h"      // imstatus2char, foreach_buddy, buddy_*, current_buddy, BUDDATA, ROSTER_TYPE_*

#include "utils.h"       // from_utf8, jidtodisp

#include "hooks.h"       // hk_add_handler, hk_del_handler

#include "settings.h"    // settings_set, settings_del, settings_get

#include "compl.h"       // compl_new_category, compl_add_category_word, compl_del_category_word

// global lua state object, necessary for uninitialization function

static lua_State *lua = NULL;

// caller sould g_free result

char *mcabber_config_filename (const char *file)

{

	const char *home = getenv ("HOME");

	if (!home)

		return NULL;

	return g_strconcat (home, "/.mcabber/", file ? file : "", NULL);

}

/// print

/// Prints its arguments to log with default priority.

/// A: string/number, ...

static int lua_global_print (lua_State *L)

{

	lua_concat (L, lua_gettop (L));

	scr_LogPrint (LPRINT_LOGNORM, lua_tostring (L, -1));

	return 0;

}

/// dopath

/// Loads lua file from default location.

/// A: string (filename, without ".lua")

/// R: string (error message, optional)

static int lua_global_dopath (lua_State *L)

{

	const char *name = luaL_checkstring (L, 1);

	size_t      size = lua_objlen (L, 1);

	char       *path;

	int         ret = 0;

	if (!strncmp (name + size - 4, ".lua", 4))

		path = mcabber_config_filename (name);

	else {

		char *fname = g_strconcat (name, ".lua", NULL);

		path = mcabber_config_filename (fname);

		g_free (fname);

	}

	if (ret = luaL_loadfile (L, path))

		scr_LogPrint (LPRINT_LOGNORM, "lua: Unable to compile file %s: %s", path, lua_tostring (L, -1));

	else if (ret = lua_pcall (lua, 0, LUA_MULTRET, 0))

		scr_LogPrint (LPRINT_LOGNORM, "lua: Runtime error in file %s: %s", path, lua_tostring (L, -1));

	g_free (path);

	if (ret)

		return 1;

	else

		return 0;

}

/// main.config_file

/// Adds mcabber default config location path to config file name.

/// Note: deprecated, use dopath.

/// A: string (filename)

/// R: string (full path)

static int lua_main_config_file (lua_State *L)

{

	char *home = mcabber_config_filename (luaL_checkstring (L, 1));

	if (!home) {

		lua_pushstring (L, "Cannot find home dir!");

		lua_error (L);

	}

	lua_pushstring (L, home);

	g_free (home);

	return 1;

}

/// log print type

/// G:

static const string2enum_t lua_lprint[] = {

	{ "normal",  LPRINT_NORMAL  },

	{ "log",     LPRINT_LOG     },

	{ "debug",   LPRINT_DEBUG   },

	{ "notutf0", LPRINT_NOTUTF8 },

	{ NULL,      0              },

};

/// roster type

/// G:

static const string2enum_t lua_roster_type[] = {

	{ "user",    ROSTER_TYPE_USER    },

	{ "group",   ROSTER_TYPE_GROUP   },

	{ "agent",   ROSTER_TYPE_AGENT   },

	{ "room",    ROSTER_TYPE_ROOM    },

	{ "special", ROSTER_TYPE_SPECIAL },

	{ NULL,      0                   },

};

/// main.log

/// Prints message to log.

/// A: log print type, message, message...

static int lua_main_log (lua_State *L)

{

	int type = luaL_checkenum_multi (L, 1, lua_lprint);

	lua_concat (L, lua_gettop (L) - 1);

	scr_LogPrint (type, lua_tostring (L, -1));

	return 0;

}

/// main.option

/// Sets or gets value of mcabber option.

/// You can specify nil as a value to delete option.

/// XXX: Should we do types here?

/// A: string (option name), string (value, optional)

/// R: string (value, optional)

static int lua_main_option (lua_State *L)

{

	const char *name = luaL_checkstring (L, 1);

	if (lua_gettop (L) > 1) { // Set

		if (lua_type (L, 2) == LUA_TNIL) // Unset

			settings_del (SETTINGS_TYPE_OPTION, name);

		else { // Set

			const char *value = luaL_checkstring (L, 2);

			settings_set (SETTINGS_TYPE_OPTION, name, value);

		}

		return 0;

	} else { // Get

		const char *value = settings_get (SETTINGS_TYPE_OPTION, name);

		if (value)

			lua_pushstring (L, value);

		else

			lua_pushnil (L);

		return 1;

	}

}

/// main.connection

/// Returns lightuserdata of mcabber's loudmouth connection.

/// This can be very useful with lua-loudmouth, and not much otherwise.

/// R: lightuserdata

static int lua_main_connection (lua_State *L)

{

	lua_pushlightuserdata (L, lconnection);

	return 1;

}

/// main.print_info

/// Prints a system message to buddy's window.

/// A: string (jid), string (message)

static int lua_main_print_info (lua_State *L)

{

	char *to = jidtodisp (luaL_checkstring (L, 1));

	scr_WriteIncomingMessage (to, luaL_checkstring (L, 2), 0, HBB_PREFIX_INFO, 0);

	g_free (to);

	return 0;

}

/// main.beep

/// Beeps with system speaker.

static int lua_main_beep (lua_State *L)

{

	scr_Beep ();

	return 0;

}

/// main.run

/// Runs specified mcabber command.

/// A: string

static int lua_main_run (lua_State *L)

{

	process_command (luaL_checkstring (L, 1), TRUE);

	return 0;

}

/// main.status

/// Returns your current status.

/// R: string (status letter), string (status message)

static int lua_main_status (lua_State *L)

{

	char *sm = from_utf8 (xmpp_getstatusmsg ());

	lua_pushlstring (L, &imstatus2char[xmpp_getstatus ()], 1);

	lua_pushstring (L, sm);

	g_free (sm);

	return 2;

}

// expects table on top

static void lua_rosterlist_callback (gpointer buddy, lua_State *L)

{

	lua_pushnumber (L, lua_objlen (L, -1) + 1);

	lua_pushstring (L, buddy_getjid (buddy));

	lua_settable (L, -3);

}

/// main.roster

/// Returns array of jids of buddies in roster.

/// R: table

static int lua_main_roster (lua_State *L)

{

	lua_newtable (L);

	foreach_buddy (ROSTER_TYPE_USER|ROSTER_TYPE_AGENT|ROSTER_TYPE_ROOM, (void (*) (gpointer buddy, void *data)) lua_rosterlist_callback, L);

	return 1;

}

/// main.current_buddy

/// Returns jid of current selected buddy or sets current buddy to buddy with specified jid.

/// A: string (optional)

/// R: string (optional)

static int lua_main_current_buddy (lua_State *L)

{

	if (lua_gettop (L) > 0) { // Set

		char *jid = jidtodisp (luaL_checkstring (L, 1));

		scr_RosterSearch (jid);

		g_free (jid);

		return 0;

	} else { // Get

		lua_pushstring (L, buddy_getjid (BUDDATA (current_buddy)));

		return 1;

	}

}

/// main.full_jid

/// Returns full jid (with current resource) of specified buddy (or current, if not specified).

/// Note, that if there are no resources online, it will return just what it got.

/// A: string (jid, optional)

/// R: string (jid)

static int lua_main_full_jid (lua_State *L)

{

	GList *buddy;

	GSList *resources;

	GSList *resource;

	if (lua_gettop (L) > 0)

		buddy = buddy_search_jid (luaL_checkstring (L, 1));

	else

		buddy = current_buddy;

	if (!buddy)

		return 0;

	resources = buddy_getresources (BUDDATA (buddy));

	if (!resources) {

		lua_pushstring (L, buddy_getjid (BUDDATA (buddy)));

		return 1;

	}

	lua_pushfstring (L, "%s%c%s", buddy_getjid (BUDDATA (buddy)), JID_RESOURCE_SEPARATOR, g_slist_last(resources)->data);

	for (resource = resources; resource; resource = g_slist_next (resource))

		g_free (resource->data);

	g_slist_free (resources);

	return 1;

}

typedef struct {

	lua_State *L;

	gpointer   buddy;

} lua_state_and_buddy_t; // :)

// expects table on top!

static void lua_buddy_resources_callback (gpointer resource, lua_state_and_buddy_t *d)

{

	lua_pushstring (d->L, resource);

	lua_createtable (d->L, 0, 3);

	lua_pushstring (d->L, "priority");

	lua_pushnumber (d->L, buddy_getresourceprio (d->buddy, resource));

	lua_settable   (d->L, -3);

	lua_pushstring  (d->L, "status");

	lua_pushlstring (d->L, &imstatus2char[buddy_getstatus (d->buddy, resource)], 1);

	lua_settable    (d->L, -3);

	lua_pushstring (d->L, "message");

	lua_pushstring (d->L, buddy_getstatusmsg (d->buddy, resource));

	lua_settable   (d->L, -3);

	lua_settable (d->L, -3);

	g_free (resource);

}

/// main.buddy_info

/// Returns a hash table with information on specified buddy.

/// Table contains fields type, name, onserver and resources.

/// Resources is also a hash table, that contains tables with information,

/// specific for each resource. In each resource table there are fields

/// priority, status and message.

/// A: string (jid)

/// R: table

static int lua_main_buddy_info (lua_State *L)

{

	char                  *jid   = jidtodisp (luaL_checkstring (L, 1));

	GSList                *buddy = roster_find (jid, jidsearch, ROSTER_TYPE_USER|ROSTER_TYPE_AGENT|ROSTER_TYPE_ROOM);

	GSList                *resources;

	lua_state_and_buddy_t  snb;

	g_free (jid);

	if (!buddy) {

		lua_pushnil (L);

		return 1;

	}

	lua_createtable (L, 0, 3);

	lua_pushstring (L, "type");

	luaL_pushenum  (L, buddy_gettype (BUDDATA (buddy)), lua_roster_type);

	lua_settable   (L, -3);

	lua_pushstring (L, "name");

	lua_pushstring (L, buddy_getname (BUDDATA (buddy)));

	lua_settable   (L, -3);

	lua_pushstring  (L, "onserver");

	lua_pushboolean (L, buddy_getonserverflag (BUDDATA (buddy)));

	lua_settable    (L, -3);

	lua_pushstring  (L, "resources");

	lua_createtable (L, 0, 0);

	snb.L     = L;

	snb.buddy = BUDDATA (buddy);

	resources = buddy_getresources (BUDDATA (buddy));

	g_slist_foreach (buddy_getresources (BUDDATA (buddy)), (GFunc) lua_buddy_resources_callback, &snb);

	g_slist_free (resources);

	lua_settable (L, -3);

	return 1;

}

// XMPP DISCO FEATURES

GSList *lua_added_features = NULL;

/// main.add_feature

/// Adds xmlns to disco#info features list.

/// A: string (xmlns)

static int lua_main_add_feature (lua_State *L)

{

	const char *xmlns = luaL_checkstring (L, 1);

	xmpp_add_feature (xmlns);

	lua_added_features = g_slist_prepend (lua_added_features, g_strdup (xmlns));

	return 0;

}

/// main.del_feature

/// Removes xmlns from disco#info features list.

/// A: stirng (xmlns)

static int lua_main_del_feature (lua_State *L)

{

	const char *xmlns = luaL_checkstring (L, 1);

	GSList     *el    = g_slist_find_custom (lua_added_features, xmlns, (GCompareFunc) strcmp);

	xmpp_del_feature (xmlns);

	if (el) {

		g_free (el->data);

		lua_added_features = g_slist_delete_link (lua_added_features, el);

	}

	return 0;

}

// MCABBER COMMANDS

/// completion type

/// Built-it completion types can be specified as string, instead of id.

/// G:

static const string2enum_t lua_completion_type[] = {

	{ "cmd",       COMPL_CMD       },

	{ "jid",       COMPL_JID       },

	{ "urljid",    COMPL_URLJID    },

	{ "name",      COMPL_NAME      },

	{ "status",    COMPL_STATUS    },

	{ "filename",  COMPL_FILENAME  },

	{ "roster",    COMPL_ROSTER    },

	{ "buffer",    COMPL_BUFFER    },

	{ "group",     COMPL_GROUP     },

	{ "groupname", COMPL_GROUPNAME },

	{ "multiline", COMPL_MULTILINE },

	{ "room",      COMPL_ROOM      },

	{ "resource",  COMPL_RESOURCE  },

	{ "auth",      COMPL_AUTH      },

	{ "request",   COMPL_REQUEST   },

	{ "events",    COMPL_EVENTS    },

	{ "eventsid",  COMPL_EVENTSID  },

	{ "pgp",       COMPL_PGP       },

	{ "color",     COMPL_COLOR     },

	{ "otr",       COMPL_OTR       },

	{ "ortpolicy", COMPL_OTRPOLICY },

	{ NULL,        0               },

};

typedef struct {

	int        reference;