From eaccffba6cbdcad496b8beebed94e7e2e5d87fec Mon Sep 17 00:00:00 2001
From: Jake Bailey <5341706+jakebailey@users.noreply.github.com>
Date: Thu, 9 Apr 2026 08:37:49 -0700
Subject: [PATCH] Fix inlay hint range

---
 src/utils/exerciseLspServer.ts | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/src/utils/exerciseLspServer.ts b/src/utils/exerciseLspServer.ts
index 7cc59ed..3466e28 100644
--- a/src/utils/exerciseLspServer.ts
+++ b/src/utils/exerciseLspServer.ts
@@ -315,7 +315,8 @@ async function exerciseLspServerWorker(testDir: string, lspServerPath: string, r
             let line = 0; // LSP uses 0-based lines
             let character = 0; // LSP uses 0-based characters
             let characterDelta = 0; // Net character offset from insertions/deletions on the current line
-            const totalLines = openFileContents.split(/\r\n|\r|\n/).length;
+            const lines = openFileContents.split(/\r\n|\r|\n/);
+            const totalLines = lines.length;
 
             let prev = "";
 
@@ -367,9 +368,10 @@ async function exerciseLspServerWorker(testDir: string, lspServerPath: string, r
 
             const inlayHintsPromise = request("textDocument/inlayHint", {
                 textDocument: { uri: openFileUri },
-                // TODO - this is still a hack.
-                // Could just move it to the end after we've iterated through the file.
-                range: { start: { line: 0, character: 0 }, end: { line: 0, character: openFileContents.length } },
+                range: {
+                    start: { line: 0, character: 0 },
+                    end: { line: totalLines - 1, character: lines[totalLines - 1].length },
+                },
             });
 
             const [diagResult, codeLenses, _inlayHints] = await Promise.all([diagnosticsPromise, codeLensesPromise, inlayHintsPromise]);
@@ -808,4 +810,4 @@ async function exerciseLspServerWorker(testDir: string, lspServerPath: string, r
             process.exit(EXIT_SERVER_ERROR);
         }
     }
-}
\ No newline at end of file
+}