Nova-Rewards/src/common/middleware/validation.middleware.ts at main · ghzhost/Nova-Rewards · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
import { Request, Response, NextFunction } from 'express';
import { ZodSchema, ZodError } from 'zod';
import { fromZodError } from 'zod-validation-error';

export function validateBody(schema: ZodSchema) {
  return (req: Request, res: Response, next: NextFunction) => {
    try {
      const validated = schema.parse(req.body);
      req.body = validated;
      next();
    } catch (error) {
      if (error instanceof ZodError) {
        const formattedError = fromZodError(error, {
          prefix: 'Validation failed',
          includePath: true,
        });

        return res.status(400).json({
          error: 'Validation failed',
          details: formattedError.details.map(detail => ({
            field: detail.path.join('.'),
            message: detail.message,
          })),
          statusCode: 400,
        });
      }
      next(error);
    }
  };
}

export function validateQuery(schema: ZodSchema) {
  return (req: Request, res: Response, next: NextFunction) => {
    try {
      const validated = schema.parse(req.query);
      req.query = validated;
      next();
    } catch (error) {
      if (error instanceof ZodError) {
        const formattedError = fromZodError(error, {
          prefix: 'Validation failed',
          includePath: true,
        });

        return res.status(400).json({
          error: 'Validation failed',
          details: formattedError.details.map(detail => ({
            field: detail.path.join('.'),
            message: detail.message,
          })),
          statusCode: 400,
        });
      }
      next(error);
    }
  };
}

export function validateParams(schema: ZodSchema) {
  return (req: Request, res: Response, next: NextFunction) => {
    try {
      const validated = schema.parse(req.params);
      req.params = validated;
      next();
    } catch (error) {
      if (error instanceof ZodError) {
        const formattedError = fromZodError(error, {
          prefix: 'Validation failed',
          includePath: true,
        });

        return res.status(400).json({
          error: 'Validation failed',
          details: formattedError.details.map(detail => ({
            field: detail.path.join('.'),
            message: detail.message,
          })),
          statusCode: 400,
        });
      }
      next(error);
    }
  };
}