#include <glib.h>	// g_ascii_strcasecmp

#include <lua.h>

#include <lauxlib.h>

#include "util.h"

enum_value_t string2enum (const char *string, const string2enum_t *set)

{

	while (set->string) {

		if (!g_ascii_strcasecmp (string, set->string))

			return set->value;

		++set;

	}

	return set->value;

}

const char *enum2string (enum_value_t value, const string2enum_t *set)

{

	while (set->string) {

		if (value == set->value)

			return set->string;

		++set;

	}

	return NULL;

}

enum_value_t luaL_checkenum (lua_State *L, int index, const string2enum_t *set)

{

	if (lua_type (L, index) == LUA_TNUMBER)

		return lua_tointeger (L, index);

	else

		return string2enum (luaL_checkstring (L, index), set);

}

void luaL_pushenum (lua_State *L, enum_value_t value, const string2enum_t *set)

{

	const char *string = enum2string (value, set);

	if (string != NULL)

		lua_pushstring (L, string);

	else

		lua_pushinteger (L, value);

}

enum_value_t luaL_checkenum_multi (lua_State *L, int index, const string2enum_t *set)

{

	int type = lua_type (L, index);

	if (type == LUA_TNUMBER)

		return lua_tointeger (L, index);

	else if (type == LUA_TSTRING)

		return string2enum (lua_tostring (L, index), set);

	else if (type == LUA_TTABLE) {

		enum_value_t retval = 0;

		lua_pushnil (L);

		while (lua_next (L, index) != 0) {

			type = lua_type (L, -2);

			if (type == LUA_TNUMBER) {

				type = lua_type (L, -1);

				if (type == LUA_TNUMBER)

					retval |= lua_tointeger (L, -1);

				else if (type == LUA_TSTRING)

					retval |= string2enum (lua_tostring (L, -1), set);

				else

					luaL_argerror (L, index, "wrong value type of flag");

			} else if (type == LUA_TSTRING) {

				if (lua_toboolean (L, -1))

					retval |= string2enum (lua_tostring (L, -2), set);

			} else

				luaL_argerror (L, index, "wrong key type of flags table");

			lua_pop (L, 1);

		}

	} else

		luaL_argerror (L, index, "integer, string, or table of ones expected");

	return 0; // never happens

}

void luaL_pushenum_multi (lua_State *L, enum_value_t value, const string2enum_t *set)

{

	enum_value_t matched = 0;

	lua_newtable (L);

	while (set->string) {

		if (value & set->value) {

			matched |= set->value & value;

			lua_pushstring (L, set->string);

			lua_pushboolean (L, 1);

			lua_settable (L, -3);

		}

		++set;

	}

	if (value ^ matched) {

		lua_pushinteger (L, 1);

		lua_pushinteger (L, value ^ matched);

		lua_settable (L, -3);

	}

}

void *luaL_malloc (lua_State *L, size_t size)

{

	void      *ud;

	lua_Alloc  allocf = lua_getallocf (L, &ud);

	return (*allocf) (ud, NULL, 0, size);

}

void *luaL_realloc (lua_State *L, void *ptr, size_t osize, size_t nsize)

{

	void      *ud;

	lua_Alloc  allocf = lua_getallocf (L, &ud);

	return (*allocf) (ud, ptr, osize, nsize);

}

void  luaL_free (lua_State *L, void *ptr)

{

	void      *ud;

	lua_Alloc  allocf = lua_getallocf (L, &ud);

	(*allocf) (ud, ptr, 1, 0);

}