From 4c7bed83eee61d7093675e472a3578ba77d68d04 Mon Sep 17 00:00:00 2001
From: fmodf <fmodf.ios@gmail.com>
Date: Tue, 27 Aug 2024 17:28:29 +0200
Subject: [PATCH] wip

---
 .../AppData/Client/Client+MartinOMEMO.swift   | 85 +++++++++++++++++--
 1 file changed, 76 insertions(+), 9 deletions(-)

diff --git a/ConversationsClassic/AppData/Client/Client+MartinOMEMO.swift b/ConversationsClassic/AppData/Client/Client+MartinOMEMO.swift
index 2cc0236..0a4aa3b 100644
--- a/ConversationsClassic/AppData/Client/Client+MartinOMEMO.swift
+++ b/ConversationsClassic/AppData/Client/Client+MartinOMEMO.swift
@@ -269,27 +269,94 @@ extension ClientMartinOMEMO: SignalPreKeyStoreProtocol {
 // MARK: - SignedPreKey
 extension ClientMartinOMEMO: SignalSignedPreKeyStoreProtocol {
     func countSignedPreKeys() -> Int {
-        0
+        do {
+            let data = try Database.shared.dbQueue.read { db in
+                try Row.fetchOne(
+                    db,
+                    sql: "SELECT count(1) FROM omemo_signed_pre_keys WHERE account = :account",
+                    arguments: ["account": credentials.bareJid]
+                )
+            }
+            return data?["count(1)"] ?? 0
+        } catch {
+            logIt(.error, "Error fetching chats: \(error.localizedDescription)")
+            return 0
+        }
     }
 
     func loadSignedPreKey(withId: UInt32) -> Data? {
-        print(withId)
-        return nil
+        do {
+            let data = try Database.shared.dbQueue.read { db in
+                try Row.fetchOne(
+                    db,
+                    sql: "SELECT key FROM omemo_signed_pre_keys WHERE account = :account AND id = :id",
+                    arguments: ["account": credentials.bareJid, "id": withId]
+                )
+            }
+            return data?["key"]
+        } catch {
+            logIt(.error, "Error fetching chats: \(error.localizedDescription)")
+            return nil
+        }
     }
 
     func storeSignedPreKey(_ data: Data, withId: UInt32) -> Bool {
-        print(data, withId)
-        return false
+        do {
+            try Database.shared.dbQueue.write { db in
+                try db.execute(
+                    sql: "INSERT INTO omemo_signed_pre_keys (account, id, key) VALUES (:account, :id, :key)",
+                    arguments: ["account": credentials.bareJid, "id": withId, "key": data]
+                )
+            }
+            return true
+        } catch {
+            logIt(.error, "Error fetching chats: \(error.localizedDescription)")
+            return false
+        }
     }
 
     func containsSignedPreKey(withId: UInt32) -> Bool {
-        print(withId)
-        return false
+        do {
+            let rec = try Database.shared.dbQueue.read { db in
+                try Row.fetchOne(
+                    db,
+                    sql: "SELECT key FROM omemo_signed_pre_keys WHERE account = :account AND id = :id",
+                    arguments: ["account": credentials.bareJid, "id": withId]
+                )
+            }
+            return rec != nil
+        } catch {
+            logIt(.error, "Error fetching chats: \(error.localizedDescription)")
+            return false
+        }
     }
 
     func deleteSignedPreKey(withId: UInt32) -> Bool {
-        print(withId)
-        return false
+        do {
+            try Database.shared.dbQueue.write { db in
+                try db.execute(
+                    sql: "DELETE FROM omemo_signed_pre_keys WHERE account = :account AND id = :id",
+                    arguments: ["account": credentials.bareJid, "id": withId]
+                )
+            }
+            return true
+        } catch {
+            logIt(.error, "Error fetching chats: \(error.localizedDescription)")
+            return false
+        }
+    }
+
+    func wipeSignedPreKeys() {
+        do {
+            try Database.shared.dbQueue.write { db in
+                try db.execute(
+                    sql: "DELETE FROM omemo_signed_pre_keys WHERE account = :account",
+                    arguments: ["account": credentials.bareJid]
+                )
+            }
+        } catch {
+            logIt(.error, "Error fetching chats: \(error.localizedDescription)")
+        }
     }
 }