query.cljs 1.5 KB

12345678910111213141516171819202122232425262728293031323334353637
  1. (ns query
  2. "An example script that queries any db graph from the commandline e.g.
  3. $ yarn -s nbb-logseq script/query.cljs db-name '[:find (pull ?b [:block/name :block/content]) :where [?b :block/created-at]]'"
  4. (:require [datascript.core :as d]
  5. [clojure.edn :as edn]
  6. [cljs-bean.core :as bean]
  7. [logseq.db.sqlite.db :as sqlite-db]
  8. [logseq.db.sqlite.restore :as sqlite-restore]
  9. [logseq.db.rules :as rules]
  10. [nbb.core :as nbb]
  11. ["path" :as path]
  12. ["os" :as os]))
  13. (defn read-graph
  14. "The db graph bare version of gp-cli/parse-graph"
  15. [graph-name]
  16. (let [graphs-dir (path/join (os/homedir) "logseq/graphs")
  17. _ (sqlite-db/open-db! graphs-dir graph-name)
  18. {:keys [uuid->db-id-map conn]}
  19. (sqlite-restore/restore-initial-data (bean/->js (sqlite-db/get-initial-data graph-name)))
  20. db (sqlite-restore/restore-other-data conn (sqlite-db/get-other-data graph-name []) uuid->db-id-map)]
  21. (d/conn-from-db db)))
  22. (defn -main [args]
  23. (when (not= 2 (count args))
  24. (println "Usage: $0 GRAPH QUERY")
  25. (js/process.exit 1))
  26. (let [[graph-name query*] args
  27. conn (read-graph graph-name)
  28. query (into (edn/read-string query*) [:in '$ '%]) ;; assumes no :in are in queries
  29. results (mapv first (apply d/q query @conn [(vals rules/db-query-dsl-rules)]))]
  30. (println "DB contains" (count (d/datoms @conn :eavt)) "datoms")
  31. (prn results)))
  32. (when (= nbb/*file* (:file (meta #'-main)))
  33. (-main *command-line-args*))