Use library URI (not name) to look up annotation

Fixes #100

Author: kevmoo

CHANGELOG.md

@@ -1,3 +1,7 @@
+## 0.5.2-dev
+
+* Use library URIs (not names) to look up annotations in the mirror system.
+
 ## 0.5.1+7
 
 * Generate valid strong-mode code for typed lists.

lib/src/annotation.dart

@@ -192,9 +192,22 @@ DeclarationMirror _getDeclarationMirrorFromType(InterfaceType type) {
   var system = currentMirrorSystem();
 
   // find library
-  var libElement = type.element.library;
-  var libNameSymbol = new Symbol(libElement.name);
-  var libMirror = system.findLibrary(libNameSymbol);
+  var libraryUri = type.element.library.librarySource.uri;
+
+  if (libraryUri.scheme == 'asset') {
+    // see if it looks like a package
+    var segs = libraryUri.pathSegments.toList();
+    if (segs.length >= 2 && segs[1] == 'lib') {
+      segs.removeAt(1);
+      libraryUri = new Uri(scheme: 'package', pathSegments: segs);
+    } else {
+      // TODO: we should support this. Just would take some time.
+      throw new UnsupportedError('Generator annotations must be defined within '
+          'a package lib directory.');
+    }
+  }
+
+  var libMirror = system.libraries[libraryUri];
 
   // find class symbol
   var typeNameSymbol = new Symbol(type.name);

pubspec.yaml

@@ -1,5 +1,5 @@
 name: source_gen
-version: 0.5.1+7
+version: 0.5.2-dev
 author: Dart Team <[email protected]>
 description: Automatic sourcecode generation for Dart
 homepage: https://github.com/dart-lang/source_gen