From a88f36e2f439a42080a7b665843f7a8fd6bb3fba Mon Sep 17 00:00:00 2001
From: Mughur <Michael.cheitel@gmail.com>
Date: Sat, 30 Jul 2022 18:57:33 +0300
Subject: [PATCH 1/2] add corp get functions, fix coffee hover text

---
 src/Corporation/data/Constants.ts          | 63 ++++++++++++++++++++++
 src/Corporation/ui/IndustryOffice.tsx      |  2 +-
 src/NetscriptFunctions/Corporation.ts      | 25 +++++++++
 src/ScriptEditor/NetscriptDefinitions.d.ts | 25 +++++++++
 4 files changed, 114 insertions(+), 1 deletion(-)

diff --git a/src/Corporation/data/Constants.ts b/src/Corporation/data/Constants.ts
index c83e3e0d7..954a596b5 100644
--- a/src/Corporation/data/Constants.ts
+++ b/src/Corporation/data/Constants.ts
@@ -26,6 +26,10 @@ export const CorporationConstants: {
   BaseMaxProducts: number;
   AllCorporationStates: string[];
   AllMaterials: string[];
+  AllIndustryTypes: string[];
+  AllUnlocks: string[];
+  AllUpgrades: string[];
+  AllResearch: string[];
   FundingRoundShares: number[];
   FundingRoundMultiplier: number[];
   AvgProfitLength: number;
@@ -82,6 +86,65 @@ export const CorporationConstants: {
     "AI Cores",
     "Real Estate",
   ],
+  AllIndustryTypes: [
+    "Energy",
+    "Utilities",
+    "Agriculture",
+    "Fishing",
+    "Mining",
+    "Food",
+    "Tobacco",
+    "Chemical",
+    "Pharmaceutical",
+    "Hardware",
+    "Robotics",
+    "Software",
+    "Healthcare",
+    "RealEstate",
+  ],
+  AllUnlocks: [
+    "Export",
+    "Smart Supply",
+    "Market Research - Demand",
+    "Market Data - Competition",
+    "VeChain",
+    "Shady Accounting",
+    "Government Partnership",
+    "Warehouse API",
+    "Office API",
+  ],
+  AllUpgrades: [
+    "Smart Factories",
+    "Smart Storage",
+    "DreamSense",
+    "Wilson Analytics",
+    "Nuoptimal Nootropic Injector Implants",
+    "Speech Processor Implants",
+    "Neural Accelerators",
+    "FocusWires",
+    "ABC SalesBots",
+    "Project Insight",
+  ],
+  AllResearch: [
+    "Hi-Tech R&D Laboratory",
+    "AutoBrew",
+    "AutoPartyManager",
+    "Automatic Drug Administration",
+    "Bulk Purchasing",
+    "CPH4 Injections",
+    "Drones",
+    "Drones - Assembly",
+    "Drones - Transport",
+    "Go-Juice",
+    "HRBuddy-Recruitment",
+    "HRBuddy-Training",
+    "JoyWire",
+    "Market-TA.I",
+    "Market-TA.II",
+    "Overclock",
+    "Self-Correcting Assemblers",
+    "Sti.mu",
+  ],
   FundingRoundShares: [0.1, 0.35, 0.25, 0.2],
   FundingRoundMultiplier: [4, 3, 3, 2.5],
 
diff --git a/src/Corporation/ui/IndustryOffice.tsx b/src/Corporation/ui/IndustryOffice.tsx
index 17760f36b..87509efce 100644
--- a/src/Corporation/ui/IndustryOffice.tsx
+++ b/src/Corporation/ui/IndustryOffice.tsx
@@ -459,7 +459,7 @@ export function IndustryOffice(props: IProps): React.ReactElement {
           {!division.hasResearch("AutoBrew") && (
             <>
               <Tooltip
-                title={<Typography>Throw an office party to increase your employee's morale and happiness</Typography>}
+                title={<Typography>Provide your employees with coffee, increasing their energy by 5%</Typography>}
               >
                 <span>
                   <Button
diff --git a/src/NetscriptFunctions/Corporation.ts b/src/NetscriptFunctions/Corporation.ts
index 023d990db..f154c4025 100644
--- a/src/NetscriptFunctions/Corporation.ts
+++ b/src/NetscriptFunctions/Corporation.ts
@@ -871,6 +871,31 @@ export function NetscriptCorporation(player: IPlayer): InternalAPI<NSCorporation
   return {
     ...warehouseAPI,
     ...officeAPI,
+  getMaterialNames:
+    (ctx: NetscriptContext) => 
+    (): string[] =>{
+      return CorporationConstants.AllMaterials;
+  },
+  getIndustryTypes:
+    (ctx: NetscriptContext) => 
+    (): string[] =>{
+      return CorporationConstants.AllIndustryTypes;
+  },
+  getUnlockables:
+    (ctx: NetscriptContext) => 
+    (): string[] =>{
+      return CorporationConstants.AllUnlocks;
+  },
+  getUpgradeNames:
+    (ctx: NetscriptContext) => 
+    (): string[] =>{
+      return CorporationConstants.AllUpgrades;
+  },
+  getResarchNames:
+    (ctx: NetscriptContext) => 
+    (): string[] =>{
+      return CorporationConstants.AllResearch;
+  },
     expandIndustry:
       (ctx: NetscriptContext) =>
       (_industryName: unknown, _divisionName: unknown): void => {
diff --git a/src/ScriptEditor/NetscriptDefinitions.d.ts b/src/ScriptEditor/NetscriptDefinitions.d.ts
index 391044699..890405d2d 100644
--- a/src/ScriptEditor/NetscriptDefinitions.d.ts
+++ b/src/ScriptEditor/NetscriptDefinitions.d.ts
@@ -6920,6 +6920,31 @@ export interface Corporation extends WarehouseAPI, OfficeAPI {
    * @returns An offer of investment
    */
   getInvestmentOffer(): InvestmentOffer;
+  /**
+   * Get list of materials
+   * @returns material names
+   */
+  getMaterialNames():string[];
+  /**
+   * Get list of industry types
+   * @returns industry names
+   */
+  getIndustryTypes():string[];
+  /**
+   * Get list of one-time unlockable upgrades
+   * @returns unlockable upgrades names
+   */
+  getUnlockables():string[];
+  /**
+   * Get list of upgrade names
+   * @returns upgrade names
+   */
+  getUpgradeNames():string[];
+  /**
+   * Get list of research names
+   * @returns research names
+   */
+  getResarchNames():string[];
   /**
    * Accept investment based on you companies current valuation
    * @remarks

From 451bd0fb545354fdc75e37324648fa1834467b0a Mon Sep 17 00:00:00 2001
From: Mughur <Michael.cheitel@gmail.com>
Date: Sat, 30 Jul 2022 19:08:17 +0300
Subject: [PATCH 2/2] add missing corp researches to constant

---
 src/Corporation/data/Constants.ts | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/Corporation/data/Constants.ts b/src/Corporation/data/Constants.ts
index 954a596b5..5c4fa3f0d 100644
--- a/src/Corporation/data/Constants.ts
+++ b/src/Corporation/data/Constants.ts
@@ -144,6 +144,11 @@ export const CorporationConstants: {
     "Overclock",
     "Self-Correcting Assemblers",
     "Sti.mu",
+    "sudo.Assist",
+    "uPgrade: Capacity.I",
+    "uPgrade: Capacity.II",
+    "uPgrade: Dashboard",
+    "uPgrade: Fulcrum",
   ],
   FundingRoundShares: [0.1, 0.35, 0.25, 0.2],
   FundingRoundMultiplier: [4, 3, 3, 2.5],