zsjz-ai/ai-frontend/packages/core/utils/compareOperator.ts

export type CompareOperator = 'GE' | 'LE' | 'EQ' | 'GT' | 'LT';

export const DEFAULT_COMPARE_OPERATOR: CompareOperator = 'GE';
export const DEFAULT_LE_COMPARE_OPERATOR: CompareOperator = 'LE';

export const COMPARE_OPERATOR_OPTIONS = [
  { label: '>=', value: 'GE' },
  { label: '<=', value: 'LE' },
  { label: '=', value: 'EQ' },
  { label: '>', value: 'GT' },
  { label: '<', value: 'LT' },
] as const;

const COMPARE_OPERATOR_MAP: Record<string, CompareOperator> = {
  GE: 'GE',
  ge: 'GE',
  '>=': 'GE',
  '1': 'GE',
  LE: 'LE',
  le: 'LE',
  '<=': 'LE',
  '2': 'LE',
  EQ: 'EQ',
  eq: 'EQ',
  '=': 'EQ',
  GT: 'GT',
  gt: 'GT',
  '>': 'GT',
  LT: 'LT',
  lt: 'LT',
  '<': 'LT',
};

export function normalizeCompareOperator(
  value?: string | null,
  fallback: CompareOperator = DEFAULT_COMPARE_OPERATOR,
): CompareOperator {
  if (!value) return fallback;
  return COMPARE_OPERATOR_MAP[String(value).trim()] || fallback;
}