import DBClient from '@/prisma/DBClient';
import { BeerImage } from '@prisma/client';
import NextConnectOptions from '@/config/nextConnect/NextConnectOptions';
import APIResponseValidationSchema from '@/validation/APIResponseValidationSchema';
import { UserExtendedNextApiRequest } from '@/config/auth/types';
import { createRouter, expressWrapper } from 'next-connect';

import getCurrentUser from '@/config/nextConnect/middleware/getCurrentUser';

import multer from 'multer';

import cloudinaryConfig from '@/config/cloudinary';
import { NextApiResponse } from 'next';
import { z } from 'zod';
import ServerError from '@/config/util/ServerError';
import validateRequest from '@/config/nextConnect/middleware/validateRequest';

const { storage } = cloudinaryConfig;

const fileFilter: multer.Options['fileFilter'] = (req, file, cb) => {
  const { mimetype } = file;

  const isImage = mimetype.startsWith('image/');

  if (!isImage) {
    cb(null, false);
  }
  cb(null, true);
};

const uploadMiddleware = expressWrapper(
  multer({ storage, fileFilter, limits: { files: 3 } }).array('images'),
);

const BeerPostImageValidationSchema = z.object({
  caption: z.string(),
  alt: z.string(),
});

interface UploadBeerPostImagesRequest extends UserExtendedNextApiRequest {
  files?: Express.Multer.File[];
  query: { id: string };
  body: z.infer<typeof BeerPostImageValidationSchema>;
}

const processImageData = async (
  req: UploadBeerPostImagesRequest,
  res: NextApiResponse<z.infer<typeof APIResponseValidationSchema>>,
) => {
  const { files, user, body } = req;

  if (!files || !files.length) {
    throw new ServerError('No images uploaded', 400);
  }
  const beerImagePromises: Promise<BeerImage>[] = [];

  files.forEach((file) => {
    beerImagePromises.push(
      DBClient.instance.beerImage.create({
        data: {
          alt: body.alt,
          postedBy: { connect: { id: user!.id } },
          beerPost: { connect: { id: req.query.id } },
          path: file.path,
          caption: body.caption,
        },
      }),
    );
  });

  const beerImages = await Promise.all(beerImagePromises);

  res.status(200).json({
    success: true,
    message: `Successfully uploaded ${beerImages.length} image${
      beerImages.length > 1 ? 's' : ''
    }`,
    statusCode: 200,
  });
};

const router = createRouter<
  UploadBeerPostImagesRequest,
  NextApiResponse<z.infer<typeof APIResponseValidationSchema>>
>();

router.post(
  getCurrentUser,
  // @ts-expect-error
  uploadMiddleware,
  validateRequest({ bodySchema: BeerPostImageValidationSchema }),
  processImageData,
);

const handler = router.handler(NextConnectOptions);
export default handler;

export const config = { api: { bodyParser: false } };