Fix AST generation for case statements (#8908)

Fixes #7534
2018-09-07 21:07:06 +02:00 · 2018-09-07 21:07:06 +02:00 · 91b37311d9
commit 91b37311d9
parent eae3c305a7
2 changed files with 12 additions and 1 deletions
--- a/compiler/transf.nim
+++ b/compiler/transf.nim
@ -624,7 +624,11 @@ proc transformCase(c: PTransf, n: PNode): PTransNode =
    case it.kind
    of nkElifBranch:
      if ifs.PNode == nil:
-        ifs = newTransNode(nkIfStmt, it.info, 0)
+        # Generate the right node depending on whether `n` is used as a stmt or
+        # as an expr
+        let kind = if n.typ != nil: nkIfExpr else: nkIfStmt
+        ifs = newTransNode(kind, it.info, 0)
+        ifs.PNode.typ = n.typ
      ifs.add(e)
    of nkElse:
      if ifs.PNode == nil: result.add(e)