added atExit

doc/source/netscript/advancedfunctions/atExit.rst

@@ -0,0 +1,18 @@
+atExit() Netscript Function
+============================
+
+.. js:function:: atExit(f)
+
+    :RAM cost: 0 GB
+    :param function f: function to call when the script dies.
+
+    Runs when the script dies.
+
+    Example:
+
+    .. code-block:: javascript
+
+        function onDeath() {
+            console.log('I died!!!')
+        }
+        atExit(onDeath);

src/Netscript/WorkerScript.ts

@@ -106,6 +106,11 @@ export class WorkerScript {
    */
   hostname: string;
 
+  /**
+   * Function called when the script ends.
+   */
+  atExit: any;
+
   constructor(runningScriptObj: RunningScript, pid: number, nsFuncsGenerator?: (ws: WorkerScript) => any) {
     this.name = runningScriptObj.filename;
     this.hostname = runningScriptObj.server;

src/Netscript/killWorkerScript.ts

@@ -10,6 +10,7 @@ import { RunningScript } from "../Script/RunningScript";
 import { GetServer } from "../Server/AllServers";
 
 import { compareArrays } from "../utils/helpers/compareArrays";
+import { dialogBoxCreate } from "../ui/React/DialogBox";
 
 export function killWorkerScript(runningScriptObj: RunningScript, hostname: string, rerenderUi?: boolean): boolean;
 export function killWorkerScript(workerScript: WorkerScript): boolean;
@@ -67,6 +68,16 @@ function killWorkerScriptByPid(pid: number, rerenderUi = true): boolean {
 function stopAndCleanUpWorkerScript(workerScript: WorkerScript, rerenderUi = true): void {
   workerScript.env.stopFlag = true;
   killNetscriptDelay(workerScript);
+  if (typeof workerScript.atExit === "function") {
+    try {
+      workerScript.atExit();
+    } catch (e: any) {
+      dialogBoxCreate(
+        `Error trying to call atExit for script ${workerScript.name} on ${workerScript.hostname} ${workerScript.scriptRef.args} ${e}`,
+      );
+    }
+    workerScript.atExit = undefined;
+  }
   removeWorkerScript(workerScript, rerenderUi);
 }
 

src/NetscriptFunctions.ts

@@ -3277,6 +3277,13 @@ function NetscriptFunctions(workerScript: WorkerScript): NS {
     corporation: corporation,
 
     formulas: formulas,
+
+    atExit: function (f: any): void {
+      if (typeof f !== "function") {
+        throw makeRuntimeErrorMsg("atExit", "argument should be function");
+      }
+      workerScript.atExit = f;
+    },
     flags: function (data: any): any {
       data = toNative(data);
       // We always want the help flag.