From 26618657fd64dad4293bad888d19ad863c838d71 Mon Sep 17 00:00:00 2001
From: Matan Lurey <matan@lurey.org>
Date: Mon, 26 Jun 2017 18:02:48 -0700
Subject: [PATCH] Patch internal cases, bump.

---
 CHANGELOG.md         |  6 ++++++
 lib/src/builder.dart | 17 +++++++++--------
 pubspec.yaml         |  2 +-
 3 files changed, 16 insertions(+), 9 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 5657c6e6..d8ff66fc 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -1,3 +1,9 @@
+## 0.5.10
+
+* Bug fixes:
+  * Do not fail when "library" is omitted but nothing would be output.
+  * Do not fail on extensions that don't end in ".dart" (valid use case).
+
 ## 0.5.9
 
 * Update the minimum Dart SDK to `1.22.1`.
diff --git a/lib/src/builder.dart b/lib/src/builder.dart
index 42ab5591..47cad45e 100644
--- a/lib/src/builder.dart
+++ b/lib/src/builder.dart
@@ -60,11 +60,9 @@ class GeneratorBuilder extends Builder {
     if (generatedExtension == null) {
       throw new ArgumentError.notNull('generatedExtension');
     }
-    if (generatedExtension.isEmpty ||
-        !generatedExtension.startsWith('.') ||
-        !generatedExtension.endsWith('.dart')) {
+    if (generatedExtension.isEmpty || !generatedExtension.startsWith('.')) {
       throw new ArgumentError.value(generatedExtension, 'generatedExtension',
-          'Extension must be in the format of .dart or .*.dart');
+          'Extension must be in the format of .*');
     }
     if (this.isStandalone && this.generators.length > 1) {
       throw new ArgumentError(
@@ -94,8 +92,14 @@ class GeneratorBuilder extends Builder {
     var generatedOutputs =
         await _generate(library, generators, buildStep).toList();
 
-    var contentBuffer = new StringBuffer();
+    // Don't output useless files.
+    //
+    // NOTE: It is important to do this check _before_ checking for valid
+    // library/part definitions because users expect some files to be skipped
+    // therefore they do not have "library".
+    if (generatedOutputs.isEmpty) return;
 
+    var contentBuffer = new StringBuffer();
     if (!isStandalone) {
       var asset = buildStep.inputId;
       var name = nameOfPartial(
@@ -115,9 +119,6 @@ class GeneratorBuilder extends Builder {
       contentBuffer.writeln();
     }
 
-    // Don't output useless files.
-    if (generatedOutputs.isEmpty) return;
-
     for (GeneratedOutput output in generatedOutputs) {
       contentBuffer.writeln('');
       contentBuffer.writeln(_headerLine);
diff --git a/pubspec.yaml b/pubspec.yaml
index 718051db..3079e55f 100644
--- a/pubspec.yaml
+++ b/pubspec.yaml
@@ -1,5 +1,5 @@
 name: source_gen
-version: 0.5.10-dev
+version: 0.5.10
 author: Dart Team <misc@dartlang.org>
 description: Automated source code generation for Dart.
 homepage: https://github.com/dart-lang/source_gen