kb-server 0.0.1-beta.21 → 0.0.1-beta.23

package/dist/common/create-errors.d.ts

@@ -21,6 +21,7 @@ declare const CommonErrorsMap: {
         readonly EmptyParameter: "请求参数不能为空";
         readonly EmptyAPIRequest: "未指定API的请求";
         readonly ValidationError: "参数校验失败";
+        readonly RouteError: "路由错误";
     };
     readonly ResourceNotFound: {
         readonly APINotFound: "不存在的API";
@@ -77,6 +78,7 @@ declare function createErrorsClasses<T extends MessageMap = MessageMap>(messageM
         readonly EmptyParameter: "请求参数不能为空";
         readonly EmptyAPIRequest: "未指定API的请求";
         readonly ValidationError: "参数校验失败";
+        readonly RouteError: "路由错误";
     };
     readonly ResourceNotFound: {
         readonly APINotFound: "不存在的API";
@@ -102,6 +104,7 @@ export declare const CommonErrors: ErrorClasses<MergeMessageMap<{
         readonly EmptyParameter: "请求参数不能为空";
         readonly EmptyAPIRequest: "未指定API的请求";
         readonly ValidationError: "参数校验失败";
+        readonly RouteError: "路由错误";
     };
     readonly ResourceNotFound: {
         readonly APINotFound: "不存在的API";

package/dist/common/create-errors.js

@@ -23,6 +23,7 @@ const CommonErrorsMap = {
         EmptyParameter: "请求参数不能为空",
         EmptyAPIRequest: "未指定API的请求",
         ValidationError: "参数校验失败",
+        RouteError: "路由错误",
     },
     ResourceNotFound: {
         APINotFound: "不存在的API",

package/dist/common/create-server.d.ts

@@ -1,7 +1,7 @@
 import express from "express";
 import { AuthFunction } from "./api-middleware";
 import { APIs } from "./create-api";
-import { SSEHandlers } from "./sse-middleware";
+import { SseHandlers } from "./create-sse";
 export interface ICreateServerParams {
     /**
      * API列表
@@ -11,7 +11,17 @@ export interface ICreateServerParams {
      * SSE配置
      */
     sse?: {
-        handlers: SSEHandlers;
+        /**
+         * SSE API列表
+         */
+        handlers: SseHandlers;
+        /**
+         * SSE 监听的路由，例如：`/sse/generate`
+         *
+         * 注意：路由要以 `/` 开头
+         *
+         * 默认为：`/sse`
+         */
         route?: string;
     };
     /**

package/dist/common/create-sse.d.ts

@@ -0,0 +1,23 @@
+import { Class } from "utility-types";
+import { ServerContext } from "./create-api";
+export type API<P, R> = (param: P) => Promise<R>;
+export type SseExecution<P, R, A> = (
+/** 请求入参 */
+params: P, 
+/** 请求上下文 */
+ctx: ServerContext<A>, 
+/**
+ * 操作函数
+ */
+operations: {
+    /**
+     * 向客户端推送消息
+     */
+    send: (event: string, message: string) => void;
+}) => Promise<R>;
+export type AnySseExecution = SseExecution<any, any, any>;
+export type SseHandlers = Record<string, AnySseExecution>;
+export declare function createSseAPI<P, R, A>(ParamsClass: Class<P>, execution: SseExecution<P, R, A>): API<P, R>;
+type StubParam<T> = T extends (params: infer P) => any ? P : never;
+export declare function implementSseAPI<F extends (params: any) => any, A = Record<string, any>>(_actionStub: F, ParamsClass: Class<StubParam<F>>, execution: SseExecution<StubParam<F>, ReturnType<F>, A>): API<StubParam<F>, ReturnType<F>>;
+export {};

package/dist/common/create-sse.js

@@ -0,0 +1,31 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.implementSseAPI = exports.createSseAPI = void 0;
+const class_transformer_1 = require("class-transformer");
+const class_validator_1 = require("class-validator");
+const create_errors_1 = require("./create-errors");
+const create_api_1 = require("./create-api");
+function createSseAPI(ParamsClass, execution) {
+    const runtime = async (params, ctx, operations) => {
+        if (!params) {
+            throw new create_errors_1.CommonErrors.InvalidParameter.EmptyParameter();
+        }
+        // 校验参数
+        const paramsToCheck = params instanceof ParamsClass
+            ? params
+            : (0, class_transformer_1.plainToClass)(ParamsClass, params);
+        const errors = await (0, class_validator_1.validate)(paramsToCheck || {});
+        if (errors?.length) {
+            const errorMessages = errors.flatMap((error) => (0, create_api_1.getMessage)(error));
+            throw new create_errors_1.CommonErrors.InvalidParameter.ValidationError(errorMessages.join("\n"));
+        }
+        // 执行函数
+        return await execution(params, ctx, operations);
+    };
+    return runtime;
+}
+exports.createSseAPI = createSseAPI;
+function implementSseAPI(_actionStub, ParamsClass, execution) {
+    return createSseAPI(ParamsClass, execution);
+}
+exports.implementSseAPI = implementSseAPI;

package/dist/common/sse-middleware.d.ts

@@ -1,25 +1,11 @@
 import express from "express";
-import { ServerContext } from "./create-api";
 import { AuthFunction } from "./api-middleware";
-export type SseExecution<P, R, A> = (
-/** 请求入参 */
-params: P, 
-/** 请求上下文 */
-ctx: ServerContext<A>, 
-/**
- * 允许执行的操作
- */
-operations: {
-    /** 发送消息 */
-    send: (event: string, message: string) => void;
-}) => Promise<R>;
-export type AnySseExecution = SseExecution<any, any, any>;
-export type SSEHandlers = Record<string, AnySseExecution>;
+import { SseHandlers } from "./create-sse";
 interface IPackSSEOptions {
     authFn?: AuthFunction;
     route?: string;
     log?: boolean;
 }
-export declare const packSSE: (sseHandlers: SSEHandlers, options?: IPackSSEOptions) => express.RequestHandler;
+export declare const packSSE: (sseHandlers: SseHandlers, options?: IPackSSEOptions) => express.RequestHandler;
 export declare const createSSEMessage: (event: string, message: string) => string;
 export {};

package/dist/common/sse-middleware.js

@@ -6,14 +6,20 @@ const logger_1 = require("../helper/logger");
 const create_errors_1 = require("./create-errors");
 const short_id_1 = require("../helper/short-id");
 const packSSE = (sseHandlers, options) => {
-    const sseMap = new Map(Object.entries(sseHandlers).map(([action, execution]) => [
-        action,
-        execution,
-    ]));
     const { authFn, log = true, route = "/sse" } = options || {};
     return async (req, res, next) => {
+        // 生成SSE API映射
+        const sseMap = new Map(Object.entries(sseHandlers).map(([action, execution]) => [
+            action,
+            execution,
+        ]));
+        // 检查SSE路由
+        const isValidRoute = route.startsWith("/");
+        if (!isValidRoute) {
+            throw new create_errors_1.CommonErrors.InvalidParameter.RouteError();
+        }
         const { path } = req;
-        // 路径不是SSE的路径
+        // 非SSE的请求直接放过
         if (path !== route) {
             next();
             return;
@@ -56,17 +62,7 @@ const packSSE = (sseHandlers, options) => {
             if (typeof execution !== "function") {
                 throw new create_errors_1.CommonErrors.ResourceNotFound.APINotFound();
             }
-            /**
-             * 开始处理SSE请求就要统计
-             *
-             * 请求处理完毕的时间
-             *
-             * 不要让 API 随意改写 req res 里面的东西
-             *
-             * 统一管理请求头
-             *
-             */
-            // 处理SSE
+            // 写请求头
             res.writeHead(200, {
                 "Content-Type": "text/event-stream",
                 "Cache-Control": "no-cache",
@@ -75,7 +71,7 @@ const packSSE = (sseHandlers, options) => {
             const send = (event, message) => {
                 const response = (0, exports.createSSEMessage)(event, message);
                 res.write(response);
-                logger_1.logger.log("info", `发送消息：${response}`);
+                logger_1.logger.log("info", `发送消息：\n${response}`);
             };
             await execution(params, ctx, { send });
             // 完成响应
@@ -106,7 +102,7 @@ const packSSE = (sseHandlers, options) => {
             }));
             // 完成响应
             took = Date.now() - start;
-            logger_1.logger.log("info", `响应：${response}`);
+            logger_1.logger.log("info", `发送消息：\n${response}`);
             logger_1.logger.log("info", `耗时：${took} ms - RequestId: ${requestId}`);
             res.write(response);
             return res.end();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "kb-server",
-  "version": "0.0.1-beta.21",
+  "version": "0.0.1-beta.23",
   "description": "A fast server for Node.JS,made by express.",
   "main": "dist/index.js",
   "scripts": {