/* Copyright 2009 Myhailo Danylenko

This file is part of lua-lm.

lua-lm is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 2 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>. */

#include <lua.h>
#include <lauxlib.h>
#include <loudmouth/loudmouth.h>

#include "config.h"
#include "lm_types.h"
#include "util.h"

void llm_callback_destroy (llm_callback_t *cb)
{
	D ("Destroying callback %p", cb);
	luaL_unref (cb->L, LUA_REGISTRYINDEX, cb->reference);
	luaL_free (cb->L, cb);
}

#define LLM_CHECK(WHAT, LWHAT, TYPE)                                              \
llm_##WHAT##_t *luaL_checklm_##WHAT (lua_State *L, int index)                     \
{                                                                                 \
	llm_##WHAT##_t *object = luaL_checkudata (L, index, "loudmouth." #LWHAT); \
	luaL_argcheck (L, object != NULL, index, "loudmouth " #WHAT " expected"); \
	return object;                                                            \
}

#define LLM_BLESS(WHAT, LWHAT, TYPE)                                            \
llm_##WHAT##_t *bless_lm_##WHAT (lua_State *L, TYPE *WHAT)                      \
{                                                                               \
	llm_##WHAT##_t *object;                              /* top of stack */ \
	lua_pushliteral (L, LLM_OBJREGISTRY);      /* 1 registry table name  */ \
	lua_rawget (L, LUA_REGISTRYINDEX);               /* 1 registry table */ \
	lua_pushlightuserdata (L, WHAT);                 /* 2 light userdata */ \
	lua_rawget (L, -2);                                  /* 2 object/nil */ \
	if (!lua_isnil (L, -1)) {                                /* 2 object */ \
		lua_remove (L, -2);                              /* 1 object */ \
		object = lua_touserdata (L, -1);                                \
		D ("Existing " #LWHAT " object %p requested", object);          \
		return object;                                                  \
	}                                                                       \
	                                                            /* 2 nil */ \
	lua_remove (L, -1);                              /* 1 registry table */ \
	object = lua_newuserdata (L, sizeof (llm_##WHAT##_t)); /* 2 userdata */ \
	luaL_getmetatable (L, "loudmouth." #LWHAT);           /* 3 metatable */ \
	lua_setmetatable (L, -2);                                /* 2 object */ \
	lua_pushlightuserdata (L, WHAT);                 /* 3 light userdata */ \
	lua_pushvalue (L, -2);                                   /* 4 object */ \
	lua_rawset (L, -4);                                      /* 2 object */ \
	lua_remove (L, -2);                                      /* 1 object */ \
	object->WHAT = WHAT;                                                    \
	lm_##LWHAT##_ref (WHAT);                                                \
	D ("New " #LWHAT " object %p blessed", object);                         \
	return object;                                                          \
}

#define LLM_DEFINE(WHAT, LWHAT, TYPE) \
LLM_CHECK (WHAT, LWHAT, TYPE)         \
LLM_BLESS (WHAT, LWHAT, TYPE)

LLM_DEFINE (connection, connection, LmConnection)
LLM_DEFINE (message, message, LmMessage)
LLM_DEFINE (handler, message_handler, LmMessageHandler)
LLM_DEFINE (node, message_node, LmMessageNode)
LLM_DEFINE (proxy, proxy, LmProxy)
LLM_DEFINE (ssl, ssl, LmSSL)

LmMessageNode *luaL_checkLmMessageNode (lua_State *L, int index)
{
	void *object;
	luaL_argcheck (L, lua_type (L, index) == LUA_TUSERDATA, index, "loudmouth message or message node expected");
	object = lua_touserdata (L, index);
	lua_getmetatable (L, index);
	luaL_getmetatable (L, "loudmouth.message");
	luaL_getmetatable (L, "loudmouth.message_node");
	if (lua_rawequal (L, -2, -3)) // Message
		object = (void *) lm_message_get_node (((llm_message_t *) object)->message);
	else if (lua_rawequal (L, -1, -3)) // Node
		object = (void *) (((llm_node_t *) object)->node);
	else
		luaL_argerror (L, index, "loudmouth message or message node expected");
	lua_pop (L, 3);
	return (LmMessageNode *) object;
}

#undef LLM_DEFINE
#undef LLM_BLESS
#undef LLM_CHECK