Merge pull request #2343 from MartinFournier/feature/set-theme

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

markdown/bitburner.userinterface.md

@@ -17,4 +17,6 @@ interface UserInterface
 |  Method | Description |
 |  --- | --- |
 |  [getTheme()](./bitburner.userinterface.gettheme.md) | Get the current theme |
+|  [resetTheme()](./bitburner.userinterface.resettheme.md) | Resets the player's theme to the default values |
+|  [setTheme(newTheme)](./bitburner.userinterface.settheme.md) | Sets the current theme |
 

markdown/bitburner.userinterface.resettheme.md

@@ -0,0 +1,21 @@
+<!-- Do not edit this file. It is automatically generated by API Documenter. -->
+
+[Home](./index.md) &gt; [bitburner](./bitburner.md) &gt; [UserInterface](./bitburner.userinterface.md) &gt; [resetTheme](./bitburner.userinterface.resettheme.md)
+
+## UserInterface.resetTheme() method
+
+Resets the player's theme to the default values
+
+<b>Signature:</b>
+
+```typescript
+resetTheme(): void;
+```
+<b>Returns:</b>
+
+void
+
+## Remarks
+
+RAM cost: cost: 0 GB
+

markdown/bitburner.userinterface.settheme.md

@@ -0,0 +1,38 @@
+<!-- Do not edit this file. It is automatically generated by API Documenter. -->
+
+[Home](./index.md) &gt; [bitburner](./bitburner.md) &gt; [UserInterface](./bitburner.userinterface.md) &gt; [setTheme](./bitburner.userinterface.settheme.md)
+
+## UserInterface.setTheme() method
+
+Sets the current theme
+
+<b>Signature:</b>
+
+```typescript
+setTheme(newTheme: UserInterfaceTheme): void;
+```
+
+## Parameters
+
+|  Parameter | Type | Description |
+|  --- | --- | --- |
+|  newTheme | UserInterfaceTheme |  |
+
+<b>Returns:</b>
+
+void
+
+## Remarks
+
+RAM cost: cost: 0 GB
+
+## Example
+
+Usage example (NS2)
+
+```ts
+const theme = ns.ui.getTheme();
+theme.primary = '#ff5500';
+ns.ui.setTheme(theme);
+```
+

src/Netscript/RamCostGenerator.ts

@@ -362,6 +362,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

@@ -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;
 }
 
 /**