From 952256ae8bf5514a246db8f33bef207f5747b138 Mon Sep 17 00:00:00 2001
From: Akshay Nair <phenax5@gmail.com>
Date: Fri, 11 Aug 2023 17:13:57 +0530
Subject: feat: adds parser for literal numbers

---
 src/parser.ts | 11 +++++++++++
 1 file changed, 11 insertions(+)

(limited to 'src/parser.ts')

diff --git a/src/parser.ts b/src/parser.ts
index 1c5c08f..8230825 100644
--- a/src/parser.ts
+++ b/src/parser.ts
@@ -1,11 +1,14 @@
 import { Enum, constructors, match } from './utils/adt'
 import * as P from './utils/parser-comb'
 
+type Unit = '' | 's' | 'ms'
+
 export type Expr = Enum<{
   Call: { name: string; args: Expr[] }
   Identifier: string
   VarIdentifier: string
   LiteralString: string
+  LiteralNumber: { value: number, unit: Unit }
 }>
 
 export const Expr = constructors<Expr>()
@@ -41,9 +44,17 @@ const stringLiteralParser: P.Parser<Expr> = P.map(
   Expr.LiteralString
 )
 
+const numberParser = P.regex(/^[-+]?((\d*\.\d+)|\d+)/)
+
+const numberExprParser: P.Parser<Expr> = P.map(
+  P.zip2(numberParser, P.optional(P.regex(/^(s|ms)/i))),
+  ([value, unit]) => Expr.LiteralNumber({ value: Number(value), unit: (unit ?? '') as Unit }),
+)
+
 const exprParser: P.Parser<Expr> = P.or([
   stringLiteralParser,
   varIdentifierExprParser,
+  numberExprParser,
   callExprParser,
   identifierExprParser,
 ])
-- 
cgit v1.3.1