CopilotForXcode/Core/Sources/SuggestionService/SuggestionService.swift at e50927a675f7e33551d77355a2d05c6ef3836f11 · GoodOpenRepo/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
import Foundation
import GitHubCopilotService
import SuggestionModel

public protocol SuggestionServiceType {
    func getSuggestions(
        fileURL: URL,
        content: String,
        cursorPosition: CursorPosition,
        tabSize: Int,
        indentSize: Int,
        usesTabsForIndentation: Bool,
        ignoreSpaceOnlySuggestions: Bool,
        referenceFileURLs: [URL]
    ) async throws -> [CodeSuggestion]

    func notifyAccepted(_ suggestion: CodeSuggestion) async
    func notifyRejected(_ suggestions: [CodeSuggestion]) async
    func notifyOpenTextDocument(fileURL: URL, content: String) async throws
    func notifyChangeTextDocument(fileURL: URL, content: String) async throws
    func notifyCloseTextDocument(fileURL: URL) async throws
    func notifySaveTextDocument(fileURL: URL) async throws
}

public final class SuggestionService: SuggestionServiceType {
    let gitHubCopilotService: GitHubCopilotSuggestionServiceType

    public init(projectRootURL: URL) {
        gitHubCopilotService = GitHubCopilotSuggestionService(projectRootURL: projectRootURL)
    }

    public func getSuggestions(
        fileURL: URL,
        content: String,
        cursorPosition: SuggestionModel.CursorPosition,
        tabSize: Int,
        indentSize: Int,
        usesTabsForIndentation: Bool,
        ignoreSpaceOnlySuggestions: Bool,
        referenceFileURLs: [URL]
    ) async throws -> [SuggestionModel.CodeSuggestion] {
        try await gitHubCopilotService.getCompletions(
            fileURL: fileURL,
            content: content,
            cursorPosition: cursorPosition,
            tabSize: tabSize,
            indentSize: indentSize,
            usesTabsForIndentation: usesTabsForIndentation,
            ignoreSpaceOnlySuggestions: ignoreSpaceOnlySuggestions
        )
    }

    public func notifyAccepted(_ suggestion: SuggestionModel.CodeSuggestion) async {
        await gitHubCopilotService.notifyAccepted(suggestion)
    }

    public func notifyRejected(_ suggestions: [SuggestionModel.CodeSuggestion]) async {
        await gitHubCopilotService.notifyRejected(suggestions)
    }

    public func notifyOpenTextDocument(fileURL: URL, content: String) async throws {
        try await gitHubCopilotService.notifyOpenTextDocument(fileURL: fileURL, content: content)
    }

    public func notifyChangeTextDocument(fileURL: URL, content: String) async throws {
        try await gitHubCopilotService.notifyChangeTextDocument(fileURL: fileURL, content: content)
    }

    public func notifyCloseTextDocument(fileURL: URL) async throws {
        try await gitHubCopilotService.notifyCloseTextDocument(fileURL: fileURL)
    }

    public func notifySaveTextDocument(fileURL: URL) async throws {
        try await gitHubCopilotService.notifySaveTextDocument(fileURL: fileURL)
    }
}