/*
 * UnitCommitment.jl: Optimization Package for Security-Constrained Unit Commitment
 * Copyright (C) 2020-2025, UChicago Argonne, LLC. All rights reserved.
 * Released under the modified BSD license. See COPYING.md for more details.
 */

import { ValidationError } from "../Data/validate";
import { ColumnSpec } from "../../components/Common/Forms/DataTable";
import { UnitCommitmentScenario } from "../Data/types";

export const renameItemInObject = <T>(
  oldName: string,
  newName: string,
  container: { [key: string]: T },
): [{ [key: string]: T }, ValidationError | null] => {
  if (newName in container) {
    return [container, { message: `${newName} already exists` }];
  }
  const newContainer = Object.keys(container).reduce(
    (acc, val) => {
      if (val === oldName) {
        acc[newName] = container[val]!;
      } else {
        acc[val] = container[val]!;
      }
      return acc;
    },
    {} as { [key: string]: T },
  );
  return [newContainer, null];
};

export const generateUniqueName = (container: any, prefix: string): string => {
  let counter = 1;
  let name = `${prefix}${counter}`;
  while (name in container) {
    counter++;
    name = `${prefix}${counter}`;
  }
  return name;
};

export const parseNumber = (
  valueStr: string,
): [number, ValidationError | null] => {
  if (valueStr === "") {
    return [0, { message: "Field must not be blank" }];
  }
  const valueFloat = parseFloat(valueStr);
  if (isNaN(valueFloat)) {
    return [0, { message: `"${valueStr}" is not a valid number` }];
  } else {
    return [valueFloat, null];
  }
};

export const parseNullableNumber = (
  valueStr: string,
): [number | null, ValidationError | null] => {
  if (valueStr === "") return [null, null];
  return parseNumber(valueStr);
};

export const parseBool = (
  valueStr: string,
): [boolean, ValidationError | null] => {
  if (["true", "1"].includes(valueStr.toLowerCase())) {
    return [true, null];
  }
  if (["false", "0"].includes(valueStr.toLowerCase())) {
    return [false, null];
  }
  return [true, { message: `"${valueStr}" is not a valid boolean value` }];
};

export const changeStringData = (
  field: string,
  newValue: string,
  container: { [key: string]: any },
): [{ [key: string]: any }, ValidationError | null] => {
  return [
    {
      ...container,
      [field]: newValue,
    },
    null,
  ];
};

export const changeBusRefData = (
  field: string,
  newValue: string,
  container: { [key: string]: any },
  scenario: UnitCommitmentScenario,
): [{ [key: string]: any }, ValidationError | null] => {
  if (!(newValue in scenario.Buses)) {
    return [scenario, { message: `Bus "${newValue}" does not exist` }];
  }
  return changeStringData(field, newValue, container);
};

export const changeNumberData = (
  field: string,
  newValueStr: string,
  container: { [key: string]: any },
  nullable: boolean = false,
): [{ [key: string]: any }, ValidationError | null] => {
  // Parse value
  const [newValueFloat, err] = nullable
    ? parseNullableNumber(newValueStr)
    : parseNumber(newValueStr);
  if (err) return [container, err];

  // Build the new object
  return [
    {
      ...container,
      [field]: newValueFloat,
    },
    null,
  ];
};

export const changeBooleanData = (
  field: string,
  newValueStr: string,
  container: { [key: string]: any },
): [{ [key: string]: any }, ValidationError | null] => {
  // Parse value
  const [newValueBool, err] = parseBool(newValueStr);
  if (err) return [container, err];

  // Build the new object
  return [
    {
      ...container,
      [field]: newValueBool,
    },
    null,
  ];
};

export const changeNumberVecTData = (
  field: string,
  time: string,
  newValueStr: string,
  container: { [key: string]: any },
  scenario: UnitCommitmentScenario,
): [{ [key: string]: any }, ValidationError | null] => {
  // Parse value
  const [newValueFloat, err] = parseNumber(newValueStr);
  if (err) return [container, err];

  // Convert HH:MM to offset
  const hours = parseInt(time.split(":")[0]!, 10);
  const min = parseInt(time.split(":")[1]!, 10);
  const idx = (hours * 60 + min) / scenario.Parameters["Time step (min)"];

  // Build the new vector
  const newVec = [...container[field]];
  newVec[idx] = newValueFloat;
  return [
    {
      ...container,
      [field]: newVec,
    },
    null,
  ];
};

export const changeNumberVecNData = (
  field: string,
  offset: string,
  newValueStr: string,
  container: { [key: string]: any },
): [{ [key: string]: any }, ValidationError | null] => {
  const oldVec = container[field];
  const newVec = [...container[field]];
  const idx = parseInt(offset) - 1;

  if (newValueStr === "") {
    // Trim the vector
    newVec.splice(idx, oldVec.length - idx);
  } else {
    // Parse new value
    const [newValueFloat, err] = parseNumber(newValueStr);
    if (err) return [container, err];

    // Increase the length of the vector
    if (idx >= oldVec.length) {
      for (let i = oldVec.length; i < idx; i++) {
        newVec[i] = 0;
      }
    }

    // Assign new value
    newVec[idx] = newValueFloat;
  }
  return [
    {
      ...container,
      [field]: newVec,
    },
    null,
  ];
};

export const changeData = (
  field: string,
  newValueStr: string,
  container: { [key: string]: any },
  colSpecs: ColumnSpec[],
  scenario: UnitCommitmentScenario,
): [{ [key: string]: any }, ValidationError | null] => {
  const match = field.match(/^([^0-9]+)([0-9:]+)?$/);
  const fieldName = match![1]!.trim();
  const fieldOffset = match![2];
  for (const spec of colSpecs) {
    if (spec.title !== fieldName) continue;
    switch (spec.type) {
      case "string":
        return changeStringData(fieldName, newValueStr, container);
      case "busRef":
        return changeBusRefData(fieldName, newValueStr, container, scenario);
      case "number":
        return changeNumberData(fieldName, newValueStr, container);
      case "number?":
        return changeNumberData(fieldName, newValueStr, container, true);
      case "number[T]":
        return changeNumberVecTData(
          fieldName,
          fieldOffset!,
          newValueStr,
          container,
          scenario,
        );
      case "number[N]":
        return changeNumberVecNData(
          fieldName,
          fieldOffset!,
          newValueStr,
          container,
        );
      case "boolean":
        return changeBooleanData(fieldName, newValueStr, container);
      default:
        throw Error(`Unknown type: ${spec.type}`);
    }
  }
  throw Error(`Unknown field: ${fieldName}`);
};
export const assertBusesNotEmpty = (
  scenario: UnitCommitmentScenario,
): ValidationError | null => {
  if (Object.keys(scenario.Buses).length === 0)
    return { message: "This component requires an existing bus." };
  return null;
};