squirrelVsis/dist/providers/documentFormattingProvider.js

"use strict";
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
    if (k2 === undefined) k2 = k;
    var desc = Object.getOwnPropertyDescriptor(m, k);
    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
      desc = { enumerable: true, get: function() { return m[k]; } };
    }
    Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
    if (k2 === undefined) k2 = k;
    o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
    Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
    o["default"] = v;
});
var __importStar = (this && this.__importStar) || (function () {
    var ownKeys = function(o) {
        ownKeys = Object.getOwnPropertyNames || function (o) {
            var ar = [];
            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
            return ar;
        };
        return ownKeys(o);
    };
    return function (mod) {
        if (mod && mod.__esModule) return mod;
        var result = {};
        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
        __setModuleDefault(result, mod);
        return result;
    };
})();
Object.defineProperty(exports, "__esModule", { value: true });
exports.DocumentFormattingProvider = void 0;
const vscode = __importStar(require("vscode"));
// 文档格式化提供者类
class DocumentFormattingProvider {
    // 提供整个文档的代码格式化功能
    async provideDocumentFormattingEdits(document, options, token) {
        const edits = [];
        const lines = [];
        // 逐行处理文档内容
        for (let i = 0; i < document.lineCount; i++) {
            const line = document.lineAt(i);
            lines.push(line.text);
        }
        // 格式化文档
        const formattedLines = this.formatDocument(lines, options);
        // 创建编辑操作
        const fullRange = new vscode.Range(document.positionAt(0), document.positionAt(document.getText().length));
        edits.push(vscode.TextEdit.replace(fullRange, formattedLines.join('\n')));
        return edits;
    }
    // 格式化文档
    formatDocument(lines, options) {
        const formattedLines = [];
        let indentLevel = 0;
        for (let i = 0; i < lines.length; i++) {
            let line = lines[i];
            const trimmedLine = line.trim();
            if (trimmedLine.length === 0) {
                // 空行保持原样或删除
                formattedLines.push('');
                continue;
            }
            // 检查是否需要减少缩进（遇到右大括号等）
            if (trimmedLine.startsWith('}') || trimmedLine.startsWith(')') || trimmedLine.startsWith(']')) {
                indentLevel = Math.max(0, indentLevel - 1);
            }
            // 添加适当的缩进
            const indent = this.createIndent(indentLevel, options);
            const formattedLine = indent + trimmedLine;
            formattedLines.push(formattedLine);
            // 检查是否需要增加缩进（遇到左大括号等）
            if (trimmedLine.endsWith('{') || trimmedLine.endsWith('(') || trimmedLine.endsWith('[')) {
                indentLevel++;
            }
            // 特殊处理else、catch等关键字，它们应该与前面的右大括号在同一行
            if (trimmedLine.startsWith('else') || trimmedLine.startsWith('catch') || trimmedLine.startsWith('elif')) {
                if (formattedLines.length > 1) {
                    const prevLine = formattedLines[formattedLines.length - 2];
                    if (prevLine.trim().endsWith('}')) {
                        // 将else等关键字与前面的右大括号放在同一行
                        formattedLines[formattedLines.length - 2] = prevLine.trim() + ' ' + trimmedLine;
                        formattedLines.pop(); // 移除当前行
                        indentLevel--; // 修正缩进级别
                    }
                }
            }
        }
        // 规范化空格（在操作符周围添加适当空格）
        return formattedLines.map(line => this.normalizeSpaces(line));
    }
    // 在操作符周围添加适当空格
    normalizeSpaces(line) {
        // 在常见的操作符周围添加空格
        return line
            .replace(/([^\s])(==|!=|<=|>=|<|>|=|\+|-|\*|\/|%)([^\s])/g, '$1 $2 $3')
            .replace(/([^\s])(,)([^\s])/g, '$1$2 $3')
            .replace(/\s+/g, ' ') // 将多个空格替换为单个空格
            .trim();
    }
    // 创建指定级别的缩进
    createIndent(level, options) {
        if (options.insertSpaces) {
            return ' '.repeat(level * options.tabSize);
        }
        else {
            return '\t'.repeat(level);
        }
    }
}
exports.DocumentFormattingProvider = DocumentFormattingProvider;
//# sourceMappingURL=documentFormattingProvider.js.map