Code lens for running main

src/effektManager.ts

@@ -508,7 +508,7 @@ export class EffektManager {
      * Gets the command arguments for starting the Effekt server.
      * @returns An array of command arguments.
      */
-    public getEffektArgs(): string[] {
+    public getEffektArgs(server: boolean = true): string[] {
         const args: string[] = [];
         const effektBackend = this.config.get<string>("backend");
         const effektLib = this.config.get<string>("lib");
@@ -519,7 +519,9 @@ export class EffektManager {
         const folders = vscode.workspace.workspaceFolders || [];
         folders.forEach(folder => args.push("--includes", folder.uri.fsPath));
 
-        args.push("--server");
+        if (server) {
+            args.push("--server");
+        }
         return args;
     }
 }

src/extension.ts

@@ -6,9 +6,11 @@ import { EffektManager } from './effektManager';
 import { Monto } from './monto';
 
 import * as net from 'net';
+import * as path from 'path';
 
 let client: LanguageClient;
 let effektManager: EffektManager;
+let effektRepl: vscode.Terminal | null = null;
 
 function registerCommands(context: vscode.ExtensionContext) {
     context.subscriptions.push(
@@ -18,10 +20,70 @@ function registerCommands(context: vscode.ExtensionContext) {
         vscode.commands.registerCommand('effekt.restartServer', async () => {
             await client?.stop();
             client?.start();
-        })
+        }),
+        vscode.commands.registerCommand('effekt.runFile', runEffektFile),
     );
 }
 
+async function getEffektRepl() {
+    if (effektRepl === null || effektRepl.exitStatus !== undefined) {
+        const effektExecutable = await effektManager.locateEffektExecutable();
+        effektRepl = vscode.window.createTerminal({
+            name: 'Effekt REPL',
+            shellPath: effektExecutable.path,
+            shellArgs: effektManager.getEffektArgs(/* server = */ false),
+            isTransient: true, // Don't persist across VSCode restarts
+        });
+        effektRepl.show();
+    }
+    return effektRepl;
+}
+
+async function runEffektFile(uri: vscode.Uri) {
+    // Save the document if it has unsaved changes
+    const document = await vscode.workspace.openTextDocument(uri);
+    if (document.isDirty) {
+        await document.save();
+    }
+
+    const repl = await getEffektRepl();
+
+    const relativePath = vscode.workspace.asRelativePath(uri);
+    const parsedPath = path.parse(relativePath);
+
+    // Remove .effekt or .effekt.md extension
+    const moduleName = parsedPath.base.replace(/\.effekt(\.md)?$/, '');
+
+    const module = path.join(parsedPath.dir, moduleName).split(path.sep).join('/');
+
+    repl.sendText(':reset')
+    repl.sendText(`import ${module}`);
+    repl.sendText('main()');
+    repl.show();
+}
+
+class EffektRunCodeLensProvider implements vscode.CodeLensProvider {
+    public provideCodeLenses(document: vscode.TextDocument): vscode.CodeLens[] {
+        const codeLenses: vscode.CodeLens[] = [];
+        const text = document.getText();
+        const mainFunctionRegex = /^def\s+main\s*\(\s*\)/gm;
+        let match: RegExpExecArray | null;
+
+        while ((match = mainFunctionRegex.exec(text)) !== null) {
+            const line = document.lineAt(document.positionAt(match.index).line);
+            const range = new vscode.Range(line.range.start, line.range.end);
+
+            codeLenses.push(new vscode.CodeLens(range, {
+                title: '$(play) Run',
+                command: 'effekt.runFile',
+                arguments: [document.uri]
+            }));
+        }
+
+        return codeLenses;
+    }
+}
+
 export async function activate(context: vscode.ExtensionContext) {
     effektManager = new EffektManager();
 
@@ -32,6 +94,27 @@ export async function activate(context: vscode.ExtensionContext) {
 
     registerCommands(context);
 
+    // Register the CodeLens provider
+    context.subscriptions.push(
+        vscode.languages.registerCodeLensProvider(
+            { language: 'effekt', scheme: 'file' },
+            new EffektRunCodeLensProvider()
+        ),
+        vscode.languages.registerCodeLensProvider(
+            { language: 'literate effekt', scheme: 'file' },
+            new EffektRunCodeLensProvider()
+        )
+    );
+
+    // Clean up REPL when closed
+    context.subscriptions.push(
+        vscode.window.onDidCloseTerminal(terminal => {
+            if (terminal === effektRepl) {
+                effektRepl = null;
+            }
+        })
+    );
+
     const config = vscode.workspace.getConfiguration("effekt");
 
     let serverOptions: ServerOptions;