pyrefly 0.18.1__tar.gz → 0.19.0__tar.gz

{pyrefly-0.18.1 → pyrefly-0.19.0}/Cargo.lock

@@ -13,9 +13,9 @@ dependencies = [
 
 [[package]]
 name = "adler2"
-version = "2.0.0"
+version = "2.0.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "512761e0bb2578dd7380c6baaa0f4ce03e84f95e960231d1dec8bf4d7d6e2627"
+checksum = "320119579fcad9c21884f5c4861d16174d0e06250625266f50fe6898340abefa"
 
 [[package]]
 name = "ahash"
@@ -307,9 +307,9 @@ dependencies = [
 
 [[package]]
 name = "cfg-if"
-version = "1.0.0"
+version = "1.0.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd"
+checksum = "9555578bc9e57714c812a1f84e4fc5b4d21fcb063490c624de019f7464c91268"
 
 [[package]]
 name = "chrono"
@@ -325,9 +325,9 @@ dependencies = [
 
 [[package]]
 name = "clap"
-version = "4.5.39"
+version = "4.5.40"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "fd60e63e9be68e5fb56422e397cf9baddded06dae1d2e523401542383bc72a9f"
+checksum = "40b6887a1d8685cebccf115538db5c0efe625ccac9696ad45c409d96566e910f"
 dependencies = [
  "clap_builder",
  "clap_derive",
@@ -335,9 +335,9 @@ dependencies = [
 
 [[package]]
 name = "clap_builder"
-version = "4.5.39"
+version = "4.5.40"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "89cc6392a1f72bbeb820d71f32108f61fdaf18bc526e1d23954168a67759ef51"
+checksum = "e0c66c08ce9f0c698cbce5c0279d0bb6ac936d8674174fe48f736533b964f59e"
 dependencies = [
  "anstream",
  "anstyle",
@@ -345,14 +345,14 @@ dependencies = [
  "strsim 0.11.1",
  "terminal_size",
  "unicase",
- "unicode-width 0.2.0",
+ "unicode-width",
 ]
 
 [[package]]
 name = "clap_derive"
-version = "4.5.32"
+version = "4.5.40"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "09176aae279615badda0765c0c0b3f6ed53f4709118af73cf4655d85d1530cd7"
+checksum = "d2c7947ae4cc3d851207c1adb5b5e260ff0cca11446b1d6d1423788e442257ce"
 dependencies = [
  "heck",
  "proc-macro2",
@@ -362,9 +362,9 @@ dependencies = [
 
 [[package]]
 name = "clap_lex"
-version = "0.7.4"
+version = "0.7.5"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "f46ad14479a25103f283c0f10005961cf086d8dc42205bb44c46ac563475dca6"
+checksum = "b94f61472cee1439c0b966b47e3aca9ae07e45d070759512cd390ea2bebc6675"
 
 [[package]]
 name = "colorchoice"
@@ -418,7 +418,7 @@ dependencies = [
  "encode_unicode",
  "libc",
  "once_cell",
- "unicode-width 0.2.0",
+ "unicode-width",
  "windows-sys 0.59.0",
 ]
 
@@ -834,11 +834,11 @@ dependencies = [
 
 [[package]]
 name = "getopts"
-version = "0.2.21"
+version = "0.2.23"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "14dbbfd5c71d70241ecf9e6f13737f7b5ce823821063188d7e46c41d371eebd5"
+checksum = "cba6ae63eb948698e300f645f87c70f76630d505f23b8907cf1e193ee85048c1"
 dependencies = [
- "unicode-width 0.1.14",
+ "unicode-width",
 ]
 
 [[package]]
@@ -890,9 +890,9 @@ dependencies = [
 
 [[package]]
 name = "hashbrown"
-version = "0.15.3"
+version = "0.15.4"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "84b26c544d002229e640969970a2e74021aadf6e2f96372b9c58eff97de08eb3"
+checksum = "5971ac85611da7067dbfcabef3c70ebb5606018acd9e2a3903a0da507521e0d5"
 
 [[package]]
 name = "heck"
@@ -1065,7 +1065,7 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "cea70ddb795996207ad57735b50c5982d8844f38ba9ee5f1aedcfb708a2aa11e"
 dependencies = [
  "equivalent",
- "hashbrown 0.15.3",
+ "hashbrown 0.15.4",
 ]
 
 [[package]]
@@ -1081,7 +1081,7 @@ dependencies = [
  "rayon",
  "tokio",
  "unicode-segmentation",
- "unicode-width 0.2.0",
+ "unicode-width",
  "web-time",
 ]
 
@@ -1367,9 +1367,9 @@ dependencies = [
 
 [[package]]
 name = "miniz_oxide"
-version = "0.8.8"
+version = "0.8.9"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "3be647b768db090acb35d5ec5db2b0e1f1de11133ca123b9eacf5137868f892a"
+checksum = "1fa76a2c86f704bdb222d66965fb3d63269ce38518b83cb0575fca855ebb6316"
 dependencies = [
  "adler2",
 ]
@@ -1754,7 +1754,7 @@ dependencies = [
 
 [[package]]
 name = "pyrefly"
-version = "0.18.1"
+version = "0.19.0"
 dependencies = [
  "anstream",
  "anyhow",
@@ -1771,7 +1771,6 @@ dependencies = [
  "indicatif",
  "itertools 0.14.0",
  "jemallocator",
- "lock_free_hashtable",
  "lsp-server",
  "lsp-types",
  "num-bigint",
@@ -1819,7 +1818,7 @@ dependencies = [
 
 [[package]]
 name = "pyrefly_util"
-version = "0.18.1"
+version = "0.19.0"
 dependencies = [
  "anstream",
  "anyhow",
@@ -2018,7 +2017,7 @@ checksum = "ba39f3699c378cd8970968dcbff9c43159ea4cfbd88d43c00b22f2ef10a435d2"
 [[package]]
 name = "ruff_python_ast"
 version = "0.0.0"
-source = "git+https://github.com/astral-sh/ruff/?rev=aee3af0f7a018e9fcf921b746ad8ef76d3b84b83#aee3af0f7a018e9fcf921b746ad8ef76d3b84b83"
+source = "git+https://github.com/astral-sh/ruff/?rev=6e785867c3a97691972e8a49c372b3effba90032#6e785867c3a97691972e8a49c372b3effba90032"
 dependencies = [
  "aho-corasick",
  "bitflags 2.9.1",
@@ -2035,7 +2034,7 @@ dependencies = [
 [[package]]
 name = "ruff_python_parser"
 version = "0.0.0"
-source = "git+https://github.com/astral-sh/ruff/?rev=aee3af0f7a018e9fcf921b746ad8ef76d3b84b83#aee3af0f7a018e9fcf921b746ad8ef76d3b84b83"
+source = "git+https://github.com/astral-sh/ruff/?rev=6e785867c3a97691972e8a49c372b3effba90032#6e785867c3a97691972e8a49c372b3effba90032"
 dependencies = [
  "bitflags 2.9.1",
  "bstr",
@@ -2054,7 +2053,7 @@ dependencies = [
 [[package]]
 name = "ruff_python_trivia"
 version = "0.0.0"
-source = "git+https://github.com/astral-sh/ruff/?rev=aee3af0f7a018e9fcf921b746ad8ef76d3b84b83#aee3af0f7a018e9fcf921b746ad8ef76d3b84b83"
+source = "git+https://github.com/astral-sh/ruff/?rev=6e785867c3a97691972e8a49c372b3effba90032#6e785867c3a97691972e8a49c372b3effba90032"
 dependencies = [
  "itertools 0.14.0",
  "ruff_source_file",
@@ -2065,7 +2064,7 @@ dependencies = [
 [[package]]
 name = "ruff_source_file"
 version = "0.0.0"
-source = "git+https://github.com/astral-sh/ruff/?rev=aee3af0f7a018e9fcf921b746ad8ef76d3b84b83#aee3af0f7a018e9fcf921b746ad8ef76d3b84b83"
+source = "git+https://github.com/astral-sh/ruff/?rev=6e785867c3a97691972e8a49c372b3effba90032#6e785867c3a97691972e8a49c372b3effba90032"
 dependencies = [
  "memchr",
  "ruff_text_size",
@@ -2074,7 +2073,7 @@ dependencies = [
 [[package]]
 name = "ruff_text_size"
 version = "0.0.0"
-source = "git+https://github.com/astral-sh/ruff/?rev=aee3af0f7a018e9fcf921b746ad8ef76d3b84b83#aee3af0f7a018e9fcf921b746ad8ef76d3b84b83"
+source = "git+https://github.com/astral-sh/ruff/?rev=6e785867c3a97691972e8a49c372b3effba90032#6e785867c3a97691972e8a49c372b3effba90032"
 
 [[package]]
 name = "rustc-demangle"
@@ -2795,15 +2794,9 @@ checksum = "f6ccf251212114b54433ec949fd6a7841275f9ada20dddd2f29e9ceea4501493"
 
 [[package]]
 name = "unicode-width"
-version = "0.1.14"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "7dd6e30e90baa6f72411720665d41d89b9a3d039dc45b8faea1ddd07f617f6af"
-
-[[package]]
-name = "unicode-width"
-version = "0.2.0"
+version = "0.2.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "1fc81956842c57dac11422a97c3b8195a1ff727f06e85c84ed2e8aa277c9a0fd"
+checksum = "4a1a07cc7db3810833284e8d372ccdc6da29741639ecc70c9ec107df0fa6154c"
 
 [[package]]
 name = "unicode_names2"

{pyrefly-0.18.1 → pyrefly-0.19.0}/PKG-INFO

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

{pyrefly-0.18.1 → pyrefly-0.19.0}/pyrefly/Cargo.toml

@@ -2,7 +2,7 @@
 
 [package]
 name = "pyrefly"
-version = "0.18.1"
+version = "0.19.0"
 authors = ["Meta"]
 edition = "2024"
 repository = "https://github.com/facebook/pyrefly"
@@ -31,7 +31,6 @@ enum-iterator = "1.4.1"
 equivalent = "1.0"
 indicatif = { version = "0.17.6", features = ["futures", "improved_unicode", "rayon", "tokio"] }
 itertools = "0.14.0"
-lock_free_hashtable = "0.1.1"
 lsp-server = "0.7.2"
 lsp-types = "0.94.1"
 num-bigint = { version = "0.4.4", features = ["rand"] }
@@ -43,10 +42,10 @@ pyrefly_derive = { path = "../pyrefly_derive" }
 pyrefly_util = { path = "../pyrefly_util" }
 regex = "1.11.1"
 regex-syntax = "0.7.5"
-ruff_python_ast = { git = "https://github.com/astral-sh/ruff/", rev = "aee3af0f7a018e9fcf921b746ad8ef76d3b84b83" }
-ruff_python_parser = { git = "https://github.com/astral-sh/ruff/", rev = "aee3af0f7a018e9fcf921b746ad8ef76d3b84b83" }
-ruff_source_file = { git = "https://github.com/astral-sh/ruff/", rev = "aee3af0f7a018e9fcf921b746ad8ef76d3b84b83" }
-ruff_text_size = { git = "https://github.com/astral-sh/ruff/", rev = "aee3af0f7a018e9fcf921b746ad8ef76d3b84b83" }
+ruff_python_ast = { git = "https://github.com/astral-sh/ruff/", rev = "6e785867c3a97691972e8a49c372b3effba90032" }
+ruff_python_parser = { git = "https://github.com/astral-sh/ruff/", rev = "6e785867c3a97691972e8a49c372b3effba90032" }
+ruff_source_file = { git = "https://github.com/astral-sh/ruff/", rev = "6e785867c3a97691972e8a49c372b3effba90032" }
+ruff_text_size = { git = "https://github.com/astral-sh/ruff/", rev = "6e785867c3a97691972e8a49c372b3effba90032" }
 serde = { version = "1.0.185", features = ["derive", "rc"] }
 serde_json = { version = "1.0.140", features = ["float_roundtrip", "unbounded_depth"] }
 serde_jsonrc = "0.1"

{pyrefly-0.18.1 → pyrefly-0.19.0}/pyrefly/bin/main.rs

@@ -22,6 +22,7 @@ use library::ModulePath;
 use library::ProjectLayout;
 use library::finder::ConfigFinder;
 use library::finder::debug_log;
+use library::run::AutotypeArgs;
 use library::run::BuckCheckArgs;
 use library::run::CheckArgs;
 use library::run::CommandExitStatus;
@@ -88,7 +89,7 @@ struct FullCheckArgs {
     /// When not set, Pyre will perform an upward-filesystem-walk approach to find the nearest
     /// pyrefly.toml or pyproject.toml with `tool.pyre` section'. If no config is found, Pyre exits with error.
     /// If both a pyrefly.toml and valid pyproject.toml are found, pyrefly.toml takes precedence.
-    #[arg(long, short, env = clap_env("CONFIG"))]
+    #[arg(long, short, env = clap_env("CONFIG"), value_name = "FILE")]
     config: Option<PathBuf>,
 
     #[command(flatten)]
@@ -112,6 +113,8 @@ enum Command {
 
     /// Start an LSP server
     Lsp(LspArgs),
+
+    Autotype(FullCheckArgs),
 }
 
 fn exit_on_panic() {
@@ -122,6 +125,14 @@ fn exit_on_panic() {
     }));
 }
 
+async fn run_autotype(
+    args: library::run::AutotypeArgs,
+    files_to_check: FilteredGlobs,
+    config_finder: ConfigFinder,
+) -> anyhow::Result<CommandExitStatus> {
+    args.run(files_to_check, config_finder, None)
+}
+
 async fn run_check(
     args: library::run::CheckArgs,
     watch: bool,
@@ -267,6 +278,17 @@ async fn run_command(command: Command, allow_forget: bool) -> anyhow::Result<Com
         Command::BuckCheck(args) => args.run(),
         Command::Lsp(args) => args.run(),
         Command::Init(args) => args.run(),
+        Command::Autotype(FullCheckArgs {
+            files,
+            project_excludes,
+            config,
+            watch: _,
+            mut args,
+        }) => {
+            let (files_to_check, config_finder) =
+                get_globs_and_config(files, project_excludes, config, &mut args)?;
+            run_autotype(AutotypeArgs::new(), files_to_check, config_finder).await
+        }
         // We intentionally make DumpConfig take the same arguments as Check so that dumping the
         // config is as easy as changing the command name.
         Command::DumpConfig(FullCheckArgs {

{pyrefly-0.18.1 → pyrefly-0.19.0}/pyrefly/lib/alt/answers.rs

@@ -61,6 +61,7 @@ use crate::table;
 use crate::table_for_each;
 use crate::table_mut_for_each;
 use crate::table_try_for_each;
+use crate::types::callable::Callable;
 use crate::types::class::Class;
 use crate::types::equality::TypeEq;
 use crate::types::equality::TypeEqCtx;
@@ -81,9 +82,18 @@ pub struct Index {
     pub externally_defined_attribute_references: SmallMap<ModulePath, Vec<(TextRange, TextRange)>>,
 }
 
+#[derive(Debug)]
+struct OverloadedCallee {
+    all_overloads: Vec<Callable>,
+    closest_overload: Callable,
+    is_closest_overload_chosen: bool,
+}
+
 #[derive(Debug, Default)]
 pub struct Traces {
     types: SmallMap<TextRange, Arc<Type>>,
+    /// A map from (range of callee, overload information)
+    overloaded_callees: SmallMap<TextRange, OverloadedCallee>,
 }
 
 /// Invariants:
@@ -535,6 +545,41 @@ impl Answers {
         let lock = self.trace.as_ref()?.lock();
         lock.types.get(&range).duped()
     }
+
+    pub fn get_chosen_overload_trace(&self, range: TextRange) -> Option<Callable> {
+        let lock = self.trace.as_ref()?.lock();
+        let overloaded_callee = lock.overloaded_callees.get(&range)?;
+        if overloaded_callee.is_closest_overload_chosen {
+            Some(overloaded_callee.closest_overload.clone())
+        } else {
+            None
+        }
+    }
+
+    /// Returns all the overload, and the index of a chosen one
+    pub fn get_all_overload_trace(
+        &self,
+        range: TextRange,
+    ) -> Option<(Vec<Callable>, Option<usize>)> {
+        let lock = self.trace.as_ref()?.lock();
+        let overloaded_callee = lock.overloaded_callees.get(&range)?;
+        let chosen_overload_index =
+            overloaded_callee
+                .all_overloads
+                .iter()
+                .enumerate()
+                .find_map(|(index, signature)| {
+                    if signature == &overloaded_callee.closest_overload {
+                        Some(index)
+                    } else {
+                        None
+                    }
+                });
+        Some((
+            overloaded_callee.all_overloads.clone(),
+            chosen_overload_index,
+        ))
+    }
 }
 
 impl<'a, Ans: LookupAnswer> AnswersSolver<'a, Ans> {
@@ -699,6 +744,27 @@ impl<'a, Ans: LookupAnswer> AnswersSolver<'a, Ans> {
         }
     }
 
+    /// Record all the overloads and the chosen overload.
+    /// The trace will be used to power signature help and hover for overloaded functions.
+    pub fn record_overload_trace(
+        &self,
+        loc: TextRange,
+        all_overloads: &[Callable],
+        closest_overload: &Callable,
+        is_closest_overload_chosen: bool,
+    ) {
+        if let Some(trace) = &self.current.trace {
+            trace.lock().overloaded_callees.insert(
+                loc,
+                OverloadedCallee {
+                    all_overloads: all_overloads.to_vec(),
+                    closest_overload: closest_overload.clone(),
+                    is_closest_overload_chosen,
+                },
+            );
+        }
+    }
+
     /// Check if `want` matches `got` returning `want` if the check fails.
     pub fn check_and_return_type_info(
         &self,

{pyrefly-0.18.1 → pyrefly-0.19.0}/pyrefly/lib/alt/call.rs

@@ -670,6 +670,8 @@ impl<'a, Ans: LookupAnswer> AnswersSolver<'a, Ans> {
                 None,
             );
             if arg_errors.is_empty() && call_errors.is_empty() {
+                // An overload is chosen, we should record it to power IDE services.
+                self.record_overload_trace(range, overloads.as_slice(), callable, true);
                 // It's only safe to return immediately if both arg_errors and call_errors are
                 // empty, as parameter types from the overload signature may be used as hints when
                 // evaluating arguments, producing arg_errors for some overloads but not others.
@@ -692,6 +694,12 @@ impl<'a, Ans: LookupAnswer> AnswersSolver<'a, Ans> {
         }
         // We're guaranteed to have at least one overload.
         let closest_overload = closest_overload.unwrap();
+        self.record_overload_trace(
+            range,
+            overloads.as_slice(),
+            &closest_overload.signature,
+            false,
+        );
         errors.extend(closest_overload.arg_errors);
         if closest_overload.call_errors.is_empty() {
             // No overload evaluated completely successfully, but we still say we found a match if

pyrefly-0.19.0/pyrefly/lib/alt/debugging.rs

@@ -0,0 +1,54 @@
+use pyrefly_util::display::DisplayWithCtx;
+
+use crate::alt::answers::AnswersSolver;
+use crate::alt::answers::LookupAnswer;
+use crate::binding::binding::Binding;
+use crate::binding::binding::Key;
+use crate::binding::binding::Keyed;
+use crate::binding::bindings::BindingEntry;
+use crate::binding::bindings::BindingTable;
+use crate::binding::table::TableKeyed;
+use crate::graph::index::Idx;
+
+/// Debugging helpers for the AnswersSolver.
+///
+/// These are all string-returning functions, which make them potentially less efficient
+/// but more convienient than `Display` implementations because they are easy to use
+/// for string-based comparisons for filtered debugging.
+///
+/// For example, one useful snippet in unit tests is:
+///   let debug = self.show_current_module() == "main";
+///   if debug {
+///      ... dump some information that would be too verbose if printed for stdlib modules ...
+///   }
+#[expect(dead_code)]
+impl<'a, Ans: LookupAnswer> AnswersSolver<'a, Ans> {
+    pub fn show_idx<K>(&self, idx: Idx<K>) -> String
+    where
+        K: Keyed,
+        BindingTable: TableKeyed<K, Value = BindingEntry<K>>,
+    {
+        format!(
+            "{}",
+            self.bindings()
+                .idx_to_key(idx)
+                .display_with(self.module_info())
+        )
+    }
+
+    pub fn show_binding_generic<K>(&self, binding: &K::Value) -> String
+    where
+        K: Keyed,
+        BindingTable: TableKeyed<K, Value = BindingEntry<K>>,
+    {
+        format!("{}", binding.display_with(self.bindings()))
+    }
+
+    pub fn show_binding(&self, binding: &Binding) -> String {
+        self.show_binding_generic::<Key>(binding)
+    }
+
+    pub fn show_current_module(&self) -> String {
+        format!("{}", self.module_info().name())
+    }
+}

{pyrefly-0.18.1 → pyrefly-0.19.0}/pyrefly/lib/alt/expr.rs

@@ -1364,6 +1364,13 @@ impl<'a, Ans: LookupAnswer> AnswersSolver<'a, Ans> {
                     _ => self.stdlib.str().clone().to_type(),
                 }
             }
+            Expr::TString(x) => self.error(
+                errors,
+                x.range,
+                ErrorKind::Unsupported,
+                None,
+                "t-strings are not yet supported".to_owned(),
+            ),
             Expr::StringLiteral(x) => Lit::from_string_literal(x).to_type(),
             Expr::BytesLiteral(x) => Lit::from_bytes_literal(x).to_type(),
             Expr::NumberLiteral(x) => match &x.value {

{pyrefly-0.18.1 → pyrefly-0.19.0}/pyrefly/lib/alt/mod.rs

@@ -10,6 +10,7 @@ pub mod attr;
 pub mod call;
 pub mod callable;
 pub mod class;
+pub mod debugging;
 pub mod expr;
 pub mod function;
 pub mod narrow;

{pyrefly-0.18.1 → pyrefly-0.19.0}/pyrefly/lib/alt/solve.rs

@@ -366,6 +366,7 @@ impl<'a, Ans: LookupAnswer> AnswersSolver<'a, Ans> {
             Expr::BooleanLiteral(..) => "bool literal",
             Expr::BoolOp(..) => "boolean operation",
             Expr::FString(..) => "f-string",
+            Expr::TString(..) => "t-string",
             Expr::UnaryOp(..) => "unary operation",
             // There are many Expr variants. Not all of them are likely to be used
             // in annotations, even accidentally. We can add branches for specific