cleanup - ensure project cleanup is called on abnormal termination

Author: cscheid

src/core/main.ts

@@ -27,8 +27,12 @@ export async function mainRunner(runner: Runner) {
     await initializeLogger(logOptions(args));
 
     // install termination signal handlers
+
+    // Even though windows doesn't technically have signals, Deno
+    // does the "expected" thing here and calls abend
+    // on interruption.
+    Deno.addSignalListener("SIGINT", abend);
     if (!isWindows) {
-      Deno.addSignalListener("SIGINT", abend);
       Deno.addSignalListener("SIGTERM", abend);
     }
 

src/core/once.ts

@@ -0,0 +1,15 @@
+/*
+ * once.ts
+ *
+ * Copyright (C) 2025 Posit Software, PBC
+ */
+
+export const once = (fn: () => void) => {
+  let called = false;
+  return () => {
+    if (!called) {
+      called = true;
+      fn();
+    }
+  };
+};

src/project/project-context.ts

@@ -104,6 +104,9 @@ import { makeTimedFunctionAsync } from "../core/performance/function-times.ts";
 import { createProjectCache } from "../core/cache/cache.ts";
 import { createTempContext } from "../core/temp.ts";
 
+import { onCleanup } from "../core/cleanup.ts";
+import { once } from "../core/once.ts";
+
 export async function projectContext(
   path: string,
   notebookContext: NotebookContext,
@@ -314,11 +317,11 @@ export async function projectContext(
           isSingleFile: false,
           diskCache: await createProjectCache(join(dir, ".quarto")),
           temp,
-          cleanup: () => {
+          cleanup: once(() => {
             cleanupFileInformationCache(result);
             result.diskCache.close();
             temp.cleanup();
-          },
+          }),
         };
 
         // see if the project [kProjectType] wants to filter the project config
@@ -357,7 +360,7 @@ export async function projectContext(
           config: configFiles,
           configResources: projectConfigResources(dir, projectConfig, type),
         };
-
+        onCleanup(result.cleanup);
         return result;
       } else {
         debug(`projectContext: Found Quarto project in ${dir}`);
@@ -408,11 +411,11 @@ export async function projectContext(
           isSingleFile: false,
           diskCache: await createProjectCache(join(dir, ".quarto")),
           temp,
-          cleanup: () => {
+          cleanup: once(() => {
             cleanupFileInformationCache(result);
             result.diskCache.close();
             temp.cleanup();
-          },
+          }),
         };
         const { files, engines } = await projectInputFiles(
           result,
@@ -425,6 +428,7 @@ export async function projectContext(
           config: configFiles,
           configResources: projectConfigResources(dir, projectConfig),
         };
+        onCleanup(result.cleanup);
         return result;
       }
     } else {
@@ -486,11 +490,11 @@ export async function projectContext(
             isSingleFile: false,
             diskCache: await createProjectCache(join(temp.baseDir, ".quarto")),
             temp,
-            cleanup: () => {
+            cleanup: once(() => {
               cleanupFileInformationCache(context);
               context.diskCache.close();
               temp.cleanup();
-            },
+            }),
           };
           if (Deno.statSync(path).isDirectory) {
             const { files, engines } = await projectInputFiles(context);
@@ -503,6 +507,7 @@ export async function projectContext(
             context.files.input = [input];
           }
           debug(`projectContext: Found Quarto project in ${originalDir}`);
+          onCleanup(context.cleanup);
           return context;
         } else {
           return undefined;

src/project/types/single-file/single-file.ts

@@ -20,13 +20,15 @@ import { RenderFlags } from "../../../command/render/types.ts";
 import { MappedString } from "../../../core/mapped-text.ts";
 import { fileExecutionEngineAndTarget } from "../../../execute/engine.ts";
 import {
+  cleanupFileInformationCache,
   projectFileMetadata,
   projectResolveBrand,
   projectResolveFullMarkdownForFile,
 } from "../../project-shared.ts";
 import { ExecutionEngine } from "../../../execute/types.ts";
 import { createProjectCache } from "../../../core/cache/cache.ts";
 import { globalTempContext } from "../../../core/temp.ts";
+import { once } from "../../../core/once.ts";
 
 export async function singleFileProjectContext(
   source: string,
@@ -77,10 +79,15 @@ export async function singleFileProjectContext(
     isSingleFile: true,
     diskCache: await createProjectCache(projectCacheBaseDir),
     temp,
-    cleanup: () => {
+    cleanup: once(() => {
+      cleanupFileInformationCache(result);
       result.diskCache.close();
-    },
+    }),
   };
+  // because the single-file project is cleaned up with
+  // the global text context, we don't need to register it
+  // in the same way that we need to register the multi-file
+  // projects.
   temp.onCleanup(result.cleanup);
   return result;
 }