zipline/src/server/session.ts

import { config } from '@/lib/config';
import { prisma } from '@/lib/db';
import { User } from '@/lib/db/models/user';
import { randomCharacters } from '@/lib/random';
import { FastifyReply, FastifyRequest } from 'fastify';
import { IncomingMessage, ServerResponse } from 'http';
import { getIronSession } from 'iron-session';

const cookieOptions = {
  // week
  maxAge: 60 * 60 * 24 * 7,
  expires: new Date(Date.now() + 60 * 60 * 24 * 7 * 1000),
  path: '/',
  sameSite: 'lax',
};

export type ZiplineSession = {
  id: string | null;
  sessionId: string | null;
};

export async function getSession(
  req: FastifyRequest | IncomingMessage,
  reply: FastifyReply | ServerResponse<IncomingMessage>,
) {
  if (!(req as any).raw || !(req as any).raw) {
    const session = await getIronSession<ZiplineSession>(
      req as IncomingMessage,
      reply as ServerResponse<IncomingMessage>,
      {
        password: config.core.secret,
        cookieName: 'zipline_session',
        cookieOptions,
      },
    );

    return session;
  }

  const session = await getIronSession<ZiplineSession>(
    (req as FastifyRequest).raw,
    (reply as FastifyReply).raw,
    {
      password: config.core.secret,
      cookieName: 'zipline_session',
      cookieOptions,
    },
  );

  return session;
}

export async function saveSession(
  session: Awaited<ReturnType<typeof getSession>>,
  user: User,
  overwriteSessions = true,
) {
  session.id = user.id;

  const sessionId = randomCharacters(32);
  session.sessionId = sessionId;

  await prisma.user.update({
    where: {
      id: user.id,
    },
    data: {
      sessions: overwriteSessions ? { set: [sessionId] } : { push: sessionId },
    },
  });

  await session.save();
}