blitzkit/packages/website-ue/src/core/tankopedia/computeCharacteristics.ts

import {
  ShellUpgrageSingleChange_AttributeName,
  TankAttributeChange_AttributeName,
} from "@protos/blitz_static_tank_upgrade_single_stage";
import type { Tank } from "../../protos/tank";
import { aggregateStageParameters } from "./aggregateStageParameters";
import {
  characteristics,
  type CharacteristicName,
  type CharacteristicOutput,
} from "./characteristics";
import type { TankState } from "./tankState";

export type ComputedCharacteristics = Record<
  CharacteristicName,
  CharacteristicOutput
>;

export function computeCharacteristics(
  id: string,
  tank: Tank,
  state: TankState
) {
  const parameters = aggregateStageParameters(
    tank.tank!.base_stats!,
    tank.tank!.upgrade_stages.slice(0, state.stage)
  );

  const computed: Partial<ComputedCharacteristics> = {};

  function characteristic(name: CharacteristicName) {
    if (name in computed) return computed[name] as CharacteristicOutput;
    throw new Error(`${name} not computed yet`);
  }

  function attributeSafe(name: TankAttributeChange_AttributeName) {
    const attribute = parameters.attributes.find(
      (attribute) => attribute.attribute_name === name
    );

    return attribute ? attribute.value : null;
  }

  function attribute(name: TankAttributeChange_AttributeName) {
    const attribute = attributeSafe(name);

    if (attribute === null) {
      throw new Error(
        `Attribute ${TankAttributeChange_AttributeName[name]} (${name}) not found`
      );
    }

    return attribute;
  }

  function shellSafe(name: ShellUpgrageSingleChange_AttributeName) {
    const shell = parameters.shells_upgrades.find(
      (shell) => shell.shell_id === state.shell
    );

    if (!shell) {
      throw new Error(
        `Shell ${ShellUpgrageSingleChange_AttributeName[name]} (${name}) not found`
      );
    }

    const attribute = shell.changes.find(
      (change) => change.attribute_name === name
    );

    return attribute ? attribute.value : null;
  }

  function shell(name: ShellUpgrageSingleChange_AttributeName) {
    const attribute = shellSafe(name);

    if (attribute === null) {
      throw new Error(
        `Shell attribute ${ShellUpgrageSingleChange_AttributeName[name]} (${name}) not found`
      );
    }

    return attribute;
  }

  for (const _name in characteristics) {
    const name = _name as CharacteristicName;
    const value = characteristics[name]({
      characteristic,
      attribute,
      attributeSafe,
      shell,
      shellSafe,
      state,
      tank,
      parameters,
      id,
    });
    computed[name] = value;
  }

  return {
    characteristics: computed as ComputedCharacteristics,
    parameters,
  };
}