antd-lua-plugin/lua-api.c

#define PLUGIN_IMPLEMENT 1
#include <dlfcn.h>
#include <antd/plugin.h>
#include <antd/scheduler.h>
#include <antd/dbhelper.h>
#include <sys/stat.h>
#include "lib/lualib.h"

static void* core_handle = NULL;


void init()
{
	char* path = __s("%s/lua/core.so", __plugin__.pdir);
	core_handle = dlopen(path, RTLD_NOW| RTLD_GLOBAL);
	if(!core_handle)
	{
		ERROR("Cannot load Lua core; %s", dlerror());
	}
	else
	{
		LOG("Lua core loaded");
	}
	free(path);
}
/**

* Plugin handler, reads request from the server and processes it
* 
*/
static void push_dict_to_lua(lua_State* L, dictionary_t d)
{
	lua_newtable(L);
	
	chain_t as;
	if(d)
		for_each_assoc(as, d)
		{
			lua_pushstring(L,as->key);
			//printf("KEY %s\n", as->key);
			if(EQU(as->key,"COOKIE") || EQU(as->key,"REQUEST_HEADER") || EQU(as->key,"REQUEST_DATA") )
				push_dict_to_lua(L, (dictionary_t)as->value);
			else
			{
				lua_pushstring(L,as->value);
				//printf("VALUE : %s\n",as->value );
			}
			lua_settable(L, -3);
		}
}
void* handle(void* data)
{
	antd_request_t* rq = (antd_request_t*) data;
	char buf[BUFFLEN];
	plugin_header_t* __plugin__ = meta();
	lua_State* L = NULL;
	//char * index = __s("%s/%s",__plugin__.htdocs,"router.lua");
	char* cnf = config_dir();
	char * apis = __s("%s/%s",cnf,"api.lua");
	L = luaL_newstate();
	luaL_openlibs(L);
	//module loader
	//luaL_newlib(L, modules);
	//lua_setglobal(L, "modules");
	// set up global variable
	// API header
	lua_newtable(L);
	lua_pushstring(L,"name");
	lua_pushstring(L, __plugin__->name);
	lua_settable(L,-3);
	
	lua_pushstring(L,"root");
	htdocs(rq, buf);
	lua_pushstring(L, buf);
	lua_settable(L,-3);
	
	lua_pushstring(L,"apiroot");
	lua_pushstring(L, cnf);
	lua_settable(L,-3);

	lua_pushstring(L,"tmpdir");
	tmpdir(buf);
	lua_pushstring(L, buf);
	lua_settable(L,-3);

	lua_pushstring(L,"dbpath");
	lua_pushstring(L, __plugin__->dbpath);
	lua_settable(L,-3);
	
	lua_setglobal(L, "__api__");
	
	// Request
	lua_newtable(L);
	lua_pushstring(L,"id");
	lua_pushlightuserdata(L, rq->client);
	//lua_pushnumber(L,client);
	lua_settable(L, -3);
	lua_pushstring(L,"request");
	push_dict_to_lua(L,rq->request);
	lua_settable(L, -3);
	lua_setglobal(L, "HTTP_REQUEST");
	
	// load major apis
	if(is_file(apis))
		if (luaL_loadfile(L, apis) || lua_pcall(L, 0, 0, 0))
		{
			ERROR( "cannot start API file: [%s] %s\n", apis, lua_tostring(L, -1));
		}
	
	/*if (luaL_loadfile(L, index) || lua_pcall(L, 0, 0, 0))
	{
		text(client);
		__t(client, "Cannot run router: %s", lua_tostring(L, -1));
	}
	free(index);*/
	// clear request
	if(L)
		lua_close(L);
	if(cnf)
		free(cnf);
	if(apis)
		free(apis);
	return antd_create_task(NULL, (void*)rq, NULL,rq->client->last_io);
	//lua_close(L);
}
void destroy()
{
	if(core_handle)
		dlclose(core_handle);
	LOG("Exit LUA Handle");
}