From 489300240852ee20b04b501819338e0a768cc07d Mon Sep 17 00:00:00 2001
From: yjbanov <yjbanov@google.com>
Date: Thu, 30 Jul 2015 11:24:53 -0700
Subject: [PATCH] chore(transfomer): code style in rewriter

---
 .../src/transform/reflection_remover/rewriter.dart     | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/modules/angular2/src/transform/reflection_remover/rewriter.dart b/modules/angular2/src/transform/reflection_remover/rewriter.dart
index edae3b24e4..ae8121f596 100644
--- a/modules/angular2/src/transform/reflection_remover/rewriter.dart
+++ b/modules/angular2/src/transform/reflection_remover/rewriter.dart
@@ -128,7 +128,8 @@ class _RewriterVisitor extends Object with RecursiveAstVisitor<Object> {
       var args = node.argumentList.arguments;
       int numArgs = node.argumentList.arguments.length;
       if (numArgs < 1 || numArgs > 2) {
-        logger.warning('`bootstrap` does not support $numArgs arguments. Found bootstrap${node.argumentList}. Transform may not succeed.');
+        logger.warning('`bootstrap` does not support $numArgs arguments. '
+            'Found bootstrap${node.argumentList}. Transform may not succeed.');
       }
 
       var reflectorInit = _setupAdded
@@ -136,17 +137,20 @@ class _RewriterVisitor extends Object with RecursiveAstVisitor<Object> {
           : ', () { ${_getStaticReflectorInitBlock()} }';
 
       // rewrite `bootstrap(...)` to `bootstrapStatic(...)`
-      buf.write('bootstrapStatic(');
-      buf.write(args[0]);
+      buf.write('bootstrapStatic(${args[0]}');
       if (numArgs == 1) {
+        // bootstrap args are positional, so before we pass reflectorInit code
+        // we need to pass `null` for DI bindings.
         if (reflectorInit.isNotEmpty) {
           buf.write(', null');
         }
       } else {
+        // pass DI bindings
         buf.write(', ${args[1]}');
       }
       buf.write(reflectorInit);
       buf.write(')');
+      _setupAdded = true;
     } else {
       // leave it as is
       buf.write(_rewriter._code.substring(_currentIndex, node.end));