webmcp: libraries/mondelefant/mondelefant

webmcp

annotate libraries/mondelefant/mondelefant_native.c @ 394:912a1b9c2551

Documentation of return values of <db_handle>:wait(...) call in case of error or timeout

author	jbe
date	Wed Dec 09 03:46:43 2015 +0100 (2015-12-09)
parents	b2d2ff6e4385
children	762ab1e87702

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