fix: flyimport triggers on enum variant declarations

[rust.git] / crates / ide_completion / src / context.rs

diff --git a/crates/ide_completion/src/context.rs b/crates/ide_completion/src/context.rs
index 05ae95769b002251330f864327d7d8f30e25ecf4..8759e2d108c7ad2f7061991b4c0d5d063fe76430 100644 (file)
--- a/crates/ide_completion/src/context.rs
+++ b/crates/ide_completion/src/context.rs
@@ -171,6 +171,10 @@ pub(crate) fn expects_assoc_item(&self) -> bool {
         matches!(self.completion_location, Some(ImmediateLocation::Trait | ImmediateLocation::Impl))
     }
 
+    pub(crate) fn expects_variant(&self) -> bool {
+        matches!(self.completion_location, Some(ImmediateLocation::Variant))
+    }
+
     pub(crate) fn expects_non_trait_assoc_item(&self) -> bool {
         matches!(self.completion_location, Some(ImmediateLocation::Impl))
     }