import { ValidationOptions } from "../ValidationOptions";
import { buildMessage, ValidateBy } from "../common/ValidateBy";
import ValidatorJS from "validator";

export type IsIpVersion = "4" | "6" | 4 | 6;

export const IS_IP = "isIp";

/**
 * Checks if the string is an IP (version 4 or 6).
 * If given value is not a string, then it returns false.
 */
export function isIP(value: unknown, version?: IsIpVersion): boolean {
    const versionStr = version ? (`${version}` as "4" | "6") : undefined;
    return typeof value === "string" && ValidatorJS.isIP(value, versionStr);
}

/**
 * Checks if the string is an IP (version 4 or 6).
 * If given value is not a string, then it returns false.
 */
export function IsIP(version?: IsIpVersion, validationOptions?: ValidationOptions): PropertyDecorator {
    return ValidateBy(
        {
            name: IS_IP,
            constraints: [version],
            validator: {
                validate: (value, args): boolean => isIP(value, args.constraints[0]),
                defaultMessage: buildMessage(
                    (eachPrefix) => eachPrefix + "$property must be an ip address",
                    validationOptions
                )
            }
        },
        validationOptions
    );
}