support variable declarations without expressions

Angry-Potato · Jul 18, 2019 · 9962c2f · 9962c2f
1 parent d9ebd19
commit 9962c2f
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 11 deletions.
diff --git a/src/nodes.js b/src/nodes.js
@@ -59,8 +59,7 @@ const isVariableDeclaration = node =>
   node.mods &&
   node.typeParams &&
   node.vars &&
-  node.typeConstraints &&
-  node.expr;
+  node.typeConstraints;
 
 const isName = node => node.hasOwnProperty("name");
 

diff --git a/src/nodes/variable-declaration.js b/src/nodes/variable-declaration.js
@@ -18,12 +18,9 @@ module.exports = (path, opts, print) => {
       ? concat([...path.map(print, "mods"), " "])
       : "";
 
-  return concat([
-    prefix,
-    varType,
-    ...path.map(print, "vars"),
-    " = ",
-    path.call(print, "expr"),
-    hardline
-  ]);
+  const expr = node.expr
+    ? concat([" = ", path.call(print, "expr"), hardline])
+    : "";
+
+  return concat([prefix, varType, ...path.map(print, "vars"), expr]);
 };
diff --git a/src/nodes/variable-declaration.spec.js b/src/nodes/variable-declaration.spec.js
@@ -23,5 +23,6 @@ describe("variable-declaration", () => {
 
   test("standard declarations", () =>
     expect(`val a = 1
-var b = 1`).toMatchFormat());
+var b = 1
+val c: Int`).toMatchFormat());
 });