webmcp

annotate libraries/mondelefant/mondelefant_native.c @ 28:ea2e8f3a2776

find changesets by author, revision, files, or words in the commit message
allow webmcp path to be set in cgi script

this allows another script to include the webmcp script when the cwd is not the cgi-bin. the script needs to set the WEBMCP_PATH variable.
author Daniel Poelzleithner <poelzi@poelzi.org>
date Sun Sep 19 01:36:08 2010 +0200 (2010-09-19)
parents 3a6fe8663b26
children ed00b972f40e
rev   line source
jbe/bsw@0 1 #include <lua.h>
jbe/bsw@0 2 #include <lauxlib.h>
jbe/bsw@0 3 #include <libpq-fe.h>
jbe/bsw@0 4 #include <postgres.h>
jbe/bsw@0 5 #include <catalog/pg_type.h>
jbe/bsw@0 6 #include <stdint.h>
jbe/bsw@0 7
jbe@23 8 // NOTE: Comments with format "// <number>" denote the Lua stack position
jbe@23 9
jbe@23 10 // prefix for all Lua registry entries of this library:
jbe@23 11 #define MONDELEFANT_REGKEY "e449ba8d9a53d353_mondelefant_"
jbe/bsw@0 12
jbe@23 13 // registry key of module "mondelefant_native":
jbe@23 14 #define MONDELEFANT_MODULE_REGKEY (MONDELEFANT_REGKEY "module")
jbe@23 15 // registry key of meta-table for database connections:
jbe@23 16 #define MONDELEFANT_CONN_MT_REGKEY (MONDELEFANT_REGKEY "connection")
jbe@23 17 // registry key of table storing connection specific data:
jbe@23 18 #define MONDELEFANT_CONN_DATA_REGKEY (MONDELEFANT_REGKEY "connection_data")
jbe@23 19 // registry key of meta-table for database result lists and objects:
jbe@23 20 #define MONDELEFANT_RESULT_MT_REGKEY (MONDELEFANT_REGKEY "result")
jbe@23 21 // registry key of meta-table for database error objects:
jbe@23 22 #define MONDELEFANT_ERROROBJECT_MT_REGKEY (MONDELEFANT_REGKEY "errorobject")
jbe@23 23 // registry key of meta-table for models (named classes here):
jbe@23 24 #define MONDELEFANT_CLASS_MT_REGKEY (MONDELEFANT_REGKEY "class")
jbe@23 25 // registry key of default prototype for models/classes:
jbe@23 26 #define MONDELEFANT_CLASS_PROTO_REGKEY (MONDELEFANT_REGKEY "class_proto")
jbe/bsw@0 27
jbe@23 28 // C-structure for database connection userdata:
jbe/bsw@0 29 typedef struct {
jbe/bsw@0 30 PGconn *pgconn;
jbe/bsw@0 31 int server_encoding;
jbe/bsw@0 32 } mondelefant_conn_t;
jbe@23 33 #define MONDELEFANT_SERVER_ENCODING_ASCII 0
jbe@23 34 #define MONDELEFANT_SERVER_ENCODING_UTF8 1
jbe/bsw@0 35
jbe@23 36 // transform codepoint-position to byte-position for a given UTF-8 string:
jbe/bsw@0 37 static size_t utf8_position_to_byte(const char *str, size_t utf8pos) {
jbe/bsw@0 38 size_t bytepos;
jbe/bsw@0 39 for (bytepos = 0; utf8pos > 0; bytepos++) {
jbe/bsw@0 40 uint8_t c;
jbe/bsw@0 41 c = ((const uint8_t *)str)[bytepos];
jbe/bsw@0 42 if (!c) break;
jbe/bsw@0 43 if (c <= 0x7f || c >= 0xc0) utf8pos--;
jbe/bsw@0 44 }
jbe/bsw@0 45 return bytepos;
jbe/bsw@0 46 }
jbe/bsw@0 47
jbe@23 48 // PostgreSQL's OID for binary data type (bytea):
jbe/bsw@0 49 #define MONDELEFANT_POSTGRESQL_BINARY_OID ((Oid)17)
jbe/bsw@0 50
jbe@23 51 // mapping a PostgreSQL type given by its OID to a string identifier:
jbe/bsw@0 52 static const char *mondelefant_oid_to_typestr(Oid oid) {
jbe/bsw@0 53 switch (oid) {
jbe/bsw@0 54 case 16: return "bool";
jbe/bsw@0 55 case 17: return "bytea";
jbe/bsw@0 56 case 18: return "char";
jbe/bsw@0 57 case 19: return "name";
jbe/bsw@0 58 case 20: return "int8";
jbe/bsw@0 59 case 21: return "int2";
jbe/bsw@0 60 case 23: return "int4";
jbe/bsw@0 61 case 25: return "text";
jbe/bsw@0 62 case 26: return "oid";
jbe/bsw@0 63 case 27: return "tid";
jbe/bsw@0 64 case 28: return "xid";
jbe/bsw@0 65 case 29: return "cid";
jbe/bsw@0 66 case 600: return "point";
jbe/bsw@0 67 case 601: return "lseg";
jbe/bsw@0 68 case 602: return "path";
jbe/bsw@0 69 case 603: return "box";
jbe/bsw@0 70 case 604: return "polygon";
jbe/bsw@0 71 case 628: return "line";
jbe/bsw@0 72 case 700: return "float4";
jbe/bsw@0 73 case 701: return "float8";
jbe/bsw@0 74 case 705: return "unknown";
jbe/bsw@0 75 case 718: return "circle";
jbe/bsw@0 76 case 790: return "money";
jbe/bsw@0 77 case 829: return "macaddr";
jbe/bsw@0 78 case 869: return "inet";
jbe/bsw@0 79 case 650: return "cidr";
jbe/bsw@0 80 case 1042: return "bpchar";
jbe/bsw@0 81 case 1043: return "varchar";
jbe/bsw@0 82 case 1082: return "date";
jbe/bsw@0 83 case 1083: return "time";
jbe/bsw@0 84 case 1114: return "timestamp";
jbe/bsw@0 85 case 1184: return "timestamptz";
jbe/bsw@0 86 case 1186: return "interval";
jbe/bsw@0 87 case 1266: return "timetz";
jbe/bsw@0 88 case 1560: return "bit";
jbe/bsw@0 89 case 1562: return "varbit";
jbe/bsw@0 90 case 1700: return "numeric";
jbe/bsw@0 91 default: return NULL;
jbe/bsw@0 92 }
jbe/bsw@0 93 }
jbe/bsw@0 94
jbe@23 95 // This library maps PostgreSQL's error codes to CamelCase string
jbe@23 96 // identifiers, which consist of CamelCase identifiers and are seperated
jbe@23 97 // by dots (".") (no leading or trailing dots).
jbe@23 98 // There are additional error identifiers which do not have a corresponding
jbe@23 99 // PostgreSQL error associated with it.
jbe@23 100
jbe@23 101 // matching start of local variable 'pgcode' against string 'incode',
jbe@23 102 // returning string 'outcode' on match:
jbe/bsw@0 103 #define mondelefant_errcode_item(incode, outcode) \
jbe/bsw@0 104 if (!strncmp(pgcode, (incode), strlen(incode))) return outcode; else
jbe/bsw@0 105
jbe@23 106 // additional error identifiers without corresponding PostgreSQL error:
jbe/bsw@0 107 #define MONDELEFANT_ERRCODE_UNKNOWN "unknown"
jbe/bsw@0 108 #define MONDELEFANT_ERRCODE_CONNECTION "ConnectionException"
jbe/bsw@0 109 #define MONDELEFANT_ERRCODE_RESULTCOUNT_LOW "WrongResultSetCount.ResultSetMissing"
jbe/bsw@0 110 #define MONDELEFANT_ERRCODE_RESULTCOUNT_HIGH "WrongResultSetCount.TooManyResults"
jbe/bsw@0 111 #define MONDELEFANT_ERRCODE_QUERY1_NO_ROWS "NoData.OneRowExpected"
jbe/bsw@0 112 #define MONDELEFANT_ERRCODE_QUERY1_MULTIPLE_ROWS "CardinalityViolation.OneRowExpected"
jbe/bsw@0 113
jbe@23 114 // mapping PostgreSQL error code to error code as returned by this library:
jbe/bsw@0 115 static const char *mondelefant_translate_errcode(const char *pgcode) {
jbe/bsw@0 116 if (!pgcode) abort(); // should not happen
jbe/bsw@0 117 mondelefant_errcode_item("02", "NoData")
jbe/bsw@0 118 mondelefant_errcode_item("03", "SqlStatementNotYetComplete")
jbe/bsw@0 119 mondelefant_errcode_item("08", "ConnectionException")
jbe/bsw@0 120 mondelefant_errcode_item("09", "TriggeredActionException")
jbe/bsw@0 121 mondelefant_errcode_item("0A", "FeatureNotSupported")
jbe/bsw@0 122 mondelefant_errcode_item("0B", "InvalidTransactionInitiation")
jbe/bsw@0 123 mondelefant_errcode_item("0F", "LocatorException")
jbe/bsw@0 124 mondelefant_errcode_item("0L", "InvalidGrantor")
jbe/bsw@0 125 mondelefant_errcode_item("0P", "InvalidRoleSpecification")
jbe/bsw@0 126 mondelefant_errcode_item("21", "CardinalityViolation")
jbe/bsw@0 127 mondelefant_errcode_item("22", "DataException")
jbe/bsw@0 128 mondelefant_errcode_item("23001", "IntegrityConstraintViolation.RestrictViolation")
jbe/bsw@0 129 mondelefant_errcode_item("23502", "IntegrityConstraintViolation.NotNullViolation")
jbe/bsw@0 130 mondelefant_errcode_item("23503", "IntegrityConstraintViolation.ForeignKeyViolation")
jbe/bsw@0 131 mondelefant_errcode_item("23505", "IntegrityConstraintViolation.UniqueViolation")
jbe/bsw@0 132 mondelefant_errcode_item("23514", "IntegrityConstraintViolation.CheckViolation")
jbe/bsw@0 133 mondelefant_errcode_item("23", "IntegrityConstraintViolation")
jbe/bsw@0 134 mondelefant_errcode_item("24", "InvalidCursorState")
jbe/bsw@0 135 mondelefant_errcode_item("25", "InvalidTransactionState")
jbe/bsw@0 136 mondelefant_errcode_item("26", "InvalidSqlStatementName")
jbe/bsw@0 137 mondelefant_errcode_item("27", "TriggeredDataChangeViolation")
jbe/bsw@0 138 mondelefant_errcode_item("28", "InvalidAuthorizationSpecification")
jbe/bsw@0 139 mondelefant_errcode_item("2B", "DependentPrivilegeDescriptorsStillExist")
jbe/bsw@0 140 mondelefant_errcode_item("2D", "InvalidTransactionTermination")
jbe/bsw@0 141 mondelefant_errcode_item("2F", "SqlRoutineException")
jbe/bsw@0 142 mondelefant_errcode_item("34", "InvalidCursorName")
jbe/bsw@0 143 mondelefant_errcode_item("38", "ExternalRoutineException")
jbe/bsw@0 144 mondelefant_errcode_item("39", "ExternalRoutineInvocationException")
jbe/bsw@0 145 mondelefant_errcode_item("3B", "SavepointException")
jbe/bsw@0 146 mondelefant_errcode_item("3D", "InvalidCatalogName")
jbe/bsw@0 147 mondelefant_errcode_item("3F", "InvalidSchemaName")
jbe/bsw@0 148 mondelefant_errcode_item("40", "TransactionRollback")
jbe/bsw@0 149 mondelefant_errcode_item("42", "SyntaxErrorOrAccessRuleViolation")
jbe/bsw@0 150 mondelefant_errcode_item("44", "WithCheckOptionViolation")
jbe/bsw@0 151 mondelefant_errcode_item("53", "InsufficientResources")
jbe/bsw@0 152 mondelefant_errcode_item("54", "ProgramLimitExceeded")
jbe/bsw@0 153 mondelefant_errcode_item("55", "ObjectNotInPrerequisiteState")
jbe/bsw@0 154 mondelefant_errcode_item("57", "OperatorIntervention")
jbe/bsw@0 155 mondelefant_errcode_item("58", "SystemError")
jbe/bsw@0 156 mondelefant_errcode_item("F0", "ConfigurationFileError")
jbe/bsw@0 157 mondelefant_errcode_item("P0", "PlpgsqlError")
jbe/bsw@0 158 mondelefant_errcode_item("XX", "InternalError")
jbe/bsw@0 159 return "unknown";
jbe/bsw@0 160 }
jbe/bsw@0 161
jbe@23 162 // C-function, checking if a given error code (as defined by this library)
jbe@23 163 // is belonging to a certain class of errors (strings are equal or error
jbe@23 164 // code begins with error class followed by a dot):
jbe/bsw@0 165 static int mondelefant_check_error_class(
jbe/bsw@0 166 const char *errcode, const char *errclass
jbe/bsw@0 167 ) {
jbe/bsw@0 168 size_t i = 0;
jbe/bsw@0 169 while (1) {
jbe/bsw@0 170 if (errclass[i] == 0) {
jbe/bsw@0 171 if (errcode[i] == 0 || errcode[i] == '.') return 1;
jbe/bsw@0 172 else return 0;
jbe/bsw@0 173 }
jbe/bsw@0 174 if (errcode[i] != errclass[i]) return 0;
jbe/bsw@0 175 i++;
jbe/bsw@0 176 }
jbe/bsw@0 177 }
jbe/bsw@0 178
jbe@23 179 // pushing first line of a string on Lua's stack (without trailing CR/LF):
jbe/bsw@0 180 static void mondelefant_push_first_line(lua_State *L, const char *str) {
jbe/bsw@0 181 char *str2;
jbe/bsw@0 182 size_t i = 0;
jbe/bsw@0 183 if (!str) abort(); // should not happen
jbe/bsw@0 184 str2 = strdup(str);
jbe/bsw@0 185 while (1) {
jbe/bsw@0 186 char c = str2[i];
jbe/bsw@0 187 if (c == '\n' || c == '\r' || c == 0) { str2[i] = 0; break; }
jbe/bsw@0 188 i++;
jbe/bsw@0 189 };
jbe/bsw@0 190 lua_pushstring(L, str2);
jbe/bsw@0 191 free(str2);
jbe/bsw@0 192 }
jbe/bsw@0 193
jbe@23 194 // "connect" function of library, which establishes a database connection
jbe@23 195 // and returns a database connection handle:
jbe/bsw@0 196 static int mondelefant_connect(lua_State *L) {
jbe@23 197 luaL_Buffer buf; // Lua string buffer to create 'conninfo' (see below)
jbe@23 198 const char *conninfo; // string for PQconnectdb function
jbe@23 199 PGconn *pgconn; // PGconn object as returned by PQconnectdb function
jbe@23 200 mondelefant_conn_t *conn; // C-structure for userdata
jbe@23 201 // if engine is anything but "postgresql", then raise error:
jbe/bsw@0 202 lua_settop(L, 1);
jbe/bsw@0 203 lua_getfield(L, 1, "engine"); // 2
jbe/bsw@0 204 if (!lua_toboolean(L, 2)) {
jbe/bsw@0 205 return luaL_error(L, "No database engine selected.");
jbe/bsw@0 206 }
jbe/bsw@0 207 lua_pushliteral(L, "postgresql"); // 3
jbe/bsw@0 208 if (!lua_rawequal(L, 2, 3)) {
jbe/bsw@0 209 return luaL_error(L,
jbe/bsw@0 210 "Only database engine 'postgresql' is supported."
jbe/bsw@0 211 );
jbe/bsw@0 212 }
jbe@23 213 // create 'conninfo' string for PQconnectdb function, which contains all
jbe@23 214 // options except "engine" option:
jbe/bsw@0 215 lua_settop(L, 1);
jbe/bsw@0 216 lua_pushnil(L); // slot for key at stack position 2
jbe/bsw@0 217 lua_pushnil(L); // slot for value at stack position 3
jbe/bsw@0 218 luaL_buffinit(L, &buf);
jbe/bsw@0 219 {
jbe/bsw@0 220 int need_seperator = 0;
jbe/bsw@0 221 while (lua_pushvalue(L, 2), lua_next(L, 1)) {
jbe/bsw@0 222 lua_replace(L, 3);
jbe/bsw@0 223 lua_replace(L, 2);
jbe/bsw@0 224 // NOTE: numbers will be converted to strings automatically here,
jbe/bsw@0 225 // but perhaps this will change in future versions of lua
jbe/bsw@0 226 luaL_argcheck(L,
jbe/bsw@0 227 lua_isstring(L, 2) && lua_isstring(L, 3), 1, "non-string contained"
jbe/bsw@0 228 );
jbe/bsw@0 229 lua_pushvalue(L, 2);
jbe/bsw@0 230 lua_pushliteral(L, "engine");
jbe/bsw@0 231 if (!lua_rawequal(L, -2, -1)) {
jbe/bsw@0 232 const char *value;
jbe/bsw@0 233 size_t value_len;
jbe/bsw@0 234 size_t value_pos = 0;
jbe/bsw@0 235 lua_pop(L, 1);
jbe/bsw@0 236 if (need_seperator) luaL_addchar(&buf, ' ');
jbe/bsw@0 237 luaL_addvalue(&buf);
jbe/bsw@0 238 luaL_addchar(&buf, '=');
jbe/bsw@0 239 luaL_addchar(&buf, '\'');
jbe/bsw@0 240 value = lua_tolstring(L, 3, &value_len);
jbe/bsw@0 241 do {
jbe/bsw@0 242 char c;
jbe/bsw@0 243 c = value[value_pos++];
jbe/bsw@0 244 if (c == '\'') luaL_addchar(&buf, '\\');
jbe/bsw@0 245 luaL_addchar(&buf, c);
jbe/bsw@0 246 } while (value_pos < value_len);
jbe/bsw@0 247 luaL_addchar(&buf, '\'');
jbe/bsw@0 248 need_seperator = 1;
jbe/bsw@0 249 } else {
jbe/bsw@0 250 lua_pop(L, 1);
jbe/bsw@0 251 }
jbe/bsw@0 252 }
jbe/bsw@0 253 }
jbe/bsw@0 254 luaL_pushresult(&buf);
jbe/bsw@0 255 lua_replace(L, 2);
jbe/bsw@0 256 lua_settop(L, 2);
jbe/bsw@0 257 conninfo = lua_tostring(L, 2);
jbe@23 258 // call PQconnectdb function of libpq:
jbe/bsw@0 259 pgconn = PQconnectdb(conninfo);
jbe@23 260 // throw errors, if neccessary:
jbe/bsw@0 261 if (!pgconn) {
jbe/bsw@0 262 return luaL_error(L,
jbe/bsw@0 263 "Error in libpq while creating 'PGconn' structure."
jbe/bsw@0 264 );
jbe/bsw@0 265 }
jbe/bsw@0 266 if (PQstatus(pgconn) != CONNECTION_OK) {
jbe/bsw@0 267 const char *errmsg;
jbe/bsw@0 268 lua_pushnil(L);
jbe/bsw@0 269 errmsg = PQerrorMessage(pgconn);
jbe/bsw@0 270 if (errmsg) {
jbe/bsw@0 271 mondelefant_push_first_line(L, errmsg);
jbe/bsw@0 272 } else {
jbe/bsw@0 273 lua_pushliteral(L,
jbe/bsw@0 274 "Error while connecting to database, but no error message given."
jbe/bsw@0 275 );
jbe/bsw@0 276 }
jbe/bsw@0 277 lua_pushliteral(L, MONDELEFANT_ERRCODE_CONNECTION);
jbe/bsw@0 278 PQfinish(pgconn);
jbe/bsw@0 279 return 3;
jbe/bsw@0 280 }
jbe@23 281 // create userdata:
jbe/bsw@0 282 lua_settop(L, 0);
jbe/bsw@0 283 conn = lua_newuserdata(L, sizeof(*conn)); // 1
jbe@23 284 // set 'pgconn' in C-struct of userdata:
jbe/bsw@0 285 conn->pgconn = pgconn;
jbe@23 286 // set 'server_encoding' in C-struct of userdata:
jbe/bsw@0 287 {
jbe/bsw@0 288 const char *charset;
jbe/bsw@0 289 charset = PQparameterStatus(pgconn, "server_encoding");
jbe/bsw@0 290 if (charset && !strcmp(charset, "UTF8")) {
jbe/bsw@0 291 conn->server_encoding = MONDELEFANT_SERVER_ENCODING_UTF8;
jbe/bsw@0 292 } else {
jbe/bsw@0 293 conn->server_encoding = MONDELEFANT_SERVER_ENCODING_ASCII;
jbe/bsw@0 294 }
jbe/bsw@0 295 }
jbe@23 296 // set meta-table of userdata:
jbe/bsw@0 297 luaL_getmetatable(L, MONDELEFANT_CONN_MT_REGKEY); // 2
jbe/bsw@0 298 lua_setmetatable(L, 1);
jbe@23 299 // create entry in table storing connection specific data and associate
jbe@23 300 // created userdata with it:
jbe/bsw@0 301 lua_getfield(L, LUA_REGISTRYINDEX, MONDELEFANT_CONN_DATA_REGKEY); // 2
jbe/bsw@0 302 lua_pushvalue(L, 1); // 3
jbe/bsw@0 303 lua_newtable(L); // 4
jbe/bsw@0 304 lua_settable(L, 2);
jbe/bsw@0 305 lua_settop(L, 1);
jbe@23 306 // store key "engine" with value "postgresql" as connection specific data:
jbe/bsw@0 307 lua_pushliteral(L, "postgresql");
jbe/bsw@0 308 lua_setfield(L, 1, "engine");
jbe@23 309 // return userdata:
jbe/bsw@0 310 return 1;
jbe/bsw@0 311 }
jbe/bsw@0 312
jbe@23 313 // returns pointer to libpq handle 'pgconn' of userdata at given index
jbe@23 314 // (or throws error, if database connection has been closed):
jbe/bsw@0 315 static mondelefant_conn_t *mondelefant_get_conn(lua_State *L, int index) {
jbe/bsw@0 316 mondelefant_conn_t *conn;
jbe/bsw@0 317 conn = luaL_checkudata(L, index, MONDELEFANT_CONN_MT_REGKEY);
jbe/bsw@0 318 if (!conn->pgconn) {
jbe/bsw@0 319 luaL_error(L, "PostgreSQL connection has been closed.");
jbe/bsw@0 320 return NULL;
jbe/bsw@0 321 }
jbe/bsw@0 322 return conn;
jbe/bsw@0 323 }
jbe/bsw@0 324
jbe@23 325 // meta-method "__index" of database handles (userdata):
jbe/bsw@0 326 static int mondelefant_conn_index(lua_State *L) {
jbe@23 327 // try table for connection specific data:
jbe/bsw@0 328 lua_settop(L, 2);
jbe/bsw@0 329 lua_getfield(L, LUA_REGISTRYINDEX, MONDELEFANT_CONN_DATA_REGKEY); // 3
jbe/bsw@0 330 lua_pushvalue(L, 1); // 4
jbe/bsw@0 331 lua_gettable(L, 3); // 4
jbe/bsw@0 332 lua_remove(L, 3); // connection specific data-table at stack position 3
jbe/bsw@0 333 lua_pushvalue(L, 2); // 4
jbe/bsw@0 334 lua_gettable(L, 3); // 4
jbe/bsw@0 335 if (!lua_isnil(L, 4)) return 1;
jbe@23 336 // try to use prototype stored in connection specific data:
jbe/bsw@0 337 lua_settop(L, 3);
jbe/bsw@0 338 lua_getfield(L, 3, "prototype"); // 4
jbe/bsw@0 339 if (lua_toboolean(L, 4)) {
jbe/bsw@0 340 lua_pushvalue(L, 2); // 5
jbe/bsw@0 341 lua_gettable(L, 4); // 5
jbe/bsw@0 342 if (!lua_isnil(L, 5)) return 1;
jbe/bsw@0 343 }
jbe@23 344 // try to use "postgresql_connection_prototype" of library:
jbe/bsw@0 345 lua_settop(L, 2);
jbe/bsw@0 346 lua_getfield(L, LUA_REGISTRYINDEX, MONDELEFANT_MODULE_REGKEY); // 3
jbe/bsw@0 347 lua_getfield(L, 3, "postgresql_connection_prototype"); // 4
jbe/bsw@0 348 if (lua_toboolean(L, 4)) {
jbe/bsw@0 349 lua_pushvalue(L, 2); // 5
jbe/bsw@0 350 lua_gettable(L, 4); // 5
jbe/bsw@0 351 if (!lua_isnil(L, 5)) return 1;
jbe/bsw@0 352 }
jbe@23 353 // try to use "connection_prototype" of library:
jbe/bsw@0 354 lua_settop(L, 3);
jbe/bsw@0 355 lua_getfield(L, 3, "connection_prototype"); // 4
jbe/bsw@0 356 if (lua_toboolean(L, 4)) {
jbe/bsw@0 357 lua_pushvalue(L, 2); // 5
jbe/bsw@0 358 lua_gettable(L, 4); // 5
jbe/bsw@0 359 if (!lua_isnil(L, 5)) return 1;
jbe/bsw@0 360 }
jbe@23 361 // give up and return nothing:
jbe/bsw@0 362 return 0;
jbe/bsw@0 363 }
jbe/bsw@0 364
jbe@23 365 // meta-method "__newindex" of database handles (userdata):
jbe/bsw@0 366 static int mondelefant_conn_newindex(lua_State *L) {
jbe@23 367 // store key-value pair in table for connection specific data:
jbe/bsw@0 368 lua_settop(L, 3);
jbe/bsw@0 369 lua_getfield(L, LUA_REGISTRYINDEX, MONDELEFANT_CONN_DATA_REGKEY); // 4
jbe/bsw@0 370 lua_pushvalue(L, 1); // 5
jbe/bsw@0 371 lua_gettable(L, 4); // 5
jbe@23 372 lua_remove(L, 4); // connection specific data-table at stack position 4
jbe/bsw@0 373 lua_pushvalue(L, 2);
jbe/bsw@0 374 lua_pushvalue(L, 3);
jbe/bsw@0 375 lua_settable(L, 4);
jbe@23 376 // return nothing:
jbe/bsw@0 377 return 0;
jbe/bsw@0 378 }
jbe/bsw@0 379
jbe@23 380 // meta-method "__gc" of database handles:
jbe/bsw@0 381 static int mondelefant_conn_free(lua_State *L) {
jbe/bsw@0 382 mondelefant_conn_t *conn;
jbe/bsw@0 383 conn = luaL_checkudata(L, 1, MONDELEFANT_CONN_MT_REGKEY);
jbe/bsw@0 384 if (conn->pgconn) PQfinish(conn->pgconn);
jbe/bsw@0 385 conn->pgconn = NULL;
jbe/bsw@0 386 return 0;
jbe/bsw@0 387 }
jbe/bsw@0 388
jbe@23 389 // method "close" of database handles:
jbe/bsw@0 390 static int mondelefant_conn_close(lua_State *L) {
jbe/bsw@0 391 mondelefant_conn_t *conn;
jbe/bsw@0 392 lua_settop(L, 1);
jbe/bsw@0 393 conn = mondelefant_get_conn(L, 1);
jbe/bsw@0 394 PQfinish(conn->pgconn);
jbe/bsw@0 395 conn->pgconn = NULL;
jbe/bsw@0 396 return 0;
jbe/bsw@0 397 }
jbe/bsw@0 398
jbe@23 399 // method "is_okay" of database handles:
jbe/bsw@0 400 static int mondelefant_conn_is_ok(lua_State *L) {
jbe/bsw@0 401 mondelefant_conn_t *conn;
jbe/bsw@0 402 lua_settop(L, 1);
jbe/bsw@0 403 conn = mondelefant_get_conn(L, 1);
jbe/bsw@0 404 lua_pushboolean(L, PQstatus(conn->pgconn) == CONNECTION_OK);
jbe/bsw@0 405 return 1;
jbe/bsw@0 406 }
jbe/bsw@0 407
jbe@23 408 // method "get_transaction_status" of database handles:
jbe/bsw@0 409 static int mondelefant_conn_get_transaction_status(lua_State *L) {
jbe/bsw@0 410 mondelefant_conn_t *conn;
jbe/bsw@0 411 lua_settop(L, 1);
jbe/bsw@0 412 conn = mondelefant_get_conn(L, 1);
jbe/bsw@0 413 switch (PQtransactionStatus(conn->pgconn)) {
jbe/bsw@0 414 case PQTRANS_IDLE:
jbe/bsw@0 415 lua_pushliteral(L, "idle");
jbe/bsw@0 416 break;
jbe/bsw@0 417 case PQTRANS_ACTIVE:
jbe/bsw@0 418 lua_pushliteral(L, "active");
jbe/bsw@0 419 break;
jbe/bsw@0 420 case PQTRANS_INTRANS:
jbe/bsw@0 421 lua_pushliteral(L, "intrans");
jbe/bsw@0 422 break;
jbe/bsw@0 423 case PQTRANS_INERROR:
jbe/bsw@0 424 lua_pushliteral(L, "inerror");
jbe/bsw@0 425 break;
jbe/bsw@0 426 default:
jbe/bsw@0 427 lua_pushliteral(L, "unknown");
jbe/bsw@0 428 }
jbe/bsw@0 429 return 1;
jbe/bsw@0 430 }
jbe/bsw@0 431
jbe@23 432 // method "create_list" of database handles:
jbe/bsw@0 433 static int mondelefant_conn_create_list(lua_State *L) {
jbe@23 434 // ensure that first argument is a database connection:
jbe/bsw@0 435 lua_settop(L, 2);
jbe/bsw@0 436 luaL_checkudata(L, 1, MONDELEFANT_CONN_MT_REGKEY);
jbe@23 437 // if no second argument is given, use an empty table:
jbe/bsw@0 438 if (!lua_toboolean(L, 2)) {
jbe/bsw@0 439 lua_newtable(L);
jbe/bsw@0 440 lua_replace(L, 2); // new result at stack position 2
jbe/bsw@0 441 }
jbe@23 442 // set meta-table for database result lists/objects:
jbe/bsw@0 443 lua_getfield(L, LUA_REGISTRYINDEX, MONDELEFANT_RESULT_MT_REGKEY); // 3
jbe/bsw@0 444 lua_setmetatable(L, 2);
jbe@23 445 // set "_connection" attribute to self:
jbe/bsw@0 446 lua_pushvalue(L, 1); // 3
jbe/bsw@0 447 lua_setfield(L, 2, "_connection");
jbe@23 448 // set "_type" attribute to string "list":
jbe/bsw@0 449 lua_pushliteral(L, "list"); // 3
jbe/bsw@0 450 lua_setfield(L, 2, "_type");
jbe@23 451 // return created database result list:
jbe/bsw@0 452 return 1;
jbe/bsw@0 453 }
jbe/bsw@0 454
jbe@23 455 // method "create_object" of database handles:
jbe/bsw@0 456 static int mondelefant_conn_create_object(lua_State *L) {
jbe@23 457 // ensure that first argument is a database connection:
jbe/bsw@0 458 lua_settop(L, 2);
jbe/bsw@0 459 luaL_checkudata(L, 1, MONDELEFANT_CONN_MT_REGKEY);
jbe@23 460 // if no second argument is given, use an empty table:
jbe/bsw@0 461 if (!lua_toboolean(L, 2)) {
jbe/bsw@0 462 lua_newtable(L);
jbe/bsw@0 463 lua_replace(L, 2); // new result at stack position 2
jbe/bsw@0 464 }
jbe@23 465 // set meta-table for database result lists/objects:
jbe/bsw@0 466 lua_getfield(L, LUA_REGISTRYINDEX, MONDELEFANT_RESULT_MT_REGKEY); // 3
jbe/bsw@0 467 lua_setmetatable(L, 2);
jbe@23 468 // set "_connection" attribute to self:
jbe/bsw@0 469 lua_pushvalue(L, 1); // 3
jbe/bsw@0 470 lua_setfield(L, 2, "_connection");
jbe@23 471 // set "_type" attribute to string "object":
jbe/bsw@0 472 lua_pushliteral(L, "object"); // 3
jbe/bsw@0 473 lua_setfield(L, 2, "_type"); // "object" or "list"
jbe@23 474 // create empty tables for "_data", "_dirty" and "_ref" attributes:
jbe/bsw@0 475 lua_newtable(L); // 3
jbe/bsw@0 476 lua_setfield(L, 2, "_data");
jbe/bsw@0 477 lua_newtable(L); // 3
jbe/bsw@0 478 lua_setfield(L, 2, "_dirty");
jbe/bsw@0 479 lua_newtable(L); // 3
jbe/bsw@0 480 lua_setfield(L, 2, "_ref"); // nil=no info, false=nil, else table
jbe@23 481 // return created database result object:
jbe/bsw@0 482 return 1;
jbe/bsw@0 483 }
jbe/bsw@0 484
jbe@23 485 // method "quote_string" of database handles:
jbe/bsw@0 486 static int mondelefant_conn_quote_string(lua_State *L) {
jbe/bsw@0 487 mondelefant_conn_t *conn;
jbe/bsw@0 488 const char *input;
jbe/bsw@0 489 size_t input_len;
jbe/bsw@0 490 char *output;
jbe/bsw@0 491 size_t output_len;
jbe@23 492 // get 'conn' attribute of C-struct of database connection:
jbe/bsw@0 493 lua_settop(L, 2);
jbe/bsw@0 494 conn = mondelefant_get_conn(L, 1);
jbe@23 495 // get second argument, which must be a string:
jbe/bsw@0 496 input = luaL_checklstring(L, 2, &input_len);
jbe@23 497 // throw error, if string is too long:
jbe/bsw@0 498 if (input_len > (SIZE_MAX / sizeof(char) - 3) / 2) {
jbe/bsw@0 499 return luaL_error(L, "String to be escaped is too long.");
jbe/bsw@0 500 }
jbe@23 501 // allocate memory for quoted string:
jbe/bsw@0 502 output = malloc((2 * input_len + 3) * sizeof(char));
jbe/bsw@0 503 if (!output) {
jbe/bsw@0 504 return luaL_error(L, "Could not allocate memory for string quoting.");
jbe/bsw@0 505 }
jbe@23 506 // do escaping by calling PQescapeStringConn and enclosing result with
jbe@23 507 // single quotes:
jbe/bsw@0 508 output[0] = '\'';
jbe/bsw@0 509 output_len = PQescapeStringConn(
jbe/bsw@0 510 conn->pgconn, output + 1, input, input_len, NULL
jbe/bsw@0 511 );
jbe/bsw@0 512 output[output_len + 1] = '\'';
jbe/bsw@0 513 output[output_len + 2] = 0;
jbe@23 514 // create Lua string:
jbe/bsw@0 515 lua_pushlstring(L, output, output_len + 2);
jbe@23 516 // free allocated memory:
jbe/bsw@0 517 free(output);
jbe@23 518 // return Lua string:
jbe/bsw@0 519 return 1;
jbe/bsw@0 520 }
jbe/bsw@0 521
jbe@23 522 // method "quote_binary" of database handles:
jbe/bsw@0 523 static int mondelefant_conn_quote_binary(lua_State *L) {
jbe/bsw@0 524 mondelefant_conn_t *conn;
jbe/bsw@0 525 const char *input;
jbe/bsw@0 526 size_t input_len;
jbe/bsw@0 527 char *output;
jbe/bsw@0 528 size_t output_len;
jbe/bsw@0 529 luaL_Buffer buf;
jbe@23 530 // get 'conn' attribute of C-struct of database connection:
jbe/bsw@0 531 lua_settop(L, 2);
jbe/bsw@0 532 conn = mondelefant_get_conn(L, 1);
jbe@23 533 // get second argument, which must be a string:
jbe/bsw@0 534 input = luaL_checklstring(L, 2, &input_len);
jbe@23 535 // call PQescapeByteaConn, which allocates memory itself:
jbe/bsw@0 536 output = (char *)PQescapeByteaConn(
jbe/bsw@0 537 conn->pgconn, (const unsigned char *)input, input_len, &output_len
jbe/bsw@0 538 );
jbe@23 539 // if PQescapeByteaConn returned NULL, then throw error:
jbe/bsw@0 540 if (!output) {
jbe/bsw@0 541 return luaL_error(L, "Could not allocate memory for binary quoting.");
jbe/bsw@0 542 }
jbe@23 543 // create Lua string enclosed by single quotes:
jbe/bsw@0 544 luaL_buffinit(L, &buf);
jbe/bsw@0 545 luaL_addchar(&buf, '\'');
jbe/bsw@0 546 luaL_addlstring(&buf, output, output_len - 1);
jbe/bsw@0 547 luaL_addchar(&buf, '\'');
jbe/bsw@0 548 luaL_pushresult(&buf);
jbe@23 549 // free memory allocated by PQescapeByteaConn:
jbe/bsw@0 550 PQfreemem(output);
jbe@23 551 // return Lua string:
jbe/bsw@0 552 return 1;
jbe/bsw@0 553 }
jbe/bsw@0 554
jbe@23 555 // method "assemble_command" of database handles:
jbe/bsw@0 556 static int mondelefant_conn_assemble_command(lua_State *L) {
jbe/bsw@0 557 mondelefant_conn_t *conn;
jbe/bsw@0 558 int paramidx = 2;
jbe/bsw@0 559 const char *template;
jbe/bsw@0 560 size_t template_pos = 0;
jbe/bsw@0 561 luaL_Buffer buf;
jbe@23 562 // get 'conn' attribute of C-struct of database connection:
jbe/bsw@0 563 lua_settop(L, 2);
jbe/bsw@0 564 conn = mondelefant_get_conn(L, 1);
jbe@23 565 // if second argument is a string, return this string:
jbe/bsw@0 566 if (lua_isstring(L, 2)) {
jbe/bsw@0 567 lua_tostring(L, 2);
jbe/bsw@0 568 return 1;
jbe/bsw@0 569 }
jbe@23 570 // if second argument has __tostring meta-method,
jbe@23 571 // then use this method and return its result:
jbe/bsw@0 572 if (luaL_callmeta(L, 2, "__tostring")) return 1;
jbe@23 573 // otherwise, require that second argument is a table:
jbe/bsw@0 574 luaL_checktype(L, 2, LUA_TTABLE);
jbe@23 575 // get first element of table, which must be a string:
jbe/bsw@0 576 lua_rawgeti(L, 2, 1); // 3
jbe/bsw@0 577 luaL_argcheck(L,
jbe/bsw@0 578 lua_isstring(L, 3),
jbe/bsw@0 579 2,
jbe/bsw@0 580 "First entry of SQL command structure is not a string."
jbe/bsw@0 581 );
jbe/bsw@0 582 template = lua_tostring(L, 3);
jbe@23 583 // get value of "input_converter" attribute of database connection:
jbe/bsw@0 584 lua_pushliteral(L, "input_converter"); // 4
jbe/bsw@0 585 lua_gettable(L, 1); // input_converter at stack position 4
jbe@23 586 // reserve space on Lua stack:
jbe/bsw@0 587 lua_pushnil(L); // free space at stack position 5
jbe/bsw@0 588 lua_pushnil(L); // free space at stack position 6
jbe@23 589 // initialize Lua buffer for result string:
jbe/bsw@0 590 luaL_buffinit(L, &buf);
jbe@23 591 // fill buffer in loop:
jbe/bsw@0 592 while (1) {
jbe@23 593 // variable declaration:
jbe/bsw@0 594 char c;
jbe@23 595 // get next character:
jbe/bsw@0 596 c = template[template_pos++];
jbe@23 597 // break, when character is NULL byte:
jbe/bsw@0 598 if (!c) break;
jbe@23 599 // question-mark and dollar-sign are special characters:
jbe@23 600 if (c == '?' || c == '$') { // special character found
jbe@23 601 // check, if same character follows:
jbe@23 602 if (template[template_pos] == c) { // special character is escaped
jbe@23 603 // consume two characters of input and add one character to buffer:
jbe/bsw@0 604 template_pos++;
jbe/bsw@0 605 luaL_addchar(&buf, c);
jbe@23 606 } else { // special character is not escaped
jbe/bsw@0 607 luaL_Buffer keybuf;
jbe/bsw@0 608 int subcmd;
jbe@23 609 // set 'subcmd' = true, if special character was a dollar-sign,
jbe@23 610 // set 'subcmd' = false, if special character was a question-mark:
jbe/bsw@0 611 subcmd = (c == '$');
jbe@23 612 // read any number of alpha numeric chars or underscores
jbe@23 613 // and store them on Lua stack:
jbe/bsw@0 614 luaL_buffinit(L, &keybuf);
jbe/bsw@0 615 while (1) {
jbe/bsw@0 616 c = template[template_pos];
jbe/bsw@0 617 if (
jbe/bsw@0 618 (c < 'A' || c > 'Z') &&
jbe/bsw@0 619 (c < 'a' || c > 'z') &&
jbe/bsw@0 620 (c < '0' || c > '9') &&
jbe/bsw@0 621 (c != '_')
jbe/bsw@0 622 ) break;
jbe/bsw@0 623 luaL_addchar(&keybuf, c);
jbe/bsw@0 624 template_pos++;
jbe/bsw@0 625 }
jbe/bsw@0 626 luaL_pushresult(&keybuf);
jbe@23 627 // check, if any characters matched:
jbe/bsw@0 628 if (lua_objlen(L, -1)) {
jbe@23 629 // if any alpha numeric chars or underscores were found,
jbe@23 630 // push them on stack as a Lua string and use them to lookup
jbe@23 631 // value from second argument:
jbe/bsw@0 632 lua_pushvalue(L, -1); // save key on stack
jbe@23 633 lua_gettable(L, 2); // fetch value (raw-value)
jbe/bsw@0 634 } else {
jbe@23 635 // otherwise push nil and use numeric lookup based on 'paramidx':
jbe/bsw@0 636 lua_pop(L, 1);
jbe/bsw@0 637 lua_pushnil(L); // put nil on key position
jbe@23 638 lua_rawgeti(L, 2, paramidx++); // fetch value (raw-value)
jbe/bsw@0 639 }
jbe@23 640 // Lua stack contains: ..., <buffer>, key, raw-value
jbe@23 641 // branch according to type of special character ("?" or "$"):
jbe@23 642 if (subcmd) { // dollar-sign
jbe/bsw@0 643 size_t i;
jbe/bsw@0 644 size_t count;
jbe@23 645 // store fetched value (which is supposed to be sub-structure)
jbe@23 646 // on Lua stack position 5 and drop key:
jbe@23 647 lua_replace(L, 5);
jbe@23 648 lua_pop(L, 1);
jbe@23 649 // Lua stack contains: ..., <buffer>
jbe@23 650 // check, if fetched value is really a sub-structure:
jbe/bsw@0 651 luaL_argcheck(L,
jbe/bsw@0 652 !lua_isnil(L, 5),
jbe/bsw@0 653 2,
jbe/bsw@0 654 "SQL sub-structure not found."
jbe/bsw@0 655 );
jbe/bsw@0 656 luaL_argcheck(L,
jbe/bsw@0 657 lua_type(L, 5) == LUA_TTABLE,
jbe/bsw@0 658 2,
jbe/bsw@0 659 "SQL sub-structure must be a table."
jbe/bsw@0 660 );
jbe@23 661 // Lua stack contains: ..., <buffer>
jbe@23 662 // get value of "sep" attribute of sub-structure,
jbe@23 663 // and place it on Lua stack position 6:
jbe/bsw@0 664 lua_getfield(L, 5, "sep");
jbe@23 665 lua_replace(L, 6);
jbe@23 666 // if seperator is nil, then use ", " as default,
jbe@23 667 // if seperator is neither nil nor a string, then throw error:
jbe/bsw@0 668 if (lua_isnil(L, 6)) {
jbe/bsw@0 669 lua_pushstring(L, ", ");
jbe/bsw@0 670 lua_replace(L, 6);
jbe/bsw@0 671 } else {
jbe/bsw@0 672 luaL_argcheck(L,
jbe/bsw@0 673 lua_isstring(L, 6),
jbe/bsw@0 674 2,
jbe/bsw@0 675 "Seperator of SQL sub-structure has to be a string."
jbe/bsw@0 676 );
jbe/bsw@0 677 }
jbe@23 678 // iterate over items of sub-structure:
jbe/bsw@0 679 count = lua_objlen(L, 5);
jbe/bsw@0 680 for (i = 0; i < count; i++) {
jbe@23 681 // add seperator, unless this is the first run:
jbe/bsw@0 682 if (i) {
jbe/bsw@0 683 lua_pushvalue(L, 6);
jbe/bsw@0 684 luaL_addvalue(&buf);
jbe/bsw@0 685 }
jbe@23 686 // recursivly apply assemble function and add results to buffer:
jbe/bsw@0 687 lua_pushcfunction(L, mondelefant_conn_assemble_command);
jbe/bsw@0 688 lua_pushvalue(L, 1);
jbe/bsw@0 689 lua_rawgeti(L, 5, i+1);
jbe/bsw@0 690 lua_call(L, 2, 1);
jbe/bsw@0 691 luaL_addvalue(&buf);
jbe/bsw@0 692 }
jbe@23 693 } else { // question-mark
jbe/bsw@0 694 if (lua_toboolean(L, 4)) {
jbe@23 695 // call input_converter with connection handle, raw-value and
jbe@23 696 // an info-table which contains a "field_name" entry with the
jbe@23 697 // used key:
jbe/bsw@0 698 lua_pushvalue(L, 4);
jbe/bsw@0 699 lua_pushvalue(L, 1);
jbe/bsw@0 700 lua_pushvalue(L, -3);
jbe/bsw@0 701 lua_newtable(L);
jbe/bsw@0 702 lua_pushvalue(L, -6);
jbe/bsw@0 703 lua_setfield(L, -2, "field_name");
jbe/bsw@0 704 lua_call(L, 3, 1);
jbe@23 705 // Lua stack contains: ..., <buffer>, key, raw-value, final-value
jbe@23 706 // remove key and raw-value:
jbe/bsw@0 707 lua_remove(L, -2);
jbe/bsw@0 708 lua_remove(L, -2);
jbe@23 709 // Lua stack contains: ..., <buffer>, final-value
jbe@23 710 // throw error, if final-value is not a string:
jbe/bsw@0 711 if (!lua_isstring(L, -1)) {
jbe/bsw@0 712 return luaL_error(L, "input_converter returned non-string.");
jbe/bsw@0 713 }
jbe/bsw@0 714 } else {
jbe@23 715 // remove key from stack:
jbe/bsw@0 716 lua_remove(L, -2);
jbe@23 717 // Lua stack contains: ..., <buffer>, raw-value
jbe@23 718 // branch according to type of value:
jbe@23 719 if (lua_isnil(L, -1)) { // value is nil
jbe@23 720 // push string "NULL" to stack:
jbe/bsw@0 721 lua_pushliteral(L, "NULL");
jbe@23 722 } else if (lua_type(L, -1) == LUA_TBOOLEAN) { // value is boolean
jbe@23 723 // push strings "TRUE" or "FALSE" to stack:
jbe/bsw@0 724 lua_pushstring(L, lua_toboolean(L, -1) ? "TRUE" : "FALSE");
jbe@23 725 } else if (lua_isstring(L, -1)) { // value is string or number
jbe/bsw@0 726 // NOTE: In this version of lua a number will be converted
jbe@23 727 // push output of "quote_string" method of database
jbe@23 728 // connection to stack:
jbe/bsw@0 729 lua_tostring(L, -1);
jbe/bsw@0 730 lua_pushcfunction(L, mondelefant_conn_quote_string);
jbe/bsw@0 731 lua_pushvalue(L, 1);
jbe/bsw@0 732 lua_pushvalue(L, -3);
jbe/bsw@0 733 lua_call(L, 2, 1);
jbe@23 734 } else { // value is of other type
jbe@23 735 // throw error:
jbe/bsw@0 736 return luaL_error(L,
jbe/bsw@0 737 "Unable to convert SQL value due to unknown type "
jbe/bsw@0 738 "or missing input_converter."
jbe/bsw@0 739 );
jbe/bsw@0 740 }
jbe@23 741 // Lua stack contains: ..., <buffer>, raw-value, final-value
jbe@23 742 // remove raw-value:
jbe/bsw@0 743 lua_remove(L, -2);
jbe@23 744 // Lua stack contains: ..., <buffer>, final-value
jbe/bsw@0 745 }
jbe@23 746 // append final-value to buffer:
jbe/bsw@0 747 luaL_addvalue(&buf);
jbe/bsw@0 748 }
jbe/bsw@0 749 }
jbe@23 750 } else { // character is not special
jbe@23 751 // just copy character:
jbe/bsw@0 752 luaL_addchar(&buf, c);
jbe/bsw@0 753 }
jbe/bsw@0 754 }
jbe@23 755 // return string in buffer:
jbe/bsw@0 756 luaL_pushresult(&buf);
jbe/bsw@0 757 return 1;
jbe/bsw@0 758 }
jbe/bsw@0 759
jbe@23 760 // max number of SQL statements executed by one "query" method call:
jbe/bsw@0 761 #define MONDELEFANT_MAX_COMMAND_COUNT 64
jbe@23 762 // max number of columns in a database result:
jbe/bsw@0 763 #define MONDELEFANT_MAX_COLUMN_COUNT 1024
jbe@23 764 // enum values for 'modes' array in C-function below:
jbe/bsw@0 765 #define MONDELEFANT_QUERY_MODE_LIST 1
jbe/bsw@0 766 #define MONDELEFANT_QUERY_MODE_OBJECT 2
jbe/bsw@0 767 #define MONDELEFANT_QUERY_MODE_OPT_OBJECT 3
jbe/bsw@0 768
jbe@23 769 // method "try_query" of database handles:
jbe/bsw@0 770 static int mondelefant_conn_try_query(lua_State *L) {
jbe/bsw@0 771 mondelefant_conn_t *conn;
jbe/bsw@0 772 int command_count;
jbe/bsw@0 773 int command_idx;
jbe/bsw@0 774 int modes[MONDELEFANT_MAX_COMMAND_COUNT];
jbe/bsw@0 775 luaL_Buffer buf;
jbe/bsw@0 776 int sent_success;
jbe/bsw@0 777 PGresult *res;
jbe/bsw@0 778 int rows, cols, row, col;
jbe@23 779 // get 'conn' attribute of C-struct of database connection:
jbe/bsw@0 780 conn = mondelefant_get_conn(L, 1);
jbe@23 781 // calculate number of commands (2 arguments for one command):
jbe/bsw@0 782 command_count = lua_gettop(L) / 2;
jbe@23 783 // push nil on stack, which is needed, if last mode was ommitted:
jbe@23 784 lua_pushnil(L);
jbe@23 785 // throw error, if number of commands is too high:
jbe/bsw@0 786 if (command_count > MONDELEFANT_MAX_COMMAND_COUNT) {
jbe/bsw@0 787 return luaL_error(L, "Exceeded maximum command count in one query.");
jbe/bsw@0 788 }
jbe@23 789 // create SQL string, store query modes and push SQL string on stack:
jbe/bsw@0 790 luaL_buffinit(L, &buf);
jbe/bsw@0 791 for (command_idx = 0; command_idx < command_count; command_idx++) {
jbe/bsw@0 792 int mode;
jbe/bsw@0 793 int mode_idx; // stack index of mode string
jbe/bsw@0 794 if (command_idx) luaL_addchar(&buf, ' ');
jbe/bsw@0 795 lua_pushcfunction(L, mondelefant_conn_assemble_command);
jbe/bsw@0 796 lua_pushvalue(L, 1);
jbe/bsw@0 797 lua_pushvalue(L, 2 + 2 * command_idx);
jbe/bsw@0 798 lua_call(L, 2, 1);
jbe/bsw@0 799 luaL_addvalue(&buf);
jbe/bsw@0 800 luaL_addchar(&buf, ';');
jbe/bsw@0 801 mode_idx = 3 + 2 * command_idx;
jbe/bsw@0 802 if (lua_isnil(L, mode_idx)) {
jbe/bsw@0 803 mode = MONDELEFANT_QUERY_MODE_LIST;
jbe/bsw@0 804 } else {
jbe/bsw@0 805 const char *modestr;
jbe/bsw@0 806 modestr = luaL_checkstring(L, mode_idx);
jbe/bsw@0 807 if (!strcmp(modestr, "list")) {
jbe/bsw@0 808 mode = MONDELEFANT_QUERY_MODE_LIST;
jbe/bsw@0 809 } else if (!strcmp(modestr, "object")) {
jbe/bsw@0 810 mode = MONDELEFANT_QUERY_MODE_OBJECT;
jbe/bsw@0 811 } else if (!strcmp(modestr, "opt_object")) {
jbe/bsw@0 812 mode = MONDELEFANT_QUERY_MODE_OPT_OBJECT;
jbe/bsw@0 813 } else {
jbe/bsw@0 814 return luaL_error(L, "Unknown query mode specified.");
jbe/bsw@0 815 }
jbe/bsw@0 816 }
jbe/bsw@0 817 modes[command_idx] = mode;
jbe/bsw@0 818 }
jbe/bsw@0 819 luaL_pushresult(&buf); // stack position unknown
jbe/bsw@0 820 lua_replace(L, 2); // SQL command string to stack position 2
jbe@23 821 // call sql_tracer, if set:
jbe/bsw@0 822 lua_settop(L, 2);
jbe/bsw@0 823 lua_getfield(L, 1, "sql_tracer"); // tracer at stack position 3
jbe/bsw@0 824 if (lua_toboolean(L, 3)) {
jbe/bsw@0 825 lua_pushvalue(L, 1); // 4
jbe/bsw@0 826 lua_pushvalue(L, 2); // 5
jbe/bsw@0 827 lua_call(L, 2, 1); // trace callback at stack position 3
jbe/bsw@0 828 }
jbe@23 829 // NOTE: If no tracer was found, then nil or false is stored at stack
jbe@23 830 // position 3.
jbe@23 831 // call PQsendQuery function and store result in 'sent_success' variable:
jbe/bsw@0 832 sent_success = PQsendQuery(conn->pgconn, lua_tostring(L, 2));
jbe@23 833 // create preliminary result table:
jbe/bsw@0 834 lua_newtable(L); // results in table at stack position 4
jbe@23 835 // iterate over results using function PQgetResult to fill result table:
jbe/bsw@0 836 for (command_idx = 0; ; command_idx++) {
jbe/bsw@0 837 int mode;
jbe/bsw@0 838 char binary[MONDELEFANT_MAX_COLUMN_COUNT];
jbe/bsw@0 839 ExecStatusType pgstatus;
jbe@23 840 // fetch mode which was given for the command:
jbe/bsw@0 841 mode = modes[command_idx];
jbe@23 842 // if PQsendQuery call was successful, then fetch result data:
jbe/bsw@0 843 if (sent_success) {
jbe@23 844 // NOTE: PQgetResult called one extra time. Break only, if all
jbe@23 845 // queries have been processed and PQgetResult returned NULL.
jbe/bsw@0 846 res = PQgetResult(conn->pgconn);
jbe/bsw@0 847 if (command_idx >= command_count && !res) break;
jbe/bsw@0 848 if (res) {
jbe/bsw@0 849 pgstatus = PQresultStatus(res);
jbe/bsw@0 850 rows = PQntuples(res);
jbe/bsw@0 851 cols = PQnfields(res);
jbe/bsw@0 852 }
jbe/bsw@0 853 }
jbe@23 854 // handle errors:
jbe/bsw@0 855 if (
jbe/bsw@0 856 !sent_success || command_idx >= command_count || !res ||
jbe/bsw@0 857 (pgstatus != PGRES_TUPLES_OK && pgstatus != PGRES_COMMAND_OK) ||
jbe/bsw@0 858 (rows < 1 && mode == MONDELEFANT_QUERY_MODE_OBJECT) ||
jbe/bsw@0 859 (rows > 1 && mode != MONDELEFANT_QUERY_MODE_LIST)
jbe/bsw@0 860 ) {
jbe/bsw@0 861 const char *command;
jbe/bsw@0 862 command = lua_tostring(L, 2);
jbe/bsw@0 863 lua_newtable(L); // 5
jbe/bsw@0 864 lua_getfield(L,
jbe/bsw@0 865 LUA_REGISTRYINDEX,
jbe/bsw@0 866 MONDELEFANT_ERROROBJECT_MT_REGKEY
jbe/bsw@0 867 );
jbe/bsw@0 868 lua_setmetatable(L, 5);
jbe/bsw@0 869 lua_pushvalue(L, 1);
jbe/bsw@0 870 lua_setfield(L, 5, "connection");
jbe/bsw@0 871 lua_pushinteger(L, command_idx + 1);
jbe/bsw@0 872 lua_setfield(L, 5, "command_number");
jbe/bsw@0 873 lua_pushvalue(L, 2);
jbe/bsw@0 874 lua_setfield(L, 5, "sql_command");
jbe/bsw@0 875 if (!res) {
jbe/bsw@0 876 lua_pushliteral(L, MONDELEFANT_ERRCODE_RESULTCOUNT_LOW);
jbe/bsw@0 877 lua_setfield(L, 5, "code");
jbe/bsw@0 878 lua_pushliteral(L, "Received too few database result sets.");
jbe/bsw@0 879 lua_setfield(L, 5, "message");
jbe/bsw@0 880 } else if (command_idx >= command_count) {
jbe/bsw@0 881 lua_pushliteral(L, MONDELEFANT_ERRCODE_RESULTCOUNT_HIGH);
jbe/bsw@0 882 lua_setfield(L, 5, "code");
jbe/bsw@0 883 lua_pushliteral(L, "Received too many database result sets.");
jbe/bsw@0 884 lua_setfield(L, 5, "message");
jbe/bsw@0 885 } else if (
jbe/bsw@0 886 pgstatus != PGRES_TUPLES_OK && pgstatus != PGRES_COMMAND_OK
jbe/bsw@0 887 ) {
jbe/bsw@0 888 const char *sqlstate;
jbe/bsw@0 889 const char *errmsg;
jbe/bsw@0 890 lua_pushstring(L, PQresultErrorField(res, PG_DIAG_SEVERITY));
jbe/bsw@0 891 lua_setfield(L, 5, "pg_severity");
jbe/bsw@0 892 sqlstate = PQresultErrorField(res, PG_DIAG_SQLSTATE);
jbe/bsw@0 893 if (sqlstate) {
jbe/bsw@0 894 lua_pushstring(L, sqlstate);
jbe/bsw@0 895 lua_setfield(L, 5, "pg_sqlstate");
jbe/bsw@0 896 lua_pushstring(L, mondelefant_translate_errcode(sqlstate));
jbe/bsw@0 897 lua_setfield(L, 5, "code");
jbe/bsw@0 898 } else {
jbe/bsw@0 899 lua_pushliteral(L, MONDELEFANT_ERRCODE_UNKNOWN);
jbe/bsw@0 900 lua_setfield(L, 5, "code");
jbe/bsw@0 901 }
jbe/bsw@0 902 errmsg = PQresultErrorField(res, PG_DIAG_MESSAGE_PRIMARY);
jbe/bsw@0 903 if (errmsg) {
jbe/bsw@0 904 mondelefant_push_first_line(L, errmsg);
jbe/bsw@0 905 lua_setfield(L, 5, "message");
jbe/bsw@0 906 lua_pushstring(L, errmsg);
jbe/bsw@0 907 lua_setfield(L, 5, "pg_message_primary");
jbe/bsw@0 908 } else {
jbe/bsw@0 909 lua_pushliteral(L,
jbe/bsw@0 910 "Error while fetching result, but no error message given."
jbe/bsw@0 911 );
jbe/bsw@0 912 lua_setfield(L, 5, "message");
jbe/bsw@0 913 }
jbe/bsw@0 914 lua_pushstring(L, PQresultErrorField(res, PG_DIAG_MESSAGE_DETAIL));
jbe/bsw@0 915 lua_setfield(L, 5, "pg_message_detail");
jbe/bsw@0 916 lua_pushstring(L, PQresultErrorField(res, PG_DIAG_MESSAGE_HINT));
jbe/bsw@0 917 lua_setfield(L, 5, "pg_message_hint");
jbe/bsw@0 918 // NOTE: "position" and "pg_internal_position" are recalculated to
jbe/bsw@0 919 // byte offsets, as Lua 5.1 is not Unicode aware.
jbe/bsw@0 920 {
jbe/bsw@0 921 char *tmp;
jbe/bsw@0 922 tmp = PQresultErrorField(res, PG_DIAG_STATEMENT_POSITION);
jbe/bsw@0 923 if (tmp) {
jbe/bsw@0 924 int pos;
jbe/bsw@0 925 pos = atoi(tmp) - 1;
jbe/bsw@0 926 if (conn->server_encoding == MONDELEFANT_SERVER_ENCODING_UTF8) {
jbe/bsw@0 927 pos = utf8_position_to_byte(command, pos);
jbe/bsw@0 928 }
jbe/bsw@0 929 lua_pushinteger(L, pos + 1);
jbe/bsw@0 930 lua_setfield(L, 5, "position");
jbe/bsw@0 931 }
jbe/bsw@0 932 }
jbe/bsw@0 933 {
jbe/bsw@0 934 const char *internal_query;
jbe/bsw@0 935 internal_query = PQresultErrorField(res, PG_DIAG_INTERNAL_QUERY);
jbe/bsw@0 936 lua_pushstring(L, internal_query);
jbe/bsw@0 937 lua_setfield(L, 5, "pg_internal_query");
jbe/bsw@0 938 char *tmp;
jbe/bsw@0 939 tmp = PQresultErrorField(res, PG_DIAG_INTERNAL_POSITION);
jbe/bsw@0 940 if (tmp) {
jbe/bsw@0 941 int pos;
jbe/bsw@0 942 pos = atoi(tmp) - 1;
jbe/bsw@0 943 if (conn->server_encoding == MONDELEFANT_SERVER_ENCODING_UTF8) {
jbe/bsw@0 944 pos = utf8_position_to_byte(internal_query, pos);
jbe/bsw@0 945 }
jbe/bsw@0 946 lua_pushinteger(L, pos + 1);
jbe/bsw@0 947 lua_setfield(L, 5, "pg_internal_position");
jbe/bsw@0 948 }
jbe/bsw@0 949 }
jbe/bsw@0 950 lua_pushstring(L, PQresultErrorField(res, PG_DIAG_CONTEXT));
jbe/bsw@0 951 lua_setfield(L, 5, "pg_context");
jbe/bsw@0 952 lua_pushstring(L, PQresultErrorField(res, PG_DIAG_SOURCE_FILE));
jbe/bsw@0 953 lua_setfield(L, 5, "pg_source_file");
jbe/bsw@0 954 lua_pushstring(L, PQresultErrorField(res, PG_DIAG_SOURCE_LINE));
jbe/bsw@0 955 lua_setfield(L, 5, "pg_source_line");
jbe/bsw@0 956 lua_pushstring(L, PQresultErrorField(res, PG_DIAG_SOURCE_FUNCTION));
jbe/bsw@0 957 lua_setfield(L, 5, "pg_source_function");
jbe/bsw@0 958 } else if (rows < 1 && mode == MONDELEFANT_QUERY_MODE_OBJECT) {
jbe/bsw@0 959 lua_pushliteral(L, MONDELEFANT_ERRCODE_QUERY1_NO_ROWS);
jbe/bsw@0 960 lua_setfield(L, 5, "code");
jbe/bsw@0 961 lua_pushliteral(L, "Expected one row, but got empty set.");
jbe/bsw@0 962 lua_setfield(L, 5, "message");
jbe/bsw@0 963 } else if (rows > 1 && mode != MONDELEFANT_QUERY_MODE_LIST) {
jbe/bsw@0 964 lua_pushliteral(L, MONDELEFANT_ERRCODE_QUERY1_MULTIPLE_ROWS);
jbe/bsw@0 965 lua_setfield(L, 5, "code");
jbe/bsw@0 966 lua_pushliteral(L, "Got more than one result row.");
jbe/bsw@0 967 lua_setfield(L, 5, "message");
jbe/bsw@0 968 } else {
jbe/bsw@0 969 // should not happen
jbe/bsw@0 970 abort();
jbe/bsw@0 971 }
jbe/bsw@0 972 if (res) {
jbe/bsw@0 973 PQclear(res);
jbe/bsw@0 974 while ((res = PQgetResult(conn->pgconn))) PQclear(res);
jbe/bsw@0 975 }
jbe/bsw@0 976 if (lua_toboolean(L, 3)) {
jbe/bsw@0 977 lua_pushvalue(L, 3);
jbe/bsw@0 978 lua_pushvalue(L, 5);
jbe/bsw@0 979 lua_call(L, 1, 0);
jbe/bsw@0 980 }
jbe/bsw@0 981 return 1;
jbe/bsw@0 982 }
jbe@23 983 // call "create_list" or "create_object" method of database handle,
jbe@23 984 // result will be at stack position 5:
jbe/bsw@0 985 if (modes[command_idx] == MONDELEFANT_QUERY_MODE_LIST) {
jbe/bsw@0 986 lua_pushcfunction(L, mondelefant_conn_create_list); // 5
jbe/bsw@0 987 lua_pushvalue(L, 1); // 6
jbe/bsw@0 988 lua_call(L, 1, 1); // 5
jbe/bsw@0 989 } else {
jbe/bsw@0 990 lua_pushcfunction(L, mondelefant_conn_create_object); // 5
jbe/bsw@0 991 lua_pushvalue(L, 1); // 6
jbe/bsw@0 992 lua_call(L, 1, 1); // 5
jbe/bsw@0 993 }
jbe@23 994 // set "_column_info":
jbe@23 995 lua_newtable(L); // 6
jbe/bsw@0 996 for (col = 0; col < cols; col++) {
jbe/bsw@0 997 lua_newtable(L); // 7
jbe/bsw@0 998 lua_pushstring(L, PQfname(res, col));
jbe/bsw@0 999 lua_setfield(L, 7, "field_name");
jbe/bsw@0 1000 {
jbe/bsw@0 1001 Oid tmp;
jbe/bsw@0 1002 tmp = PQftable(res, col);
jbe/bsw@0 1003 if (tmp == InvalidOid) lua_pushnil(L);
jbe/bsw@0 1004 else lua_pushinteger(L, tmp);
jbe/bsw@0 1005 lua_setfield(L, 7, "table_oid");
jbe/bsw@0 1006 }
jbe/bsw@0 1007 {
jbe/bsw@0 1008 int tmp;
jbe/bsw@0 1009 tmp = PQftablecol(res, col);
jbe/bsw@0 1010 if (tmp == 0) lua_pushnil(L);
jbe/bsw@0 1011 else lua_pushinteger(L, tmp);
jbe/bsw@0 1012 lua_setfield(L, 7, "table_column_number");
jbe/bsw@0 1013 }
jbe/bsw@0 1014 {
jbe/bsw@0 1015 Oid tmp;
jbe/bsw@0 1016 tmp = PQftype(res, col);
jbe/bsw@0 1017 binary[col] = (tmp == MONDELEFANT_POSTGRESQL_BINARY_OID);
jbe/bsw@0 1018 lua_pushinteger(L, tmp);
jbe/bsw@0 1019 lua_setfield(L, 7, "type_oid");
jbe/bsw@0 1020 lua_pushstring(L, mondelefant_oid_to_typestr(tmp));
jbe/bsw@0 1021 lua_setfield(L, 7, "type");
jbe/bsw@0 1022 }
jbe/bsw@0 1023 {
jbe/bsw@0 1024 int tmp;
jbe/bsw@0 1025 tmp = PQfmod(res, col);
jbe/bsw@0 1026 if (tmp == -1) lua_pushnil(L);
jbe/bsw@0 1027 else lua_pushinteger(L, tmp);
jbe/bsw@0 1028 lua_setfield(L, 7, "type_modifier");
jbe/bsw@0 1029 }
jbe/bsw@0 1030 lua_rawseti(L, 6, col+1);
jbe/bsw@0 1031 }
jbe@23 1032 lua_setfield(L, 5, "_column_info");
jbe@23 1033 // set "_rows_affected":
jbe/bsw@0 1034 {
jbe/bsw@0 1035 char *tmp;
jbe/bsw@0 1036 tmp = PQcmdTuples(res);
jbe/bsw@0 1037 if (tmp[0]) {
jbe/bsw@0 1038 lua_pushinteger(L, atoi(tmp));
jbe/bsw@0 1039 lua_setfield(L, 5, "_rows_affected");
jbe/bsw@0 1040 }
jbe/bsw@0 1041 }
jbe@23 1042 // set "_oid":
jbe/bsw@0 1043 {
jbe/bsw@0 1044 Oid tmp;
jbe/bsw@0 1045 tmp = PQoidValue(res);
jbe/bsw@0 1046 if (tmp != InvalidOid) {
jbe/bsw@0 1047 lua_pushinteger(L, tmp);
jbe/bsw@0 1048 lua_setfield(L, 5, "_oid");
jbe/bsw@0 1049 }
jbe/bsw@0 1050 }
jbe@23 1051 // copy data as strings or nil, while performing binary unescaping
jbe@23 1052 // automatically:
jbe/bsw@0 1053 if (modes[command_idx] == MONDELEFANT_QUERY_MODE_LIST) {
jbe/bsw@0 1054 for (row = 0; row < rows; row++) {
jbe/bsw@0 1055 lua_pushcfunction(L, mondelefant_conn_create_object); // 6
jbe/bsw@0 1056 lua_pushvalue(L, 1); // 7
jbe/bsw@0 1057 lua_call(L, 1, 1); // 6
jbe/bsw@0 1058 for (col = 0; col < cols; col++) {
jbe/bsw@0 1059 if (PQgetisnull(res, row, col)) {
jbe/bsw@0 1060 lua_pushnil(L);
jbe/bsw@0 1061 } else if (binary[col]) {
jbe/bsw@0 1062 size_t binlen;
jbe/bsw@0 1063 char *binval;
jbe/bsw@0 1064 binval = (char *)PQunescapeBytea(
jbe/bsw@0 1065 (unsigned char *)PQgetvalue(res, row, col), &binlen
jbe/bsw@0 1066 );
jbe/bsw@0 1067 if (!binval) {
jbe/bsw@0 1068 return luaL_error(L,
jbe/bsw@0 1069 "Could not allocate memory for binary unescaping."
jbe/bsw@0 1070 );
jbe/bsw@0 1071 }
jbe/bsw@0 1072 lua_pushlstring(L, binval, binlen);
jbe/bsw@0 1073 PQfreemem(binval);
jbe/bsw@0 1074 } else {
jbe/bsw@0 1075 lua_pushstring(L, PQgetvalue(res, row, col));
jbe/bsw@0 1076 }
jbe/bsw@0 1077 lua_rawseti(L, 6, col+1);
jbe/bsw@0 1078 }
jbe/bsw@0 1079 lua_rawseti(L, 5, row+1);
jbe/bsw@0 1080 }
jbe/bsw@0 1081 } else if (rows == 1) {
jbe/bsw@0 1082 for (col = 0; col < cols; col++) {
jbe/bsw@0 1083 if (PQgetisnull(res, 0, col)) {
jbe/bsw@0 1084 lua_pushnil(L);
jbe/bsw@0 1085 } else if (binary[col]) {
jbe/bsw@0 1086 size_t binlen;
jbe/bsw@0 1087 char *binval;
jbe/bsw@0 1088 binval = (char *)PQunescapeBytea(
jbe/bsw@0 1089 (unsigned char *)PQgetvalue(res, 0, col), &binlen
jbe/bsw@0 1090 );
jbe/bsw@0 1091 if (!binval) {
jbe/bsw@0 1092 return luaL_error(L,
jbe/bsw@0 1093 "Could not allocate memory for binary unescaping."
jbe/bsw@0 1094 );
jbe/bsw@0 1095 }
jbe/bsw@0 1096 lua_pushlstring(L, binval, binlen);
jbe/bsw@0 1097 PQfreemem(binval);
jbe/bsw@0 1098 } else {
jbe/bsw@0 1099 lua_pushstring(L, PQgetvalue(res, 0, col));
jbe/bsw@0 1100 }
jbe/bsw@0 1101 lua_rawseti(L, 5, col+1);
jbe/bsw@0 1102 }
jbe/bsw@0 1103 } else {
jbe/bsw@0 1104 // no row in optrow mode
jbe/bsw@0 1105 lua_pop(L, 1);
jbe/bsw@0 1106 lua_pushnil(L);
jbe/bsw@0 1107 }
jbe@23 1108 // save result in result list:
jbe/bsw@0 1109 lua_rawseti(L, 4, command_idx+1);
jbe@23 1110 // extra assertion:
jbe/bsw@0 1111 if (lua_gettop(L) != 4) abort(); // should not happen
jbe@23 1112 // free memory acquired by libpq:
jbe/bsw@0 1113 PQclear(res);
jbe/bsw@0 1114 }
jbe/bsw@0 1115 // trace callback at stack position 3
jbe/bsw@0 1116 // result at stack position 4 (top of stack)
jbe@23 1117 // if a trace callback is existent, then call:
jbe/bsw@0 1118 if (lua_toboolean(L, 3)) {
jbe/bsw@0 1119 lua_pushvalue(L, 3);
jbe/bsw@0 1120 lua_call(L, 0, 0);
jbe/bsw@0 1121 }
jbe@23 1122 // put result at stack position 3:
jbe@23 1123 lua_replace(L, 3);
jbe@23 1124 // get output converter to stack position 4:
jbe@23 1125 lua_getfield(L, 1, "output_converter");
jbe@23 1126 // apply output converters and fill "_data" table according to column names:
jbe/bsw@0 1127 for (command_idx = 0; command_idx < command_count; command_idx++) {
jbe/bsw@0 1128 int mode;
jbe/bsw@0 1129 mode = modes[command_idx];
jbe/bsw@0 1130 lua_rawgeti(L, 3, command_idx+1); // raw result at stack position 5
jbe/bsw@0 1131 if (lua_toboolean(L, 5)) {
jbe/bsw@0 1132 lua_getfield(L, 5, "_column_info"); // column_info list at position 6
jbe/bsw@0 1133 cols = lua_objlen(L, 6);
jbe/bsw@0 1134 if (mode == MONDELEFANT_QUERY_MODE_LIST) {
jbe/bsw@0 1135 rows = lua_objlen(L, 5);
jbe/bsw@0 1136 for (row = 0; row < rows; row++) {
jbe/bsw@0 1137 lua_rawgeti(L, 5, row+1); // row at stack position 7
jbe/bsw@0 1138 lua_getfield(L, 7, "_data"); // _data table at stack position 8
jbe/bsw@0 1139 for (col = 0; col < cols; col++) {
jbe/bsw@0 1140 lua_rawgeti(L, 6, col+1); // this column info at position 9
jbe/bsw@0 1141 lua_getfield(L, 9, "field_name"); // 10
jbe/bsw@0 1142 if (lua_toboolean(L, 4)) {
jbe/bsw@0 1143 lua_pushvalue(L, 4); // output-converter
jbe/bsw@0 1144 lua_pushvalue(L, 1); // connection
jbe/bsw@0 1145 lua_rawgeti(L, 7, col+1); // raw-value
jbe/bsw@0 1146 lua_pushvalue(L, 9); // this column info
jbe/bsw@0 1147 lua_call(L, 3, 1); // converted value at position 11
jbe/bsw@0 1148 } else {
jbe/bsw@0 1149 lua_rawgeti(L, 7, col+1); // raw-value at position 11
jbe/bsw@0 1150 }
jbe/bsw@0 1151 lua_pushvalue(L, 11); // 12
jbe/bsw@0 1152 lua_rawseti(L, 7, col+1);
jbe/bsw@0 1153 lua_rawset(L, 8);
jbe/bsw@0 1154 lua_settop(L, 8);
jbe/bsw@0 1155 }
jbe/bsw@0 1156 lua_settop(L, 6);
jbe/bsw@0 1157 }
jbe/bsw@0 1158 } else {
jbe/bsw@0 1159 lua_getfield(L, 5, "_data"); // _data table at stack position 7
jbe/bsw@0 1160 for (col = 0; col < cols; col++) {
jbe/bsw@0 1161 lua_rawgeti(L, 6, col+1); // this column info at position 8
jbe/bsw@0 1162 lua_getfield(L, 8, "field_name"); // 9
jbe/bsw@0 1163 if (lua_toboolean(L, 4)) {
jbe/bsw@0 1164 lua_pushvalue(L, 4); // output-converter
jbe/bsw@0 1165 lua_pushvalue(L, 1); // connection
jbe/bsw@0 1166 lua_rawgeti(L, 5, col+1); // raw-value
jbe/bsw@0 1167 lua_pushvalue(L, 8); // this column info
jbe/bsw@0 1168 lua_call(L, 3, 1); // converted value at position 10
jbe/bsw@0 1169 } else {
jbe/bsw@0 1170 lua_rawgeti(L, 5, col+1); // raw-value at position 10
jbe/bsw@0 1171 }
jbe/bsw@0 1172 lua_pushvalue(L, 10); // 11
jbe/bsw@0 1173 lua_rawseti(L, 5, col+1);
jbe/bsw@0 1174 lua_rawset(L, 7);
jbe/bsw@0 1175 lua_settop(L, 7);
jbe/bsw@0 1176 }
jbe/bsw@0 1177 }
jbe/bsw@0 1178 }
jbe/bsw@0 1179 lua_settop(L, 4);
jbe/bsw@0 1180 }
jbe@23 1181 // return nil as first result value, followed by result lists/objects:
jbe/bsw@0 1182 lua_settop(L, 3);
jbe/bsw@0 1183 lua_pushnil(L);
jbe/bsw@0 1184 for (command_idx = 0; command_idx < command_count; command_idx++) {
jbe/bsw@0 1185 lua_rawgeti(L, 3, command_idx+1);
jbe/bsw@0 1186 }
jbe/bsw@0 1187 return command_count+1;
jbe/bsw@0 1188 }
jbe/bsw@0 1189
jbe@23 1190 // method "escalate" of error objects:
jbe/bsw@0 1191 static int mondelefant_errorobject_escalate(lua_State *L) {
jbe@23 1192 // check, if we may throw an error object instead of an error string:
jbe/bsw@0 1193 lua_settop(L, 1);
jbe/bsw@0 1194 lua_getfield(L, 1, "connection"); // 2
jbe/bsw@0 1195 lua_getfield(L, 2, "error_objects"); // 3
jbe/bsw@0 1196 if (lua_toboolean(L, 3)) {
jbe@23 1197 // throw error object:
jbe/bsw@0 1198 lua_settop(L, 1);
jbe/bsw@0 1199 return lua_error(L);
jbe/bsw@0 1200 } else {
jbe@23 1201 // throw error string:
jbe/bsw@0 1202 lua_getfield(L, 1, "message"); // 4
jbe/bsw@0 1203 if (lua_isnil(L, 4)) {
jbe/bsw@0 1204 return luaL_error(L, "No error message given for escalation.");
jbe/bsw@0 1205 }
jbe/bsw@0 1206 return lua_error(L);
jbe/bsw@0 1207 }
jbe/bsw@0 1208 }
jbe/bsw@0 1209
jbe@23 1210 // method "is_kind_of" of error objects:
jbe/bsw@0 1211 static int mondelefant_errorobject_is_kind_of(lua_State *L) {
jbe/bsw@0 1212 lua_settop(L, 2);
jbe/bsw@0 1213 lua_getfield(L, 1, "code"); // 3
jbe/bsw@0 1214 if (lua_isstring(L, 3)) {
jbe/bsw@0 1215 lua_pushboolean(L,
jbe/bsw@0 1216 mondelefant_check_error_class(
jbe/bsw@0 1217 lua_tostring(L, 3), luaL_checkstring(L, 2)
jbe/bsw@0 1218 )
jbe/bsw@0 1219 );
jbe/bsw@0 1220 } else {
jbe/bsw@0 1221 // only happens for errors where code is not set
jbe/bsw@0 1222 lua_pushboolean(L, 0);
jbe/bsw@0 1223 }
jbe/bsw@0 1224 return 1;
jbe/bsw@0 1225 }
jbe/bsw@0 1226
jbe@23 1227 // method "query" of database handles:
jbe/bsw@0 1228 static int mondelefant_conn_query(lua_State *L) {
jbe/bsw@0 1229 int argc;
jbe@23 1230 // count number of arguments:
jbe/bsw@0 1231 argc = lua_gettop(L);
jbe@23 1232 // insert "try_query" function/method at stack position 1:
jbe/bsw@0 1233 lua_pushcfunction(L, mondelefant_conn_try_query);
jbe@23 1234 lua_insert(L, 1);
jbe@23 1235 // call "try_query" method:
jbe@23 1236 lua_call(L, argc, LUA_MULTRET); // results (with error) starting at index 1
jbe@23 1237 // check, if error occurred:
jbe@23 1238 if (lua_toboolean(L, 1)) {
jbe@23 1239 // invoke escalate method of error object:
jbe/bsw@0 1240 lua_pushcfunction(L, mondelefant_errorobject_escalate);
jbe@23 1241 lua_pushvalue(L, 1);
jbe/bsw@0 1242 lua_call(L, 1, 0); // will raise an error
jbe/bsw@0 1243 return 0; // should not be executed
jbe/bsw@0 1244 } else {
jbe@23 1245 // return everything but nil error object:
jbe@23 1246 return lua_gettop(L) - 1;
jbe/bsw@0 1247 }
jbe/bsw@0 1248 }
jbe/bsw@0 1249
jbe@23 1250 // library function "set_class":
jbe/bsw@0 1251 static int mondelefant_set_class(lua_State *L) {
jbe@23 1252 // ensure that first argument is a database result list/object:
jbe/bsw@0 1253 lua_settop(L, 2);
jbe/bsw@0 1254 lua_getmetatable(L, 1); // 3
jbe/bsw@0 1255 lua_getfield(L, LUA_REGISTRYINDEX, MONDELEFANT_RESULT_MT_REGKEY); // 4
jbe/bsw@0 1256 luaL_argcheck(L, lua_equal(L, 3, 4), 1, "not a database result");
jbe@23 1257 // ensure that second argument is a database class (model):
jbe/bsw@0 1258 lua_settop(L, 2);
jbe/bsw@0 1259 lua_getmetatable(L, 2); // 3
jbe/bsw@0 1260 lua_getfield(L, LUA_REGISTRYINDEX, MONDELEFANT_CLASS_MT_REGKEY); // 4
jbe/bsw@0 1261 luaL_argcheck(L, lua_equal(L, 3, 4), 2, "not a database class");
jbe@23 1262 // set attribute "_class" of result list/object to given class:
jbe/bsw@0 1263 lua_settop(L, 2);
jbe/bsw@0 1264 lua_pushvalue(L, 2); // 3
jbe/bsw@0 1265 lua_setfield(L, 1, "_class");
jbe@23 1266 // test, if database result is a list (and not a single object):
jbe/bsw@0 1267 lua_getfield(L, 1, "_type"); // 3
jbe/bsw@0 1268 lua_pushliteral(L, "list"); // 4
jbe/bsw@0 1269 if (lua_rawequal(L, 3, 4)) {
jbe/bsw@0 1270 int i;
jbe@23 1271 // set attribute "_class" of all elements to given class:
jbe/bsw@0 1272 for (i=0; i < lua_objlen(L, 1); i++) {
jbe/bsw@0 1273 lua_settop(L, 2);
jbe/bsw@0 1274 lua_rawgeti(L, 1, i+1); // 3
jbe/bsw@0 1275 lua_pushvalue(L, 2); // 4
jbe/bsw@0 1276 lua_setfield(L, 3, "_class");
jbe/bsw@0 1277 }
jbe/bsw@0 1278 }
jbe@23 1279 // return first argument:
jbe/bsw@0 1280 lua_settop(L, 1);
jbe/bsw@0 1281 return 1;
jbe/bsw@0 1282 }
jbe/bsw@0 1283
jbe@23 1284 // library function "new_class":
jbe/bsw@0 1285 static int mondelefant_new_class(lua_State *L) {
jbe@23 1286 // use first argument as template or create new table:
jbe/bsw@0 1287 lua_settop(L, 1);
jbe/bsw@0 1288 if (!lua_toboolean(L, 1)) {
jbe/bsw@0 1289 lua_settop(L, 0);
jbe/bsw@0 1290 lua_newtable(L); // 1
jbe/bsw@0 1291 }
jbe@23 1292 // set meta-table for database classes (models):
jbe/bsw@0 1293 lua_getfield(L, LUA_REGISTRYINDEX, MONDELEFANT_CLASS_MT_REGKEY); // 2
jbe/bsw@0 1294 lua_setmetatable(L, 1);
jbe@23 1295 // check, if "prototype" attribute is not set:
jbe/bsw@0 1296 lua_pushliteral(L, "prototype"); // 2
jbe/bsw@0 1297 lua_rawget(L, 1); // 2
jbe/bsw@0 1298 if (!lua_toboolean(L, 2)) {
jbe@23 1299 // set "prototype" attribute to default prototype:
jbe/bsw@0 1300 lua_pushliteral(L, "prototype"); // 3
jbe/bsw@0 1301 lua_getfield(L,
jbe/bsw@0 1302 LUA_REGISTRYINDEX,
jbe/bsw@0 1303 MONDELEFANT_CLASS_PROTO_REGKEY
jbe/bsw@0 1304 ); // 4
jbe/bsw@0 1305 lua_rawset(L, 1);
jbe/bsw@0 1306 }
jbe@23 1307 // set "object" attribute to empty table, unless it is already set:
jbe/bsw@0 1308 lua_settop(L, 1);
jbe/bsw@0 1309 lua_pushliteral(L, "object"); // 2
jbe/bsw@0 1310 lua_rawget(L, 1); // 2
jbe/bsw@0 1311 if (!lua_toboolean(L, 2)) {
jbe/bsw@0 1312 lua_pushliteral(L, "object"); // 3
jbe/bsw@0 1313 lua_newtable(L); // 4
jbe/bsw@0 1314 lua_rawset(L, 1);
jbe/bsw@0 1315 }
jbe@23 1316 // set "object_get" attribute to empty table, unless it is already set:
jbe/bsw@0 1317 lua_settop(L, 1);
jbe/bsw@0 1318 lua_pushliteral(L, "object_get"); // 2
jbe/bsw@0 1319 lua_rawget(L, 1); // 2
jbe/bsw@0 1320 if (!lua_toboolean(L, 2)) {
jbe/bsw@0 1321 lua_pushliteral(L, "object_get"); // 3
jbe/bsw@0 1322 lua_newtable(L); // 4
jbe/bsw@0 1323 lua_rawset(L, 1);
jbe/bsw@0 1324 }
jbe@23 1325 // set "object_set" attribute to empty table, unless it is already set:
jbe/bsw@0 1326 lua_settop(L, 1);
jbe/bsw@0 1327 lua_pushliteral(L, "object_set"); // 2
jbe/bsw@0 1328 lua_rawget(L, 1); // 2
jbe/bsw@0 1329 if (!lua_toboolean(L, 2)) {
jbe/bsw@0 1330 lua_pushliteral(L, "object_set"); // 3
jbe/bsw@0 1331 lua_newtable(L); // 4
jbe/bsw@0 1332 lua_rawset(L, 1);
jbe/bsw@0 1333 }
jbe@23 1334 // set "list" attribute to empty table, unless it is already set:
jbe/bsw@0 1335 lua_settop(L, 1);
jbe/bsw@0 1336 lua_pushliteral(L, "list"); // 2
jbe/bsw@0 1337 lua_rawget(L, 1); // 2
jbe/bsw@0 1338 if (!lua_toboolean(L, 2)) {
jbe/bsw@0 1339 lua_pushliteral(L, "list"); // 3
jbe/bsw@0 1340 lua_newtable(L); // 4
jbe/bsw@0 1341 lua_rawset(L, 1);
jbe/bsw@0 1342 }
jbe@23 1343 // set "references" attribute to empty table, unless it is already set:
jbe/bsw@0 1344 lua_settop(L, 1);
jbe/bsw@0 1345 lua_pushliteral(L, "references"); // 2
jbe/bsw@0 1346 lua_rawget(L, 1); // 2
jbe/bsw@0 1347 if (!lua_toboolean(L, 2)) {
jbe/bsw@0 1348 lua_pushliteral(L, "references"); // 3
jbe/bsw@0 1349 lua_newtable(L); // 4
jbe/bsw@0 1350 lua_rawset(L, 1);
jbe/bsw@0 1351 }
jbe@23 1352 // set "foreign_keys" attribute to empty table, unless it is already set:
jbe/bsw@0 1353 lua_settop(L, 1);
jbe/bsw@0 1354 lua_pushliteral(L, "foreign_keys"); // 2
jbe/bsw@0 1355 lua_rawget(L, 1); // 2
jbe/bsw@0 1356 if (!lua_toboolean(L, 2)) {
jbe/bsw@0 1357 lua_pushliteral(L, "foreign_keys"); // 3
jbe/bsw@0 1358 lua_newtable(L); // 4
jbe/bsw@0 1359 lua_rawset(L, 1);
jbe/bsw@0 1360 }
jbe@23 1361 // return table:
jbe/bsw@0 1362 lua_settop(L, 1);
jbe/bsw@0 1363 return 1;
jbe/bsw@0 1364 }
jbe/bsw@0 1365
jbe@23 1366 // method "get_reference" of classes (models):
jbe/bsw@0 1367 static int mondelefant_class_get_reference(lua_State *L) {
jbe/bsw@0 1368 lua_settop(L, 2);
jbe/bsw@0 1369 while (lua_toboolean(L, 1)) {
jbe@23 1370 // get "references" table:
jbe/bsw@0 1371 lua_getfield(L, 1, "references"); // 3
jbe@23 1372 // perform lookup:
jbe/bsw@0 1373 lua_pushvalue(L, 2); // 4
jbe/bsw@0 1374 lua_gettable(L, 3); // 4
jbe@23 1375 // return result, if lookup was successful:
jbe/bsw@0 1376 if (!lua_isnil(L, 4)) return 1;
jbe@23 1377 // replace current table by its prototype:
jbe/bsw@0 1378 lua_settop(L, 2);
jbe/bsw@0 1379 lua_pushliteral(L, "prototype"); // 3
jbe/bsw@0 1380 lua_rawget(L, 1); // 3
jbe/bsw@0 1381 lua_replace(L, 1);
jbe/bsw@0 1382 }
jbe@23 1383 // return nothing:
jbe/bsw@0 1384 return 0;
jbe/bsw@0 1385 }
jbe/bsw@0 1386
jbe@23 1387 // method "iterate_over_references" of classes (models):
jbe/bsw@0 1388 static int mondelefant_class_iterate_over_references(lua_State *L) {
jbe/bsw@0 1389 return luaL_error(L, "Reference iterator not implemented yet."); // TODO
jbe/bsw@0 1390 }
jbe/bsw@0 1391
jbe@23 1392 // method "get_foreign_key_reference_name" of classes (models):
jbe/bsw@0 1393 static int mondelefant_class_get_foreign_key_reference_name(lua_State *L) {
jbe/bsw@0 1394 lua_settop(L, 2);
jbe/bsw@0 1395 while (lua_toboolean(L, 1)) {
jbe@23 1396 // get "foreign_keys" table:
jbe/bsw@0 1397 lua_getfield(L, 1, "foreign_keys"); // 3
jbe@23 1398 // perform lookup:
jbe/bsw@0 1399 lua_pushvalue(L, 2); // 4
jbe/bsw@0 1400 lua_gettable(L, 3); // 4
jbe@23 1401 // return result, if lookup was successful:
jbe/bsw@0 1402 if (!lua_isnil(L, 4)) return 1;
jbe@23 1403 // replace current table by its prototype:
jbe/bsw@0 1404 lua_settop(L, 2);
jbe/bsw@0 1405 lua_pushliteral(L, "prototype"); // 3
jbe/bsw@0 1406 lua_rawget(L, 1); // 3
jbe/bsw@0 1407 lua_replace(L, 1);
jbe/bsw@0 1408 }
jbe@23 1409 // return nothing:
jbe/bsw@0 1410 return 0;
jbe/bsw@0 1411 }
jbe/bsw@0 1412
jbe@23 1413 // meta-method "__index" of database result lists and objects:
jbe/bsw@0 1414 static int mondelefant_result_index(lua_State *L) {
jbe/bsw@0 1415 const char *result_type;
jbe@23 1416 // only lookup, when key is a string not beginning with an underscore:
jbe/bsw@0 1417 if (lua_type(L, 2) != LUA_TSTRING || lua_tostring(L, 2)[0] == '_') {
jbe@23 1418 return 0;
jbe/bsw@0 1419 }
jbe@23 1420 // get value of "_class" attribute, or default class, when unset:
jbe@23 1421 lua_settop(L, 2);
jbe/bsw@0 1422 lua_getfield(L, 1, "_class"); // 3
jbe/bsw@0 1423 if (!lua_toboolean(L, 3)) {
jbe/bsw@0 1424 lua_settop(L, 2);
jbe/bsw@0 1425 lua_getfield(L,
jbe/bsw@0 1426 LUA_REGISTRYINDEX,
jbe/bsw@0 1427 MONDELEFANT_CLASS_PROTO_REGKEY
jbe/bsw@0 1428 ); // 3
jbe/bsw@0 1429 }
jbe@23 1430 // get value of "_type" attribute:
jbe/bsw@0 1431 lua_getfield(L, 1, "_type"); // 4
jbe/bsw@0 1432 result_type = lua_tostring(L, 4);
jbe@23 1433 // different lookup for lists and objects:
jbe@23 1434 if (result_type && !strcmp(result_type, "object")) { // object
jbe/bsw@0 1435 lua_settop(L, 3);
jbe/bsw@0 1436 // try inherited attributes, methods or getter functions:
jbe/bsw@0 1437 while (lua_toboolean(L, 3)) {
jbe/bsw@0 1438 lua_getfield(L, 3, "object"); // 4
jbe/bsw@0 1439 lua_pushvalue(L, 2); // 5
jbe/bsw@0 1440 lua_gettable(L, 4); // 5
jbe/bsw@0 1441 if (!lua_isnil(L, 5)) return 1;
jbe/bsw@0 1442 lua_settop(L, 3);
jbe/bsw@0 1443 lua_getfield(L, 3, "object_get"); // 4
jbe/bsw@0 1444 lua_pushvalue(L, 2); // 5
jbe/bsw@0 1445 lua_gettable(L, 4); // 5
jbe/bsw@0 1446 if (lua_toboolean(L, 5)) {
jbe/bsw@0 1447 lua_pushvalue(L, 1); // 6
jbe/bsw@0 1448 lua_call(L, 1, 1); // 5
jbe/bsw@0 1449 return 1;
jbe/bsw@0 1450 }
jbe/bsw@0 1451 lua_settop(L, 3);
jbe/bsw@0 1452 lua_pushliteral(L, "prototype"); // 4
jbe/bsw@0 1453 lua_rawget(L, 3); // 4
jbe/bsw@0 1454 lua_replace(L, 3);
jbe/bsw@0 1455 }
jbe/bsw@0 1456 lua_settop(L, 2);
jbe/bsw@0 1457 // try primary keys of referenced objects:
jbe/bsw@0 1458 lua_pushcfunction(L,
jbe/bsw@0 1459 mondelefant_class_get_foreign_key_reference_name
jbe/bsw@0 1460 ); // 3
jbe/bsw@0 1461 lua_getfield(L, 1, "_class"); // 4
jbe/bsw@0 1462 lua_pushvalue(L, 2); // 5
jbe/bsw@0 1463 lua_call(L, 2, 1); // 3
jbe/bsw@0 1464 if (!lua_isnil(L, 3)) {
jbe/bsw@0 1465 // reference name at stack position 3
jbe/bsw@0 1466 lua_pushcfunction(L, mondelefant_class_get_reference); // 4
jbe/bsw@0 1467 lua_getfield(L, 1, "_class"); // 5
jbe/bsw@0 1468 lua_pushvalue(L, 3); // 6
jbe/bsw@0 1469 lua_call(L, 2, 1); // reference info at stack position 4
jbe/bsw@0 1470 lua_getfield(L, 1, "_ref"); // 5
jbe/bsw@0 1471 lua_getfield(L, 4, "ref"); // 6
jbe/bsw@0 1472 lua_gettable(L, 5); // 6
jbe/bsw@0 1473 if (!lua_isnil(L, 6)) {
jbe/bsw@0 1474 if (lua_toboolean(L, 6)) {
jbe/bsw@0 1475 lua_getfield(L, 4, "that_key"); // 7
jbe/bsw@0 1476 if (lua_isnil(L, 7)) {
jbe/bsw@0 1477 return luaL_error(L, "Missing 'that_key' entry in model reference.");
jbe/bsw@0 1478 }
jbe/bsw@0 1479 lua_gettable(L, 6); // 7
jbe/bsw@0 1480 } else {
jbe/bsw@0 1481 lua_pushnil(L);
jbe/bsw@0 1482 }
jbe/bsw@0 1483 return 1;
jbe/bsw@0 1484 }
jbe/bsw@0 1485 }
jbe/bsw@0 1486 lua_settop(L, 2);
jbe/bsw@0 1487 // try normal data field info:
jbe/bsw@0 1488 lua_getfield(L, 1, "_data"); // 3
jbe/bsw@0 1489 lua_pushvalue(L, 2); // 4
jbe/bsw@0 1490 lua_gettable(L, 3); // 4
jbe/bsw@0 1491 if (!lua_isnil(L, 4)) return 1;
jbe/bsw@0 1492 lua_settop(L, 2);
jbe/bsw@0 1493 // try cached referenced object (or cached NULL reference):
jbe/bsw@0 1494 lua_getfield(L, 1, "_ref"); // 3
jbe/bsw@0 1495 lua_pushvalue(L, 2); // 4
jbe/bsw@0 1496 lua_gettable(L, 3); // 4
jbe/bsw@0 1497 if (lua_isboolean(L, 4) && !lua_toboolean(L, 4)) {
jbe/bsw@0 1498 lua_pushnil(L);
jbe/bsw@0 1499 return 1;
jbe/bsw@0 1500 } else if (!lua_isnil(L, 4)) {
jbe/bsw@0 1501 return 1;
jbe/bsw@0 1502 }
jbe/bsw@0 1503 lua_settop(L, 2);
jbe/bsw@0 1504 // try to load a referenced object:
jbe/bsw@0 1505 lua_pushcfunction(L, mondelefant_class_get_reference); // 3
jbe/bsw@0 1506 lua_getfield(L, 1, "_class"); // 4
jbe/bsw@0 1507 lua_pushvalue(L, 2); // 5
jbe/bsw@0 1508 lua_call(L, 2, 1); // 3
jbe/bsw@0 1509 if (!lua_isnil(L, 3)) {
jbe/bsw@0 1510 lua_settop(L, 2);
jbe/bsw@0 1511 lua_getfield(L, 1, "load"); // 3
jbe/bsw@0 1512 lua_pushvalue(L, 1); // 4 (self)
jbe/bsw@0 1513 lua_pushvalue(L, 2); // 5
jbe/bsw@0 1514 lua_call(L, 2, 0);
jbe/bsw@0 1515 lua_settop(L, 2);
jbe/bsw@0 1516 lua_getfield(L, 1, "_ref"); // 3
jbe/bsw@0 1517 lua_pushvalue(L, 2); // 4
jbe/bsw@0 1518 lua_gettable(L, 3); // 4
jbe/bsw@0 1519 if (lua_isboolean(L, 4) && !lua_toboolean(L, 4)) lua_pushnil(L); // TODO: use special object instead of false
jbe/bsw@0 1520 return 1;
jbe/bsw@0 1521 }
jbe/bsw@0 1522 return 0;
jbe@23 1523 } else if (result_type && !strcmp(result_type, "list")) { // list
jbe/bsw@0 1524 lua_settop(L, 3);
jbe/bsw@0 1525 // try inherited list attributes or methods:
jbe/bsw@0 1526 while (lua_toboolean(L, 3)) {
jbe/bsw@0 1527 lua_getfield(L, 3, "list"); // 4
jbe/bsw@0 1528 lua_pushvalue(L, 2); // 5
jbe/bsw@0 1529 lua_gettable(L, 4); // 5
jbe/bsw@0 1530 if (!lua_isnil(L, 5)) return 1;
jbe/bsw@0 1531 lua_settop(L, 3);
jbe/bsw@0 1532 lua_pushliteral(L, "prototype"); // 4
jbe/bsw@0 1533 lua_rawget(L, 3); // 4
jbe/bsw@0 1534 lua_replace(L, 3);
jbe/bsw@0 1535 }
jbe/bsw@0 1536 }
jbe@23 1537 // return nothing:
jbe/bsw@0 1538 return 0;
jbe/bsw@0 1539 }
jbe/bsw@0 1540
jbe@23 1541 // meta-method "__newindex" of database result lists and objects:
jbe/bsw@0 1542 static int mondelefant_result_newindex(lua_State *L) {
jbe/bsw@0 1543 const char *result_type;
jbe@23 1544 // perform rawset, unless key is a string not starting with underscore:
jbe/bsw@0 1545 lua_settop(L, 3);
jbe/bsw@0 1546 if (lua_type(L, 2) != LUA_TSTRING || lua_tostring(L, 2)[0] == '_') {
jbe/bsw@0 1547 lua_rawset(L, 1);
jbe/bsw@0 1548 return 1;
jbe/bsw@0 1549 }
jbe@23 1550 // get value of "_class" attribute, or default class, when unset:
jbe/bsw@0 1551 lua_getfield(L, 1, "_class"); // 4
jbe/bsw@0 1552 if (!lua_toboolean(L, 4)) {
jbe/bsw@0 1553 lua_settop(L, 3);
jbe/bsw@0 1554 lua_getfield(L,
jbe/bsw@0 1555 LUA_REGISTRYINDEX,
jbe/bsw@0 1556 MONDELEFANT_CLASS_PROTO_REGKEY
jbe/bsw@0 1557 ); // 4
jbe/bsw@0 1558 }
jbe@23 1559 // get value of "_type" attribute:
jbe/bsw@0 1560 lua_getfield(L, 1, "_type"); // 5
jbe/bsw@0 1561 result_type = lua_tostring(L, 5);
jbe@23 1562 // distinguish between lists and objects:
jbe@23 1563 if (result_type && !strcmp(result_type, "object")) { // objects
jbe/bsw@0 1564 lua_settop(L, 4);
jbe/bsw@0 1565 // try object setter functions:
jbe/bsw@0 1566 while (lua_toboolean(L, 4)) {
jbe/bsw@0 1567 lua_getfield(L, 4, "object_set"); // 5
jbe/bsw@0 1568 lua_pushvalue(L, 2); // 6
jbe/bsw@0 1569 lua_gettable(L, 5); // 6
jbe/bsw@0 1570 if (lua_toboolean(L, 6)) {
jbe/bsw@0 1571 lua_pushvalue(L, 1); // 7
jbe/bsw@0 1572 lua_pushvalue(L, 3); // 8
jbe/bsw@0 1573 lua_call(L, 2, 0);
jbe/bsw@0 1574 return 0;
jbe/bsw@0 1575 }
jbe/bsw@0 1576 lua_settop(L, 4);
jbe/bsw@0 1577 lua_pushliteral(L, "prototype"); // 5
jbe/bsw@0 1578 lua_rawget(L, 4); // 5
jbe/bsw@0 1579 lua_replace(L, 4);
jbe/bsw@0 1580 }
jbe/bsw@0 1581 lua_settop(L, 3);
jbe/bsw@0 1582 // check, if a object reference is changed:
jbe/bsw@0 1583 lua_pushcfunction(L, mondelefant_class_get_reference); // 4
jbe/bsw@0 1584 lua_getfield(L, 1, "_class"); // 5
jbe/bsw@0 1585 lua_pushvalue(L, 2); // 6
jbe/bsw@0 1586 lua_call(L, 2, 1); // 4
jbe/bsw@0 1587 if (!lua_isnil(L, 4)) {
jbe/bsw@0 1588 // store object in _ref table (use false for nil): // TODO: use special object instead of false
jbe/bsw@0 1589 lua_getfield(L, 1, "_ref"); // 5
jbe/bsw@0 1590 lua_pushvalue(L, 2); // 6
jbe/bsw@0 1591 if (lua_isnil(L, 3)) lua_pushboolean(L, 0); // 7
jbe/bsw@0 1592 else lua_pushvalue(L, 3); // 7
jbe/bsw@0 1593 lua_settable(L, 5);
jbe/bsw@0 1594 lua_settop(L, 4);
jbe/bsw@0 1595 // delete referencing key from _data table:
jbe/bsw@0 1596 lua_getfield(L, 4, "this_key"); // 5
jbe/bsw@0 1597 if (lua_isnil(L, 5)) {
jbe/bsw@0 1598 return luaL_error(L, "Missing 'this_key' entry in model reference.");
jbe/bsw@0 1599 }
jbe/bsw@0 1600 lua_getfield(L, 1, "_data"); // 6
jbe/bsw@0 1601 lua_pushvalue(L, 5); // 7
jbe/bsw@0 1602 lua_pushnil(L); // 8
jbe/bsw@0 1603 lua_settable(L, 6);
jbe/bsw@0 1604 lua_settop(L, 5);
jbe/bsw@0 1605 lua_getfield(L, 1, "_dirty"); // 6
jbe/bsw@0 1606 lua_pushvalue(L, 5); // 7
jbe/bsw@0 1607 lua_pushboolean(L, 1); // 8
jbe/bsw@0 1608 lua_settable(L, 6);
jbe/bsw@0 1609 return 0;
jbe/bsw@0 1610 }
jbe/bsw@0 1611 lua_settop(L, 3);
jbe/bsw@0 1612 // store value in data field info:
jbe/bsw@0 1613 lua_getfield(L, 1, "_data"); // 4
jbe/bsw@0 1614 lua_pushvalue(L, 2); // 5
jbe/bsw@0 1615 lua_pushvalue(L, 3); // 6
jbe/bsw@0 1616 lua_settable(L, 4);
jbe/bsw@0 1617 lua_settop(L, 3);
jbe/bsw@0 1618 // mark field as dirty (needs to be UPDATEd on save):
jbe/bsw@0 1619 lua_getfield(L, 1, "_dirty"); // 4
jbe/bsw@0 1620 lua_pushvalue(L, 2); // 5
jbe/bsw@0 1621 lua_pushboolean(L, 1); // 6
jbe/bsw@0 1622 lua_settable(L, 4);
jbe/bsw@0 1623 lua_settop(L, 3);
jbe/bsw@0 1624 // reset reference cache, if neccessary:
jbe/bsw@0 1625 lua_pushcfunction(L,
jbe/bsw@0 1626 mondelefant_class_get_foreign_key_reference_name
jbe/bsw@0 1627 ); // 4
jbe/bsw@0 1628 lua_getfield(L, 1, "_class"); // 5
jbe/bsw@0 1629 lua_pushvalue(L, 2); // 6
jbe/bsw@0 1630 lua_call(L, 2, 1); // 4
jbe/bsw@0 1631 if (!lua_isnil(L, 4)) {
jbe/bsw@0 1632 lua_getfield(L, 1, "_ref"); // 5
jbe/bsw@0 1633 lua_pushvalue(L, 4); // 6
jbe/bsw@0 1634 lua_pushnil(L); // 7
jbe/bsw@0 1635 lua_settable(L, 5);
jbe/bsw@0 1636 }
jbe/bsw@0 1637 return 0;
jbe@23 1638 } else { // non-objects (i.e. lists)
jbe@23 1639 // perform rawset:
jbe/bsw@0 1640 lua_settop(L, 3);
jbe/bsw@0 1641 lua_rawset(L, 1);
jbe/bsw@0 1642 return 0;
jbe/bsw@0 1643 }
jbe/bsw@0 1644 }
jbe/bsw@0 1645
jbe@23 1646 // meta-method "__index" of classes (models):
jbe/bsw@0 1647 static int mondelefant_class_index(lua_State *L) {
jbe@23 1648 // perform lookup in prototype:
jbe/bsw@0 1649 lua_settop(L, 2);
jbe/bsw@0 1650 lua_pushliteral(L, "prototype"); // 3
jbe/bsw@0 1651 lua_rawget(L, 1); // 3
jbe/bsw@0 1652 lua_pushvalue(L, 2); // 4
jbe/bsw@0 1653 lua_gettable(L, 3); // 4
jbe/bsw@0 1654 return 1;
jbe/bsw@0 1655 }
jbe/bsw@0 1656
jbe@23 1657 // registration information for functions of library:
jbe/bsw@0 1658 static const struct luaL_Reg mondelefant_module_functions[] = {
jbe/bsw@0 1659 {"connect", mondelefant_connect},
jbe/bsw@0 1660 {"set_class", mondelefant_set_class},
jbe/bsw@0 1661 {"new_class", mondelefant_new_class},
jbe/bsw@0 1662 {NULL, NULL}
jbe/bsw@0 1663 };
jbe/bsw@0 1664
jbe@23 1665 // registration information for meta-methods of database connections:
jbe/bsw@0 1666 static const struct luaL_Reg mondelefant_conn_mt_functions[] = {
jbe/bsw@0 1667 {"__gc", mondelefant_conn_free},
jbe/bsw@0 1668 {"__index", mondelefant_conn_index},
jbe/bsw@0 1669 {"__newindex", mondelefant_conn_newindex},
jbe/bsw@0 1670 {NULL, NULL}
jbe/bsw@0 1671 };
jbe/bsw@0 1672
jbe@23 1673 // registration information for methods of database connections:
jbe/bsw@0 1674 static const struct luaL_Reg mondelefant_conn_methods[] = {
jbe/bsw@0 1675 {"close", mondelefant_conn_close},
jbe/bsw@0 1676 {"is_ok", mondelefant_conn_is_ok},
jbe/bsw@0 1677 {"get_transaction_status", mondelefant_conn_get_transaction_status},
jbe/bsw@0 1678 {"create_list", mondelefant_conn_create_list},
jbe/bsw@0 1679 {"create_object", mondelefant_conn_create_object},
jbe/bsw@0 1680 {"quote_string", mondelefant_conn_quote_string},
jbe/bsw@0 1681 {"quote_binary", mondelefant_conn_quote_binary},
jbe/bsw@0 1682 {"assemble_command", mondelefant_conn_assemble_command},
jbe/bsw@0 1683 {"try_query", mondelefant_conn_try_query},
jbe/bsw@0 1684 {"query", mondelefant_conn_query},
jbe/bsw@0 1685 {NULL, NULL}
jbe/bsw@0 1686 };
jbe/bsw@0 1687
jbe@23 1688 // registration information for meta-methods of error objects:
jbe/bsw@0 1689 static const struct luaL_Reg mondelefant_errorobject_mt_functions[] = {
jbe/bsw@0 1690 {NULL, NULL}
jbe/bsw@0 1691 };
jbe/bsw@0 1692
jbe@23 1693 // registration information for methods of error objects:
jbe/bsw@0 1694 static const struct luaL_Reg mondelefant_errorobject_methods[] = {
jbe/bsw@0 1695 {"escalate", mondelefant_errorobject_escalate},
jbe/bsw@0 1696 {"is_kind_of", mondelefant_errorobject_is_kind_of},
jbe/bsw@0 1697 {NULL, NULL}
jbe/bsw@0 1698 };
jbe/bsw@0 1699
jbe@23 1700 // registration information for meta-methods of database result lists/objects:
jbe/bsw@0 1701 static const struct luaL_Reg mondelefant_result_mt_functions[] = {
jbe/bsw@0 1702 {"__index", mondelefant_result_index},
jbe/bsw@0 1703 {"__newindex", mondelefant_result_newindex},
jbe/bsw@0 1704 {NULL, NULL}
jbe/bsw@0 1705 };
jbe/bsw@0 1706
jbe@23 1707 // registration information for methods of database result lists/objects:
jbe/bsw@0 1708 static const struct luaL_Reg mondelefant_class_mt_functions[] = {
jbe/bsw@0 1709 {"__index", mondelefant_class_index},
jbe/bsw@0 1710 {NULL, NULL}
jbe/bsw@0 1711 };
jbe/bsw@0 1712
jbe@23 1713 // registration information for methods of classes (models):
jbe/bsw@0 1714 static const struct luaL_Reg mondelefant_class_methods[] = {
jbe/bsw@0 1715 {"get_reference", mondelefant_class_get_reference},
jbe/bsw@0 1716 {"iterate_over_references", mondelefant_class_iterate_over_references},
jbe/bsw@0 1717 {"get_foreign_key_reference_name",
jbe/bsw@0 1718 mondelefant_class_get_foreign_key_reference_name},
jbe/bsw@0 1719 {NULL, NULL}
jbe/bsw@0 1720 };
jbe/bsw@0 1721
jbe@23 1722 // registration information for methods of database result objects (not lists!):
jbe/bsw@0 1723 static const struct luaL_Reg mondelefant_object_methods[] = {
jbe/bsw@0 1724 {NULL, NULL}
jbe/bsw@0 1725 };
jbe/bsw@0 1726
jbe@23 1727 // registration information for methods of database result lists (not single objects!):
jbe/bsw@0 1728 static const struct luaL_Reg mondelefant_list_methods[] = {
jbe/bsw@0 1729 {NULL, NULL}
jbe/bsw@0 1730 };
jbe/bsw@0 1731
jbe@23 1732 // luaopen function to initialize/register library:
jbe/bsw@0 1733 int luaopen_mondelefant_native(lua_State *L) {
jbe/bsw@0 1734 lua_settop(L, 0);
jbe/bsw@0 1735 lua_newtable(L); // module at stack position 1
jbe/bsw@0 1736 luaL_register(L, NULL, mondelefant_module_functions);
jbe/bsw@0 1737
jbe/bsw@0 1738 lua_pushvalue(L, 1); // 2
jbe/bsw@0 1739 lua_setfield(L, LUA_REGISTRYINDEX, MONDELEFANT_MODULE_REGKEY);
jbe/bsw@0 1740
jbe/bsw@0 1741 lua_newtable(L); // 2
jbe/bsw@0 1742 // NOTE: only PostgreSQL is supported yet:
jbe/bsw@0 1743 luaL_register(L, NULL, mondelefant_conn_methods);
jbe/bsw@0 1744 lua_setfield(L, 1, "postgresql_connection_prototype");
jbe/bsw@0 1745 lua_newtable(L); // 2
jbe/bsw@0 1746 lua_setfield(L, 1, "connection_prototype");
jbe/bsw@0 1747
jbe/bsw@0 1748 luaL_newmetatable(L, MONDELEFANT_CONN_MT_REGKEY); // 2
jbe/bsw@0 1749 luaL_register(L, NULL, mondelefant_conn_mt_functions);
jbe/bsw@0 1750 lua_settop(L, 1);
jbe/bsw@0 1751 luaL_newmetatable(L, MONDELEFANT_RESULT_MT_REGKEY); // 2
jbe/bsw@0 1752 luaL_register(L, NULL, mondelefant_result_mt_functions);
jbe/bsw@0 1753 lua_setfield(L, 1, "result_metatable");
jbe/bsw@0 1754 luaL_newmetatable(L, MONDELEFANT_CLASS_MT_REGKEY); // 2
jbe/bsw@0 1755 luaL_register(L, NULL, mondelefant_class_mt_functions);
jbe/bsw@0 1756 lua_setfield(L, 1, "class_metatable");
jbe/bsw@0 1757
jbe/bsw@0 1758 lua_newtable(L); // 2
jbe/bsw@0 1759 lua_newtable(L); // 3
jbe/bsw@0 1760 luaL_register(L, NULL, mondelefant_object_methods);
jbe/bsw@0 1761 lua_setfield(L, 2, "object");
jbe/bsw@0 1762 lua_newtable(L); // 3
jbe/bsw@0 1763 lua_setfield(L, 2, "object_get");
jbe/bsw@0 1764 lua_newtable(L); // 3
jbe/bsw@0 1765 lua_setfield(L, 2, "object_set");
jbe/bsw@0 1766 lua_newtable(L); // 3
jbe/bsw@0 1767 luaL_register(L, NULL, mondelefant_list_methods);
jbe/bsw@0 1768 lua_setfield(L, 2, "list");
jbe/bsw@0 1769 lua_newtable(L); // 3
jbe/bsw@0 1770 lua_setfield(L, 2, "references");
jbe/bsw@0 1771 lua_newtable(L); // 3
jbe/bsw@0 1772 lua_setfield(L, 2, "foreign_keys");
jbe/bsw@0 1773 lua_pushvalue(L, 2); // 3
jbe/bsw@0 1774 lua_setfield(L, LUA_REGISTRYINDEX, MONDELEFANT_CLASS_PROTO_REGKEY);
jbe/bsw@0 1775 lua_setfield(L, 1, "class_prototype");
jbe/bsw@0 1776
jbe/bsw@0 1777 lua_newtable(L); // 2
jbe/bsw@0 1778 lua_pushliteral(L, "k"); // 3
jbe/bsw@0 1779 lua_setfield(L, 2, "__mode");
jbe/bsw@0 1780 lua_newtable(L); // 3
jbe/bsw@0 1781 lua_pushvalue(L, 2); // 4
jbe/bsw@0 1782 lua_setmetatable(L, 3);
jbe/bsw@0 1783 lua_setfield(L, LUA_REGISTRYINDEX, MONDELEFANT_CONN_DATA_REGKEY);
jbe/bsw@0 1784 lua_settop(L, 1);
jbe/bsw@0 1785
jbe/bsw@0 1786 luaL_newmetatable(L, MONDELEFANT_ERROROBJECT_MT_REGKEY); // 2
jbe/bsw@0 1787 luaL_register(L, NULL, mondelefant_errorobject_mt_functions);
jbe/bsw@0 1788 lua_newtable(L); // 3
jbe/bsw@0 1789 luaL_register(L, NULL, mondelefant_errorobject_methods);
jbe/bsw@0 1790 lua_setfield(L, 2, "__index");
jbe/bsw@0 1791 lua_setfield(L, 1, "errorobject_metatable");
jbe/bsw@0 1792
jbe/bsw@0 1793 return 1;
jbe/bsw@0 1794 }

Impressum / About Us