From 87e242d87c8749c987b3889b89c8fbdd458f8171 Mon Sep 17 00:00:00 2001
From: Remy Willems <rwillems@amazon.com>
Date: Thu, 13 Jul 2023 14:50:24 +0200
Subject: [PATCH] Enable custom server arguments to contain spaces

---
 src/language/dafnyLanguageClient.ts | 42 +++++++++++++++++++++++++++--
 1 file changed, 40 insertions(+), 2 deletions(-)

diff --git a/src/language/dafnyLanguageClient.ts b/src/language/dafnyLanguageClient.ts
index ae6caa01..0b0e7395 100644
--- a/src/language/dafnyLanguageClient.ts
+++ b/src/language/dafnyLanguageClient.ts
@@ -24,7 +24,7 @@ function getLanguageServerLaunchArgsNew(): string[] {
   };
   const verifyOn: string = map[oldVerifyOnValue];
 
-  const launchArgs = Configuration.get<string[]>(ConfigurationConstants.LanguageServer.LaunchArgs);
+  const launchArgs = getCustomLaunchArgs();
   const specifiedCores = parseInt(Configuration.get<string>(ConfigurationConstants.LanguageServer.VerificationVirtualCores));
   // This is a temporary fix to prevent 0 cores from being used, since the languages server currently does not handle 0 cores correctly: https://github.com/dafny-lang/dafny/pull/3276
   const cores = isNaN(specifiedCores) || specifiedCores === 0 ? Math.ceil((os.cpus().length + 1) / 2) : Math.max(1, specifiedCores);
@@ -41,7 +41,7 @@ function getLanguageServerLaunchArgsNew(): string[] {
 }
 
 function getLanguageServerLaunchArgsOld(): string[] {
-  const launchArgs = Configuration.get<string[]>(ConfigurationConstants.LanguageServer.LaunchArgs);
+  const launchArgs = getCustomLaunchArgs();
   return [
     getVerificationArgument(),
     getVerifierTimeLimitArgument(),
@@ -54,6 +54,44 @@ function getLanguageServerLaunchArgsOld(): string[] {
   ];
 }
 
+function getCustomLaunchArgs(): string[] {
+  const launchArgs = Configuration.get<string[]>(ConfigurationConstants.LanguageServer.LaunchArgs);
+  return launchArgs.flatMap(a => splitArguments(a));
+}
+
+function splitArguments(args: string): string[] {
+  if(args) {
+    return [];
+  }
+
+  let inSingleQuote = false;
+  let inDoubleQuote = false;
+  const result: string[] = [];
+  let start = 0;
+  for(let end = 0; end < args.length; end++) {
+    let store = false;
+    if(args[end] === '"' && !inSingleQuote) {
+      store = inDoubleQuote;
+      inDoubleQuote = !inDoubleQuote;
+    }
+    if(args[end] === '\'' && !inDoubleQuote) {
+      store = inSingleQuote;
+      inSingleQuote = !inSingleQuote;
+    }
+    if(!inSingleQuote && !inDoubleQuote && args[end] === ' ') {
+      store = true;
+    }
+
+    if(store) {
+      result.push(args.substring(start, end - start));
+      start = end + 1; // Skip the single or double quote or space in the next entry
+    }
+  }
+  result.push(args.substring(start, args.length - start));
+  return result;
+}
+
+
 function getVerificationArgument(): string {
   return `--documents:verify=${Configuration.get<string>(ConfigurationConstants.LanguageServer.AutomaticVerification)}`;
 }