Merge pull request #589 from devchat-ai/auto_custom_workflow

Auto custom workflow

Author: yangbobo2021

gui

@@ -208,13 +208,16 @@ export function registerInstallCommandsCommand(
       if (!fs.existsSync(sysDirPath)) {
         // Directory does not exist, wait for updateWorkflows to finish
         await dcClient.updateWorkflows();
+        await dcClient.updateCustomWorkflows();
         sendCommandListByDevChatRun();
       } else {
         // Directory exists, execute sendCommandListByDevChatRun immediately
         await sendCommandListByDevChatRun();
 
         // Then asynchronously execute updateWorkflows
         await dcClient.updateWorkflows();
+        await dcClient.updateCustomWorkflows();
+        
         await sendCommandListByDevChatRun();
       }
     }

src/contributes/commands.ts

@@ -82,3 +82,5 @@ messageHandler.registerHandler('getIDEServicePort', getIDEServicePort);
 
 messageHandler.registerHandler('readServerConfigBase', readServerConfigBase);
 messageHandler.registerHandler('writeServerConfigBase', writeServerConfigBase);
+
+

src/handler/handlerRegister.ts

@@ -57,4 +57,4 @@ export async function sendCommandListByDevChatRun() {
     if (existPannel) {
         await getWorkflowCommandList({}, existPannel!);
     }
-}
+}

src/handler/workflowCommandHandler.ts

@@ -7,3 +7,10 @@ export async function getServicePort() {
     // return await UiUtilWrapper.getLSPBrigePort();
     return process.env.DEVCHAT_IDE_SERVICE_PORT;
 }
+
+export async function getLocalServicePort() {
+    logger
+        .channel()
+        ?.info(`get local service port: ${process.env.DC_LOCALSERVICE_PORT}`);
+    return process.env.DC_LOCALSERVICE_PORT;
+}

src/ide_services/endpoints/getServicePort.ts

@@ -3,7 +3,7 @@ import * as http from "http";
 import * as querystring from "querystring";
 import { logger } from "../util/logger";
 
-import { getServicePort } from "./endpoints/getServicePort";
+import { getServicePort, getLocalServicePort } from "./endpoints/getServicePort";
 import { installPythonEnv } from "./endpoints/installPythonEnv";
 import { ideLogging} from "./endpoints/ideLogging";
 import { updateSlashCommands } from "./endpoints/updateSlashCommands";
@@ -26,6 +26,10 @@ const functionRegistry: any = {
         keys: [],
         handler: getServicePort,
     },
+    "/get_local_service_port": {
+        keys: [],
+        handler: getLocalServicePort,
+    },
     "/install_python_env": {
         keys: ["command_name", "requirements_file"],
         handler: installPythonEnv,

src/ide_services/services.ts

@@ -229,6 +229,19 @@ export class DevChatClient {
             );
     }
 
+    @timeThis
+    @catchAndReturn(undefined)
+    async updateCustomWorkflows(): Promise<void> {
+        const response = await this._post("/workflows/custom_update");
+        logger
+            .channel()
+            ?.trace(
+                `updateCustomWorkflows response data: \n${JSON.stringify(
+                    response.data
+                )}`
+            );
+    }
+
     @timeThis
     async message(
         message: ChatRequest,