import { Query, Selector, SelectorPart } from "@molgenis/vip-report-api";
import { FieldMetadata } from "@molgenis/vip-report-vcf";
import {
  ConfigFilterField,
  ConfigFilterFormat,
  FilterValue,
  FilterValueCategorical,
  FilterValueFlag,
  FilterValueInterval,
  FilterValueString,
} from "../../types/configFilter";
import { UnexpectedEnumValueException } from "../error.ts";
import { createSelectorInfo, createSelectorSample } from "./selector.ts";
import { createQueryFilterClosedInterval, createQueryFilterString } from "./queryFilter.ts";

export function createQueryFilterField(filter: ConfigFilterField, filterValue: FilterValue): Query {
  const selector = createSelectorFilter(filter);
  const field = filter.field;

  let query: Query;
  switch (filter.field.type) {
    case "CATEGORICAL":
      query = createQueryFilterFieldCategorical(selector, field, filterValue as FilterValueCategorical);
      break;
    case "CHARACTER":
    case "STRING":
      query = createQueryFilterFieldString(selector, field, filterValue as FilterValueString);
      break;
    case "FLOAT":
    case "INTEGER":
      query = createQueryFilterClosedInterval(selector, filterValue as FilterValueInterval);
      break;
    case "FLAG":
      query = createQueryFilterFlag(selector, field, filterValue as FilterValueCategorical);
      break;
    default:
      throw new UnexpectedEnumValueException(field.type);
  }
  return query;
}

function createSelectorFilter(filter: ConfigFilterField) {
  let selector: SelectorPart[];
  switch (filter.type) {
    case "genotype":
      selector = createSelectorSample((filter as ConfigFilterFormat).sample, filter.field);
      break;
    case "info":
      selector = createSelectorInfo(filter.field);
      break;
    case "composed":
    default:
      throw new UnexpectedEnumValueException(filter.type);
  }
  return selector;
}

export function createQueryFilterFieldCategorical(
  selector: Selector,
  field: FieldMetadata,
  filterValue: FilterValueCategorical,
): Query {
  const multiValue = field.number.count !== 1;
  return createQueryFilterString(selector, filterValue, multiValue);
}

function createQueryFilterFlag(selector: Selector, field: FieldMetadata, filterValue: FilterValueFlag): Query {
  const values: string[] = [];
  for (const value of filterValue) {
    if (value === "true") {
      values.push("1");
    } else if (value === "false") {
      values.push("__null");
    }
  }
  return createQueryFilterFieldCategorical(selector, field, values);
}

function createQueryFilterFieldString(selector: Selector, field: FieldMetadata, filterValue: FilterValueString): Query {
  const multiValue = field.number.count !== 1;
  return createQueryFilterString(selector, filterValue, multiValue);
}