import { ConfigCell, ConfigCellGroup, ConfigCellItem } from "../../types/configCells";
import { initConfigCellComposed } from "./configCellsComposed.ts";
import { ConfigCells, ConfigJsonField, ConfigJsonFieldItem, ConfigJsonFieldItemGroup } from "../../types/config";
import { VariantType } from "../variantType.ts";
import { MetadataContainer, SampleContainer } from "../api.ts";
import { initConfigCellFixed } from "./configCellsFixed.ts";
import { initConfigCellGenotype, initConfigCellInfo } from "./configCellsField.ts";

import { ConfigInvalidError } from "../error.ts";

export function initConfigCells(
  configs: ConfigJsonField[],
  variantType: VariantType,
  metadata: MetadataContainer,
  sample: SampleContainer | null,
): ConfigCells {
  const configCells: (ConfigCell | null)[] = configs.flatMap((configStaticField) => {
    let configCells: (ConfigCell | null)[];
    if (configStaticField.type === "group") {
      configCells = [
        initConfigCellItemGroup(configStaticField as ConfigJsonFieldItemGroup, variantType, metadata, sample),
      ];
    } else {
      configCells = initConfigCellItem(configStaticField as ConfigJsonFieldItem, variantType, metadata, sample);
    }
    return configCells;
  });
  return configCells.filter((configCell) => configCell !== null);
}

function initConfigCellItemGroup(
  config: ConfigJsonFieldItemGroup,
  variantType: VariantType,
  metadata: MetadataContainer,
  sample: SampleContainer | null,
) {
  const configFields = initConfigCells(config.fields, variantType, metadata, sample) as ConfigCellItem[];

  let fieldConfig: ConfigCellGroup | null;
  if (configFields.length === 0) {
    fieldConfig = null;
  } else {
    fieldConfig = { type: "group", fieldConfigs: configFields };
  }
  return fieldConfig;
}

function initConfigCellItem(
  config: ConfigJsonFieldItem,
  variantType: VariantType,
  metadata: MetadataContainer,
  sample: SampleContainer | null,
): ConfigCell[] {
  const type = config.type;

  let configFields: (ConfigCell | null)[];
  switch (type) {
    case "fixed":
      configFields = [initConfigCellFixed(config)];
      break;
    case "info":
      configFields = initConfigCellInfo(config, metadata.records);
      break;
    case "format":
      throw new ConfigInvalidError(`unsupported field type '${type}', did you mean to use 'genotype'?`);
    case "genotype":
      configFields = sample ? initConfigCellGenotype(config, metadata.records, sample) : [];
      break;
    case "composed":
      configFields = [initConfigCellComposed(config, variantType, metadata, sample)];
      break;
  }
  return configFields.filter((configField) => configField !== null);
}