feat: support Github Copilot (#2432)

* feat: support Github Copilot * feat: finish i18n translate * fix: add safeStorage * clean code * chore: remove vision model * ✨ feat: add Model Context Protocol (MCP) support (#2809) * ✨ feat: add Model Context Protocol (MCP) server configuration (main) - Added `@modelcontextprotocol/sdk` dependency for MCP integration. - Introduced MCP server configuration UI in settings with add, edit, delete, and activation functionalities. - Created `useMCPServers` hook to manage MCP server state and actions. - Added i18n support for MCP settings with translation keys. - Integrated MCP settings into the application's settings navigation and routing. - Implemented Redux state management for MCP servers. - Updated `yarn.lock` with new dependencies and their resolutions. * 🌟 feat: implement mcp service and integrate with ipc handlers - Added `MCPService` class to manage Model Context Protocol servers. - Implemented various handlers in `ipc.ts` for managing MCP servers including listing, adding, updating, deleting, and activating/deactivating servers. - Integrated MCP related types into existing type declarations for consistency across the application. - Updated `preload` to expose new MCP related APIs to the renderer process. - Enhanced `MCPSettings` component to interact directly with the new MCP service for adding, updating, deleting servers and setting their active states. - Introduced selectors in the MCP Redux slice for fetching active and all servers from the store. - Moved MCP types to a centralized location in `@renderer/types` for reuse across different parts of the application. * feat: enhance MCPService initialization to prevent recursive calls and improve error handling * feat: enhance MCP integration by adding MCPTool type and updating related methods * feat: implement streaming support for tool calls in OpenAIProvider and enhance message processing * fix: finish_reason undefined * fix migrate * feat: add rate limit and warning * feat: add delete copilot token file feat: add login message feat: add default headers and change getCopilotToken algorithm * fix * feat: add rate limit * chore: change apihost * fix: remove duplicate apikey * fix: change api host * chore: add vertify first tooltip --------- Co-authored-by: 亢奋猫 <kangfenmao@qq.com> Co-authored-by: LiuVaayne <10231735+vaayne@users.noreply.github.com>
2025-03-19 13:24:50 +08:00
parent f9f2586dc4
commit 0ddcecabdf
22 changed files with 936 additions and 31 deletions
@@ -0,0 +1,247 @@
+import axios, { AxiosRequestConfig } from 'axios'
+import { app, safeStorage } from 'electron'
+import fs from 'fs/promises'
+import path from 'path'
+
+// 配置常量，集中管理
+const CONFIG = {
+  GITHUB_CLIENT_ID: 'Iv1.b507a08c87ecfe98',
+  POLLING: {
+    MAX_ATTEMPTS: 8,
+    INITIAL_DELAY_MS: 1000,
+    MAX_DELAY_MS: 16000 // 最大延迟16秒
+  },
+  DEFAULT_HEADERS: {
+    accept: 'application/json',
+    'editor-version': 'Neovim/0.6.1',
+    'editor-plugin-version': 'copilot.vim/1.16.0',
+    'content-type': 'application/json',
+    'user-agent': 'GithubCopilot/1.155.0',
+    'accept-encoding': 'gzip,deflate,br'
+  },
+  // API端点集中管理
+  API_URLS: {
+    GITHUB_USER: 'https://api.github.com/user',
+    GITHUB_DEVICE_CODE: 'https://github.com/login/device/code',
+    GITHUB_ACCESS_TOKEN: 'https://github.com/login/oauth/access_token',
+    COPILOT_TOKEN: 'https://api.github.com/copilot_internal/v2/token'
+  }
+}
+
+// 接口定义移到顶部，便于查阅
+interface UserResponse {
+  login: string
+  avatar: string
+}
+
+interface AuthResponse {
+  device_code: string
+  user_code: string
+  verification_uri: string
+}
+
+interface TokenResponse {
+  access_token: string
+}
+
+interface CopilotTokenResponse {
+  token: string
+}
+
+// 自定义错误类，统一错误处理
+class CopilotServiceError extends Error {
+  constructor(
+    message: string,
+    public readonly cause?: unknown
+  ) {
+    super(message)
+    this.name = 'CopilotServiceError'
+  }
+}
+
+class CopilotService {
+  private readonly tokenFilePath: string
+  private headers: Record<string, string>
+
+  constructor() {
+    this.tokenFilePath = path.join(app.getPath('userData'), '.copilot_token')
+    this.headers = { ...CONFIG.DEFAULT_HEADERS }
+  }
+
+  /**
+   * 设置自定义请求头
+   */
+  private updateHeaders = (headers?: Record<string, string>): void => {
+    if (headers && Object.keys(headers).length > 0) {
+      this.headers = { ...headers }
+    }
+  }
+
+  /**
+   * 获取GitHub登录信息
+   */
+  public getUser = async (_: Electron.IpcMainInvokeEvent, token: string): Promise<UserResponse> => {
+    try {
+      const config: AxiosRequestConfig = {
+        headers: {
+          Connection: 'keep-alive',
+          'user-agent': 'Visual Studio Code (desktop)',
+          'Sec-Fetch-Site': 'none',
+          'Sec-Fetch-Mode': 'no-cors',
+          'Sec-Fetch-Dest': 'empty',
+          authorization: `token ${token}`
+        }
+      }
+
+      const response = await axios.get(CONFIG.API_URLS.GITHUB_USER, config)
+      return {
+        login: response.data.login,
+        avatar: response.data.avatar_url
+      }
+    } catch (error) {
+      console.error('Failed to get user information:', error)
+      throw new CopilotServiceError('无法获取GitHub用户信息', error)
+    }
+  }
+
+  /**
+   * 获取GitHub设备授权信息
+   */
+  public getAuthMessage = async (
+    _: Electron.IpcMainInvokeEvent,
+    headers?: Record<string, string>
+  ): Promise<AuthResponse> => {
+    try {
+      this.updateHeaders(headers)
+
+      const response = await axios.post<AuthResponse>(
+        CONFIG.API_URLS.GITHUB_DEVICE_CODE,
+        {
+          client_id: CONFIG.GITHUB_CLIENT_ID,
+          scope: 'read:user'
+        },
+        { headers: this.headers }
+      )
+
+      return response.data
+    } catch (error) {
+      console.error('Failed to get auth message:', error)
+      throw new CopilotServiceError('无法获取GitHub授权信息', error)
+    }
+  }
+
+  /**
+   * 使用设备码获取访问令牌 - 优化轮询逻辑
+   */
+  public getCopilotToken = async (
+    _: Electron.IpcMainInvokeEvent,
+    device_code: string,
+    headers?: Record<string, string>
+  ): Promise<TokenResponse> => {
+    this.updateHeaders(headers)
+
+    let currentDelay = CONFIG.POLLING.INITIAL_DELAY_MS
+
+    for (let attempt = 0; attempt < CONFIG.POLLING.MAX_ATTEMPTS; attempt++) {
+      await this.delay(currentDelay)
+
+      try {
+        const response = await axios.post<TokenResponse>(
+          CONFIG.API_URLS.GITHUB_ACCESS_TOKEN,
+          {
+            client_id: CONFIG.GITHUB_CLIENT_ID,
+            device_code,
+            grant_type: 'urn:ietf:params:oauth:grant-type:device_code'
+          },
+          { headers: this.headers }
+        )
+
+        const { access_token } = response.data
+        if (access_token) {
+          return { access_token }
+        }
+      } catch (error) {
+        // 指数退避策略
+        currentDelay = Math.min(currentDelay * 2, CONFIG.POLLING.MAX_DELAY_MS)
+
+        // 仅在最后一次尝试失败时记录详细错误
+        const isLastAttempt = attempt === CONFIG.POLLING.MAX_ATTEMPTS - 1
+        if (isLastAttempt) {
+          console.error(`Token polling failed after ${CONFIG.POLLING.MAX_ATTEMPTS} attempts:`, error)
+        }
+      }
+    }
+
+    throw new CopilotServiceError('获取访问令牌超时，请重试')
+  }
+
+  /**
+   * 保存Copilot令牌到本地文件
+   */
+  public saveCopilotToken = async (_: Electron.IpcMainInvokeEvent, token: string): Promise<void> => {
+    try {
+      const encryptedToken = safeStorage.encryptString(token)
+      await fs.writeFile(this.tokenFilePath, encryptedToken)
+    } catch (error) {
+      console.error('Failed to save token:', error)
+      throw new CopilotServiceError('无法保存访问令牌', error)
+    }
+  }
+
+  /**
+   * 从本地文件读取令牌并获取Copilot令牌
+   */
+  public getToken = async (
+    _: Electron.IpcMainInvokeEvent,
+    headers?: Record<string, string>
+  ): Promise<CopilotTokenResponse> => {
+    try {
+      this.updateHeaders(headers)
+
+      const encryptedToken = await fs.readFile(this.tokenFilePath)
+      const access_token = safeStorage.decryptString(Buffer.from(encryptedToken))
+
+      const config: AxiosRequestConfig = {
+        headers: {
+          ...this.headers,
+          authorization: `token ${access_token}`
+        }
+      }
+
+      const response = await axios.get<CopilotTokenResponse>(CONFIG.API_URLS.COPILOT_TOKEN, config)
+
+      return response.data
+    } catch (error) {
+      console.error('Failed to get Copilot token:', error)
+      throw new CopilotServiceError('无法获取Copilot令牌，请重新授权', error)
+    }
+  }
+
+  /**
+   * 退出登录，删除本地token文件
+   */
+  public logout = async (): Promise<void> => {
+    try {
+      try {
+        await fs.access(this.tokenFilePath)
+        await fs.unlink(this.tokenFilePath)
+        console.log('Successfully logged out from Copilot')
+      } catch (error) {
+        // 文件不存在不是错误，只是记录一下
+        console.log('Token file not found, nothing to delete')
+      }
+    } catch (error) {
+      console.error('Failed to logout:', error)
+      throw new CopilotServiceError('无法完成退出登录操作', error)
+    }
+  }
+
+  /**
+   * 辅助方法：延迟执行
+   */
+  private delay = (ms: number): Promise<void> => {
+    return new Promise((resolve) => setTimeout(resolve, ms))
+  }
+}
+
+export default new CopilotService()