@lark-apaas/nestjs-capability 0.0.1-alpha.4 → 0.0.1-alpha.6

package/dist/index.cjs

@@ -36,6 +36,7 @@ __export(index_exports, {
   CapabilityNotFoundError: () => CapabilityNotFoundError,
   CapabilityService: () => CapabilityService,
   DebugController: () => DebugController,
+  ErrorCodes: () => ErrorCodes,
   PluginLoadError: () => PluginLoadError,
   PluginLoaderService: () => PluginLoaderService,
   PluginNotFoundError: () => PluginNotFoundError,
@@ -44,6 +45,22 @@ __export(index_exports, {
 });
 module.exports = __toCommonJS(index_exports);
 
+// src/interfaces/error-codes.ts
+var ErrorCodes = {
+  /** 成功 */
+  SUCCESS: "0",
+  /** 能力不存在 */
+  CAPABILITY_NOT_FOUND: "k_ec_cap_001",
+  /** 插件不存在 */
+  PLUGIN_NOT_FOUND: "k_ec_cap_002",
+  /** Action 不存在 */
+  ACTION_NOT_FOUND: "k_ec_cap_003",
+  /** 参数验证失败 */
+  PARAMS_VALIDATION_ERROR: "k_ec_cap_004",
+  /** 执行失败 */
+  EXECUTION_ERROR: "k_ec_cap_005"
+};
+
 // src/services/template-engine.service.ts
 var import_common = require("@nestjs/common");
 function _ts_decorate(decorators, target, key, desc) {
@@ -311,6 +328,9 @@ var CapabilityService = class _CapabilityService {
       callStream: /* @__PURE__ */ __name((actionName, input, contextOverride) => {
         return this.executeCallStream(config, actionName, input, contextOverride);
       }, "callStream"),
+      callStreamWithEvents: /* @__PURE__ */ __name((actionName, input, contextOverride) => {
+        return this.executeCallStreamWithEvents(config, actionName, input, contextOverride);
+      }, "callStreamWithEvents"),
       isStream: /* @__PURE__ */ __name(async (actionName) => {
         return this.checkIsStream(config, actionName);
       }, "isStream")
@@ -421,6 +441,79 @@ var CapabilityService = class _CapabilityService {
       throw error;
     }
   }
+  /**
+  * 流式执行 capability，返回带事件协议的流
+  * - 优先使用 pluginInstance.runStreamWithEvents
+  * - 如果插件不支持，则包装 runStream/run 为 StreamEvent
+  */
+  async *executeCallStreamWithEvents(config, actionName, input, contextOverride) {
+    const startTime = Date.now();
+    let chunkCount = 0;
+    try {
+      const pluginInstance = await this.pluginLoaderService.loadPlugin(config.pluginKey);
+      if (!pluginInstance.hasAction(actionName)) {
+        throw new ActionNotFoundError(config.pluginKey, actionName);
+      }
+      const resolvedParams = this.templateEngineService.resolve(config.formValue, input);
+      const context = this.buildActionContext(contextOverride);
+      this.logger.log({
+        message: "Executing capability (streamWithEvents)",
+        capabilityId: config.id,
+        action: actionName,
+        pluginKey: config.pluginKey
+      });
+      if (pluginInstance.runStreamWithEvents) {
+        yield* pluginInstance.runStreamWithEvents(actionName, context, resolvedParams);
+      } else {
+        const isStream = pluginInstance.isStreamAction?.(actionName) ?? false;
+        if (isStream && pluginInstance.runStream) {
+          for await (const chunk of pluginInstance.runStream(actionName, context, resolvedParams)) {
+            chunkCount++;
+            yield {
+              type: "data",
+              data: chunk
+            };
+          }
+        } else {
+          const result = await pluginInstance.run(actionName, context, resolvedParams);
+          chunkCount = 1;
+          yield {
+            type: "data",
+            data: result
+          };
+        }
+        yield {
+          type: "done",
+          metadata: {
+            chunks: chunkCount,
+            duration: Date.now() - startTime
+          }
+        };
+      }
+      this.logger.log({
+        message: "Capability streamWithEvents completed",
+        capabilityId: config.id,
+        action: actionName,
+        duration: Date.now() - startTime,
+        chunks: chunkCount
+      });
+    } catch (error) {
+      this.logger.error({
+        message: "Capability streamWithEvents execution failed",
+        capabilityId: config.id,
+        action: actionName,
+        error: error instanceof Error ? error.message : String(error),
+        duration: Date.now() - startTime
+      });
+      yield {
+        type: "error",
+        error: {
+          code: "EXECUTION_ERROR",
+          message: error instanceof Error ? error.message : String(error)
+        }
+      };
+    }
+  }
   buildActionContext(override) {
     return {
       logger: this.logger,
@@ -482,14 +575,15 @@ var DebugController = class {
   list() {
     const capabilities = this.capabilityService.listCapabilities();
     return {
-      code: 0,
-      message: "success",
-      data: capabilities.map((c) => ({
-        id: c.id,
-        name: c.name,
-        pluginKey: c.pluginKey,
-        pluginVersion: c.pluginVersion
-      }))
+      status_code: ErrorCodes.SUCCESS,
+      data: {
+        capabilities: capabilities.map((c) => ({
+          id: c.id,
+          name: c.name,
+          pluginID: c.pluginKey,
+          pluginVersion: c.pluginVersion
+        }))
+      }
     };
   }
   /**
@@ -538,63 +632,40 @@ var DebugController = class {
     try {
       const result = await this.capabilityService.loadWithConfig(config).call(action, params);
       return {
-        code: 0,
-        message: "success",
-        data: result,
-        debug: {
-          capabilityConfig: config,
-          resolvedParams,
-          duration: Date.now() - startTime,
-          pluginKey: config.pluginKey,
-          action
+        status_code: ErrorCodes.SUCCESS,
+        data: {
+          output: result,
+          debug: {
+            capabilityConfig: config,
+            resolvedParams,
+            duration: Date.now() - startTime,
+            pluginID: config.pluginKey,
+            action
+          }
         }
       };
     } catch (error) {
-      const duration = Date.now() - startTime;
       if (error instanceof CapabilityNotFoundError) {
         throw new import_common4.HttpException({
-          code: 1,
-          message: error.message,
-          error: "CAPABILITY_NOT_FOUND",
-          debug: {
-            duration
-          }
+          status_code: ErrorCodes.CAPABILITY_NOT_FOUND,
+          error_msg: `Capability not found: ${capabilityId}`
         }, import_common4.HttpStatus.NOT_FOUND);
       }
       if (error instanceof PluginNotFoundError) {
         throw new import_common4.HttpException({
-          code: 1,
-          message: error.message,
-          error: "PLUGIN_NOT_FOUND",
-          debug: {
-            duration,
-            pluginKey: config.pluginKey,
-            action
-          }
+          status_code: ErrorCodes.PLUGIN_NOT_FOUND,
+          error_msg: `Plugin not found: ${config.pluginKey}`
         }, import_common4.HttpStatus.INTERNAL_SERVER_ERROR);
       }
       if (error instanceof ActionNotFoundError) {
         throw new import_common4.HttpException({
-          code: 1,
-          message: error.message,
-          error: "ACTION_NOT_FOUND",
-          debug: {
-            duration,
-            pluginKey: config.pluginKey,
-            action
-          }
+          status_code: ErrorCodes.ACTION_NOT_FOUND,
+          error_msg: `Action '${action}' not found in plugin ${config.pluginKey}`
         }, import_common4.HttpStatus.BAD_REQUEST);
       }
       throw new import_common4.HttpException({
-        code: 1,
-        message: error instanceof Error ? error.message : String(error),
-        error: "EXECUTION_ERROR",
-        debug: {
-          duration,
-          pluginKey: config.pluginKey,
-          action,
-          resolvedParams
-        }
+        status_code: ErrorCodes.EXECUTION_ERROR,
+        error_msg: `Execution failed: ${error instanceof Error ? error.message : String(error)}`
       }, import_common4.HttpStatus.INTERNAL_SERVER_ERROR);
     }
   }
@@ -607,33 +678,76 @@ var DebugController = class {
       const config = this.getCapabilityConfig(capabilityId, body.capability);
       const action = await this.getActionName(config.pluginKey, body.action);
       const capability = this.capabilityService.loadWithConfig(config);
-      const stream = capability.callStream(action, params);
-      for await (const chunk of stream) {
-        res.write(`data: ${JSON.stringify(chunk)}
+      const eventStream = capability.callStreamWithEvents(action, params);
+      for await (const event of eventStream) {
+        switch (event.type) {
+          case "data": {
+            const response = {
+              status_code: ErrorCodes.SUCCESS,
+              data: {
+                type: "content",
+                delta: {
+                  content: event.data
+                }
+              }
+            };
+            res.write(`data: ${JSON.stringify(response)}
+
+`);
+            break;
+          }
+          case "done": {
+            const response = {
+              status_code: ErrorCodes.SUCCESS,
+              data: {
+                type: "content",
+                delta: {
+                  content: null
+                },
+                finished: true
+              }
+            };
+            res.write(`data: ${JSON.stringify(response)}
+
+`);
+            res.end();
+            return;
+          }
+          case "error": {
+            const response = {
+              status_code: ErrorCodes.SUCCESS,
+              data: {
+                type: "error",
+                error: {
+                  code: 0,
+                  message: event.error.message
+                }
+              }
+            };
+            res.write(`data: ${JSON.stringify(response)}
 
 `);
+            res.end();
+            return;
+          }
+        }
       }
-      res.write("data: [DONE]\n\n");
+      res.end();
     } catch (error) {
-      const message = error instanceof Error ? error.message : String(error);
-      let errorCode = "EXECUTION_ERROR";
-      if (error instanceof CapabilityNotFoundError) {
-        errorCode = "CAPABILITY_NOT_FOUND";
-      } else if (error instanceof PluginNotFoundError) {
-        errorCode = "PLUGIN_NOT_FOUND";
-      } else if (error instanceof ActionNotFoundError) {
-        errorCode = "ACTION_NOT_FOUND";
-      } else if (error instanceof import_common4.HttpException) {
-        const response = error.getResponse();
-        errorCode = response.error ?? "EXECUTION_ERROR";
-      }
-      res.write(`data: ${JSON.stringify({
-        error: message,
-        code: errorCode
-      })}
+      const errorMsg = error instanceof Error ? error.message : String(error);
+      const response = {
+        status_code: ErrorCodes.SUCCESS,
+        data: {
+          type: "error",
+          error: {
+            code: 0,
+            message: errorMsg
+          }
+        }
+      };
+      res.write(`data: ${JSON.stringify(response)}
 
 `);
-    } finally {
       res.end();
     }
   }
@@ -642,7 +756,7 @@ _ts_decorate4([
   (0, import_common4.Get)("list"),
   _ts_metadata2("design:type", Function),
   _ts_metadata2("design:paramtypes", []),
-  _ts_metadata2("design:returntype", typeof ListResponse === "undefined" ? Object : ListResponse)
+  _ts_metadata2("design:returntype", typeof SuccessResponse === "undefined" ? Object : SuccessResponse)
 ], DebugController.prototype, "list", null);
 _ts_decorate4([
   (0, import_common4.Post)("debug/:capability_id"),
@@ -708,51 +822,48 @@ var WebhookController = class {
   list() {
     const capabilities = this.capabilityService.listCapabilities();
     return {
-      code: 0,
-      message: "success",
-      data: capabilities.map((c) => ({
-        id: c.id,
-        name: c.name,
-        description: c.description,
-        pluginKey: c.pluginKey,
-        pluginVersion: c.pluginVersion
-      }))
+      status_code: ErrorCodes.SUCCESS,
+      data: {
+        capabilities: capabilities.map((c) => ({
+          id: c.id,
+          name: c.name,
+          pluginID: c.pluginKey,
+          pluginVersion: c.pluginVersion
+        }))
+      }
     };
   }
   async execute(capabilityId, body) {
     try {
       const result = await this.capabilityService.load(capabilityId).call(body.action, body.params);
       return {
-        code: 0,
-        message: "success",
-        data: result
+        status_code: ErrorCodes.SUCCESS,
+        data: {
+          output: result
+        }
       };
     } catch (error) {
       if (error instanceof CapabilityNotFoundError) {
         throw new import_common5.HttpException({
-          code: 1,
-          message: error.message,
-          error: "CAPABILITY_NOT_FOUND"
+          status_code: ErrorCodes.CAPABILITY_NOT_FOUND,
+          error_msg: `Capability not found: ${capabilityId}`
         }, import_common5.HttpStatus.NOT_FOUND);
       }
       if (error instanceof PluginNotFoundError) {
         throw new import_common5.HttpException({
-          code: 1,
-          message: error.message,
-          error: "PLUGIN_NOT_FOUND"
+          status_code: ErrorCodes.PLUGIN_NOT_FOUND,
+          error_msg: `Plugin not found`
         }, import_common5.HttpStatus.INTERNAL_SERVER_ERROR);
       }
       if (error instanceof ActionNotFoundError) {
         throw new import_common5.HttpException({
-          code: 1,
-          message: error.message,
-          error: "ACTION_NOT_FOUND"
+          status_code: ErrorCodes.ACTION_NOT_FOUND,
+          error_msg: `Action '${body.action}' not found`
         }, import_common5.HttpStatus.BAD_REQUEST);
       }
       throw new import_common5.HttpException({
-        code: 1,
-        message: error instanceof Error ? error.message : String(error),
-        error: "EXECUTION_ERROR"
+        status_code: ErrorCodes.EXECUTION_ERROR,
+        error_msg: `Execution failed: ${error instanceof Error ? error.message : String(error)}`
       }, import_common5.HttpStatus.INTERNAL_SERVER_ERROR);
     }
   }
@@ -762,30 +873,76 @@ var WebhookController = class {
     res.setHeader("Connection", "keep-alive");
     try {
       const capability = this.capabilityService.load(capabilityId);
-      const stream = capability.callStream(body.action, body.params);
-      for await (const chunk of stream) {
-        res.write(`data: ${JSON.stringify(chunk)}
+      const eventStream = capability.callStreamWithEvents(body.action, body.params);
+      for await (const event of eventStream) {
+        switch (event.type) {
+          case "data": {
+            const response = {
+              status_code: ErrorCodes.SUCCESS,
+              data: {
+                type: "content",
+                delta: {
+                  content: event.data
+                }
+              }
+            };
+            res.write(`data: ${JSON.stringify(response)}
+
+`);
+            break;
+          }
+          case "done": {
+            const response = {
+              status_code: ErrorCodes.SUCCESS,
+              data: {
+                type: "content",
+                delta: {
+                  content: null
+                },
+                finished: true
+              }
+            };
+            res.write(`data: ${JSON.stringify(response)}
+
+`);
+            res.end();
+            return;
+          }
+          case "error": {
+            const response = {
+              status_code: ErrorCodes.SUCCESS,
+              data: {
+                type: "error",
+                error: {
+                  code: 0,
+                  message: event.error.message
+                }
+              }
+            };
+            res.write(`data: ${JSON.stringify(response)}
 
 `);
+            res.end();
+            return;
+          }
+        }
       }
-      res.write("data: [DONE]\n\n");
+      res.end();
     } catch (error) {
-      const message = error instanceof Error ? error.message : String(error);
-      let errorCode = "EXECUTION_ERROR";
-      if (error instanceof CapabilityNotFoundError) {
-        errorCode = "CAPABILITY_NOT_FOUND";
-      } else if (error instanceof PluginNotFoundError) {
-        errorCode = "PLUGIN_NOT_FOUND";
-      } else if (error instanceof ActionNotFoundError) {
-        errorCode = "ACTION_NOT_FOUND";
-      }
-      res.write(`data: ${JSON.stringify({
-        error: message,
-        code: errorCode
-      })}
+      const errorMsg = error instanceof Error ? error.message : String(error);
+      const response = {
+        status_code: ErrorCodes.SUCCESS,
+        data: {
+          type: "error",
+          error: {
+            code: 0,
+            message: errorMsg
+          }
+        }
+      };
+      res.write(`data: ${JSON.stringify(response)}
 
 `);
-    } finally {
       res.end();
     }
   }
@@ -794,7 +951,7 @@ _ts_decorate5([
   (0, import_common5.Get)("list"),
   _ts_metadata3("design:type", Function),
   _ts_metadata3("design:paramtypes", []),
-  _ts_metadata3("design:returntype", typeof ListResponse === "undefined" ? Object : ListResponse)
+  _ts_metadata3("design:returntype", typeof SuccessResponse === "undefined" ? Object : SuccessResponse)
 ], WebhookController.prototype, "list", null);
 _ts_decorate5([
   (0, import_common5.Post)(":capability_id"),
@@ -909,6 +1066,7 @@ CapabilityModule = _ts_decorate6([
   CapabilityNotFoundError,
   CapabilityService,
   DebugController,
+  ErrorCodes,
   PluginLoadError,
   PluginLoaderService,
   PluginNotFoundError,