MISC: tiny corp code style improvement (#586)

src/Corporation/Actions.ts

@@ -1,7 +1,11 @@
+import { isInteger } from "lodash";
+
 import { Player } from "@player";
+import { CorpResearchName, CorpSmartSupplyOption } from "@nsdefs";
+
 import { MaterialInfo } from "./MaterialInfo";
 import { Corporation } from "./Corporation";
-import { IndustryResearchTrees, IndustriesData } from "./IndustryData";
+import { IndustryResearchTrees, IndustriesData } from "./data/IndustryData";
 import { Division } from "./Division";
 import * as corpConstants from "./data/Constants";
 import { OfficeSpace } from "./OfficeSpace";
@@ -13,8 +17,6 @@ import { ResearchMap } from "./ResearchMap";
 import { isRelevantMaterial } from "./ui/Helpers";
 import { CityName } from "../Enums";
 import { getRandomInt } from "../utils/helpers/getRandomInt";
-import { CorpResearchName } from "@nsdefs";
-import { isInteger } from "lodash";
 import { getRecordValues } from "../Types/Record";
 
 export function NewDivision(corporation: Corporation, industry: IndustryType, name: string): void {
@@ -206,6 +208,7 @@ export function SellProduct(product: Product, city: CityName, amt: string, price
     if (temp == null || isNaN(parseFloat(temp))) {
       throw new Error("Invalid value or expression for sell quantity field");
     }
+
     if (all) {
       for (const cityName of Object.values(CityName)) {
         product.cityData[cityName].desiredSellAmount = qty; //Use sanitized input
@@ -242,10 +245,7 @@ export function SetSmartSupply(warehouse: Warehouse, smartSupply: boolean): void
   warehouse.smartSupplyEnabled = smartSupply;
 }
 
-export function SetSmartSupplyOption(warehouse: Warehouse, material: Material, useOption: string): void {
-  if (!corpConstants.smartSupplyUseOptions.includes(useOption)) {
-    throw new Error(`Invalid Smart Supply option '${useOption}'`);
-  }
+export function SetSmartSupplyOption(warehouse: Warehouse, material: Material, useOption: CorpSmartSupplyOption): void {
   warehouse.smartSupplyOptions[material.name] = useOption;
 }
 

src/Corporation/Corporation.ts

@@ -36,8 +36,10 @@ export class Corporation {
   revenue = 0;
   expenses = 0;
   fundingRound = 0;
-  public = false; //Publicly traded
-  totalShares = corpConstants.initialShares; // Total existing shares
+  /** Publicly traded */
+  public = false;
+  /** Total existing shares */
+  totalShares = corpConstants.initialShares;
   numShares = corpConstants.initialShares; // Total shares owned by player
   shareSalesUntilPriceUpdate = corpConstants.sharesPerPriceUpdate;
   shareSaleCooldown = 0; // Game cycles until player can sell shares again
@@ -64,7 +66,7 @@ export class Corporation {
   state = new CorporationState();
 
   constructor(params: IParams = {}) {
-    this.name = params.name ? params.name : "The Corporation";
+    this.name = params.name || "The Corporation";
     this.seedFunded = params.seedFunded ?? false;
   }
 
@@ -73,14 +75,14 @@ export class Corporation {
       console.error("Trying to add invalid amount of funds. Report to a developer.");
       return;
     }
-    this.funds = this.funds + amt;
+    this.funds += amt;
   }
 
   getState(): CorpStateName {
     return this.state.getState();
   }
 
-  storeCycles(numCycles = 1): void {
+  storeCycles(numCycles: number): void {
     this.storedCycles += numCycles;
   }
 

src/Corporation/Division.ts

@@ -1,6 +1,6 @@
 import { constructorsForReviver, Generic_toJSON, Generic_fromJSON, IReviverValue } from "../utils/JSONReviver";
 import { CityName } from "../Enums";
-import { IndustryResearchTrees, IndustriesData } from "./IndustryData";
+import { IndustryResearchTrees, IndustriesData } from "./data/IndustryData";
 import * as corpConstants from "./data/Constants";
 import { CorpEmployeeJob, IndustryType } from "./data/Enums";
 import { getRandomInt } from "../utils/helpers/getRandomInt";

src/Corporation/Product.ts

@@ -1,7 +1,7 @@
 import { CorpEmployeeJob } from "./data/Enums";
+import { IndustriesData } from "./data/IndustryData";
 import { MaterialInfo } from "./MaterialInfo";
 import { Division } from "./Division";
-import { IndustriesData } from "./IndustryData";
 
 import { Generic_fromJSON, Generic_toJSON, IReviverValue, constructorsForReviver } from "../utils/JSONReviver";
 import { getRandomInt } from "../utils/helpers/getRandomInt";

src/Corporation/Warehouse.ts

@@ -1,3 +1,6 @@
+import { Player } from "@player";
+import { CorpMaterialName, CorpSmartSupplyOption } from "@nsdefs";
+
 import { Material } from "./Material";
 import { Corporation } from "./Corporation";
 import { Division } from "./Division";
@@ -7,7 +10,6 @@ import { CityName } from "../Enums";
 import { materialNames } from "./data/Constants";
 import { createFullRecordFromEntries, getRecordEntries } from "../Types/Record";
 import { CorpUnlockName } from "./data/Enums";
-import { Player } from "@player";
 
 interface IConstructorParams {
   division: Division;
@@ -35,7 +37,9 @@ export class Warehouse {
   smartSupplyEnabled = false;
 
   // Decide if smart supply should use the amount of materials imported into account when deciding on the amount to buy.
-  smartSupplyOptions = createFullRecordFromEntries(materialNames.map((matName) => [matName, "leftovers"]));
+  smartSupplyOptions = createFullRecordFromEntries<CorpMaterialName, CorpSmartSupplyOption>(
+    materialNames.map((matName) => [matName, "leftovers"]),
+  );
 
   // Stores the amount of product to be produced. Used for Smart Supply unlock.
   // The production tracked by smart supply is always based on the previous cycle,

src/Corporation/data/Constants.ts

@@ -3,6 +3,7 @@ import {
   CorpIndustryName,
   CorpMaterialName,
   CorpResearchName,
+  CorpSmartSupplyOption,
   CorpStateName,
   CorpUnlockName,
   CorpUpgradeName,
@@ -121,8 +122,8 @@ export const stateNames: CorpStateName[] = ["START", "PURCHASE", "PRODUCTION", "
   valuationLength = 10,
   /** Minimum decay value for employee morale/energy */
   minEmployeeDecay = 10,
-  /**smart supply ot */
-  smartSupplyUseOptions = ["leftovers", "imports", "none"],
+  /** smart supply options */
+  smartSupplyOptions: CorpSmartSupplyOption[] = ["leftovers", "imports", "none"],
   PurchaseMultipliers = {
     x1: 1 as PositiveInteger,
     x5: 5 as PositiveInteger,

src/Corporation/data/IndustryData.ts

@@ -1,10 +1,8 @@
-import React from "react";
-import { Corporation } from "./Corporation";
-import { getBaseResearchTreeCopy, getProductIndustryResearchTreeCopy } from "./data/BaseResearchTree";
-import { MoneyCost } from "./ui/MoneyCost";
 import { CorpIndustryData } from "@nsdefs";
-import { IndustryType } from "./data/Enums";
-import { createFullRecordFromEntries } from "../Types/Record";
+
+import { createFullRecordFromEntries } from "../../Types/Record";
+import { IndustryType } from "./Enums";
+import { getBaseResearchTreeCopy, getProductIndustryResearchTreeCopy } from "./BaseResearchTree";
 
 export const IndustriesData: Record<IndustryType, CorpIndustryData> = {
   [IndustryType.Agriculture]: {
@@ -293,28 +291,6 @@ export const IndustriesData: Record<IndustryType, CorpIndustryData> = {
 
 export const IndustryStartingCosts = {};
 
-// Map of description for each industry
-export const IndustryDescriptions = (industry: IndustryType, corp: Corporation) => {
-  const data = IndustriesData[industry];
-  return (
-    <>
-      {data.description}
-      <br />
-      <br />
-      Required Materials: {Object.keys(data.requiredMaterials).toString().replace(/,/gi, ", ")}
-      <br />
-      Produces Materials: {data.producedMaterials ? data.producedMaterials.toString().replace(/,/gi, ", ") : "NONE"}
-      <br />
-      Produces products: {data.product ? "YES" : "NO"}
-      <br />
-      <br />
-      Starting cost: <MoneyCost money={data.startingCost} corp={corp} />
-      <br />
-      Recommended starting Industry: {data.recommendStarting ? "YES" : "NO"}
-    </>
-  );
-};
-
 export const IndustryResearchTrees = createFullRecordFromEntries(
   Object.values(IndustryType).map((industryType) => {
     return [

src/Corporation/ui/IndustryDescription.tsx

@@ -0,0 +1,34 @@
+import React from "react";
+
+import Typography from "@mui/material/Typography";
+
+import { MoneyCost } from "./MoneyCost";
+import { Corporation } from "../Corporation";
+import { IndustryType } from "../data/Enums";
+import { IndustriesData } from "../data/IndustryData";
+
+interface IProps {
+  industry: IndustryType;
+  corp: Corporation;
+}
+
+export const IndustryDescription = ({ industry, corp }: IProps) => {
+  const data = IndustriesData[industry];
+  return (
+    <Typography>
+      {data.description}
+      <br />
+      <br />
+      Required Materials: {Object.keys(data.requiredMaterials).toString().replace(/,/gi, ", ")}
+      <br />
+      Produces Materials: {data.producedMaterials ? data.producedMaterials.toString().replace(/,/gi, ", ") : "NONE"}
+      <br />
+      Produces products: {data.product ? "YES" : "NO"}
+      <br />
+      <br />
+      Starting cost: <MoneyCost money={data.startingCost} corp={corp} />
+      <br />
+      Recommended starting Industry: {data.recommendStarting ? "YES" : "NO"}
+    </Typography>
+  );
+};

src/Corporation/ui/NewDivisionTab.tsx

@@ -1,17 +1,20 @@
 import React, { useState } from "react";
+
+import Box from "@mui/material/Box";
+import MenuItem from "@mui/material/MenuItem";
+import Select, { SelectChangeEvent } from "@mui/material/Select";
+import TextField from "@mui/material/TextField";
+import Typography from "@mui/material/Typography";
+
 import { dialogBoxCreate } from "../../ui/React/DialogBox";
-import { IndustryDescriptions, IndustriesData } from "../IndustryData";
+import { IndustriesData } from "../data/IndustryData";
 import { IndustryType } from "../data/Enums";
 import { useCorporation } from "./Context";
 import { NewDivision } from "../Actions";
 
-import Typography from "@mui/material/Typography";
 import { ButtonWithTooltip } from "../../ui/Components/ButtonWithTooltip";
-import TextField from "@mui/material/TextField";
-import MenuItem from "@mui/material/MenuItem";
-import Box from "@mui/material/Box";
-import Select, { SelectChangeEvent } from "@mui/material/Select";
 import { KEY } from "../../utils/helpers/keyCodes";
+import { IndustryDescription } from "./IndustryDescription";
 interface IProps {
   setDivisionName: (name: string) => void;
 }
@@ -58,9 +61,6 @@ export function NewDivisionTab(props: IProps): React.ReactElement {
     setIndustry(event.target.value as IndustryType);
   }
 
-  const desc = IndustryDescriptions(industry, corp);
-  if (desc === undefined) throw new Error(`Desired industry for new division doesn't exists: '${industry}'`);
-
   return (
     <>
       <Typography>
@@ -74,7 +74,7 @@ export function NewDivisionTab(props: IProps): React.ReactElement {
           </MenuItem>
         ))}
       </Select>
-      <Typography>{desc}</Typography>
+      <IndustryDescription industry={industry} corp={corp} />
       <br />
       <br />
 

src/Corporation/ui/modals/MakeProductModal.tsx

@@ -1,7 +1,7 @@
 import React, { useState } from "react";
 import { dialogBoxCreate } from "../../../ui/React/DialogBox";
 import { Modal } from "../../../ui/React/Modal";
-import { IndustriesData } from "../../IndustryData";
+import { IndustriesData } from "../../data/IndustryData";
 import { IndustryType } from "../../data/Enums";
 import { MakeProduct } from "../../Actions";
 import { useCorporation, useDivision } from "../Context";

src/Corporation/ui/modals/ResearchModal.tsx

@@ -1,6 +1,6 @@
 import React, { useState } from "react";
 import { Modal } from "../../../ui/React/Modal";
-import { IndustryResearchTrees } from "../../IndustryData";
+import { IndustryResearchTrees } from "../../data/IndustryData";
 import * as corpConstants from "../../data/Constants";
 import { Division } from "../../Division";
 import { Research } from "../../Actions";

src/NetscriptFunctions/Corporation.ts

@@ -52,7 +52,7 @@ import {
 import { CorpUnlocks } from "../Corporation/data/CorporationUnlocks";
 import { CorpUpgrades } from "../Corporation/data/CorporationUpgrades";
 import { CorpUnlockName, CorpUpgradeName, CorpEmployeeJob, IndustryType } from "../Corporation/data/Enums";
-import { IndustriesData, IndustryResearchTrees } from "../Corporation/IndustryData";
+import { IndustriesData, IndustryResearchTrees } from "../Corporation/data/IndustryData";
 import * as corpConstants from "../Corporation/data/Constants";
 import { ResearchMap } from "../Corporation/ResearchMap";
 import { Factions } from "../Faction/Factions";
@@ -413,6 +413,7 @@ export function NetscriptCorporation(): InternalAPI<NSCorporation> {
       const warehouse = getWarehouse(divisionName, cityName);
       const material = getMaterial(divisionName, cityName, materialName);
       const option = helpers.string(ctx, "option", _option);
+      assertMember(ctx, corpConstants.smartSupplyOptions, "Smart Supply Option", "option", option);
       if (!hasUnlock(CorpUnlockName.SmartSupply))
         throw helpers.makeRuntimeErrorMsg(ctx, `You have not purchased the Smart Supply upgrade!`);
       SetSmartSupplyOption(warehouse, material, option);
@@ -682,8 +683,8 @@ export function NetscriptCorporation(): InternalAPI<NSCorporation> {
       /* TODO 2.2: possibly just rework the whole corp constants structure to be more readable, and just use cloneDeep
        *           to provide it directly to player.
        * TODO 2.2: Roll product information into industriesData, there's no reason to look up a product separately */
-      return cloneDeep(omit(corpConstants, "fundingRoundShares", "fundingRoundMultiplier", "valuationLength"));
       // TODO: add functions for getting materialInfo and research info
+      return cloneDeep(omit(corpConstants, "fundingRoundShares", "fundingRoundMultiplier", "valuationLength"));
     },
     getIndustryData: (ctx) => (_industryName) => {
       checkAccess(ctx);

src/PersonObjects/Player/PlayerObjectCorporationMethods.ts

@@ -1,6 +1,6 @@
 import { CorpUnlockName } from "../../Corporation/data/Enums";
+import { resetIndustryResearchTrees } from "../../Corporation/data/IndustryData";
 import { Corporation } from "../../Corporation/Corporation";
-import { resetIndustryResearchTrees } from "../../Corporation/IndustryData";
 
 import type { PlayerObject } from "./PlayerObject";
 

src/Prestige.ts

@@ -5,7 +5,7 @@ import { augmentationExists, initAugmentations } from "./Augmentation/Augmentati
 import { AugmentationNames } from "./Augmentation/data/AugmentationNames";
 import { initBitNodeMultipliers } from "./BitNode/BitNode";
 import { Companies, initCompanies } from "./Company/Companies";
-import { resetIndustryResearchTrees } from "./Corporation/IndustryData";
+import { resetIndustryResearchTrees } from "./Corporation/data/IndustryData";
 import { CompletedProgramName } from "./Programs/Programs";
 import { Factions, initFactions } from "./Faction/Factions";
 import { joinFaction } from "./Faction/FactionHelpers";

src/ScriptEditor/NetscriptDefinitions.d.ts

@@ -6747,6 +6747,9 @@ type CorpIndustryName =
   | "Healthcare"
   | "Real Estate";
 
+/** @public */
+type CorpSmartSupplyOption = "leftovers" | "imports" | "none";
+
 /** Names of all cities
  * @public */
 declare enum CityName {
@@ -7002,7 +7005,7 @@ export interface WarehouseAPI {
     divisionName: string,
     city: CityName | `${CityName}`,
     materialName: string,
-    option: string,
+    option: CorpSmartSupplyOption,
   ): void;
   /**
    * Set material buy data
@@ -7369,6 +7372,8 @@ interface CorporationInfo {
 interface CorpConstants {
   /** Names of all corporation game states */
   stateNames: CorpStateName[];
+  /** Names of all employee positions */
+  employeePositions: CorpEmployeePosition[];
   /** Names of all industries */
   industryNames: CorpIndustryName[];
   /** Names of all materials */
@@ -7412,6 +7417,7 @@ interface CorpConstants {
   maxProductsBase: number;
   /** The minimum decay value for morale/energy */
   minEmployeeDecay: number;
+  smartSupplyOptions: CorpSmartSupplyOption[];
 }
 /** @public */
 type CorpStateName = "START" | "PURCHASE" | "PRODUCTION" | "EXPORT" | "SALE";