fix Command-line entry-point for LSP features #3148

pyrefly/lib/commands.rs

@@ -21,6 +21,8 @@ pub mod init;
 #[cfg(not(target_arch = "wasm32"))]
 pub mod lsp;
 #[cfg(not(target_arch = "wasm32"))]
+pub mod query;
+#[cfg(not(target_arch = "wasm32"))]
 pub mod stubgen;
 #[cfg(not(target_arch = "wasm32"))]
 pub mod suppress;

pyrefly/lib/commands/all.rs

@@ -23,6 +23,7 @@ use crate::commands::dump_config::DumpConfigArgs;
 use crate::commands::infer::InferArgs;
 use crate::commands::init::InitArgs;
 use crate::commands::lsp::LspArgs;
+use crate::commands::query::QueryArgs;
 use crate::commands::stubgen::StubgenArgs;
 use crate::commands::suppress::SuppressArgs;
 use crate::commands::tsp::TspArgs;
@@ -55,6 +56,9 @@ pub enum Command {
     /// Start an LSP server
     Lsp(LspArgs),
 
+    /// Query semantic information and print the result as JSON
+    Query(QueryArgs),
+
     /// Start a TSP server
     Tsp(TspArgs),
     /// Automatically add type annotations to a file or directory.
@@ -102,6 +106,7 @@ impl Command {
                 )?,
                 None,
             )),
+            Command::Query(args) => Ok((args.run(config_configurer_wrapper, thread_count)?, None)),
             Command::Tsp(args) => Ok((
                 args.run(
                     telemetry,