added a shorthand syntax for function one liner

TomTriple · Dec 6, 2012 · 8718cb4 · 8718cb4
1 parent d26c13e
commit 8718cb4
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/Ex1Test.g b/Ex1Test.g
@@ -8,6 +8,7 @@ options {
 tokens {
   FN_CALL;
   FN_DEF;
+  FN_DEF_INLINE;
   EX;
 }
 
@@ -20,6 +21,7 @@ prog: exprStmt* EOF -> exprStmt*
 
 exprStmt: ID ':' expr ';' -> ^(':' ID expr)
         | name=ID ':' '{' (arg+=ID)? (',' arg+=ID)* '=>' body+=exprStmt* '}' ';' -> ^(FN_DEF $name $arg* $body*)
+        | name=ID ':' '=>' inline=exprStmt -> ^(FN_DEF_INLINE $name $inline)
         | expr ';' -> ^(EX expr)
         ;
 

diff --git a/Ex1Walker.g b/Ex1Walker.g
@@ -410,6 +410,10 @@ exprStmt returns [node]
           this.popScope();
         }
         | ^(EX expr) { $node = $expr.node; }
+        | ^(FN_DEF_INLINE name=ID {var node = new L.FunctionDef(this.currentScope, $name.text, this.pushScope()); } (e=exprStmt { node.body.push($e.node); }) ) {
+          $node = node;
+          this.popScope();
+        }
         ;
 
 expr returns [node]

diff --git a/index.html b/index.html
@@ -29,8 +29,10 @@
   };
   fnResult: someFn(77, 9);
   assert(fnResult(10) == 200);
-
-
+
+  voidFn: => puts(777);
+  voidFn();
+
 </script>