@kevisual/router 0.0.82 → 0.0.84

package/dist/router.d.ts

@@ -32,12 +32,20 @@ declare class CustomError extends Error {
      * @returns
      */
     static isError(error: unknown): error is CustomError;
+    static throw(code?: number | string, message?: string): void;
+    static throw(code?: number | string, opts?: CustomErrorOptions): void;
+    static throw(opts?: CustomErrorOptions): void;
     parse(e?: CustomError): {
         code: number;
         data: any;
         message: string;
     };
 }
+interface throwError {
+    throw(code?: number | string, message?: string): void;
+    throw(code?: number | string, opts?: CustomErrorOptions): void;
+    throw(opts?: CustomErrorOptions): void;
+}
 
 declare class MockProcess {
     emitter?: EventEmitter;
@@ -72,9 +80,20 @@ type RouterContextT = {
     code?: number;
     [key: string]: any;
 };
+type BuildRouteContext<M, U> = M extends {
+    args?: infer A;
+} ? A extends z.ZodObject<any> ? RouteContext<{
+    args?: z.infer<A>;
+}, U> : A extends Record<string, z.ZodTypeAny> ? RouteContext<{
+    args?: {
+        [K in keyof A]: z.infer<A[K]>;
+    };
+}, U> : RouteContext<U> : RouteContext<U>;
 type RouteContext<T = {
     code?: number;
-}, S = any> = {
+}, U extends SimpleObject$1 = {}, S = {
+    [key: string]: any;
+}> = {
     /**
      * 本地自己调用的时候使用，可以标识为当前自调用，那么 auth 就不许重复的校验
      * 或者不需要登录的，直接调用
@@ -97,7 +116,14 @@ type RouteContext<T = {
     code?: number;
     /** return msg */
     message?: string;
+    /**
+     * 传递状态
+     */
     state?: S;
+    /**
+     * 当前routerId
+     */
+    currentId?: string;
     /**
      * 当前路径
      */
@@ -138,14 +164,12 @@ type RouteContext<T = {
         path: string;
         key?: string;
         payload?: any;
-    }, ctx?: RouteContext & {
-        [key: string]: any;
-    }) => Promise<any>;
+    }, ctx?: RouteContext) => Promise<any>;
     index?: number;
-    throw?: (code?: number | string, message?: string, tips?: string) => void;
+    throw?: throwError['throw'];
     /** 是否需要序列化, 使用JSON.stringify和JSON.parse */
     needSerialize?: boolean;
-} & T;
+} & T & U;
 type SimpleObject$1 = Record<string, any>;
 type Run<T extends SimpleObject$1 = {}> = (ctx: Required<RouteContext<T>>) => Promise<typeof ctx | null | void>;
 type RunMessage = {
@@ -156,7 +180,7 @@ type RunMessage = {
 };
 type NextRoute = Pick<Route, 'id' | 'path' | 'key'>;
 type RouteMiddleware = {
-    path: string;
+    path?: string;
     key?: string;
     id?: string;
 } | string;
@@ -197,9 +221,11 @@ declare const tool: {
 /** */
 declare const createSkill: <T = SimpleObject$1>(skill: Skill<T>) => Skill<T>;
 type RouteInfo = Pick<Route, (typeof pickValue)[number]>;
-declare class Route<U = {
-    [key: string]: any;
-}, T extends SimpleObject$1 = SimpleObject$1> {
+/**
+ * @M 是 route的 metadate的类型，默认是 SimpleObject
+ * @U 是 RouteContext 里 state的类型
+ */
+declare class Route<M extends SimpleObject$1 = SimpleObject$1, U extends SimpleObject$1 = SimpleObject$1> implements throwError {
     /**
      * 一级路径
      */
@@ -209,10 +235,10 @@ declare class Route<U = {
      */
     key?: string;
     id?: string;
-    run?: Run;
+    run?: Run<BuildRouteContext<M, U>>;
     nextRoute?: NextRoute;
     description?: string;
-    metadata?: T;
+    metadata?: M;
     middleware?: RouteMiddleware[];
     type?: string;
     /**
@@ -227,19 +253,19 @@ declare class Route<U = {
     } = RouterContextT>(opts: DefineRouteOpts): this;
     define<T extends {
         [key: string]: any;
-    } = RouterContextT>(fn: Run<T & U>): this;
+    } = RouterContextT>(fn: Run<T & BuildRouteContext<M, U>>): this;
     define<T extends {
         [key: string]: any;
-    } = RouterContextT>(key: string, fn: Run<T & U>): this;
+    } = RouterContextT>(key: string, fn: Run<T & BuildRouteContext<M, U>>): this;
     define<T extends {
         [key: string]: any;
-    } = RouterContextT>(path: string, key: string, fn: Run<T & U>): this;
+    } = RouterContextT>(path: string, key: string, fn: Run<T & BuildRouteContext<M, U>>): this;
     update(opts: DefineRouteOpts, onlyUpdateList?: string[]): this;
     addTo(router: QueryRouter | {
         add: (route: Route) => void;
         [key: string]: any;
     }, opts?: AddOpts): void;
-    throw(code?: number | string, message?: string, tips?: string): void;
+    throw(...args: any[]): void;
 }
 declare const toJSONSchema: (args: any, opts?: {
     mergeObject?: boolean;
@@ -264,11 +290,11 @@ declare const fromJSONSchema: <Merge extends boolean = false>(args?: any, opts?:
 type AddOpts = {
     overwrite?: boolean;
 };
-declare class QueryRouter {
+declare class QueryRouter<T extends SimpleObject$1 = SimpleObject$1> implements throwError {
     appId: string;
     routes: Route[];
     maxNextRoute: number;
-    context?: RouteContext;
+    context?: RouteContext<T>;
     constructor();
     /**
      * add route
@@ -296,7 +322,7 @@ declare class QueryRouter {
      * @param ctx
      * @returns
      */
-    runRoute(path: string, key: string, ctx?: RouteContext): any;
+    runRoute(path: string, key: string, ctx?: RouteContext<T>): Promise<RouteContext<T>>;
     /**
      * 第一次执行
      * @param message
@@ -307,9 +333,11 @@ declare class QueryRouter {
         path: string;
         key?: string;
         payload?: any;
-    }, ctx?: RouteContext & {
+    }, ctx?: RouteContext<T> & {
         [key: string]: any;
-    }): Promise<any>;
+    }): Promise<RouteContext<T, {}, {
+        [key: string]: any;
+    }>>;
     /**
      * 返回的数据包含所有的context的请求返回的内容，可做其他处理
      * @param message
@@ -321,9 +349,15 @@ declare class QueryRouter {
         path?: string;
         key?: string;
         payload?: any;
-    }, ctx?: RouteContext & {
+    }, ctx?: RouteContext<T> & {
         [key: string]: any;
-    }): Promise<any>;
+    }): Promise<RouteContext<T, {}, {
+        [key: string]: any;
+    }> | {
+        code: number;
+        body: any;
+        message: string;
+    }>;
     /**
      * 请求 result 的数据
      * @param message
@@ -339,9 +373,9 @@ declare class QueryRouter {
     }, ctx?: RouteContext & {
         [key: string]: any;
     }): Promise<{
-        code: any;
+        code: number;
         data: any;
-        message: any;
+        message: string;
     }>;
     /**
      * Router Run获取数据
@@ -354,12 +388,12 @@ declare class QueryRouter {
         path?: string;
         key?: string;
         payload?: any;
-    }, ctx?: RouteContext & {
+    }, ctx?: RouteContext<T> & {
         [key: string]: any;
     }): Promise<{
-        code: any;
+        code: number;
         data: any;
-        message: any;
+        message: string;
     }>;
     /**
      * 设置上下文
@@ -371,12 +405,12 @@ declare class QueryRouter {
     /**
      * 获取handle函数, 这里会去执行parse函数
      */
-    getHandle<T = any>(router: QueryRouter, wrapperFn?: HandleFn<T>, ctx?: RouteContext): (msg: {
+    getHandle<T = any>(router: QueryRouter, wrapperFn?: HandleFn, ctx?: RouteContext): (msg: {
         id?: string;
         path?: string;
         key?: string;
         [key: string]: any;
-    }, handleContext?: RouteContext) => Promise<{
+    }, handleContext?: RouteContext<T>) => Promise<{
         [key: string]: any;
         code: string;
         data?: any;
@@ -390,24 +424,16 @@ declare class QueryRouter {
         message: any;
         data?: undefined;
     }>;
-    exportRoutes(): Route<{
-        [key: string]: any;
-    }, SimpleObject$1>[];
+    exportRoutes(): Route<SimpleObject$1, SimpleObject$1>[];
     importRoutes(routes: Route[]): void;
     importRouter(router: QueryRouter): void;
-    throw(code?: number | string, message?: string): void;
-    throw(code?: number | string, opts?: CustomErrorOptions): void;
-    throw(opts?: CustomErrorOptions): void;
-    hasRoute(path: string, key?: string): Route<{
-        [key: string]: any;
-    }, SimpleObject$1>;
+    throw(...args: any[]): void;
+    hasRoute(path: string, key?: string): Route<SimpleObject$1, SimpleObject$1>;
     findRoute(opts?: {
         path?: string;
         key?: string;
         id?: string;
-    }): Route<{
-        [key: string]: any;
-    }, SimpleObject$1>;
+    }): Route<SimpleObject$1, SimpleObject$1>;
     createRouteList(opts?: {
         force?: boolean;
         filter?: (route: Route) => boolean;
@@ -449,10 +475,11 @@ declare class QueryRouter {
         [key: string]: z.ZodType<unknown, unknown, z.core.$ZodTypeInternals<unknown, unknown>>;
     };
 }
-type QueryRouterServerOpts = {
+type QueryRouterServerOpts<C extends SimpleObject$1 = SimpleObject$1> = {
     handleFn?: HandleFn;
-    context?: RouteContext;
+    context?: RouteContext<C>;
     appId?: string;
+    initHandle?: boolean;
 };
 interface HandleFn<T = any> {
     (msg: {
@@ -469,20 +496,27 @@ interface HandleFn<T = any> {
 /**
  * QueryRouterServer
  * @description 移除server相关的功能，只保留router相关的功能，和http.createServer不相关，独立
+ * @template C 自定义 RouteContext 类型
  */
-declare class QueryRouterServer extends QueryRouter {
+declare class QueryRouterServer<C extends SimpleObject$1 = SimpleObject$1> extends QueryRouter<C> {
     appId: string;
     handle: any;
-    constructor(opts?: QueryRouterServerOpts);
+    context: RouteContext<C>;
+    constructor(opts?: QueryRouterServerOpts<C>);
     setHandle(wrapperFn?: HandleFn, ctx?: RouteContext): void;
     addRoute(route: Route, opts?: AddOpts): void;
     Route: typeof Route;
-    route(opts: RouteOpts): Route<Required<RouteContext>>;
-    route(path: string, key?: string): Route<Required<RouteContext>>;
-    route(path: string, opts?: RouteOpts): Route<Required<RouteContext>>;
-    route(path: string, key?: string, opts?: RouteOpts): Route<Required<RouteContext>>;
-    prompt(description: string): Route<Required<RouteContext>>;
-    prompt(description: Function): Route<Required<RouteContext>>;
+    route<M extends SimpleObject$1 = SimpleObject$1>(opts: RouteOpts & {
+        metadata?: M;
+    }): Route<M, Required<RouteContext<C>>>;
+    route<M extends SimpleObject$1 = SimpleObject$1>(path: string, opts?: RouteOpts & {
+        metadata?: M;
+    }): Route<M, Required<RouteContext<C>>>;
+    route<M extends SimpleObject$1 = SimpleObject$1>(path: string, key?: string): Route<M, Required<RouteContext<C>>>;
+    route<M extends SimpleObject$1 = SimpleObject$1>(path: string, key?: string, opts?: RouteOpts & {
+        metadata?: M;
+    }): Route<M, Required<RouteContext<C>>>;
+    prompt(description: string): Route<SimpleObject$1, SimpleObject$1>;
     /**
      * 调用了handle
      * @param param0
@@ -493,9 +527,7 @@ declare class QueryRouterServer extends QueryRouter {
         path?: string;
         key?: string;
         payload?: any;
-    }, ctx?: RouteContext & {
-        [key: string]: any;
-    }): Promise<any>;
+    }, ctx?: Partial<RouteContext<C>>): Promise<any>;
 }
 declare class Mini extends QueryRouterServer {
 }
@@ -951,7 +983,7 @@ type RouterHandle = (msg: {
     [key: string]: any;
 };
 type AppOptions<T = {}> = {
-    router?: QueryRouter;
+    router?: QueryRouterServer;
     server?: ServerType;
     /** handle msg 关联 */
     routerHandle?: RouterHandle;
@@ -959,17 +991,18 @@ type AppOptions<T = {}> = {
     serverOptions?: ServerNodeOpts;
     appId?: string;
 };
-type AppRouteContext<T = {}> = HandleCtx & RouteContext<T> & {
+type AppRouteContext<T> = HandleCtx & RouteContext<T> & {
     app: App<T>;
 };
 /**
  *  封装了 Router 和 Server 的 App 模块，处理http的请求和响应，内置了 Cookie 和 Token 和 res 的处理
  *  U - Route Context的扩展类型
  */
-declare class App<U = {}> extends QueryRouter {
+declare class App<U = {}> extends QueryRouterServer<AppRouteContext<U>> {
     appId: string;
-    router: QueryRouter;
+    router: QueryRouterServer;
     server: ServerType;
+    context: AppRouteContext<U>;
     constructor(opts?: AppOptions<U>);
     listen(port: number, hostname?: string, backlog?: number, listeningListener?: () => void): void;
     listen(port: number, hostname?: string, listeningListener?: () => void): void;
@@ -979,34 +1012,7 @@ declare class App<U = {}> extends QueryRouter {
     listen(path: string, listeningListener?: () => void): void;
     listen(handle: any, backlog?: number, listeningListener?: () => void): void;
     listen(handle: any, listeningListener?: () => void): void;
-    addRoute(route: Route, opts?: AddOpts): void;
     Route: typeof Route;
-    route(opts: RouteOpts<AppRouteContext<U>>): Route<AppRouteContext<U>>;
-    route(path: string, key?: string): Route<AppRouteContext<U>>;
-    route(path: string, opts?: RouteOpts<AppRouteContext<U>>): Route<AppRouteContext<U>>;
-    route(path: string, key?: string, opts?: RouteOpts<AppRouteContext<U>>): Route<AppRouteContext<U>>;
-    prompt(description: string): Route<AppRouteContext<U>>;
-    prompt(description: Function): Route<AppRouteContext<U>>;
-    call(message: {
-        id?: string;
-        path?: string;
-        key?: string;
-        payload?: any;
-    }, ctx?: AppRouteContext<U> & {
-        [key: string]: any;
-    }): Promise<any>;
-    run(msg: {
-        id?: string;
-        path?: string;
-        key?: string;
-        payload?: any;
-    }, ctx?: Partial<AppRouteContext<U>> & {
-        [key: string]: any;
-    }): Promise<{
-        code: any;
-        data: any;
-        message: any;
-    }>;
     static handleRequest(req: IncomingMessage$1, res: ServerResponse$1): Promise<{
         cookies: Record<string, string>;
         token: string;

package/dist/router.js

@@ -3051,6 +3051,18 @@ class CustomError extends Error {
   static isError(error) {
     return error instanceof CustomError || typeof error === "object" && error !== null && "code" in error;
   }
+  static throw(...args) {
+    const [args0, args1] = args;
+    if (args0 && typeof args0 === "object") {
+      throw new CustomError(args0);
+    }
+    if (args1 && typeof args1 === "object") {
+      throw new CustomError(args0, args1);
+    } else if (args1) {
+      throw new CustomError(args0, { message: args1 });
+    }
+    throw new CustomError(args0);
+  }
   parse(e) {
     if (e) {
       return CustomError.parseError(e);
@@ -16997,7 +17009,7 @@ class Route {
     router.add(this, opts);
   }
   throw(...args) {
-    throw new CustomError(...args);
+    CustomError.throw(...args);
   }
 }
 var toJSONSchemaRoute = (route) => {
@@ -17040,6 +17052,7 @@ class QueryRouter {
     const maxNextRoute = this.maxNextRoute;
     ctx = ctx || {};
     ctx.currentPath = path;
+    ctx.currentId = route?.id;
     ctx.currentKey = key;
     ctx.currentRoute = route;
     ctx.index = (ctx.index || 0) + 1;
@@ -17053,7 +17066,7 @@ class QueryRouter {
       ctx.code = 500;
       ctx.message = "Too many nextRoute";
       ctx.body = null;
-      return;
+      return ctx;
     }
     if (route && route.middleware && route.middleware.length > 0) {
       const errorMiddleware = [];
@@ -17128,7 +17141,9 @@ class QueryRouter {
             }
             return ctx;
           }
-          if (ctx.end) {}
+          if (ctx.end) {
+            return ctx;
+          }
         }
       }
     }
@@ -17295,16 +17310,7 @@ class QueryRouter {
     this.importRoutes(router.routes);
   }
   throw(...args) {
-    const [args0, args1] = args;
-    if (args0 && typeof args0 === "object") {
-      throw new CustomError(args0);
-    }
-    if (args1 && typeof args1 === "object") {
-      throw new CustomError(args0, args1);
-    } else if (args1) {
-      throw new CustomError(args0, { message: args1 });
-    }
-    throw new CustomError(args0);
+    CustomError.throw(...args);
   }
   hasRoute(path, key = "") {
     return this.routes.find((r) => r.path === path && r.key === key);
@@ -17331,7 +17337,7 @@ class QueryRouter {
         description: "列出当前应用下的所有的路由信息",
         middleware: opts?.middleware || [],
         run: async (ctx) => {
-          const tokenUser = ctx.state.tokenUser;
+          const tokenUser = ctx.state;
           let isUser = !!tokenUser;
           const list = this.getList(opts?.filter).filter((item) => {
             if (item.id === "auth" || item.id === "auth-can" || item.id === "check-auth-admin" || item.id === "auth-admin") {
@@ -17366,7 +17372,10 @@ class QueryRouterServer extends QueryRouter {
   handle;
   constructor(opts) {
     super();
-    this.handle = this.getHandle(this, opts?.handleFn, opts?.context);
+    const initHandle = opts?.initHandle ?? true;
+    if (initHandle || opts?.handleFn) {
+      this.handle = this.getHandle(this, opts?.handleFn, opts?.context);
+    }
     this.setContext({ needSerialize: false, ...opts?.context });
     if (opts?.appId) {
       this.appId = opts.appId;
@@ -17397,15 +17406,8 @@ class QueryRouterServer extends QueryRouter {
     }
     return new Route(path, key, opts);
   }
-  prompt(...args) {
-    const [desc] = args;
-    let description = "";
-    if (typeof desc === "string") {
-      description = desc;
-    } else if (typeof desc === "function") {
-      description = desc() || "";
-    }
-    return new Route("", "", { description });
+  prompt(description) {
+    return new Route(undefined, undefined, { description });
   }
   async run(msg, ctx) {
     const handle = this.handle;
@@ -18481,11 +18483,11 @@ class BunServer extends ServerBase {
   }
 }
 // src/app.ts
-class App extends QueryRouter {
+class App extends QueryRouterServer {
   router;
   server;
   constructor(opts) {
-    super();
+    super({ initHandle: false, context: { needSerialize: true, ...opts?.routerContext } });
     const router = this;
     let server = opts?.server;
     if (!server) {
@@ -18497,7 +18499,6 @@ class App extends QueryRouter {
       }
     }
     server.setHandle(router.getHandle(router, opts?.routerHandle, opts?.routerContext));
-    router.setContext({ needSerialize: true, ...opts?.routerContext });
     this.router = router;
     this.server = server;
     if (opts?.appId) {
@@ -18510,42 +18511,7 @@ class App extends QueryRouter {
   listen(...args) {
     this.server.listen(...args);
   }
-  addRoute(route, opts) {
-    super.add(route, opts);
-  }
   Route = Route;
-  route(...args) {
-    const [path, key, opts] = args;
-    if (typeof path === "object") {
-      return new Route(path.path, path.key, path);
-    }
-    if (typeof path === "string") {
-      if (opts) {
-        return new Route(path, key, opts);
-      }
-      if (key && typeof key === "object") {
-        return new Route(path, key?.key || "", key);
-      }
-      return new Route(path, key);
-    }
-    return new Route(path, key, opts);
-  }
-  prompt(...args) {
-    const [desc] = args;
-    let description = "";
-    if (typeof desc === "string") {
-      description = desc;
-    } else if (typeof desc === "function") {
-      description = desc() || "";
-    }
-    return new Route("", "", { description });
-  }
-  async call(message, ctx) {
-    return await super.call(message, ctx);
-  }
-  async run(msg, ctx) {
-    return await super.run(msg, ctx);
-  }
   static handleRequest(req, res) {
     return handleServer(req, res);
   }