Lots of tweaks

packages/plugos/environments/custom_logger.ts

@@ -0,0 +1,61 @@
+export type LogLevel = "info" | "warn" | "error" | "log";
+
+export class ConsoleLogger {
+  print: boolean;
+  callback: (level: LogLevel, entry: string) => void;
+
+  constructor(
+    callback: (level: LogLevel, entry: string) => void,
+    print: boolean = true
+  ) {
+    this.print = print;
+    this.callback = callback;
+  }
+
+  log(...args: any[]): void {
+    this.push("log", args);
+  }
+
+  warn(...args: any[]): void {
+    this.push("warn", args);
+  }
+
+  error(...args: any[]): void {
+    this.push("error", args);
+  }
+
+  info(...args: any[]): void {
+    this.push("info", args);
+  }
+
+  push(level: LogLevel, args: any[]) {
+    this.callback(level, this.logMessage(args));
+    if (this.print) {
+      console[level](...args);
+    }
+  }
+
+  logMessage(values: any[]): string {
+    let pieces: string[] = [];
+    for (let val of values) {
+      switch (typeof val) {
+        case "string":
+        case "number":
+          pieces.push("" + val);
+          break;
+        default:
+          try {
+            let s = JSON.stringify(val, null, 2);
+            if (s.length > 500) {
+              s = s.substring(0, 500) + "...";
+            }
+            pieces.push(s);
+          } catch {
+            // May be cyclical reference
+            pieces.push("[circular object]");
+          }
+      }
+    }
+    return pieces.join(" ");
+  }
+}

packages/plugos/environments/node_worker.ts

@@ -1,3 +1,5 @@
+import { ConsoleLogger } from "./custom_logger";
+
 const {
   parentPort,
   workerData: { preloadedModules, nodeModulesPath },
@@ -16,10 +18,18 @@ let pendingRequests = new Map<
 
 let syscallReqId = 0;
 
+let consoleLogger = new ConsoleLogger((level, message) => {
+  parentPort.postMessage({
+    type: "log",
+    level,
+    message,
+  });
+}, false);
+
 let vm = new VM({
   sandbox: {
     // Exposing some "safe" APIs
-    console,
+    console: consoleLogger,
     setTimeout,
     clearTimeout,
     setInterval,

packages/plugos/environments/sandbox_worker.ts

@@ -1,4 +1,5 @@
 import { safeRun } from "../util";
+import { ConsoleLogger } from "./custom_logger";
 import { ControllerMessage, WorkerMessage } from "./worker";
 
 let loadedFunctions = new Map<string, Function>();
@@ -53,6 +54,11 @@ self.require = (moduleName: string): any => {
   return preloadedModules[moduleName];
 };
 
+// @ts-ignore
+self.console = new ConsoleLogger((level, message) => {
+  workerPostMessage({ type: "log", level, message });
+}, false);
+
 function wrapScript(code: string) {
   return `return (${code})["default"]`;
 }

packages/plugos/environments/worker.ts

@@ -1,11 +1,14 @@
-export type ControllerMessageType = "inited" | "result" | "syscall";
+import type { LogLevel } from "./custom_logger";
 
+export type ControllerMessageType = "inited" | "result" | "syscall" | "log";
 export type ControllerMessage = {
   type: ControllerMessageType;
   id?: number;
   name?: string;
   args?: any[];
   error?: string;
+  level?: LogLevel;
+  message?: string;
   result?: any;
 };