From 1c5b21022c4b54d0497f9a0f270a71aca839f1d3 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Andr=C3=A9=20Lichtenth=C3=A4ler?=
 <andre.lichtenthaeler@gmail.com>
Date: Mon, 25 Jul 2022 11:22:41 +0200
Subject: [PATCH] feat: add arrowFunctionExpression (wip)

#119
---
 src/parser/JavaScript.ts | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/src/parser/JavaScript.ts b/src/parser/JavaScript.ts
index 7e12f1c..e73940d 100644
--- a/src/parser/JavaScript.ts
+++ b/src/parser/JavaScript.ts
@@ -6,11 +6,15 @@ import {
   optional,
   whitespace,
   between,
+  lazy,
 } from "rudus";
 import { NodeType } from "../enums";
 import { IParser } from "../Interfaces";
 
 const betweenDoubleQuotes = between(string('"'));
+const betweenParanthesis = between(string("("), string(")"));
+const betweenBraces = between(string("{"), string("}"));
+
 const optionalWhitespace = optional(whitespace()).map(state => [
   NodeType.Whitespace,
   state.result,
@@ -24,6 +28,7 @@ const optionalSemicolon = optional(string(";")).map(state => [
 const declarationKeyword = regex(/^var|let|const/);
 const word = regex(/\w+/); // word
 const equalSign = string("=");
+const bigArrow = string("=>");
 
 const literalBool = regex(/^true|false$/);
 const literalUndefined = regex(/undefined/);
@@ -31,7 +36,18 @@ const literalNull = regex(/null/);
 const literalNumber = regex(/^(\d+\.)?\d+$|^0b[0-1]+(n)?$/);
 const literalString = betweenDoubleQuotes(word);
 
+const functionBody = lazy(() => literalNumber); // for testing purposes
+
+const arrowFunctionExpression = sequenceOf([
+  betweenParanthesis(optional(word)),
+  optionalWhitespace,
+  bigArrow,
+  optionalWhitespace,
+  betweenBraces(optional(functionBody)),
+]);
+
 const variableValue = anyOf([
+  arrowFunctionExpression.map(state => [NodeType.CallExpression, state.result]),
   literalBool.map(state => [NodeType.BooleanLiteral, state.result]),
   literalString.map(state => [NodeType.StringLiteral, state.result]),
   literalNumber.map(state => [NodeType.NumberLiteral, state.result]),