lsup_lua/src/lua_namespace.c

#include "lua_lsup.h"

#define check_nsm(L) \
    *(LSUP_NSMap **)luaL_checkudata(L, 1, "LSUP.NSMap")


/*
 * Factory methods.
 */

int l_nsmap_new (lua_State *L)
{
    LSUP_NSMap **nsm_p = lua_newuserdatauv (L, sizeof (*nsm_p), 1);
    luaL_getmetatable (L, "LSUP.NSMap");
    lua_setmetatable (L, -2);

    *nsm_p = LSUP_nsmap_new ();
    if (!*nsm_p) return luaL_error (L, "Error while creating namespace map.");

    return 1;
}


/*
 * Class methods.
 */

static int l_nsmap_gc (lua_State *L)
{
    LSUP_NSMap *nsm = check_nsm (L);
    LOG_DEBUG ("Garbage collecting NSM @%p", nsm);
    if (nsm) LSUP_nsmap_free (nsm);

    return 0;
}


static int l_nsmap_add (lua_State *L)
{
    LSUP_NSMap *nsm = check_nsm (L);
    const char
        *pfx = luaL_checkstring (L, 2),
        *nsstr = luaL_checkstring (L, 3);

    LUA_PCHECK (
            LSUP_nsmap_add (nsm, pfx, nsstr),
            "Error adding member to NS map");

    return 0;
}


static int l_nsmap_remove (lua_State *L)
{
    LSUP_NSMap *nsm = check_nsm (L);
    const char *pfx = luaL_checkstring (L, 2);

    LUA_PCHECK (
            LSUP_nsmap_remove (nsm, pfx),
            "Error removing member to NS map");

    return 0;
}


static int l_nsmap_get_ns (lua_State *L)
{
    LSUP_NSMap *nsm = check_nsm (L);
    const char *pfx = luaL_checkstring (L, 2);

    const char *ns = LSUP_nsmap_get_ns (nsm, pfx);

    if (ns) lua_pushstring (L, ns);
    else return 0;

    return 1;
}


static int l_nsmap_get_pfx (lua_State *L)
{
    LSUP_NSMap *nsm = check_nsm (L);
    const char *ns = luaL_checkstring (L, 2);

    const char *pfx = LSUP_nsmap_get_pfx (nsm, ns);

    if (pfx) lua_pushstring (L, pfx);
    else lua_pushnil (L);

    return 1;
}


static int l_nsmap_normalize_uri (lua_State *L)
{
    LSUP_NSMap *nsm = check_nsm (L);
    const char *pfx_uri = luaL_checkstring (L, 2);

    char *fq_uri;
    LUA_PCHECK (
            LSUP_nsmap_normalize_uri (nsm, pfx_uri, &fq_uri),
            "Error normalizing URI");

    if (fq_uri) lua_pushstring (L, fq_uri);
    else return 0;

    free (fq_uri);
    return 1;
}


static int l_nsmap_denormalize_uri (lua_State *L)
{
    LSUP_NSMap *nsm = check_nsm (L);
    const char *fq_uri = luaL_checkstring (L, 2);

    char *pfx_uri;
    LUA_PCHECK (
            LSUP_nsmap_denormalize_uri (nsm, fq_uri, &pfx_uri),
            "Error denormalizing URI");

    if (pfx_uri) lua_pushstring (L, pfx_uri);
    else return 0;

    free (pfx_uri);
    return 1;
}


/*
static int l_nsmap_iter_next (lua_State *L)
{
    // TODO
    return 0;
}
*/

static int nsmap_iter_next (lua_State *L)
{
    const char ****data_p = lua_touserdata (L, lua_upvalueindex (1));
    const char ***data = *data_p;
    size_t *i = lua_touserdata (L, lua_upvalueindex (2));

    if ((data)[*i]) {
        lua_pushstring (L, data[*i][0]);
        lua_pushstring (L, data[*i][1]);
        (*i)++;

        return 2;
    } else {
        free (data);
        *data_p = NULL;
        free (i);

        return 0;
    }
}


static int l_nsmap_iter (lua_State *L)
{
    LSUP_NSMap *nsm = check_nsm (L);
    const char ****data_p = lua_newuserdata (L, sizeof (*data_p));
    //luaL_getmetatable (L, "LSUP.String2DIterator");
    //lua_setmetatable (L, -2);
    *data_p = LSUP_nsmap_dump (nsm);
    LUA_NLCHECK (*data_p, "Error creating NS map iterator.");

    size_t *i = malloc (sizeof (*i));
    LUA_NLCHECK (i, "Memory error in NS map iterator.");
    *i = 0;
    lua_pushlightuserdata (L, i);

    lua_pushcclosure (L, nsmap_iter_next, 2);

    return 1;
}


/*
 * Library setup.
 */

static const luaL_Reg nsmap_lib_fn [] = {
    {"new", l_nsmap_new},
    {NULL}
};


static const luaL_Reg nsmap_lib_meth [] = {
    {"__gc", l_nsmap_gc},
    {"__pairs", l_nsmap_iter},

    {"add", l_nsmap_add},
    {"remove", l_nsmap_remove},
    {"get_ns", l_nsmap_get_ns},
    {"get_pfx", l_nsmap_get_pfx},
    {"normalize_uri", l_nsmap_normalize_uri},
    {"denormalize_uri", l_nsmap_denormalize_uri},
    //{"iter", l_nsmap_iter},

    {NULL}
};


int luaopen_lsup_namespace (lua_State *L)
{
    luaL_newmetatable (L, "LSUP.NSMap");
    lua_pushvalue (L, -1);
    lua_setfield (L, -2, "__index");
    luaL_setfuncs (L, nsmap_lib_meth, 0);
    luaL_newlib (L, nsmap_lib_fn);

    //luaL_newmetatable (L, "LSUP.String2DIterator");

    return 1;
}