49 lines
1.6 KiB
TypeScript
49 lines
1.6 KiB
TypeScript
import { cookies } from 'next/headers';
|
|
import { NextRequest } from 'next/server';
|
|
import { sessionCookieName } from '@/lib/auth/session-cookie';
|
|
import { readStalwartAuthContextFromStore } from '@/lib/stalwart/auth-context';
|
|
import { MAX_ACCOUNT_SLOTS } from '@/lib/account-utils';
|
|
|
|
export interface StalwartCredentials {
|
|
/** URL of the JMAP server (used for JMAP + management method calls) */
|
|
serverUrl: string;
|
|
authHeader: string;
|
|
username: string;
|
|
hasSessionCookie: boolean;
|
|
slot: number;
|
|
}
|
|
|
|
function parseSlot(raw: string | null): number | null {
|
|
if (raw === null) return null;
|
|
const slot = parseInt(raw, 10);
|
|
return Number.isNaN(slot) || slot < 0 || slot >= MAX_ACCOUNT_SLOTS ? null : slot;
|
|
}
|
|
|
|
const ALL_SLOTS = Array.from({ length: MAX_ACCOUNT_SLOTS }, (_, i) => i);
|
|
|
|
function getCandidateSlots(request: NextRequest): number[] {
|
|
const requestedSlot = parseSlot(request.headers.get('X-JMAP-Cookie-Slot'))
|
|
?? parseSlot(request.nextUrl.searchParams.get('slot'));
|
|
|
|
return requestedSlot === null ? ALL_SLOTS : [requestedSlot];
|
|
}
|
|
|
|
export async function getStalwartCredentials(request: NextRequest): Promise<StalwartCredentials | null> {
|
|
const cookieStore = await cookies();
|
|
|
|
for (const slot of getCandidateSlots(request)) {
|
|
const context = readStalwartAuthContextFromStore(cookieStore, slot);
|
|
if (!context) continue;
|
|
|
|
return {
|
|
serverUrl: context.serverUrl.replace(/\/+$/, ''),
|
|
authHeader: context.authHeader,
|
|
username: context.username,
|
|
hasSessionCookie: !!cookieStore.get(sessionCookieName(slot))?.value,
|
|
slot,
|
|
};
|
|
}
|
|
|
|
return null;
|
|
}
|