MISC: Add source map to transformed scripts

package.json

@@ -29,6 +29,7 @@
     "bcryptjs": "^2.4.3",
     "better-react-mathjax": "^2.0.3",
     "clsx": "^1.2.1",
+    "convert-source-map": "^2.0.0",
     "date-fns": "^2.30.0",
     "escodegen": "^2.1.0",
     "jszip": "^3.10.1",
@@ -61,6 +62,7 @@
     "@swc/core": "^1.4.14",
     "@types/babel__standalone": "^7.1.7",
     "@types/bcryptjs": "^2.4.4",
+    "@types/convert-source-map": "^2.0.3",
     "@types/escodegen": "^0.0.7",
     "@types/file-saver": "^2.0.5",
     "@types/jest": "^29.5.5",

src/NetscriptJSEvaluator.ts

@@ -5,6 +5,7 @@
 import * as walk from "acorn-walk";
 import { parse } from "acorn";
 import type * as acorn from "acorn";
+import { fromObject } from "convert-source-map";
 
 import { LoadedModule, type ScriptURL, type ScriptModule } from "./Script/LoadedModule";
 import type { Script } from "./Script/Script";
@@ -82,6 +83,7 @@ function generateLoadedModule(script: Script, scripts: Map<ScriptFilePath, Scrip
   }
 
   let scriptCode;
+  let sourceMap;
   const fileType = getFileType(script.filename);
   switch (fileType) {
     case FileType.JS:
@@ -90,7 +92,7 @@ function generateLoadedModule(script: Script, scripts: Map<ScriptFilePath, Scrip
     case FileType.JSX:
     case FileType.TS:
     case FileType.TSX:
-      scriptCode = transformScript(script.code, fileType);
+      ({ scriptCode, sourceMap } = transformScript(script.code, fileType));
       break;
     default:
       throw new Error(`Invalid file type: ${fileType}. Filename: ${script.filename}, server: ${script.server}.`);
@@ -176,7 +178,19 @@ function generateLoadedModule(script: Script, scripts: Map<ScriptFilePath, Scrip
     // servers; it will be listed under the first server it was compiled for.
     // We don't include this in the cache key, so that other instances of the
     // script dedupe properly.
-    const adjustedCode = newCode + `\n//# sourceURL=${script.server}/${script.filename}`;
+    const sourceURL = `${script.server}/${script.filename}`;
+    let adjustedCode = newCode + `\n//# sourceURL=${sourceURL}`;
+    if (sourceMap) {
+      let inlineSourceMap;
+      try {
+        inlineSourceMap = fromObject({ ...JSON.parse(sourceMap), sources: [sourceURL], sourceRoot: "/" }).toComment();
+      } catch (error) {
+        console.warn(`Cannot generate inline source map for ${script.filename} in ${script.server}`, error);
+      }
+      if (inlineSourceMap) {
+        adjustedCode += `\n${inlineSourceMap}`;
+      }
+    }
     // At this point we have the full code and can construct a new blob / assign the URL.
 
     const url = URL.createObjectURL(makeScriptBlob(adjustedCode)) as ScriptURL;

src/utils/ScriptTransformer.ts

@@ -123,12 +123,11 @@ export function getModuleScript(
 /**
  * This function must be synchronous to avoid race conditions. Check https://github.com/bitburner-official/bitburner-src/pull/1173#issuecomment-2026940461
  * for more information.
- *
- * @param code
- * @param fileType
- * @returns
  */
-export function transformScript(code: string, fileType: FileType): string | null | undefined {
+export function transformScript(
+  code: string,
+  fileType: FileType,
+): { scriptCode: string; sourceMap: string | undefined } {
   if (supportedFileTypes.every((v) => v !== fileType)) {
     throw new Error(`Invalid file type: ${fileType}`);
   }
@@ -149,10 +148,15 @@ export function transformScript(code: string, fileType: FileType): string | null
       parserConfig.jsx = true;
     }
   }
-  return transformSync(code, {
+  const result = transformSync(code, {
     jsc: {
       parser: parserConfig,
       target: "es2020",
     },
-  }).code;
+    sourceMaps: true,
+  });
+  return {
+    scriptCode: result.code,
+    sourceMap: result.map,
+  };
 }