clone/logseq
1
0
Fork 0
mirror of https://github.com/logseq/logseq.git synced 2026-04-25 22:54:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

enhance: add basic script to test db graph import

Browse Source 
also tweaked query script to work with db graphs by relative path
This commit is contained in:
Gabriel Horner
2024-02-24 00:42:31 -05:00
parent 356657d36b
commit a61ee21e25
4 changed files with 94 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
deps/db/script/query.cljs vendored
View File

@@ -1,27 +1,32 @@
(ns query
"An example script that queries any db graph from the commandline e.g.
(ns query
"An example script that queries any db graph from the commandline e.g.
$ yarn -s nbb-logseq script/query.cljs db-name '[:find (pull ?b [:block/name :block/content]) :where [?b :block/created-at]]'"
(:require [datascript.core :as d]
[clojure.edn :as edn]
[logseq.db.sqlite.db :as sqlite-db]
[logseq.db.frontend.rules :as rules]
[nbb.core :as nbb]
["path" :as path]
["os" :as os]))
$ yarn -s nbb-logseq script/query.cljs db-name '[:find (pull ?b [:block/name :block/content]) :where [?b :block/created-at]]'"
(:require [datascript.core :as d]
[clojure.edn :as edn]
[logseq.db.sqlite.db :as sqlite-db]
[logseq.db.frontend.rules :as rules]
[nbb.core :as nbb]
[clojure.string :as string]
["path" :as node-path]
["os" :as os]))
(defn read-graph
"The db graph bare version of gp-cli/parse-graph"
[graph-name]
(let [graphs-dir (path/join (os/homedir) "logseq/graphs")]
(sqlite-db/open-db! graphs-dir graph-name)))
(defn- get-dir-and-db-name
"Gets dir and db name for use with open-db!"
[graph-dir]
(if (string/includes? graph-dir "/")
(let [graph-dir'
(node-path/join (or js/process.env.ORIGINAL_PWD ".") graph-dir)]
((juxt node-path/dirname node-path/basename) graph-dir'))
[(node-path/join (os/homedir) "logseq" "graphs") graph-dir]))
(defn -main [args]
(when (< (count args) 2)
(println "Usage: $0 GRAPH QUERY")
(js/process.exit 1))
(let [[graph-name query*] args
conn (read-graph graph-name)
(let [[graph-dir query*] args
[dir db-name] (get-dir-and-db-name graph-dir)
conn (sqlite-db/open-db! dir db-name)
query (into (edn/read-string query*) [:in '$ '%]) ;; assumes no :in are in queries
results (mapv first (d/q query @conn (rules/extract-rules rules/db-query-dsl-rules)))]
(when ((set args) "-v") (println "DB contains" (count (d/datoms @conn :eavt)) "datoms"))