suggest after import prefix in part files - fixes #25023

pkg/analysis_server/lib/src/services/completion/dart/completion_manager.dart

 // Copyright (c) 2015, 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.
 
 library services.completion.dart.manager;
 
 import 'dart:async';
 
 import 'package:analysis_server/plugin/protocol/protocol.dart';
 import 'package:analysis_server/src/provisional/completion/completion_core.dart'
@@ skipping 42 matching lines @@
     return suggestions;
   }
 }
 
 /**
  * The information about a requested list of completions within a Dart file.
  */
 class DartCompletionRequestImpl extends CompletionRequestImpl
     implements DartCompletionRequest {
   /**
+   * The source for the library containing the completion request.
+   * This may be different from the source in which the completion is requested
+   * if the completion is being requested in a part file.
+   * This may be `null` if the library for a part file cannot be determined.
+   */
+  Source _librarySource;
+
+  /**
    * The cached completion target or `null` if not computed yet.
    */
   CompletionTarget _target;
 
   /**
    * The [DartType] for Object in dart:core
    */
   InterfaceType _objectType;
 
   /**
@@ skipping 13 matching lines @@
         request.source,
         request.offset);
   }
 
   DartCompletionRequestImpl._(
       AnalysisContext context,
       ResourceProvider resourceProvider,
       SearchEngine searchEngine,
       Source source,
       int offset)
-      : super(context, resourceProvider, searchEngine, source, offset);
+      : super(context, resourceProvider, searchEngine, source, offset) {
+    if (target.unit.directives.any((d) => d is PartOfDirective)) {
+      List<Source> libraries = context.getLibrariesContaining(source);
+      if (libraries.isNotEmpty) {
+        _librarySource = libraries[0];
+      }
+    } else {
+      _librarySource = source;
+    }
+  }
 
   @override
   Future<LibraryElement> get libraryElement async {
     //TODO(danrubel) build the library element rather than all the declarations
     CompilationUnit unit = await resolveDeclarationsInScope();
     if (unit != null) {
       CompilationUnitElement elem = unit.element;
       if (elem != null) {
         return elem.library;
       }
@@ skipping 20 matching lines @@
     return _target;
   }
 
   @override
   Future<CompilationUnit> resolveDeclarationsInScope() async {
     CompilationUnit unit = target.unit;
     if (_haveResolveDeclarationsInScope) {
       return unit;
     }
 
-    // Determine the library source
-    Source librarySource;
-    if (unit.directives.any((d) => d is PartOfDirective)) {
-      List<Source> libraries = context.getLibrariesContaining(source);
-      if (libraries.isEmpty) {
-        return null;
-      }
-      librarySource = libraries[0];
-    } else {
-      librarySource = source;
+    // Gracefully degrade if librarySource cannot be determined
+    if (_librarySource == null) {
+      return null;
     }
 
     // Resolve declarations in the target unit
     CompilationUnit resolvedUnit =
-        await new AnalysisFutureHelper<CompilationUnit>(
-            context,
-            new LibrarySpecificUnit(librarySource, source),
-            RESOLVED_UNIT3).computeAsync();
+        await new AnalysisFutureHelper<CompilationUnit>(context,
+                new LibrarySpecificUnit(_librarySource, source), RESOLVED_UNIT3)
+            .computeAsync();
 
     // TODO(danrubel) determine if the underlying source has been modified
     // in a way that invalidates the completion request
     // and return null
 
     // Gracefully degrade if unit cannot be resolved
     if (resolvedUnit == null) {
       return null;
     }
 
     // Recompute the target for the newly resolved unit
     _target = new CompletionTarget.forOffset(resolvedUnit, offset);
     _haveResolveDeclarationsInScope = true;
     return resolvedUnit;
   }
 
   @override
+  Future<List<Directive>> resolveDirectives() async {
+    CompilationUnit libUnit;
+    if (_librarySource == source) {
+      libUnit = await resolveDeclarationsInScope();
+    } else if (_librarySource != null) {
+      libUnit =
+          await new AnalysisFutureHelper<CompilationUnit>(
+                  context,
+                  new LibrarySpecificUnit(_librarySource, _librarySource),
+                  RESOLVED_UNIT3)
+              .computeAsync();
+    }
+    return libUnit?.directives;
+  }
+
+  @override
   Future resolveExpression(Expression expression) async {
     //TODO(danrubel) resolve the expression or containing method
     // rather than the entire complilation unit
 
-    CompilationUnit unit = target.unit;
-
-    // Determine the library source
-    Source librarySource;
-    if (unit.directives.any((d) => d is PartOfDirective)) {
-      List<Source> libraries = context.getLibrariesContaining(source);
-      if (libraries.isEmpty) {
-        return;
-      }
-      librarySource = libraries[0];
-    } else {
-      librarySource = source;
+    // Gracefully degrade if librarySource cannot be determined
+    if (_librarySource == null) {
+      return null;
     }
 
     // Resolve declarations in the target unit
     CompilationUnit resolvedUnit =
-        await new AnalysisFutureHelper<CompilationUnit>(
-            context,
-            new LibrarySpecificUnit(librarySource, source),
-            RESOLVED_UNIT).computeAsync();
+        await new AnalysisFutureHelper<CompilationUnit>(context,
+                new LibrarySpecificUnit(_librarySource, source), RESOLVED_UNIT)
+            .computeAsync();
 
     // TODO(danrubel) determine if the underlying source has been modified
     // in a way that invalidates the completion request
     // and return null
 
     // Gracefully degrade if unit cannot be resolved
     if (resolvedUnit == null) {
-      return;
+      return null;
     }
 
     // Recompute the target for the newly resolved unit
     _target = new CompletionTarget.forOffset(resolvedUnit, offset);
     _haveResolveDeclarationsInScope = true;
   }
 }