follow references on component properties

also add support for providing a template/templateUrl via an injectable
function

fixes #7

Author: Andrew Schmadel

ng-annotate-main.js

@@ -390,12 +390,8 @@ function matchRegular(path, ctx) {
         args.length === 2 && t.isLiteral(args[0]) && is.string(args[0].value) && argPaths[1]);
 
     if (method.name === "component") {
-        const controllers = target.get("properties").filter(prop => prop.node.key.name == "controller");
-        if(controllers.length === 1) {
-            target = controllers[0].get("value");
-        } else {
-            return false;
-        }
+        target.node.$chained = chainedRegular;
+        return matchComponent(target);
     }
 
     if (target) {
@@ -453,6 +449,34 @@ function matchResolve(props) {
     return [];
 }
 
+function matchComponent(path){
+    let chained = path.node.$chained;
+    if(t.isIdentifier(path)) {
+        path = followReference(path);
+        if(t.isVariableDeclarator(path)){
+            path = path.get('init');
+        }
+    }
+    if(t.isObjectExpression(path)){
+        path.node.chained = chained;
+        const props = path.get("properties");
+
+        const ctrl = matchProp("controller", props);
+        const tmpl =  matchProp("template", props);
+        const tmplUrl =  matchProp("templateUrl", props);
+
+        let res = [];
+        ctrl && res.push(ctrl);
+        tmpl && res.push(tmpl);
+        tmplUrl && res.push(tmplUrl);
+
+        res.forEach(t => t.node.$chained = chained);
+        return res;
+    } else {
+        return false;
+    }
+}
+
 function renamedString(ctx, originalString) {
     if (ctx.rename) {
         return ctx.rename.get(originalString) || originalString;

tests/references.js

@@ -180,6 +180,35 @@ module.exports = {
             });
         }
       }
+    },
+    {
+        name: "chained component controller",
+        implicit: true,
+        input: function() {
+            var tmplProvider = function(bar){};
+            var testFeedback = {
+                controller: testFeedbackController,
+                template: tmplProvider
+            };
+            function testFeedbackController(foo) {}
+
+            angular.module("test.feedback.pkg", [])
+                .component("testFeedback", testFeedback);
+        },
+        expected: function() {
+            testFeedbackController.$inject = ["foo"];
+            var tmplProvider = function(bar){};
+            tmplProvider.$inject = ["bar"];
+            var testFeedback = {
+                controller: testFeedbackController,
+                template: tmplProvider
+            };
+            function testFeedbackController(foo) {
+            }
+
+            angular.module("test.feedback.pkg", [])
+                .component("testFeedback", testFeedback);
+        }
     }
   ]
 }

tests/simple.js

@@ -563,6 +563,32 @@ module.exports = {
       })();
       angular.module("MyMod").controller("MyCtrl", myCtrl10);
     }
+  },
+  {
+    name: "injectable component templates/controller/templateurl",
+    implicit: true,
+    input: function(){
+      angular.module("mod").component("cmp", {
+        controller: function(a){},
+        template: function(b){},
+        templateUrl: function(c){},
+      }).component("cmp2", {
+        controller: "myCtrl",
+        template: "tmpl",
+        templateUrl: "template.html"
+      });
+    },
+    expected: function(){
+      angular.module("mod").component("cmp", {
+        controller: ["a", function(a){}],
+        template: ["b", function(b){}],
+        templateUrl: ["c", function(c){}],
+      }).component("cmp2", {
+        controller: "myCtrl",
+        template: "tmpl",
+        templateUrl: "template.html"
+      });
+    }
   }
  ]
 };