Function setupCompletion

packages/lsp/src/server/completion.ts:46–103

(connection: Connection, documents: TextDocuments<AlphaTexTextDocument>)

];

export function setupCompletion(connection: Connection, documents: TextDocuments<AlphaTexTextDocument>) {
    connection.onCompletion((params: TextDocumentPositionParams): CompletionItem[] => {
        const document = documents.get(params.textDocument.uri);
        if (!document?.ast) {
            return topLevelCompletions;
        }

        const offset = document.offsetAt(params.position);

        const bar = binaryNodeSearch(document.ast.bars, offset, Number.MAX_SAFE_INTEGER);
        const barIndex = bar ? document.ast.bars.indexOf(bar) : 0;

        // no bar
        if (!bar) {
            return sortCompletions(
                createMetaDataCompletions(barIndex, undefined, offset, Number.MAX_SAFE_INTEGER),
                connection.console
            );
        }

        const endOfBar =
            bar.pipe?.start?.offset ??
            (barIndex === document.ast.bars.length - 1
                ? Number.MAX_SAFE_INTEGER
                : document.ast.bars[barIndex + 1].start!.offset);

        const endOfBarMetaData = bar.beats[0]?.start?.offset ?? endOfBar;
        const metaData = bar ? binaryNodeSearch(bar.metaData, offset, endOfBarMetaData) : undefined;
        if (metaData) {
            const metaDataIndex = bar.metaData.indexOf(metaData);
            const endOfMetaData =
                metaDataIndex === bar.metaData.length - 1
                    ? endOfBarMetaData
                    : bar.metaData[metaDataIndex + 1].start!.offset;
            return sortCompletions(
                createMetaDataCompletions(barIndex, metaData, offset, endOfMetaData),
                connection.console
            );
        }

        const beat = binaryNodeSearch(bar.beats, offset, endOfBar);
        if (beat) {
            const beatIndex = bar.beats.indexOf(beat);
            const endOfBeat = beatIndex === bar.beats.length - 1 ? endOfBar : bar.beats[beatIndex + 1].start!.offset;

            return sortCompletions(createBeatCompletions(beat, offset, endOfBeat), connection.console);
        }

        return sortCompletions(
            createMetaDataCompletions(barIndex, undefined, offset, Number.MAX_SAFE_INTEGER),
            connection.console
        );
    });

    connection.onCompletionResolve((item: CompletionItem): CompletionItem => {
        return item;
    });
}