resume mode

Author: dimitropoulos

packages/playground/evaluate/config.ts

@@ -1,5 +1,5 @@
 import { dirname, join } from "node:path";
-import { fileURLToPath } from "url";
+import { fileURLToPath } from "node:url";
 
 export interface Config {
   /** if `Infinity`, the fun never stops.  otherwise it will forcibly halt at a the specified instructions count */
@@ -45,6 +45,7 @@ export const stringResultTypeName = `String${resultTypeName}`;
 export const simpleTypeMode = process.argv.includes('--simple');
 export const statsOnlyMode = process.argv.includes("--stats-only");
 export const reportDiagnosticsMode = process.argv.includes("--report-ts-diagnostics");
+export const resumeMode = process.argv.includes("--resume");
 
 const __filename = fileURLToPath(import.meta.url);
 const __dirname = dirname(__filename);

packages/playground/evaluate/run.ts

@@ -1,5 +1,5 @@
-import { join } from "path";
-import { statSync, mkdirSync, readdirSync, unlinkSync } from "fs";
+import { join } from "node:path";
+import { statSync, mkdirSync, readdirSync, unlinkSync } from "node:fs";
 import {
   clearStats,
   encourage,
@@ -8,14 +8,14 @@ import {
 } from "./stats";
 import {
   config,
-  startFilePath,
   resultsDirectory,
   simpleTypeMode,
   statsOnlyMode,
   reportDiagnosticsMode,
+  resumeMode,
 } from "./config";
 import { Meter } from "./metering";
-import { getFuncImportLine, printType, programIsComplete, ProgramRun } from './utils';
+import { getStartFilePath as getStartFilePath, getFuncImportLine, printType, programIsComplete, ProgramRun } from './utils';
 import { createEnv, createNewFile, evaluateType, reportErrors } from "./ts";
 
 const isolatedProgram = async ({
@@ -128,6 +128,7 @@ const clearResults = () => {
 };
 
 const bootstrap = async () => {
+  const startFilePath = getStartFilePath();
   const startProgramTime = performance.now();
   const env = createEnv(startFilePath);
   const program = env.languageService.getProgram();
@@ -191,8 +192,10 @@ const mainLoop = async () => {
 
 const startProgram = async () => {
   encourage();
-  clearResults();
-  clearStats();
+  if (!resumeMode) {
+    clearResults();
+    clearStats();
+  }
   await mainLoop();
 };
 

packages/playground/evaluate/utils.ts

@@ -1,10 +1,11 @@
 
-import { readFileSync, writeFileSync } from "node:fs";
+import { readdirSync, readFileSync, writeFileSync } from "node:fs";
 import { Biome, Distribution } from "@biomejs/js-api";
-import { config, createResultFilePath, errorFilePath, finalResultPath, formatterErrorFilePath, shouldTakeABreath, startFilePath } from "./config";
+import { config, createResultFilePath, errorFilePath, finalResultPath, formatterErrorFilePath, resultsDirectory, resumeMode, shouldTakeABreath, startFilePath } from "./config";
 import { VirtualTypeScriptEnvironment } from "@typescript/vfs";
 import ts from "typescript";
 import { inspect } from "node:util";
+import { join } from "node:path";
 
 export interface ProgramRun {
   env: VirtualTypeScriptEnvironment;
@@ -172,7 +173,8 @@ export const finalizeProgram = async ({
   resultTypeName: string;
   env: VirtualTypeScriptEnvironment;
 }) => {
-  const relativeLastResult = createResultFilePath(lastInstructionCount)
+  const absoluteLastResult = createResultFilePath(lastInstructionCount)
+  const relativeLastResult = absoluteLastResult.replace(resultsDirectory, ".");
 
   const readStringImportLine = config.readStringFromMemory
     ? [
@@ -252,4 +254,27 @@ export const printType = (typeString: string) => {
   }
 
   return output;
-};
+};
+
+export const getStartFilePath = () => {
+  if (!resumeMode) {
+    return startFilePath;
+  }
+
+  const files = readdirSync(resultsDirectory);
+
+  const resumeIndex = process.argv.indexOf("--resume");
+  const maybeResumeFile = process.argv[resumeIndex + 1];
+  if (maybeResumeFile) {
+    if (!files.includes(maybeResumeFile)) {
+      throw new Error(`specified resume file '${maybeResumeFile}' not found in ${resultsDirectory}`);
+    }
+    return join(resultsDirectory, maybeResumeFile);
+  }
+
+  const filenames = files.filter(file => {
+    return file.startsWith("result-") && file.endsWith(".ts");
+  }).sort();
+
+  return join(resultsDirectory, filenames[filenames.length - 1]);
+}