Implement BazelPackageUriResolver.restoreAbsolute().

pkg/analyzer/lib/src/generated/bazel.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.
 
 library analyzer.src.generated.bazel;
 
 import 'dart:collection';
 import 'dart:core';
 
 import 'package:analyzer/file_system/file_system.dart';
 import 'package:analyzer/src/generated/source.dart';
 import 'package:analyzer/src/generated/source_io.dart';
+import 'package:analyzer/src/util/fast_uri.dart';
 import 'package:path/path.dart';
 
 /**
  * Instances of the class `BazelFileUriResolver` resolve `file` URI's by first
  * resolving file uri's in the expected way, and then by looking in the
  * corresponding generated directories.
  */
 class BazelFileUriResolver extends ResourceUriResolver {
   final BazelWorkspace workspace;
 
@@ skipping 56 matching lines @@
         return file?.createSource(uri);
       } else {
         String packagePath = packageName.replaceAll('.', _context.separator);
         String path =
             _context.join(_workspace.root, packagePath, 'lib', filePath);
         File file = _workspace.findFile(path);
         return file?.createSource(uri);
       }
     });
   }
+
+  @override
+  Uri restoreAbsolute(Source source) {
+    Context context = _workspace.provider.pathContext;
+    String path = source.fullName;
+
+    Uri restore(String root, String path) {
+      if (root != null && context.isWithin(root, path)) {
+        String relative = context.relative(path, from: root);
+        List<String> components = context.split(relative);
+        if (components.length >= 1 && components[0] == 'third_party') {
+          if (components.length > 4 &&
+              components[1] == 'dart' &&
+              components[3] == 'lib') {
+            String packageName = components[2];
+            String pathInLib = components.skip(4).join('/');
+            return FastUri.parse('package:$packageName/$pathInLib');
+          }
+        } else {
+          for (int i = 2; i < components.length - 1; i++) {
+            String component = components[i];
+            if (component == 'lib') {
+              String packageName = components.getRange(0, i).join('.');
+              String pathInLib = components.skip(i + 1).join('/');
+              return FastUri.parse('package:$packageName/$pathInLib');
+            }
+          }
+        }
+      }
+      return null;
+    }
+
+    // Search in each root.
+    for (String root in [
+      _workspace.bin,
+      _workspace.genfiles,
+      _workspace.readonly,
+      _workspace.root
+    ]) {
+      Uri uri = restore(root, path);
+      if (uri != null) {
+        return uri;
+      }
+    }
+
+    return null;
+  }
 }
 
 /**
  * Information about a Bazel workspace.
  */
 class BazelWorkspace {
   static const String _WORKSPACE = 'WORKSPACE';
   static const String _READONLY = 'READONLY';
 
   final ResourceProvider provider;
@@ skipping 141 matching lines @@
     Context context = provider.pathContext;
     if (provider.getFolder(context.join(root, 'blaze-genfiles')).exists) {
       return 'blaze';
     }
     if (provider.getFolder(context.join(root, 'bazel-genfiles')).exists) {
       return 'bazel';
     }
     return null;
   }
 }