From 41e9051f2d809c42c3dfecc2eb11ad544cbd27b7 Mon Sep 17 00:00:00 2001
From: Chris Xiong <chirs241097@gmail.com>
Date: Mon, 19 Sep 2022 02:39:03 -0400
Subject: You break it, you fix it!

The GUI is now working again, with scanning built-in.
---
 xsig/src/signature_db.cpp | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

(limited to 'xsig/src')

diff --git a/xsig/src/signature_db.cpp b/xsig/src/signature_db.cpp
index 393b756..ba1a372 100644
--- a/xsig/src/signature_db.cpp
+++ b/xsig/src/signature_db.cpp
@@ -214,6 +214,22 @@ void signature_db::batch_get_signature_end()
     p->batch_end(batch_status::getsig);
 }
 
+std::vector<size_t> signature_db::get_image_ids()
+{
+    sqlite3_stmt *st = nullptr;
+    sqlite3_prepare_v2(p->db, "select id from images;", -1, &st, 0);
+    std::vector<size_t> ret;
+    while (1)
+    {
+        int r = sqlite3_step(st);
+        if (r != SQLITE_ROW) break;
+        size_t id = (size_t)sqlite3_column_int(st, 0);
+        ret.push_back(id);
+    }
+    sqlite3_finalize(st);
+    return ret;
+}
+
 void signature_db::batch_put_subslice_begin()
 {
     if (!p->db) [[ unlikely ]] return;
-- 
cgit v1.2.3