12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485 |
- #include <dirent.h>
- #include <stdlib.h>
- #include <string.h>
- #include <lub/string.h>
- #include "private.h"
- char *scripts_path = NULL;
- static bool_t
- load_scripts(lua_State *L, char *path)
- {
- struct dirent *entry;
- DIR *dir = opendir(path);
- bool_t result = BOOL_FALSE;
- int res = 0;
- const char *ext_lua = ".lua";
- const char *ext_bin = ".bin";
- if (!dir) {
- printf("%s: Failed to open '%s' directory\n", __func__, path);
- return BOOL_FALSE;
- }
- for (entry = readdir(dir); entry; entry = readdir(dir)) {
- const char *extension = strrchr(entry->d_name, '.');
- /* check the filename */
- if ((extension) &&
- ((!strcmp(ext_lua, extension)) ||
- (!strcmp(ext_bin, extension)))) {
- char *filename = NULL;
- lub_string_cat(&filename, path);
- lub_string_cat(&filename, "/");
- lub_string_cat(&filename, entry->d_name);
- result = BOOL_FALSE;
- if ((res = luaL_loadfile(L, filename))) {
- l_print_error(L, __func__, "load", res);
- } else if ((res = lua_pcall(L, 0, 0, 0))) {
- l_print_error(L, __func__, "exec", res);
- } else
- result = BOOL_TRUE;
- lub_string_free(filename);
- /* Shouldn't happen, but we can't be too sure ;-) */
- while(lua_gettop(L))
- lua_pop(L, 1);
- if (!result)
- break;
- }
- }
- closedir(dir);
- return result;
- }
- int clish_plugin_init_lua(clish_shell_t *shell)
- {
- lua_State *L = NULL;
- if (!(L = luaL_newstate())) {
- printf("%s: Failed to instantiate Lua interpreter\n", __func__);
- return (-1);
- }
- luaL_openlibs(L);
- if (scripts_path && !load_scripts(L, scripts_path)) {
- lub_string_free(scripts_path);
- return (-1);
- }
- lub_string_free(scripts_path);
- clish_shell__set_udata(shell, LUA_UDATA, L);
- lua_pushlightuserdata(L, shell);
- lua_setglobal(L, "clish_shell");
- return (0);
- }
|