]>
Commit | Line | Data |
---|---|---|
1 | #include "lua.h" | |
2 | #include "luabot.h" | |
3 | ||
4 | #include "../config.h" | |
5 | ||
6 | #ifdef HAVE_PGSQL | |
7 | #include "../dbapi/dbapi.h" | |
8 | ||
9 | #warning Needs testing with new db API. | |
10 | ||
11 | static int lua_dbcreatequery(lua_State *ps) { | |
12 | char *s = (char *)lua_tostring(ps, 1); | |
13 | ||
14 | if(!s) | |
15 | LUA_RETURN(ps, LUA_FAIL); | |
16 | ||
17 | dbcreatequery("%s", s); | |
18 | LUA_RETURN(ps, LUA_OK); | |
19 | } | |
20 | ||
21 | /* TODO */ | |
22 | /* | |
23 | static int lua_dbloadtable(lua_State *ps) { | |
24 | lua_list *l = lua_listfromstate(ps); | |
25 | if(!l) | |
26 | LUA_RETURN(ps, LUA_FAIL); | |
27 | ||
28 | } | |
29 | */ | |
30 | ||
31 | typedef struct lua_callback { | |
32 | lua_list *l; | |
33 | long handler, args; | |
34 | } lua_callback; | |
35 | ||
36 | /* hack */ | |
37 | DBResult *pgres; | |
38 | ||
39 | static int lua_dbnumfields(lua_State *ps) { | |
40 | lua_pushint(ps, dbnumfields(pgres)); | |
41 | ||
42 | return 1; | |
43 | } | |
44 | ||
45 | /* TODO */ | |
46 | /* | |
47 | static int lua_dbgetrow(lua_State *ps) { | |
48 | char *r; | |
49 | int tuple, field, len; | |
50 | ||
51 | if(!lua_isint(ps, 1)) | |
52 | return 0; | |
53 | ||
54 | lua_pushstring(ps, dbgetvalue(pgres, lua_toint(ps, 1)), len); | |
55 | ||
56 | return 1; | |
57 | } | |
58 | */ | |
59 | ||
60 | void lua_dbcallback(DBConn *dbconn, void *arg) { | |
61 | pgres = dbgetresult(dbconn); | |
62 | lua_callback *c = (lua_callback *)arg; | |
63 | ||
64 | if(!lua_listexists(c->l)) { | |
65 | luafree(c); | |
66 | return; | |
67 | } | |
68 | ||
69 | if(!dbquerysuccessful(pgres)) { | |
70 | _lua_vpcall(c->l->l, (void *)c->handler, LUA_POINTERMODE, "bR", 0, c->args); | |
71 | ||
72 | luaL_unref(c->l->l, LUA_REGISTRYINDEX, c->handler); | |
73 | luaL_unref(c->l->l, LUA_REGISTRYINDEX, c->args); | |
74 | luafree(c); | |
75 | return; | |
76 | } | |
77 | ||
78 | _lua_vpcall(c->l->l, (void *)c->handler, LUA_POINTERMODE, "bR", 1, c->args); | |
79 | ||
80 | luaL_unref(c->l->l, LUA_REGISTRYINDEX, c->handler); | |
81 | luaL_unref(c->l->l, LUA_REGISTRYINDEX, c->args); | |
82 | luafree(c); | |
83 | dbclear(pgres); | |
84 | pgres = NULL; | |
85 | } | |
86 | ||
87 | static int lua_dbquery(lua_State *ps) { | |
88 | lua_list *l = lua_listfromstate(ps); | |
89 | char *q = (char *)lua_tostring(ps, 1); | |
90 | lua_callback *cb; | |
91 | ||
92 | /* what happens if 3 isn't there? */ | |
93 | if(!l || !q) | |
94 | LUA_RETURN(ps, LUA_FAIL); | |
95 | ||
96 | if(!lua_isfunction(ps, 2)) { | |
97 | dbquery("%s", q); | |
98 | LUA_RETURN(ps, LUA_OK); | |
99 | } | |
100 | ||
101 | cb = (lua_callback *)luamalloc(sizeof(lua_callback)); | |
102 | if(!cb) | |
103 | LUA_RETURN(ps, LUA_FAIL); | |
104 | ||
105 | cb->l = l; | |
106 | cb->args = luaL_ref(ps, LUA_REGISTRYINDEX); | |
107 | cb->handler = luaL_ref(ps, LUA_REGISTRYINDEX); | |
108 | ||
109 | dbasyncquery(lua_dbcallback, cb, "%s", q); | |
110 | ||
111 | LUA_RETURN(ps, LUA_OK); | |
112 | } | |
113 | ||
114 | static int lua_dbescape(lua_State *ps) { | |
115 | char ebuf[8192 * 2 + 1]; | |
116 | char *s = (char *)lua_tostring(ps, 1); | |
117 | int len; | |
118 | ||
119 | if(!s) | |
120 | return 0; | |
121 | ||
122 | len = lua_strlen(ps, 1); | |
123 | if(len > sizeof(ebuf) / 2 - 5) | |
124 | return 0; | |
125 | ||
126 | dbescapestring(ebuf, s, len); | |
127 | lua_pushstring(ps, ebuf); | |
128 | ||
129 | return 1; | |
130 | } | |
131 | ||
132 | void lua_registerdbcommands(lua_State *l) { | |
133 | lua_register(l, "db_createquery", lua_dbcreatequery); | |
134 | ||
135 | /* lua_register(l, "db_loadtable", lua_dbloadtable); */ | |
136 | ||
137 | lua_register(l, "db_query", lua_dbquery); | |
138 | lua_register(l, "db_escape", lua_dbescape); | |
139 | lua_register(l, "db_numfields", lua_dbnumfields); | |
140 | /* lua_register(l, "db_getvalue", lua_dbgetvalue);*/ | |
141 | } | |
142 | ||
143 | #else | |
144 | ||
145 | void lua_registerdbcommands(lua_State *l) { | |
146 | } | |
147 | ||
148 | #endif |