pyrefly 0.36.0__tar.gz → 0.36.1__tar.gz

{pyrefly-0.36.0 → pyrefly-0.36.1}/Cargo.lock

@@ -2079,7 +2079,7 @@ dependencies = [
 
 [[package]]
 name = "pyrefly"
-version = "0.36.0"
+version = "0.36.1"
 dependencies = [
  "anstream",
  "anyhow",
@@ -2130,7 +2130,7 @@ dependencies = [
 
 [[package]]
 name = "pyrefly_build"
-version = "0.36.0"
+version = "0.36.1"
 dependencies = [
  "anyhow",
  "dupe",
@@ -2148,7 +2148,7 @@ dependencies = [
 
 [[package]]
 name = "pyrefly_bundled"
-version = "0.36.0"
+version = "0.36.1"
 dependencies = [
  "anyhow",
  "starlark_map",
@@ -2158,7 +2158,7 @@ dependencies = [
 
 [[package]]
 name = "pyrefly_config"
-version = "0.36.0"
+version = "0.36.1"
 dependencies = [
  "anyhow",
  "clap",
@@ -2193,7 +2193,7 @@ dependencies = [
 
 [[package]]
 name = "pyrefly_derive"
-version = "0.36.0"
+version = "0.36.1"
 dependencies = [
  "proc-macro2",
  "quote",
@@ -2202,7 +2202,7 @@ dependencies = [
 
 [[package]]
 name = "pyrefly_python"
-version = "0.36.0"
+version = "0.36.1"
 dependencies = [
  "anyhow",
  "dupe",
@@ -2226,7 +2226,7 @@ dependencies = [
 
 [[package]]
 name = "pyrefly_types"
-version = "0.36.0"
+version = "0.36.1"
 dependencies = [
  "compact_str 0.8.0",
  "dupe",
@@ -2246,7 +2246,7 @@ dependencies = [
 
 [[package]]
 name = "pyrefly_util"
-version = "0.36.0"
+version = "0.36.1"
 dependencies = [
  "anstream",
  "anyhow",
@@ -3309,7 +3309,7 @@ dependencies = [
 
 [[package]]
 name = "tsp_types"
-version = "0.36.0"
+version = "0.36.1"
 dependencies = [
  "lsp-server",
  "lsp-types",

{pyrefly-0.36.0 → pyrefly-0.36.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyrefly
-Version: 0.36.0
+Version: 0.36.1
 Classifier: Development Status :: 3 - Alpha
 Classifier: Programming Language :: Python
 Classifier: Programming Language :: Python :: 3

{pyrefly-0.36.0 → pyrefly-0.36.1}/crates/pyrefly_build/Cargo.toml

@@ -2,7 +2,7 @@
 
 [package]
 name = "pyrefly_build"
-version = "0.36.0"
+version = "0.36.1"
 authors = ["Meta"]
 edition = "2024"
 repository = "https://github.com/facebook/pyrefly"

{pyrefly-0.36.0 → pyrefly-0.36.1}/crates/pyrefly_build/src/lib.rs

@@ -27,21 +27,28 @@
 #![feature(if_let_guard)]
 
 use std::path::PathBuf;
+use std::sync::Arc;
 
 use serde::Deserialize;
 use serde::Serialize;
 
-use crate::buck::query::BxlArgs;
-
-pub mod buck;
 pub mod handle;
-pub mod map_db;
 pub mod source_db;
+pub use source_db::SourceDatabase;
+mod query;
+
+use crate::query::SourceDbQuerier;
+use crate::query::buck::BxlArgs;
+use crate::query::buck::BxlQuerier;
+use crate::query::custom::CustomQuerier;
+use crate::query::custom::CustomQueryArgs;
+use crate::source_db::query_source_db::QuerySourceDatabase;
 
 #[derive(Debug, Clone, Deserialize, Serialize, PartialEq, Eq)]
 #[serde(rename_all = "kebab-case", tag = "type")]
 pub enum BuildSystem {
     Buck(BxlArgs),
+    Custom(CustomQueryArgs),
 }
 
 impl BuildSystem {
@@ -49,11 +56,10 @@ impl BuildSystem {
         &self,
         config_root: PathBuf,
     ) -> Box<dyn source_db::SourceDatabase + 'static> {
-        match &self {
-            Self::Buck(args) => Box::new(buck::bxl::BuckSourceDatabase::new(
-                config_root,
-                args.clone(),
-            )),
-        }
+        let querier: Arc<dyn SourceDbQuerier> = match &self {
+            Self::Buck(args) => Arc::new(BxlQuerier::new(args.clone())),
+            Self::Custom(args) => Arc::new(CustomQuerier::new(args.clone())),
+        };
+        Box::new(QuerySourceDatabase::new(config_root, querier))
     }
 }

pyrefly-0.36.1/crates/pyrefly_build/src/query/buck.rs

@@ -0,0 +1,47 @@
+/*
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ */
+
+use std::fmt::Debug;
+use std::process::Command;
+
+use serde::Deserialize;
+use serde::Serialize;
+
+use crate::query::SourceDbQuerier;
+
+#[derive(Debug, Clone, Deserialize, Serialize, PartialEq, Eq, Default)]
+#[serde(rename_all = "kebab-case")]
+pub struct BxlArgs {
+    isolation_dir: Option<String>,
+    extras: Option<Vec<String>>,
+}
+
+#[derive(Debug)]
+pub struct BxlQuerier(BxlArgs);
+
+impl BxlQuerier {
+    pub fn new(args: BxlArgs) -> Self {
+        Self(args)
+    }
+}
+
+impl SourceDbQuerier for BxlQuerier {
+    fn construct_command(&self) -> Command {
+        let mut cmd = Command::new("buck2");
+        if let Some(isolation_dir) = &self.0.isolation_dir {
+            cmd.arg("--isolation-dir");
+            cmd.arg(isolation_dir);
+        }
+        cmd.arg("bxl");
+        cmd.arg("--reuse-current-config");
+        if let Some(metadata) = &self.0.extras {
+            cmd.args(metadata);
+        }
+        cmd.arg("prelude//python/sourcedb/pyrefly.bxl:main");
+        cmd
+    }
+}

pyrefly-0.36.1/crates/pyrefly_build/src/query/custom.rs

@@ -0,0 +1,52 @@
+/*
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ */
+
+use std::fmt::Debug;
+use std::process::Command;
+
+use serde::Deserialize;
+use serde::Serialize;
+use vec1::Vec1;
+
+use crate::query::SourceDbQuerier;
+
+/// Args and settings for querying a custom source DB.
+#[derive(Debug, Clone, Deserialize, Serialize, PartialEq, Eq, Default)]
+#[serde(rename_all = "kebab-case")]
+pub struct CustomQueryArgs {
+    /// The command to run.
+    /// Pyrefly will call this in the form `<command> @<argfile>`,
+    /// where `<argfile>` has the format
+    /// ```text
+    /// --
+    /// <arg-flag>
+    /// <arg>
+    /// ...
+    /// ```
+    /// and `<arg-flag>` is either `--file` or `--target`, depending on the type
+    /// of `<arg>`
+    /// and `<arg>` is an absolute path to a file or a build system's target.
+    command: Vec1<String>,
+}
+
+/// A querier allowing for a custom command when querying and constructing source DB.
+#[derive(Debug)]
+pub struct CustomQuerier(CustomQueryArgs);
+
+impl CustomQuerier {
+    pub fn new(args: CustomQueryArgs) -> Self {
+        Self(args)
+    }
+}
+
+impl SourceDbQuerier for CustomQuerier {
+    fn construct_command(&self) -> Command {
+        let mut cmd = Command::new(self.0.command.first());
+        cmd.args(self.0.command.iter().skip(1));
+        cmd
+    }
+}

pyrefly-0.36.0/crates/pyrefly_build/src/buck/query.rs → pyrefly-0.36.1/crates/pyrefly_build/src/query/mod.rs

@@ -7,8 +7,8 @@
 
 use std::ffi::OsStr;
 use std::ffi::OsString;
-use std::fmt::Debug;
-use std::io::Write;
+use std::fmt;
+use std::io::Write as _;
 use std::path::Path;
 use std::path::PathBuf;
 use std::process::Command;
@@ -18,7 +18,6 @@ use dupe::Dupe as _;
 use pyrefly_python::module_name::ModuleName;
 use pyrefly_python::sys_info::SysInfo;
 use serde::Deserialize;
-use serde::Serialize;
 use starlark_map::small_map::SmallMap;
 use starlark_map::small_set::SmallSet;
 use tempfile::NamedTempFile;
@@ -26,6 +25,9 @@ use vec1::Vec1;
 
 use crate::source_db::Target;
 
+pub mod buck;
+pub mod custom;
+
 /// An enum representing something that has been included by the build system, and
 /// which the build system should query for when building the sourcedb.
 #[derive(Debug, PartialEq, Eq, Hash)]
@@ -40,7 +42,7 @@ impl Include {
         Self::Path(path)
     }
 
-    fn to_bxl_args(&self) -> impl Iterator<Item = &OsStr> {
+    fn to_cli_arg(&self) -> impl Iterator<Item = &OsStr> {
         match self {
             Include::Target(target) => [OsStr::new("--target"), target.to_os_str()].into_iter(),
             Include::Path(path) => [OsStr::new("--file"), path.as_os_str()].into_iter(),
@@ -48,68 +50,60 @@ impl Include {
     }
 }
 
-#[derive(Debug, Clone, Deserialize, Serialize, PartialEq, Eq, Default)]
-#[serde(rename_all = "kebab-case")]
-pub struct BxlArgs {
-    isolation_dir: Option<String>,
-    extras: Option<Vec<String>>,
-}
+pub trait SourceDbQuerier: Send + Sync + fmt::Debug {
+    fn query_source_db(
+        &self,
+        files: &SmallSet<Include>,
+        cwd: &Path,
+    ) -> anyhow::Result<TargetManifestDatabase> {
+        if files.is_empty() {
+            return Ok(TargetManifestDatabase {
+                db: SmallMap::new(),
+                root: cwd.to_path_buf(),
+            });
+        }
 
-pub(crate) fn query_source_db<'a>(
-    files: impl Iterator<Item = &'a Include>,
-    cwd: &Path,
-    bxl_args: &BxlArgs,
-) -> anyhow::Result<TargetManifestDatabase> {
-    let mut files = files.peekable();
-    if files.peek().is_none() {
-        return Ok(TargetManifestDatabase {
-            db: SmallMap::new(),
-            root: cwd.to_path_buf(),
+        let mut argfile =
+            NamedTempFile::with_prefix("pyrefly_build_query_").with_context(|| {
+                "Failed to create temporary argfile for querying source DB".to_owned()
+            })?;
+        let mut argfile_args = OsString::from("--");
+        files.iter().flat_map(Include::to_cli_arg).for_each(|arg| {
+            argfile_args.push("\n");
+            argfile_args.push(arg);
         });
-    }
 
-    let mut argfile = NamedTempFile::with_prefix("pyrefly_buck_query_")
-        .with_context(|| "Failed to create temporary argfile for querying Buck".to_owned())?;
-    let mut argfile_args = OsString::from("--");
-    files.flat_map(Include::to_bxl_args).for_each(|arg| {
-        argfile_args.push("\n");
-        argfile_args.push(arg);
-    });
-
-    argfile
-        .as_file_mut()
-        .write_all(argfile_args.as_encoded_bytes())
-        .with_context(|| "Could not write to argfile when querying Buck".to_owned())?;
-
-    let mut cmd = Command::new("buck2");
-    if let Some(isolation_dir) = &bxl_args.isolation_dir {
-        cmd.arg("--isolation-dir");
-        cmd.arg(isolation_dir);
-    }
-    cmd.arg("bxl");
-    cmd.arg("--reuse-current-config");
-    if let Some(metadata) = &bxl_args.extras {
-        cmd.args(metadata);
-    }
-    cmd.arg("prelude//python/sourcedb/pyrefly.bxl:main");
-    cmd.arg(format!("@{}", argfile.path().display()));
-    cmd.current_dir(cwd);
-
-    let result = cmd.output()?;
-    if !result.status.success() {
-        let stdout = String::from_utf8(result.stdout)
-            .unwrap_or_else(|_| "<Failed to parse stdout from Buck source db query>".to_owned());
-        let stderr = String::from_utf8(result.stderr)
-            .unwrap_or_else(|_| "<Failed to parse stderr from Buck source db query>".to_owned());
-
-        return Err(anyhow::anyhow!(
-            "Buck source db query failed...\nSTDOUT: {stdout}\nSTDERR: {stderr}"
-        ));
+        argfile
+            .as_file_mut()
+            .write_all(argfile_args.as_encoded_bytes())
+            .with_context(|| "Could not write to argfile when querying source DB".to_owned())?;
+
+        let mut cmd = self.construct_command();
+        cmd.arg(format!("@{}", argfile.path().display()));
+        cmd.current_dir(cwd);
+
+        let result = cmd.output()?;
+        if !result.status.success() {
+            let stdout = String::from_utf8(result.stdout)
+                .unwrap_or_else(|_| "<Failed to parse stdout from source DB query>".to_owned());
+            let stderr = String::from_utf8(result.stderr).unwrap_or_else(|_| {
+                "<Failed to parse stderr from Buck source DB query>".to_owned()
+            });
+
+            return Err(anyhow::anyhow!(
+                "Source DB query failed...\nSTDOUT: {stdout}\nSTDERR: {stderr}"
+            ));
+        }
+
+        serde_json::from_slice(&result.stdout).with_context(|| {
+            format!(
+                "Failed to construct valid `TargetManifestDatabase` from querier result. Command run: `{}`",
+                cmd.get_program().display(),
+            )
+        })
     }
 
-    serde_json::from_slice(&result.stdout).with_context(|| {
-        "Failed to construct valid `TargetManifestDatabase` from BXL query result".to_owned()
-    })
+    fn construct_command(&self) -> Command;
 }
 
 #[derive(Debug, PartialEq, Eq, Deserialize, Clone)]

pyrefly-0.36.0/crates/pyrefly_build/src/source_db.rs → pyrefly-0.36.1/crates/pyrefly_build/src/source_db/mod.rs

@@ -24,6 +24,10 @@ use static_interner::Interner;
 
 use crate::handle::Handle;
 
+pub mod buck_check;
+pub mod map_db;
+pub(crate) mod query_source_db;
+
 // We're interning `Target`s, since they'll be duplicated all over the place,
 // and it would be nice to have something that implements `Copy`.
 // We choose Interning over `Arc`, since we want to make sure all `Target`s

pyrefly-0.36.0/crates/pyrefly_build/src/buck/bxl.rs → pyrefly-0.36.1/crates/pyrefly_build/src/source_db/query_source_db.rs

@@ -8,6 +8,7 @@
 use std::collections::VecDeque;
 use std::path::Path;
 use std::path::PathBuf;
+use std::sync::Arc;
 
 use dupe::Dupe as _;
 use pyrefly_python::module_name::ModuleName;
@@ -20,12 +21,11 @@ use starlark_map::small_set::SmallSet;
 use tracing::debug;
 use tracing::info;
 
-use crate::buck::query::BxlArgs;
-use crate::buck::query::Include;
-use crate::buck::query::PythonLibraryManifest;
-use crate::buck::query::TargetManifestDatabase;
-use crate::buck::query::query_source_db;
 use crate::handle::Handle;
+use crate::query::Include;
+use crate::query::PythonLibraryManifest;
+use crate::query::SourceDbQuerier;
+use crate::query::TargetManifestDatabase;
 use crate::source_db::SourceDatabase;
 use crate::source_db::Target;
 
@@ -52,7 +52,7 @@ impl Inner {
 }
 
 #[derive(Debug)]
-pub struct BuckSourceDatabase {
+pub struct QuerySourceDatabase {
     inner: RwLock<Inner>,
     /// The set of items the sourcedb has been queried for. Not all of the targets
     /// or files listed here will necessarily appear in the sourcedb, for example,
@@ -61,16 +61,16 @@ pub struct BuckSourceDatabase {
     /// The directory that will be passed into the sourcedb query shell-out. Should
     /// be the same as the directory containing the config this sourcedb is a part of.
     cwd: PathBuf,
-    bxl_args: BxlArgs,
+    querier: Arc<dyn SourceDbQuerier>,
 }
 
-impl BuckSourceDatabase {
-    pub fn new(cwd: PathBuf, bxl_args: BxlArgs) -> Self {
-        BuckSourceDatabase {
+impl QuerySourceDatabase {
+    pub fn new(cwd: PathBuf, querier: Arc<dyn SourceDbQuerier>) -> Self {
+        QuerySourceDatabase {
             cwd,
             inner: RwLock::new(Inner::new()),
             includes: Mutex::new(SmallSet::new()),
-            bxl_args,
+            querier,
         }
     }
 
@@ -100,7 +100,7 @@ impl BuckSourceDatabase {
     }
 }
 
-impl SourceDatabase for BuckSourceDatabase {
+impl SourceDatabase for QuerySourceDatabase {
     fn modules_to_check(&self) -> Vec<crate::handle::Handle> {
         // TODO(connernilsen): implement modules_to_check
         vec![]
@@ -176,7 +176,7 @@ impl SourceDatabase for BuckSourceDatabase {
         }
         *includes = new_includes;
         info!("Querying Buck for source DB");
-        let raw_db = query_source_db(includes.iter(), &self.cwd, &self.bxl_args)?;
+        let raw_db = self.querier.query_source_db(&includes, &self.cwd)?;
         info!("Finished querying Buck for source DB");
         Ok(self.update_with_target_manifest(raw_db))
     }
@@ -206,9 +206,26 @@ mod tests {
     use starlark_map::smallset;
 
     use super::*;
-    use crate::buck::query::TargetManifest;
+    use crate::query::TargetManifest;
+
+    #[derive(Debug)]
+    struct DummyQuerier {}
+
+    impl SourceDbQuerier for DummyQuerier {
+        fn query_source_db(
+            &self,
+            _: &SmallSet<Include>,
+            _: &Path,
+        ) -> anyhow::Result<TargetManifestDatabase> {
+            Ok(TargetManifestDatabase::get_test_database())
+        }
+
+        fn construct_command(&self) -> std::process::Command {
+            panic!("We shouldn't be calling this...");
+        }
+    }
 
-    impl BuckSourceDatabase {
+    impl QuerySourceDatabase {
         fn from_target_manifest_db(
             raw_db: TargetManifestDatabase,
             files: &SmallSet<PathBuf>,
@@ -222,14 +239,14 @@ mod tests {
                         .collect(),
                 ),
                 cwd: PathBuf::new(),
-                bxl_args: Default::default(),
+                querier: Arc::new(DummyQuerier {}),
             };
             new.update_with_target_manifest(raw_db);
             new
         }
     }
 
-    fn get_db() -> (BuckSourceDatabase, PathBuf) {
+    fn get_db() -> (QuerySourceDatabase, PathBuf) {
         let raw_db = TargetManifestDatabase::get_test_database();
         let root = raw_db.root.to_path_buf();
         let files = smallset! {
@@ -238,7 +255,7 @@ mod tests {
         };
 
         (
-            BuckSourceDatabase::from_target_manifest_db(raw_db, &files),
+            QuerySourceDatabase::from_target_manifest_db(raw_db, &files),
             root,
         )
     }

{pyrefly-0.36.0 → pyrefly-0.36.1}/crates/pyrefly_bundled/Cargo.toml

@@ -2,7 +2,7 @@
 
 [package]
 name = "pyrefly_bundled"
-version = "0.36.0"
+version = "0.36.1"
 authors = ["Meta"]
 edition = "2024"
 repository = "https://github.com/facebook/pyrefly"

{pyrefly-0.36.0 → pyrefly-0.36.1}/crates/pyrefly_config/Cargo.toml

@@ -2,7 +2,7 @@
 
 [package]
 name = "pyrefly_config"
-version = "0.36.0"
+version = "0.36.1"
 authors = ["Meta"]
 edition = "2024"
 repository = "https://github.com/facebook/pyrefly"

{pyrefly-0.36.0 → pyrefly-0.36.1}/crates/pyrefly_config/src/environment/mod.rs

@@ -9,5 +9,5 @@ pub(crate) mod active_environment;
 pub(crate) mod conda;
 pub mod environment;
 pub(crate) mod finder;
-pub(crate) mod interpreters;
+pub mod interpreters;
 pub(crate) mod venv;

{pyrefly-0.36.0 → pyrefly-0.36.1}/crates/pyrefly_derive/Cargo.toml

@@ -2,7 +2,7 @@
 
 [package]
 name = "pyrefly_derive"
-version = "0.36.0"
+version = "0.36.1"
 authors = ["Meta"]
 edition = "2024"
 repository = "https://github.com/facebook/pyrefly"

{pyrefly-0.36.0 → pyrefly-0.36.1}/crates/pyrefly_python/Cargo.toml

@@ -2,7 +2,7 @@
 
 [package]
 name = "pyrefly_python"
-version = "0.36.0"
+version = "0.36.1"
 authors = ["Meta"]
 edition = "2024"
 repository = "https://github.com/facebook/pyrefly"

{pyrefly-0.36.0 → pyrefly-0.36.1}/crates/pyrefly_types/Cargo.toml

@@ -2,7 +2,7 @@
 
 [package]
 name = "pyrefly_types"
-version = "0.36.0"
+version = "0.36.1"
 authors = ["Meta"]
 edition = "2024"
 repository = "https://github.com/facebook/pyrefly"

{pyrefly-0.36.0 → pyrefly-0.36.1}/crates/pyrefly_util/Cargo.toml

@@ -2,7 +2,7 @@
 
 [package]
 name = "pyrefly_util"
-version = "0.36.0"
+version = "0.36.1"
 authors = ["Meta"]
 edition = "2024"
 repository = "https://github.com/facebook/pyrefly"

{pyrefly-0.36.0 → pyrefly-0.36.1}/crates/tsp_types/Cargo.toml

@@ -2,7 +2,7 @@
 
 [package]
 name = "tsp_types"
-version = "0.36.0"
+version = "0.36.1"
 authors = ["Meta"]
 edition = "2024"
 repository = "https://github.com/facebook/pyrefly"

{pyrefly-0.36.0 → pyrefly-0.36.1}/pyrefly/Cargo.toml

@@ -2,7 +2,7 @@
 
 [package]
 name = "pyrefly"
-version = "0.36.0"
+version = "0.36.1"
 authors = ["Meta"]
 edition = "2024"
 repository = "https://github.com/facebook/pyrefly"