umami/src/lib/yup.ts

import moment from 'moment';
import * as yup from 'yup';
import { UNIT_TYPES } from './constants';

export const DateRangeValidation = {
  startAt: yup.number().integer().required(),
  endAt: yup.number().integer().moreThan(yup.ref('startAt')).required(),
};

// ex: /funnel|insights|retention/i
export function getFilterValidation(matchRegex) {
  return {
    filter: yup.string(),
    filterType: yup.string().matches(matchRegex),
    pageSize: yup.number().integer().positive().max(200),
    page: yup.number().integer().positive(),
    orderBy: yup.string(),
  };
}

export const TimezoneTest = yup.string().test(
  'timezone',
  () => `Invalid timezone`,
  value => !value || !moment.tz.zone(value),
);

export const UnitTypeTest = yup.string().test(
  'unit',
  () => `Invalid unit`,
  value => !value || !UNIT_TYPES.includes(value),
);