From df7264f016db4fdf16b6f0031cb04d67f8763fe9 Mon Sep 17 00:00:00 2001
From: Tienson Qin <tiensonqin@gmail.com>
Date: Mon, 11 Dec 2023 16:52:21 +0800
Subject: [PATCH] Add helper to remove all OPFS files

---
 src/main/frontend/db_worker.cljs  | 50 +++++++++++++++++--------------
 src/main/frontend/persist_db.cljs |  1 +
 2 files changed, 29 insertions(+), 22 deletions(-)

diff --git a/src/main/frontend/db_worker.cljs b/src/main/frontend/db_worker.cljs
index fe9309a589..49e79141a0 100644
--- a/src/main/frontend/db_worker.cljs
+++ b/src/main/frontend/db_worker.cljs
@@ -27,9 +27,7 @@
   (or @*opfs-pool
       (p/let [^js pool (.installOpfsSAHPoolVfs @*sqlite #js {:name (str "logseq-pool-" graph)
                                                              :initialCapacity 10})]
-        ;; (.removeVfs ^js pool)
         (reset! *opfs-pool pool)
-        (js/console.dir pool)
         pool)))
 
 (defn- init-sqlite-module!
@@ -50,13 +48,6 @@
   [repo]
   (str "/" repo ".sqlite"))
 
-(comment
-  (defn- get-file-names
-   [graph]
-   (p/let [^js pool (<get-opfs-pool graph)]
-     (when pool
-       (.getFileNames pool)))))
-
 (comment
   (defn- export-db-file
     [repo]
@@ -106,11 +97,13 @@
   [repo]
   (doseq [[r db] @*sqlite-conns]
     (when-not (= repo r)
+      (swap! *sqlite-conns dissoc r)
       (.close ^Object db))))
 
 (defn- close-db!
   [repo]
   (when-let [db (@*sqlite-conns repo)]
+    (swap! *sqlite-conns dissoc repo)
     (.close ^Object db)))
 
 (defn- create-or-open-db!
@@ -154,6 +147,29 @@
                         (rest dirs))]
             (p/recur result dirs)))))))
 
+(comment
+  (defn <remove-all-files!
+    "!! Dangerous: use it only for development."
+    []
+    (p/let [all-files (<list-all-files)
+            files (filter #(= (.-kind %) "file") all-files)
+            dirs (filter #(= (.-kind %) "directory") all-files)
+            _ (p/all (map (fn [file] (.remove file)) files))]
+      (p/all (map (fn [dir] (.remove dir)) dirs)))))
+
+(defn- <get-pool-files
+  [^js pool]
+  (.getFileNames pool))
+
+(defn- remove-vfs!
+  [repo]
+  (p/let [^js pool (<get-opfs-pool repo)]
+    (when pool
+      (p/let [files (<get-pool-files pool)
+              _ (p/all (map (fn [file] (.unlink pool file)) files))
+              _ (.wipeFiles pool)]
+        (.removeVfs ^js pool)))))
+
 #_:clj-kondo/ignore
 (defclass SQLiteDB
   (extends js/Object)
@@ -226,22 +242,12 @@
 
   (unsafeUnlinkDB
    [_this repo]
-   (p/let [_ (close-db! repo)
-           pool (<get-opfs-pool repo)
-           path (get-repo-path repo)]
-     (when pool
-       (.unlink pool path)))))
+   (p/let [_ (close-db! repo)]
+     (remove-vfs! repo)
+     nil)))
 
 (defn init
   "web worker entry"
   []
   (let [^js obj (SQLiteDB.)]
     (Comlink/expose obj)))
-
-(comment
-  (defn- remove-pfs!
-   "!! use it only for development"
-   []
-   (p/let [^js pool (<get-opfs-pool)]
-     (when pool
-       (.removeVfs ^js pool)))))
diff --git a/src/main/frontend/persist_db.cljs b/src/main/frontend/persist_db.cljs
index 7b159bb55d..0f3559a84f 100644
--- a/src/main/frontend/persist_db.cljs
+++ b/src/main/frontend/persist_db.cljs
@@ -18,6 +18,7 @@
    (protocol/<unsafe-delete (get-impl) repo))
 
 ;; FIXME: limit repo name's length
+;; @shuyu Do we still need this?
 (defn <new [repo]
   {:pre [(<= (count repo) 56)]}
   (protocol/<new (get-impl) repo))