Fix SdkExtUriResolver.restoreAbsolute

pkg/analysis_server/lib/src/sdk_ext.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 analysis_server.src.sdk_ext;
 
 import 'dart:async';
 import 'dart:collection';
 import 'dart:convert';
 import 'dart:core' hide Resource;
 
-import 'package:analysis_server/uri/resolver_provider.dart';
 import 'package:analyzer/file_system/file_system.dart';
 import 'package:analyzer/source/package_map_resolver.dart';
 import 'package:analyzer/src/generated/source.dart';
 import 'package:analyzer/src/generated/java_io.dart' show JavaFile;
 import 'package:analyzer/src/generated/source_io.dart' show FileBasedSource;
 import 'package:path/path.dart' as pathos;
 
+class _SdkExtFileBasedSource extends FileBasedSource {
+  _SdkExtFileBasedSource(JavaFile file, Uri uri)
+      : super(file, uri);
+}
+
 /// Given a packageMap (see [PackageMapProvider]), check in each package's lib
 /// directory for the existence of a `.sdkext` file. This file must contain a
 /// JSON encoded map. Each key in the map is a `dart:` library name. Each value
 /// is a path (relative to the directory containing `.sdkext`) to a dart script
 /// for the given library. For example:
 /// {
 ///   "dart:sky": "../sdk_ext/dart_sky.dart"
 /// }
 ///
 /// If a key doesn't begin with `dart:` it is ignored.
@@ skipping 24 matching lines @@
   /// Number of sdk extensions.
   int get length => _urlMappings.length;
 
   /// Resolve a 'part' statement inside an sdk extension.
   Source _resolvePart(Uri libraryEntry, String partPath, Uri importUri) {
     // Library part.
     var directory = pathos.dirname(libraryEntry.path);
     var partUri = new Uri.file(pathos.join(directory, partPath));
     assert(partUri.isAbsolute);
     JavaFile javaFile = new JavaFile.fromUri(partUri);
-    return new FileBasedSource(javaFile, importUri);
+    return new _SdkExtFileBasedSource(javaFile, importUri);
   }
 
   /// Resolve an import of an sdk extension.
   Source _resolveEntry(Uri libraryEntry, Uri importUri) {
     // Library entry.
     JavaFile javaFile = new JavaFile.fromUri(libraryEntry);
-    return new FileBasedSource(javaFile, importUri);
+    return new _SdkExtFileBasedSource(javaFile, importUri);
   }
 
   @override
   Source resolveAbsolute(Uri importUri) {
     // Split import uri into library name and (optionally) a part path.
     var uri = importUri.toString();
     String libraryName;
     String partPath;
     int index = uri.indexOf('/');
     if (index >= 0) {
@@ skipping 17 matching lines @@
 
     if (index >= 0) {
       return _resolvePart(libraryEntry, partPath, importUri);
     } else {
       return _resolveEntry(libraryEntry, importUri);
     }
   }
 
   @override
   Uri restoreAbsolute(Source source) {
-    return source.uri;
+    if (source is _SdkExtFileBasedSource) {
+      return source.uri;
+    }
+    return null;
   }
 
   /// Given a package [name] and a list of folders ([libDirs]),
   /// add any found sdk extensions.
   void _processPackage(String name, List<Folder> libDirs) {
     for (var libDir in libDirs) {
       var sdkExt = _readDotSdkExt(libDir);
       if (sdkExt != null) {
         _processSdkExt(sdkExt, libDir);
       }
@@ skipping 31 matching lines @@
   void _processSdkExtension(String name, String file, Folder libDir) {
     if (!name.startsWith(DART_COLON_PREFIX)) {
       // SDK extensions must begin with 'dart:'.
       return;
     }
     var key = name;
     var value = libDir.canonicalizePath(file);
     _urlMappings[key] = value;
   }
 }