feat: support to set the server arguments on desktop

LinuxSuRen · LinuxSuRen · commit cf62b369c692 · 2025-07-04T18:27:04.000+08:00
diff --git a/console/atest-desktop/index.html b/console/atest-desktop/index.html
@@ -9,19 +9,41 @@
 </head>
 <body>
 
-<div style="margin: 5px">
+<div style="margin: 5px; display: flex; justify-content: center">
   <div>
     <div>Server Status</div>
-    <button type="button" id="action">Start</button>
-    <button type="button" id="open-server-page">Open Server Page</button>
-    <div>
-      <span>Port:</span><input name="port" id="port" type="text"/>
-    </div>
-  </div>
-
-  <div>
-    <div>Log</div>
-    <button type="button" id="open-log-file">Open Log File</button>
+      <table>
+          <tr>
+              <td>
+                  <label for="port">Port</label>
+              </td>
+              <td>
+                  <input name="port" id="port" type="text"/>
+              </td>
+          </tr>
+          <tr>
+              <td>
+                  <label for="extension-registry">Extension Registry</label>
+              </td>
+              <td>
+                  <input name="extension-registry" id="extension-registry" type="text"/>
+              </td>
+          </tr>
+          <tr>
+              <td>Log</td>
+              <td>
+                  <button type="button" id="open-log-file">Open Log File</button>
+              </td>
+          </tr>
+          <tr>
+              <td>
+                  <button type="button" id="action">Start</button>
+              </td>
+              <td>
+                  <button type="button" id="open-server-page">Open Server Page</button>
+              </td>
+          </tr>
+      </table>
   </div>
 </div>
 
@@ -63,9 +85,11 @@
 window.setInterval(loadServerStatus, 2000)
 
 const portInput = document.getElementById('port');
+const extensionRegistry = document.getElementById('extension-registry');
 (async function() {
   portInput.value = await window.electronAPI.getPort()
+  extensionRegistry.value = await window.electronAPI.getExtensionRegistry()
 })();
 </script>
 </body>
-</html>
+</html>
diff --git a/console/atest-desktop/main.js b/console/atest-desktop/main.js
@@ -1,5 +1,5 @@
 /*
-Copyright 2024 API Testing Authors.
+Copyright 2024-2025 API Testing Authors.
 
 Licensed under the Apache License, Version 2.0 (the "License");
 you may not use this file except in compliance with the License.
@@ -117,6 +117,9 @@ menu.append(new MenuItem({
 Menu.setApplicationMenu(menu)
 
 let serverProcess;
+let serverPort;
+let setExtensionRegistry = "ghcr.io";
+
 // This method will be called when Electron has finished
 // initialization and is ready to create browser windows.
 // Some APIs can only be used after this event occurs.
@@ -129,6 +132,15 @@ app.whenReady().then(() => {
   ipcMain.on('control', (e, okCallback, errCallback) => {
     server.control(okCallback, errCallback)
   })
+  ipcMain.handle('setPort', (port) => {
+      serverPort = port;
+  })
+  ipcMain.handle('setExtensionRegistry', (registry) => {
+    setExtensionRegistry = registry
+  })
+  ipcMain.handle('getExtensionRegistry', () => {
+    return setExtensionRegistry
+  })
   ipcMain.handle('getHomePage', server.getHomePage)
   ipcMain.handle('getPort', () => {
     return server.getPort()
@@ -181,6 +193,7 @@ const startServer = () => {
     "server",
     "--http-port", server.getPort(),
     "--port=0",
+    `--extension-registry=${setExtensionRegistry}`,
     "--local-storage", path.join(homeData, "*.yaml")
   ])
   serverProcess.stdout.on('data', (data) => {
@@ -222,4 +235,4 @@ function getLogLevel() {
 }
 
 // In this file you can include the rest of your app's specific main process
-// code. You can also put them in separate files and require them here.
+// code. You can also put them in separate files and require them here.
diff --git a/console/atest-desktop/preload.js b/console/atest-desktop/preload.js
@@ -36,5 +36,8 @@ contextBridge.exposeInMainWorld('electronAPI', {
   control: (okCallback, errCallback) => ipcRenderer.send('control', okCallback, errCallback),
   getHomePage: () => ipcRenderer.invoke('getHomePage'),
   getPort: () => ipcRenderer.invoke('getPort'),
+  setPort: (port) => ipcRenderer.send('setPort', port),
+  setExtensionRegistry: (registry) => ipcRenderer.send('setExtensionRegistry', registry),
+  getExtensionRegistry: () => ipcRenderer.invoke('getExtensionRegistry'),
   getHealthzUrl: () => ipcRenderer.invoke('getHealthzUrl'),
 })
