n4s/src/eager/ruleCallGenerator.ts

typed schema validation version of enforce

import { invariant } from 'vest-utils';

import { ctx } from '../enforceContext';
import { enforceMessage, transformResult } from '../ruleResult';

import type { UnmodifiedRules, SchemaRules } from './ruleRegistry';

type RuleCallConfig = {
  target: any;
  rule: UnmodifiedRules | SchemaRules;
  ruleName: string;
  value: any;
  customMessage: string | undefined;
  clearMessage: () => void;
};

export function createRuleCall(config: RuleCallConfig) {
  const { target, rule, ruleName, value, customMessage, clearMessage } = config;

  return function ruleCall(...args: any[]): any {
    const transformedResult = ctx.run({ value }, () =>
      transformResult(
        (rule as (...args: any[]) => any)(value, ...args),
        ruleName,
        value,
        ...args,
      ),
    );

    invariant(
      transformedResult.pass,
      enforceMessage(ruleName, transformedResult, value, customMessage),
    );

    // Clear message after each rule - it only applies to the next rule
    clearMessage();
    target.pass = transformedResult.pass;

    return target;
  };
}