feat(gpt-runner-vscode): vscode is success load gpt-runner

2214962083 · 2214962083 · commit d439ff79217d · 2023-06-05T00:50:05.000+08:00
diff --git a/.vscode/settings.json b/.vscode/settings.json
@@ -11,6 +11,7 @@
   "typescript.inlayHints.parameterTypes.enabled": true,
   // "typescript.inlayHints.functionLikeReturnTypes.enabled": true,
   "cSpell.words": [
+    "cachedir",
     "Chatgpt",
     "clsx",
     "codicon",
diff --git a/packages/gpt-runner-vscode/.vscode/settings.json b/packages/gpt-runner-vscode/.vscode/settings.json
@@ -0,0 +1,10 @@
+{
+    "workbench.colorCustomizations": {
+        "activityBar.background": "#312F04",
+        "titleBar.activeBackground": "#454106",
+        "titleBar.activeForeground": "#FDFBE1"
+    },
+    "cSpell.words": [
+        "nicepkg"
+    ]
+}
diff --git a/packages/gpt-runner-vscode/package.json b/packages/gpt-runner-vscode/package.json
@@ -24,11 +24,39 @@
     "onStartupFinished"
   ],
   "contributes": {
+    "viewsContainers": {
+      "activitybar": [
+        {
+          "id": "gpt-runner",
+          "title": "GPT Runner",
+          "icon": "res/sidebar-icon.png"
+        }
+      ]
+    },
+    "views": {
+      "gpt-runner": [
+        {
+          "type": "webview",
+          "id": "gpt-runner.chatView",
+          "name": "GPT Runner Chat"
+        }
+      ]
+    },
     "commands": [
       {
         "command": "gpt-runner.reload",
         "title": "Reload GPT Runner",
         "category": "GPT Runner"
+      },
+      {
+        "command": "gpt-runner.restartServer",
+        "title": "GPT Runner Restart Server",
+        "category": "GPT Runner"
+      },
+      {
+        "command": "gpt-runner.openChat",
+        "title": "GPT Runner Open Chat",
+        "category": "GPT Runner"
       }
     ],
     "configuration": {
@@ -50,10 +78,10 @@
   },
   "devDependencies": {
     "@types/vscode": "^1.71.0",
-    "@nicepkg/gpt-runner": "workspace:*",
+    "@nicepkg/gpt-runner-web": "workspace:*",
     "@nicepkg/gpt-runner-shared": "workspace:*",
     "esno": "^0.16.3",
     "eventemitter3": "^5.0.1",
-    "unconfig": "^0.3.7"
+    "unconfig": "^0.3.9"
   }
 }
diff --git a/packages/gpt-runner-vscode/res/sidebar-icon.png b/packages/gpt-runner-vscode/res/sidebar-icon.png
diff --git a/packages/gpt-runner-vscode/src/index.ts b/packages/gpt-runner-vscode/src/index.ts
@@ -5,6 +5,8 @@ import { log } from './log'
 import { ContextLoader } from './contextLoader'
 import { getExtConfiguration } from './utils'
 import { EXT_DISPLAY_NAME, EXT_NAME } from './constant'
+import { registerWebview } from './register/webview'
+import { registerServer } from './register/server'
 
 async function registerRoot(ext: ExtensionContext, status: StatusBarItem, cwd: string) {
   const contextLoader = new ContextLoader(cwd)
@@ -13,7 +15,8 @@ async function registerRoot(ext: ExtensionContext, status: StatusBarItem, cwd: s
 
   log.appendLine(`📁 Root context loaded: ${cwd}`)
 
-  // registerAnnotations(cwd, contextLoader, status, ext)
+  registerServer(cwd, contextLoader, ext)
+  registerWebview(cwd, contextLoader, ext)
 
   return contextLoader
 }
diff --git a/packages/gpt-runner-vscode/src/register/server.ts b/packages/gpt-runner-vscode/src/register/server.ts
@@ -0,0 +1,66 @@
+import child_process from 'child_process'
+import type { Disposable, ExtensionContext } from 'vscode'
+import * as vscode from 'vscode'
+import type { ContextLoader } from '../contextLoader'
+import { EXT_NAME } from '../constant'
+
+export async function registerServer(
+  cwd: string,
+  contextLoader: ContextLoader,
+  ext: ExtensionContext,
+) {
+  let serverDisposable: Disposable
+  let serverProcess: child_process.ChildProcessWithoutNullStreams
+
+  const dispose = () => {
+    serverProcess?.kill?.()
+    serverDisposable?.dispose?.()
+  }
+
+  const registerProvider = () => {
+    dispose()
+
+    serverDisposable = vscode.commands.registerCommand(`${EXT_NAME}.restartServer`, () => {
+      serverProcess?.kill?.()
+
+      const { extensionUri } = ext
+      const serverUri = vscode.Uri.joinPath(extensionUri, './node_modules/@nicepkg/gpt-runner-web/dist/server.cjs')
+
+      serverProcess = child_process.spawn('node', [serverUri.fsPath], {
+        env: {
+          ...process.env,
+          NODE_OPTIONS: '--experimental-fetch',
+          NODE_NO_WARNINGS: '1',
+        },
+      })
+
+      serverProcess.stdout.on('data', (data: string) => {
+        console.log(`stdout: ${data}`)
+        vscode.window.showInformationMessage(`Server running: ${data}`)
+      })
+
+      serverProcess.stderr.on('data', (data: string) => {
+        console.error(`stderr: ${data}`)
+        vscode.window.showErrorMessage(`Server error: ${data}`)
+      })
+
+      serverProcess.on('close', (code: string) => {
+        console.log(`child process exited with code ${code}`)
+        vscode.window.showWarningMessage(`Server stopped with code: ${code}`)
+      })
+    })
+
+    return serverDisposable
+  }
+
+  ext.subscriptions.push(
+    registerProvider(),
+  )
+
+  contextLoader.emitter.on('contextReload', () => {
+    registerProvider()
+  })
+  contextLoader.emitter.on('contextUnload', () => {
+    dispose()
+  })
+}
diff --git a/packages/gpt-runner-vscode/src/register/webview.ts b/packages/gpt-runner-vscode/src/register/webview.ts
@@ -0,0 +1,93 @@
+import fs from 'fs'
+import path from 'path'
+import type { ExtensionContext } from 'vscode'
+import * as vscode from 'vscode'
+import type { ContextLoader } from '../contextLoader'
+import { EXT_NAME } from '../constant'
+import { createHash } from '../utils'
+
+class ChatViewProvider implements vscode.WebviewViewProvider {
+  static readonly viewType = `${EXT_NAME}.chatView`
+
+  #view?: vscode.WebviewView
+  #extContext: ExtensionContext
+
+  constructor(
+    extContext: ExtensionContext,
+  ) {
+    this.#extContext = extContext
+  }
+
+  resolveWebviewView(
+    webviewView: vscode.WebviewView,
+    _context: vscode.WebviewViewResolveContext,
+    _token: vscode.CancellationToken,
+  ) {
+    const { extensionUri } = this.#extContext
+    this.#view = webviewView
+
+    const baseUri = vscode.Uri.joinPath(extensionUri, './node_modules/@nicepkg/gpt-runner-web/dist/browser')
+
+    webviewView.webview.options = {
+      // Allow scripts in the webview
+      enableScripts: true,
+
+      localResourceRoots: [baseUri],
+    }
+
+    webviewView.webview.html = this.#getHtmlForWebview(webviewView.webview)
+  }
+
+  #getHtmlForWebview(webview: vscode.Webview) {
+    const { extensionUri } = this.#extContext
+
+    const baseUri = vscode.Uri.joinPath(extensionUri, './node_modules/@nicepkg/gpt-runner-web/dist/browser')
+
+    webview.options = {
+      enableScripts: true,
+      localResourceRoots: [baseUri],
+    }
+
+    const indexHtml = fs.readFileSync(path.join(baseUri.fsPath, 'index.html'), 'utf8')
+    const nonce = createHash()
+
+    const indexHtmlWithBaseUri = indexHtml.replace(
+      /\s+(href|src)="(.+?)"/g,
+      (_, attr, url) => ` ${attr}="${webview.asWebviewUri(vscode.Uri.joinPath(baseUri, url))}"`,
+    ).replace(/\/\/\s*before-script/g, `
+      window.globalConfig = {
+        serverBaseUrl: 'http://localhost:3003',
+        initialRoutePath: '/chat'
+      }
+    `).replace(/<script\s*([\w\W]+)?>/g,
+      (_, attr) => `<script ${attr} nonce="${nonce}">`,
+    )
+
+    console.log('indexHtmlWithBaseUri', indexHtmlWithBaseUri)
+
+    return indexHtmlWithBaseUri
+  }
+}
+
+export async function registerWebview(
+  cwd: string,
+  contextLoader: ContextLoader,
+  ext: ExtensionContext,
+) {
+  const provider = new ChatViewProvider(ext)
+  let webviewDisposer: vscode.Disposable | undefined
+
+  const registerProvider = () => {
+    webviewDisposer = vscode.window.registerWebviewViewProvider(ChatViewProvider.viewType, provider)
+    return webviewDisposer
+  }
+
+  ext.subscriptions.push(registerProvider())
+
+  contextLoader.emitter.on('contextReload', () => {
+    registerProvider()
+  })
+  contextLoader.emitter.on('contextUnload', () => {
+    webviewDisposer?.dispose?.()
+  })
+}
diff --git a/packages/gpt-runner-vscode/src/utils.ts b/packages/gpt-runner-vscode/src/utils.ts
@@ -30,3 +30,12 @@ export function getExtConfiguration(): ExtensionConfiguration {
     disabled,
   }
 }
+
+export function createHash() {
+  let text = ''
+  const possible = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789'
+  for (let i = 0; i < 32; i++)
+    text += possible.charAt(Math.floor(Math.random() * possible.length))
+
+  return text
+}
diff --git a/packages/gpt-runner-web/client/index.html b/packages/gpt-runner-web/client/index.html
@@ -5,12 +5,19 @@
   <meta charset="utf-8">
   <meta name="viewport" content="width=device-width,initial-scale=1,shrink-to-fit=no">
   <title>GPT Runner</title>
+  <script>
+    // before-script
+  </script>
+  <base href="/">
   <link href="/codicon/codicon.css" rel="stylesheet" />
 </head>
 
 <body>
   <div id="root"></div>
   <script type="module" src="/src/main.tsx"></script>
+  <script>
+    // after-script
+  </script>
 </body>
 
 </html>
diff --git a/packages/gpt-runner-web/client/src/helpers/global-config.ts b/packages/gpt-runner-web/client/src/helpers/global-config.ts
@@ -1,5 +1,17 @@
 import { getSearchParams } from '@nicepkg/gpt-runner-shared/browser'
 
+declare global {
+  interface Window {
+    globalConfig?: {
+      rootPath?: string
+      serverBaseUrl?: string
+      initialRoutePath?: string
+    }
+  }
+}
+
 export const globalConfig = {
-  rootPath: '/Users/yangxiaoming/Documents/codes/gpt-runner' || getSearchParams('rootPath'),
+  rootPath: window?.globalConfig?.rootPath || '/Users/yangxiaoming/Documents/codes/gpt-runner' || getSearchParams('rootPath'),
+  initialRoutePath: window?.globalConfig?.initialRoutePath || '/',
+  serverBaseUrl: window?.globalConfig?.serverBaseUrl || 'http://localhost:3003',
 }
diff --git a/packages/gpt-runner-web/client/src/networks/chatgpt.ts b/packages/gpt-runner-web/client/src/networks/chatgpt.ts
@@ -1,6 +1,7 @@
 import type { EventSourceMessage } from '@microsoft/fetch-event-source'
 import { fetchEventSource } from '@microsoft/fetch-event-source'
 import type { ChatStreamReqParams } from '@nicepkg/gpt-runner-shared/common'
+import { globalConfig } from '../helpers/global-config'
 
 export interface FetchChatStreamReqParams extends ChatStreamReqParams {
   namespace?: string
@@ -25,7 +26,7 @@ export async function fetchChatgptStream(
   } = params
 
   try {
-    await fetchEventSource('/api/chatgpt/chat-stream', {
+    await fetchEventSource(`${globalConfig.serverBaseUrl}/api/chatgpt/chat-stream`, {
       method: 'POST',
       signal,
       headers: {
diff --git a/packages/gpt-runner-web/client/src/networks/config.ts b/packages/gpt-runner-web/client/src/networks/config.ts
@@ -1,4 +1,5 @@
 import type { BaseResponse, GetUserConfigReqParams, GetUserConfigResData } from '@nicepkg/gpt-runner-shared/common'
+import { globalConfig } from '../helpers/global-config'
 
 export interface FetchUserConfigParams extends GetUserConfigReqParams {
 }
@@ -8,7 +9,7 @@ export type FetchUserConfigRes = BaseResponse<GetUserConfigResData>
 export async function fetchUserConfig(params: FetchUserConfigParams): Promise<FetchUserConfigRes> {
   const { rootPath } = params
 
-  const res = await fetch(`/api/config/user-config?rootPath=${rootPath}`, {
+  const res = await fetch(`${globalConfig.serverBaseUrl}/api/config/user-config?rootPath=${rootPath}`, {
     method: 'GET',
     headers: {
       'Content-Type': 'application/json',
diff --git a/packages/gpt-runner-web/client/src/networks/gpt-files.ts b/packages/gpt-runner-web/client/src/networks/gpt-files.ts
@@ -1,4 +1,5 @@
 import type { BaseResponse, GetGptFilesReqParams, GptFilesTreeResData } from '@nicepkg/gpt-runner-shared/common'
+import { globalConfig } from '../helpers/global-config'
 
 export interface FetchGptFilesTreeParams extends GetGptFilesReqParams {
 }
@@ -8,7 +9,7 @@ export type FetchGptFilesTreeRes = BaseResponse<GptFilesTreeResData>
 export async function fetchGptFilesTree(params: FetchGptFilesTreeParams): Promise<FetchGptFilesTreeRes> {
   const { rootPath } = params
 
-  const res = await fetch(`/api/gpt-files?rootPath=${rootPath}`, {
+  const res = await fetch(`${globalConfig.serverBaseUrl}/api/gpt-files?rootPath=${rootPath}`, {
     method: 'GET',
     headers: {
       'Content-Type': 'application/json',
diff --git a/packages/gpt-runner-web/client/src/networks/state.ts b/packages/gpt-runner-web/client/src/networks/state.ts
@@ -1,4 +1,5 @@
 import type { BaseResponse, GetStateReqParams, GetStateResData, SaveStateReqParams, SaveStateResData } from '@nicepkg/gpt-runner-shared/common'
+import { globalConfig } from '../helpers/global-config'
 
 export interface FetchStateParams extends GetStateReqParams {
 }
@@ -8,7 +9,7 @@ export type FetchStateRes = BaseResponse<GetStateResData>
 export async function fetchState(params: FetchStateParams): Promise<FetchStateRes> {
   const { key } = params
 
-  const res = await fetch(`/api/state?key=${key}`, {
+  const res = await fetch(`${globalConfig.serverBaseUrl}/api/state?key=${key}`, {
     method: 'GET',
     headers: {
       'Content-Type': 'application/json',
@@ -26,7 +27,7 @@ export type SaveStateRes = BaseResponse<SaveStateResData>
 export async function saveState(params: SaveStateParams): Promise<SaveStateRes> {
   const { key, state } = params
 
-  const res = await fetch('/api/state', {
+  const res = await fetch(`${globalConfig.serverBaseUrl}/api/state`, {
     method: 'POST',
     headers: {
       'Content-Type': 'application/json',
diff --git a/packages/gpt-runner-web/client/src/router.tsx b/packages/gpt-runner-web/client/src/router.tsx
diff --git a/packages/gpt-runner-web/package.json b/packages/gpt-runner-web/package.json
diff --git a/packages/gpt-runner-web/server/src/controllers/state.controller.ts b/packages/gpt-runner-web/server/src/controllers/state.controller.ts
diff --git a/packages/gpt-runner-web/server/src/helpers/get-cache-dir.ts b/packages/gpt-runner-web/server/src/helpers/get-cache-dir.ts
diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml

Original file line number	Diff line number	Diff line change
`@@ -30,3 +30,12 @@ export function getExtConfiguration(): ExtensionConfiguration {`
`30`	`30`	`disabled,`
`31`	`31`	`}`
`32`	`32`	`}`
	`33`	`+`
	`34`	`+export function createHash() {`
	`35`	`+ let text = ''`
	`36`	`+ const possible = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789'`
	`37`	`+ for (let i = 0; i < 32; i++)`
	`38`	`+ text += possible.charAt(Math.floor(Math.random() * possible.length))`
	`39`	`+`
	`40`	`+ return text`
	`41`	`+}`