Fix deprecated ByteSourceRange

Author: kimdv

Sources/SourceKitLSP/Rename.swift

@@ -989,7 +989,7 @@ extension SwiftLanguageService {
     }
 
     if let startToken, let endToken {
-      return snapshot.range(
+      return snapshot.absolutePositionRange(
         of: startToken.positionAfterSkippingLeadingTrivia..<endToken.endPositionBeforeTrailingTrivia
       )
     }

Sources/SourceKitLSP/Swift/CodeActions/PackageManifestEdits.swift

@@ -138,7 +138,7 @@ fileprivate extension PackageEditResult {
     // The edits to perform on the manifest itself.
     let manifestTextEdits = manifestEdits.map { edit in
       TextEdit(
-        range: snapshot.range(of: edit.range),
+        range: snapshot.absolutePositionRange(of: edit.range),
         newText: edit.replacement
       )
     }

Sources/SourceKitLSP/Swift/CodeActions/SyntaxRefactoringCodeActionProvider.swift

@@ -36,7 +36,7 @@ extension SyntaxRefactoringCodeActionProvider where Self.Context == Void {
 
     let textEdits = sourceEdits.compactMap { (edit) -> TextEdit? in
       let edit = TextEdit(
-        range: scope.snapshot.range(of: edit.range),
+        range: scope.snapshot.absolutePositionRange(of: edit.range),
         newText: edit.replacement
       )
       if edit.isNoOp(in: scope.snapshot) {

Sources/SourceKitLSP/Swift/Diagnostic.swift

@@ -61,7 +61,7 @@ extension CodeAction {
   init?(_ fixIt: FixIt, in snapshot: DocumentSnapshot) {
     var textEdits = [TextEdit]()
     for edit in fixIt.edits {
-      textEdits.append(TextEdit(range: snapshot.range(of: edit.range), newText: edit.replacement))
+      textEdits.append(TextEdit(range: snapshot.absolutePositionRange(of: edit.range), newText: edit.replacement))
     }
 
     self.init(
@@ -281,7 +281,7 @@ extension Diagnostic {
     var range = Range(snapshot.position(of: diag.position))
     for highlight in diag.highlights {
       let swiftSyntaxRange = highlight.positionAfterSkippingLeadingTrivia..<highlight.endPositionBeforeTrailingTrivia
-      let highlightRange = snapshot.range(of: swiftSyntaxRange)
+      let highlightRange = snapshot.absolutePositionRange(of: swiftSyntaxRange)
       if range.upperBound == highlightRange.lowerBound {
         range = range.lowerBound..<highlightRange.upperBound
       } else {
@@ -345,7 +345,7 @@ extension DiagnosticRelatedInformation {
   init(_ note: Note, in snapshot: DocumentSnapshot) {
     let nodeRange = note.node.positionAfterSkippingLeadingTrivia..<note.node.endPositionBeforeTrailingTrivia
     self.init(
-      location: Location(uri: snapshot.uri, range: snapshot.range(of: nodeRange)),
+      location: Location(uri: snapshot.uri, range: snapshot.absolutePositionRange(of: nodeRange)),
       message: note.message
     )
   }

Sources/SourceKitLSP/Swift/DocumentSymbols.swift

@@ -74,8 +74,8 @@ fileprivate final class DocumentSymbolsFinder: SyntaxAnyVisitor {
     if !self.range.overlaps(range) {
       return .skipChildren
     }
-    let positionRange = snapshot.range(of: range)
-    let selectionPositionRange = snapshot.range(of: selection)
+    let positionRange = snapshot.absolutePositionRange(of: range)
+    let selectionPositionRange = snapshot.absolutePositionRange(of: selection)
 
     // Record MARK comments on the node's leading and trailing trivia in `result` not as a child of `node`.
     visit(node.leadingTrivia, position: node.position)
@@ -141,7 +141,9 @@ fileprivate final class DocumentSymbolsFinder: SyntaxAnyVisitor {
         let trimmedComment = commentText.trimmingCharacters(in: CharacterSet(["/", "*"]).union(.whitespaces))
         if trimmedComment.starts(with: markPrefix) {
           let markText = trimmedComment.dropFirst(markPrefix.count)
-          let range = snapshot.range(of: position..<position.advanced(by: piece.sourceLength.utf8Length))
+          let range = snapshot.absolutePositionRange(
+            of: position..<position.advanced(by: piece.sourceLength.utf8Length)
+          )
           result.append(
             DocumentSymbol(
               name: String(markText),

Sources/SourceKitLSP/Swift/SemanticTokens.swift

@@ -58,7 +58,7 @@ extension SwiftLanguageService {
       if let range = range.flatMap({ snapshot.byteSourceRange(of: $0) }) {
         range
       } else {
-        ByteSourceRange(offset: 0, length: await tree.totalLength.utf8Length)
+        await tree.range
       }
 
     let tokens =
@@ -107,7 +107,7 @@ extension SyntaxClassifiedRange {
       return SyntaxHighlightingTokens(tokens: [])
     }
 
-    let multiLineRange = snapshot.positionOf(utf8Offset: self.offset)..<snapshot.positionOf(utf8Offset: self.endOffset)
+    let multiLineRange = snapshot.absolutePositionRange(of: self.range)
     let ranges = multiLineRange.splitToSingleLineRanges(in: snapshot)
 
     let tokens = ranges.map {

Sources/SourceKitLSP/Swift/SwiftLanguageService.swift

@@ -690,7 +690,7 @@ extension SwiftLanguageService {
 
         result.append(
           ColorInformation(
-            range: snapshot.range(of: node.position..<node.endPosition),
+            range: snapshot.absolutePositionRange(of: node.position..<node.endPosition),
             color: Color(red: red, green: green, blue: blue, alpha: alpha)
           )
         )
@@ -1138,7 +1138,7 @@ extension DocumentSnapshot {
   /// If the bounds of the range do not refer to a valid positions with in the snapshot, this function adjusts them to
   /// the closest valid positions and logs a fault containing the file and line of the caller (from `callerFile` and
   /// `callerLine`).
-  func range(
+  func absolutePositionRange(
     of range: Range<AbsolutePosition>,
     callerFile: StaticString = #fileID,
     callerLine: UInt = #line
@@ -1169,9 +1169,12 @@ extension DocumentSnapshot {
     of range: Range<Position>,
     callerFile: StaticString = #fileID,
     callerLine: UInt = #line
-  ) -> ByteSourceRange {
+  ) -> Range<AbsolutePosition> {
     let utf8OffsetRange = utf8OffsetRange(of: range, callerFile: callerFile, callerLine: callerLine)
-    return ByteSourceRange(offset: utf8OffsetRange.startIndex, length: utf8OffsetRange.count)
+    return Range<AbsolutePosition>(
+      position: AbsolutePosition(utf8Offset: utf8OffsetRange.startIndex),
+      length: SourceLength(utf8Length: utf8OffsetRange.count)
+    )
   }
 
   // MARK: Position <-> RenameLocation

Sources/SourceKitLSP/Swift/SwiftTestingScanner.swift

@@ -248,7 +248,9 @@ final class SyntacticSwiftTestingTestScanner: SyntaxVisitor {
       return .skipChildren
     }
 
-    let range = snapshot.range(of: node.positionAfterSkippingLeadingTrivia..<node.endPositionBeforeTrailingTrivia)
+    let range = snapshot.absolutePositionRange(
+      of: node.positionAfterSkippingLeadingTrivia..<node.endPositionBeforeTrailingTrivia
+    )
     // Members won't be extensions since extensions will only be at the top level.
     let testItem = AnnotatedTestItem(
       testItem: TestItem(
@@ -306,7 +308,9 @@ final class SyntacticSwiftTestingTestScanner: SyntaxVisitor {
     let name =
       node.name.text + "(" + node.signature.parameterClause.parameters.map { "\($0.firstName.text):" }.joined() + ")"
 
-    let range = snapshot.range(of: node.positionAfterSkippingLeadingTrivia..<node.endPositionBeforeTrailingTrivia)
+    let range = snapshot.absolutePositionRange(
+      of: node.positionAfterSkippingLeadingTrivia..<node.endPositionBeforeTrailingTrivia
+    )
     let testItem = AnnotatedTestItem(
       testItem: TestItem(
         id: (parentTypeNames + [name]).joined(separator: "/"),

Sources/SourceKitLSP/TestDiscovery.swift

@@ -402,7 +402,7 @@ final class SyntacticSwiftXCTestScanner: SyntaxVisitor {
         // declarations are probably less common than helper functions that start with `test` and have a return type.
         return nil
       }
-      let range = snapshot.range(
+      let range = snapshot.absolutePositionRange(
         of: function.positionAfterSkippingLeadingTrivia..<function.endPositionBeforeTrailingTrivia
       )
 
@@ -436,7 +436,9 @@ final class SyntacticSwiftXCTestScanner: SyntaxVisitor {
       // Don't report a test class if it doesn't contain any test methods.
       return .visitChildren
     }
-    let range = snapshot.range(of: node.positionAfterSkippingLeadingTrivia..<node.endPositionBeforeTrailingTrivia)
+    let range = snapshot.absolutePositionRange(
+      of: node.positionAfterSkippingLeadingTrivia..<node.endPositionBeforeTrailingTrivia
+    )
     let testItem = AnnotatedTestItem(
       testItem: TestItem(
         id: node.name.text,