drop/server/internal/session/memory.ts

import moment from "moment";
import { Session, SessionProvider } from "./types";
import { v4 as uuidv4 } from 'uuid';

export default function createMemorySessionHandler() {
    const sessions: { [key: string]: Session } = {}

    const sessionCookieName = "drop-session";

    const memoryProvider: SessionProvider = {
        async setSession(h3, data) {
            const existingCookie = getCookie(h3, sessionCookieName);
            if (existingCookie) delete sessions[existingCookie]; // Clear any previous session

            const cookie = uuidv4();
            const expiry = moment().add(31, 'day');
            setCookie(h3, sessionCookieName, cookie, { expires: expiry.toDate() });

            sessions[cookie] = data;
            return true;
        },
        async updateSession(h3, key, data) {
            const cookie = getCookie(h3, sessionCookieName);
            if (!cookie) return false;

            sessions[cookie] = Object.assign({}, sessions[cookie], { [key]: data });
            return true;
        },
        async getSession(h3) {
            const cookie = getCookie(h3, sessionCookieName);
            if (!cookie) return undefined;

            return sessions[cookie] as any; // Wild type cast because we let the user specify types if they want
        },
        async clearSession(h3) {
            const cookie = getCookie(h3, sessionCookieName);
            if (!cookie) return;

            delete sessions[cookie];
            deleteCookie(h3, sessionCookieName);
        },
    };

    return memoryProvider;
}