fix: Fix import insertion inserting after last comment in a file

This commit is contained in:
Lukas Wirth 2022-05-09 11:52:49 +02:00
parent 5d5bbec9b6
commit c0feb389ed
2 changed files with 25 additions and 7 deletions

View File

@ -8,7 +8,7 @@
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()
}

View File

@ -5,6 +5,24 @@
use super::*;
#[test]
fn trailing_comment_in_empty_file() {
check(
"foo::bar",
r#"
struct Struct;
// 0 = 1
"#,
r#"
use foo::bar;
struct Struct;
// 0 = 1
"#,
ImportGranularity::Crate,
);
}
#[test]
fn respects_cfg_attr_fn() {
check(