Adding functions to manipulate stickers in DB

inc/lektor/database.h

@@ -124,3 +124,14 @@ bool database_plt_add_uri(sqlite3 *db, const char *name, struct lkt_uri_t *uri);
 /* User control, yeah, MPD authentification sucks. */
 bool database_user_authentificate(sqlite3 *db, const char *password);
 bool database_user_add(sqlite3 *db, const char *username, const char *password);
+
+/* Stickers manipulations. */
+typedef bool (*database_sticker_callback_t)(void *args, const char *sticker, const char *type,
+        int uri, int value);
+
+bool database_sticker_create(sqlite3 *db, const char *name);
+bool database_sticker_delete(sqlite3 *db, const char *name);
+bool database_sticker_list(sqlite3 *db, const char *type, void *args, database_sticker_callback_t call);
+bool database_sticker_set(sqlite3 *db, const char *type, const char *name, int uri, int value);
+bool database_sticker_get(sqlite3 *db, const char *type, const char *name, int uri, void *args,
+                          database_sticker_callback_t call);

meson.build

@@ -26,6 +26,7 @@ dep_mpv = dependency('mpv', required : false)
 core_sources =  [ 'src/mkv/bufferfd.c'
                 , 'src/mkv/write.c'
                 , 'src/commands.c'
+                , 'src/database/stickers.c'
                 , 'src/database/open.c'
                 , 'src/database/queue.c'
                 , 'src/database/update.c'

src/commands.c

@@ -251,7 +251,8 @@ command_stop(sqlite3 *db, struct lkt_win *win, enum mpd_idle_flag *watch_mask_pt
 }
 
 bool
-command_add(sqlite3 *db, struct lkt_win *win, char *args[LKT_MESSAGE_ARGS_MAX], enum mpd_idle_flag *watch_mask_ptr)
+command_add(sqlite3 *db, struct lkt_win *win, char *args[LKT_MESSAGE_ARGS_MAX],
+            enum mpd_idle_flag *watch_mask_ptr)
 {
     if (args == NULL) {
         fprintf(stderr, " ! command_add: invalid NULL arguments\n");

src/database/stickers.c

+#define _POSIX_C_SOURCE 200809L
+
+#include <lektor/macro.h>
+#include <lektor/database.h>
+#include <string.h>
+#include <strings.h>
+#include <stdlib.h>
+#include <stdio.h>
+
+bool
+database_sticker_create(sqlite3 *db, const char *name)
+{
+    static const char *INSERT =
+        "WITH id_max (id) AS (SELECT MAX(id) FROM 'stickers')"
+        "INSERT INTO 'stickers' (id, name)"
+        " SELECT"
+        "  CASE WHEN id_max.id IS NULL THEN 1 ELSE id_max.id + 1 END,"
+        "  ?"
+        " FROM id_max"
+        "WHERE NOT EXISTS (SELECT 1 FROM 'stickers' WHERE name = ?);";
+    sqlite3_stmt *stmt;
+    int ret = false;
+
+    if (strlen(name) == 0) {
+        fprintf(stderr, " ! database_sticker_create: A sticker name must be at least one character long\n");
+        return ret;
+    }
+
+    SQLITE_PREPARE(db, stmt, INSERT, error);
+    SQLITE_BIND_TEXT(db, stmt, 1, name, error);
+    SQLITE_BIND_TEXT(db, stmt, 2, name, error);
+
+    if (sqlite3_step(stmt) != SQLITE_OK) {
+        fprintf(stderr, " ! database_sticker_create: Failed to create sticker '%s': %s\n",
+                name, sqlite3_errmsg(db));
+        goto error;
+    }
+
+    ret = true;
+error:
+    sqlite3_finalize(stmt);
+    return ret;
+}
+
+bool
+database_sticker_delete(sqlite3 *db, const char *name)
+{
+    static const char *INSERT =
+        "DELETE FROM 'stickers' WHERE name = ?;";
+    sqlite3_stmt *stmt;
+    int ret = false;
+
+    if (strlen(name) == 0) {
+        fprintf(stderr, " ! database_sticker_delete: A sticker name must be at least one character long\n");
+        return ret;
+    }
+
+    SQLITE_PREPARE(db, stmt, INSERT, error);
+    SQLITE_BIND_TEXT(db, stmt, 1, name, error);
+
+    if (sqlite3_step(stmt) != SQLITE_OK) {
+        fprintf(stderr, " ! database_sticker_delete: Failed to delete sticker '%s': %s\n",
+                name, sqlite3_errmsg(db));
+        goto error;
+    }
+
+    ret = true;
+error:
+    sqlite3_finalize(stmt);
+    return ret;
+}
+
+bool
+database_sticker_list(sqlite3 *db, const char *type, void *args,
+                      database_sticker_callback_t call)
+{
+    const char *SQL = NULL;
+    int ret = false, uri, value;
+    sqlite3_stmt *stmt;
+    const char *sticker;
+
+    if (type == NULL)
+        SQL =
+            "SELECT name, sts.id, value FROM 'stickers' LEFT OUTER JOIN "
+            "( SELECT id, sticker, value FROM 'stickers.song'"
+            "  UNION"
+            "  SELECT id, sticker, value FROM 'stickers.plt'"
+            ") AS sts"
+            "ON sts.sticker = 'stickers'.id;";
+    else if (!strcasecmp(type, "plt"))
+        SQL =
+            "SELECT name, 'stickers.plt'.id, value "
+            "FROM 'stickers' "
+            "LEFT OUTER JOIN 'stickers' "
+            "ON 'stickers'.id = 'stickers.plt'.sticker;";
+    else if (!strcasecmp(type, "song"))
+        SQL =
+            "SELECT name, 'stickers.song'.id, value "
+            "FROM 'stickers' "
+            "LEFT OUTER JOIN 'stickers' "
+            "ON 'stickers'.id = 'stickers.song'.sticker;";
+    else {
+        fprintf(stderr, " . database_sticker_list: type '%s' is invalid\n", type);
+        return false;
+    }
+
+    SQLITE_PREPARE(db, stmt, SQL, error);
+
+    for (;;) {
+        switch (sqlite3_step(stmt)) {
+        case SQLITE_ROW:
+            sticker = (const char *) sqlite3_column_text(stmt, 0);
+            uri     = sqlite3_column_int(stmt, 1);
+            value   = sqlite3_column_int(stmt, 2);
+            if (!call(args, sticker, type, uri, value))
+                goto error;
+            continue;
+
+        case SQLITE_DONE:
+            goto end_loop;
+
+        default:
+            goto error;
+        }
+    }
+
+end_loop:
+    ret = true;
+error:
+    sqlite3_finalize(stmt);
+    return ret;
+}
+
+bool
+database_sticker_set(sqlite3 *db, const char *type, const char *name, int uri, int value)
+{
+    const char *SQL = NULL;
+    sqlite3_stmt *stmt;
+    int ret = false;
+
+    /* Bindings:
+     * 1 -> the uri of the plalist or kara
+     * 2 -> the value of the sticker
+     * 3 -> the name of the sticker
+     */
+
+    if (!strcasecmp(type, "song"))
+        SQL =
+            "INSERT OR REPLACE INTO 'stickers.song' (id, sticker, value) "
+            "SELECT ?, 'stickers'.id, ? "
+            "FROM 'stickers'"
+            "WHERE 'stickers'.id = ?;\n";
+    else if (!strcasecmp(type, "song"))
+        SQL =
+            "INSERT OR REPLACE INTO 'stickers.plt' (id, sticker, value) "
+            "SELECT ?, 'stickers'.id, ? "
+            "FROM 'stickers'"
+            "WHERE 'stickers'.id = ?;\n";
+    else {
+        fprintf(stderr, " . database_sticker_set: Type '%s' is invalid\n", type);
+        return false;
+    }
+
+    SQLITE_PREPARE(db, stmt, SQL, error);
+    SQLITE_BIND_INT(db, stmt, 1, uri, error);
+    SQLITE_BIND_INT(db, stmt, 2, value, error);
+    SQLITE_BIND_TEXT(db, stmt, 3, name, error);
+
+    if (sqlite3_step(stmt) != SQLITE_OK) {
+        fprintf(stderr, " ! database_sticker_set: Failed to update or set sticker '%s' for "
+                "'%s' %d: %s\n", name, type, uri, sqlite3_errmsg(db));
+        goto error;
+    }
+
+    ret = true;
+error:
+    sqlite3_finalize(stmt);
+    return ret;
+}
+
+bool
+database_sticker_get(sqlite3 *db, const char *type, const char *name, int uri, void *args,
+                     database_sticker_callback_t call)
+{
+    const char *SQL = NULL;
+    sqlite3_stmt *stmt;
+    int ret = false, value;
+
+    if (!strcasecmp(type, "song"))
+        SQL =
+            "SELECT value "
+            "FROM 'stickers' "
+            "JOIN 'stickers.song'"
+            " ON 'stickers'.id = 'stickers.song'.sticker"
+            " AND 'stickers'.name = ?;\n";
+    else if (!strcasecmp(type, "song"))
+        SQL =
+            "SELECT value "
+            "FROM 'stickers' "
+            "JOIN 'stickers.song'"
+            " ON 'stickers'.id = 'stickers.song'.sticker"
+            " AND 'stickers'.name = ?;\n";
+    else {
+        fprintf(stderr, " . database_sticker_get: Type '%s' is invalid\n", type);
+        return false;
+    }
+
+    SQLITE_PREPARE(db, stmt, SQL, error);
+    SQLITE_BIND_TEXT(db, stmt, 1, name, error);
+
+    for (;;) {
+        switch (sqlite3_step(stmt)) {
+        case SQLITE_ROW:
+            value = sqlite3_column_int(stmt, 1);
+            if (!call(args, name, type, uri, value))
+                goto error;
+            continue;
+
+        case SQLITE_DONE:
+            goto end_loop;
+
+        default:
+            goto error;
+        }
+    }
+
+end_loop:
+    ret = true;
+error:
+    sqlite3_finalize(stmt);
+    return ret;
+}