import { TRPCError } from '@trpc/server';
import type { ReadModelDelta } from '@sammo-ts/common';
import { z } from 'zod';

import { accessLimitAuthedProcedure, router } from '../../trpc.js';
import {
    canUseDashboardSourceRevision,
    readDashboardSourceRevisionState,
    type DashboardSourceRevisionState,
    type DashboardSourceSlice,
} from '../../services/dashboardSourceRevision.js';
import { createReadModelDelta } from '../../services/readModelDeltaCache.js';
import {
    DASHBOARD_PROJECTION_ACCESS_WEIGHT,
    formatGeneralAccessLimitMessage,
    getGeneralAccessState,
    recordGeneralAccessWeight,
} from '../../services/generalAccess.js';
import { getBoardAccess } from '../board/index.js';
import { getGeneralContext } from '../general/index.js';
import { getTurnCommandTable } from '../turns/index.js';

const zRevision = z.string().regex(/^[A-Za-z0-9_-]{22}$/u);

const zContextBundleInput = z.object({
    include: z.object({
        context: z.boolean(),
        commandTable: z.boolean(),
        boardAccess: z.boolean(),
    }),
    known: z
        .object({
            context: zRevision.optional(),
            commandTable: zRevision.optional(),
            boardAccess: zRevision.optional(),
        })
        .optional(),
    knownSource: z
        .object({
            context: zRevision.optional(),
            commandTable: zRevision.optional(),
            boardAccess: zRevision.optional(),
        })
        .optional(),
    forceSnapshot: z.boolean().optional(),
});

type DashboardSliceRequest = {
    included: boolean;
    sourceState: DashboardSourceRevisionState | null;
    slice: DashboardSourceSlice;
    knownContent?: string;
    knownSource?: string;
    forceSnapshot?: boolean;
};

export const requiresDashboardProjection = (request: DashboardSliceRequest): boolean => {
    if (!request.included) return false;
    const sourceRevision = request.sourceState?.sourceRevisions[request.slice];
    return !(
        sourceRevision !== undefined &&
        request.knownContent !== undefined &&
        canUseDashboardSourceRevision({
            state: request.sourceState,
            slice: request.slice,
            knownContent: request.knownContent,
            knownSource: request.knownSource,
            forceSnapshot: request.forceSnapshot,
        })
    );
};

const createDashboardSliceDelta = async <T>(options: {
    included: boolean;
    sourceState: DashboardSourceRevisionState | null;
    slice: DashboardSourceSlice;
    knownContent?: string;
    knownSource?: string;
    forceSnapshot?: boolean;
    load: () => Promise<T | undefined>;
    create: (value: T) => Promise<ReadModelDelta<T>>;
}): Promise<ReadModelDelta<T> | undefined> => {
    if (!options.included) {
        return undefined;
    }

    const sourceRevision = options.sourceState?.sourceRevisions[options.slice];
    if (!requiresDashboardProjection(options) && options.knownContent !== undefined) {
        return {
            kind: 'unchanged',
            revision: options.knownContent,
            sourceRevision,
        };
    }

    const value = await options.load();
    if (value === undefined) {
        return undefined;
    }
    const delta = await options.create(value);
    return sourceRevision ? { ...delta, sourceRevision } : delta;
};

export const dashboardRouter = router({
    getContextBundleDelta: accessLimitAuthedProcedure.input(zContextBundleInput).query(async ({ ctx, input }) => {
        const authUser = ctx.auth?.user;
        const viewerId = authUser?.id;
        if (!authUser || !viewerId) {
            throw new TRPCError({ code: 'UNAUTHORIZED' });
        }

        const includesProjection = Object.values(input.include).some(Boolean);
        let generalId: number | null = null;
        if (includesProjection) {
            generalId =
                ctx.realtimeAccessGeneralId ??
                (
                    await ctx.db.general.findFirst({
                        where: { userId: viewerId },
                        orderBy: { id: 'asc' },
                        select: { id: true },
                    })
                )?.id ??
                null;
        }
        let sourceState = generalId ? await readDashboardSourceRevisionState(ctx.db, generalId, authUser) : null;
        const buildSliceRequests = (): DashboardSliceRequest[] => [
            {
                included: input.include.context,
                sourceState,
                slice: 'context',
                knownContent: input.known?.context,
                knownSource: input.knownSource?.context,
                forceSnapshot: input.forceSnapshot,
            },
            {
                included: input.include.commandTable && generalId !== null,
                sourceState,
                slice: 'commandTable',
                knownContent: input.known?.commandTable,
                knownSource: input.knownSource?.commandTable,
                forceSnapshot: input.forceSnapshot,
            },
            {
                included: input.include.boardAccess && generalId !== null,
                sourceState,
                slice: 'boardAccess',
                knownContent: input.known?.boardAccess,
                knownSource: input.knownSource?.boardAccess,
                forceSnapshot: input.forceSnapshot,
            },
        ];
        const sliceRequests = buildSliceRequests();
        const rebuildsPostgresProjection = sliceRequests.some(requiresDashboardProjection);
        if (rebuildsPostgresProjection && ctx.generalAccessTracking === true && ctx.realtimeAccessGranted !== true) {
            const recorded = await recordGeneralAccessWeight(ctx, DASHBOARD_PROJECTION_ACCESS_WEIGHT);
            const accessState = await getGeneralAccessState(ctx);
            if (accessState?.level === 2) {
                throw new TRPCError({
                    code: 'TOO_MANY_REQUESTS',
                    message: formatGeneralAccessLimitMessage(accessState),
                });
            }
            if (recorded && generalId !== null) {
                sourceState = await readDashboardSourceRevisionState(ctx.db, generalId, authUser);
            }
        }

        const [contextDelta, commandTableDelta, boardAccessDelta] = await Promise.all([
            createDashboardSliceDelta({
                included: input.include.context,
                sourceState,
                slice: 'context',
                knownContent: input.known?.context,
                knownSource: input.knownSource?.context,
                forceSnapshot: input.forceSnapshot,
                load: () => getGeneralContext(ctx),
                create: (value) =>
                    createReadModelDelta({
                        store: ctx.redis,
                        profile: ctx.profile.name,
                        viewerId,
                        slice: `main-context:${generalId ?? 'none'}`,
                        value,
                        knownRevision: input.known?.context,
                        forceSnapshot: input.forceSnapshot,
                    }),
            }),
            createDashboardSliceDelta({
                included: input.include.commandTable && generalId !== null,
                sourceState,
                slice: 'commandTable',
                knownContent: input.known?.commandTable,
                knownSource: input.knownSource?.commandTable,
                forceSnapshot: input.forceSnapshot,
                load: () => (generalId ? getTurnCommandTable(ctx, generalId) : Promise.resolve(undefined)),
                create: (value) =>
                    createReadModelDelta({
                        store: ctx.redis,
                        profile: ctx.profile.name,
                        viewerId,
                        slice: `main-command-table:${generalId}`,
                        value,
                        knownRevision: input.known?.commandTable,
                        forceSnapshot: input.forceSnapshot,
                    }),
            }),
            createDashboardSliceDelta({
                included: input.include.boardAccess && generalId !== null,
                sourceState,
                slice: 'boardAccess',
                knownContent: input.known?.boardAccess,
                knownSource: input.knownSource?.boardAccess,
                forceSnapshot: input.forceSnapshot,
                load: () => (generalId ? getBoardAccess(ctx) : Promise.resolve(undefined)),
                create: (value) =>
                    createReadModelDelta({
                        store: ctx.redis,
                        profile: ctx.profile.name,
                        viewerId,
                        slice: `main-board-access:${generalId}`,
                        value,
                        knownRevision: input.known?.boardAccess,
                        forceSnapshot: input.forceSnapshot,
                    }),
            }),
        ]);

        return {
            context: contextDelta,
            commandTable: commandTableDelta,
            boardAccess: boardAccessDelta,
        };
    }),
});