This commit is contained in:
rizky 2024-09-12 00:47:59 -07:00
parent ab9b40b1c4
commit dd8ea2e1ec
11 changed files with 5 additions and 299 deletions

View File

@ -1,6 +1,4 @@
import { useLocal } from "lib/utils/use-local";
import { glb } from "app/lib/goal";
import { Button } from "lib/comps/ui/button";
import * as DialogPrimitive from "@radix-ui/react-dialog";
import {
Dialog,
DialogContent,
@ -9,9 +7,9 @@ import {
DialogTitle,
DialogTrigger,
} from "lib/comps/ui/dialog";
import * as DialogPrimitive from "@radix-ui/react-dialog";
import { useLocal } from "lib/utils/use-local";
import { X } from "lucide-react";
import { FC, useEffect } from "react";
import { FC } from "react";
export const Pop: FC<{
child: any;

View File

@ -66,7 +66,7 @@ export const useServerRouter = <T extends ReturnType<typeof newServerRouter>>(
return {
async handle(arg: ServerContext | SessionContext<any>) {
const { url, req } = arg;
const { url, req, handle } = arg;
const found = findRoute(rou, undefined, url.pathname);
if (found) {
@ -93,6 +93,7 @@ export const useServerRouter = <T extends ReturnType<typeof newServerRouter>>(
return new Response(JSON.stringify(result));
}
return handle(req);
},
};
};

View File

@ -1,121 +0,0 @@
import { ConsoleLogWriter } from "drizzle-orm";
import { newClientRouter } from "../server/server-route";
import { sessionRouter } from "./router/router";
import { sessionClientStore } from "./store/client-store";
import { ClientSession, SessionAuth, SessionData } from "./type";
export const newClientSession = <T>(arg?: {
tracker?: { enabled?: boolean };
on?: Partial<{
messageReceived: (session: ClientSession<T>) => Promise<void>;
afterLogin: (session: ClientSession<T>) => Promise<void>;
afterLogout: (session: ClientSession<T>) => Promise<void>;
afterInit: (session: ClientSession<T>) => Promise<void>;
}>;
}) => {
const store = sessionClientStore<T>();
const client = newClientRouter(sessionRouter);
const login_promise = { resolve: null as any, reject: null as any };
const logout_promise = { resolve: null as any, reject: null as any };
const session: ClientSession<T> = {
status: "checking",
current: null,
connected: false,
async connect(auth) {},
async init() {
const current = await store.load();
if (!current) {
this.status = "guest";
} else {
this.current = current;
this.status = await client.check(current.uid, current.sid);
if (this.status !== "active") {
await store.clear();
this.current = null;
} else {
await this.connect();
}
}
if (arg?.on?.afterInit) {
arg.on.afterInit(session);
}
return { status: this.status };
},
login(auth: SessionAuth) {
return new Promise<SessionData<T>>(async (resolve, reject) => {
if (isEditor) {
resolve({} as any);
return;
}
login_promise.resolve = resolve;
login_promise.reject = reject;
if (this.status === "checking") {
await new Promise<void>((done) => {
const ival = setInterval(() => {
if (this.status !== "checking") {
clearInterval(ival);
done();
}
}, 100);
});
}
if (this.status === "guest") {
this.connect(auth);
} else {
console.error(
`\
Session login failed, current status is: ${this.status}.
Login is prevented, please logout first before re-login!`
);
if (this.current) {
resolve(this.current);
} else {
if (auth) {
this.connect(auth);
} else {
reject("Current session not found");
}
}
}
});
},
logout() {
return new Promise<void>(async (resolve, reject) => {
if (isEditor) {
resolve({} as any);
return;
}
logout_promise.resolve = resolve;
logout_promise.reject = reject;
});
},
};
if (!isEditor) {
session.init();
}
return session;
};
const wsReady = async (ws: WebSocket) => {
if (ws) {
if (ws.readyState === ws.OPEN) return;
else {
ws.close();
await new Promise<void>((done) => {
const ival = setInterval(() => {
if (ws.readyState === ws.CLOSED) {
clearInterval(ival);
done();
}
}, 100);
});
}
}
};

View File

@ -1,7 +0,0 @@
import { newServerRouter } from "lib/server/server-route";
export const sessionRouter = newServerRouter({
check: ["/_session/check", () => import("./session-check")],
logout: ["/_session/logout", () => import("./session-logout")],
track: ["/_session/track", () => import("./session-track")],
});

View File

@ -1,18 +0,0 @@
import { EsensiSession } from "app/server/session";
import { sessionContext } from "lib/server/context";
import { ClientSessionStatus } from "../type";
export default async function (this: any, uid: string, sid: string) {
const ctx = sessionContext<EsensiSession>(this);
let result = "invalid" as ClientSessionStatus;
const session = ctx.session.findFirst({ uid, sid });
if (session) {
if (!session.expired_at || session.expired_at > Date.now()) {
result = "active";
} else {
result = "expired";
}
}
return result;
}

View File

@ -1,18 +0,0 @@
import { EsensiSession } from "app/server/session";
import { sessionContext } from "lib/server/context";
import { SessionAuth } from "../type";
export default async function (this: any, arg: SessionAuth) {
const ctx = sessionContext<EsensiSession>(this);
// let result = "invalid" as ClientSessionStatus;
// const session = ctx.session.findFirst({ uid, sid });
// if (session) {
// if (!session.expired_at || session.expired_at > Date.now()) {
// result = "active";
// } else {
// result = "expired";
// }
// }
return { status: "ok" };
}

View File

@ -1,18 +0,0 @@
import { EsensiSession } from "app/server/session";
import { sessionContext } from "lib/server/context";
import { SessionAuth } from "../type";
export default async function (this: any, arg: SessionAuth) {
const ctx = sessionContext<EsensiSession>(this);
// let result = "invalid" as ClientSessionStatus;
// const session = ctx.session.findFirst({ uid, sid });
// if (session) {
// if (!session.expired_at || session.expired_at > Date.now()) {
// result = "active";
// } else {
// result = "expired";
// }
// }
return { status: "ok" };
}

View File

@ -1,62 +0,0 @@
/// <reference types="bun-types" />
import { ServerWebSocket } from "bun";
import { useServerRouter } from "../server/server-route";
import { sessionRouter } from "./router/router";
import { newSessionStore } from "./store/session-store";
import {
ServerContext,
SessionAuth,
SessionStore,
SingleSession,
} from "./type";
type WS = ServerWebSocket<{ url: string }>;
type SessionServerHandler = {
cleanup: () => Promise<void>;
handle: (arg: ServerContext) => Promise<Response>;
};
export const initSessionServer = <T>(
server: PrasiServer,
arg: {
encrypt?: boolean;
router?: ReturnType<typeof useServerRouter>;
}
) => {
try {
const session_store = newSessionStore<T>(server.site_id);
const session_router = useServerRouter(sessionRouter);
const server_handler: SessionServerHandler = {
async cleanup() {},
async handle(server_arg) {
const { req, handle, url } = server_arg;
const route_arg = {
...server_arg,
session: {
...session_store,
current: undefined,
},
};
if (url.pathname.startsWith("/_session/")) {
const res = await session_router.handle(route_arg);
if (res) return res;
}
if (arg.router) {
const res = await arg.router.handle(route_arg);
if (res) return res;
}
return handle(req);
},
};
server.session = server_handler;
} catch (e) {
console.log(e);
}
};

View File

@ -1,49 +0,0 @@
import { createId } from "@paralleldrive/cuid2";
import { index, integer, sqliteTable, text } from "drizzle-orm/sqlite-core";
export const session = sqliteTable(
"session",
{
sid: text("sid")
.notNull()
.primaryKey()
.$defaultFn(() => createId()),
uid: text("uid").notNull(),
created_at: integer("created_at", { mode: "timestamp_ms" }).default(
new Date()
),
active: integer("active", { mode: "boolean" }),
data: text("data", { mode: "json" }).notNull(),
wsid: text("wsid", { mode: "json" }).default([]),
expired_at: integer("expired_at", { mode: "timestamp_ms" }),
},
(table) => {
return {
expired_at_idx: index("expired_at_idx").on(table.expired_at),
};
}
);
export const track = sqliteTable(
"track",
{
id: text("id")
.notNull()
.primaryKey()
.$defaultFn(() => createId()),
created_at: integer("created_at", { mode: "timestamp_ms" }).default(
new Date()
),
session_id: text("session_id"),
url: text("url").notNull(),
referer: text("referer"),
user_ip: text("user_ip"),
data: text("data", { mode: "json" }),
tstamp: integer("tstamp", { mode: "timestamp_ms" }).default(new Date()),
},
(table) => {
return {
session_id: index("session_id").on(table.session_id),
};
}
);

0
utils/crypto-alias.ts Normal file → Executable file
View File

0
utils/fetch-link-params.ts Normal file → Executable file
View File