feat(router): implement configuration customization

aurelia · Aug 14, 2019 · c7f6fa5 · c7f6fa5
1 parent 1e8b61d
commit c7f6fa5
Showing 1 changed file with 17 additions and 2 deletions.
diff --git a/packages/router/src/configuration.ts b/packages/router/src/configuration.ts
@@ -32,6 +32,11 @@ export const DefaultResources: IRegistry[] = [
   NavCustomElement as unknown as IRegistry,
 ];
 
+let configurationOptions: IRouterOptions = {};
+let configurationCall: ((router: IRouter) => void) = (router: IRouter) => {
+  router.activate(configurationOptions);
+};
+
 /**
  * A DI configuration object containing router resource registrations.
  */
@@ -43,7 +48,7 @@ const routerConfiguration = {
     return container.register(
       ...DefaultComponents,
       ...DefaultResources,
-      StartTask.with(IRouter).beforeBind().call(router => router.activate()),
+      StartTask.with(IRouter).beforeBind().call(configurationCall),
       StartTask.with(IRouter).beforeAttach().call(router => router.loadUrl()),
     );
   },
@@ -58,7 +63,17 @@ export const RouterConfiguration = {
   /**
    * Make it possible to specify options to Router activation
    */
-  customize(config: IRouterOptions = {}) {
+  customize(config?: IRouterOptions | (() => void)) {
+    if (config === undefined) {
+      configurationOptions = {};
+      configurationCall = (router: IRouter) => {
+        router.activate(configurationOptions);
+      };
+    } else if (config instanceof Function) {
+      configurationCall = config;
+    } else {
+      configurationOptions = config;
+    }
     return { ...routerConfiguration };
   },
   ...routerConfiguration,