1.0.1 代码格式化移植新版

src/providers/documentFormattingProvider.ts

@@ -1,100 +1,50 @@
 import * as vscode from 'vscode';
+import { beautifyCode, isSquirrelCode } from '../utils/beautifyHelper';
 
-// 文档格式化提供者类
+/**
+ * 文档格式化提供者类 - 使用 js-beautify.js
+ */
 export class DocumentFormattingProvider implements vscode.DocumentFormattingEditProvider {
-    // 提供整个文档的代码格式化功能
+    /**
+     * 提供整个文档的代码格式化功能
+     */
     async provideDocumentFormattingEdits(
         document: vscode.TextDocument,
         options: vscode.FormattingOptions,
         token: vscode.CancellationToken
     ): Promise<vscode.TextEdit[]> {
         const edits: vscode.TextEdit[] = [];
-        const lines: string[] = [];
 
-        // 逐行处理文档内容
-        for (let i = 0; i < document.lineCount; i++) {
-            const line = document.lineAt(i);
-            lines.push(line.text);
+        try {
+            // 获取文档全文
+            const fullText = document.getText();
+
+            // 检查是否是 Squirrel 代码
+            if (!isSquirrelCode(fullText)) {
+                return edits; // 如果不是 Squirrel 代码，不应用格式化
+            }
+
+            // 使用 js-beautify 格式化代码
+            const formattedText = beautifyCode(fullText, options);
+
+            // 如果格式化后的代码与原代码相同，不需要修改
+            if (formattedText === fullText) {
+                return edits;
+            }
+
+            // 创建编辑操作，替换整个文档
+            const fullRange = new vscode.Range(
+                document.positionAt(0),
+                document.positionAt(fullText.length)
+            );
+
+            edits.push(vscode.TextEdit.replace(fullRange, formattedText));
+
+        } catch (error) {
+            console.error('文档格式化失败:', error);
+            // 如果格式化失败，返回空编辑数组，保持原代码不变
         }
 
-        // 格式化文档
-        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;
     }
-
-    // 格式化文档
-    private formatDocument(lines: string[], options: vscode.FormattingOptions): string[] {
-        const formattedLines: string[] = [];
-        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));
-    }
-
-    // 在操作符周围添加适当空格
-    private normalizeSpaces(line: string): string {
-        // 在常见的操作符周围添加空格
-        return line
-            .replace(/([^\s])(==|!=|<=|>=|<|>|=|\+|-|\*|\/|%)([^\s])/g, '$1 $2 $3')
-            .replace(/([^\s])(,)([^\s])/g, '$1$2 $3')
-            .replace(/\s+/g, ' ') // 将多个空格替换为单个空格
-            .trim();
-    }
-
-    // 创建指定级别的缩进
-    private createIndent(level: number, options: vscode.FormattingOptions): string {
-        if (options.insertSpaces) {
-            return ' '.repeat(level * options.tabSize);
-        } else {
-            return '\t'.repeat(level);
-        }
-    }
 }