better comparison

diff --git a/crates/ide_assists/src/handlers/extract_function.rs b/crates/ide_assists/src/handlers/extract_function.rs
index 3750ac6127362030d8798241ad04c4f8d21f5eca..e03f4d3c82aa3e01d343db397699d37fecce75f6 100644 (file)
--- a/crates/ide_assists/src/handlers/extract_function.rs
+++ b/crates/ide_assists/src/handlers/extract_function.rs
@@ -482,9 +482,7 @@ fn from_range(parent: ast::StmtList, selected: TextRange) -> FunctionBody {
         let full_body = parent.syntax().children_with_tokens();
 
         let mut text_range = full_body
-            .filter(|it| {
-                matches!(it.kind().is_punct() || it.kind() == SyntaxKind::WHITESPACE, false)
-            })
+            .filter(|it| ast::Stmt::can_cast(it.kind()) || it.kind() == COMMENT)
             .map(|element| element.text_range())
             .filter(|&range| selected.intersect(range).filter(|it| !it.is_empty()).is_some())
             .reduce(|acc, stmt| acc.cover(stmt));