typechecking, add set_confdir and set_savedir
[project/uci.git] / lua / uci.c
index 026569f..c69cf83 100644 (file)
--- a/lua/uci.c
+++ b/lua/uci.c
@@ -63,18 +63,33 @@ done:
 }
 
 static int
-uci_lua_load(lua_State *L)
+uci_lua_unload(lua_State *L)
 {
        struct uci_package *p;
        const char *s;
 
-       luaL_checkany(L, 1);
+       luaL_checkstring(L, 1);
        s = lua_tostring(L, -1);
        p = find_package(s);
        if (p) {
                uci_unload(ctx, p);
-               p = NULL;
+               lua_pushboolean(L, 1);
+       } else {
+               lua_pushboolean(L, 0);
        }
+       return 1;
+}
+
+static int
+uci_lua_load(lua_State *L)
+{
+       struct uci_package *p = NULL;
+       const char *s;
+
+       uci_lua_unload(L);
+       lua_pop(L, 1); /* bool ret value of unload */
+       s = lua_tostring(L, -1);
+
        if (uci_load(ctx, s, &p)) {
                uci_lua_perror(L, "uci.load");
                lua_pushboolean(L, 0);
@@ -97,7 +112,7 @@ uci_lua_get(lua_State *L)
        char *s;
        int err = UCI_ERR_MEM;
 
-       luaL_checkany(L, 1);
+       luaL_checkstring(L, 1);
        s = strdup(lua_tostring(L, -1));
        if (!s)
                goto error;
@@ -148,9 +163,35 @@ error:
 }
 
 
+static int
+uci_lua_set_confdir(lua_State *L)
+{
+       int ret;
+
+       luaL_checkstring(L, 1);
+       ret = uci_set_confdir(ctx, lua_tostring(L, -1));
+       lua_pushboolean(L, (ret == 0));
+       return 1;
+}
+
+static int
+uci_lua_set_savedir(lua_State *L)
+{
+       int ret;
+
+       luaL_checkstring(L, 1);
+       ret = uci_set_savedir(ctx, lua_tostring(L, -1));
+       lua_pushboolean(L, (ret == 0));
+
+       return 1;
+}
+
 static const luaL_Reg uci[] = {
        { "load", uci_lua_load },
+       { "unload", uci_lua_unload },
        { "get", uci_lua_get },
+       { "set_confdir", uci_lua_set_confdir },
+       { "set_savedir", uci_lua_set_savedir },
        { NULL, NULL },
 };