fix

server/server-route.ts

@@ -1,21 +1,6 @@
 import { _post } from "lib/utils/post";
 import { addRoute, createRouter, findRoute } from "rou3";
-import { ServerSession } from "./session/server-session";
-import { SessionData } from "./session/store/session-store";
-
-export type ServerContext = {
-  req: Request;
-  handle: (req: Request) => Promise<Response>;
-  mode: "dev" | "prod";
-  url: {
-    raw: URL;
-    pathname: string;
-  };
-};
-
-export interface SessionContext<T> extends ServerContext {
-  session: ServerSession<T>;
-}
+import { ServerContext, SessionContext } from "./session/type";
 
 type RouteFn = (...arg: any[]) => Promise<any>;
 
@@ -38,18 +23,22 @@ export const newServerRouter = <
 };
 
 export const newClientRouter = <T extends ReturnType<typeof newServerRouter>>(
-  router: T
+  ...routers: T[]
 ) => {
   return new Proxy(
     {},
     {
       get(target, api_name, receiver) {
         return (...args: any[]) => {
-          const [url, _, opt] = router[api_name as any];
-          if (opt && opt.response_as)
-            return _post(url, args, { response_as: opt.response_as });
+          for (const router of routers) {
+            if (router[api_name as any]) {
+              const [url, _, opt] = router[api_name as any];
+              if (opt && opt.response_as)
+                return _post(url, args, { response_as: opt.response_as });
 
-          return _post(url, args);
+              return _post(url, args);
+            }
+          }
         };
       },
     }

server/session/client-session.ts

@@ -5,11 +5,10 @@ import { ClientSession } from "./type";
 
 export const newClientSession = <T>(arg?: {
   on: Partial<{
-    session: ClientSession<T>;
-    messageReceived: () => Promise<void>;
-    afterLogin: () => Promise<void>;
-    afterLogout: () => Promise<void>;
-    afterRecheck: () => Promise<void>;
+    messageReceived: (session: ClientSession<T>) => Promise<void>;
+    afterLogin: (session: ClientSession<T>) => Promise<void>;
+    afterLogout: (session: ClientSession<T>) => Promise<void>;
+    afterRecheck: (session: ClientSession<T>) => Promise<void>;
   }>;
 }) => {
   const store = sessionClientStore<T>();
@@ -18,13 +17,23 @@ export const newClientSession = <T>(arg?: {
   const session: ClientSession<T> = {
     status: "checking",
     current: null,
-    async recheck() {
-      const current = await store.load();
-      if (!current) {
-        this.status = "guest";
-      } else {
-        this.status = await client.check(current.uid, current.sid);
-      }
+    async connect() {
+      const url = new URL(location.href);
+      url.protocol = "wss:";
+      const ws = new WebSocket(url);
+
+      ws.onopen = () => {
+        ws.send("ok");
+      };
+      ws.onmessage = (m) => {
+        console.log(m);
+      };
+      // const current = await store.load();
+      // if (!current) {
+      //   this.status = "guest";
+      // } else {
+      //   this.status = await client.check(current.uid, current.sid);
+      // }
       return { status: this.status };
     },
     async login(arg: {
@@ -35,9 +44,6 @@ export const newClientSession = <T>(arg?: {
     async logout() {},
   };
 
-  session.recheck().then((e) => {
-    console.log(e);
-  });
-
+  session.connect();
   return session;
 };

server/session/server-session.ts

@@ -1,8 +1,9 @@
 /// <reference types="bun-types" />
 
-import { ServerContext, useServerRouter } from "../server-route";
+import { useServerRouter } from "../server-route";
 import { sessionRouter } from "./router/session-router";
 import { newSessionStore } from "./store/session-store";
+import { ServerContext } from "./type";
 
 type SessionServerHandler = {
   cleanup: () => Promise<void>;

server/session/type.ts

@@ -44,7 +44,7 @@ export type ClientSessionStatus = "checking" | "guest" | "expired" | "active";
 export type ClientSession<T> = {
   status: ClientSessionStatus;
   current: null | SessionData<T>;
-  recheck(): Promise<{ status: ClientSessionStatus }>;
+  connect(): Promise<{ status: ClientSessionStatus }>;
   login(arg: {
     method: "user-pass";
     username: string;
@@ -52,3 +52,17 @@ export type ClientSession<T> = {
   }): Promise<void>;
   logout(): Promise<void>;
 };
+
+export interface SessionContext<T> extends ServerContext {
+  session: ServerSession<T>;
+}
+
+export type ServerContext = {
+  req: Request;
+  handle: (req: Request) => Promise<Response>;
+  mode: "dev" | "prod";
+  url: {
+    raw: URL;
+    pathname: string;
+  };
+};