Better visibility check.

This commit is contained in:
Andrea Pretto 2019-01-26 13:21:41 +01:00
parent 27c071f702
commit b9ca8ab868

View File

@ -83,7 +83,11 @@ fn fold_kind(kind: SyntaxKind) -> Option<FoldKind> {
}
fn has_visibility(node: &SyntaxNode) -> bool {
return node.descendants().any(|n| n.kind() == VISIBILITY);
use ast::VisibilityOwner;
return ast::Module::cast(node)
.and_then(|m| m.visibility())
.is_some();
}
fn has_newline(node: &SyntaxNode) -> bool {