feat: 添加使用 PawSQL 优化 SQL 的上下文菜单功能

在编辑器中添加了一个新的上下文菜单项，允许用户使用 PawSQL 优化选中的 SQL 语句。该功能通过解析当前 SQL 文件并找到最接近光标的查询来实现优化。

Author: lhy

package.json

@@ -113,6 +113,11 @@
         "command": "pawsql.setDefaultWorkspace",
         "title": "%commands.setDefaultWorkspace%",
         "icon": "$(star)"
+      },
+      {
+        "command": "pawsql.contextMenu.optimizeWithSelectedWorkspace",
+        "title": "%commands.optimizeSQL%",
+        "icon": "$(sparkle)"
       }
     ],
     "configuration": [
@@ -226,6 +231,13 @@
           "when": "view == pawsqlSidebar && viewItem == workspaceItem",
           "group": "inline"
         }
+      ],
+      "editor/context": [
+        {
+          "command": "pawsql.contextMenu.optimizeWithSelectedWorkspace",
+          "when": "editorLangId =~ /sql/",
+          "group": "navigation"
+        }
       ]
     },
     "icons": {

package.nls.en.json

@@ -5,7 +5,7 @@
   "commands.refreshTree": "Refresh",
   "commands.createWorkspace": "Create a Workspace",
   "commands.setDefaultWorkspace": "Set as Default Workspace",
-
+  "commands.optimizeSQL": "Optimize SQL with PawSQL",
   "config.initialization.title": "PawSQL Setup",
   "config.initialization.version.title": "PawSQL Edition",
   "config.initialization.backendUrl.title": "PawSQL Server",

package.nls.json

@@ -5,6 +5,7 @@
   "commands.refreshTree": "刷新",
   "commands.createWorkspace": "创建工作空间",
   "commands.setDefaultWorkspace": "设置为默认工作空间",
+  "commands.optimizeSQL": "使用 PawSQL 优化此 SQL",
   "config.initialization.title": "PawSQL Client",
   "config.initialization.version.title": "PawSQL 版本",
   "config.initialization.backendUrl.title": "PawSQL 服务器",

src/CommandManager.ts

@@ -8,6 +8,7 @@ import { SqlCodeLensProvider } from "./SqlCodeLensProvider";
 import { ConfigurationService } from "./configurationService";
 import { LanguageService } from "./LanguageService";
 import path from "path";
+import parse from "./utils/parse";
 
 export class CommandManager {
   constructor(
@@ -19,6 +20,7 @@ export class CommandManager {
   public async initializeCommands(): Promise<void> {
     try {
       this.registerApiKeyCommands();
+      this.registerContextMenuCommands();
     } catch (error) {
       ErrorHandler.handle("initialize.commands.failed", error);
     }
@@ -32,6 +34,55 @@ export class CommandManager {
     this.context.subscriptions.push(configFileDefaultDisposable);
   }
 
+  private registerContextMenuCommands(): void {
+    const contextMenuOptimizeDisposable = vscode.commands.registerCommand(
+      "pawsql.contextMenu.optimizeWithSelectedWorkspace",
+      () => this.handleContextMenuOptimize()
+    );
+    this.context.subscriptions.push(contextMenuOptimizeDisposable);
+  }
+
+  public async handleContextMenuOptimize(): Promise<void> {
+    const editor = vscode.window.activeTextEditor;
+    if (!editor || editor.document.languageId !== "sql") {
+      return;
+    }
+
+    const cursorPosition = editor.selection.active;
+    const document = editor.document;
+    const text = document.getText();
+    const queries = await parse(text);
+
+    let closestQuery: { range: vscode.Range; text: string } | null = null;
+    let minDistance = Number.MAX_VALUE;
+
+    for (const query of queries) {
+      const queryIndex = text.indexOf(query);
+      if (queryIndex === -1) continue;
+
+      const startPos = document.positionAt(queryIndex);
+      const endPos = document.positionAt(queryIndex + query.length);
+      const range = new vscode.Range(startPos, endPos);
+
+      const distance = Math.min(
+        Math.abs(cursorPosition.line - startPos.line),
+        Math.abs(cursorPosition.line - endPos.line)
+      );
+
+      if (distance < minDistance) {
+        minDistance = distance;
+        closestQuery = { range, text: query };
+      }
+    }
+
+    if (closestQuery) {
+      await this.extension.handleWorkspaceSelectionWithRangeQuery(
+        closestQuery.text,
+        closestQuery.range
+      );
+    }
+  }
+
   public async handleFileDefaultWorkspaceSelection(): Promise<void> {
     const apiKey = await ConfigurationService.getApiKey();