Function dbtable_insert

lua/sp.c:2424–2504

}

static int dbtable_insert(Lua lua)
{
    SP sp = getsp(lua);
    int rc, nargs;
    strbuf *columns, *params, *sql;

    nargs = lua_gettop(lua);
    if (nargs != 2) {
        return luabb_error(lua, NULL, "bad arguments to 'insert'");
    }

    luaL_checkudata(lua, 1, dbtypes.dbtable);
    dbtable_t *table = lua_touserdata(lua, 1);

    columns = strbuf_new();
    params = strbuf_new();

    strbuf_append(columns, "(");
    strbuf_append(params, "(");

    luaL_checktype(lua, 2, LUA_TTABLE);

    // Iterate lua table to create prepared statement
    lua_pushnil(lua);
    while (lua_next(lua, 2)) {
        const char *col = lua_tostring(lua, -2);
        char *quoted_col = sqlite3_mprintf("\"%w\",", col);
        strbuf_append(columns, quoted_col);
        sqlite3_free(quoted_col);
        strbuf_appendf(params, "?,");
        lua_pop(lua, 1);
    }
    strbuf_del(columns, 1);
    strbuf_del(params, 1);

    strbuf_append(columns, ")");
    strbuf_append(params, ")");

    char n1[MAXTABLELEN], separator[2], n2[MAXTABLELEN];
    query_tbl_name(table->table_name, n1, separator, n2);
    sql = strbuf_new();
    strbuf_appendf(sql, "INSERT INTO %s%s\"%s\" %s VALUES %s", n1, separator,
                   n2, strbuf_buf(columns), strbuf_buf(params));
    char *sqlstr = strbuf_disown(sql);

    strbuf_free(sql);
    strbuf_free(columns);
    strbuf_free(params);

    db_reset(lua);
    sqlite3_stmt *stmt = NULL;
    rc = lua_prepare_sql(sp, sqlstr, &stmt);
    free(sqlstr);
    if (rc != 0) {
        lua_pushinteger(lua, rc); /* Failure return code. */
        return 1;
    }