Allow creating the non-existing file

Author: celonymire

src/extension/providers/JudgeViewProvider.ts

@@ -1,7 +1,6 @@
 import * as vscode from "vscode";
 import * as v from "valibot";
 import * as crypto from "crypto";
-import * as fs from "fs";
 
 import {
   TestcaseSchema,
@@ -23,6 +22,7 @@ import {
   getFileRunSettings,
   openInNewEditor,
   openInTerminalTab,
+  openOrCreateFile,
   ReadonlyStringProvider,
   resolveVariables,
   showOpenRunSettingsErrorWindow,
@@ -1442,12 +1442,10 @@ export default class extends BaseViewProvider<typeof ProviderMessageSchema, Webv
       return;
     }
 
-    if (settings.interactorFile && fs.existsSync(settings.interactorFile)) {
-      void vscode.window.showTextDocument(vscode.Uri.file(settings.interactorFile));
+    if (settings.interactorFile) {
+      void openOrCreateFile(settings.interactorFile);
     } else {
-      void vscode.window.showWarningMessage(
-        `${settings.interactorFile ?? "Interactor file"} does not exist`
-      );
+      void vscode.window.showWarningMessage("Interactor file not specified");
     }
   }
 

src/extension/providers/StressViewProvider.ts

@@ -1,7 +1,6 @@
 import * as vscode from "vscode";
 import * as v from "valibot";
 import * as crypto from "crypto";
-import * as fs from "fs";
 
 import { StatusSchema, type Status } from "../../shared/enums";
 import BaseViewProvider from "./BaseViewProvider";
@@ -17,6 +16,7 @@ import {
   getFileRunSettings,
   openInNewEditor,
   openInTerminalTab,
+  openOrCreateFile,
   showOpenRunSettingsErrorWindow,
   TextHandler,
   type FileRunSettings,
@@ -706,11 +706,7 @@ export default class extends BaseViewProvider<typeof ProviderMessageSchema, Webv
       });
     }
 
-    if (fs.existsSync(resolvedFile)) {
-      void vscode.window.showTextDocument(vscode.Uri.file(resolvedFile));
-    } else {
-      void vscode.window.showWarningMessage(`${resolvedFile} does not exist`);
-    }
+    void openOrCreateFile(resolvedFile);
   }
 
   private _open({ id }: v.InferOutput<typeof OpenMessageSchema>) {
@@ -738,11 +734,7 @@ export default class extends BaseViewProvider<typeof ProviderMessageSchema, Webv
       return;
     }
 
-    if (fs.existsSync(resolvedFile)) {
-      void vscode.window.showTextDocument(vscode.Uri.file(resolvedFile));
-    } else {
-      void vscode.window.showWarningMessage(`${resolvedFile} does not exist`);
-    }
+    void openOrCreateFile(resolvedFile);
   }
 
   clear() {

src/extension/utils/vscode.ts

@@ -234,6 +234,33 @@ export function openInTerminalTab(content: string, title: string = "Output") {
   terminal.show();
 }
 
+/**
+ * Opens a file if it exists, otherwise prompts the user to create it.
+ */
+export async function openOrCreateFile(file: string): Promise<void> {
+  if (fs.existsSync(file)) {
+    const document = await vscode.workspace.openTextDocument(file);
+    await vscode.window.showTextDocument(document);
+  } else {
+    const choice = await vscode.window.showWarningMessage(
+      `${file} does not exist`,
+      "Create File",
+      "Cancel"
+    );
+    if (choice === "Create File") {
+      try {
+        fs.mkdirSync(path.dirname(file), { recursive: true });
+        fs.writeFileSync(file, "");
+        const document = await vscode.workspace.openTextDocument(file);
+        await vscode.window.showTextDocument(document);
+      } catch (error) {
+        getLogger("vscode").error(`Failed to create file ${file}: ${error}`);
+        vscode.window.showErrorMessage(`Failed to create file ${file}: ${error}`);
+      }
+    }
+  }
+}
+
 function getRelativeFileDirname(relativeFilePath: string | undefined): string {
   if (!relativeFilePath) return "";
   const dir = path.dirname(relativeFilePath);