Merge branch 'dev' into ns2_recompile_when_script_write

This commit is contained in:
J 2019-06-02 23:50:57 -04:00 committed by GitHub
commit 3eaefa01f9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 90 additions and 26 deletions

@ -1139,7 +1139,7 @@ function NetscriptFunctions(workerScript) {
var oldScript = destServer.scripts[i]; var oldScript = destServer.scripts[i];
oldScript.code = sourceScript.code; oldScript.code = sourceScript.code;
oldScript.ramUsage = sourceScript.ramUsage; oldScript.ramUsage = sourceScript.ramUsage;
oldScript.markUpdated();; oldScript.markUpdated();
return true; return true;
} }
} }

@ -1,10 +1,22 @@
import { makeRuntimeRejectMsg } from "./NetscriptEvaluator"; import { makeRuntimeRejectMsg } from "./NetscriptEvaluator";
import { Script } from "./Script/Script";
// Makes a blob that contains the code of a given script. // Makes a blob that contains the code of a given script.
export function makeScriptBlob(code) { export function makeScriptBlob(code) {
return new Blob([code], {type: "text/javascript"}); return new Blob([code], {type: "text/javascript"});
} }
class ScriptUrl {
/**
* @param {string} filename
* @param {string} url
*/
constructor(filename, url) {
this.filename = filename;
this.url = url;
}
}
// Begin executing a user JS script, and return a promise that resolves // Begin executing a user JS script, and return a promise that resolves
// or rejects when the script finishes. // or rejects when the script finishes.
// - script is a script to execute (see Script.js). We depend only on .filename and .code. // - script is a script to execute (see Script.js). We depend only on .filename and .code.
@ -15,9 +27,9 @@ export function makeScriptBlob(code) {
// running the main function of the script. // running the main function of the script.
export async function executeJSScript(scripts = [], workerScript) { export async function executeJSScript(scripts = [], workerScript) {
let loadedModule; let loadedModule;
let urlStack = null; let urls = null;
let script = workerScript.getScript(); let script = workerScript.getScript();
if (script.module === "") { if (shouldCompile(script, scripts)) {
// The URL at the top is the one we want to import. It will // The URL at the top is the one we want to import. It will
// recursively import all the other modules in the urlStack. // recursively import all the other modules in the urlStack.
// //
@ -25,10 +37,11 @@ export async function executeJSScript(scripts = [], workerScript) {
// but not really behaves like import. Particularly, it cannot // but not really behaves like import. Particularly, it cannot
// load fully dynamic content. So we hide the import from webpack // load fully dynamic content. So we hide the import from webpack
// by placing it inside an eval call. // by placing it inside an eval call.
urlStack = _getScriptUrls(script, scripts, []); urls = _getScriptUrls(script, scripts, []);
script.module = await eval('import(urlStack[urlStack.length - 1])'); script.module = new Promise(resolve => resolve(eval('import(urls[urls.length - 1].url)')));
script.dependencies = urls.map(u => u.filename);
} }
loadedModule = script.module; loadedModule = await script.module;
let ns = workerScript.env.vars; let ns = workerScript.env.vars;
@ -41,12 +54,31 @@ export async function executeJSScript(scripts = [], workerScript) {
return loadedModule.main(ns); return loadedModule.main(ns);
} finally { } finally {
// Revoke the generated URLs // Revoke the generated URLs
if (urlStack != null) { if (urls != null) {
for (const url in urlStack) URL.revokeObjectURL(url); for (const b in urls) URL.revokeObjectURL(b.url);
} }
}; };
} }
/** Returns whether we should compile the script parameter.
*
* @param {Script} script
* @param {Script[]} scripts
*/
function shouldCompile(script, scripts) {
if (script.module === "") return true;
return script.dependencies.some(dep => {
const depScript = scripts.find(s => s.filename == dep);
// If the script is not present on the server, we should recompile, if only to get any necessary
// compilation errors.
if (!depScript) return true;
const depIsMoreRecent = depScript.moduleSequenceNumber > script.moduleSequenceNumber
return depIsMoreRecent;
});
}
// Gets a stack of blob urls, the top/right-most element being // Gets a stack of blob urls, the top/right-most element being
// the blob url for the named script on the named server. // the blob url for the named script on the named server.
// //
@ -58,8 +90,18 @@ export async function executeJSScript(scripts = [], workerScript) {
// different parts of the tree. That hasn't presented any problem with during // different parts of the tree. That hasn't presented any problem with during
// testing, but it might be an idea for the future. Would require a topo-sort // testing, but it might be an idea for the future. Would require a topo-sort
// then url-izing from leaf-most to root-most. // then url-izing from leaf-most to root-most.
/**
* @param {Script} script
* @param {Script[]} scripts
* @param {Script[]} seen
* @returns {ScriptUrl[]} All of the compiled scripts, with the final one
* in the list containing the blob corresponding to
* the script parameter.
*/
// BUG: apparently seen is never consulted. Oops.
export function _getScriptUrls(script, scripts, seen) { export function _getScriptUrls(script, scripts, seen) {
// Inspired by: https://stackoverflow.com/a/43834063/91401 // Inspired by: https://stackoverflow.com/a/43834063/91401
/** @type {ScriptUrl[]} */
const urlStack = []; const urlStack = [];
seen.push(script); seen.push(script);
try { try {
@ -86,7 +128,7 @@ export function _getScriptUrls(script, scripts, seen) {
// The top url in the stack is the replacement import file for this script. // The top url in the stack is the replacement import file for this script.
urlStack.push(...urls); urlStack.push(...urls);
return [prefix, urls[urls.length - 1], suffix].join(''); return [prefix, urls[urls.length - 1].url, suffix].join('');
} }
); );
@ -96,7 +138,7 @@ export function _getScriptUrls(script, scripts, seen) {
// If we successfully transformed the code, create a blob url for it and // If we successfully transformed the code, create a blob url for it and
// push that URL onto the top of the stack. // push that URL onto the top of the stack.
urlStack.push(URL.createObjectURL(makeScriptBlob(transformedCode))); urlStack.push(new ScriptUrl(script.filename, URL.createObjectURL(makeScriptBlob(transformedCode))));
return urlStack; return urlStack;
} catch (err) { } catch (err) {
// If there is an error, we need to clean up the URLs. // If there is an error, we need to clean up the URLs.

@ -15,6 +15,8 @@ import {
} from "../../utils/JSONReviver"; } from "../../utils/JSONReviver";
import { roundToTwo } from "../../utils/helpers/roundToTwo"; import { roundToTwo } from "../../utils/helpers/roundToTwo";
let globalModuleSequenceNumber = 0;
export class Script { export class Script {
// Initializes a Script Object from a JSON save state // Initializes a Script Object from a JSON save state
static fromJSON(value: any): Script { static fromJSON(value: any): Script {
@ -31,6 +33,14 @@ export class Script {
// This is only applicable for NetscriptJS // This is only applicable for NetscriptJS
module: any = ""; module: any = "";
// The timestamp when when the script was last updated.
moduleSequenceNumber: number;
// Only used with NS2 scripts; the list of dependency script filenames. This is constructed
// whenever the script is first evaluated, and therefore may be out of date if the script
// has been updated since it was last run.
dependencies: string[] = [];
// Amount of RAM this Script requres to run // Amount of RAM this Script requres to run
ramUsage: number = 0; ramUsage: number = 0;
@ -43,6 +53,7 @@ export class Script {
this.ramUsage = 0; this.ramUsage = 0;
this.server = server; // IP of server this script is on this.server = server; // IP of server this script is on
this.module = ""; this.module = "";
this.moduleSequenceNumber = ++globalModuleSequenceNumber;
if (this.code !== "") { this.updateRamUsage(otherScripts); } if (this.code !== "") { this.updateRamUsage(otherScripts); }
}; };
@ -98,6 +109,12 @@ export class Script {
} }
} }
// Marks that this script has been updated. Causes recompilation of NS2 modules.
markUpdated() {
this.module = "";
this.moduleSequenceNumber = ++globalModuleSequenceNumber;
}
/** /**
* Calculates and updates the script's RAM usage based on its code * Calculates and updates the script's RAM usage based on its code
* @param {Script[]} otherScripts - Other scripts on the server. Used to process imports * @param {Script[]} otherScripts - Other scripts on the server. Used to process imports

@ -1,3 +1,5 @@
import { KEY } from "./helpers/keyCodes";
/** /**
* Create and display a pop-up dialog box. * Create and display a pop-up dialog box.
* This dialog box does not allow for any interaction and should close when clicking * This dialog box does not allow for any interaction and should close when clicking
@ -9,29 +11,32 @@ let dialogBoxes = [];
$(document).click(function(event) { $(document).click(function(event) {
if (dialogBoxOpened && dialogBoxes.length >= 1) { if (dialogBoxOpened && dialogBoxes.length >= 1) {
if (!$(event.target).closest(dialogBoxes[0]).length){ if (!$(event.target).closest(dialogBoxes[0]).length){
dialogBoxes[0].remove(); closeTopmostDialogBox();
dialogBoxes.splice(0, 1);
if (dialogBoxes.length == 0) {
dialogBoxOpened = false;
} else {
dialogBoxes[0].style.visibility = "visible";
}
} }
} }
}); });
function closeTopmostDialogBox() {
// Dialog box close buttons if (!dialogBoxOpened || dialogBoxes.length === 0) return;
$(document).on('click', '.dialog-box-close-button', function( event ) {
if (dialogBoxOpened && dialogBoxes.length >= 1) {
dialogBoxes[0].remove(); dialogBoxes[0].remove();
dialogBoxes.splice(0, 1); dialogBoxes.shift();
if (dialogBoxes.length == 0) { if (dialogBoxes.length == 0) {
dialogBoxOpened = false; dialogBoxOpened = false;
} else { } else {
dialogBoxes[0].style.visibility = "visible"; dialogBoxes[0].style.visibility = "visible";
} }
} }
// Dialog box close buttons
$(document).on('click', '.dialog-box-close-button', function( event ) {
closeTopmostDialogBox();
});
document.addEventListener("keydown", function (event) {
if (event.keyCode == KEY.ESC && dialogBoxOpened) {
closeTopmostDialogBox();
event.preventDefault();
}
}); });
let dialogBoxOpened = false; let dialogBoxOpened = false;