add function to return one string column

[zpackage] / lib / dbquery.c

diff --git a/lib/dbquery.c b/lib/dbquery.c
index 3f818418d88486e477df17a7544f8bfc04f1efe3..ab03984697b2f255fc88aaa9ebc9d0c0a8d78ba5 100644 (file)
--- a/lib/dbquery.c
+++ b/lib/dbquery.c
@@ -1,3 +1,5 @@
+#define _POSIX_C_SOURCE 200809L
+
 #include <stdarg.h>
 #include <string.h>
 
@@ -5,10 +7,10 @@
 #include "zpm.h"
 
 #define SQLERROR(x) fprintf(stderr, "%s %d: %s\n", __func__, __LINE__, (x))
-sqlite3_stmt *zpm_dbquery(struct zpm *zpm, char *query, ...) {
+
+sqlite3_stmt *zpm_dbqueryv(struct zpm *zpm, char *query, va_list args) {
        sqlite3 *db;
        char *sql;
-       va_list args;
        sqlite3_stmt *st;
        int rv;
 
@@ -23,9 +25,7 @@ sqlite3_stmt *zpm_dbquery(struct zpm *zpm, char *query, ...) {
 
        db = zpm->db;
 
-       va_start(args, query);
        sql = sqlite3_vmprintf(query, args);
-       va_end(args);
 
        if (!sql) {
                zpm->error = 1;
@@ -41,3 +41,46 @@ sqlite3_stmt *zpm_dbquery(struct zpm *zpm, char *query, ...) {
 
        return st;
 }
+
+sqlite3_stmt *zpm_dbquery(struct zpm *zpm, char *query, ...) {
+       va_list args;
+       sqlite3_stmt *st;
+
+       if (!zpm || zpm->error) {
+               return 0;
+       }
+
+       if (!zpm->db) {
+               zpm->error = 1;
+               return 0;
+       }
+
+       va_start(args, query);
+       st = zpm_dbqueryv(zpm, query, args);
+       va_end(args);
+
+       return st;
+}
+
+char *zpm_db_string(struct zpm *zpm, char *query, ...) {
+       sqlite3_stmt *st;
+       va_list args;
+       int rv;
+       char *result = 0;
+
+       va_start(args, query);
+       st = zpm_dbqueryv(zpm, query, args);
+       va_end(args);
+
+       rv = sqlite3_step(st);
+
+       if (rv == SQLITE_ROW) {
+               result = (char *)sqlite3_column_text(st, 0);
+               if (result) {
+                       result = strdup(result);
+               }
+       }
+
+       sqlite3_finalize(st);
+       return result;
+}