import { Router, Response, NextFunction } from 'express'

import React from 'react'

import { renderToBuffer } from '@react-pdf/renderer'

import { z } from 'zod'

import { requireAuth, AuthenticatedRequest } from '@app/middleware/auth'

import { buildReportData } from './report.data'

import ReportDocument from './ReportDocument'

const router = Router()

const ReportSchema = z

  .object({

    periodStart: z.string().regex(/^\d{4}-\d{2}-\d{2}$/),

    periodEnd: z.string().regex(/^\d{4}-\d{2}-\d{2}$/),

  })

  .refine(

    (d) => {

      const start = new Date(d.periodStart)

      const end = new Date(d.periodEnd)

      const diff = (end.getTime() - start.getTime()) / 86_400_000

      return diff > 0 && diff <= 90

    },

    { message: 'periodEnd must be after periodStart and range ≤ 90 days' },

  )

router.post(

  '/',

  requireAuth,

  async (req: AuthenticatedRequest, res: Response, next: NextFunction) => {

    try {

      const parsed = ReportSchema.safeParse(req.body)

      if (!parsed.success) {

        return res.status(400).json({ error: parsed.error.issues[0]?.message ?? 'Invalid request' })

      }

      const { periodStart, periodEnd } = parsed.data

      const data = await buildReportData(

        req.userId!,

        new Date(periodStart),

        new Date(periodEnd),

      )

      const element = React.createElement(ReportDocument, { data })

      const buffer = await renderToBuffer(

        element as React.ReactElement<any>,

      )

      res.setHeader('Content-Type', 'application/pdf')

      res.setHeader(

        'Content-Disposition',

        `attachment; filename="Nexo-report-${periodStart.slice(0, 7)}.pdf"`,

      )

      return res.send(buffer)

    } catch (err) {

      return next(err)

    }

  },

)

export default router