From 8a67a6e7c3b190873842e821d294bb32e79a239f Mon Sep 17 00:00:00 2001
From: Jake Bailey <5341706+jakebailey@users.noreply.github.com>
Date: Wed, 6 Nov 2024 11:03:30 -0800
Subject: [PATCH] Use findArgument in createCancellationToken

---
 src/tsserver/nodeServer.ts | 12 +++---------
 1 file changed, 3 insertions(+), 9 deletions(-)

diff --git a/src/tsserver/nodeServer.ts b/src/tsserver/nodeServer.ts
index 69ea459231c2b..dc71afe816dae 100644
--- a/src/tsserver/nodeServer.ts
+++ b/src/tsserver/nodeServer.ts
@@ -275,7 +275,7 @@ export function initializeNodeSystem(): StartInput {
         sys.gc = () => global.gc?.();
     }
 
-    const cancellationToken = createCancellationToken(sys.args);
+    const cancellationToken = createCancellationToken();
 
     const localeStr = ts.server.findArgument("--locale");
     if (localeStr) {
@@ -673,14 +673,8 @@ function pipeExists(name: string): boolean {
     return fs.existsSync(name);
 }
 
-function createCancellationToken(args: string[]): ts.server.ServerCancellationToken {
-    let cancellationPipeName: string | undefined;
-    for (let i = 0; i < args.length - 1; i++) {
-        if (args[i] === "--cancellationPipeName") {
-            cancellationPipeName = args[i + 1];
-            break;
-        }
-    }
+function createCancellationToken(): ts.server.ServerCancellationToken {
+    const cancellationPipeName = ts.server.findArgument("--cancellationPipeName");
     if (!cancellationPipeName) {
         return ts.server.nullCancellationToken;
     }