@kevisual/router 0.0.83 → 0.0.85

package/src/app.ts

@@ -1,4 +1,4 @@
-import { AddOpts, QueryRouter, Route, RouteContext, RouteOpts } from './route.ts';
+import { AddOpts, QueryRouter, QueryRouterServer, Route, RouteContext, RouteOpts } from './route.ts';
 import { ServerNode, ServerNodeOpts } from './server/server.ts';
 import { HandleCtx } from './server/server-base.ts';
 import { ServerType } from './server/server-type.ts';
@@ -10,7 +10,7 @@ import { randomId } from './utils/random.ts';
 
 type RouterHandle = (msg: { path: string;[key: string]: any }) => { code: string; data?: any; message?: string;[key: string]: any };
 type AppOptions<T = {}> = {
-  router?: QueryRouter;
+  router?: QueryRouterServer;
   server?: ServerType;
   /** handle msg 关联 */
   routerHandle?: RouterHandle;
@@ -19,18 +19,19 @@ type AppOptions<T = {}> = {
   appId?: string;
 };
 
-export type AppRouteContext<T = {}> = HandleCtx & RouteContext<T> & { app: App<T> };
+export type AppRouteContext<T> = HandleCtx & RouteContext<T> & { app: App<T> };
 
 /**
  *  封装了 Router 和 Server 的 App 模块，处理http的请求和响应，内置了 Cookie 和 Token 和 res 的处理
  *  U - Route Context的扩展类型
  */
-export class App<U = {}> extends QueryRouter {
+export class App<U = {}> extends QueryRouterServer<AppRouteContext<U>> {
   declare appId: string;
-  router: QueryRouter;
+  router: QueryRouterServer;
   server: ServerType;
+  declare context: AppRouteContext<U>;
   constructor(opts?: AppOptions<U>) {
-    super();
+    super({ initHandle: false, context: { needSerialize: true, ...opts?.routerContext } as any });
     const router = this;
     let server = opts?.server;
     if (!server) {
@@ -42,7 +43,6 @@ export class App<U = {}> 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) {
@@ -64,50 +64,7 @@ export class App<U = {}> extends QueryRouter {
     // @ts-ignore
     this.server.listen(...args);
   }
-  addRoute(route: Route, opts?: AddOpts) {
-    super.add(route, opts);
-  }
-
   Route = 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>>;
-  route(...args: any[]) {
-    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(description: string): Route<AppRouteContext<U>>
-  prompt(description: Function): Route<AppRouteContext<U>>
-  prompt(...args: any[]) {
-    const [desc] = args;
-    let description = ''
-    if (typeof desc === 'string') {
-      description = desc;
-    } else if (typeof desc === 'function') {
-      description = desc() || ''; // 如果是Promise，需要addTo App之前就要获取应有的函数了。
-    }
-    return new Route('', '', { description });
-  }
-
-  async call(message: { id?: string, path?: string; key?: string; payload?: any }, ctx?: AppRouteContext<U> & { [key: string]: any }) {
-    return await super.call(message, ctx);
-  }
-  async run(msg: { id?: string, path?: string; key?: string; payload?: any }, ctx?: Partial<AppRouteContext<U>> & { [key: string]: any }) {
-    return await super.run(msg, ctx);
-  }
   static handleRequest(req: IncomingMessage, res: ServerResponse) {
     return handleServer(req, res);
   }

package/src/opencode.ts

@@ -10,7 +10,7 @@ export const addCallFn = (app: App) => {
     path: 'call',
     key: '',
     description: '调用',
-    middleware: ['auth'],
+    middleware: ['auth-admin'],
     metadata: {
       tags: ['opencode'],
       ...createSkill({
@@ -24,7 +24,7 @@ export const addCallFn = (app: App) => {
         args: {
           path: tool.schema.string().describe('应用路径，例如 cnb'),
           key: tool.schema.string().optional().describe('应用key，例如 list-repos'),
-          payload: tool.schema.object({}).optional().describe('调用参数'),
+          payload: tool.schema.object({}).optional().describe('调用参数, 为对象, 例如 { "query": "javascript" }'),
         }
       })
     },
@@ -59,9 +59,16 @@ export const createRouterAgentPluginFn = (opts?: {
   if (!router.hasRoute('call', '')) {
     addCallFn(router as App)
   }
-  if (!router.hasRoute('auth', '')) {
-    router.route({ path: 'auth', key: '', id: 'auth', description: '认证' }).define(async (ctx) => { }).addTo(router as App)
+  if (router) {
+    (router as any).route({ path: 'auth', key: '', id: 'auth', description: '认证' }).define(async (ctx) => { }).addTo(router as App, {
+      overwrite: false
+    });
+
+    (router as any).route({ path: 'auth-admin', key: '', id: 'auth-admin', description: '认证' }).define(async (ctx) => { }).addTo(router as App, {
+      overwrite: false
+    })
   }
+
   const _routes = filter(router.routes, opts?.query || '')
   const routes = _routes.filter(r => {
     const metadata = r.metadata as Skill
@@ -75,7 +82,7 @@ export const createRouterAgentPluginFn = (opts?: {
   });
   // opencode run "使用技能查看系统信息"
   const AgentPlugin: Plugin = async (pluginInput) => {
-    useContextKey<PluginInput>('plugin-input', () => pluginInput, true)
+    useContextKey<PluginInput>('plugin-input', () => pluginInput, { isNew: true })
     const hooks = opts?.hooks ? await opts.hooks(pluginInput) : {}
     return {
       ...hooks,

package/src/route.ts

@@ -6,7 +6,16 @@ import { randomId } from './utils/random.ts';
 import * as schema from './validator/schema.ts';
 
 export type RouterContextT = { code?: number;[key: string]: any };
-export type RouteContext<T = { code?: number }, S = 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>;
+
+export type RouteContext<T = { code?: number }, U extends SimpleObject = {}, S = { [key: string]: any }> = {
   /**
    * 本地自己调用的时候使用，可以标识为当前自调用，那么 auth 就不许重复的校验
    * 或者不需要登录的，直接调用
@@ -23,9 +32,15 @@ export type RouteContext<T = { code?: number }, S = any> = {
   code?: number;
   /** return msg */
   message?: string;
-  // 传递状态
+  /**
+   * 传递状态
+   */
   state?: S;
   // transfer data
+  /**
+   * 当前routerId
+   */
+  currentId?: string;
   /**
    * 当前路径
    */
@@ -54,19 +69,19 @@ export type RouteContext<T = { code?: number }, S = any> = {
     ctx?: RouteContext & { [key: string]: any },
   ) => Promise<any>;
   /** 请求 route的返回结果，解析了body为data，就类同于 query.post获取的数据*/
-  run?: (message: { path: string; key?: string; payload?: any }, ctx?: RouteContext & { [key: string]: any }) => Promise<any>;
+  run?: (message: { path: string; key?: string; payload?: any }, ctx?: RouteContext) => Promise<any>;
   index?: number;
   throw?: throwError['throw'];
   /** 是否需要序列化, 使用JSON.stringify和JSON.parse */
   needSerialize?: boolean;
-} & T;
+} & T & U;
 export type SimpleObject = Record<string, any>;
 export type Run<T extends SimpleObject = {}> = (ctx: Required<RouteContext<T>>) => Promise<typeof ctx | null | void>;
 export type RunMessage = { path?: string; key?: string; id?: string; payload?: any; };
 export type NextRoute = Pick<Route, 'id' | 'path' | 'key'>;
 export type RouteMiddleware =
   | {
-    path: string;
+    path?: string;
     key?: string;
     id?: string;
   }
@@ -80,7 +95,7 @@ export type RouteOpts<U = {}, T = SimpleObject> = {
   description?: string;
   metadata?: T;
   middleware?: RouteMiddleware[]; // middleware
-  type?: 'route' | 'middleware';
+  type?: 'route' | 'middleware' | 'compound'; // compound表示这个 route 作为一个聚合体，没有实际的 run，而是一个 router 的聚合列表
   /**
    * $#$ will be used to split path and key
    */
@@ -123,7 +138,11 @@ export const createSkill = <T = SimpleObject>(skill: Skill<T>): Skill<T> => {
 
 export type RouteInfo = Pick<Route, (typeof pickValue)[number]>;
 
-export class Route<U = { [key: string]: any }, T extends SimpleObject = SimpleObject> implements throwError {
+/**
+ * @M 是 route的 metadate的类型，默认是 SimpleObject
+ * @U 是 RouteContext 里 state的类型
+ */
+export class Route<M extends SimpleObject = SimpleObject, U extends SimpleObject = SimpleObject> implements throwError {
   /**
    * 一级路径
    */
@@ -133,10 +152,10 @@ export class Route<U = { [key: string]: any }, T extends SimpleObject = SimpleOb
    */
   key?: string;
   id?: string;
-  run?: Run;
+  run?: Run<BuildRouteContext<M, U>>;
   nextRoute?: NextRoute; // route to run after this route
   description?: string;
-  metadata?: T;
+  metadata?: M;
   middleware?: RouteMiddleware[]; // middleware
   type? = 'route';
   /**
@@ -154,13 +173,13 @@ export class Route<U = { [key: string]: any }, T extends SimpleObject = SimpleOb
     if (opts) {
       this.id = opts.id || randomId(12, 'rand-');
       if (!opts.id && opts.idUsePath) {
-        const delimiter = opts.delimiter ?? '$#$';
+        const delimiter = opts.delimiter ?? '$$';
         this.id = path + delimiter + key;
       }
-      this.run = opts.run;
+      this.run = opts.run as Run<BuildRouteContext<M, U>>;
       this.nextRoute = opts.nextRoute;
       this.description = opts.description;
-      this.metadata = opts.metadata as T;
+      this.metadata = opts.metadata as M;
       this.type = opts.type || 'route';
       this.middleware = opts.middleware || [];
       this.key = opts.key || key;
@@ -184,9 +203,9 @@ export class Route<U = { [key: string]: any }, T extends SimpleObject = SimpleOb
     return this;
   }
   define<T extends { [key: string]: any } = RouterContextT>(opts: DefineRouteOpts): this;
-  define<T extends { [key: string]: any } = RouterContextT>(fn: Run<T & U>): this;
-  define<T extends { [key: string]: any } = RouterContextT>(key: string, fn: Run<T & U>): this;
-  define<T extends { [key: string]: any } = RouterContextT>(path: string, key: string, fn: Run<T & U>): this;
+  define<T extends { [key: string]: any } = RouterContextT>(fn: Run<T & BuildRouteContext<M, U>>): this;
+  define<T extends { [key: string]: any } = RouterContextT>(key: string, fn: Run<T & BuildRouteContext<M, U>>): this;
+  define<T extends { [key: string]: any } = RouterContextT>(path: string, key: string, fn: Run<T & BuildRouteContext<M, U>>): this;
   define(...args: any[]) {
     const [path, key, opts] = args;
     // 全覆盖，所以opts需要准确，不能由idUsePath 需要check的变量
@@ -209,7 +228,7 @@ export class Route<U = { [key: string]: any }, T extends SimpleObject = SimpleOb
       return this;
     }
     if (typeof path === 'function') {
-      this.run = path;
+      this.run = path as Run<BuildRouteContext<M, U>>;
       return this;
     }
     if (typeof path === 'string' && typeof key === 'function') {
@@ -262,11 +281,11 @@ export const fromJSONSchema = schema.fromJSONSchema;
  * @parmas overwrite 是否覆盖已存在的route，默认true
  */
 export type AddOpts = { overwrite?: boolean };
-export class QueryRouter implements throwError {
+export class QueryRouter<T extends SimpleObject = SimpleObject> implements throwError {
   appId: string = '';
   routes: Route[];
   maxNextRoute = 40;
-  context?: RouteContext = {}; // default context for call
+  context?: RouteContext<T> = {} as RouteContext<T>; // default context for call
   constructor() {
     this.routes = [];
   }
@@ -309,11 +328,12 @@ export class QueryRouter implements throwError {
    * @param ctx
    * @returns
    */
-  async runRoute(path: string, key: string, ctx?: RouteContext) {
+  async runRoute(path: string, key: string, ctx?: RouteContext<T>): Promise<RouteContext<T>> {
     const route = this.routes.find((r) => r.path === path && r.key === key);
     const maxNextRoute = this.maxNextRoute;
-    ctx = (ctx || {}) as RouteContext;
+    ctx = (ctx || {}) as RouteContext<T>;
     ctx.currentPath = path;
+    ctx.currentId = route?.id;
     ctx.currentKey = key;
     ctx.currentRoute = route;
     ctx.index = (ctx.index || 0) + 1;
@@ -327,7 +347,7 @@ export class QueryRouter implements throwError {
       ctx.code = 500;
       ctx.message = 'Too many nextRoute';
       ctx.body = null;
-      return;
+      return ctx;
     }
     // run middleware
     if (route && route.middleware && route.middleware.length > 0) {
@@ -382,7 +402,7 @@ export class QueryRouter implements throwError {
         const middleware = routeMiddleware[i];
         if (middleware) {
           try {
-            await middleware.run(ctx as Required<RouteContext>);
+            await middleware.run(ctx as Required<RouteContext<T>>);
           } catch (e) {
             if (route?.isDebug) {
               console.error('=====debug====:middlerware error');
@@ -404,6 +424,7 @@ export class QueryRouter implements throwError {
             return ctx;
           }
           if (ctx.end) {
+            return ctx;
           }
         }
       }
@@ -412,7 +433,7 @@ export class QueryRouter implements throwError {
     if (route) {
       if (route.run) {
         try {
-          await route.run(ctx as Required<RouteContext>);
+          await route.run(ctx as Required<RouteContext<T>>);
         } catch (e) {
           if (route?.isDebug) {
             console.error('=====debug====:route error');
@@ -467,7 +488,7 @@ export class QueryRouter implements throwError {
       }
     }
     // 如果没有找到route，返回404，这是因为出现了错误
-    return Promise.resolve({ code: 404, body: 'Not found' });
+    return Promise.resolve({ code: 404, body: 'Not found' } as RouteContext<T>);
   }
   /**
    * 第一次执行
@@ -475,12 +496,12 @@ export class QueryRouter implements throwError {
    * @param ctx
    * @returns
    */
-  async parse(message: { path: string; key?: string; payload?: any }, ctx?: RouteContext & { [key: string]: any }) {
+  async parse(message: { path: string; key?: string; payload?: any }, ctx?: RouteContext<T> & { [key: string]: any }) {
     if (!message?.path) {
-      return Promise.resolve({ code: 404, body: null, message: 'Not found path' });
+      return Promise.resolve({ code: 404, body: null, message: 'Not found path' } as RouteContext<T>);
     }
     const { path, key = '', payload = {}, ...query } = message;
-    ctx = ctx || {};
+    ctx = ctx || {} as RouteContext<T>;
     ctx.query = { ...ctx.query, ...query, ...payload };
     ctx.args = ctx.query;
     ctx.state = { ...ctx?.state };
@@ -514,7 +535,7 @@ export class QueryRouter implements throwError {
    * @param ctx
    * @returns
    */
-  async call(message: { id?: string; path?: string; key?: string; payload?: any }, ctx?: RouteContext & { [key: string]: any }) {
+  async call(message: { id?: string; path?: string; key?: string; payload?: any }, ctx?: RouteContext<T> & { [key: string]: any }) {
     let path = message.path;
     let key = message.key;
     // 优先 path + key
@@ -555,7 +576,7 @@ export class QueryRouter implements throwError {
    * @param ctx 
    * @returns 
    */
-  async run(message: { id?: string; path?: string; key?: string; payload?: any }, ctx?: RouteContext & { [key: string]: any }) {
+  async run(message: { id?: string; path?: string; key?: string; payload?: any }, ctx?: RouteContext<T> & { [key: string]: any }) {
     const res = await this.call(message, { ...this.context, ...ctx });
     return {
       code: res.code,
@@ -569,7 +590,7 @@ export class QueryRouter implements throwError {
    * @param ctx
    */
   setContext(ctx: RouteContext) {
-    this.context = ctx;
+    this.context = ctx as RouteContext<T>;
   }
   getList(filter?: (route: Route) => boolean): RouteInfo[] {
     return this.routes.filter(filter || (() => true)).map((r) => {
@@ -580,11 +601,11 @@ export class QueryRouter implements throwError {
   /**
    * 获取handle函数, 这里会去执行parse函数
    */
-  getHandle<T = any>(router: QueryRouter, wrapperFn?: HandleFn<T>, ctx?: RouteContext) {
-    return async (msg: { id?: string; path?: string; key?: string;[key: string]: any }, handleContext?: RouteContext) => {
+  getHandle<T = any>(router: QueryRouter, wrapperFn?: HandleFn, ctx?: RouteContext) {
+    return async (msg: { id?: string; path?: string; key?: string;[key: string]: any }, handleContext?: RouteContext<T>) => {
       try {
         const context = { ...ctx, ...handleContext };
-        const res = await router.call(msg, context);
+        const res = await router.call(msg, context) as any;
         if (wrapperFn) {
           res.data = res.body;
           return wrapperFn(res, context);
@@ -637,7 +658,7 @@ export class QueryRouter implements throwError {
         description: '列出当前应用下的所有的路由信息',
         middleware: opts?.middleware || [],
         run: async (ctx: RouteContext) => {
-          const tokenUser = ctx.state.tokenUser;
+          const tokenUser = ctx.state as unknown as { tokenUser?: any };
           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') {
@@ -683,10 +704,11 @@ export class QueryRouter implements throwError {
   fromJSONSchema = fromJSONSchema;
 }
 
-type QueryRouterServerOpts = {
+type QueryRouterServerOpts<C extends SimpleObject = SimpleObject> = {
   handleFn?: HandleFn;
-  context?: RouteContext;
+  context?: RouteContext<C>;
   appId?: string;
+  initHandle?: boolean;
 };
 interface HandleFn<T = any> {
   (msg: { path: string;[key: string]: any }, ctx?: any): { code: string; data?: any; message?: string;[key: string]: any };
@@ -695,13 +717,18 @@ interface HandleFn<T = any> {
 /**
  * QueryRouterServer
  * @description 移除server相关的功能，只保留router相关的功能，和http.createServer不相关，独立
+ * @template C 自定义 RouteContext 类型
  */
-export class QueryRouterServer extends QueryRouter {
+export class QueryRouterServer<C extends SimpleObject = SimpleObject> extends QueryRouter<C> {
   declare appId: string;
   handle: any;
-  constructor(opts?: QueryRouterServerOpts) {
+  declare context: RouteContext<C>;
+  constructor(opts?: QueryRouterServerOpts<C>) {
     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;
@@ -716,37 +743,28 @@ export class QueryRouterServer extends QueryRouter {
     this.add(route, opts);
   }
   Route = 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>>;
-  route(...args: any[]) {
+  route<M extends SimpleObject = SimpleObject>(opts: RouteOpts & { metadata?: M }): Route<M, Required<RouteContext<C>>>;
+  route<M extends SimpleObject = SimpleObject>(path: string, opts?: RouteOpts & { metadata?: M }): Route<M, Required<RouteContext<C>>>;
+  route<M extends SimpleObject = SimpleObject>(path: string, key?: string): Route<M, Required<RouteContext<C>>>;
+  route<M extends SimpleObject = SimpleObject>(path: string, key?: string, opts?: RouteOpts & { metadata?: M }): Route<M, Required<RouteContext<C>>>;
+  route<M extends SimpleObject = SimpleObject>(...args: any[]) {
     const [path, key, opts] = args;
     if (typeof path === 'object') {
-      return new Route(path.path, path.key, path);
+      return new Route<M, Required<RouteContext<C>>>(path.path, path.key, path);
     }
     if (typeof path === 'string') {
       if (opts) {
-        return new Route(path, key, opts);
+        return new Route<M, Required<RouteContext<C>>>(path, key, opts);
       }
       if (key && typeof key === 'object') {
-        return new Route(path, key?.key || '', key);
+        return new Route<M, Required<RouteContext<C>>>(path, key?.key || '', key);
       }
-      return new Route(path, key);
+      return new Route<M, Required<RouteContext<C>>>(path, key);
     }
-    return new Route(path, key, opts);
+    return new Route<M, Required<RouteContext<C>>>(path, key, opts);
   }
-  prompt(description: string): Route<Required<RouteContext>>;
-  prompt(description: Function): Route<Required<RouteContext>>;
-  prompt(...args: any[]) {
-    const [desc] = args;
-    let description = ''
-    if (typeof desc === 'string') {
-      description = desc;
-    } else if (typeof desc === 'function') {
-      description = desc() || ''; // 如果是Promise，需要addTo App之前就要获取应有的函数了。
-    }
-    return new Route('', '', { description });
+  prompt(description: string) {
+    return new Route(undefined, undefined, { description });
   }
 
   /**
@@ -754,12 +772,12 @@ export class QueryRouterServer extends QueryRouter {
    * @param param0
    * @returns
    */
-  async run(msg: { id?: string; path?: string; key?: string; payload?: any }, ctx?: RouteContext & { [key: string]: any }) {
+  async run(msg: { id?: string; path?: string; key?: string; payload?: any }, ctx?: Partial<RouteContext<C>>) {
     const handle = this.handle;
     if (handle) {
       return handle(msg, ctx);
     }
-    return super.run(msg, ctx);
+    return super.run(msg, ctx as RouteContext<C>);
   }
 }
 

package/src/server/server-base.ts

@@ -284,7 +284,7 @@ export class ServerBase implements ServerType {
    * @param ws 
    */
   async onWsClose(ws: WS) {
-    const id = ws?.data?.id || '';
+    const id = ws?.wsId || '';
     if (id) {
       this.emitter.emit('close--' + id, { type: 'close', ws, id });
       setTimeout(() => {
@@ -298,4 +298,7 @@ export class ServerBase implements ServerType {
     if (this.showConnected)
       ws.send(JSON.stringify({ type: 'connected' }));
   }
+  createId() {
+    return Math.random().toString(36).substring(2, 15);
+  }
 }

package/src/server/server-bun.ts

@@ -4,7 +4,7 @@
  * @tags bun, server, websocket, http
  * @createdAt 2025-12-20
  */
-import { ServerType, type ServerOpts, type Cors, RouterRes, RouterReq } from './server-type.ts';
+import { ServerType, type ServerOpts, type Cors, RouterRes, RouterReq, WS } from './server-type.ts';
 import { ServerBase } from './server-base.ts';
 
 export class BunServer extends ServerBase implements ServerType {
@@ -264,10 +264,14 @@ export class BunServer extends ServerBase implements ServerType {
         open: (ws: any) => {
           this.sendConnected(ws);
         },
-        message: async (ws: any, message: string | Buffer) => {
+        message: async (bunWs: any, message: string | Buffer) => {
+          const ws = bunWs as WS;
           const pathname = ws.data.pathname || '';
           const token = ws.data.token || '';
           const id = ws.data.id || '';
+          if (!ws.wsId) {
+            ws.wsId = this.createId();
+          }
           await this.onWebSocket({ ws, message, pathname, token, id });
         },
         close: (ws: any) => {

package/src/server/server-type.ts

@@ -49,16 +49,33 @@ export type OnWebSocketOptions<T = {}> = {
   message: string | Buffer;
   pathname: string,
   token?: string,
+  /** data 的id提取出来 */
   id?: string,
 }
 export type OnWebSocketFn = (options: OnWebSocketOptions) => Promise<void> | void;
 export type WS<T = {}> = {
   send: (data: any) => void;
   close: (code?: number, reason?: string) => void;
+  /**
+ * ws 自己生成的一个id，主要是为了区分不同的ws连接，方便在onWebSocket中使用
+ */
+  wsId?: string;
   data?: {
+    /**
+     * ws连接时的url，包含pathname和searchParams
+     */
     url: URL;
+    /**
+     * ws连接时的pathname
+     */
     pathname: string;
+    /**
+     * ws连接时的url中的token参数
+     */
     token?: string;
+    /**
+     * ws连接时的url中的id参数.
+     */
     id?: string;
     /**
      * 鉴权后的获取的信息

package/src/server/ws-server.ts

@@ -56,6 +56,11 @@ export class WsServerBase {
         token,
         id,
       }
+      // @ts-ignore
+      if (!ws.wsId) {
+        // @ts-ignore
+        ws.wsId = this.createId();
+      }
       ws.on('message', async (message: string | Buffer) => {
         await this.server.onWebSocket({ ws, message, pathname, token, id });
       });
@@ -66,7 +71,9 @@ export class WsServerBase {
         this.server.onWsClose(ws);
       });
     });
-
+  }
+  createId() {
+    return Math.random().toString(36).substring(2, 15);
   }
 }
 // TODO: ws handle and path and routerContext