blitzkit/packages/website/src/components/Tankopedia/TechTreeSection/index.tsx

import { alias } from "@blitzkit/core";
import { literals } from "@blitzkit/i18n/src/literals";
import { TankType } from "@blitzkit/protos";
import { CaretLeftIcon, CaretRightIcon, PlusIcon } from "@radix-ui/react-icons";
import { Flex, Heading, IconButton, ScrollArea, Text } from "@radix-ui/themes";
import { Fragment, useEffect, useMemo, useRef, useState } from "react";
import { api } from "../../../api/dynamic";
import { useLocale } from "../../../hooks/useLocale";
import { Duel } from "../../../stores/duel";
import { Tankopedia } from "../../../stores/tankopedia";
import type { MaybeSkeletonComponentProps } from "../../../types/maybeSkeletonComponentProps";
import { Arrow } from "./components/Arrow";
import { Node } from "./components/Node";

type Line = number[];

const tankDefinitions = await api.tanks();

export const XP_MULTIPLIERS = [1, 2, 3, 4, 5, 10];

export function TechTreeSection({ skeleton }: MaybeSkeletonComponentProps) {
  const { locale, strings } = useLocale();
  const master = Duel.use((state) => state.protagonist.tank);
  const xpMultiplier = Tankopedia.use((state) => state.xpMultiplier);
  const container = useRef<HTMLDivElement>(null);
  const lines = useMemo(() => {
    function extend(line: Line): Line[] {
      const root = tankDefinitions.tanks[line.at(-1)!];

      if (
        root.ancestors === undefined ||
        root.ancestors.length === 0 ||
        root.tier === 1
      ) {
        return [line];
      } else {
        if (root.ancestors.length === 1 || root.tier === 2) {
          line.push(
            root.ancestors.find(
              (ancestor) => !tankDefinitions.tanks[ancestor].deprecated,
            ) ?? root.ancestors[0],
          );
          return extend(line);
        } else {
          return root.ancestors
            .map((ancestor) => {
              const newLine = [...line, ancestor];
              return extend(newLine);
            })
            .flat();
        }
      }
    }

    return extend([master.id]);
  }, [master]);
  const [lineIndex, setLineIndex] = useState(0);
  const line = useMemo(
    () => (lines.length === 0 ? [master.id] : [...lines[lineIndex]].reverse()),
    [master, lineIndex],
  );
  const totalXp = line.reduce(
    (xp, id) =>
      xp +
      (tankDefinitions.tanks[id].research_cost === undefined ||
      tankDefinitions.tanks[id].tier === 1
        ? 0
        : (tankDefinitions.tanks[id].research_cost.research_cost_type!
            .value as number)),
    0,
  );
  const totalCredits = line.reduce(
    (credits, id) => credits + tankDefinitions.tanks[id].price!.value,
    0,
  );

  useEffect(() => {
    if (!container.current) return;
    container.current.scrollLeft = container.current.scrollWidth;
  });

  if (
    master.type !== TankType.TANK_TYPE_RESEARCHABLE ||
    master.ancestors === undefined ||
    master.ancestors.length === 0
  ) {
    return null;
  }

  return (
    <Flex
      direction="column"
      align="center"
      gap="0"
      style={{ backgroundColor: "var(--color-surface)" }}
      py="6"
    >
      <Flex direction="column" align="center">
        <Heading size="6">
          {strings.website.tools.tankopedia.tech_tree.title}
        </Heading>

        {master.tier > 1 && (
          <Flex gap="4" align="center">
            <Text color="gray" wrap="nowrap">
              <Flex gap="1" align="center">
                <img
                  alt="XP"
                  src={alias("api", "/icons/currencies/xp.webp")}
                  style={{
                    width: "1em",
                    height: "1em",
                    objectFit: "contain",
                    objectPosition: "center",
                  }}
                />
                {Math.round(totalXp / xpMultiplier).toLocaleString(locale)}
              </Flex>
            </Text>

            <Text color="gray" wrap="nowrap">
              <Flex gap="1" align="center">
                <img
                  alt="Silver"
                  src={alias("api", "/icons/currencies/silver.webp")}
                  style={{
                    width: "1em",
                    height: "1em",
                    objectFit: "contain",
                    objectPosition: "center",
                  }}
                />
                {totalCredits.toLocaleString(locale)}
              </Flex>
            </Text>
          </Flex>
        )}

        <Flex gap="1" mt="2">
          {XP_MULTIPLIERS.map((multiplier) => {
            const selected = xpMultiplier === multiplier;

            return (
              <Flex
                key={multiplier}
                style={{
                  background: "url(/assets/images/icons/xp-multiplier.png)",
                  backgroundSize: "contain",
                  backgroundRepeat: "no-repeat",
                  backgroundPosition: "center",
                  opacity: selected ? 1 : 0.5,
                  cursor: "pointer",
                }}
                onClick={() => {
                  Tankopedia.mutate((draft) => {
                    draft.xpMultiplier = multiplier;
                  });
                }}
                width="2.5rem"
                justify="center"
                align="center"
                pt="1"
                pl="1"
              >
                <Text size="2">
                  {literals(strings.common.units.x, { value: multiplier })}
                </Text>
              </Flex>
            );
          })}
        </Flex>
      </Flex>

      <ScrollArea type="hover" scrollbars="horizontal" ref={container}>
        <Flex align="center" gap="2" justify="center" p="4">
          {line.map((id, index) => {
            const last = index === line.length - 1;
            const tank = tankDefinitions.tanks[id];

            return (
              <Fragment key={id}>
                {index > 0 && <Arrow key={`${id}-arrow`} />}

                <Node
                  key={`${id}-node`}
                  id={id}
                  nextIds={last ? tank.successors : [line[index + 1]]}
                  highlight={last}
                  skeleton={skeleton}
                />
              </Fragment>
            );
          })}

          {master.successors.length > 0 && (
            <>
              <Arrow />
              {master.successors!.map((id, index) => (
                <Fragment key={id}>
                  {index > 0 && (
                    <Text color="gray" key={`${id}-plus`}>
                      <PlusIcon />
                    </Text>
                  )}
                  <Node key={`${id}-node}`} id={id} skeleton={skeleton} />
                </Fragment>
              ))}
            </>
          )}
        </Flex>
      </ScrollArea>

      <Flex>
        {lines.length > 1 && (
          <Flex align="center" gap="2">
            <IconButton
              size="2"
              variant="soft"
              onClick={() =>
                setLineIndex((lines.length + lineIndex - 1) % lines.length)
              }
            >
              <CaretLeftIcon />
            </IconButton>
            <Text>
              {literals(strings.website.tools.tankopedia.tech_tree.route, {
                index: lineIndex + 1,
                count: lines.length,
              })}
            </Text>
            <IconButton
              size="2"
              variant="soft"
              onClick={() => setLineIndex((lineIndex + 1) % lines.length)}
            >
              <CaretRightIcon />
            </IconButton>
          </Flex>
        )}
      </Flex>
    </Flex>
  );
}