bitburner-src/src/Corporation/Warehouse.ts
2021-10-13 02:27:55 -04:00

120 lines
3.7 KiB
TypeScript

import { Material } from "./Material";
import { ICorporation } from "./ICorporation";
import { IIndustry } from "./IIndustry";
import { MaterialSizes } from "./MaterialSizes";
import { IMap } from "../types";
import { Generic_fromJSON, Generic_toJSON, Reviver } from "../utils/JSONReviver";
import { exceptionAlert } from "../utils/helpers/exceptionAlert";
interface IConstructorParams {
corp?: ICorporation;
industry?: IIndustry;
loc?: string;
size?: number;
}
export class Warehouse {
// Warehouse's level, which affects its maximum size
level = 1;
// City that this Warehouse is in
loc: string;
// Map of Materials held by this Warehouse
materials: IMap<Material>;
// Maximum amount warehouse can hold
size: number;
// Amount of space currently used by warehouse
sizeUsed = 0;
// Whether Smart Supply is enabled for this Industry (the Industry that this Warehouse is for)
smartSupplyEnabled = false;
// Decide if smart supply should use the materials already in the warehouse when deciding on the amount to buy.
smartSupplyUseLeftovers: { [key: string]: boolean | undefined } = {};
// 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,
// so it will always trail the "true" production by 1 cycle
smartSupplyStore = 0;
constructor(params: IConstructorParams = {}) {
this.loc = params.loc ? params.loc : "";
this.size = params.size ? params.size : 0;
this.materials = {
Water: new Material({ name: "Water" }),
Energy: new Material({ name: "Energy" }),
Food: new Material({ name: "Food" }),
Plants: new Material({ name: "Plants" }),
Metal: new Material({ name: "Metal" }),
Hardware: new Material({ name: "Hardware" }),
Chemicals: new Material({ name: "Chemicals" }),
Drugs: new Material({ name: "Drugs" }),
Robots: new Material({ name: "Robots" }),
AICores: new Material({ name: "AI Cores" }),
RealEstate: new Material({ name: "Real Estate" }),
};
this.smartSupplyUseLeftovers = {
Water: true,
Energy: true,
Food: true,
Plants: true,
Metal: true,
Hardware: true,
Chemicals: true,
Drugs: true,
Robots: true,
AICores: true,
RealEstate: true,
};
if (params.corp && params.industry) {
this.updateSize(params.corp, params.industry);
}
// Default smart supply to being enabled if the upgrade is unlocked
if (params.corp?.unlockUpgrades[1]) {
this.smartSupplyEnabled = true;
}
}
// Re-calculate how much space is being used by this Warehouse
updateMaterialSizeUsed(): void {
this.sizeUsed = 0;
for (const matName in this.materials) {
const mat = this.materials[matName];
if (MaterialSizes.hasOwnProperty(matName)) {
this.sizeUsed += mat.qty * MaterialSizes[matName];
}
}
if (this.sizeUsed > this.size) {
console.warn("Warehouse size used greater than capacity, something went wrong");
}
}
updateSize(corporation: ICorporation, industry: IIndustry): void {
try {
this.size = this.level * 100 * corporation.getStorageMultiplier() * industry.getStorageMultiplier();
} catch (e: any) {
exceptionAlert(e);
}
}
// Serialize the current object to a JSON save state.
toJSON(): any {
return Generic_toJSON("Warehouse", this);
}
// Initiatizes a Warehouse object from a JSON save state.
// eslint-disable-next-line @typescript-eslint/explicit-module-boundary-types
static fromJSON(value: any): Warehouse {
return Generic_fromJSON(Warehouse, value.data);
}
}
Reviver.constructors.Warehouse = Warehouse;