From db97b1a9c42fe945ca37256f5ee18d52c6aa32b4 Mon Sep 17 00:00:00 2001
From: Akshay Nair <phenax5@gmail.com>
Date: Thu, 10 Aug 2023 18:18:46 +0530
Subject: feat: adds parser for simple expressions

---
 src/utils/adt.ts         |  6 ++---
 src/utils/parser-comb.ts | 63 ++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 65 insertions(+), 4 deletions(-)
 create mode 100644 src/utils/parser-comb.ts

(limited to 'src/utils')

diff --git a/src/utils/adt.ts b/src/utils/adt.ts
index c19398b..e9a6c51 100644
--- a/src/utils/adt.ts
+++ b/src/utils/adt.ts
@@ -1,11 +1,9 @@
 type TagValue<T, N> = T extends Tag<N, infer V> ? V : never
 
 export const match =
-  <R, T extends Tag<string, any>>(pattern: {
+  <R, T extends Tag<string, any>>(tag: T, pattern: {
     [key in T['tag'] | '_']?: (v: TagValue<T, key>) => R
-  }) =>
-  (tag: T): R =>
-    ((pattern as any)[tag.tag] || (pattern._ as any))(tag.value)
+  }): R => ((pattern as any)[tag.tag] || (pattern._ as any))(tag.value)
 
 type Tag<N, V> = { tag: N; value: V }
 export type Enum<T> = { [N in keyof T]: Tag<N, T[N]> }[keyof T]
diff --git a/src/utils/parser-comb.ts b/src/utils/parser-comb.ts
new file mode 100644
index 0000000..9cea86f
--- /dev/null
+++ b/src/utils/parser-comb.ts
@@ -0,0 +1,63 @@
+import { Enum, constructors, match } from './adt';
+
+export type Result<V, E> = Enum<{ Ok: V, Err: E }>
+export const Result = constructors<Result<any, any>>()
+
+export const mapResult = <A, B, E>(res: Result<A, E>, fn: (_: A) => B): Result<B, E> =>
+  chainResult(res, a => Result.Ok(fn(a)))
+
+export const chainResult = <A, B, E>(res: Result<A, E>, fn: (_: A) => Result<B, E>): Result<B, E> =>
+  match(res, {
+    Ok: a => fn(a),
+    Err: e => Result.Err(e),
+  });
+
+export type ParseResult<T> = Result<{ value: T, input: string }, { error: string, input: string }>;
+
+export type Parser<T> = (input: string) => ParseResult<T>;
+
+export const regex = (re: RegExp): Parser<string> => input => {
+  const res = input.match(re)
+  if (!res) return Result.Err({ error: 'fucked', input });
+  return Result.Ok({ value: res[0], input: input.replace(re, '') });
+}
+
+export const string = (str: string): Parser<string> => input => {
+  if (!input.startsWith(str)) return Result.Err({ error: 'fuckedstring', input })
+  return Result.Ok({ value: str, input: input.slice(str.length) });
+}
+
+export const or = <T>([parser, ...rest]: Array<Parser<T>>): Parser<T> => input => {
+  if (rest.length === 0) return parser(input);
+  const result = parser(input)
+  return match(result, {
+    Ok: () => result,
+    Err: (_) => or(rest)(input),
+  });
+}
+
+export const mapParseResult = <T, R>(parser: Parser<T>, fn: (_: { value: T, input: string }) => { value: R, input: string }): Parser<R> => input =>
+  mapResult(parser(input), fn)
+
+export const map = <T, R>(parser: Parser<T>, fn: (_: T) => R): Parser<R> =>
+  mapParseResult(parser, ({ value, ...rest }) => ({ ...rest, value: fn(value) }));
+
+export const zip2 = <A, B>(parserA: Parser<A>, parserB: Parser<B>): Parser<readonly [A, B]> => input => {
+  // TODO: refactor please. shit code
+  const resa: Result<{ value: A, input: string }, { error: string, input: string }> = parserA(input);
+  return chainResult(resa, ({ value: a, input: inputB }) => {
+    const res: Result<{ value: readonly [A, B], input: string }, { error: string, input: string }> =
+      map(parserB, (b) => [a, b] as const)(inputB)
+    return res
+  })
+}
+
+export const prefixed = <A>(parserPrefix: Parser<any>, parser: Parser<A>): Parser<A> =>
+  map(zip2(parserPrefix, parser), ([_, a]) => a);
+
+export const suffixed = <A>(parser: Parser<A>, parserSuffix: Parser<any>): Parser<A> =>
+  map(zip2(parser, parserSuffix), ([a, _]) => a);
+
+export const between = <A>(prefix: Parser<any>, parser: Parser<A>, suffix: Parser<any>): Parser<A> =>
+  suffixed(prefixed(prefix, parser), suffix)
+
-- 
cgit v1.3.1