diff_graphs.cljs 2.9 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364
  1. (ns diff-graphs
  2. "A script that diffs two DB graphs through their sqlite.build EDN"
  3. (:require ["os" :as os]
  4. ["path" :as node-path]
  5. [babashka.cli :as cli]
  6. [clojure.data :as data]
  7. [clojure.pprint :as pprint]
  8. [clojure.string :as string]
  9. [logseq.common.config :as common-config]
  10. #_:clj-kondo/ignore
  11. [logseq.db.sqlite.cli :as sqlite-cli]
  12. [logseq.db.sqlite.export :as sqlite-export]
  13. [nbb.core :as nbb]))
  14. (defn- get-dir-and-db-name
  15. "Gets dir and db name for use with open-db!"
  16. [graph-dir]
  17. (if (string/includes? graph-dir "/")
  18. (let [graph-dir'
  19. (node-path/join (or js/process.env.ORIGINAL_PWD ".") graph-dir)]
  20. ((juxt node-path/dirname node-path/basename) graph-dir'))
  21. [(node-path/join (os/homedir) "logseq" "graphs") graph-dir]))
  22. (def spec
  23. "Options spec"
  24. {:help {:alias :h
  25. :desc "Print help"}
  26. :exclude-namespaces {:alias :e
  27. :coerce #{}
  28. :desc "Namespaces to exclude from properties and classes"}
  29. :exclude-built-in-pages? {:alias :b
  30. :desc "Exclude built-in pages"}
  31. :include-timestamps? {:alias :t
  32. :desc "Include timestamps in export"}})
  33. (defn -main [args]
  34. (let [{options :opts args' :args} (cli/parse-args args {:spec spec})
  35. [graph-dir graph-dir2] args'
  36. _ (when (or (nil? graph-dir) (nil? graph-dir2) (:help options))
  37. (println (str "Usage: $0 GRAPH-NAME GRAPH-NAME2 [& ARGS] [OPTIONS]\nOptions:\n"
  38. (cli/format-opts {:spec spec})))
  39. (js/process.exit 1))
  40. conn (apply sqlite-cli/open-db! (get-dir-and-db-name graph-dir))
  41. conn2 (apply sqlite-cli/open-db! (get-dir-and-db-name graph-dir2))
  42. export-options (select-keys options [:include-timestamps? :exclude-namespaces :exclude-built-in-pages?])
  43. export-map (sqlite-export/build-export @conn {:export-type :graph :graph-options export-options})
  44. export-map2 (sqlite-export/build-export @conn2 {:export-type :graph :graph-options export-options})
  45. prepare-export-to-diff
  46. (fn [m]
  47. (-> m
  48. (update :classes update-vals (fn [m]
  49. (update m :build/class-properties sort)))
  50. ;; TODO: fix built-in views for schema export
  51. (update :pages-and-blocks (fn [pbs]
  52. (vec (remove #(= (:block/title (:page %)) common-config/views-page-name) pbs))))))
  53. diff (->> (data/diff (prepare-export-to-diff export-map) (prepare-export-to-diff export-map2))
  54. butlast)]
  55. (if (= diff [nil nil])
  56. (println "The two graphs are equal!")
  57. (do (pprint/pprint diff)
  58. (js/process.exit 1)))))
  59. (when (= nbb/*file* (nbb/invoked-file))
  60. (-main *command-line-args*))