CopilotForXcode/Tool/Sources/GitHubCopilotService/Services/GitHubCopilotSuggestionService.swift at 0.46.158 · github/CopilotForXcode · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
import CopilotForXcodeKit
import Foundation
import SuggestionBasic
import Workspace
import SuggestionProvider

public final class GitHubCopilotSuggestionService: SuggestionServiceType, NESSuggestionServiceType {
    public var configuration: SuggestionServiceConfiguration {
        .init(
            acceptsRelevantCodeSnippets: true,
            mixRelevantCodeSnippetsInSource: true,
            acceptsRelevantSnippetsFromOpenedFiles: false
        )
    }

    let serviceLocator: ServiceLocatorType

    init(serviceLocator: ServiceLocatorType) {
        self.serviceLocator = serviceLocator
    }

    public func getSuggestions(
        _ request: CopilotForXcodeKit.SuggestionRequest,
        workspace: WorkspaceInfo
    ) async throws -> [CopilotForXcodeKit.CodeSuggestion] {
        guard let service = await serviceLocator.getService(from: workspace) else { return [] }
        return try await service.getCompletions(
            fileURL: request.fileURL,
            content: request.content,
            originalContent: request.originalContent,
            cursorPosition: .init(
                line: request.cursorPosition.line,
                character: request.cursorPosition.character
            ),
            tabSize: request.tabSize,
            indentSize: request.indentSize,
            usesTabsForIndentation: request.usesTabsForIndentation
        ).map(Self.convert)
    }

    public func getNESSuggestions(
        _ request: CopilotForXcodeKit.SuggestionRequest,
        workspace: WorkspaceInfo
    ) async throws -> [CopilotForXcodeKit.CodeSuggestion] {
        guard let service = await serviceLocator.getService(from: workspace) else { return [] }

        return try await service
            .getCopilotInlineEdit(
                fileURL: request.fileURL,
                content: request.content,
                cursorPosition: .init(line: request.cursorPosition.line, character: request.cursorPosition.character)
            )
            .map(Self.convert)
    }

    public func notifyAccepted(
        _ suggestion: CopilotForXcodeKit.CodeSuggestion,
        workspace: WorkspaceInfo
    ) async {
        guard let service = await serviceLocator.getService(from: workspace) else { return }
        await service.notifyAccepted(Self.convert(suggestion))
    }

    public func notifyRejected(
        _ suggestions: [CopilotForXcodeKit.CodeSuggestion],
        workspace: WorkspaceInfo
    ) async {
        guard let service = await serviceLocator.getService(from: workspace) else { return }
        await service.notifyRejected(suggestions.map(Self.convert))
    }

    public func cancelRequest(workspace: WorkspaceInfo) async {
        guard let service = await serviceLocator.getService(from: workspace) else { return }
        await service.cancelRequest()
    }

    static func convert(
        _ suggestion: SuggestionBasic.CodeSuggestion
    ) -> CopilotForXcodeKit.CodeSuggestion {
        .init(
            id: suggestion.id,
            text: suggestion.text,
            position: .init(
                line: suggestion.position.line,
                character: suggestion.position.character
            ),
            range: .init(
                start: .init(
                    line: suggestion.range.start.line,
                    character: suggestion.range.start.character
                ),
                end: .init(
                    line: suggestion.range.end.line,
                    character: suggestion.range.end.character
                )
            )
        )
    }

    static func convert(
        _ suggestion: CopilotForXcodeKit.CodeSuggestion
    ) -> SuggestionBasic.CodeSuggestion {
        .init(
            id: suggestion.id,
            text: suggestion.text,
            position: .init(
                line: suggestion.position.line,
                character: suggestion.position.character
            ),
            range: .init(
                start: .init(
                    line: suggestion.range.start.line,
                    character: suggestion.range.start.character
                ),
                end: .init(
                    line: suggestion.range.end.line,
                    character: suggestion.range.end.character
                )
            )
        )
    }
}