add from option to run

lib/src/mirror_loader.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 initialize.mirror_loader;
 
 import 'dart:collection' show Queue;
 import 'dart:mirrors';
 import 'package:path/path.dart' as path;
 import 'package:initialize/initialize.dart';
 
 final _root = currentMirrorSystem().isolate.rootLibrary;
 
 Queue<Function> loadInitializers(
-    {List<Type> typeFilter, InitializerFilter customFilter}) {
-  return new InitializationCrawler(typeFilter, customFilter).run();
+    {List<Type> typeFilter, InitializerFilter customFilter, Uri from}) {
+  return new InitializationCrawler(typeFilter, customFilter, from: from).run();
 }
 
 // Crawls a library and all its dependencies for `Initializer` annotations using
 // mirrors
 class InitializationCrawler {
   // Set of all visited annotations, keys are the declarations that were
   // annotated, values are the annotations that have been processed.
   static final _annotationsFound =
       new Map<DeclarationMirror, Set<InstanceMirror>>();
 
   // If non-null, then only these annotations should be processed.
   final List<Type> typeFilter;
 
   // If non-null, then only annotations which return true when passed to this
   // function will be processed.
   final InitializerFilter customFilter;
 
-  InitializationCrawler(this.typeFilter, this.customFilter);
+  /// The library to start crawling from.
+  final LibraryMirror _rootLibrary;
+
+  /// Note: The [from] argument is only supported in the mirror_loader.dart. It
+  /// is not supported statically.
+  InitializationCrawler(this.typeFilter, this.customFilter, {Uri from})
+      : _rootLibrary = from == null
+          ? _root
+          : currentMirrorSystem().libraries[from] {
+    if (_rootLibrary == null) throw 'Unable to find library at $from.';
+  }
 
   // The primary function in this class, invoke it to crawl and collect all the
   // annotations into a queue of init functions.
   Queue<Function> run() {
     var librariesSeen = new Set<LibraryMirror>();
     var queue = new Queue<Function>();
     var libraries = currentMirrorSystem().libraries;
 
-    var trampolineUri = Uri.parse('${_root.uri}\$trampoline');
-    if (libraries.containsKey(trampolineUri)) {
-      // In dartium, process all relative libraries in reverse order of when
-      // they were seen.
-      // TODO(jakemac): This is an approximation of what we actually want.
-      // https://github.com/dart-lang/initialize/issues/25
-      var relativeLibraryUris = new List.from(libraries.keys
-          .where((uri) => uri.scheme != 'package' && uri.scheme != 'dart'));
-
-      for (var import in relativeLibraryUris.reversed) {
-        // Always load the package: version of a library if available for
-        // canonicalization purposes.
-        var libToRun;
-        if (_isHttpStylePackageUrl(import)) {
-          var packageUri = _packageUriFor(import);
-          libToRun = libraries[packageUri];
-        }
-        if (libToRun == null) libToRun = libraries[import];
-
-        // Dartium creates an extra trampoline lib that loads the main dart script
-        // and breaks our ordering, we should skip it.
-        if (librariesSeen.contains(libToRun) ||
-            libToRun.uri.path.endsWith('\$trampoline')) {
-          continue;
-        }
-        _readLibraryDeclarations(libToRun, librariesSeen, queue);
-      }
-    } else {
-      // Not in dartium, just process from the root library.
-      _readLibraryDeclarations(_root, librariesSeen, queue);
-    }
-
+    _readLibraryDeclarations(_rootLibrary, librariesSeen, queue);
     return queue;
   }
 
   /// Whether [uri] is an http URI that contains a 'packages' segment, and
   /// therefore could be converted into a 'package:' URI.
   bool _isHttpStylePackageUrl(Uri uri) {
     var uriPath = uri.path;
     return uri.scheme == _root.uri.scheme &&
         // Don't process cross-domain uris.
         uri.authority == _root.uri.authority &&
@@ skipping 155 matching lines @@
     return true;
   }
 }
 
 final _TOP_LEVEL_FUNCTIONS_ONLY = new UnsupportedError(
     'Only top level methods are supported for initializers');
 
 final _UNSUPPORTED_DECLARATION = new UnsupportedError(
     'Initializers are only supported on libraries, classes, and top level '
     'methods');