bitburner-src/src/Locations/ui/GymLocation.tsx

/**
 * React Subcomponent for displaying a location's UI, when that location is a gym
 *
 * This subcomponent renders all of the buttons for training at the gym
 */
import * as React from "react";
import Button from "@mui/material/Button";

import { Location } from "../Location";

import { IPlayer } from "../../PersonObjects/IPlayer";

import { Money } from "../../ui/React/Money";
import { IRouter } from "../../ui/Router";
import { Box } from "@mui/material";
import { ClassWork, ClassType, Classes } from "../../Work/ClassWork";
import { calculateCost } from "../../Work/formulas/Class";

type IProps = {
  loc: Location;
  p: IPlayer;
  router: IRouter;
};

export function GymLocation(props: IProps): React.ReactElement {
  function train(stat: ClassType): void {
    props.p.startNEWWork(
      new ClassWork({
        classType: stat,
        location: props.loc.name,
        singularity: false,
      }),
    );
    props.p.startFocusing();
    props.router.toWork();
  }

  const cost = calculateCost(Classes[ClassType.GymStrength], props.loc);

  return (
    <Box sx={{ display: "grid", width: "fit-content" }}>
      <Button onClick={() => train(ClassType.GymStrength)}>
        Train Strength (<Money money={cost} player={props.p} /> / sec)
      </Button>
      <Button onClick={() => train(ClassType.GymDefense)}>
        Train Defense (<Money money={cost} player={props.p} /> / sec)
      </Button>
      <Button onClick={() => train(ClassType.GymDexterity)}>
        Train Dexterity (<Money money={cost} player={props.p} /> / sec)
      </Button>
      <Button onClick={() => train(ClassType.GymAgility)}>
        Train Agility (<Money money={cost} player={props.p} /> / sec)
      </Button>
    </Box>
  );
}