Issue 26063. Index declarations in even partically analyzed units.

pkg/analysis_server/lib/src/services/index/index.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:async';
 
 import 'package:analyzer/dart/ast/ast.dart';
 import 'package:analyzer/dart/element/element.dart';
 import 'package:analyzer/src/generated/engine.dart' show AnalysisContext;
 import 'package:analyzer/src/generated/source.dart';
@@ skipping 59 matching lines @@
    * Complete with a list of locations where a class members with the given
    * [name] is referenced with a qualifier, but is not resolved.
    */
   Future<List<Location>> getUnresolvedMemberReferences(String name) {
     return _mergeLocations((_ContextIndex index) {
       return index.getUnresolvedMemberReferences(name);
     });
   }
 
   /**
+   * Index declarations in the given partially resolved [unit].
+   */
+  void indexDeclarations(CompilationUnit unit) {
+    if (unit?.element?.library == null) {
+      return;
+    }
+    AnalysisContext context = unit.element.context;
+    _getContextIndex(context).indexDeclarations(unit);
+  }
+
+  /**
    * Index the given fully resolved [unit].
    */
   void indexUnit(CompilationUnit unit) {
-    if (unit == null || unit.element == null) {
+    if (unit?.element?.library == null) {
       return;
     }
     AnalysisContext context = unit.element.context;
     _getContextIndex(context).indexUnit(unit);
   }
 
   /**
    * Remove all index information for the given [context].
    */
   void removeContext(AnalysisContext context) {
@@ skipping 155 matching lines @@
    * Complete with a list of locations where a class members with the given
    * [name] is referenced with a qualifier, but is not resolved.
    */
   Future<List<Location>> getUnresolvedMemberReferences(String name) async {
     return _mergeLocations((_PackageIndexRequester requester) {
       return requester.getUnresolvedMemberReferences(context, name);
     });
   }
 
   /**
+   * Index declarations in the given partially resolved [unit].
+   */
+  void indexDeclarations(CompilationUnit unit) {
+    PackageIndexAssembler assembler = new PackageIndexAssembler();
+    assembler.indexDeclarations(unit);
+    _putUnitIndexBuilder(unit, assembler);
+  }
+
+  /**
    * Index the given fully resolved [unit].
    */
   void indexUnit(CompilationUnit unit) {
-    // Index the unit.
     PackageIndexAssembler assembler = new PackageIndexAssembler();
-    assembler.index(unit);
-    PackageIndexBuilder indexBuilder = assembler.assemble();
-    // Put the index into the map.
-    List<int> indexBytes = indexBuilder.toBuffer();
-    PackageIndex index = new PackageIndex.fromBuffer(indexBytes);
-    String key = _getUnitKeyForElement(unit.element);
-    indexMap[key] = index;
+    assembler.indexUnit(unit);
+    _putUnitIndexBuilder(unit, assembler);
   }
 
   /**
    * Remove index information about the unit.
    */
   void removeUnit(Source librarySource, Source unitSource) {
     String key = _getUnitKeyForSource(librarySource, unitSource);
     indexMap.remove(key);
   }
 
@@ skipping 12 matching lines @@
   Future<List<Location>> _mergeLocations(
       List<Location> callback(_PackageIndexRequester requester)) async {
     List<Location> locations = <Location>[];
     for (PackageIndex index in indexMap.values) {
       _PackageIndexRequester requester = new _PackageIndexRequester(index);
       List<Location> indexLocations = callback(requester);
       locations.addAll(indexLocations);
     }
     return locations;
   }
+
+  void _putUnitIndexBuilder(
+      CompilationUnit unit, PackageIndexAssembler assembler) {
+    PackageIndexBuilder indexBuilder = assembler.assemble();
+    // Put the index into the map.
+    List<int> indexBytes = indexBuilder.toBuffer();
+    PackageIndex index = new PackageIndex.fromBuffer(indexBytes);
+    String key = _getUnitKeyForElement(unit.element);
+    indexMap[key] = index;
+  }
 }
 
 /**
  * Helper for requesting information from a single [PackageIndex].
  */
 class _PackageIndexRequester {
   final PackageIndex index;
 
   _PackageIndexRequester(this.index);
 
@@ skipping 237 matching lines @@
           unitUnitUri,
           unitIndex.usedNameKinds[i],
           unitIndex.usedNameOffsets[i],
           name.length,
           unitIndex.usedNameIsQualifiedFlags[i],
           false));
     }
     return locations;
   }
 }