pd.if.org Git - zpackage/blob - lib/dbquery.c

   1 #define _POSIX_C_SOURCE 200809L
   2
   3 #include <stdarg.h>
   4 #include <string.h>
   5
   6 #include "sqlite3.h"
   7 #include "zpm.h"
   8
   9 #define SQLERROR(x) fprintf(stderr, "%s %d: %s\n", __func__, __LINE__, (x))
  10
  11 sqlite3_stmt *zpm_dbqueryv(struct zpm *zpm, char *query, va_list args) {
  12         sqlite3 *db;
  13         char *sql;
  14         sqlite3_stmt *st;
  15         int rv;
  16
  17         if (!zpm || zpm->error) {
  18                 return 0;
  19         }
  20
  21         if (!zpm->db) {
  22                 zpm->error = 1;
  23                 return 0;
  24         }
  25
  26         db = zpm->db;
  27
  28         sql = sqlite3_vmprintf(query, args);
  29
  30         if (!sql) {
  31                 zpm->error = 1;
  32                 return 0;
  33         }
  34
  35         rv = sqlite3_prepare_v2(db, sql, strlen(sql), &st, NULL);
  36         if (rv != SQLITE_OK) {
  37                 SQLERROR(sqlite3_errmsg(db));
  38                 zpm->error = rv;
  39                 return 0;
  40         }
  41
  42         return st;
  43 }
  44
  45 sqlite3_stmt *zpm_dbquery(struct zpm *zpm, char *query, ...) {
  46         va_list args;
  47         sqlite3_stmt *st;
  48
  49         if (!zpm || zpm->error) {
  50                 return 0;
  51         }
  52
  53         if (!zpm->db) {
  54                 zpm->error = 1;
  55                 return 0;
  56         }
  57
  58         va_start(args, query);
  59         st = zpm_dbqueryv(zpm, query, args);
  60         va_end(args);
  61
  62         return st;
  63 }
  64
  65 char *zpm_db_string(struct zpm *zpm, char *query, ...) {
  66         sqlite3_stmt *st;
  67         va_list args;
  68         int rv;
  69         char *result = 0;
  70
  71         va_start(args, query);
  72         st = zpm_dbqueryv(zpm, query, args);
  73         va_end(args);
  74
  75         rv = sqlite3_step(st);
  76
  77         if (rv == SQLITE_ROW) {
  78                 result = (char *)sqlite3_column_text(st, 0);
  79                 if (result) {
  80                         result = strdup(result);
  81                 }
  82         }
  83
  84         sqlite3_finalize(st);
  85         return result;
  86 }