1 files changed, 153 insertions, 0 deletions
diff --git a/src/data.c b/src/data.c
new file mode 100644
index 0000000..0013ed2
--- /dev/null
+++ b/src/data.c
@@ -0,0 +1,153 @@
+#include "data.h"
+#include <sqlite3.h>
+#include <gtk/gtk.h>
+#include <unistd.h>
+#include <string.h>
+#include <stdlib.h>
+#include <sys/stat.h>
+#include <errno.h>
+#include "collections.h"
+
+static gboolean data_check_and_create_database(gchar *data_file);
+static gint data_create_new_database(const char *filename);
+
+GList *data_get_series(void)
+{
+  sqlite3 *database;
+  sqlite3_stmt *statement;
+  gchar *data_file;
+  GList *list = NULL;
+
+  data_file = collections_get_data_file();
+
+  if (data_check_and_create_database(data_file)) {
+    if (sqlite3_open(data_file, &database) == SQLITE_OK) {
+      int res;
+      const char *sqlStatement =
+        " SELECT   name, "
+        "          current_qty, "
+        "          total_qty "
+        " FROM     collection "
+        " ORDER BY name "
+        " COLLATE NOCASE ";
+
+      res = sqlite3_prepare_v2(database,
+                               sqlStatement,
+                               strlen(sqlStatement),
+                               &statement, NULL);
+      if (res == SQLITE_OK) {
+        while (sqlite3_step(statement) == SQLITE_ROW) {
+          struct collection *col =
+            (struct collection *)malloc(sizeof(struct collection *));
+
+          col->name = g_strdup(sqlite3_column_text(statement, 0));
+          col->current_qty = sqlite3_column_int(statement, 1);
+          col->total_qty = sqlite3_column_int(statement, 2);
+
+          list = g_list_append(list, (gpointer)col);
+        }
+      }
+      else
+        g_print("error %d: %s\n", res, sqlite3_errmsg(database));
+      /* Release the compiled statement from memory */
+      sqlite3_finalize(statement);
+    }
+    sqlite3_close(database);
+  }
+
+  return list;
+}
+
+gboolean data_add_series(gchar *name, int total_qty)
+{
+  sqlite3 *database;
+  sqlite3_stmt *statement;
+  gchar *data_file;
+
+  data_file = collections_get_data_file();
+
+  if (data_check_and_create_database(data_file)) {
+    if (sqlite3_open(data_file, &database) == SQLITE_OK) {
+      int res;
+      const char *sqlStatement =
+        g_strdup_printf("INSERT INTO collection (name, current_qty, total_qty) "
+                        " VALUES ('%s', 0, %d)", name, total_qty);
+
+      res = sqlite3_prepare_v2(database,
+                               sqlStatement,
+                               strlen(sqlStatement),
+                               &statement, NULL);
+      if (res == SQLITE_OK) {
+        if (sqlite3_step(statement) == SQLITE_DONE)
+          return TRUE;
+      }
+    }
+  }
+
+  return FALSE;
+}
+
+static gboolean data_check_and_create_database(gchar *data_file)
+{
+  if (!access(data_file, R_OK) == 0)
+    if (data_create_new_database(data_file)) {
+      g_printerr("Couldn't create a new database\n");
+      return FALSE; /* Couldn't create database, can't continue */
+    }
+
+  return TRUE;
+}
+
+static gint data_create_new_database(const char *filename)
+{
+  sqlite3 *db;
+  char *zErrMsg = 0;
+  int rc;
+
+  if (access(collections_get_config_dir(), R_OK) == -1) {
+    if (mkdir(collections_get_config_dir(), S_IRWXU) == -1)
+      g_printerr("Can't create directory: %s\n", strerror(errno));
+  }
+
+  /* Open database connection, create file */
+  rc = sqlite3_open(filename, &db);
+  if (rc) {
+    g_printerr("Can't open database: %s\n", sqlite3_errmsg(db));
+    if (db)
+      sqlite3_close(db);
+    return -1;
+  }
+
+  /* Create collections table */
+  rc = sqlite3_exec(db, "CREATE TABLE collection("
+                    "id INTEGER PRIMARY KEY,"
+                    "name VARCHAR(100),"
+                    "current_qty INTEGER,"
+                    "total_qty INTEGER"
+                    ")", NULL, NULL, &zErrMsg);
+  if (rc != SQLITE_OK) {
+    g_printerr("Can't create collection table: %s\n", zErrMsg);
+    sqlite3_free(zErrMsg);
+    sqlite3_close(db);
+    return -1;
+  }
+
+  /* Create items table */
+  rc = sqlite3_exec(db, "CREATE TABLE item("
+                    "collection_id INTEGER,"
+                    "id INTEGER,"
+                    "title VARCHAR(100),"
+                    "PRIMARY KEY(collection_id, id),"
+                    "FOREIGN KEY(collection_id)"
+                    "  REFERENCES collection(id)"
+                    ")", NULL, NULL, &zErrMsg);
+  if (rc != SQLITE_OK) {
+    g_printerr("Can't create item table: %s\n", zErrMsg);
+    sqlite3_free(zErrMsg);
+    sqlite3_close(db);
+    return -1;
+  }
+
+  sqlite3_close(db);
+  return 0;
+}