99 lines
2.4 KiB
TypeScript
99 lines
2.4 KiB
TypeScript
import { Pos, startPos } from './position.js';
|
|
|
|
export enum TokenType {
|
|
SPACE,
|
|
NEWLINE,
|
|
ATOM,
|
|
STRING,
|
|
OPEN,
|
|
CLOSE,
|
|
}
|
|
|
|
export interface Token {
|
|
type: TokenType;
|
|
start: Pos;
|
|
end: Pos;
|
|
text: string;
|
|
}
|
|
|
|
export interface Group {
|
|
start: Token;
|
|
end: Token | null;
|
|
items: Items;
|
|
}
|
|
|
|
export type Item = Token | Group;
|
|
export type Items = Array<Item>;
|
|
|
|
export function makeToken(text: string, name?: string | null, type: TokenType = TokenType.ATOM): Token {
|
|
const p = startPos(name ?? null);
|
|
return {
|
|
start: p,
|
|
end: p,
|
|
type,
|
|
text
|
|
};
|
|
}
|
|
|
|
export function makeGroup(start: Token, items: Array<Items>, end?: Token) {
|
|
return { start, end: end ?? null, items };
|
|
}
|
|
|
|
export function isSpace(i: Item): i is Token {
|
|
return isTokenType(i, TokenType.SPACE) || isTokenType(i, TokenType.NEWLINE);
|
|
}
|
|
|
|
export function isGroup(i: Item): i is Group {
|
|
return i && ('items' in i);
|
|
}
|
|
|
|
export function isToken(i: Item): i is Token {
|
|
return i && ('type' in i);
|
|
}
|
|
|
|
export function isTokenType(i: Item, t: TokenType): i is Token {
|
|
return isToken(i) && i.type === t;
|
|
}
|
|
|
|
export type ItemTextOptions = {
|
|
missing?: string,
|
|
color?: boolean,
|
|
};
|
|
|
|
export function foldItems<T>(i: Items,
|
|
fToken: (t: Token) => T,
|
|
fGroup: (start: Token, end: Token | null, t: T, k: (t: Token) => T) => T,
|
|
fItems: (ts: T[]) => T): T
|
|
{
|
|
const walk = (i: Item): T => {
|
|
if (isGroup(i)) {
|
|
return fGroup(i.start, i.end, fItems(i.items.map(walk)), walk);
|
|
} else {
|
|
return fToken(i);
|
|
}
|
|
};
|
|
return fItems(i.map(walk));
|
|
}
|
|
|
|
export function itemText(items: Items, options: ItemTextOptions = {}): string {
|
|
return foldItems(
|
|
items,
|
|
i => {
|
|
if (options.color ?? false) {
|
|
switch (i.type) {
|
|
case TokenType.SPACE:
|
|
case TokenType.NEWLINE:
|
|
return '\x1b[31m' + i.text + '\x1b[0m';
|
|
case TokenType.STRING:
|
|
return '\x1b[34m' + i.text + '\x1b[0m';
|
|
default:
|
|
return i.text;
|
|
}
|
|
} else {
|
|
return i.text;
|
|
}
|
|
},
|
|
(start, end, inner, k) => k(start) + inner + (end ? k(end) : options.missing ?? ''),
|
|
strs => strs.join(''));
|
|
}
|