@vscode/chat-lib 0.0.1

package/dist/src/_internal/platform/notebook/common/alternativeContentProvider.text.js

@@ -0,0 +1,231 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AlternativeTextNotebookContentProvider = void 0;
+exports.generateCellTextMarker = generateCellTextMarker;
+exports.lineMightHaveCellMarker = lineMightHaveCellMarker;
+exports.getBlockComment = getBlockComment;
+exports.getLineCommentStart = getLineCommentStart;
+const languages_1 = require("../../../util/common/languages");
+const types_1 = require("../../../util/common/types");
+const vscodeTypesShim_1 = require("../../../util/common/test/shims/vscodeTypesShim");
+const alternativeContentProvider_1 = require("./alternativeContentProvider");
+const alternativeNotebookDocument_1 = require("./alternativeNotebookDocument");
+const helpers_1 = require("./helpers");
+const arraysFind_1 = require("../../../util/vs/base/common/arraysFind");
+function generateCellTextMarker(cell, lineComment) {
+    const cellIdStr = cell.id ? `[id=${cell.id}] ` : '';
+    return `${lineComment}%% vscode.cell ${cellIdStr}[language=${cell.language}]`;
+}
+function lineMightHaveCellMarker(line) {
+    return line.toLowerCase().includes('vscode.cell');
+}
+class AlternativeTextDocument extends alternativeNotebookDocument_1.AlternativeNotebookDocument {
+    constructor(text, cellOffsetMap, notebook) {
+        super(text, notebook);
+        this.cellOffsetMap = cellOffsetMap;
+    }
+    fromCellPosition(cell, position) {
+        const cellSummary = (0, helpers_1.summarize)(cell);
+        const lineCommentStart = getLineCommentStart(this.notebook);
+        const cellMarker = generateCellTextMarker(cellSummary, lineCommentStart);
+        const eolLength = cell.document.eol === vscodeTypesShim_1.EndOfLine.LF ? 1 : 2;
+        const blockComment = getBlockComment(this.notebook);
+        const alternativeContentText = this.getText();
+        const offsetInCell = cell.document.offsetAt(position);
+        const markdownOffset = cell.kind === vscodeTypesShim_1.NotebookCellKind.Markup ? blockComment[0].length + eolLength : 0;
+        const offset = alternativeContentText.indexOf(cellMarker) + cellMarker.length + eolLength + markdownOffset + offsetInCell;
+        return this.positionAt(offset);
+    }
+    toCellPosition(position) {
+        const offset = this.offsetAt(position);
+        const cell = (0, arraysFind_1.findLast)(this.cellOffsetMap, (cell) => cell.sourceOffset <= offset);
+        if (!cell) {
+            return undefined;
+        }
+        const cellPosition = cell.cell.document.positionAt(offset - cell.sourceOffset);
+        return { cell: cell.cell, position: cellPosition };
+    }
+}
+class AlternativeTextNotebookContentProvider extends alternativeContentProvider_1.BaseAlternativeNotebookContentProvider {
+    constructor() {
+        super('text');
+    }
+    stripCellMarkers(text) {
+        const lines = text.split(helpers_1.EOL);
+        if (lines.length && lineMightHaveCellMarker(lines[0])) {
+            lines.shift();
+            return lines.join(helpers_1.EOL);
+        }
+        else {
+            return text;
+        }
+    }
+    getSummaryOfStructure(notebook, cellsToInclude, existingCodeMarker) {
+        const blockComment = getBlockComment(notebook);
+        const lineCommentStart = getLineCommentStart(notebook);
+        const existingCodeMarkerWithComment = `${lineCommentStart} ${existingCodeMarker}`;
+        const lines = [];
+        notebook.getCells().forEach((cell) => {
+            if (cellsToInclude.includes(cell)) {
+                const cellSummary = (0, helpers_1.summarize)(cell);
+                if (cellSummary.source.length && cellSummary.source[0].trim().length) {
+                    cellSummary.source = [cellSummary.source[0], existingCodeMarkerWithComment];
+                }
+                else if (cellSummary.source.length && cellSummary.source.some(line => line.trim().length)) {
+                    cellSummary.source = [existingCodeMarkerWithComment, cellSummary.source.filter(line => line.trim().length)[0], existingCodeMarkerWithComment];
+                }
+                else {
+                    cellSummary.source = [existingCodeMarkerWithComment];
+                }
+                lines.push(generateAlternativeCellTextContent(cellSummary, lineCommentStart, blockComment).content);
+            }
+            else if (!lines.length || lines[lines.length - 1] !== existingCodeMarkerWithComment) {
+                lines.push(existingCodeMarkerWithComment);
+            }
+        });
+        return lines.join(helpers_1.EOL);
+    }
+    async *parseAlternateContent(notebookOrUri, inputStream, token) {
+        const isNotebook = !(0, types_1.isUri)(notebookOrUri);
+        const cellIdMap = isNotebook ? (0, helpers_1.getCellIdMap)(notebookOrUri) : new Map();
+        let inMarkdownCell = false;
+        let isInTripleQuotes = false;
+        let pendingTripleQuotes = false;
+        let emittedStart = false;
+        let cellIndex = -1;
+        const lineCommentStart = getLineCommentStart(isNotebook ? notebookOrUri : undefined);
+        const blockComment = getBlockComment(isNotebook ? notebookOrUri : undefined);
+        const defaultLanguage = isNotebook ? (0, languages_1.getLanguage)((0, helpers_1.getDefaultLanguage)(notebookOrUri)).languageId : undefined;
+        const cellIdsSeen = new Set();
+        for await (const lineOfText of inputStream) {
+            if (token.isCancellationRequested) {
+                break;
+            }
+            const line = lineOfText.value;
+            // Check for new cell delimiter
+            // Sometimes LLM returns cells without the `vscode.cell` marker such as .
+            const isLineCommentForEmptyCellWithoutCellMarker = line.startsWith(`${lineCommentStart}%% [`) && line.trimEnd().endsWith(']');
+            const isLineCommentWithCellMarker = line.startsWith(`${lineCommentStart}%% vscode.cell`);
+            // Attempt to extract only if we think we have a cell marker, else we end up doing this for every single line and thats expensive.
+            const cellParts = (isLineCommentWithCellMarker || isLineCommentForEmptyCellWithoutCellMarker) ? extractCellParts(line, defaultLanguage) : undefined;
+            if ((isLineCommentWithCellMarker || isLineCommentForEmptyCellWithoutCellMarker) && cellParts?.language) {
+                if (pendingTripleQuotes) {
+                    pendingTripleQuotes = false;
+                }
+                const lineOfCellText = { index: -1, uri: undefined, language: undefined, kind: vscodeTypesShim_1.NotebookCellKind.Code, emitted: false, type: 'start' };
+                lineOfCellText.index = cellIndex += 1;
+                lineOfCellText.emitted = false;
+                // LLM returns duplicate cell with the same id.
+                if (cellParts.id && cellIdMap.get(cellParts.id)?.document.languageId === cellParts.language) {
+                    if (cellIdsSeen.has(cellParts.id)) {
+                        cellParts.id = '';
+                    }
+                    else {
+                        cellIdsSeen.add(cellParts.id);
+                    }
+                }
+                else {
+                    // Possible duplicate cell with the same id but different language.
+                    // In such cases, treat them as new cells.
+                    cellParts.id = '';
+                }
+                const cell = cellIdMap.get(cellParts.id);
+                lineOfCellText.id = cellParts.id;
+                lineOfCellText.language = cellParts.language;
+                lineOfCellText.uri = cell?.document.uri;
+                lineOfCellText.kind = cell?.kind || (lineOfCellText.language === 'markdown' ? vscodeTypesShim_1.NotebookCellKind.Markup : vscodeTypesShim_1.NotebookCellKind.Code);
+                inMarkdownCell = lineOfCellText.language === 'markdown';
+                isInTripleQuotes = false;
+                if (emittedStart) {
+                    yield { index: cellIndex - 1, type: 'end' };
+                }
+                emittedStart = true;
+                yield lineOfCellText;
+                continue;
+            }
+            if (!emittedStart) {
+                continue;
+            }
+            if (inMarkdownCell) {
+                if (!isInTripleQuotes) {
+                    // Look for the opening triple quotes
+                    if (line === blockComment[0]) {
+                        isInTripleQuotes = true;
+                    }
+                    else {
+                        // lineEmitted = true;
+                        yield { index: cellIndex, line, type: 'line' };
+                    }
+                }
+                else {
+                    // We are in triple quotes
+                    if (line === blockComment[1]) {
+                        // Closing triple quotes found
+                        isInTripleQuotes = false;
+                        pendingTripleQuotes = true;
+                    }
+                    else {
+                        yield { index: cellIndex, line, type: 'line' };
+                    }
+                }
+            }
+            else {
+                // Non-markdown cell or default
+                yield { index: cellIndex, line, type: 'line' };
+            }
+        }
+        if (emittedStart) {
+            yield { index: cellIndex, type: 'end' };
+        }
+    }
+    getAlternativeDocument(notebook, excludeMarkdownCells) {
+        const cells = notebook.getCells().filter(cell => excludeMarkdownCells ? cell.kind !== vscodeTypesShim_1.NotebookCellKind.Markup : true).map(cell => (0, helpers_1.summarize)(cell));
+        const blockComment = getBlockComment(notebook);
+        const lineCommentStart = getLineCommentStart(notebook);
+        const cellContent = cells.map(cell => ({ ...generateAlternativeCellTextContent(cell, lineCommentStart, blockComment), cell: notebook.cellAt(cell.index) }));
+        const content = cellContent.map(cell => cell.content).join(helpers_1.EOL);
+        const cellOffsetMap = cellContent.map(cellContent => {
+            const offset = content.indexOf(cellContent.content);
+            const sourceOffset = offset + cellContent.prefix.length;
+            return { offset, sourceOffset, cell: notebook.cellAt(cellContent.cell.index) };
+        });
+        return new AlternativeTextDocument(content, cellOffsetMap, notebook);
+    }
+}
+exports.AlternativeTextNotebookContentProvider = AlternativeTextNotebookContentProvider;
+function generateAlternativeCellTextContent(cell, lineCommentStart, blockComment) {
+    const cellMarker = generateCellTextMarker(cell, lineCommentStart);
+    const src = cell.source.join(helpers_1.EOL);
+    const prefix = cell.language === 'markdown' ? `${cellMarker}${helpers_1.EOL}${blockComment[0]}${helpers_1.EOL}` : `${cellMarker}${helpers_1.EOL}`;
+    const content = cell.language === 'markdown'
+        ? `${prefix}${src}${helpers_1.EOL}${blockComment[1]}`
+        : `${prefix}${src}`;
+    return { content, prefix };
+}
+function getBlockComment(notebook) {
+    if (!notebook) {
+        return ['"""', '"""'];
+    }
+    const language = (0, languages_1.getLanguage)((0, helpers_1.getDefaultLanguage)(notebook));
+    return language.blockComment ?? ['```', '```'];
+}
+function getLineCommentStart(notebook) {
+    if (!notebook) {
+        return '#';
+    }
+    const language = (0, languages_1.getLanguage)((0, helpers_1.getDefaultLanguage)(notebook));
+    return language.lineComment.start || '#';
+}
+function extractCellParts(line, defaultLanguage) {
+    const idMatch = line.match(/\[id=(.+?)\]/);
+    const languageMatch = line.match(/\[language=(.+?)\]/);
+    if (!languageMatch) {
+        if (lineMightHaveCellMarker(line) && typeof defaultLanguage === 'string') {
+            // If we have a cell marker but no language, we assume the default language.
+            return { id: idMatch ? idMatch[1].trim() : '', language: defaultLanguage };
+        }
+        return;
+    }
+    return { id: idMatch ? idMatch[1].trim() : '', language: languageMatch[1].trim() };
+}
+//# sourceMappingURL=alternativeContentProvider.text.js.map

package/dist/src/_internal/platform/notebook/common/alternativeContentProvider.text.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"alternativeContentProvider.text.js","sourceRoot":"","sources":["../../../../../../src/_internal/platform/notebook/common/alternativeContentProvider.text.ts"],"names":[],"mappings":";;;AAaA,wDAGC;AAED,0DAEC;AAmMD,0CAMC;AAED,kDAMC;AAhOD,8DAA6D;AAC7D,sDAAmD;AACnD,qFAA8F;AAC9F,6EAAsF;AACtF,+EAA4E;AAC5E,uCAAsH;AACtH,wEAAmE;AAEnE,SAAgB,sBAAsB,CAAC,IAAiB,EAAE,WAAmB;IAC5E,MAAM,SAAS,GAAG,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;IACpD,OAAO,GAAG,WAAW,kBAAkB,SAAS,aAAa,IAAI,CAAC,QAAQ,GAAG,CAAC;AAC/E,CAAC;AAED,SAAgB,uBAAuB,CAAC,IAAY;IACnD,OAAO,IAAI,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;AACnD,CAAC;AAED,MAAM,uBAAwB,SAAQ,yDAA2B;IAChE,YAAY,IAAY,EAAmB,aAA6E,EAAE,QAA0B;QACnJ,KAAK,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QADoB,kBAAa,GAAb,aAAa,CAAgE;IAExH,CAAC;IAEQ,gBAAgB,CAAC,IAAkB,EAAE,QAAkB;QAC/D,MAAM,WAAW,GAAG,IAAA,mBAAS,EAAC,IAAI,CAAC,CAAC;QACpC,MAAM,gBAAgB,GAAG,mBAAmB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC5D,MAAM,UAAU,GAAG,sBAAsB,CAAC,WAAW,EAAE,gBAAgB,CAAC,CAAC;QAEzE,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,KAAK,2BAAS,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC7D,MAAM,YAAY,GAAG,eAAe,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACpD,MAAM,sBAAsB,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC;QAC9C,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACtD,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,KAAK,kCAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,MAAM,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;QACtG,MAAM,MAAM,GAAG,sBAAsB,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,UAAU,CAAC,MAAM,GAAG,SAAS,GAAG,cAAc,GAAG,YAAY,CAAC;QAC1H,OAAO,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;IAEQ,cAAc,CAAC,QAAkB;QACzC,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACvC,MAAM,IAAI,GAAG,IAAA,qBAAQ,EAAC,IAAI,CAAC,aAAa,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,IAAI,MAAM,CAAC,CAAC;QACjF,IAAI,CAAC,IAAI,EAAE,CAAC;YACX,OAAO,SAAS,CAAC;QAClB,CAAC;QACD,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC;QAC/E,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,QAAQ,EAAE,YAAY,EAAE,CAAC;IACpD,CAAC;CACD;AAGD,MAAa,sCAAuC,SAAQ,mEAAsC;IACjG;QACC,KAAK,CAAC,MAAM,CAAC,CAAC;IACf,CAAC;IAEM,gBAAgB,CAAC,IAAY;QACnC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,aAAG,CAAC,CAAC;QAC9B,IAAI,KAAK,CAAC,MAAM,IAAI,uBAAuB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YACvD,KAAK,CAAC,KAAK,EAAE,CAAC;YACd,OAAO,KAAK,CAAC,IAAI,CAAC,aAAG,CAAC,CAAC;QACxB,CAAC;aAAM,CAAC;YACP,OAAO,IAAI,CAAC;QACb,CAAC;IACF,CAAC;IAEe,qBAAqB,CAAC,QAA0B,EAAE,cAA8B,EAAE,kBAA0B;QAC3H,MAAM,YAAY,GAAG,eAAe,CAAC,QAAQ,CAAC,CAAC;QAC/C,MAAM,gBAAgB,GAAG,mBAAmB,CAAC,QAAQ,CAAC,CAAC;QACvD,MAAM,6BAA6B,GAAG,GAAG,gBAAgB,IAAI,kBAAkB,EAAE,CAAC;QAClF,MAAM,KAAK,GAAa,EAAE,CAAC;QAC3B,QAAQ,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;YACpC,IAAI,cAAc,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;gBACnC,MAAM,WAAW,GAAG,IAAA,mBAAS,EAAC,IAAI,CAAC,CAAC;gBACpC,IAAI,WAAW,CAAC,MAAM,CAAC,MAAM,IAAI,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,MAAM,EAAE,CAAC;oBACtE,WAAW,CAAC,MAAM,GAAG,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,6BAA6B,CAAC,CAAC;gBAC7E,CAAC;qBAAM,IAAI,WAAW,CAAC,MAAM,CAAC,MAAM,IAAI,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC;oBAC7F,WAAW,CAAC,MAAM,GAAG,CAAC,6BAA6B,EAAE,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,6BAA6B,CAAC,CAAC;gBAC/I,CAAC;qBAAM,CAAC;oBACP,WAAW,CAAC,MAAM,GAAG,CAAC,6BAA6B,CAAC,CAAC;gBACtD,CAAC;gBACD,KAAK,CAAC,IAAI,CAAC,kCAAkC,CAAC,WAAW,EAAE,gBAAgB,EAAE,YAAY,CAAC,CAAC,OAAO,CAAC,CAAC;YACrG,CAAC;iBAAM,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,KAAK,6BAA6B,EAAE,CAAC;gBACvF,KAAK,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;YAC3C,CAAC;QACF,CAAC,CAAC,CAAC;QACH,OAAO,KAAK,CAAC,IAAI,CAAC,aAAG,CAAC,CAAC;IACxB,CAAC;IAGe,KAAK,CAAC,CAAC,qBAAqB,CAAC,aAAqC,EAAE,WAAsC,EAAE,KAAwB;QACnJ,MAAM,UAAU,GAAG,CAAC,IAAA,aAAK,EAAC,aAAa,CAAC,CAAC;QACzC,MAAM,SAAS,GAAG,UAAU,CAAC,CAAC,CAAC,IAAA,sBAAY,EAAC,aAAa,CAAC,CAAC,CAAC,CAAC,IAAI,GAAG,EAAwB,CAAC;QAE7F,IAAI,cAAc,GAAG,KAAK,CAAC;QAC3B,IAAI,gBAAgB,GAAG,KAAK,CAAC;QAC7B,IAAI,mBAAmB,GAAG,KAAK,CAAC;QAChC,IAAI,YAAY,GAAG,KAAK,CAAC;QACzB,IAAI,SAAS,GAAG,CAAC,CAAC,CAAC;QAEnB,MAAM,gBAAgB,GAAG,mBAAmB,CAAC,UAAU,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QACrF,MAAM,YAAY,GAAG,eAAe,CAAC,UAAU,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QAC7E,MAAM,eAAe,GAAG,UAAU,CAAC,CAAC,CAAC,IAAA,uBAAW,EAAC,IAAA,4BAAkB,EAAC,aAAa,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC;QAC3G,MAAM,WAAW,GAAG,IAAI,GAAG,EAAU,CAAC;QACtC,IAAI,KAAK,EAAE,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;YAC5C,IAAI,KAAK,CAAC,uBAAuB,EAAE,CAAC;gBACnC,MAAM;YACP,CAAC;YACD,MAAM,IAAI,GAAG,UAAU,CAAC,KAAK,CAAC;YAE9B,+BAA+B;YAC/B,yEAAyE;YACzE,MAAM,0CAA0C,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,gBAAgB,MAAM,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;YAC9H,MAAM,2BAA2B,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,gBAAgB,gBAAgB,CAAC,CAAC;YACzF,kIAAkI;YAClI,MAAM,SAAS,GAAG,CAAC,2BAA2B,IAAI,0CAA0C,CAAC,CAAC,CAAC,CAAC,gBAAgB,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YACpJ,IAAI,CAAC,2BAA2B,IAAI,0CAA0C,CAAC,IAAI,SAAS,EAAE,QAAQ,EAAE,CAAC;gBACxG,IAAI,mBAAmB,EAAE,CAAC;oBACzB,mBAAmB,GAAG,KAAK,CAAC;gBAC7B,CAAC;gBACD,MAAM,cAAc,GAA0C,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,GAAG,EAAE,SAAS,EAAE,QAAQ,EAAE,SAAS,EAAE,IAAI,EAAE,kCAAgB,CAAC,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;gBAC7K,cAAc,CAAC,KAAK,GAAG,SAAS,IAAI,CAAC,CAAC;gBACtC,cAAc,CAAC,OAAO,GAAG,KAAK,CAAC;gBAC/B,+CAA+C;gBAC/C,IAAI,SAAS,CAAC,EAAE,IAAI,SAAS,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC,EAAE,QAAQ,CAAC,UAAU,KAAK,SAAS,CAAC,QAAQ,EAAE,CAAC;oBAC7F,IAAI,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC,EAAE,CAAC;wBACnC,SAAS,CAAC,EAAE,GAAG,EAAE,CAAC;oBACnB,CAAC;yBAAM,CAAC;wBACP,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;oBAC/B,CAAC;gBACF,CAAC;qBAAM,CAAC;oBACP,mEAAmE;oBACnE,0CAA0C;oBAC1C,SAAS,CAAC,EAAE,GAAG,EAAE,CAAC;gBACnB,CAAC;gBAED,MAAM,IAAI,GAAG,SAAS,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;gBACzC,cAAc,CAAC,EAAE,GAAG,SAAS,CAAC,EAAE,CAAC;gBACjC,cAAc,CAAC,QAAQ,GAAG,SAAS,CAAC,QAAQ,CAAC;gBAC7C,cAAc,CAAC,GAAG,GAAG,IAAI,EAAE,QAAQ,CAAC,GAAG,CAAC;gBACxC,cAAc,CAAC,IAAI,GAAG,IAAI,EAAE,IAAI,IAAI,CAAC,cAAc,CAAC,QAAQ,KAAK,UAAU,CAAC,CAAC,CAAC,kCAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,kCAAgB,CAAC,IAAI,CAAC,CAAC;gBAC/H,cAAc,GAAG,cAAc,CAAC,QAAQ,KAAK,UAAU,CAAC;gBACxD,gBAAgB,GAAG,KAAK,CAAC;gBAEzB,IAAI,YAAY,EAAE,CAAC;oBAClB,MAAM,EAAE,KAAK,EAAE,SAAS,GAAG,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;gBAC7C,CAAC;gBAED,YAAY,GAAG,IAAI,CAAC;gBACpB,MAAM,cAAc,CAAC;gBACrB,SAAS;YACV,CAAC;YAED,IAAI,CAAC,YAAY,EAAE,CAAC;gBACnB,SAAS;YACV,CAAC;YACD,IAAI,cAAc,EAAE,CAAC;gBACpB,IAAI,CAAC,gBAAgB,EAAE,CAAC;oBACvB,qCAAqC;oBACrC,IAAI,IAAI,KAAK,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC;wBAC9B,gBAAgB,GAAG,IAAI,CAAC;oBACzB,CAAC;yBAAM,CAAC;wBACP,sBAAsB;wBACtB,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;oBAChD,CAAC;gBACF,CAAC;qBAAM,CAAC;oBACP,0BAA0B;oBAC1B,IAAI,IAAI,KAAK,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC;wBAC9B,8BAA8B;wBAC9B,gBAAgB,GAAG,KAAK,CAAC;wBACzB,mBAAmB,GAAG,IAAI,CAAC;oBAC5B,CAAC;yBAAM,CAAC;wBACP,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;oBAChD,CAAC;gBACF,CAAC;YACF,CAAC;iBAAM,CAAC;gBACP,+BAA+B;gBAC/B,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;YAChD,CAAC;QACF,CAAC;QAED,IAAI,YAAY,EAAE,CAAC;YAClB,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;QACzC,CAAC;IACF,CAAC;IAEe,sBAAsB,CAAC,QAA0B,EAAE,oBAA8B;QAChG,MAAM,KAAK,GAAG,QAAQ,CAAC,QAAQ,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,KAAK,kCAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAA,mBAAS,EAAC,IAAI,CAAC,CAAC,CAAC;QACnJ,MAAM,YAAY,GAAG,eAAe,CAAC,QAAQ,CAAC,CAAC;QAC/C,MAAM,gBAAgB,GAAG,mBAAmB,CAAC,QAAQ,CAAC,CAAC;QACvD,MAAM,WAAW,GAAG,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,EAAE,GAAG,kCAAkC,CAAC,IAAI,EAAE,gBAAgB,EAAE,YAAY,CAAC,EAAE,IAAI,EAAE,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC;QAC5J,MAAM,OAAO,GAAG,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,aAAG,CAAC,CAAC;QAChE,MAAM,aAAa,GAAG,WAAW,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE;YACnD,MAAM,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;YACpD,MAAM,YAAY,GAAG,MAAM,GAAG,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC;YACxD,OAAO,EAAE,MAAM,EAAE,YAAY,EAAE,IAAI,EAAE,QAAQ,CAAC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;QAChF,CAAC,CAAC,CAAC;QAEH,OAAO,IAAI,uBAAuB,CAAC,OAAO,EAAE,aAAa,EAAE,QAAQ,CAAC,CAAC;IACtE,CAAC;CAED;AAtJD,wFAsJC;AAED,SAAS,kCAAkC,CAAC,IAAiB,EAAE,gBAAwB,EAAE,YAA8B;IACtH,MAAM,UAAU,GAAG,sBAAsB,CAAC,IAAI,EAAE,gBAAgB,CAAC,CAAC;IAClE,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,aAAG,CAAC,CAAC;IAClC,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,KAAK,UAAU,CAAC,CAAC,CAAC,GAAG,UAAU,GAAG,aAAG,GAAG,YAAY,CAAC,CAAC,CAAC,GAAG,aAAG,EAAE,CAAC,CAAC,CAAC,GAAG,UAAU,GAAG,aAAG,EAAE,CAAC;IACpH,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,KAAK,UAAU;QAC3C,CAAC,CAAC,GAAG,MAAM,GAAG,GAAG,GAAG,aAAG,GAAG,YAAY,CAAC,CAAC,CAAC,EAAE;QAC3C,CAAC,CAAC,GAAG,MAAM,GAAG,GAAG,EAAE,CAAC;IACrB,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;AAC5B,CAAC;AAED,SAAgB,eAAe,CAAC,QAA2B;IAC1D,IAAI,CAAC,QAAQ,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IACvB,CAAC;IACD,MAAM,QAAQ,GAAG,IAAA,uBAAW,EAAC,IAAA,4BAAkB,EAAC,QAAQ,CAAC,CAAC,CAAC;IAC3D,OAAO,QAAQ,CAAC,YAAY,IAAI,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;AAChD,CAAC;AAED,SAAgB,mBAAmB,CAAC,QAA2B;IAC9D,IAAI,CAAC,QAAQ,EAAE,CAAC;QACf,OAAO,GAAG,CAAC;IACZ,CAAC;IACD,MAAM,QAAQ,GAAG,IAAA,uBAAW,EAAC,IAAA,4BAAkB,EAAC,QAAQ,CAAC,CAAC,CAAC;IAC3D,OAAO,QAAQ,CAAC,WAAW,CAAC,KAAK,IAAI,GAAG,CAAC;AAC1C,CAAC;AAED,SAAS,gBAAgB,CAAC,IAAY,EAAE,eAAmC;IAC1E,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;IAC3C,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,oBAAoB,CAAC,CAAC;IACvD,IAAI,CAAC,aAAa,EAAE,CAAC;QACpB,IAAI,uBAAuB,CAAC,IAAI,CAAC,IAAI,OAAO,eAAe,KAAK,QAAQ,EAAE,CAAC;YAC1E,4EAA4E;YAC5E,OAAO,EAAE,EAAE,EAAE,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,QAAQ,EAAE,eAAe,EAAE,CAAC;QAC5E,CAAC;QACD,OAAO;IACR,CAAC;IACD,OAAO,EAAE,EAAE,EAAE,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,QAAQ,EAAE,aAAa,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC;AACpF,CAAC"}

package/dist/src/_internal/platform/notebook/common/alternativeContentProvider.xml.d.ts

@@ -0,0 +1,13 @@
+import type { CancellationToken, NotebookCell, NotebookDocument, Uri } from 'vscode';
+import { BaseAlternativeNotebookContentProvider } from './alternativeContentProvider';
+import { AlternativeNotebookDocument } from './alternativeNotebookDocument';
+import { LineOfCellText, LineOfText } from './helpers';
+export declare function isXmlContent(text: string): boolean;
+export declare class AlternativeXmlNotebookContentProvider extends BaseAlternativeNotebookContentProvider {
+    constructor();
+    stripCellMarkers(text: string): string;
+    getSummaryOfStructure(notebook: NotebookDocument, cellsToInclude: NotebookCell[], existingCodeMarker: string): string;
+    parseAlternateContent(notebookOrUri: NotebookDocument | Uri, inputStream: AsyncIterable<LineOfText>, token: CancellationToken): AsyncIterable<LineOfCellText>;
+    getAlternativeDocument(notebook: NotebookDocument, excludeMarkdownCells?: boolean): AlternativeNotebookDocument;
+}
+//# sourceMappingURL=alternativeContentProvider.xml.d.ts.map

package/dist/src/_internal/platform/notebook/common/alternativeContentProvider.xml.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"alternativeContentProvider.xml.d.ts","sourceRoot":"","sources":["../../../../../../src/_internal/platform/notebook/common/alternativeContentProvider.xml.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,iBAAiB,EAAE,YAAY,EAAE,gBAAgB,EAAE,GAAG,EAAE,MAAM,QAAQ,CAAC;AAIrF,OAAO,EAAE,sCAAsC,EAAE,MAAM,8BAA8B,CAAC;AACtF,OAAO,EAAE,2BAA2B,EAAE,MAAM,+BAA+B,CAAC;AAC5E,OAAO,EAAyC,cAAc,EAAE,UAAU,EAA0B,MAAM,WAAW,CAAC;AAetH,wBAAgB,YAAY,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAElD;AA+BD,qBAAa,qCAAsC,SAAQ,sCAAsC;;IAIzF,gBAAgB,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM;IAW7B,qBAAqB,CAAC,QAAQ,EAAE,gBAAgB,EAAE,cAAc,EAAE,YAAY,EAAE,EAAE,kBAAkB,EAAE,MAAM,GAAG,MAAM;IAuBvH,qBAAqB,CAAC,aAAa,EAAE,gBAAgB,GAAG,GAAG,EAAE,WAAW,EAAE,aAAa,CAAC,UAAU,CAAC,EAAE,KAAK,EAAE,iBAAiB,GAAG,aAAa,CAAC,cAAc,CAAC;IA8E3J,sBAAsB,CAAC,QAAQ,EAAE,gBAAgB,EAAE,oBAAoB,CAAC,EAAE,OAAO,GAAG,2BAA2B;CAc/H"}

package/dist/src/_internal/platform/notebook/common/alternativeContentProvider.xml.js

@@ -0,0 +1,191 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AlternativeXmlNotebookContentProvider = void 0;
+exports.isXmlContent = isXmlContent;
+const languages_1 = require("../../../util/common/languages");
+const types_1 = require("../../../util/common/types");
+const vscodeTypesShim_1 = require("../../../util/common/test/shims/vscodeTypesShim");
+const alternativeContentProvider_1 = require("./alternativeContentProvider");
+const alternativeNotebookDocument_1 = require("./alternativeNotebookDocument");
+const helpers_1 = require("./helpers");
+const arraysFind_1 = require("../../../util/vs/base/common/arraysFind");
+const StartDelimter = `<VSCode.Cell `;
+const StartEmptyCellDelimter = `<VSCode.Cell>`;
+const EndDelimter = `</VSCode.Cell>`;
+function generatePartialStartDelimiterWithId(id) {
+    return `${StartDelimter}id="${id}" `;
+}
+function generateCellMarker(cell) {
+    return `${generatePartialStartDelimiterWithId(cell.id)}language="${cell.language}">`;
+}
+function isXmlContent(text) {
+    return text.includes(StartDelimter) || text.includes(EndDelimter) || text.includes(StartEmptyCellDelimter);
+}
+class AlternativeXmlDocument extends alternativeNotebookDocument_1.AlternativeNotebookDocument {
+    constructor(text, cellOffsetMap, notebook) {
+        super(text, notebook);
+        this.cellOffsetMap = cellOffsetMap;
+    }
+    fromCellPosition(cell, position) {
+        const cellSummary = (0, helpers_1.summarize)(cell);
+        const cellMarker = generateCellMarker(cellSummary);
+        const eolLength = cell.document.eol === vscodeTypesShim_1.EndOfLine.LF ? 1 : 2;
+        const alternativeContentText = this.getText();
+        const offsetInCell = cell.document.offsetAt(position);
+        const offset = alternativeContentText.indexOf(cellMarker) + cellMarker.length + eolLength + offsetInCell;
+        return this.positionAt(offset);
+    }
+    toCellPosition(position) {
+        const offset = this.offsetAt(position);
+        const cell = (0, arraysFind_1.findLast)(this.cellOffsetMap, (cell) => cell.offset <= offset);
+        if (!cell) {
+            return undefined;
+        }
+        const cellPosition = cell.cell.document.positionAt(offset - cell.offset);
+        return { cell: cell.cell, position: cellPosition };
+    }
+}
+class AlternativeXmlNotebookContentProvider extends alternativeContentProvider_1.BaseAlternativeNotebookContentProvider {
+    constructor() {
+        super('xml');
+    }
+    stripCellMarkers(text) {
+        const lines = text.split(helpers_1.EOL);
+        if (lines.length && (lines[0].startsWith(StartDelimter) || lines[0].startsWith(StartEmptyCellDelimter))) {
+            lines.shift();
+        }
+        if (lines.length && lines[lines.length - 1].trim().endsWith(EndDelimter)) {
+            lines[lines.length - 1] = lines[lines.length - 1].substring(0, lines[lines.length - 1].lastIndexOf(EndDelimter));
+        }
+        return lines.join(helpers_1.EOL);
+    }
+    getSummaryOfStructure(notebook, cellsToInclude, existingCodeMarker) {
+        const lines = [];
+        const existingCodeMarkerWithComment = `// ${existingCodeMarker}`;
+        notebook.getCells().forEach((cell) => {
+            if (cellsToInclude.includes(cell)) {
+                const cellSummary = (0, helpers_1.summarize)(cell);
+                lines.push(generateCellMarker(cellSummary));
+                if (cellSummary.source.length && cellSummary.source[0].trim().length) {
+                    lines.push(cellSummary.source[0]);
+                    lines.push(existingCodeMarkerWithComment);
+                }
+                else if (cellSummary.source.length && cellSummary.source.some(line => line.trim().length)) {
+                    cellSummary.source = [existingCodeMarkerWithComment, cellSummary.source.filter(line => line.trim().length)[0], existingCodeMarkerWithComment];
+                }
+                else {
+                    lines.push(existingCodeMarkerWithComment);
+                }
+                lines.push(EndDelimter);
+            }
+            else if (!lines.length || lines[lines.length - 1] !== existingCodeMarkerWithComment) {
+                lines.push(existingCodeMarkerWithComment);
+            }
+        });
+        return lines.join(helpers_1.EOL);
+    }
+    async *parseAlternateContent(notebookOrUri, inputStream, token) {
+        const isNotebook = !(0, types_1.isUri)(notebookOrUri);
+        const cellIdMap = isNotebook ? (0, helpers_1.getCellIdMap)(notebookOrUri) : new Map();
+        let index = -1;
+        let endDelimiterSeen = false;
+        const cellIdsSeen = new Set();
+        let previousLineEndedWithEndCellMarker = false;
+        let previousLine = undefined;
+        const defaultLanguage = isNotebook ? (0, languages_1.getLanguage)((0, helpers_1.getDefaultLanguage)(notebookOrUri)).languageId : undefined;
+        for await (const lineOfText of inputStream) {
+            if (token.isCancellationRequested) {
+                break;
+            }
+            const line = lineOfText.value;
+            if ((line.startsWith(StartDelimter) || line.startsWith(StartEmptyCellDelimter)) && (index < 0 || (endDelimiterSeen || (previousLineEndedWithEndCellMarker && previousLine)))) {
+                if (!endDelimiterSeen && previousLineEndedWithEndCellMarker && previousLine) {
+                    // Last line didn't finish, emit that, but strip the end delimiter.
+                    previousLine.line = previousLine.line.substring(0, previousLine.line.lastIndexOf(EndDelimter));
+                    yield previousLine;
+                    yield { type: 'end', index: previousLine.index };
+                }
+                previousLineEndedWithEndCellMarker = false;
+                previousLine = undefined;
+                index += 1;
+                endDelimiterSeen = false;
+                const lineOfCellText = { type: 'start', index, uri: undefined, language: undefined, kind: vscodeTypesShim_1.NotebookCellKind.Code };
+                const cellParts = extractCellParts(line, defaultLanguage);
+                // LLM returns duplicate cell with the same id.
+                // We need tests for this.
+                // this is a work around to treat subsequent cells as new cells.
+                if (cellParts.id && cellIdMap.get(cellParts.id)?.document.languageId === cellParts.language) {
+                    if (cellIdsSeen.has(cellParts.id)) {
+                        cellParts.id = '';
+                    }
+                    else {
+                        cellIdsSeen.add(cellParts.id);
+                    }
+                }
+                else {
+                    // Possible duplicate cell with the same id but different language.
+                    // In such cases, treat them as new cells.
+                    cellParts.id = '';
+                }
+                const cell = cellIdMap.get(cellParts.id)?.document.languageId === cellParts.language ? cellIdMap.get(cellParts.id) : undefined;
+                lineOfCellText.id = cellParts.id;
+                lineOfCellText.language = cellParts.language;
+                lineOfCellText.uri = cell?.document.uri;
+                lineOfCellText.kind = cell?.kind || (lineOfCellText.language === 'markdown' ? vscodeTypesShim_1.NotebookCellKind.Markup : vscodeTypesShim_1.NotebookCellKind.Code);
+                yield lineOfCellText;
+            }
+            else if (line.startsWith(EndDelimter)) {
+                if (previousLineEndedWithEndCellMarker && previousLine) {
+                    // The last line somehow ends with the cell marker (must have been added by the user),
+                    // yield the previous line.
+                    yield previousLine;
+                }
+                endDelimiterSeen = true;
+                previousLineEndedWithEndCellMarker = false;
+                previousLine = undefined;
+                yield { type: 'end', index };
+            }
+            else if (index >= 0) {
+                if (previousLineEndedWithEndCellMarker && previousLine) {
+                    // Some how we have two subsequent lines that end with the cell marker,
+                    // Weird, shoudl not happen, if it does, yield the previous line.
+                    yield previousLine;
+                    previousLine = undefined;
+                }
+                previousLineEndedWithEndCellMarker = line.endsWith(EndDelimter);
+                if (previousLineEndedWithEndCellMarker) {
+                    previousLine = { type: 'line', index, line };
+                }
+                else {
+                    yield { type: 'line', index, line };
+                }
+            }
+        }
+    }
+    getAlternativeDocument(notebook, excludeMarkdownCells) {
+        const cells = notebook.getCells().filter(cell => excludeMarkdownCells ? cell.kind !== vscodeTypesShim_1.NotebookCellKind.Markup : true).map(cell => (0, helpers_1.summarize)(cell));
+        const cellContent = cells.map(cell => {
+            const cellMarker = generateCellMarker(cell);
+            const prefix = `${cellMarker}${helpers_1.EOL}`;
+            return { content: `${prefix}${cell.source.join(helpers_1.EOL)}${helpers_1.EOL}${EndDelimter}`, prefix, cell: notebook.cellAt(cell.index) };
+        });
+        const content = cellContent.map(cell => cell.content).join(helpers_1.EOL);
+        const cellOffsetMap = cellContent.map(cellContent => ({ offset: content.indexOf(cellContent.content) + cellContent.prefix.length, cell: cellContent.cell }));
+        return new AlternativeXmlDocument(content, cellOffsetMap, notebook);
+    }
+}
+exports.AlternativeXmlNotebookContentProvider = AlternativeXmlNotebookContentProvider;
+function extractCellParts(line, defaultLanguage) {
+    const idMatch = line.match(/id="([^"]+)"/);
+    const languageMatch = line.match(/language="([^"]+)"/);
+    if (!languageMatch) {
+        if (isXmlContent(line) && typeof defaultLanguage === 'string') {
+            // If we have a cell marker but no language, we assume the default language.
+            return { id: idMatch ? idMatch[1].trim() : '', language: defaultLanguage };
+        }
+        throw new Error(`Invalid cell part in ${line}`);
+    }
+    // New cells will not have an id.
+    return { id: idMatch ? idMatch[1].trim() : '', language: languageMatch[1].trim() };
+}
+//# sourceMappingURL=alternativeContentProvider.xml.js.map

package/dist/src/_internal/platform/notebook/common/alternativeContentProvider.xml.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"alternativeContentProvider.xml.js","sourceRoot":"","sources":["../../../../../../src/_internal/platform/notebook/common/alternativeContentProvider.xml.ts"],"names":[],"mappings":";;;AAyBA,oCAEC;AAtBD,8DAA6D;AAC7D,sDAAmD;AACnD,qFAAwG;AACxG,6EAAsF;AACtF,+EAA4E;AAC5E,uCAAsH;AACtH,wEAAmE;AAEnE,MAAM,aAAa,GAAG,eAAe,CAAC;AACtC,MAAM,sBAAsB,GAAG,eAAe,CAAC;AAC/C,MAAM,WAAW,GAAG,gBAAgB,CAAC;AAErC,SAAS,mCAAmC,CAAC,EAAU;IACtD,OAAO,GAAG,aAAa,OAAO,EAAE,IAAI,CAAC;AACtC,CAAC;AAED,SAAS,kBAAkB,CAAC,IAAiB;IAC5C,OAAO,GAAG,mCAAmC,CAAC,IAAI,CAAC,EAAE,CAAC,aAAa,IAAI,CAAC,QAAQ,IAAI,CAAC;AACtF,CAAC;AAED,SAAgB,YAAY,CAAC,IAAY;IACxC,OAAO,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,sBAAsB,CAAC,CAAC;AAC5G,CAAC;AAGD,MAAM,sBAAuB,SAAQ,yDAA2B;IAC/D,YAAY,IAAY,EAAmB,aAAuD,EAAE,QAA0B;QAC7H,KAAK,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QADoB,kBAAa,GAAb,aAAa,CAA0C;IAElG,CAAC;IAEQ,gBAAgB,CAAC,IAAkB,EAAE,QAAkB;QAC/D,MAAM,WAAW,GAAG,IAAA,mBAAS,EAAC,IAAI,CAAC,CAAC;QACpC,MAAM,UAAU,GAAG,kBAAkB,CAAC,WAAW,CAAC,CAAC;QAEnD,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,KAAK,2BAAS,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAE7D,MAAM,sBAAsB,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC;QAC9C,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACtD,MAAM,MAAM,GAAG,sBAAsB,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,UAAU,CAAC,MAAM,GAAG,SAAS,GAAG,YAAY,CAAC;QACzG,OAAO,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;IAEQ,cAAc,CAAC,QAAkB;QACzC,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACvC,MAAM,IAAI,GAAG,IAAA,qBAAQ,EAAC,IAAI,CAAC,aAAa,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,CAAC;QAC3E,IAAI,CAAC,IAAI,EAAE,CAAC;YACX,OAAO,SAAS,CAAC;QAClB,CAAC;QACD,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC;QACzE,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,QAAQ,EAAE,YAAY,EAAE,CAAC;IACpD,CAAC;CACD;AAED,MAAa,qCAAsC,SAAQ,mEAAsC;IAChG;QACC,KAAK,CAAC,KAAK,CAAC,CAAC;IACd,CAAC;IACM,gBAAgB,CAAC,IAAY;QACnC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,aAAG,CAAC,CAAC;QAC9B,IAAI,KAAK,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,aAAa,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,sBAAsB,CAAC,CAAC,EAAE,CAAC;YACzG,KAAK,CAAC,KAAK,EAAE,CAAC;QACf,CAAC;QACD,IAAI,KAAK,CAAC,MAAM,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;YAC1E,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,CAAC;QAClH,CAAC;QACD,OAAO,KAAK,CAAC,IAAI,CAAC,aAAG,CAAC,CAAC;IACxB,CAAC;IAEe,qBAAqB,CAAC,QAA0B,EAAE,cAA8B,EAAE,kBAA0B;QAC3H,MAAM,KAAK,GAAa,EAAE,CAAC;QAC3B,MAAM,6BAA6B,GAAG,MAAM,kBAAkB,EAAE,CAAC;QACjE,QAAQ,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;YACpC,IAAI,cAAc,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;gBACnC,MAAM,WAAW,GAAG,IAAA,mBAAS,EAAC,IAAI,CAAC,CAAC;gBACpC,KAAK,CAAC,IAAI,CAAC,kBAAkB,CAAC,WAAW,CAAC,CAAC,CAAC;gBAC5C,IAAI,WAAW,CAAC,MAAM,CAAC,MAAM,IAAI,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,MAAM,EAAE,CAAC;oBACtE,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;oBAClC,KAAK,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;gBAC3C,CAAC;qBAAM,IAAI,WAAW,CAAC,MAAM,CAAC,MAAM,IAAI,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC;oBAC7F,WAAW,CAAC,MAAM,GAAG,CAAC,6BAA6B,EAAE,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,6BAA6B,CAAC,CAAC;gBAC/I,CAAC;qBAAM,CAAC;oBACP,KAAK,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;gBAC3C,CAAC;gBACD,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACzB,CAAC;iBAAM,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,KAAK,6BAA6B,EAAE,CAAC;gBACvF,KAAK,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;YAC3C,CAAC;QACF,CAAC,CAAC,CAAC;QACH,OAAO,KAAK,CAAC,IAAI,CAAC,aAAG,CAAC,CAAC;IACxB,CAAC;IAEM,KAAK,CAAC,CAAC,qBAAqB,CAAC,aAAqC,EAAE,WAAsC,EAAE,KAAwB;QAC1I,MAAM,UAAU,GAAG,CAAC,IAAA,aAAK,EAAC,aAAa,CAAC,CAAC;QACzC,MAAM,SAAS,GAAG,UAAU,CAAC,CAAC,CAAC,IAAA,sBAAY,EAAC,aAAa,CAAC,CAAC,CAAC,CAAC,IAAI,GAAG,EAAwB,CAAC;QAG7F,IAAI,KAAK,GAAG,CAAC,CAAC,CAAC;QACf,IAAI,gBAAgB,GAAG,KAAK,CAAC;QAC7B,MAAM,WAAW,GAAG,IAAI,GAAG,EAAU,CAAC;QACtC,IAAI,kCAAkC,GAAG,KAAK,CAAC;QAC/C,IAAI,YAAY,GAA+B,SAAS,CAAC;QACzD,MAAM,eAAe,GAAG,UAAU,CAAC,CAAC,CAAC,IAAA,uBAAW,EAAC,IAAA,4BAAkB,EAAC,aAAa,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC;QAC3G,IAAI,KAAK,EAAE,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;YAC5C,IAAI,KAAK,CAAC,uBAAuB,EAAE,CAAC;gBACnC,MAAM;YACP,CAAC;YACD,MAAM,IAAI,GAAG,UAAU,CAAC,KAAK,CAAC;YAC9B,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,sBAAsB,CAAC,CAAC,IAAI,CAAC,KAAK,GAAG,CAAC,IAAI,CAAC,gBAAgB,IAAI,CAAC,kCAAkC,IAAI,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC9K,IAAI,CAAC,gBAAgB,IAAI,kCAAkC,IAAI,YAAY,EAAE,CAAC;oBAC7E,mEAAmE;oBACnE,YAAY,CAAC,IAAI,GAAG,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,YAAY,CAAC,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,CAAC;oBAC/F,MAAM,YAAY,CAAC;oBACnB,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,YAAY,CAAC,KAAK,EAAE,CAAC;gBAClD,CAAC;gBACD,kCAAkC,GAAG,KAAK,CAAC;gBAC3C,YAAY,GAAG,SAAS,CAAC;gBAEzB,KAAK,IAAI,CAAC,CAAC;gBACX,gBAAgB,GAAG,KAAK,CAAC;gBACzB,MAAM,cAAc,GAAmB,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,SAAS,EAAE,QAAQ,EAAE,SAAS,EAAE,IAAI,EAAE,kCAAgB,CAAC,IAAI,EAAE,CAAC;gBAClI,MAAM,SAAS,GAAG,gBAAgB,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;gBAC1D,+CAA+C;gBAC/C,0BAA0B;gBAC1B,gEAAgE;gBAChE,IAAI,SAAS,CAAC,EAAE,IAAI,SAAS,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC,EAAE,QAAQ,CAAC,UAAU,KAAK,SAAS,CAAC,QAAQ,EAAE,CAAC;oBAC7F,IAAI,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC,EAAE,CAAC;wBACnC,SAAS,CAAC,EAAE,GAAG,EAAE,CAAC;oBACnB,CAAC;yBAAM,CAAC;wBACP,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;oBAC/B,CAAC;gBACF,CAAC;qBAAM,CAAC;oBACP,mEAAmE;oBACnE,0CAA0C;oBAC1C,SAAS,CAAC,EAAE,GAAG,EAAE,CAAC;gBACnB,CAAC;gBACD,MAAM,IAAI,GAAG,SAAS,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC,EAAE,QAAQ,CAAC,UAAU,KAAK,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;gBAC/H,cAAc,CAAC,EAAE,GAAG,SAAS,CAAC,EAAE,CAAC;gBACjC,cAAc,CAAC,QAAQ,GAAG,SAAS,CAAC,QAAQ,CAAC;gBAC7C,cAAc,CAAC,GAAG,GAAG,IAAI,EAAE,QAAQ,CAAC,GAAG,CAAC;gBACxC,cAAc,CAAC,IAAI,GAAG,IAAI,EAAE,IAAI,IAAI,CAAC,cAAc,CAAC,QAAQ,KAAK,UAAU,CAAC,CAAC,CAAC,kCAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,kCAAgB,CAAC,IAAI,CAAC,CAAC;gBAC/H,MAAM,cAAc,CAAC;YACtB,CAAC;iBAAM,IAAI,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;gBACzC,IAAI,kCAAkC,IAAI,YAAY,EAAE,CAAC;oBACxD,sFAAsF;oBACtF,2BAA2B;oBAC3B,MAAM,YAAY,CAAC;gBACpB,CAAC;gBAED,gBAAgB,GAAG,IAAI,CAAC;gBACxB,kCAAkC,GAAG,KAAK,CAAC;gBAC3C,YAAY,GAAG,SAAS,CAAC;gBACzB,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;YAC9B,CAAC;iBAAM,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;gBACvB,IAAI,kCAAkC,IAAI,YAAY,EAAE,CAAC;oBACxD,uEAAuE;oBACvE,iEAAiE;oBACjE,MAAM,YAAY,CAAC;oBACnB,YAAY,GAAG,SAAS,CAAC;gBAC1B,CAAC;gBACD,kCAAkC,GAAG,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;gBAChE,IAAI,kCAAkC,EAAE,CAAC;oBACxC,YAAY,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;gBAC9C,CAAC;qBAAM,CAAC;oBACP,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;gBACrC,CAAC;YACF,CAAC;QACF,CAAC;IACF,CAAC;IAEe,sBAAsB,CAAC,QAA0B,EAAE,oBAA8B;QAChG,MAAM,KAAK,GAAG,QAAQ,CAAC,QAAQ,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,KAAK,kCAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAA,mBAAS,EAAC,IAAI,CAAC,CAAC,CAAC;QAEnJ,MAAM,WAAW,GAAG,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE;YACpC,MAAM,UAAU,GAAG,kBAAkB,CAAC,IAAI,CAAC,CAAC;YAC5C,MAAM,MAAM,GAAG,GAAG,UAAU,GAAG,aAAG,EAAE,CAAC;YACrC,OAAO,EAAE,OAAO,EAAE,GAAG,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,aAAG,CAAC,GAAG,aAAG,GAAG,WAAW,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;QACxH,CAAC,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,aAAG,CAAC,CAAC;QAChE,MAAM,aAAa,GAAG,WAAW,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,GAAG,WAAW,CAAC,MAAM,CAAC,MAAM,EAAE,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QAE7J,OAAO,IAAI,sBAAsB,CAAC,OAAO,EAAE,aAAa,EAAE,QAAQ,CAAC,CAAC;IACrE,CAAC;CAED;AAlID,sFAkIC;AAGD,SAAS,gBAAgB,CAAC,IAAY,EAAE,eAAmC;IAC1E,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;IAC3C,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,oBAAoB,CAAC,CAAC;IACvD,IAAI,CAAC,aAAa,EAAE,CAAC;QACpB,IAAI,YAAY,CAAC,IAAI,CAAC,IAAI,OAAO,eAAe,KAAK,QAAQ,EAAE,CAAC;YAC/D,4EAA4E;YAC5E,OAAO,EAAE,EAAE,EAAE,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,QAAQ,EAAE,eAAe,EAAE,CAAC;QAC5E,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,wBAAwB,IAAI,EAAE,CAAC,CAAC;IACjD,CAAC;IAED,iCAAiC;IACjC,OAAO,EAAE,EAAE,EAAE,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,QAAQ,EAAE,aAAa,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC;AACpF,CAAC"}

package/dist/src/_internal/platform/notebook/common/alternativeNotebookDocument.d.ts

@@ -0,0 +1,34 @@
+import type { NotebookCell, NotebookDocument, TextLine } from 'vscode';
+import { Position, Range } from '../../../util/common/test/shims/vscodeTypesShim';
+export declare abstract class AlternativeNotebookDocument {
+    protected readonly _text: string;
+    protected readonly notebook: NotebookDocument;
+    private _transformer;
+    private get transformer();
+    getText(range?: Range): string;
+    private _getTextInRange;
+    constructor(_text: string, notebook: NotebookDocument);
+    protected positionToOffset(position: Position): number;
+    /**
+     * Translates a position in the notebook document to the corresponding alternative position.
+     */
+    abstract fromCellPosition(cell: NotebookCell, position: Position): Position;
+    /**
+     * Translates a position in the alternative document to the corresponding cell index and position in the notebook document.
+     */
+    abstract toCellPosition(position: Position): {
+        cell: NotebookCell;
+        position: Position;
+    } | undefined;
+    getWordRangeAtPosition(_position: Position): Range | undefined;
+    private _lines;
+    get lines(): string[];
+    get lineCount(): number;
+    lineAt(line: number): TextLine;
+    lineAt(position: Position): TextLine;
+    offsetAt(position: Position): number;
+    positionAt(offset: number): Position;
+    validateRange(range: Range): Range;
+    validatePosition(position: Position): Position;
+}
+//# sourceMappingURL=alternativeNotebookDocument.d.ts.map

package/dist/src/_internal/platform/notebook/common/alternativeNotebookDocument.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"alternativeNotebookDocument.d.ts","sourceRoot":"","sources":["../../../../../../src/_internal/platform/notebook/common/alternativeNotebookDocument.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,YAAY,EAAE,gBAAgB,EAAE,QAAQ,EAAE,MAAM,QAAQ,CAAC;AAEvE,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,iDAAiD,CAAC;AAKlF,8BAAsB,2BAA2B;IAwBpC,SAAS,CAAC,QAAQ,CAAC,KAAK,EAAE,MAAM;IAAE,SAAS,CAAC,QAAQ,CAAC,QAAQ,EAAE,gBAAgB;IAvB3F,OAAO,CAAC,YAAY,CAA0C;IAC9D,OAAO,KAAK,WAAW,GAKtB;IAED,OAAO,CAAC,KAAK,CAAC,EAAE,KAAK,GAAG,MAAM;IAI9B,OAAO,CAAC,eAAe;gBAWQ,KAAK,EAAE,MAAM,EAAqB,QAAQ,EAAE,gBAAgB;IAI3F,SAAS,CAAC,gBAAgB,CAAC,QAAQ,EAAE,QAAQ,GAAG,MAAM;IAKtD;;OAEG;IACH,QAAQ,CAAC,gBAAgB,CAAC,IAAI,EAAE,YAAY,EAAE,QAAQ,EAAE,QAAQ,GAAG,QAAQ;IAE3E;;OAEG;IACH,QAAQ,CAAC,cAAc,CAAC,QAAQ,EAAE,QAAQ,GAAG;QAAE,IAAI,EAAE,YAAY,CAAC;QAAC,QAAQ,EAAE,QAAQ,CAAA;KAAE,GAAG,SAAS;IAEnG,sBAAsB,CAAC,SAAS,EAAE,QAAQ,GAAG,KAAK,GAAG,SAAS;IAiB9D,OAAO,CAAC,MAAM,CAAyB;IAEvC,IAAI,KAAK,IAAI,MAAM,EAAE,CAKpB;IAED,IAAI,SAAS,IAAI,MAAM,CAEtB;IAED,MAAM,CAAC,IAAI,EAAE,MAAM,GAAG,QAAQ;IAC9B,MAAM,CAAC,QAAQ,EAAE,QAAQ,GAAG,QAAQ;IAgBpC,QAAQ,CAAC,QAAQ,EAAE,QAAQ,GAAG,MAAM;IAIpC,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,QAAQ;IAMpC,aAAa,CAAC,KAAK,EAAE,KAAK,GAAG,KAAK;IAUlC,gBAAgB,CAAC,QAAQ,EAAE,QAAQ,GAAG,QAAQ;CAgC9C"}

package/dist/src/_internal/platform/notebook/common/alternativeNotebookDocument.js

@@ -0,0 +1,123 @@
+"use strict";
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AlternativeNotebookDocument = void 0;
+const wordHelper_1 = require("../../../util/vs/editor/common/core/wordHelper");
+const vscodeTypesShim_1 = require("../../../util/common/test/shims/vscodeTypesShim");
+const positionOffsetTransformer_1 = require("../../editing/common/positionOffsetTransformer");
+const textDocumentSnapshot_1 = require("../../editing/common/textDocumentSnapshot");
+class AlternativeNotebookDocument {
+    get transformer() {
+        if (!this._transformer) {
+            this._transformer = new positionOffsetTransformer_1.PositionOffsetTransformer(this._text);
+        }
+        return this._transformer;
+    }
+    getText(range) {
+        return range ? this._getTextInRange(range) : this._text;
+    }
+    _getTextInRange(_range) {
+        const range = this.validateRange(_range);
+        if (range.isEmpty) {
+            return '';
+        }
+        const offsetRange = this.transformer.toOffsetRange(range);
+        return this._text.substring(offsetRange.start, offsetRange.endExclusive);
+    }
+    constructor(_text, notebook) {
+        this._text = _text;
+        this.notebook = notebook;
+        this._transformer = null;
+        this._lines = null;
+    }
+    positionToOffset(position) {
+        position = this.validatePosition(position);
+        return this.transformer.getOffset(position);
+    }
+    getWordRangeAtPosition(_position) {
+        const position = this.validatePosition(_position);
+        const wordAtText = (0, wordHelper_1.getWordAtText)(position.character + 1, wordHelper_1.DEFAULT_WORD_REGEXP, this.lines[position.line], 0);
+        if (wordAtText) {
+            return new vscodeTypesShim_1.Range(position.line, wordAtText.startColumn - 1, position.line, wordAtText.endColumn - 1);
+        }
+        return undefined;
+    }
+    get lines() {
+        if (!this._lines) {
+            this._lines = this._text.split(/\r\n|\r|\n/g);
+        }
+        return this._lines;
+    }
+    get lineCount() {
+        return this.lines.length;
+    }
+    lineAt(lineOrPosition) {
+        let line;
+        if (lineOrPosition instanceof vscodeTypesShim_1.Position) {
+            line = lineOrPosition.line;
+        }
+        else if (typeof lineOrPosition === 'number') {
+            line = lineOrPosition;
+        }
+        else {
+            throw new Error(`Invalid argument`);
+        }
+        if (line < 0 || line >= this.lines.length) {
+            throw new Error('Illegal value for `line`');
+        }
+        return new textDocumentSnapshot_1.SnapshotDocumentLine(line, this.lines[line], line === this.lines.length - 1);
+    }
+    offsetAt(position) {
+        return this.transformer.getOffset(position);
+    }
+    positionAt(offset) {
+        offset = Math.floor(offset);
+        offset = Math.max(0, offset);
+        return this.transformer.getPosition(offset);
+    }
+    validateRange(range) {
+        const start = this.validatePosition(range.start);
+        const end = this.validatePosition(range.end);
+        if (start === range.start && end === range.end) {
+            return range;
+        }
+        return new vscodeTypesShim_1.Range(start.line, start.character, end.line, end.character);
+    }
+    validatePosition(position) {
+        if (this._text.length === 0) {
+            return position.with(0, 0);
+        }
+        let { line, character } = position;
+        let hasChanged = false;
+        if (line < 0) {
+            line = 0;
+            character = 0;
+            hasChanged = true;
+        }
+        else if (line >= this.lines.length) {
+            line = this.lines.length - 1;
+            character = this.lines[line].length;
+            hasChanged = true;
+        }
+        else {
+            const maxCharacter = this.lines[line].length;
+            if (character < 0) {
+                character = 0;
+                hasChanged = true;
+            }
+            else if (character > maxCharacter) {
+                character = maxCharacter;
+                hasChanged = true;
+            }
+        }
+        if (!hasChanged) {
+            return position;
+        }
+        return new vscodeTypesShim_1.Position(line, character);
+    }
+}
+exports.AlternativeNotebookDocument = AlternativeNotebookDocument;
+//# sourceMappingURL=alternativeNotebookDocument.js.map