template and templateUrl can be set as injectable functions

Author: artch

README.md

@@ -179,8 +179,8 @@ Adds new segment at current pointer level.
 
     Segment's parameters hash. The following params are supported:
     
-    - `template` provides HTML for the given segment view; 
-    - `templateUrl` is a template which should be fetched from the network via this URL; if neither `template` nor `templateUrl` parameters are defined, the DOM element's transcluded content will be used;
+    - `template` provides HTML for the given segment view; if `template` is a function, it will be called with injectable arguments;
+    - `templateUrl` is a template which should be fetched from the network via this URL; if `templateUrl` is a function, it will be called with injectable arguments; if neither `template` nor `templateUrl` parameters are defined, the DOM element's transcluded content will be used;
     - `controller` is attached to the given segment view when compiled and linked, this can be any controller definition AngularJS supports;
     - `controllerAs` is a controller alias name, if present the controller will be published to scope under the
     controllerAs name;

src/route-segment.js

@@ -359,15 +359,25 @@ mod.provider( '$routeSegment',
                 locals[key] = angular.isString(value) ? $injector.get(value) : $injector.invoke(value);
             });
 
-            if(params.template)
+            if(params.template) {
+
                 locals.$template = params.template;
+                if(angular.isFunction(locals.$template))
+                    locals.$template = $injector.invoke(locals.$template);
+            }
 
-            if(options.autoLoadTemplates && params.templateUrl)
-                locals.$template = 
-                    $http.get(params.templateUrl, {cache: $templateCache})
-                        .then(function(response) {                            
+            if(options.autoLoadTemplates && params.templateUrl) {
+
+                locals.$template = params.templateUrl;
+                if(angular.isFunction(locals.$template))
+                    locals.$template = $injector.invoke(locals.$template);
+
+                locals.$template =
+                    $http.get(locals.$template, {cache: $templateCache})
+                        .then(function (response) {
                             return response.data;
                         });
+            }
 
             return $q.all(locals).then(
 

test/unit/route-segment.spec.js

@@ -184,6 +184,34 @@ describe('route segment', function() {
             expect(callback.calls[0].args[1].segment.locals.$template).toEqual('TEST');
         });
 
+        it('should work with template set as a function', function () {
+
+            $routeSegmentProvider.segment('section3', {
+                template: function(injectable) { return injectable.foo }
+            });
+            $provide.value('injectable', {foo: 'bar'});
+
+            $rootScope.$broadcast('$routeChangeSuccess', {$route: {segment: 'section3'}});
+            $rootScope.$digest();
+            expect(callback.calls[0].args[1].segment.locals.$template).toEqual('bar');
+        });
+
+        it('should work with templateUrl set as a function', function () {
+
+            $routeSegmentProvider.options.autoLoadTemplates = true;
+            $routeSegmentProvider.segment('section3', {
+                templateUrl: function(injectable) { return injectable.foo }
+            });
+            $provide.value('injectable', {foo: '/abc/def'});
+
+            $httpBackend.expectGET('/abc/def').respond(200, 'TEST');
+            $rootScope.$broadcast('$routeChangeSuccess', {$route: {segment: 'section3'}});
+
+            $rootScope.$digest();
+            $httpBackend.flush();
+            expect(callback.calls[0].args[1].segment.locals.$template).toEqual('TEST');
+        });
+
         it('`startsWith` should work', function () {
             $location.path('/X-foo');