extension.js

// The module 'vscode' contains the VS Code extensibility API
// Import the module and reference it with the alias vscode in your code below
const vscode = require('vscode');

// This method is called when your extension is activated
// Your extension is activated the very first time the command is executed

/**
 * @param {vscode.ExtensionContext} context
 */
function activate(context) {
    let disposable = vscode.window.onDidChangeActiveTextEditor(editor => {
        if (editor && editor.document.languageId === 'python') {
            foldDocstrings(editor);
        }
    });

    context.subscriptions.push(disposable);
}

/**
 * Folds all multiline docstrings in the given editor.
 *
 * @param {vscode.TextEditor} editor - The editor instance containing the document to process.
 */
function foldDocstrings(editor) {
    const document = editor.document;

    setTimeout(() => {
        const text = document.getText();
        // console.log("Document Text:", text);

        if (!text) {
            console.log("⚠️ Warning: getText() returned an empty string.");
            return;
        }

        const regex = /"""\s*([\s\S]*?)\s*"""/g; // Match multiline docstrings
        let ranges = [];
    
        let match;
        while ((match = regex.exec(text)) !== null) {
            let startLine = document.positionAt(match.index).line + 1;
            let endLine = document.positionAt(match.index + match[0].length).line + 1;
    
            if (startLine < endLine) {
                console.log(`Docstring found from line ${startLine} to ${endLine}`);
                ranges.push(new vscode.Range(startLine, 0, endLine, 0));
            }
        }

        if (ranges.length === 0) {
            console.log("✅ No docstrings found.");
            return;
        }

        // Apply folding using selections and command
        editor.selections = ranges.map(range => new vscode.Selection(range.start, range.start));
        vscode.commands.executeCommand("editor.fold");
    }, 250);
}

// This method is called when your extension is deactivated
function deactivate() {}

module.exports = {
	activate,
	deactivate
}