src/lib/lightmediascanner_db_image.c

   1 #include <lightmediascanner_db.h>
   2 #include "lightmediascanner_db_private.h"
   3 #include <stdlib.h>
   4 #include <stdio.h>
   5
   6 struct lms_db_image {
   7     sqlite3 *db;
   8     sqlite3_stmt *insert;
   9     unsigned int _references;
  10     unsigned int _is_started:1;
  11 };
  12
  13 static lms_db_image_t *_singleton = NULL;
  14
  15 static int
  16 _db_table_updater_images_0(sqlite3 *db, const char *table, unsigned int current_version, int is_last_run) {
  17     char *errmsg;
  18     int r, ret;
  19
  20     errmsg = NULL;
  21     r = sqlite3_exec(db,
  22                      "CREATE TABLE IF NOT EXISTS images ("
  23                      "id INTEGER PRIMARY KEY, "
  24                      "title TEXT, "
  25                      "artist TEXT, "
  26                      "date INTEGER NOT NULL, "
  27                      "width INTEGER NOT NULL, "
  28                      "height INTEGER NOT NULL, "
  29                      "orientation INTEGER NOT NULL, "
  30                      "gps_lat REAL DEFAULT 0.0, "
  31                      "gps_long REAL DEFAULT 0.0, "
  32                      "gps_alt REAL DEFAULT 0.0"
  33                      ")",
  34                      NULL, NULL, &errmsg);
  35     if (r != SQLITE_OK) {
  36         fprintf(stderr, "ERROR: could not create 'images' table: %s\n", errmsg);
  37         sqlite3_free(errmsg);
  38         return -1;
  39     }
  40
  41     r = sqlite3_exec(db,
  42                      "CREATE INDEX IF NOT EXISTS images_date_idx ON images ("
  43                      "date"
  44                      ")",
  45                      NULL, NULL, &errmsg);
  46     if (r != SQLITE_OK) {
  47         fprintf(stderr, "ERROR: could not create 'images_date_idx' index: %s\n",
  48                 errmsg);
  49         sqlite3_free(errmsg);
  50         return -2;
  51     }
  52
  53     ret = lms_db_create_trigger_if_not_exists(db,
  54         "delete_images_on_files_deleted "
  55         "DELETE ON files FOR EACH ROW BEGIN "
  56         " DELETE FROM images WHERE id = OLD.id; END;");
  57     if (ret != 0)
  58         goto done;
  59
  60     ret = lms_db_create_trigger_if_not_exists(db,
  61         "delete_files_on_images_deleted "
  62         "DELETE ON images FOR EACH ROW BEGIN "
  63         " DELETE FROM files WHERE id = OLD.id; END;");
  64
  65   done:
  66     return ret;
  67 }
  68
  69 static lms_db_table_updater_t _db_table_updater_images[] = {
  70     _db_table_updater_images_0
  71 };
  72
  73
  74 static int
  75 _db_create_table_if_required(sqlite3 *db)
  76 {
  77     return lms_db_table_update_if_required(db, "images",
  78          LMS_ARRAY_SIZE(_db_table_updater_images),
  79          _db_table_updater_images);
  80 }
  81
  82 lms_db_image_t *
  83 lms_db_image_new(sqlite3 *db)
  84 {
  85     lms_db_image_t *ldi;
  86
  87     if (_singleton) {
  88         _singleton->_references++;
  89         return _singleton;
  90     }
  91
  92     if (!db)
  93         return NULL;
  94
  95     if (_db_create_table_if_required(db) != 0) {
  96         fprintf(stderr, "ERROR: could not create table.\n");
  97         return NULL;
  98     }
  99
 100     ldi = calloc(1, sizeof(lms_db_image_t));
 101     ldi->_references = 1;
 102     ldi->db = db;
 103
 104     return ldi;
 105 }
 106
 107 int
 108 lms_db_image_start(lms_db_image_t *ldi)
 109 {
 110     if (!ldi)
 111         return -1;
 112     if (ldi->_is_started)
 113         return 0;
 114
 115     ldi->insert = lms_db_compile_stmt(ldi->db,
 116         "INSERT OR REPLACE INTO images ("
 117         "id, title, artist, date, width, height, orientation, "
 118         "gps_lat, gps_long, gps_alt) VALUES ("
 119         "?, ?, ?, ?, ?, ?, ?, ?, ?, ?)");
 120     if (!ldi->insert)
 121         return -2;
 122
 123     ldi->_is_started = 1;
 124     return 0;
 125 }
 126
 127 int
 128 lms_db_image_free(lms_db_image_t *ldi)
 129 {
 130     if (!ldi)
 131         return -1;
 132     if (ldi->_references == 0) {
 133         fprintf(stderr, "ERROR: over-called lms_db_image_free(%p)\n", ldi);
 134         return -1;
 135     }
 136
 137     ldi->_references--;
 138     if (ldi->_references > 0)
 139         return 0;
 140
 141     if (ldi->insert)
 142         lms_db_finalize_stmt(ldi->insert, "insert");
 143
 144     free(ldi);
 145     _singleton = NULL;
 146
 147     return 0;
 148 }
 149
 150 static int
 151 _db_insert(lms_db_image_t *ldi, const struct lms_image_info *info)
 152 {
 153     sqlite3_stmt *stmt;
 154     int r, ret;
 155
 156     stmt = ldi->insert;
 157
 158     ret = lms_db_bind_int64(stmt, 1, info->id);
 159     if (ret != 0)
 160         goto done;
 161
 162     ret = lms_db_bind_text(stmt, 2, info->title.str, info->title.len);
 163     if (ret != 0)
 164         goto done;
 165
 166     ret = lms_db_bind_text(stmt, 3, info->artist.str, info->artist.len);
 167     if (ret != 0)
 168         goto done;
 169
 170     ret = lms_db_bind_int(stmt, 4, info->date);
 171     if (ret != 0)
 172         goto done;
 173
 174     ret = lms_db_bind_int(stmt, 5, info->width);
 175     if (ret != 0)
 176         goto done;
 177
 178     ret = lms_db_bind_int(stmt, 6, info->height);
 179     if (ret != 0)
 180         goto done;
 181
 182     ret = lms_db_bind_int(stmt, 7, info->orientation);
 183     if (ret != 0)
 184         goto done;
 185
 186     ret = lms_db_bind_double(stmt, 8, info->gps.latitude);
 187     if (ret != 0)
 188         goto done;
 189
 190     ret = lms_db_bind_double(stmt, 9, info->gps.longitude);
 191     if (ret != 0)
 192         goto done;
 193
 194     ret = lms_db_bind_double(stmt, 10, info->gps.altitude);
 195     if (ret != 0)
 196         goto done;
 197
 198     r = sqlite3_step(stmt);
 199     if (r != SQLITE_DONE) {
 200         fprintf(stderr, "ERROR: could not insert image info: %s\n",
 201                 sqlite3_errmsg(ldi->db));
 202         ret = -11;
 203         goto done;
 204     }
 205
 206     ret = 0;
 207
 208   done:
 209     lms_db_reset_stmt(stmt);
 210
 211     return ret;
 212 }
 213
 214 int
 215 lms_db_image_add(lms_db_image_t *ldi, struct lms_image_info *info)
 216 {
 217     if (!ldi)
 218         return -1;
 219     if (!info)
 220         return -2;
 221     if (info->id < 1)
 222         return -3;
 223
 224     return _db_insert(ldi, info);
 225 }