| rev |
line source |
|
jbe@0
|
1 #include <lua.h>
|
|
jbe@0
|
2 #include <lauxlib.h>
|
|
jbe@25
|
3 #include "seqlualib.h"
|
|
jbe@35
|
4 #include <string.h>
|
|
jbe@8
|
5
|
|
jbe@32
|
6 static int seqlua_ipairsaux_raw(lua_State *L) {
|
|
jbe@32
|
7 lua_Integer i;
|
|
jbe@32
|
8 luaL_checktype(L, 1, LUA_TTABLE);
|
|
jbe@32
|
9 i = luaL_checkinteger(L, 2) + 1;
|
|
jbe@32
|
10 lua_pushinteger(L, i);
|
|
jbe@32
|
11 lua_rawgeti(L, 1, i); // TODO: Lua 5.3 returns type
|
|
jbe@32
|
12 return lua_isnil(L, -1) ? 1 : 2;
|
|
jbe@32
|
13 }
|
|
jbe@32
|
14
|
|
jbe@30
|
15 static int seqlua_ipairsaux_index(lua_State *L) {
|
|
jbe@30
|
16 lua_Integer i = luaL_checkinteger(L, 2) + 1;
|
|
jbe@0
|
17 lua_pushinteger(L, i);
|
|
jbe@0
|
18 lua_pushinteger(L, i);
|
|
jbe@0
|
19 lua_gettable(L, 1); // TODO: Lua 5.3 returns type
|
|
jbe@32
|
20 return lua_isnil(L, -1) ? 1 : 2;
|
|
jbe@0
|
21 }
|
|
jbe@0
|
22
|
|
jbe@30
|
23 static int seqlua_ipairsaux_call(lua_State *L) {
|
|
jbe@0
|
24 lua_pushinteger(L, luaL_checkinteger(L, 2) + 1);
|
|
jbe@30
|
25 lua_insert(L, 1); // integer on stack index 1
|
|
jbe@30
|
26 lua_settop(L, 2); // function on stack index 2
|
|
jbe@0
|
27 lua_call(L, 0, LUA_MULTRET);
|
|
jbe@0
|
28 if (lua_isnoneornil(L, 2)) {
|
|
jbe@0
|
29 lua_settop(L, 0);
|
|
jbe@0
|
30 lua_pushnil(L);
|
|
jbe@0
|
31 return 1;
|
|
jbe@0
|
32 } else {
|
|
jbe@0
|
33 return lua_gettop(L);
|
|
jbe@0
|
34 }
|
|
jbe@0
|
35 }
|
|
jbe@0
|
36
|
|
jbe@0
|
37 static int seqlua_ipairs(lua_State *L) {
|
|
jbe@52
|
38 const char *method;
|
|
jbe@52
|
39 int generated = 0;
|
|
jbe@52
|
40 seqlua_ipairs_repeat:
|
|
jbe@30
|
41 if (luaL_getmetafield(L, 1, "__ipairs")) {
|
|
jbe@30
|
42 lua_pushvalue(L, 1);
|
|
jbe@30
|
43 lua_call(L, 1, 3);
|
|
jbe@35
|
44 if (lua_type(L, -3) == LUA_TSTRING) {
|
|
jbe@35
|
45 const char *method = lua_tostring(L, -3);
|
|
jbe@35
|
46 if (!strcmp(method, "raw")) {
|
|
jbe@35
|
47 lua_pushcfunction(L, seqlua_ipairsaux_raw);
|
|
jbe@35
|
48 } else if (!strcmp(method, "index")) {
|
|
jbe@35
|
49 lua_pushcfunction(L, seqlua_ipairsaux_index);
|
|
jbe@35
|
50 } else if (!strcmp(method, "call")) {
|
|
jbe@35
|
51 lua_pushcfunction(L, seqlua_ipairsaux_call);
|
|
jbe@35
|
52 } else {
|
|
jbe@35
|
53 luaL_error(L, "Unexpected string returned by __ipairs metamethod");
|
|
jbe@35
|
54 }
|
|
jbe@35
|
55 lua_pushvalue(L, -3);
|
|
jbe@35
|
56 lua_pushinteger(L, 0);
|
|
jbe@35
|
57 }
|
|
jbe@0
|
58 } else {
|
|
jbe@30
|
59 int t = lua_type(L, 1);
|
|
jbe@30
|
60 if (t == LUA_TFUNCTION) {
|
|
jbe@52
|
61 if (generated) goto seqlua_ipairs_call;
|
|
jbe@52
|
62 method = lua_tostring(L, 2);
|
|
jbe@52
|
63 if (method) {
|
|
jbe@52
|
64 if (!strcmp(method, "call")) {
|
|
jbe@52
|
65 seqlua_ipairs_call:
|
|
jbe@52
|
66 lua_pushcfunction(L, seqlua_ipairsaux_call);
|
|
jbe@52
|
67 goto seqlua_ipairs_finish;
|
|
jbe@52
|
68 } else if (!strcmp(method, "generator")) {
|
|
jbe@52
|
69 lua_settop(L, 1);
|
|
jbe@52
|
70 lua_call(L, 0, 1);
|
|
jbe@52
|
71 generated = 1;
|
|
jbe@52
|
72 goto seqlua_ipairs_repeat;
|
|
jbe@52
|
73 }
|
|
jbe@52
|
74 }
|
|
jbe@32
|
75 } else if (luaL_getmetafield(L, 1, "__index")) {
|
|
jbe@32
|
76 lua_pushcfunction(L, seqlua_ipairsaux_index);
|
|
jbe@52
|
77 goto seqlua_ipairs_finish;
|
|
jbe@30
|
78 }
|
|
jbe@52
|
79 luaL_checktype(L, 1, LUA_TTABLE);
|
|
jbe@52
|
80 lua_pushcfunction(L, seqlua_ipairsaux_raw);
|
|
jbe@52
|
81 seqlua_ipairs_finish:
|
|
jbe@30
|
82 lua_pushvalue(L, 1);
|
|
jbe@30
|
83 lua_pushinteger(L, 0);
|
|
jbe@0
|
84 }
|
|
jbe@0
|
85 return 3;
|
|
jbe@0
|
86 }
|
|
jbe@0
|
87
|
|
jbe@25
|
88 static int seqlua_concat(lua_State *L) {
|
|
jbe@25
|
89 const char *sep;
|
|
jbe@25
|
90 size_t seplen;
|
|
jbe@25
|
91 luaL_Buffer buf;
|
|
jbe@25
|
92 seqlua_Iterator iter;
|
|
jbe@25
|
93 sep = luaL_checklstring(L, 1, &seplen);
|
|
jbe@25
|
94 luaL_checkany(L, 2);
|
|
jbe@25
|
95 lua_settop(L, 3);
|
|
jbe@25
|
96 luaL_buffinit(L, &buf);
|
|
jbe@52
|
97 seqlua_iterloop(L, &iter, SEQLUA_MODE_CALL, 2) {
|
|
jbe@25
|
98 lua_replace(L, 3);
|
|
jbe@25
|
99 if (seqlua_itercount(&iter) > 1) luaL_addlstring(&buf, sep, seplen);
|
|
jbe@25
|
100 luaL_tolstring(L, 3, NULL);
|
|
jbe@25
|
101 luaL_addvalue(&buf);
|
|
jbe@25
|
102 }
|
|
jbe@25
|
103 luaL_pushresult(&buf);
|
|
jbe@25
|
104 return 1;
|
|
jbe@25
|
105 }
|
|
jbe@25
|
106
|
|
jbe@0
|
107 int luaopen_seqlua(lua_State *L) {
|
|
jbe@0
|
108 lua_pushcfunction(L, seqlua_ipairs);
|
|
jbe@0
|
109 lua_setglobal(L, "ipairs");
|
|
jbe@25
|
110 lua_getglobal(L, "string");
|
|
jbe@25
|
111 lua_pushcfunction(L, seqlua_concat);
|
|
jbe@25
|
112 lua_setfield(L, -2, "concat");
|
|
jbe@25
|
113 return 0;
|
|
jbe@0
|
114 }
|