import { Request, RequestHandler } from 'express';
import { z, ZodError, ZodSchema } from 'zod';
import { SortOrder } from '@/types/types';
import { ApiError } from '@/utils';
// Modified from zod-express-middleware: https://github.com/Aquila169/zod-express-middleware

export type TypedRequest<
  S extends {
    body: z.AnyZodObject;
    params: z.AnyZodObject;
    query: z.AnyZodObject;
  }
> = Request<z.infer<S['params']>, any, z.infer<S['body']>, z.infer<S['query']>>;

export enum ValidationType {
  BODY = 'Body',
  PARAMS = 'Params',
  QUERY = 'Query',
}

type RequestValidation<TParams, TQuery, TBody> = {
  body?: ZodSchema<TBody>;
  params?: ZodSchema<TParams>;
  query?: ZodSchema<TQuery>;
};

type ErrorListItem = {
  errors: ZodError<any>;
  type: ValidationType;
  value: string;
};

const getErrorValue = (object: any, error: ZodError<any>) => {
  const e = error.errors[0];
  if (e.code === z.ZodIssueCode.invalid_type) {
    return e.expected;
  }

  return object[e.path[0]][e.path[1]];
};

export const validateRequest: <TParams = any, TQuery = any, TBody = any>(
  schemas: RequestValidation<TParams, TQuery, TBody>
) => RequestHandler<TParams, any, TBody, TQuery> =
  ({ params, query, body }) =>
  (req, _res, next) => {
    const errors: Array<ErrorListItem> = [];
    if (params) {
      const parsed = params.safeParse(req.params);
      if (!parsed.success) {
        errors.push({
          errors: parsed.error,
          type: ValidationType.PARAMS,
          value: getErrorValue(req.params, parsed.error),
        });
      }
    }
    if (query) {
      const parsed = query.safeParse(req.query);
      if (!parsed.success) {
        const value = getErrorValue(req.query, parsed.error);
        errors.push({
          errors: parsed.error,
          type: ValidationType.QUERY,
          value,
        });
      }
    }
    if (body) {
      const parsed = body.safeParse(req.body);
      if (!parsed.success) {
        errors.push({
          errors: parsed.error,
          type: ValidationType.BODY,
          value: getErrorValue(req.body, parsed.error),
        });
      }
    }

    if (errors.length > 0) {
      const message = JSON.stringify(
        [
          // `(${errors[0].type})`,
          `(${errors[0].type}: ${errors[0].errors.issues[0].path[0]})`,
          errors[0].value && `[${errors[0].value}]`,
          errors[0].errors.issues[0].message,
        ]
          .filter((x) => x)
          .join(' ')
      );

      throw ApiError.badRequest(message);
    }

    return next();
  };

// const requiredErrorMessage = (
//   type: 'Query' | 'Body' | 'Params',
//   key: string
// ) => {
//   return `(${type}) [${key}] Required`;
// };

export const paginationValidation = {
  skip: z.string().refine((value) => {
    const parsed = Number(value);
    return !Number.isNaN(parsed) && parsed >= 0;
  }),
  take: z.string().refine((value) => {
    const parsed = Number(value);
    return !Number.isNaN(parsed) && parsed >= 0;
  }),
};

export const serverUrlIdValidation = {
  serverUrlId: z.optional(z.string().uuid()),
};

export const idValidation = (property: string) => {
  return { [property]: z.string().uuid() };
};

export const serverFolderIdValidation = {
  serverFolderId: z.optional(z.string().uuid().array()),
};

export const orderByValidation = {
  orderBy: z.nativeEnum(SortOrder),
};