blob: 58d0a65c1776272b91abc3d076e0974fa2019b95 [file] [log] [blame]
// Copyright (c) 2017, the Dart project authors. Please see the AUTHORS file
// for details. All rights reserved. Use of this source code is governed by a
// BSD-style license that can be found in the LICENSE file.
import 'package:analyzer/dart/ast/ast.dart';
/**
* Compute the [DefinedNames] for the given [unit].
*/
DefinedNames computeDefinedNames(CompilationUnit unit) {
DefinedNames names = new DefinedNames();
void appendName(Set<String> names, SimpleIdentifier node) {
String name = node?.name;
if (name != null && name.length != 0) {
names.add(name);
}
}
void appendClassMemberName(ClassMember member) {
if (member is MethodDeclaration) {
appendName(names.classMemberNames, member.name);
} else if (member is FieldDeclaration) {
for (VariableDeclaration field in member.fields.variables) {
appendName(names.classMemberNames, field.name);
}
}
}
void appendTopLevelName(CompilationUnitMember member) {
if (member is NamedCompilationUnitMember) {
appendName(names.topLevelNames, member.name);
if (member is ClassDeclaration) {
member.members.forEach(appendClassMemberName);
}
} else if (member is TopLevelVariableDeclaration) {
for (VariableDeclaration variable in member.variables.variables) {
appendName(names.topLevelNames, variable.name);
}
}
}
unit.declarations.forEach(appendTopLevelName);
return names;
}
/**
* Defined top-level and class member names.
*/
class DefinedNames {
final Set<String> topLevelNames = new Set<String>();
final Set<String> classMemberNames = new Set<String>();
}