Unnamed repository; edit this file 'description' to name the repository.
Diffstat (limited to 'crates/ide-db/src/imports/insert_use.rs')
| -rw-r--r-- | crates/ide-db/src/imports/insert_use.rs | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/crates/ide-db/src/imports/insert_use.rs b/crates/ide-db/src/imports/insert_use.rs index a19969ecc4..66d0b2ee82 100644 --- a/crates/ide-db/src/imports/insert_use.rs +++ b/crates/ide-db/src/imports/insert_use.rs @@ -8,7 +8,7 @@ use hir::Semantics; use syntax::{ algo, ast::{self, make, AstNode, HasAttrs, HasModuleItem, HasVisibility, PathSegmentKind}, - ted, AstToken, Direction, NodeOrToken, SyntaxNode, SyntaxToken, + ted, Direction, NodeOrToken, SyntaxKind, SyntaxNode, }; use crate::{ @@ -397,12 +397,16 @@ fn insert_use_( } // there are no imports in this file at all + // so put the import after all inner module attributes and possible license header comments if let Some(last_inner_element) = scope_syntax .children_with_tokens() - .filter(|child| match child { + .take_while(|child| match child { NodeOrToken::Node(node) => is_inner_attribute(node.clone()), - NodeOrToken::Token(token) => is_comment(token.clone()), + NodeOrToken::Token(token) => { + [SyntaxKind::WHITESPACE, SyntaxKind::COMMENT].contains(&token.kind()) + } }) + .filter(|child| child.as_token().map_or(true, |t| t.kind() != SyntaxKind::WHITESPACE)) .last() { cov_mark::hit!(insert_empty_inner_attr); @@ -439,7 +443,3 @@ fn insert_use_( fn is_inner_attribute(node: SyntaxNode) -> bool { ast::Attr::cast(node).map(|attr| attr.kind()) == Some(ast::AttrKind::Inner) } - -fn is_comment(token: SyntaxToken) -> bool { - ast::Comment::cast(token).is_some() -} |