Build an API Request Validator

import { z, ZodSchema, ZodError } from "zod";
import { Redis } from "ioredis";
const redis = new Redis(process.env.REDIS_URL!);

interface ValidationRule { name: string; check: (value: any, context: any) => boolean | Promise<boolean>; message: string; }
interface ValidationResult { valid: boolean; errors: Array<{ field: string; message: string; code: string }>; sanitized: Record<string, any>; }

const SANITIZERS: Record<string, (value: any) => any> = {
  trim: (v) => typeof v === "string" ? v.trim() : v,
  lowercase: (v) => typeof v === "string" ? v.toLowerCase() : v,
  stripHtml: (v) => typeof v === "string" ? v.replace(/<[^>]*>/g, "") : v,
  escapeSQL: (v) => typeof v === "string" ? v.replace(/['"\\;]/g, "") : v,
  normalizeEmail: (v) => typeof v === "string" ? v.toLowerCase().trim().replace(/\+.*@/, "@") : v,
};

// Validate request against schema + custom rules
export async function validate(data: any, schema: ZodSchema, rules?: ValidationRule[], sanitize?: string[]): Promise<ValidationResult> {
  let sanitized = { ...data };

  // Apply sanitizers
  if (sanitize) {
    for (const key of Object.keys(sanitized)) {
      for (const s of sanitize) {
        if (SANITIZERS[s]) sanitized[key] = SANITIZERS[s](sanitized[key]);
      }
    }
  }

  const errors: ValidationResult["errors"] = [];

  // Schema validation
  try {
    sanitized = schema.parse(sanitized);
  } catch (e) {
    if (e instanceof ZodError) {
      for (const issue of e.issues) {
        errors.push({ field: issue.path.join("."), message: issue.message, code: issue.code });
      }
    }
  }

  // Custom business rules
  if (rules && errors.length === 0) {
    for (const rule of rules) {
      const passed = await rule.check(sanitized, {});
      if (!passed) errors.push({ field: "_business", message: rule.message, code: "business_rule" });
    }
  }

  // Log invalid requests for security monitoring
  if (errors.length > 0) {
    await redis.hincrby("validation:stats", "rejected", 1);
    // Detect potential attacks
    const suspicious = JSON.stringify(data);
    if (/<script/i.test(suspicious) || /union\s+select/i.test(suspicious) || /;\s*drop/i.test(suspicious)) {
      await redis.rpush("security:suspicious", JSON.stringify({ data: suspicious.slice(0, 500), errors, timestamp: new Date().toISOString() }));
    }
  } else {
    await redis.hincrby("validation:stats", "accepted", 1);
  }

  return { valid: errors.length === 0, errors, sanitized };
}

// Middleware: auto-validate request body/query/params
export function validationMiddleware(config: { body?: ZodSchema; query?: ZodSchema; params?: ZodSchema; rules?: ValidationRule[]; sanitize?: string[] }) {
  return async (c: any, next: any) => {
    const allErrors: ValidationResult["errors"] = [];

    if (config.body) {
      const body = await c.req.json().catch(() => ({}));
      const result = await validate(body, config.body, config.rules, config.sanitize);
      if (!result.valid) allErrors.push(...result.errors);
      else c.set("validatedBody", result.sanitized);
    }

    if (config.query) {
      const query = Object.fromEntries(new URL(c.req.url).searchParams);
      const result = await validate(query, config.query);
      if (!result.valid) allErrors.push(...result.errors);
      else c.set("validatedQuery", result.sanitized);
    }

    if (allErrors.length > 0) {
      return c.json({ error: { code: "VALIDATION_ERROR", message: "Request validation failed", details: allErrors } }, 400);
    }

    await next();
  };
}

// Common reusable schemas
export const CommonSchemas = {
  pagination: z.object({ page: z.coerce.number().int().min(1).default(1), limit: z.coerce.number().int().min(1).max(100).default(20) }),
  id: z.object({ id: z.string().uuid() }),
  email: z.string().email().max(255),
  password: z.string().min(8).max(128).regex(/(?=.*[a-z])(?=.*[A-Z])(?=.*\d)/, "Must contain uppercase, lowercase, and number"),
  amount: z.number().positive().max(999999.99),
  dateRange: z.object({ startDate: z.string().datetime(), endDate: z.string().datetime() }).refine((d) => new Date(d.endDate) > new Date(d.startDate), "End date must be after start date"),
};