clover's git

the-knife

1	import * as Data from "effect/Data";
2	import * as Effect from "effect/Effect";
3	import { Atom } from "effect/reactivity";
4
5	import {
6	highlightSourceFile,
7	type ReviewDiffTheme,
8	type ReviewHighlightedToken,
9	} from "../review/shikiReviewHighlighter";
10
11	const SOURCE_HIGHLIGHT_IDLE_TTL_MS = 5 * 60_000;
12
13	export interface SourceHighlightInput {
14	readonly path: string;
15	readonly contents: string;
16	readonly theme: ReviewDiffTheme;
17	}
18
19	export type SourceHighlightTokens = ReadonlyArray<ReadonlyArray<ReviewHighlightedToken>>;
20
21	type SourceHighlighter = (input: SourceHighlightInput) => Promise<SourceHighlightTokens>;
22
23	class SourceHighlightCacheKey extends Data.Class<SourceHighlightInput> {}
24
25	class SourceHighlightError extends Data.TaggedError("SourceHighlightError")<{
26	readonly cause: unknown;
27	}> {}
28
29	export function createSourceHighlightAtomFamily(options?: {
30	readonly highlight?: SourceHighlighter;
31	readonly idleTtlMs?: number;
32	}) {
33	const highlight = options?.highlight ?? highlightSourceFile;
34	const idleTtlMs = options?.idleTtlMs ?? SOURCE_HIGHLIGHT_IDLE_TTL_MS;
35	const family = Atom.family((request: SourceHighlightCacheKey) =>
36	Atom.make(
37	Effect.tryPromise({
38	try: () => highlight(request),
39	catch: (cause) => new SourceHighlightError({ cause }),
40	}),
41	).pipe(
42	Atom.setIdleTTL(idleTtlMs),
43	Atom.withLabel(`mobile:source-highlight:${request.theme}:${request.path}`),
44	),
45	);
46
47	return (input: SourceHighlightInput) => family(new SourceHighlightCacheKey(input));
48	}
49
50	export const sourceHighlightAtom = createSourceHighlightAtomFamily();