add uci.foreach()
[project/uci.git] / lua / uci.c
index 026569f..397a36d 100644 (file)
--- a/lua/uci.c
+++ b/lua/uci.c
@@ -14,6 +14,7 @@
 
 #include <sys/types.h>
 #include <sys/time.h>
 
 #include <sys/types.h>
 #include <sys/time.h>
+#include <stdbool.h>
 #include <stdlib.h>
 #include <string.h>
 #include <unistd.h>
 #include <stdlib.h>
 #include <string.h>
 #include <unistd.h>
@@ -62,19 +63,70 @@ done:
        lua_pop(L, 2);
 }
 
        lua_pop(L, 2);
 }
 
+static void uci_push_section(lua_State *L, struct uci_section *s)
+{
+       struct uci_element *e;
+
+       lua_newtable(L);
+       lua_pushstring(L, s->type);
+       lua_setfield(L, -2, "type");
+       lua_pushstring(L, s->e.name);
+       lua_setfield(L, -2, "name");
+
+       lua_newtable(L);
+       lua_pushvalue(L, -1);
+       lua_setfield(L, -3, "options");
+
+       uci_foreach_element(&s->options, e) {
+               struct uci_option *o = uci_to_option(e);
+               lua_pushstring(L, o->value);
+               lua_setfield(L, -2, o->e.name);
+       }
+       lua_pop(L, 1);
+}
+
+static void uci_push_package(lua_State *L, struct uci_package *p)
+{
+       struct uci_element *e;
+       int i = 0;
+
+       lua_newtable(L);
+       uci_foreach_element(&p->sections, e) {
+               i++;
+               luaL_setn(L, -1, i);
+               uci_push_section(L, uci_to_section(e));
+               lua_rawseti(L, -2, i);
+       }
+}
+
 static int
 static int
-uci_lua_load(lua_State *L)
+uci_lua_unload(lua_State *L)
 {
        struct uci_package *p;
        const char *s;
 
 {
        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);
        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);
        if (uci_load(ctx, s, &p)) {
                uci_lua_perror(L, "uci.load");
                lua_pushboolean(L, 0);
@@ -85,10 +137,49 @@ uci_lua_load(lua_State *L)
        return 1;
 }
 
        return 1;
 }
 
+
 static int
 static int
-uci_lua_get(lua_State *L)
+uci_lua_foreach(lua_State *L)
+{
+       struct uci_package *p;
+       struct uci_element *e;
+       const char *package, *type;
+       bool ret = false;
+
+       package = luaL_checkstring(L, 1);
+
+       if (lua_isnil(L, 2))
+               type = NULL;
+       else
+               type = luaL_checkstring(L, 2);
+
+       if (!lua_isfunction(L, 3) || !package)
+               luaL_error(L, "Invalid argument");
+
+       p = find_package(package);
+       if (!p)
+               goto done;
+
+       uci_foreach_element(&p->sections, e) {
+               struct uci_section *s = uci_to_section(e);
+
+               if (type && (strcmp(s->type, type) != 0))
+                       continue;
+
+               lua_pushvalue(L, 3); /* iterator function */
+               uci_push_section(L, s);
+               if (lua_pcall(L, 1, 0, 0) == 0)
+                       ret = true;
+       }
+
+done:
+       lua_pushboolean(L, ret);
+       return 1;
+}
+
+static int
+uci_lua_get_any(lua_State *L, bool all)
 {
 {
-       struct uci_lua_context *f;
        struct uci_element *e = NULL;
        struct uci_package *p = NULL;
        char *package = NULL;
        struct uci_element *e = NULL;
        struct uci_package *p = NULL;
        char *package = NULL;
@@ -97,7 +188,7 @@ uci_lua_get(lua_State *L)
        char *s;
        int err = UCI_ERR_MEM;
 
        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;
        s = strdup(lua_tostring(L, -1));
        if (!s)
                goto error;
@@ -105,7 +196,7 @@ uci_lua_get(lua_State *L)
        if ((err = uci_parse_tuple(ctx, s, &package, &section, &option, NULL)))
                goto error;
 
        if ((err = uci_parse_tuple(ctx, s, &package, &section, &option, NULL)))
                goto error;
 
-       if (section == NULL) {
+       if (!all && (section == NULL)) {
                err = UCI_ERR_INVAL;
                goto error;
        }
                err = UCI_ERR_INVAL;
                goto error;
        }
@@ -116,12 +207,22 @@ uci_lua_get(lua_State *L)
                goto error;
        }
 
                goto error;
        }
 
-       if ((err = uci_lookup(ctx, &e, p, section, option)))
-               goto error;
+       if (section) {
+               if ((err = uci_lookup(ctx, &e, p, section, option)))
+                       goto error;
+       } else {
+               e = &p->e;
+       }
 
        switch(e->type) {
 
        switch(e->type) {
+               case UCI_TYPE_PACKAGE:
+                       uci_push_package(L, p);
+                       break;
                case UCI_TYPE_SECTION:
                case UCI_TYPE_SECTION:
-                       lua_pushstring(L, uci_to_section(e)->type);
+                       if (all)
+                               uci_push_section(L, uci_to_section(e));
+                       else
+                               lua_pushstring(L, uci_to_section(e)->type);
                        break;
                case UCI_TYPE_OPTION:
                        lua_pushstring(L, uci_to_option(e)->value);
                        break;
                case UCI_TYPE_OPTION:
                        lua_pushstring(L, uci_to_option(e)->value);
@@ -147,10 +248,179 @@ error:
        }
 }
 
        }
 }
 
+static int
+uci_lua_get(lua_State *L)
+{
+       return uci_lua_get_any(L, false);
+}
+
+static int
+uci_lua_get_all(lua_State *L)
+{
+       return uci_lua_get_any(L, true);
+}
+
+static int
+uci_lua_set(lua_State *L)
+{
+       struct uci_package *p;
+       char *package = NULL;
+       char *section = NULL;
+       char *option = NULL;
+       char *value = NULL;
+       char *s;
+       int err = UCI_ERR_MEM;
+
+       luaL_checkstring(L, 1);
+       s = strdup(lua_tostring(L, -1));
+       if (!s)
+               goto error;
+
+       if ((err = uci_parse_tuple(ctx, s, &package, &section, &option, &value)))
+               goto error;
+
+       if ((section == NULL) || (value == NULL)) {
+               err = UCI_ERR_INVAL;
+               goto error;
+       }
+
+       p = find_package(package);
+       if (!p) {
+               err = UCI_ERR_NOTFOUND;
+               goto error;
+       }
+       err = uci_set(ctx, p, section, option, value, NULL);
+
+error:
+       if (err)
+               uci_lua_perror(L, "uci.set");
+       lua_pushboolean(L, (err == 0));
+       return 1;
+}
+
+enum pkg_cmd {
+       CMD_SAVE,
+       CMD_COMMIT,
+       CMD_REVERT
+};
+
+static int
+uci_lua_package_cmd(lua_State *L, enum pkg_cmd cmd)
+{
+       struct uci_element *e, *tmp;
+       const char *s = NULL;
+       const char *section = NULL;
+       const char *option = NULL;
+       int failed = 0;
+       int nargs;
+
+       nargs = lua_gettop(L);
+       switch(nargs) {
+       case 3:
+               if (cmd != CMD_REVERT)
+                       goto err;
+               luaL_checkstring(L, 1);
+               option = lua_tostring(L, -1);
+               lua_pop(L, 1);
+               /* fall through */
+       case 2:
+               if (cmd != CMD_REVERT)
+                       goto err;
+               luaL_checkstring(L, 1);
+               section = lua_tostring(L, -1);
+               lua_pop(L, 1);
+               /* fall through */
+       case 1:
+               luaL_checkstring(L, 1);
+               s = lua_tostring(L, -1);
+               lua_pop(L, 1);
+               break;
+       default:
+               err:
+               luaL_error(L, "Invalid argument count");
+               break;
+       }
+
+       uci_foreach_element_safe(&ctx->root, tmp, e) {
+               struct uci_package *p = uci_to_package(e);
+               int ret = UCI_ERR_INVAL;
+
+               if (s && (strcmp(s, e->name) != 0))
+                       continue;
+
+               switch(cmd) {
+               case CMD_COMMIT:
+                       ret = uci_commit(ctx, &p, false);
+                       break;
+               case CMD_SAVE:
+                       ret = uci_save(ctx, p);
+                       break;
+               case CMD_REVERT:
+                       ret = uci_revert(ctx, &p, section, option);
+                       break;
+               }
+
+               if (ret != 0)
+                       failed = 1;
+       }
+
+       lua_pushboolean(L, !failed);
+       return 1;
+}
+
+static int
+uci_lua_save(lua_State *L)
+{
+       return uci_lua_package_cmd(L, CMD_SAVE);
+}
+
+static int
+uci_lua_commit(lua_State *L)
+{
+       return uci_lua_package_cmd(L, CMD_COMMIT);
+}
+
+static int
+uci_lua_revert(lua_State *L)
+{
+       return uci_lua_package_cmd(L, CMD_REVERT);
+}
+
+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 },
 
 static const luaL_Reg uci[] = {
        { "load", uci_lua_load },
+       { "unload", uci_lua_unload },
        { "get", uci_lua_get },
        { "get", uci_lua_get },
+       { "get_all", uci_lua_get_all },
+       { "set", uci_lua_set },
+       { "save", uci_lua_save },
+       { "commit", uci_lua_commit },
+       { "revert", uci_lua_revert },
+       { "foreach", uci_lua_foreach },
+       { "set_confdir", uci_lua_set_confdir },
+       { "set_savedir", uci_lua_set_savedir },
        { NULL, NULL },
 };
 
        { NULL, NULL },
 };