Fix script entry case of SdkExtUriResolver.restoreAbsolute

pkg/analyzer/lib/source/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 source.sdk_ext;
 
 import 'dart:convert';
 import 'dart:core' hide Resource;
 
 import 'package:analyzer/file_system/file_system.dart';
@@ skipping 58 matching lines @@
 
     if (partPath != null) {
       return _resolvePart(libraryEntry, partPath, importUri);
     } else {
       return _resolveEntry(libraryEntry, importUri);
     }
   }
 
   @override
   Uri restoreAbsolute(Source source) {
-    String libraryName = _libraryName(source.uri);
-    if (_registeredSdkExtension(libraryName)) {
-      return source.uri;
+    String extensionName = _findExtensionNameFor(source.fullName);
+    if (extensionName != null) {
+      return Uri.parse(extensionName);
     }
+    // TODO(johnmccutchan): Handle restoring parts.
     return null;
   }
 
+  /// Return the extension name for [fullName] or `null`.
+  String _findExtensionNameFor(String fullName) {
+    var result;
+    _urlMappings.forEach((extensionName, pathMapping) {
+      if (pathMapping == fullName) {
+        result = extensionName;
+      }
+    });
+    return result;
+  }
+
   /// Return the library name of [importUri].
   String _libraryName(Uri importUri) {
     var uri = importUri.toString();
     int index = uri.indexOf('/');
     if (index >= 0) {
       return uri.substring(0, index);
     }
     return uri;
   }
 
@@ skipping 49 matching lines @@
   String _readDotSdkExt(Folder libDir) {
     var file = libDir.getChild(SDK_EXT_NAME);
     try {
       return file.readAsStringSync();
     } on FileSystemException {
       // File can't be read.
       return null;
     }
   }
 
-  /// Returns true if [libraryName] is a registered sdk extension.
-  bool _registeredSdkExtension(String libraryName) {
-    return _urlMappings[libraryName] != null;
-  }
-
   /// 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);
   }
 
   /// 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);
   }
 }