fix(if): set elseFactory

aurelia · Mar 29, 2019 · f01af55 · f01af55
1 parent b71b0e9
commit f01af55
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/packages/runtime/src/resources/custom-attributes/if.ts b/packages/runtime/src/resources/custom-attributes/if.ts
@@ -293,7 +293,13 @@ export class Else<T extends INode = INode> {
     container.register(Registration.transient('custom-attribute:else', this));
   }
 
-  public link(ifBehavior: If<T>): void {
-    ifBehavior.elseFactory = this.factory;
+  public link(ifBehavior: If<T> | IController<T>): void {
+    if (ifBehavior instanceof If) {
+      ifBehavior.elseFactory = this.factory;
+    } else if (ifBehavior.viewModel instanceof If) {
+      ifBehavior.viewModel.elseFactory = this.factory;
+    } else {
+      throw new Error(`Unsupported IfBehavior`); // TODO: create error code
+    }
   }
 }