Language server improvements for auto-diff. (#2521)

1 files changed, 9 insertions, 4 deletions

diff --git a/source/slang/slang-workspace-version.cpp b/source/slang/slang-workspace-version.cpp
index 914393320..d5fc62e79 100644
--- a/source/slang/slang-workspace-version.cpp
+++ b/source/slang/slang-workspace-version.cpp
@@ -458,20 +458,25 @@ UnownedStringSlice DocumentVersion::peekIdentifier(Index& offset)
     return UnownedStringSlice("");
 }
 
-
-int DocumentVersion::getTokenLength(Index line, Index col)
+int DocumentVersion::getTokenLength(Index offset)
 {
-    auto offset = getOffset(line, col);
     if (offset >= 0)
     {
         Index pos = offset;
         for (; pos < text.getLength() && _isIdentifierChar(text[pos]); ++pos)
-        {}
+        {
+        }
         return (int)(pos - offset);
     }
     return 0;
 }
 
+int DocumentVersion::getTokenLength(Index line, Index col)
+{
+    auto offset = getOffset(line, col);
+    return getTokenLength(offset);
+}
+
 ASTMarkup* WorkspaceVersion::getOrCreateMarkupAST(ModuleDecl* module)
 {
     RefPtr<ASTMarkup> astMarkup;