Add ns.ui.setTheme() and ns.ui.resetTheme()

src/Netscript/RamCostGenerator.ts

@@ -361,6 +361,8 @@ export const RamCosts: IMap<any> = {
 
   ui: {
     getTheme: 0,
+    setTheme: 0,
+    resetTheme: 0,
   },
 
   heart: {

src/NetscriptFunctions/UserInterface.ts

@@ -4,6 +4,8 @@ import { IPlayer } from "../PersonObjects/IPlayer";
 import { getRamCost } from "../Netscript/RamCostGenerator";
 import { UserInterface as IUserInterface, UserInterfaceTheme } from "../ScriptEditor/NetscriptDefinitions";
 import { Settings } from "../Settings/Settings";
+import { ThemeEvents } from "../ui/React/Theme";
+import { defaultTheme } from "../Settings/Themes";
 
 export function NetscriptUserInterface(
   player: IPlayer,
@@ -15,5 +17,37 @@ export function NetscriptUserInterface(
       helper.updateDynamicRam("getTheme", getRamCost(player, "ui", "getTheme"));
       return { ...Settings.theme };
     },
-  };
+
+    setTheme: function (newTheme: UserInterfaceTheme): void {
+      helper.updateDynamicRam("setTheme", getRamCost(player, "ui", "setTheme"));
+      const hex = /^(#)((?:[A-Fa-f0-9]{3}){1,2})$/;
+      const currentTheme = {...Settings.theme}
+      const errors: string[] = [];
+      for (const key of Object.keys(newTheme)) {
+        if (!currentTheme[key]) {
+          // Invalid key
+          errors.push(`Invalid key "${key}"`);
+        } else if (!hex.test(newTheme[key] ?? '')) {
+          errors.push(`Invalid color "${key}": ${newTheme[key]}`);
+        } else {
+          currentTheme[key] = newTheme[key];
+        }
+      }
+
+      if (errors.length === 0) {
+        Object.assign(Settings.theme, currentTheme);
+        ThemeEvents.emit();
+        workerScript.log("ui.setTheme", () => `Successfully set theme`);
+      } else {
+        workerScript.log("ui.setTheme", () => `Failed to set theme. Errors: ${errors.join(', ')}`);
+      }
+    },
+
+    resetTheme: function (): void {
+      helper.updateDynamicRam("resetTheme", getRamCost(player, "ui", "resetTheme"));
+      Settings.theme = defaultTheme;
+      ThemeEvents.emit();
+      workerScript.log("ui.resetTheme", () => `Reinitialized theme to default`);
+    },
+  }
 }

src/ScriptEditor/NetscriptDefinitions.d.ts

@@ -1321,10 +1321,10 @@ export interface Singularity {
 
   /**
    * SF4.1 - Workout at the gym.
-   * 
+   *
    * @remarks
    * RAM cost: 2 GB
-   * 
+   *
 
    * This function will automatically set you to start working out at a gym to train
    * a particular stat. If you are already in the middle of some “working” action
@@ -3833,6 +3833,27 @@ interface UserInterface {
    * @returns An object containing the theme's colors
    */
   getTheme(): UserInterfaceTheme;
+
+  /**
+   * Sets the current theme
+   * @remarks
+   * RAM cost: cost: 0 GB
+   * @example
+   * Usage example (NS2)
+   * ```ts
+   * const theme = ns.ui.getTheme();
+   * theme.primary = '#ff5500';
+   * ns.ui.setTheme(theme);
+   * ```
+   */
+  setTheme(newTheme: UserInterfaceTheme): void;
+
+  /**
+   * Resets the player's theme to the default values
+   * @remarks
+   * RAM cost: cost: 0 GB
+  */
+  resetTheme(): void;
 }
 
 /**