add zig lsp extractor

Author: dundalek

src/io/github/dundalek/stratify/lsp.clj

@@ -242,13 +242,24 @@
   (let [{:keys [start end]} (:selectionRange sym)]
     (str uri "#L" (:line start) "C" (:character start) "-L" (:line end) "C" (:character end))))
 
-(defn extract-graph [{:keys [server root-path source-paths source-pattern]}]
+(defn- send-did-open! [server uri file-path]
+  (server-message! server {:method "textDocument/didOpen"
+                           :params {:textDocument {:uri uri
+                                                   :languageId "zig"
+                                                   :version 1
+                                                   :text (slurp file-path)}}
+                           :jsonrpc "2.0"}))
+
+(defn extract-graph [{:keys [server root-path source-paths source-pattern open-documents?]}]
   (let [uri-base (str "file://" root-path "/")
-        file-uris (->> source-paths
-                       (mapcat (fn [path]
-                                 (fs/glob (fs/file root-path path) source-pattern)))
-                       (map path->uri))
+        file-paths (->> source-paths
+                        (mapcat (fn [path]
+                                  (fs/glob (fs/file root-path path) source-pattern))))
+        file-uris (map path->uri file-paths)
         file-uris-set (set file-uris)
+        _ (when open-documents?
+            (doseq [[uri path] (map vector file-uris file-paths)]
+              (send-did-open! server uri (str path))))
         symbols (->> file-uris
                      (mapcat (fn [uri]
                                (->> (server-request! server "textDocument/documentSymbol"
@@ -415,6 +426,20 @@
       (finally
         (server-stop! server)))))
 
+(defn extract-zig [opts]
+  (let [opts (normalize-opts opts)
+        {:keys [root-path]} opts
+        server (start-server {:args ["zls"]})]
+    (try
+      (server-initialize! server {:root-path root-path})
+      (extract-graph (merge {:source-paths ["src"]
+                             :source-pattern "**.zig"
+                             :open-documents? true
+                             :server server}
+                            opts))
+      (finally
+        (server-stop! server)))))
+
 (defn decode-semantic-tokens [payload token-types token-modifiers]
   (let [{:keys [data]} payload]
     (loop [[delta-line delta-start-char length token-type-id token-modifiers-bits & remaining] data

src/stratify/main.clj

@@ -21,7 +21,7 @@
 
 (def ^:private source-formats
   #{"clj" "dgml" "dot" "overarch" "pulumi" "scip"
-    "go-lsp" "lua-lsp" "rust-lsp"})
+    "go-lsp" "lua-lsp" "rust-lsp" "zig-lsp"})
 
 (def ^:private target-formats #{"codecharta" "dep-tree" "dgml"})
 
@@ -145,6 +145,10 @@
           (let [g (lsp/extract-rust {:root-path (first args)})]
             (open-studio g))
 
+          (and studio (= from "zig-lsp"))
+          (let [g (lsp/extract-zig {:root-path (first args)})]
+            (open-studio g))
+
           (and studio (= from "overarch"))
           (do
             (add-deps "overarch")
@@ -187,6 +191,10 @@
           (let [g (lsp/extract-rust {:root-path (first args)})]
             (sdgml/write-to-file output-file (lsp/graph->dgml g)))
 
+          (= from "zig-lsp")
+          (let [g (lsp/extract-zig {:root-path (first args)})]
+            (sdgml/write-to-file output-file (lsp/graph->dgml g)))
+
           (= from "overarch")
           (do
             (add-deps "overarch")
@@ -284,6 +292,8 @@
 
   (main* "-f" "rust-lsp" "test/resources/code/rust/greeting")
 
+  (main* "-f" "zig-lsp" "test/resources/code/zig/greeting")
+
   (main* "--studio" "-f" "overarch" "test/resources/overarch/model.edn")
 
   (main* "-f" "dot" "test/resources/graphviz/simple.dot")

test/resources/code/zig/greeting/zls.json

@@ -0,0 +1,4 @@
+{
+  "enable_build_on_save": true,
+  "build_on_save_step": "check"
+}

test/src/io/github/dundalek/stratify/lsp_test.clj

@@ -85,6 +85,18 @@
                    "src/main.rs#L2C3-L2C7" {:label "main", :parent "src/main.rs"}}})
          (extract-relative-graph lsp/extract-rust "test/resources/code/rust/greeting"))))
 
+(deftest extract-zig
+  (is (= (make-digraph
+          {:adj {"src/main.zig#L3C7-L3C11" #{"src/main.zig#L0C6-L0C9" "src/main.zig#L1C6-L1C14"}}
+           :attrs {"src" {:category "Namespace", :label "src", :parent nil},
+                   "src/greeting.zig" {:category "Namespace", :label "greeting.zig", :parent "src"},
+                   "src/greeting.zig#L0C7-L0C12" {:label "greet", :parent "src/greeting.zig"},
+                   "src/main.zig" {:category "Namespace", :label "main.zig", :parent "src"},
+                   "src/main.zig#L0C6-L0C9" {:label "std", :parent "src/main.zig"},
+                   "src/main.zig#L1C6-L1C14" {:label "greeting", :parent "src/main.zig"},
+                   "src/main.zig#L3C7-L3C11" {:label "main", :parent "src/main.zig"}}})
+         (extract-relative-graph lsp/extract-zig "test/resources/code/zig/greeting"))))
+
 (deftest location-less-or-equal?
   (is (true? (lsp/location-less-or-equal? {:line 0 :character 0} {:line 0 :character 15})))
   (is (true? (lsp/location-less-or-equal? {:line 0 :character 15} {:line 1 :character 1})))