26 lines
1.1 KiB
TypeScript
26 lines
1.1 KiB
TypeScript
import type { Text } from "@codemirror/state";
|
|
import type { DatabaseType } from "@/types/database";
|
|
import { executableStatementRanges, type SqlTextRange } from "@/lib/sqlStatementRanges";
|
|
|
|
export interface ExecutableStatementRangeCache {
|
|
doc: Text;
|
|
databaseType?: DatabaseType;
|
|
byStart: Map<number, SqlTextRange>;
|
|
}
|
|
|
|
export type ExecutableStatementRangeParser = (sql: string, databaseType?: DatabaseType) => SqlTextRange[];
|
|
|
|
export function executableStatementRangeCacheForDoc(cache: ExecutableStatementRangeCache | null, doc: Text, databaseType?: DatabaseType, parse: ExecutableStatementRangeParser = executableStatementRanges): ExecutableStatementRangeCache {
|
|
if (cache?.doc === doc && cache.databaseType === databaseType) return cache;
|
|
|
|
const byStart = new Map<number, SqlTextRange>();
|
|
for (const range of parse(doc.toString(), databaseType)) {
|
|
byStart.set(range.from, range);
|
|
}
|
|
return { doc, databaseType, byStart };
|
|
}
|
|
|
|
export function executableStatementRangeStartingAt(cache: ExecutableStatementRangeCache, lineFrom: number): SqlTextRange | null {
|
|
return cache.byStart.get(lineFrom) ?? null;
|
|
}
|