From bcdffd796a0cfeaea12475c2af264dcdf1623b62 Mon Sep 17 00:00:00 2001
From: steinuil <steenuil.owl@gmail.com>
Date: Wed, 14 Aug 2024 16:57:14 +0200
Subject: [PATCH] Add nu-complete cache helper

---
 custom-completions/cache/nu-complete-cache.nu | 60 +++++++++++++++++++
 1 file changed, 60 insertions(+)
 create mode 100644 custom-completions/cache/nu-complete-cache.nu

diff --git a/custom-completions/cache/nu-complete-cache.nu b/custom-completions/cache/nu-complete-cache.nu
new file mode 100644
index 00000000..f4ec3da5
--- /dev/null
+++ b/custom-completions/cache/nu-complete-cache.nu
@@ -0,0 +1,60 @@
+def db [] {
+  [$env.XDG_CACHE_HOME "nu-complete.sqlite"] | path join
+}
+
+def init [] {
+  if (db | path exists) {
+    stor import --file-name (db)
+  } else {
+    (stor create
+      --table-name "nu_complete" 
+      --columns {
+        name: str
+        value: str
+        timestamp: datetime
+      }
+    )
+  }
+}
+
+export def "nu-complete cache" [
+  --expire: duration
+  cmd: any
+] {
+  init
+
+  let name = view source $cmd
+
+  let saved = stor open
+    | (query db
+        "SELECT value, timestamp FROM nu_complete WHERE name = :name"
+        -p { name: $name })
+    | get -i 0
+
+  if $saved == null or (($saved.timestamp | into datetime) + $expire < (date now)) {
+    let output = do $cmd
+
+    if $saved == null {
+      { name: $name, value: ($output | to nuon -r), timestamp: (date now) }
+      | stor insert -t "nu_complete"
+    } else {
+      stor open
+      | query db "UPDATE nu_complete SET value = :value, timestamp = :timestamp WHERE name = :name" -p {
+        value: ($output | to nuon -r),
+        timestamp: (date now),
+        name: $name
+      }
+    }
+
+    rm -f (db)
+    stor export --file-name (db)
+
+    $output
+  } else {
+    rm -f (db)
+    stor export --file-name (db)
+
+    $saved.value | from nuon
+  }
+}
+