diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/eval/index.ts | 26 |
1 files changed, 23 insertions, 3 deletions
diff --git a/src/eval/index.ts b/src/eval/index.ts index d0d7363..efede36 100644 --- a/src/eval/index.ts +++ b/src/eval/index.ts @@ -1,5 +1,5 @@ import { identity, pipe } from 'fp-ts/function' -import { takeLeftWhile, zip } from 'fp-ts/lib/Array' +import { filter, takeLeftWhile, zip, zipWith } from 'fp-ts/lib/Array' import { chain, getOrElseW, @@ -35,9 +35,9 @@ const checkExpr = <T>( index: number, skip: ( n: index, - ) => (m: MatchGroupIndexed<T | T[]>[]) => MatchGroupIndexed<T | T[]>[] = _ => + ) => (m: MatchGroupIndexed<any>[]) => MatchGroupIndexed<any>[] = _ => identity, -): MatchGroupIndexed<T | T[]>[] => { +): MatchGroupIndexed<any>[] => { return pipe( expr, match<MatchGroupIndexed<any>[], Expr>({ @@ -111,6 +111,26 @@ const checkExpr = <T>( ) }, + Sequence: ({ exprs }) => { + const getGroups = () => { + if (exprs.length > list.length) return [] + + const indexed = <T>(ls: T[]): Array<[number, T]> => ls.map((x, i) => [i, x]) + const result = pipe( + zipWith(exprs, indexed(list), (expr, [i, val]) => checkExpr(expr, val, list, index + i)), + filter(matches => !!matches.length), + ) + if (result.length !== exprs.length) return [] + + return [group(result, index)] + }; + + return pipe( + getGroups(), + skip(exprs.length || 1), + ) + }, + _: _ => { throw new Error(`TODO: ${expr.tag} not implemented for match`) }, }), ) |
