/* 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 <glib.h>
#include <loudmouth/loudmouth.h>
#include "config.h"
#include "lm_types.h"
#include "util.h"
/// lm.message_handler
/// Module, representing message handling functions.
/// message handler callback function
/// Callback function, used on incoming messages.
/// Should return true if message has been fully handled and no more
/// handlers should be called.
/// A: lm connection object, lm message object
/// R: boolean
LmHandlerResult callback_lm_handler (LmMessageHandler *handler, LmConnection *connection, LmMessage *message, llm_callback_t *cb)
{
int ret;
lua_rawgeti (cb->L, LUA_REGISTRYINDEX, cb->reference);
bless_lm_connection (cb->L, connection);
// XXX lm_connection_unref (connection);
bless_lm_message (cb->L, message);
// XXX lm_message_unref (message);
if (lua_pcall (cb->L, 2, 1, 0)) {
W ("Message handler callback error: %s", lua_tostring (cb->L, -1));
lua_pop (cb->L, 1);
return LM_HANDLER_RESULT_ALLOW_MORE_HANDLERS;
}
ret = lua_toboolean (cb->L, -1);
lua_pop (cb->L, 1);
if (ret)
return LM_HANDLER_RESULT_REMOVE_MESSAGE;
else
return LM_HANDLER_RESULT_ALLOW_MORE_HANDLERS;
}
/// lm.message_handler.new
/// Creates new message handler object.
/// A: message handler callback function
/// R: lm message handler object
static int new_lm_handler (lua_State *L)
{
llm_callback_t *cb;
LmMessageHandler *handler;
luaL_argcheck (L, lua_isfunction (L, 1), 1, "function expected");
cb = luaL_malloc (L, sizeof (llm_callback_t));
cb->reference = luaL_ref (L, LUA_REGISTRYINDEX);
cb->L = L;
handler = lm_message_handler_new ((LmHandleMessageFunction)callback_lm_handler,
cb, (GDestroyNotify)llm_callback_destroy);
bless_lm_handler (L, handler);
lm_message_handler_unref (handler); // XXX
D ("Message handler %p created", handler);
return 1;
}
/// lm.message_handler.bless
/// Blesses given pointer to lm message handler object.
/// A: lightuserdata (C lm message handler object)
/// R: lm message handler object
static int bless_lua_lm_handler (lua_State *L)
{
luaL_argcheck (L, lua_islightuserdata (L, 1), 1, "loudmouth message handler pointer expected");
bless_lm_handler (L, lua_touserdata (L, 1));
return 1;
}
// XXX: merge with next?
/// message_handler:invalidate
/// Invalidates handler.
/// R: lm message handler object
static int invalidate_lm_handler (lua_State *L)
{
llm_handler_t *object = luaL_checklm_handler (L, 1);
lm_message_handler_invalidate (object->handler);
return 1;
}
/// message_handler:valid
/// Returns true if message handler is still valid.
/// R: boolean
static int valid_lm_handler (lua_State *L)
{
llm_handler_t *object = luaL_checklm_handler (L, 1);
lua_pushboolean (L, lm_message_handler_is_valid (object->handler));
return 1;
}
/// message_handler:pointer
/// Returns pointer to underlying C loudmouth structure.
/// R: lightuserdata
static int pointer_lm_handler (lua_State *L)
{
llm_handler_t *object = luaL_checklm_handler (L, 1);
lua_pushlightuserdata (L, object->handler);
return 1;
}
static int gc_lm_handler (lua_State *L)
{
llm_handler_t *object = luaL_checklm_handler (L, 1);
D ("Message handler %p gc called", object);
lm_message_handler_unref (object->handler);
return 0;
}
static const luaL_Reg reg_f_lm_handler[] = {
{ "new", new_lm_handler },
{ "bless", bless_lua_lm_handler },
{ NULL, NULL },
};
static const luaL_Reg reg_m_lm_handler[] = {
{ "invalidate", invalidate_lm_handler },
{ "valid", valid_lm_handler },
{ "pointer", pointer_lm_handler },
{ "__gc", gc_lm_handler },
{ NULL, NULL },
};
int luaopen_lm_message_handler (lua_State *L)
{
luaL_newmetatable (L, "loudmouth.message_handler");
lua_pushvalue (L, -1);
lua_setfield (L, -2, "__index");
luaL_register (L, NULL, reg_m_lm_handler);
lua_pop (L, 1);
lua_newtable (L); // XXX we can specify here exact amount of fields
luaL_register (L, NULL, reg_f_lm_handler);
return 1;
}