eii-frontend / lib / session.ts

import { globalCookieStore, env } from './env'
import { getIronSession } from 'iron-session'
import { cookies } from 'next/headers'

export interface Session {
  did?: string
}

const sessionOptions = {
  cookieName: 'sid',
  password: env.COOKIE_SECRET,
}

export async function getSession(): Promise<Session> {
  const cookieStore = await cookies()
  const session = await getIronSession<Session>(cookieStore, sessionOptions)

  if (!session.did) {
    const globalDid = globalCookieStore.get('session_did')
    if (globalDid) {
      session.did = globalDid
      const updatedSession = await getIronSession<Session>(cookieStore, sessionOptions)
      updatedSession.did = globalDid
      await updatedSession.save()
    }
  } else {
    globalCookieStore.set('session_did', session.did)
  }

  return session
}

export async function setSession(session: Session): Promise<void> {
  const cookieStore = await cookies()
  const ironSession = await getIronSession<Session>(cookieStore, sessionOptions)

  ironSession.did = session.did
  await ironSession.save()

  if (session.did) {
    globalCookieStore.set('session_did', session.did)
  }
}

export async function clearSession(): Promise<void> {
  const cookieStore = await cookies()
  const ironSession = await getIronSession<Session>(cookieStore, sessionOptions)

  ironSession.did = undefined
  await ironSession.save()

  globalCookieStore.delete('session_did')
}