Write out AIR_CRATE_NAMES at build time

Cargo.toml

@@ -45,6 +45,7 @@ biome_string_case = { git = "https://github.com/biomejs/biome", rev = "2648fa420
 biome_text_size = { git = "https://github.com/biomejs/biome", rev = "2648fa4201be4afd26f44eca1a4e77aac0a67272" }
 biome_unicode_table = { git = "https://github.com/biomejs/biome", rev = "2648fa4201be4afd26f44eca1a4e77aac0a67272" }
 bytes = "1.8.0"
+cargo_metadata = "0.19.1"
 clap = { version = "4.5.20", features = ["derive"] }
 crossbeam = "0.8.4"
 dissimilar = "1.0.9"

crates/lsp/Cargo.toml

@@ -41,9 +41,6 @@ triomphe.workspace = true
 url.workspace = true
 uuid = { workspace = true, features = ["v4"] }
 
-[lints]
-workspace = true
-
 [dev-dependencies]
 assert_matches.workspace = true
 bytes.workspace = true
@@ -54,3 +51,9 @@ lsp_test.workspace = true
 memchr.workspace = true
 tests_macros.workspace = true
 tokio-util.workspace = true
+
+[build-dependencies]
+cargo_metadata.workspace = true
+
+[lints]
+workspace = true

crates/lsp/build.rs

@@ -0,0 +1,34 @@
+use std::env;
+use std::fs;
+use std::path::Path;
+
+extern crate cargo_metadata;
+
+fn main() {
+    write_workspace_crate_names();
+}
+
+/// Write out a constant array of air crate names as `AIR_CRATE_NAMES` at build time
+fn write_workspace_crate_names() {
+    let dir = env::var_os("OUT_DIR").unwrap();
+    let path = Path::new(&dir).join("crates.rs");
+
+    // Equivalent to `cargo metadata --no-deps`
+    let mut cmd = cargo_metadata::MetadataCommand::new();
+    cmd.no_deps();
+    let metadata = cmd.exec().unwrap();
+
+    let packages: Vec<String> = metadata
+        .workspace_packages()
+        .iter()
+        .map(|package| package.name.clone())
+        .map(|package| String::from("\"") + package.as_str() + "\",")
+        .collect();
+
+    let packages = packages.join(" ");
+
+    let contents = format!("pub(crate) const AIR_CRATE_NAMES: &[&str] = &[{packages}];");
+
+    fs::write(&path, contents).unwrap();
+    println!("cargo::rerun-if-changed=build.rs");
+}

crates/lsp/src/crates.rs

@@ -0,0 +1,3 @@
+// Generates `AIR_CRATE_NAMES`, a const array of the crate names in the air workspace,
+// see `lsp/src/build.rs`
+include!(concat!(env!("OUT_DIR"), "/crates.rs"));

crates/lsp/src/lib.rs

@@ -4,6 +4,7 @@
 pub use tower_lsp::start_lsp;
 
 pub mod config;
+pub mod crates;
 pub mod documents;
 pub mod encoding;
 pub mod from_proto;

crates/lsp/src/logging.rs

@@ -59,6 +59,8 @@ use tracing_subscriber::{
     Layer,
 };
 
+use crate::crates;
+
 // TODO:
 // - Add `air.logLevel` and `air.dependencyLogLevels` as VS Code extension options that set
 //   the log levels, and pass them through the arbitrary `initializationOptions` field of
@@ -226,17 +228,6 @@ fn is_test_client(client_info: Option<&ClientInfo>) -> bool {
     client_info.map_or(false, |client_info| client_info.name == "AirTestClient")
 }
 
-// TODO: Is there a way to generate this at compile time?
-const TARGETS: &[&str] = &[
-    "air_r_factory",
-    "air_r_formatter",
-    "air_r_parser",
-    "air_r_syntax",
-    "fs",
-    "line_ending",
-    "lsp",
-];
-
 fn log_filter(log_level: LogLevel, dependency_log_levels: Option<String>) -> filter::Targets {
     // Initialize `filter` from dependency log levels.
     // If nothing is supplied, dependency logs are completely off.
@@ -251,7 +242,7 @@ fn log_filter(log_level: LogLevel, dependency_log_levels: Option<String>) -> fil
     let log_level = log_level.tracing_level();
 
     // Apply the air log level to each air crate that logs
-    for target in TARGETS {
+    for target in crates::AIR_CRATE_NAMES {
         filter = filter.with_target(*target, log_level);
     }