No elmi

.gitmodules

@@ -0,0 +1,3 @@
+[submodule "tree-sitter-elm"]
+	path = tree-sitter-elm
+	url = https://github.com/Razzeee/tree-sitter-elm

Cargo.toml

@@ -15,5 +15,13 @@ miniserde = "0.1"
 pathdiff = "0.1"
 rand = { version = "0.7.3", default-features = false, features = ["std"] }
 num_cpus = "1"
+thiserror = "1.0.21"
+tree-sitter = "0.17.0"
+regex = "1.4.1"
+lazy_static = "1.4.0"
 
 [dev-dependencies]
+
+[build-dependencies]
+cc="1"
+

build.rs

@@ -0,0 +1,17 @@
+use std::path::PathBuf;
+
+fn main() {
+    let dir: PathBuf = ["tree-sitter-elm", "src"].iter().collect();
+
+    cc::Build::new()
+        .include(&dir)
+        .file(dir.join("parser.c"))
+        .flag_if_supported("-Wno-unused-but-set-variable")
+        .compile("tree-sitter-parser");
+
+    cc::Build::new()
+        .cpp(true)
+        .include(&dir)
+        .file(dir.join("scanner.cc"))
+        .compile("tree-sitter-scanner");
+}

src/elmi.rs

@@ -1,62 +1,37 @@
 //! Basically a wrapper module for elmi-to-json for the time being.
 //! It reads the compiled .elmi files and extracts exposed tests.
 
-use miniserde::{json, Deserialize};
-use std::collections::HashSet;
-use std::path::{Path, PathBuf};
-use std::process::{Command, Stdio};
+use std::fs;
 
-/// Use elmi-to-json as a binary to extract all exposed tests
-/// from compiled .elmi files.
-pub fn all_tests<P: AsRef<Path>>(
-    work_dir: P,
-    src_files: &HashSet<PathBuf>,
+use std::path::Path;
+
+/// Find all possible tests (all values) in test_files.
+pub fn all_tests(
+    test_files: impl IntoIterator<Item = impl AsRef<Path>>,
 ) -> Result<Vec<TestModule>, String> {
-    let output = Command::new("elmi-to-json")
-        .arg("--for-elm-test")
-        .arg("--elm-version")
-        .arg("0.19.1")
-        // stdio config
-        .current_dir(&work_dir)
-        .stdin(Stdio::null())
-        .stdout(Stdio::piped())
-        .stderr(Stdio::inherit())
-        .output()
-        .expect("command failed to start");
-    let str_output = std::str::from_utf8(&output.stdout)
-        .map_err(|_| "Output of elmi-to-json is not valid UTF-8".to_string())?;
-    let output: ElmiToJsonOutput =
-        json::from_str(str_output).map_err(|_| "Deserialization error".to_string())?;
-    Ok(output
-        .test_modules
+    test_files
         .into_iter()
-        // Filter out modules with no test
-        .filter(|m| !m.tests.is_empty())
-        // Filter out modules not in the list of src_files (must also be canonical)
-        .filter(|m| {
-            let path =
-                work_dir.as_ref().join(&m.path).canonicalize().expect(
-                    "There was an issue when retrieving module paths from elmi-to-json output",
-                );
-            src_files.contains(&path)
-        })
-        // No need to verify that module names are valid, elm 0.19.1 already verifies that.
-        // No need to filter exposed since only exposed values are in elmi files.
-        .collect())
-}
+        .map(|test_file| {
+            let source = fs::read_to_string(&test_file).unwrap();
 
-#[derive(Deserialize, Debug)]
-/// Struct mirroring the json result of elmi-to-json --for-elm-test.
-struct ElmiToJsonOutput {
-    #[serde(rename = "testModules")]
-    test_modules: Vec<TestModule>,
+            let tree = {
+                let mut parser = tree_sitter::Parser::new();
+                let language = super::parser::tree_sitter_elm();
+                parser.set_language(language).unwrap();
+                parser.parse(&source, None).unwrap()
+            };
+
+            crate::parser::get_all_exposed_values(&tree, &source)
+                .map(|tests| TestModule {
+                    path: test_file.as_ref().to_str().unwrap().to_string(),
+                    tests: tests.iter().map(|s| s.to_string()).collect(),
+                })
+                .map_err(|s| s.to_string())
+        })
+        .collect()
 }
 
-#[derive(Deserialize, Debug)]
-/// Test modules as listed in the json result of elmi-to-json.
 pub struct TestModule {
-    #[serde(rename = "moduleName")]
-    pub module_name: String,
     pub path: String,
     pub tests: Vec<String>,
 }

src/main.rs

@@ -3,6 +3,7 @@ mod elmi;
 mod help;
 mod init;
 mod install;
+mod parser;
 mod run;
 mod utils;
 
@@ -46,9 +47,8 @@ fn main_args() -> Result<Args, Box<dyn std::error::Error>> {
 /// This happens for example with the command: `elm-test-rs /path/to/some/Module.elm`.
 fn no_subcommand_args(
     first_arg: Option<String>,
-    args: pico_args::Arguments,
+        mut args: pico_args::Arguments,
 ) -> Result<Args, Box<dyn std::error::Error>> {
-    let mut args = args;
     let mut rng = rand::thread_rng();
     Ok(Args::Run(run::Options {
         help: args.contains("--help"),