uwsgi/plugins/sqlite3/plugin.c

#include <uwsgi.h>
#include <sqlite3.h>

extern struct uwsgi_server uwsgi;

static void uwsgi_sqlite3_config(char *, char *[]);
static void uwsgi_opt_load_sqlite3(char *opt, char *filename, void *none) {
        config_magic_table_fill(filename, uwsgi.magic_table);
        uwsgi_sqlite3_config(filename, uwsgi.magic_table);
}

static struct uwsgi_option uwsgi_sqlite3_options[] = {
        {"sqlite3", required_argument, 0, "load config from sqlite3 db", uwsgi_opt_load_sqlite3, NULL, UWSGI_OPT_IMMEDIATE},
        {"sqlite", required_argument, 0, "load config from sqlite3 db", uwsgi_opt_load_sqlite3, NULL, UWSGI_OPT_IMMEDIATE},
	{0, 0, 0, 0, 0, 0, 0},
};

static int uwsgi_sqlite3_config_callback(void *magic_table, int field_count, char **fields, char **col) {
        // make a copy of the string
        if (field_count >= 2) {
                size_t value_len = strlen(fields[1]) + 1;
                char *value = magic_sub(fields[1], value_len, &value_len, (char **) magic_table);
                add_exported_option(uwsgi_strncopy(fields[0], strlen(fields[0])), value, 0);
        }

        return 0;
}

static void uwsgi_sqlite3_config(char *file, char *magic_table[]) {

        sqlite3 *db;
        char *err = NULL;
        char *query = "SELECT * FROM uwsgi";

        char *colon = uwsgi_get_last_char(file, ':');

        if (colon) {
                colon[0] = 0;
                if (colon[1] != 0) {
                        query = colon + 1;
                }
        }

        uwsgi_log("[uWSGI] getting sqlite3 configuration from %s\n", file);

#ifdef sqlite3_open_v2
        if (sqlite3_open_v2(file, &db, SQLITE_OPEN_READONLY, NULL)) {
#else
        if (sqlite3_open(file, &db)) {
#endif
                uwsgi_log("unable to open sqlite3 db: %s\n", sqlite3_errmsg(db));
                sqlite3_close(db);
                exit(1);
        }

        if (sqlite3_exec(db, query, uwsgi_sqlite3_config_callback, (void *) magic_table, &err)) {
                uwsgi_log("sqlite3 error: %s\n", err);
                sqlite3_close(db);
                exit(1);
        }

        sqlite3_close(db);

}


struct uwsgi_plugin sqlite3_plugin = {
	.name = "sqlite3",
	.options = uwsgi_sqlite3_options,
};