more work on lua module, now able to load/unload scripts

This commit is contained in:
Aaron Blakely 2024-02-29 02:51:00 -06:00
parent 5626a7844b
commit b1a4865852
7 changed files with 292 additions and 32 deletions

View File

@ -4,10 +4,12 @@
#include <string.h>
#include <stdlib.h>
int block = 0;
void lua_init_events()
{
lua_register(lua.L, "add_handler", lua_add_handler);
lua_register(lua.L, "del_handler", lua_del_handler);
lua_register(lua.L, "_add_handler", lua_add_handler);
lua_register(lua.L, "_del_handler", lua_del_handler);
}
int lua_add_handler(lua_State *L)
@ -15,6 +17,12 @@ int lua_add_handler(lua_State *L)
char *event;
int lreg;
if (block)
{
// return error
return !LUA_OK;
}
if (lua_gettop(L) < 2)
{
printf("Error: add_handler requires 2 arguments\n");
@ -118,6 +126,8 @@ void lua_fire_handlers(char *event, ...)
int i;
va_list args;
printf("lua_fire_handlers: %s\n", event);
char *user, *host, *chan, *text;
for (i = 0; i < lua.event_count; i++)
@ -137,6 +147,7 @@ void lua_fire_handlers(char *event, ...)
text = va_arg(args, char *);
printf("dbug: %s %s %s %s\n", user, host, chan, text);
printf("dbug: %s %d %d\n", lua.events[i].event, lua.events[i].lreg, lua.event_count);
lua_callfunc(lua.events[i].lreg, 4, user, host, chan, text);
}

41
mods/lua/init.lua Executable file
View File

@ -0,0 +1,41 @@
-- init.lua: xbot runtime script
-- Do not modify this file unless you know what you are doing.
--
-- Written by Aaron Blakely
local handlerstore = {}
function load() end
function unload() end
function add_handler(type, func)
local ret = _add_handler(type, func)
local callfilename = debug.getinfo(2, "S").source
if ret then
table.insert(handlerstore, {type, func, ret, callfilename})
end
end
function del_handler(type, func)
local callfilename = debug.getinfo(2, "S").source
-- loop through the handlerstore and remove the handler
for i, v in ipairs(handlerstore) do
print("called with type: " .. type .. "callfilename: " .. callfilename)
print("dbug: type: " .. v[1] .. " func: " .. v[3] .. " callfilename: " .. v[4])
if v[2] == func then
print("dbug: ref found")
end
if v[1] == type and v[2] == func and v[4] == callfilename then
_del_handler(type, v[3])
table.remove(handlerstore, i)
return
end
end
print("Handler not found")
end

View File

@ -15,21 +15,134 @@
struct lua_interp lua;
struct irc_conn *instance;
char *scriptsfile = "./mods/lua/scripts";
int append_script(char *fname)
{
FILE *fp;
char *buf = (char *)malloc(sizeof(char *) * 500);
// check if the file is already in the list
struct script_list list = get_scripts();
for (int i = 0; i < list.count; i++)
{
if (strcmp(list.scripts[i], fname) == 0)
{
free(buf);
return 0;
}
}
if ((fp = fopen(scriptsfile, "a")) == NULL)
{
free(buf);
return -1;
}
sprintf(buf, "%s\n", fname);
fputs(buf, fp);
fclose(fp);
free(buf);
return 0;
}
int remove_script(char *fname)
{
FILE *fp;
char *buf = (char *)malloc(sizeof(char *) * 500);
char *tmpfile = "./mods/lua/scripts.tmp";
if ((fp = fopen(scriptsfile, "r")) == NULL)
{
free(buf);
return -1;
}
FILE *tmp = fopen(tmpfile, "w");
while (fgets(buf, 500, fp) != NULL)
{
if (strcmp(buf, fname) != 0)
{
fputs(buf, tmp);
}
}
fclose(fp);
fclose(tmp);
remove(scriptsfile);
rename(tmpfile, scriptsfile);
free(buf);
return 0;
}
struct script_list get_scripts()
{
FILE *fp;
char *buf = (char *)malloc(sizeof(char *) * 500);
struct script_list list;
int i = 0;
if ((fp = fopen(scriptsfile, "r")) == NULL)
{
free(buf);
list.count = 0;
return list;
}
while (fgets(buf, 500, fp) != NULL)
{
list.scripts[i] = (char *)malloc(sizeof(char *) * 150);
strlcpy(list.scripts[i], buf, 150);
// remove newline
list.scripts[i][strlen(list.scripts[i]) - 1] = '\0';
i++;
}
list.count = i;
free(buf);
return list;
}
MY_API void lua_eval(struct irc_conn *bot, char *user, char *host, char *chan, const char *text)
{
int res;
block = 1;
printf("lua eval called with %s\n", text);
if (strstr(text, "!lua") != NULL)
{
// skip the command
text = skip(text, ' ');
printf("lua: %s\n", text);
if (luaL_dostring(lua.L, text))
res = luaL_loadstring(lua.L, text);
if (res == LUA_OK)
{
res = lua_pcall(lua.L, 0, 0, 0);
if (res != LUA_OK)
{
irc_privmsg(bot, chan, "Error: %s", lua_tostring(lua.L, -1));
lua_pop(lua.L, 1);
}
}
else
{
irc_privmsg(bot, chan, "Error: %s", lua_tostring(lua.L, -1));
lua_pop(lua.L, 1);
}
}
block = 0;
}
MY_API void lua_load_script(struct irc_conn *bot, char *user, char *host, char *chan, const char *text)
@ -46,18 +159,35 @@ MY_API void lua_load_script(struct irc_conn *bot, char *user, char *host, char *
strlcpy(lua.scripts[lua.script_count].fname, buf, 150);
if (luaL_loadfile(lua.L, buf) != LUA_OK)
{
if (!strcmp(chan, "-stdio-"))
{
printf("Error loading lua script: %s\n", lua_tostring(lua.L, -1));
}
else
{
irc_privmsg(bot, chan, "Error loading lua script: %s", lua_tostring(lua.L, -1));
}
return;
}
name = basename(buf);
name = basename(buf);
append_script(text);
// execute the script
if (lua_pcall(lua.L, 0, 0, 0) != LUA_OK)
{
if (!strcmp(chan, "-stdio-"))
{
printf("Error executing lua script: %s\n", lua_tostring(lua.L, -1));
}
else
{
irc_privmsg(bot, chan, "Error executing lua script: %s", lua_tostring(lua.L, -1));
}
return;
}
@ -72,8 +202,16 @@ MY_API void lua_load_script(struct irc_conn *bot, char *user, char *host, char *
else
{
lua.scripts[lua.script_count].unload = -1;
if (!strcmp(chan, "-stdio-"))
{
printf("No unload() function in %s\n", name);
}
else
{
irc_privmsg(bot, chan, "No unload() function in %s", name);
}
}
// call the load function if it exists
lua_getglobal(lua.L, "load");
@ -81,21 +219,42 @@ MY_API void lua_load_script(struct irc_conn *bot, char *user, char *host, char *
if (lua_isfunction(lua.L, -1))
{
if (lua_pcall(lua.L, 0, 0, 0) != LUA_OK)
{
if (!strcmp(chan, "-stdio-"))
{
printf("Error calling load() in %s: %s\n", buf, lua_tostring(lua.L, -1));
}
else
{
irc_privmsg(bot, chan, "Error calling load() in %s: %s", buf, lua_tostring(lua.L, -1));
}
return;
}
sprintf(buf, "Loaded %s", name);
lua.script_count++;
irc_privmsg(bot, chan, buf);
if (!strcmp(chan, "-stdio-"))
{
printf("Loaded %s\n", name);
}
else
{
irc_privmsg(bot, chan, "Loaded %s", name);
}
}
else
{
if (!strcmp(chan, "-stdio-"))
{
printf("Error: No load() function in %s\n", buf);
}
else
{
irc_privmsg(bot, chan, "Error: No load() function in %s", buf);
}
}
}
free(buf);
}
@ -128,9 +287,17 @@ MY_API void lua_unload_script(struct irc_conn *bot, char *user, char *host, char
luaL_unref(lua.L, LUA_REGISTRYINDEX, lua.scripts[i].unload);
lua.scripts[i].unload = -1;
remove_script(text);
sprintf(buf, "Unloaded %s", text);
irc_privmsg(bot, chan, buf);
while (i < lua.script_count)
{
lua.scripts[i] = lua.scripts[i + 1];
i++;
}
lua.script_count--;
free(buf);
@ -153,6 +320,8 @@ void lua_setvar(char *name, char *value)
MY_API void mod_init()
{
char *buf = (char *)malloc(sizeof(char *) * 500);
struct script_list list = get_scripts();
instance = get_bot();
lua.scripts = calloc(512, sizeof(struct lua_script));
@ -166,16 +335,42 @@ MY_API void mod_init()
lua_init_wrappers();
lua_init_events();
register_module("lua", "Aaron Blakely", "v0.1", "Lua module");
lua_init_handlers();
register_module("lua", "Aaron Blakely", "v0.1", "Lua module");
// load init.lua
if (luaL_loadfile(lua.L, "./mods/lua/init.lua") != LUA_OK)
{
printf("Error loading init.lua: %s\n", lua_tostring(lua.L, -1));
return;
}
if (lua_pcall(lua.L, 0, 0, 0) != LUA_OK)
{
printf("Error executing init.lua: %s\n", lua_tostring(lua.L, -1));
return;
}
for (int i = 0; i < list.count; i++)
{
printf("Loading %s\n", list.scripts[i]);
sprintf(buf, "!load %s", list.scripts[i]);
lua_load_script(instance, "lua", "localhost", "-stdio-", buf);
}
printf("Lua module loaded\n");
free(buf);
}
MY_API void mod_unload()
{
lua_close(lua.L);
free(lua.scripts);
free(lua.events);
unregister_module("lua");
del_handler(PRIVMSG_CHAN, lua_eval);
printf("Lua module unloaded\n");

View File

@ -34,6 +34,13 @@ struct lua_interp
struct lua_event *events;
};
struct script_list
{
int count;
char *scripts[512];
};
extern int block;
extern struct lua_interp lua;
extern struct irc_conn *instance;
@ -70,6 +77,10 @@ MY_API void quit_handler(struct irc_conn *bot, char *user, char *host, const cha
// lua.c
void lua_setvar(char *name, char *value);
int append_script(char *fname);
int remove_script(char *fname);
struct script_list get_scripts();
MY_API void lua_eval(struct irc_conn *bot, char *user, char *host, char *chan, const char *text);
MY_API void lua_load_script(struct irc_conn *bot, char *user, char *host, char *chan, const char *text);
MY_API void lua_unload_script(struct irc_conn *bot, char *user, char *host, char *chan, const char *text);

2
mods/lua/scripts.tmp Executable file
View File

@ -0,0 +1,2 @@
test.lua
hello.lua

View File

@ -3,20 +3,18 @@ VERSION = "v0.5"
AUTHOR = "Aaron Blakely"
DESCRIPTION = "A simple hello world script for xbot"
local handlers = {}
function hi(nick, host, chan, text)
privmsg(chan, "Hello, " .. nick .. "!")
privmsg(chan, "Hello, " .. nick .. " from hello.lua!!!!!")
end
function load()
-- register_script(NAME, VERSION, AUTHOR, DESCRIPTION)
table.insert(handlers, {PRIVMSG_CHAN, add_handler(PRIVMSG_CHAN, hi)})
add_handler(PRIVMSG_CHAN, hi)
end
function unload()
for i, v in ipairs(handlers) do
del_handler(v[1], v[2])
end
-- unregister_script(NAME)
del_handler(PRIVMSG_CHAN, hi)
end

View File

@ -1,23 +1,25 @@
NAME = "hello"
VERSION = "v0.5"
AUTHOR = "Aaron Blakely"
DESCRIPTION = "A simple hello world script for xbot"
local NAME = "hello"
local VERSION = "v0.5"
local AUTHOR = "Aaron Blakely"
local DESCRIPTION = "A simple hello world script for xbot"
local handlers = {}
function test(nick, host, chan, text)
-- check if text contains "hello"
function hi(nick, host, chan, text)
privmsg(chan, "Hello, " .. nick .. " from test.lua!")
if string.find(text, "hello") then
privmsg(chan, "Hello, " .. nick .. " from test.lua!!!")
end
end
function load()
-- register_script(NAME, VERSION, AUTHOR, DESCRIPTION)
table.insert(handlers, {PRIVMSG_CHAN, add_handler(PRIVMSG_CHAN, hi)})
add_handler(PRIVMSG_CHAN, test)
end
function unload()
privmsg("#lobby", "Unloading test.lua")
for i, v in ipairs(handlers) do
del_handler(v[1], v[2])
end
-- unregister_script(NAME)
del_handler(PRIVMSG_CHAN, test)
end