Compute top-level declarations in a file.

pkg/analyzer/lib/src/dart/analysis/file_state.dart

 // Copyright (c) 2016, 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 'dart:convert';
 import 'dart:typed_data';
 
 import 'package:analyzer/dart/ast/ast.dart';
 import 'package:analyzer/dart/ast/token.dart';
 import 'package:analyzer/error/listener.dart';
 import 'package:analyzer/file_system/file_system.dart';
 import 'package:analyzer/src/dart/analysis/byte_store.dart';
 import 'package:analyzer/src/dart/analysis/driver.dart';
 import 'package:analyzer/src/dart/analysis/referenced_names.dart';
+import 'package:analyzer/src/dart/analysis/top_level_declarations.dart';
 import 'package:analyzer/src/dart/scanner/reader.dart';
 import 'package:analyzer/src/dart/scanner/scanner.dart';
 import 'package:analyzer/src/generated/engine.dart';
 import 'package:analyzer/src/generated/parser.dart';
 import 'package:analyzer/src/generated/source.dart';
 import 'package:analyzer/src/generated/utilities_dart.dart';
 import 'package:analyzer/src/source/source_resource.dart';
 import 'package:analyzer/src/summary/api_signature.dart';
 import 'package:analyzer/src/summary/format.dart';
 import 'package:analyzer/src/summary/idl.dart';
+import 'package:analyzer/src/summary/name_filter.dart';
 import 'package:analyzer/src/summary/summarize_ast.dart';
 import 'package:analyzer/src/util/fast_uri.dart';
 import 'package:convert/convert.dart';
 import 'package:crypto/crypto.dart';
 import 'package:meta/meta.dart';
 
 /**
  * [FileContentOverlay] is used to temporary override content of files.
  */
 class FileContentOverlay {
@@ skipping 51 matching lines @@
   String _content;
   String _contentHash;
   LineInfo _lineInfo;
   Set<String> _referencedNames;
   UnlinkedUnit _unlinked;
   List<int> _apiSignature;
 
   List<FileState> _importedFiles;
   List<FileState> _exportedFiles;
   List<FileState> _partedFiles;
+  List<NameFilter> _exportFilters;
+
   Set<FileState> _directReferencedFiles = new Set<FileState>();
   Set<FileState> _transitiveFiles;
   String _transitiveSignature;
 
+  List<TopLevelDeclaration> _topLevelDeclarations;
+
   FileState._(this._fsState, this.path, this.uri, this.source);
 
   /**
    * The unlinked API signature of the file.
    */
   List<int> get apiSignature => _apiSignature;
 
   /**
    * The content of the file.
    */
@@ skipping 51 matching lines @@
    * The list of files this library file references as parts.
    */
   List<FileState> get partedFiles => _partedFiles;
 
   /**
    * The external names referenced by the file.
    */
   Set<String> get referencedNames => _referencedNames;
 
   /**
+   * Return top-level declarations declared in the file.
+   */
+  List<TopLevelDeclaration> get topLevelDeclarations {
+    if (_topLevelDeclarations == null) {
+      _topLevelDeclarations = <TopLevelDeclaration>[];
+      // Add types.
+      for (UnlinkedClass type in unlinked.classes) {
+        _topLevelDeclarations.add(
+            new TopLevelDeclaration(TopLevelDeclarationKind.type, type.name));
+      }
+      for (UnlinkedEnum type in unlinked.enums) {
+        _topLevelDeclarations.add(
+            new TopLevelDeclaration(TopLevelDeclarationKind.type, type.name));
+      }
+      for (UnlinkedTypedef type in unlinked.typedefs) {
+        _topLevelDeclarations.add(
+            new TopLevelDeclaration(TopLevelDeclarationKind.type, type.name));
+      }
+      // Add functions and variables.
+      Set<String> addedVariableNames = new Set<String>();
+      for (UnlinkedExecutable executable in unlinked.executables) {
+        String name = executable.name;
+        if (executable.kind == UnlinkedExecutableKind.functionOrMethod) {
+          _topLevelDeclarations.add(
+              new TopLevelDeclaration(TopLevelDeclarationKind.function, name));
+        } else if (executable.kind == UnlinkedExecutableKind.getter ||
+            executable.kind == UnlinkedExecutableKind.setter) {
+          if (executable.kind == UnlinkedExecutableKind.setter) {
+            name = name.substring(0, name.length - 1);
+          }
+          if (addedVariableNames.add(name)) {
+            _topLevelDeclarations.add(new TopLevelDeclaration(
+                TopLevelDeclarationKind.variable, name));
+          }
+        }
+      }
+      for (UnlinkedVariable variable in unlinked.variables) {
+        String name = variable.name;
+        if (addedVariableNames.add(name)) {
+          _topLevelDeclarations.add(
+              new TopLevelDeclaration(TopLevelDeclarationKind.variable, name));
+        }
+      }
+    }
+    return _topLevelDeclarations;
+  }
+
+  /**
    * Return the set of transitive files - the file itself and all of the
    * directly or indirectly referenced files.
    */
   Set<FileState> get transitiveFiles {
     if (_transitiveFiles == null) {
       _transitiveFiles = new Set<FileState>();
 
       void appendReferenced(FileState file) {
         if (_transitiveFiles.add(file)) {
           file._directReferencedFiles.forEach(appendReferenced);
@@ skipping 136 matching lines @@
     if (_partedFiles != null) {
       for (FileState part in _partedFiles) {
         _fsState._partToLibraries[part]?.remove(this);
       }
     }
 
     // Build the graph.
     _importedFiles = <FileState>[];
     _exportedFiles = <FileState>[];
     _partedFiles = <FileState>[];
+    _exportFilters = <NameFilter>[];
     for (UnlinkedImport import in _unlinked.imports) {
       if (!import.isImplicit) {
         String uri = import.uri;
         if (!_isDartUri(uri)) {
           FileState file = _fileForRelativeUri(uri);
           if (file != null) {
             _importedFiles.add(file);
           }
         }
       }
     }
     for (UnlinkedExportPublic export in _unlinked.publicNamespace.exports) {
       String uri = export.uri;
       if (!_isDartUri(uri)) {
         FileState file = _fileForRelativeUri(uri);
         if (file != null) {
           _exportedFiles.add(file);
+          _exportFilters
+              .add(new NameFilter.forUnlinkedCombinators(export.combinators));
         }
       }
     }
     for (String uri in _unlinked.publicNamespace.parts) {
       if (!_isDartUri(uri)) {
         FileState file = _fileForRelativeUri(uri);
         if (file != null) {
           _partedFiles.add(file);
           // TODO(scheglov) Sort for stable results?
           _fsState._partToLibraries
@@ skipping 201 matching lines @@
         .where((f) => f._transitiveFiles == null)
         .toSet();
   }
 
   Set<FileState> get filesWithoutTransitiveSignature {
     return state._uriToFile.values
         .where((f) => f._transitiveSignature == null)
         .toSet();
   }
 }