Repl command suffix

README.md

@@ -85,6 +85,10 @@ Since around GHC 8, the compiler GHC and its interactive REPL GHCi has gained va
 
     Change the options in workspace settings instead of user settings if you want to apply the settings to a workspace locally.
 
+- `ghcSimple.commandSuffix`: REPL command suffix
+
+    Arbitrary string appended to the REPL command line (e.g. `ghci`, `cabal new-repl`, `stack ghci`).
+
 - `ghcSimple.useObjectCode`: Speed up GHCi reloads using `-fobject-code`
 
     Enabled by default. Load everything with `-fobject-code` first before loading needed files with `-fbyte-code`. This way only changed files need to be recompiled, which greatly speeds up GHCi on large projects.

package.json

@@ -100,8 +100,8 @@
                 "ghcSimple.workspaceType": {
                     "type": "string",
                     "default": "detect",
-                    "description": "Workspace type, 'bare' means use GHCi directly",
                     "scope": "resource",
+                    "description": "Workspace type, 'bare' means use GHCi directly",
                     "enum": [
                         "detect",
                         "stack",
@@ -112,6 +112,12 @@
                         "bare"
                     ]
                 },
+                "ghcSimple.commandSuffix": {
+                    "type": "string",
+                    "default": "",
+                    "scope": "resource",
+                    "description": "Arbitrary string appended to REPL command"
+                },
                 "ghcSimple.startupCommands.all": {
                     "type": "array",
                     "items": {

src/extension-state.ts

@@ -18,6 +18,11 @@ function getWorkspaceType(ext: ExtensionState, folder: vscode.WorkspaceFolder):
     return ext.workspaceTypeMap.get(folder);
 }
 
+function splitArgs(argStr: string): string[] {
+    if (argStr === "") return [];
+    else return argStr.split(" ");
+}
+
 export async function startSession(ext: ExtensionState, doc: vscode.TextDocument): Promise<Session> {
     const folder = vscode.workspace.getWorkspaceFolder(doc.uri);
     const type = folder === undefined
@@ -27,16 +32,22 @@ export async function startSession(ext: ExtensionState, doc: vscode.TextDocument
     const session = (() => {
         if (-1 !== ['stack', 'cabal', 'cabal new', 'cabal v2'].indexOf(type)) {
             // stack or cabal
+            const cmdSuffix =
+                vscode.workspace.getConfiguration('ghcSimple', folder.uri).commandSuffix;
 
             if (! ext.workspaceManagers.has(folder))
-                ext.workspaceManagers.set(folder, new Session(ext, type, 'workspace', folder.uri));
+                ext.workspaceManagers.set(folder,
+                    new Session(ext, type, splitArgs(cmdSuffix), 'workspace', folder.uri));
 
             return ext.workspaceManagers.get(folder);
         } else {
             // bare or bare-stack
+            const cmdSuffix =
+                vscode.workspace.getConfiguration('ghcSimple', doc.uri).commandSuffix;
 
             if (! ext.documentManagers.has(doc))
-                ext.documentManagers.set(doc, new Session(ext, type, 'file', doc.uri));
+                ext.documentManagers.set(doc,
+                    new Session(ext, type, splitArgs(cmdSuffix), 'file', doc.uri));
 
             return ext.documentManagers.get(doc);
         }

src/session.ts

@@ -15,6 +15,7 @@ export class Session implements vscode.Disposable {
     constructor(
         public ext: ExtensionState,
         public workspaceType: HaskellWorkspaceType,
+        public cmdSuffix: string[],
         public resourceType: 'workspace' | 'file',
         public resource: vscode.Uri) {
         this.ghci = null;
@@ -54,6 +55,10 @@ export class Session implements vscode.Disposable {
                     return ['ghci'];
             })();
 
+            for (const arg of this.cmdSuffix) {
+                cmd.push(arg);
+            }
+
             this.ext.outputChannel.appendLine(`Starting GHCi with: ${JSON.stringify(cmd)}`);
             this.ext.outputChannel.appendLine(
                 `(Under ${