Support else if (...)

dandrei · dandrei · commit 457450541a1e · 2017-12-15T17:21:29.000+02:00
diff --git a/src/__tests__/if-then-else-spec.js b/src/__tests__/if-then-else-spec.js
@@ -55,3 +55,16 @@ test("ternary", t =>
   export function test(): i32 {
     return 1 ? 42 : 24;
   }`).then(outputIs(t, 42)));
+
+test("else if statement", t =>
+  compileAndRun(`
+  export function test(x: i32): i32 {
+   if (x == 0) {
+      return 2;
+    } else if (x == 1) {
+       return 4;
+    } else {
+      return 1;
+    }
+    test(1);
+  }`).then(outputIs(t, 4)));
diff --git a/src/parser/if-then-else.js b/src/parser/if-then-else.js
@@ -26,7 +26,12 @@ const ifThenElse = (ctx: Context) => {
 
     ctx.expect(["}"]);
 
-    if (ctx.eat(["else"])) {
+    while (ctx.eat(["else"])) {
+      if (ctx.eat(["if"])) {
+        ctx.expect(["("]);
+        node.expr = expression(ctx, "i32");
+        ctx.expect([")"]);
+      }
       ctx.expect(["{"]);
       while (ctx.token && ctx.token.value !== "}") {
         stmt = statement(ctx);
