Function getChunks

src/rmarkdown/chunks.ts:81–144

(document: vscode.TextDocument)

// Scan document and return chunk info (e.g. ID, chunk range) from all chunks
export function getChunks(document: vscode.TextDocument): RMarkdownChunk[] {
    const lines = document.getText().split(/\r?\n/);
    const chunks: RMarkdownChunk[] = [];

    let line = 0;
    let chunkId = 0;  // One-based index
    let chunkStartLine: number | undefined = undefined;
    let chunkEndLine: number | undefined = undefined;
    let codeEndLine: number | undefined = undefined;
    let chunkLanguage: string | undefined = undefined;
    let chunkOptions: string | undefined = undefined;
    let chunkEval: boolean | undefined = undefined;
    const isRDoc = isRDocument(document);

    while (line < lines.length) {
        if (chunkStartLine === undefined) {
            if (isChunkStartLine(lines[line], isRDoc)) {
                chunkId++;
                chunkStartLine = line;
                chunkLanguage = getChunkLanguage(lines[line], isRDoc);
                chunkOptions = getChunkOptions(lines[line], isRDoc);
                chunkEval = getChunkEval(chunkOptions);
            }
        } else {
            // Second condition is for the last chunk in an .R file
            const isRDocAndFinalLine = (isRDoc && line === lines.length - 1);
            if (isChunkEndLine(lines[line], isRDoc) || isRDocAndFinalLine) {
                chunkEndLine = line;
                codeEndLine = line - 1;
                
                // isChunkEndLine looks for `# %%` in `.R` files, so if found, then need to go back one line to mark end of code chunk. 
                if (isRDoc && !isRDocAndFinalLine) {
                    chunkEndLine = chunkEndLine - 1;
                    codeEndLine = chunkEndLine;
                    line = line - 1;
                }
                
                const chunkRange = new vscode.Range(
                    new vscode.Position(chunkStartLine, 0),
                    new vscode.Position(line, lines[line].length)
                );
                const codeRange = new vscode.Range(
                    new vscode.Position(chunkStartLine + 1, 0),
                    new vscode.Position(codeEndLine, lines[codeEndLine].length)
                );

                chunks.push({
                    id: chunkId, // One-based index
                    startLine: chunkStartLine,
                    endLine: chunkEndLine,
                    language: chunkLanguage,
                    options: chunkOptions,
                    eval: chunkEval,
                    chunkRange: chunkRange,
                    codeRange: codeRange
                });

                chunkStartLine = undefined;