mirror of
https://github.com/apple/sourcekit-lsp.git
synced 2026-03-02 18:23:24 +01:00
cf56bf2ce6
- Add DefinitionLocations.swift to CMakeLists.txt - Add proper documentation comments to functions in DefinitionLocations.swift - Make indexToLSPLocation private since it's only used within the file - Revert unrelated comment changes in InlayHintResolve.swift - Use .only instead of .first in InlayHintResolve to avoid ambiguous types - Refactor TypeDefinition.swift to use cleaner control flow with guard/else - Add test for jumping to generated interface (String)
91 lines
3.1 KiB
Swift
91 lines
3.1 KiB
Swift
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This source file is part of the Swift.org open source project
|
|
//
|
|
// Copyright (c) 2014 - 2026 Apple Inc. and the Swift project authors
|
|
// Licensed under Apache License v2.0 with Runtime Library Exception
|
|
//
|
|
// See https://swift.org/LICENSE.txt for license information
|
|
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
import IndexStoreDB
|
|
@_spi(SourceKitLSP) package import LanguageServerProtocol
|
|
import SemanticIndex
|
|
import SourceKitD
|
|
import SourceKitLSP
|
|
|
|
extension SwiftLanguageService {
|
|
/// Handles the textDocument/typeDefinition request.
|
|
package func typeDefinition(_ request: TypeDefinitionRequest) async throws -> LocationsOrLocationLinksResponse? {
|
|
let uri = request.textDocument.uri
|
|
let position = request.position
|
|
|
|
let snapshot = try await self.latestSnapshot(for: uri)
|
|
let compileCommand = await self.compileCommand(for: uri, fallbackAfterTimeout: false)
|
|
|
|
let skreq = sourcekitd.dictionary([
|
|
keys.cancelOnSubsequentRequest: 0,
|
|
keys.offset: snapshot.utf8Offset(of: position),
|
|
keys.sourceFile: snapshot.uri.sourcekitdSourceFile,
|
|
keys.primaryFile: snapshot.uri.primaryFile?.pseudoPath,
|
|
keys.compilerArgs: compileCommand?.compilerArgs as [any SKDRequestValue]?,
|
|
])
|
|
|
|
let dict = try await send(sourcekitdRequest: \.cursorInfo, skreq, snapshot: snapshot)
|
|
let documentManager = try self.documentManager
|
|
|
|
// Helper to get symbol from typeUsr
|
|
func symbolFromTypeUsr() async throws -> SymbolDetails? {
|
|
guard let typeUsr: String = dict[keys.typeUsr] else {
|
|
return nil
|
|
}
|
|
guard let typeInfo = try await cursorInfoFromTypeUSR(typeUsr, in: snapshot) else {
|
|
return nil
|
|
}
|
|
return typeInfo.symbolInfo
|
|
}
|
|
|
|
// If the cursor is on a type symbol itself, use its USR directly.
|
|
// Otherwise get the type of the symbol at this position.
|
|
let symbol: SymbolDetails
|
|
if let cursorInfo = CursorInfo(dict, snapshot: snapshot, documentManager: documentManager, sourcekitd: sourcekitd) {
|
|
switch cursorInfo.symbolInfo.kind {
|
|
case .class, .struct, .enum, .interface, .typeParameter:
|
|
symbol = cursorInfo.symbolInfo
|
|
default:
|
|
guard let typeSymbol = try await symbolFromTypeUsr() else {
|
|
return nil
|
|
}
|
|
symbol = typeSymbol
|
|
}
|
|
} else {
|
|
guard let typeSymbol = try await symbolFromTypeUsr() else {
|
|
return nil
|
|
}
|
|
symbol = typeSymbol
|
|
}
|
|
|
|
let locations = try await SourceKitLSP.definitionLocations(
|
|
for: symbol,
|
|
originatorUri: uri,
|
|
index: nil,
|
|
openGeneratedInterface: { document, moduleName, groupName, symbolUSR in
|
|
try await self.openGeneratedInterface(
|
|
document: document,
|
|
moduleName: moduleName,
|
|
groupName: groupName,
|
|
symbolUSR: symbolUSR
|
|
)
|
|
}
|
|
)
|
|
|
|
if locations.isEmpty {
|
|
return nil
|
|
}
|
|
|
|
return .locations(locations)
|
|
}
|
|
}
|