import {isCountingAggregateOp} from '../../../aggregate';
import {NonPositionScaleChannel, PolarPositionScaleChannel, PositionScaleChannel} from '../../../channel';
import {getFieldDef, vgField} from '../../../channeldef';
import {ScaleInvalidDataShowAs, isScaleInvalidDataIncludeAsValue} from '../../../invalid';
import {fieldValidPredicate} from '../../../predicate';
import {VgValueRef, isSignalRef} from '../../../vega.schema';
import {getScaleInvalidDataMode} from '../../invalid/ScaleInvalidDataMode';
import {ScaleComponent} from '../../scale/component';
import {scaledZeroOrMinOrMax} from './scaledZeroOrMinOrMax';
import {MidPointParams} from './valueref';

export function getConditionalValueRefForIncludingInvalidValue<
  C extends PositionScaleChannel | PolarPositionScaleChannel | NonPositionScaleChannel
>({
  scaleChannel,
  channelDef,
  scale,
  scaleName,
  markDef,
  config
}: {
  scaleChannel: C;
} & Pick<MidPointParams, 'scale' | 'scaleName' | 'channelDef' | 'markDef' | 'config'>): VgValueRef | undefined {
  const scaleType = scale?.get('type');

  const fieldDef = getFieldDef<string>(channelDef);
  const isCountAggregate = isCountingAggregateOp(fieldDef?.aggregate);

  const invalidDataMode = getScaleInvalidDataMode<C>({
    scaleChannel,
    markDef,
    config,
    scaleType,
    isCountAggregate
  });

  if (fieldDef && invalidDataMode === 'show') {
    const includeAs: ScaleInvalidDataShowAs<C> = config.scale.invalid?.[scaleChannel] ?? 'zero-or-min';
    return {
      test: fieldValidPredicate(vgField(fieldDef, {expr: 'datum'}), false),
      ...refForInvalidValues(includeAs, scale, scaleName)
    };
  }
  return undefined;
}

function refForInvalidValues<C extends PositionScaleChannel | PolarPositionScaleChannel | NonPositionScaleChannel>(
  includeAs: ScaleInvalidDataShowAs<C>,
  scale: ScaleComponent,
  scaleName: string
): VgValueRef {
  if (isScaleInvalidDataIncludeAsValue(includeAs)) {
    const {value} = includeAs;
    return isSignalRef(value) ? {signal: value.signal} : {value};
  }

  return scaledZeroOrMinOrMax({
    scale,
    scaleName,
    mode: 'zeroOrMin'
  });
}