#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

static 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 | LPRINT_NOTUTF8, lua_tostring (L, -1));

	return 0;

}

/// dopath

/// Loads lua file from default location.

/// XXX: g_filename_from_utf8?

/// 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 (size > 4 && !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;

}

/// yes or no ansvers

/// G:

static const string2enum_t lua_yesno[] = {

	{ "1",       1 },

	{ "0",       0 },

	{ "enable",  1 },

	{ "disable", 0 },

	{ "true",    1 },

	{ "false",   0 },

	{ "on",      1 },

	{ "off",     0 },

	{ "yes",     1 },

	{ "no",      0 },

	{ NULL,     -1 },

};

/// main.yesno

/// According to yes or no ansvers returns true or false.

/// If ansver is not recognized, returns nil.

/// A: anything (string expected)

/// R: boolean or nil

static int lua_main_yesno (lua_State *L)

{

	int type = lua_type (L, 1);

	if (type == LUA_TSTRING) {

		int ret = luaL_checkenum (L, 1, lua_yesno);

		if (ret == -1)

			lua_pushnil (L);

		else

			lua_pushboolean (L, ret);

	} else if (type == LUA_TNUMBER)

		lua_pushboolean (L, lua_tointeger (L, 1));

	else if (type != LUA_TBOOLEAN)

		lua_pushnil (L);

	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.

/// Note: most likely you need notutf8 flag enabled.

/// 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;

}

// expects table on top

static void lua_options_callback (char *key, char *value, lua_State *L)

{

	char *loc = from_utf8 (key);

	lua_pushstring (L, loc);

	g_free (loc);

	loc = from_utf8 (value);

	lua_pushstring (L, loc);

	g_free (loc);

	lua_settable (L, -3);

}

/// main.option

/// Sets or gets value of mcabber option.

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

/// If you omit option name, it returns hash table of all options.

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

/// R: string (value, optional)

static int lua_main_option (lua_State *L)

{

	int top = lua_gettop (L);

	if (top > 0) {

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

		if (top > 1) { // Set

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

				settings_del (SETTINGS_TYPE_OPTION, name);

			else { // Set

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

				settings_set (SETTINGS_TYPE_OPTION, name, value);

				g_free (value);

			}

			g_free (name);

			return 0;

		} else { // Get

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

			if (value) {

				lua_pushstring (L, value);

				g_free (value);

			} else

				lua_pushnil (L);

			g_free (name);

			return 1;

		}

	} else { // List

		lua_newtable (L);

		settings_foreach (SETTINGS_TYPE_OPTION, (void (*)(char *key, char *val, void *ud)) lua_options_callback, L);

		return 1;

	}

}

/// main.alias

/// Sets or gets alias.

/// You can specify nil as a command to delete alias.

/// If you omit alias name, it will return hash table of all aliases.

/// A: string (alias name, optional), string (command, optional)

/// R: string (command, optional)

static int lua_main_alias (lua_State *L)

{

	int top = lua_gettop (L);

	if (top > 0) {

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

		if (top > 1) { // Set

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

				settings_del (SETTINGS_TYPE_ALIAS, name);

				compl_del_category_word (COMPL_CMD, name);

			} else { // Set

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

				if (!settings_get (SETTINGS_TYPE_ALIAS, name))

					compl_add_category_word (COMPL_CMD, name);

				settings_set (SETTINGS_TYPE_ALIAS, name, value);

				g_free (value);

			}

			g_free (name);

			return 0;

		} else { // Get

			char *value = from_utf8 (settings_get (SETTINGS_TYPE_ALIAS, name));

			if (value) {

				lua_pushstring (L, value);

				g_free (value);

			} else

				lua_pushnil (L);

			g_free (name);

			return 1;

		}

	} else { // List

		lua_newtable (L);

		settings_foreach (SETTINGS_TYPE_ALIAS, (void (*)(char *key, char *val, void *ud)) lua_options_callback, L);

		return 1;

	}

}

/// main.bind

/// Sets or gets binding.

/// You can specify nil as a command to unbind key.

/// If you omit keycode, it will return hash table of all bindings.

/// A: string (keycode, optional), string (command, optional)

/// R: string (command, optional)

static int lua_main_binding (lua_State *L)

{

	int top = lua_gettop (L);

	if (top > 0) {

		// just to be sure...

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

		if (top > 1) { // Set

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

				settings_del (SETTINGS_TYPE_BINDING, name);

			else { // Set

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

				settings_set (SETTINGS_TYPE_BINDING, name, value);

				g_free (value);

			}

			g_free (name);

			return 0;

		} else { // Get

			char *value = from_utf8 (settings_get (SETTINGS_TYPE_BINDING, name));

			if (value) {

				lua_pushstring (L, value);

				g_free (value);

			} else

				lua_pushnil (L);

			g_free (name);

			return 1;

		}

	} else { // List

		lua_newtable (L);

		settings_foreach (SETTINGS_TYPE_BINDING, (void (*)(char *key, char *val, void *ud)) lua_options_callback, L);

		return 1;

	}

}

/// main.fileoption

/// Gets option, expanding it as filename.

/// A: string (option name)

/// R: string (expanded option value) or nil

static int lua_main_fileoption (lua_State *L)

{

	char *fname = expand_filename (settings_opt_get (luaL_checkstring (L, 1)));

	if (fname) {

		lua_pushstring (L, fname);

		g_free (fname);

	} 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 *jid  = to_utf8 (luaL_checkstring (L, 1));

	char *to   = jidtodisp (jid);

	char *mesg = to_utf8 (luaL_checkstring (L, 2));

	scr_WriteIncomingMessage (to, mesg, 0, HBB_PREFIX_INFO, 0);

	g_free (mesg);

	g_free (to);

	g_free (jid);

	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)

{

	char *jid = from_utf8 (buddy_getjid (buddy));

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

	lua_pushstring (L, jid);

	lua_settable (L, -3);

	g_free (jid);

}

/// 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

		// XXX: we need not convert to utf, RS works on jids/names in locale charset,

		// but will jidtodisp always correctly work on such tings?

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

		scr_RosterSearch (jid);

		g_free (jid);

		return 0;

	} else { // Get

		char *jid = from_utf8 (buddy_getjid (BUDDATA (current_buddy)));

		lua_pushstring (L, jid);

		g_free (jid);

		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.