zod_ip_refine.ts

import { z } from 'zod';

declare module 'zod' {
  interface ZodString {
    ip({ message }?: { message: string }): ZodString;
  }
}

// Add the ip() method to ZodString prototype
z.ZodString.prototype.ip = function ({ message }: { message: string } = { message: 'Invalid IP address' }) {
  return this.refine((val: string) => {
    const ipv4Regex = /^(?:(?:25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.){3}(?:25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)$/;
    const ipv6Regex = /^(?:[0-9a-fA-F]{1,4}:){7}[0-9a-fA-F]{1,4}$/;
    return ipv4Regex.test(val) || ipv6Regex.test(val);
  }, message);
};