1 /*
2 ** Compat-5.1
3 ** Copyright Kepler Project 2004-2006 (http://www.keplerproject.org/compat)
4 ** $Id: compat-5.1.c 1621 2008-10-20 06:06:16Z oetiker $
5 */
7 #include <stdio.h>
8 #include <string.h>
9 #include "lua.h"
10 #include "lauxlib.h"
11 #include "compat-5.1.h"
13 static void getfield(lua_State *L, int idx, const char *name) {
14 const char *end = strchr(name, '.');
15 lua_pushvalue(L, idx);
16 while (end) {
17 lua_pushlstring(L, name, end - name);
18 lua_gettable(L, -2);
19 lua_remove(L, -2);
20 if (lua_isnil(L, -1)) return;
21 name = end+1;
22 end = strchr(name, '.');
23 }
24 lua_pushstring(L, name);
25 lua_gettable(L, -2);
26 lua_remove(L, -2);
27 }
29 static void setfield(lua_State *L, int idx, const char *name) {
30 const char *end = strchr(name, '.');
31 lua_pushvalue(L, idx);
32 while (end) {
33 lua_pushlstring(L, name, end - name);
34 lua_gettable(L, -2);
35 /* create table if not found */
36 if (lua_isnil(L, -1)) {
37 lua_pop(L, 1);
38 lua_newtable(L);
39 lua_pushlstring(L, name, end - name);
40 lua_pushvalue(L, -2);
41 lua_settable(L, -4);
42 }
43 lua_remove(L, -2);
44 name = end+1;
45 end = strchr(name, '.');
46 }
47 lua_pushstring(L, name);
48 lua_pushvalue(L, -3);
49 lua_settable(L, -3);
50 lua_pop(L, 2);
51 }
53 LUALIB_API void luaL_module(lua_State *L, const char *libname,
54 const luaL_reg *l, int nup) {
55 if (libname) {
56 getfield(L, LUA_GLOBALSINDEX, libname); /* check whether lib already exists */
57 if (lua_isnil(L, -1)) {
58 int env, ns;
59 lua_pop(L, 1); /* get rid of nil */
60 lua_pushliteral(L, "require");
61 lua_gettable(L, LUA_GLOBALSINDEX); /* look for require */
62 lua_getfenv(L, -1); /* getfenv(require) */
63 lua_remove(L, -2); /* remove function require */
64 env = lua_gettop(L);
66 lua_newtable(L); /* create namespace for lib */
67 ns = lua_gettop(L);
68 getfield(L, env, "package.loaded"); /* get package.loaded table */
69 if (lua_isnil(L, -1)) { /* create package.loaded table */
70 lua_pop(L, 1); /* remove previous result */
71 lua_newtable(L);
72 lua_pushvalue(L, -1);
73 setfield(L, env, "package.loaded");
74 }
75 else if (!lua_istable(L, -1))
76 luaL_error(L, "name conflict for library `%s'", libname);
77 lua_pushstring(L, libname);
78 lua_pushvalue(L, ns);
79 lua_settable(L, -3); /* package.loaded[libname] = ns */
80 lua_pop(L, 1); /* get rid of package.loaded table */
81 lua_pushvalue(L, ns); /* copy namespace */
82 setfield(L, LUA_GLOBALSINDEX, libname);
83 lua_remove (L, env); /* remove env */
84 }
85 lua_insert(L, -(nup+1)); /* move library table to below upvalues */
86 }
87 for (; l->name; l++) {
88 int i;
89 lua_pushstring(L, l->name);
90 for (i=0; i<nup; i++) /* copy upvalues to the top */
91 lua_pushvalue(L, -(nup+1));
92 lua_pushcclosure(L, l->func, nup);
93 lua_settable(L, -(nup+3));
94 }
95 lua_pop(L, nup); /* remove upvalues */
96 }