[ { "path": ".github/FUNDING.yml", "content": "custom: [\"https://intii.lemonsqueezy.com\", \"https://www.buymeacoffee.com/intitni\"]\n" }, { "path": ".github/ISSUE_TEMPLATE/bug_report.yaml", "content": "name: Bug Report\ndescription: File a bug report\ntitle: \"[Bug]: \"\nlabels: [\"bug\"]\nassignees:\n - intitni\nbody:\n - type: markdown\n attributes:\n value: |\n Thanks for taking the time to fill out this bug report! \n - type: checkboxes\n id: before-reporting\n attributes:\n label: Before Reporting\n description: Before reporting the bug, we suggestion that you first refer to the [FAQ](https://github.com/intitni/CopilotForXcode/wiki/Frequently-Asked-Questions) to check if it may address your issue. And search for existing issues to avoid duplication. If you are reporting a bug from a beta build, please use the dedicated template for beta build.\n options:\n - label: I have checked FAQ, and there is no solution to my issue\n required: true\n - label: I have searched the existing issues, and there is no existing issue for my issue\n required: true\n - type: textarea\n id: what-happened\n attributes:\n label: What happened?\n description: Also tell us, what did you expect to happen?\n placeholder: Tell us what you see!\n value: \"A bug happened!\"\n validations:\n required: true\n - type: textarea\n id: reproduce\n attributes:\n label: How to reproduce the bug.\n description: If possible, please provide the steps to reproduce the bug and relevant settings in screenshots.\n placeholder: \"1. *****\\n2.*****\"\n value: \"It just happened!\"\n - type: textarea\n id: logs\n attributes:\n label: Relevant log output\n description: If it's a crash, please provide the crash report. You can find it in the Console.app.\n render: shell\n - type: input\n id: mac-version\n attributes:\n label: macOS version\n - type: input\n id: xcode-version\n attributes:\n label: Xcode version\n - type: input\n id: copilot-for-xcode-version\n attributes:\n label: Copilot for Xcode version\n \n" }, { "path": ".github/ISSUE_TEMPLATE/feature_reqeust.yaml", "content": "name: Feature Request\ndescription: Request a feature\ntitle: \"[Enhancement]: \"\nlabels: [\"enhancement\"]\nassignees:\n - intitni\nbody: \n - type: markdown\n attributes:\n value: |\n Thanks for taking the time to fill out this feature request! But please firstly [post your idea in discussion](https://github.com/intitni/CopilotForXcode/discussions/new?category=ideas) so that we can discuss about it in advance.\n - type: checkboxes\n id: before-reporting\n attributes:\n label: Before Requesting\n description: Before requesting the feature, we suggestion that you first search for existing issues to avoid duplication.\n options:\n - label: I have searched the existing issues, and there is no existing issue for my feature request\n required: true\n - type: textarea\n id: what-feature\n attributes:\n label: What feature do you want?\n description: Please describe the feature you want.\n placeholder: Tell us what you want!\n value: \"I want a feature!\"\n validations:\n required: true\n" }, { "path": ".github/ISSUE_TEMPLATE/help_wanted.yml", "content": "name: Help Wanted\ndescription: Ask for help from the developer and the community\ntitle: \"[Help Wanted]: \"\nlabels: [\"help wanted\"]\nbody:\n - type: checkboxes\n id: before-reporting\n attributes:\n label: Before Reporting\n description: Before asking for help, we suggestion that you first refer to the [FAQ](https://github.com/intitni/CopilotForXcode/wiki/Frequently-Asked-Questions) to check if it may address your issue. And search for existing issues to avoid duplication.\n options:\n - label: I have checked FAQ, and there is no solution to my issue\n required: true\n - label: I have searched the existing issues, and there is no existing issue for my issue\n required: true\n - type: textarea\n id: what-help\n attributes:\n label: Describe your issue\n description: Please describe the help you want.\n placeholder: My issue is...\n value: \"I want help!\"\n validations:\n required: true\n" }, { "path": ".github/ISSUE_TEMPLATE/z_bug_report_beta.yaml", "content": "name: Bug Report (Beta)\ndescription: File a bug report\ntitle: \"[Bug (Beta)]: \"\nlabels: [\"bug\", \"beta\"]\nassignees:\n - intitni\nbody:\n - type: markdown\n attributes:\n value: |\n Thanks for taking the time to fill out this bug report! \n - type: checkboxes\n id: before-reporting\n attributes:\n label: Before Reporting\n description: Before reporting the bug, we suggestion that you first refer to the [FAQ](https://github.com/intitni/CopilotForXcode/wiki/Frequently-Asked-Questions) to check if it may address your issue. And search for existing issues to avoid duplication.\n options:\n - label: I have checked FAQ, and there is no solution to my issue\n required: true\n - label: I have searched the existing issues, and there is no existing issue for my issue\n required: true\n - type: textarea\n id: what-happened\n attributes:\n label: What happened?\n description: Also tell us, what did you expect to happen?\n placeholder: Tell us what you see!\n value: \"A bug happened!\"\n validations:\n required: true\n - type: textarea\n id: reproduce\n attributes:\n label: How to reproduce the bug.\n description: If possible, please provide the steps to reproduce the bug and relevant settings in screenshots.\n placeholder: \"1. *****\\n2.*****\"\n value: \"It just happened!\"\n - type: textarea\n id: logs\n attributes:\n label: Relevant log output\n description: If it's a crash, please provide the crash report. You can find it in the Console.app.\n render: shell\n - type: input\n id: mac-version\n attributes:\n label: macOS version\n - type: input\n id: xcode-version\n attributes:\n label: Xcode version\n - type: input\n id: copilot-for-xcode-version\n attributes:\n label: Copilot for Xcode version\n \n" }, { "path": ".github/workflows/close_inactive_issues.yml", "content": "name: Close inactive issues\non:\n schedule:\n - cron: \"30 1 * * *\"\n\njobs:\n close-issues:\n runs-on: ubuntu-latest\n permissions:\n issues: write\n pull-requests: write\n steps:\n - uses: actions/stale@v5\n with:\n days-before-issue-stale: 30\n days-before-issue-close: 14\n stale-issue-label: \"stale\"\n exempt-issue-labels: \"low priority, help wanted, planned, investigating, blocked\"\n stale-issue-message: \"This issue is stale because it has been open for 30 days with no activity.\"\n close-issue-message: \"This issue was closed because it has been inactive for 14 days since being marked as stale.\"\n days-before-pr-stale: -1\n days-before-pr-close: -1\n repo-token: ${{ secrets.GITHUB_TOKEN }}\n" }, { "path": ".gitignore", "content": "# IDE\n.idea\n\n# Created by \nhttps://www.toptal.com/developers/gitignore/api/xcode,macos,swift,swiftpackagemanager\n# Edit at \nhttps://www.toptal.com/developers/gitignore?templates=xcode,macos,swift,swiftpackagemanager\n\n### macOS ###\n# General\n.DS_Store\n.AppleDouble\n.LSOverride\n\n# Files that might appear in the root of a volume\n.DocumentRevisions-V100\n.fseventsd\n.Spotlight-V100\n.TemporaryItems\n.Trashes\n.VolumeIcon.icns\n.com.apple.timemachine.donotpresent\n\n# Directories potentially created on remote AFP share\n.AppleDB\n.AppleDesktop\nNetwork Trash Folder\nTemporary Items\n.apdisk\n\n### macOS Patch ###\n# iCloud generated files\n*.icloud\n\n### Swift ###\n# Xcode\n#\n# gitignore contributors: remember to update Global/Xcode.gitignore, \nObjective-C.gitignore & Swift.gitignore\n\n## User settings\nxcuserdata/\n\n## compatibility with Xcode 8 and earlier (ignoring not required starting \nXcode 9)\n*.xcscmblueprint\n*.xccheckout\n\n## compatibility with Xcode 3 and earlier (ignoring not required starting \nXcode 4)\nbuild/\nDerivedData/\n*.moved-aside\n*.pbxuser\n!default.pbxuser\n*.mode1v3\n!default.mode1v3\n*.mode2v3\n!default.mode2v3\n*.perspectivev3\n!default.perspectivev3\n\n## Obj-C/Swift specific\n*.hmap\n\n## App packaging\n*.ipa\n*.dSYM.zip\n*.dSYM\n\n# Swift Package Manager\n# Add this line if you want to avoid checking in source code from Swift \nPackage Manager dependencies.\n# Packages/\n# Package.pins\n# Package.resolved\n# *.xcodeproj\n# Xcode automatically generates this directory with a .xcworkspacedata \nfile and xcuserdata\n# hence it is not needed unless you have added a package configuration \nfile to your project\n# .swiftpm\n\n.build/\n\n# CocoaPods\n# We recommend against adding the Pods directory to your .gitignore. \nHowever\n# you should judge for yourself, the pros and cons are mentioned at:\n# \nhttps://guides.cocoapods.org/using/using-cocoapods.html#should-i-check-the-pods-directory-into-source-control\n# Pods/\n# Add this line if you want to avoid checking in source code from the \n\n# Carthage\n# Add this line if you want to avoid checking in source code from Carthage \ndependencies.\n# Carthage/Checkouts\n\nCarthage/Build/\n\n# Accio dependency management\nDependencies/\n.accio/\n\n# fastlane\n# It is recommended to not store the screenshots in the git repo.\n# Instead, use fastlane to re-generate the screenshots whenever they are \nneeded.\n# For more information about the recommended setup visit:\n# \nhttps://docs.fastlane.tools/best-practices/source-control/#source-control\n\nfastlane/report.xml\nfastlane/Preview.html\nfastlane/screenshots/**/*.png\nfastlane/test_output\n\n# Code Injection\n# After new code Injection tools there's a generated folder \n/iOSInjectionProject\n# https://github.com/johnno1962/injectionforxcode\n\niOSInjectionProject/\n\n\n# End of \nhttps://www.toptal.com/developers/gitignore/api/xcode,macos,swift,swiftpackagemanager\n\nSecrets.xcconfig\nPython/Python.xcframework\nPython/python-stdlib\nPython/site-packages/*\n!Python/site-packages/requirements.txt\n!Python/site-packages/install.sh\n\nPython/VERSIONS\nCopilot for Xcode Plus.xcworkspace\nPLUS\n" }, { "path": ".gitmodules", "content": "" }, { "path": ".swiftformat", "content": "--allman false\n--beforemarks \n--binarygrouping 4,8\n--categorymark \"MARK: %c\"\n--classthreshold 0\n--closingparen balanced\n--commas always\n--conflictmarkers reject\n--decimalgrouping 3,6\n--elseposition same-line\n--enumthreshold 0\n--exponentcase lowercase\n--exponentgrouping disabled\n--fractiongrouping disabled\n--fragment false\n--funcattributes preserve\n--guardelse auto\n--header ignore\n--hexgrouping 4,8\n--hexliteralcase uppercase\n--ifdef no-indent\n--importgrouping testable-bottom\n--indent 4\n--indentcase false\n--lifecycle \n--linebreaks lf\n--maxwidth 100\n--modifierorder \n--nospaceoperators ...,..<\n--nowrapoperators \n--octalgrouping 4,8\n--operatorfunc spaced\n--patternlet hoist\n--ranges spaced\n--self remove\n--selfrequired \n--semicolons inline\n--shortoptionals always\n--smarttabs enabled\n--stripunusedargs unnamed-only\n--structthreshold 0\n--tabwidth unspecified\n--trailingclosures \n--trimwhitespace always\n--typeattributes preserve\n--varattributes preserve\n--voidtype void\n--wraparguments before-first\n--wrapcollections disabled\n--wrapparameters before-first\n--xcodeindentation disabled\n--yodaswap always\n\n--enable isEmpty\n\n--exclude Pods,**/Generated\n" }, { "path": "ChatPlugins/.gitignore", "content": ".DS_Store\n/.build\n/Packages\nxcuserdata/\nDerivedData/\n.swiftpm/configuration/registries.json\n.swiftpm/xcode/package.xcworkspace/contents.xcworkspacedata\n.netrc\n" }, { "path": "ChatPlugins/.swiftpm/xcode/xcshareddata/xcschemes/ChatPlugins.xcscheme", "content": "\n\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n\n" }, { "path": "ChatPlugins/Package.swift", "content": "// swift-tools-version: 5.8\n// The swift-tools-version declares the minimum version of Swift required to build this package.\n\nimport PackageDescription\n\nlet package = Package(\n name: \"ChatPlugins\",\n platforms: [.macOS(.v12)],\n products: [\n .library(\n name: \"ChatPlugins\",\n targets: [\"TerminalChatPlugin\", \"ShortcutChatPlugin\"]\n ),\n ],\n dependencies: [\n .package(path: \"../Tool\"),\n ],\n targets: [\n .target(\n name: \"TerminalChatPlugin\",\n dependencies: [\n .product(name: \"Chat\", package: \"Tool\"),\n .product(name: \"Terminal\", package: \"Tool\"),\n .product(name: \"AppMonitoring\", package: \"Tool\"),\n ]\n ),\n .target(\n name: \"ShortcutChatPlugin\",\n dependencies: [\n .product(name: \"Chat\", package: \"Tool\"),\n .product(name: \"Terminal\", package: \"Tool\"),\n .product(name: \"AppMonitoring\", package: \"Tool\"),\n ]\n ),\n ]\n)\n\n" }, { "path": "ChatPlugins/Sources/ShortcutChatPlugin/ShortcutChatPlugin.swift", "content": "import ChatBasic\nimport Foundation\nimport Terminal\n\npublic final class ShortcutChatPlugin: ChatPlugin {\n public static var id: String { \"com.intii.shortcut\" }\n public static var command: String { \"shortcut\" }\n public static var name: String { \"Shortcut\" }\n public static var description: String { \"\"\"\n Run a shortcut and use message content as input. You need to provide the shortcut name as an argument, for example, `/shortcut(Shortcut Name)`.\n \"\"\" }\n\n let terminal: TerminalType\n\n init(terminal: TerminalType) {\n self.terminal = terminal\n }\n\n public init() {\n terminal = Terminal()\n }\n\n public func sendForTextResponse(_ request: Request) async\n -> AsyncThrowingStream\n {\n let stream = await sendForComplicatedResponse(request)\n return .init { continuation in\n let task = Task {\n do {\n for try await response in stream {\n switch response {\n case let .content(.text(content)):\n continuation.yield(content)\n default:\n break\n }\n }\n continuation.finish()\n } catch {\n continuation.finish(throwing: error)\n }\n }\n\n continuation.onTermination = { _ in\n task.cancel()\n }\n }\n }\n\n public func sendForComplicatedResponse(_ request: Request) async\n -> AsyncThrowingStream\n {\n return .init { continuation in\n let task = Task {\n let id = \"\\(Self.command)-\\(UUID().uuidString)\"\n\n guard let shortcutName = request.arguments.first, !shortcutName.isEmpty else {\n continuation.yield(.content(.text(\n \"Please provide the shortcut name in format: `/\\(Self.command)(shortcut name)`\"\n )))\n return\n }\n\n var input = String(request.text).trimmingCharacters(in: .whitespacesAndNewlines)\n if input.isEmpty {\n // if no input detected, use the previous message as input\n input = request.history.last?.content ?? \"\"\n }\n\n do {\n continuation.yield(.startAction(\n id: \"run\",\n task: \"Run shortcut `\\(shortcutName)`\"\n ))\n\n let env = ProcessInfo.processInfo.environment\n let shell = env[\"SHELL\"] ?? \"/bin/bash\"\n let temporaryURL = FileManager.default.temporaryDirectory\n let temporaryInputFileURL = temporaryURL\n .appendingPathComponent(\"\\(id)-input.txt\")\n let temporaryOutputFileURL = temporaryURL\n .appendingPathComponent(\"\\(id)-output\")\n\n try input.write(to: temporaryInputFileURL, atomically: true, encoding: .utf8)\n\n let command = \"\"\"\n shortcuts run \"\\(shortcutName)\" \\\n -i \"\\(temporaryInputFileURL.path)\" \\\n -o \"\\(temporaryOutputFileURL.path)\"\n \"\"\"\n\n continuation.yield(.startAction(\n id: \"run\",\n task: \"Run shortcut \\(shortcutName)\"\n ))\n\n do {\n let result = try await terminal.runCommand(\n shell,\n arguments: [\"-i\", \"-l\", \"-c\", command],\n currentDirectoryURL: nil,\n environment: [:]\n )\n continuation.yield(.finishAction(id: \"run\", result: .success(result)))\n } catch {\n continuation.yield(.finishAction(\n id: \"run\",\n result: .failure(error.localizedDescription)\n ))\n throw error\n }\n\n await Task.yield()\n try Task.checkCancellation()\n\n if FileManager.default.fileExists(atPath: temporaryOutputFileURL.path) {\n let data = try Data(contentsOf: temporaryOutputFileURL)\n if let text = String(data: data, encoding: .utf8) {\n var response = text\n if response.isEmpty {\n response = \"Finished\"\n }\n continuation.yield(.content(.text(response)))\n } else {\n let content = \"\"\"\n [View File](\\(temporaryOutputFileURL))\n \"\"\"\n continuation.yield(.content(.text(content)))\n }\n } else {\n continuation.yield(.content(.text(\"Finished\")))\n }\n\n } catch {\n continuation.yield(.content(.text(error.localizedDescription)))\n }\n\n continuation.finish()\n }\n\n continuation.onTermination = { _ in\n task.cancel()\n }\n }\n }\n}\n\n" }, { "path": "ChatPlugins/Sources/TerminalChatPlugin/TerminalChatPlugin.swift", "content": "import ChatBasic\nimport Foundation\nimport Terminal\nimport XcodeInspector\n\npublic final class TerminalChatPlugin: ChatPlugin {\n public static var id: String { \"com.intii.terminal\" }\n public static var command: String { \"shell\" }\n public static var name: String { \"Shell\" }\n public static var description: String { \"\"\"\n Run the command in the message from shell.\n\n You can use environment variable `$FILE_PATH` and `$PROJECT_ROOT` to access the current file path and project root.\n \"\"\" }\n\n let terminal: TerminalType\n\n init(terminal: TerminalType) {\n self.terminal = terminal\n }\n\n public init() {\n terminal = Terminal()\n }\n\n public func getTextContent(from request: Request) async\n -> AsyncStream\n {\n return .init { continuation in\n let task = Task {\n do {\n let fileURL = XcodeInspector.shared.realtimeActiveDocumentURL\n let projectURL = XcodeInspector.shared.realtimeActiveProjectURL\n\n var environment = [String: String]()\n if let fileURL {\n environment[\"FILE_PATH\"] = fileURL.path\n }\n if let projectURL {\n environment[\"PROJECT_ROOT\"] = projectURL.path\n }\n\n try Task.checkCancellation()\n\n let env = ProcessInfo.processInfo.environment\n let shell = env[\"SHELL\"] ?? \"/bin/bash\"\n\n let output = terminal.streamCommand(\n shell,\n arguments: [\"-i\", \"-l\", \"-c\", request.text],\n currentDirectoryURL: projectURL,\n environment: environment\n )\n\n var accumulatedOutput = \"\"\n for try await content in output {\n try Task.checkCancellation()\n accumulatedOutput += content\n continuation.yield(accumulatedOutput)\n }\n } catch let error as Terminal.TerminationError {\n let errorMessage = \"\\n\\n[error: \\(error.reason)]\"\n continuation.yield(errorMessage)\n } catch {\n let errorMessage = \"\\n\\n[error: \\(error.localizedDescription)]\"\n continuation.yield(errorMessage)\n }\n\n continuation.finish()\n }\n\n continuation.onTermination = { _ in\n task.cancel()\n Task {\n await self.terminal.terminate()\n }\n }\n }\n }\n\n public func sendForTextResponse(_ request: Request) async\n -> AsyncThrowingStream\n {\n let stream = await getTextContent(from: request)\n return .init { continuation in\n let task = Task {\n continuation.yield(\"Executing command: `\\(request.text)`\\n\\n\")\n continuation.yield(\"```console\\n\")\n for await text in stream {\n try Task.checkCancellation()\n continuation.yield(text)\n }\n continuation.yield(\"\\n```\\n\")\n continuation.finish()\n }\n\n continuation.onTermination = { _ in\n task.cancel()\n }\n }\n }\n\n public func formatContent(_ content: Response.Content) -> Response.Content {\n switch content {\n case let .text(content):\n return .text(\"\"\"\n ```console\n \\(content)\n ```\n \"\"\")\n }\n }\n\n public func sendForComplicatedResponse(_ request: Request) async\n -> AsyncThrowingStream\n {\n return .init { continuation in\n let task = Task {\n var updateTime = Date()\n\n continuation.yield(.startAction(id: \"run\", task: \"Run `\\(request.text)`\"))\n\n let textStream = await getTextContent(from: request)\n var previousOutput = \"\"\n\n continuation.yield(.finishAction(\n id: \"run\",\n result: .success(\"Executed.\")\n ))\n\n for await accumulatedOutput in textStream {\n try Task.checkCancellation()\n\n let newContent = accumulatedOutput.dropFirst(previousOutput.count)\n previousOutput = accumulatedOutput\n\n if !newContent.isEmpty {\n if Date().timeIntervalSince(updateTime) > 60 * 2 {\n continuation.yield(.startNewMessage)\n continuation.yield(.startAction(\n id: \"run\",\n task: \"Continue `\\(request.text)`\"\n ))\n continuation.yield(.finishAction(\n id: \"run\",\n result: .success(\"Executed.\")\n ))\n continuation.yield(.content(.text(\"[continue]\\n\")))\n updateTime = Date()\n }\n\n continuation.yield(.content(.text(String(newContent))))\n }\n }\n\n continuation.finish()\n }\n\n continuation.onTermination = { _ in\n task.cancel()\n }\n }\n }\n}\n\n" }, { "path": "ChatPlugins/Tests/ChatPluginsTests/ChatPluginsTests.swift", "content": "import Testing\n@testable import ChatPlugins\n\n@Test func example() async throws {\n // Write your test here and use APIs like `#expect(...)` to check expected conditions.\n}\n" }, { "path": "CommunicationBridge/ServiceDelegate.swift", "content": "import AppKit\nimport Foundation\nimport Logger\nimport XPCShared\n\nclass ServiceDelegate: NSObject, NSXPCListenerDelegate {\n func listener(\n _: NSXPCListener,\n shouldAcceptNewConnection newConnection: NSXPCConnection\n ) -> Bool {\n newConnection.exportedInterface = NSXPCInterface(\n with: CommunicationBridgeXPCServiceProtocol.self\n )\n\n let exportedObject = XPCService()\n newConnection.exportedObject = exportedObject\n newConnection.resume()\n\n Logger.communicationBridge.info(\"Accepted new connection.\")\n\n return true\n }\n}\n\nclass XPCService: CommunicationBridgeXPCServiceProtocol {\n static let eventHandler = EventHandler()\n\n func launchExtensionServiceIfNeeded(\n withReply reply: @escaping (NSXPCListenerEndpoint?) -> Void\n ) {\n Task {\n await Self.eventHandler.launchExtensionServiceIfNeeded(withReply: reply)\n }\n }\n\n func quit(withReply reply: @escaping () -> Void) {\n Task {\n await Self.eventHandler.quit(withReply: reply)\n }\n }\n\n func updateServiceEndpoint(\n endpoint: NSXPCListenerEndpoint,\n withReply reply: @escaping () -> Void\n ) {\n Task {\n await Self.eventHandler.updateServiceEndpoint(endpoint: endpoint, withReply: reply)\n }\n }\n}\n\nactor EventHandler {\n var endpoint: NSXPCListenerEndpoint?\n let launcher = ExtensionServiceLauncher()\n var exitTask: Task?\n\n init() {\n Task { await rescheduleExitTask() }\n }\n\n func launchExtensionServiceIfNeeded(\n withReply reply: @escaping (NSXPCListenerEndpoint?) -> Void\n ) async {\n rescheduleExitTask()\n #if DEBUG\n if let endpoint, !(await testXPCListenerEndpoint(endpoint)) {\n self.endpoint = nil\n }\n reply(endpoint)\n #else\n if await launcher.isApplicationValid {\n Logger.communicationBridge.info(\"Service app is still valid\")\n reply(endpoint)\n } else {\n endpoint = nil\n await launcher.launch()\n reply(nil)\n }\n #endif\n }\n\n func quit(withReply reply: () -> Void) {\n Logger.communicationBridge.info(\"Exiting service.\")\n listener.invalidate()\n exit(0)\n }\n\n func updateServiceEndpoint(endpoint: NSXPCListenerEndpoint, withReply reply: () -> Void) {\n rescheduleExitTask()\n self.endpoint = endpoint\n reply()\n }\n\n /// The bridge will kill itself when it's not used for a period.\n /// It's fine that the bridge is killed because it will be launched again when needed.\n private func rescheduleExitTask() {\n exitTask?.cancel()\n exitTask = Task {\n #if DEBUG\n try await Task.sleep(nanoseconds: 60_000_000_000)\n Logger.communicationBridge.info(\"Exit will be called in release build.\")\n #else\n try await Task.sleep(nanoseconds: 1_800_000_000_000)\n Logger.communicationBridge.info(\"Exiting service.\")\n listener.invalidate()\n exit(0)\n #endif\n }\n }\n}\n\nactor ExtensionServiceLauncher {\n let appIdentifier = bundleIdentifierBase.appending(\".ExtensionService\")\n let appURL = Bundle.main.bundleURL.appendingPathComponent(\n \"CopilotForXcodeExtensionService.app\"\n )\n var isLaunching: Bool = false\n var application: NSRunningApplication?\n var isApplicationValid: Bool {\n guard let application else { return false }\n if application.isTerminated { return false }\n let identifier = application.processIdentifier\n if let application = NSWorkspace.shared.runningApplications.first(where: {\n $0.processIdentifier == identifier\n }) {\n Logger.communicationBridge.info(\n \"Service app found: \\(application.processIdentifier) \\(String(describing: application.bundleIdentifier))\"\n )\n return true\n }\n return false\n }\n\n func launch() {\n guard !isLaunching else { return }\n isLaunching = true\n\n Logger.communicationBridge.info(\"Launching extension service app.\")\n\n NSWorkspace.shared.openApplication(\n at: appURL,\n configuration: {\n let configuration = NSWorkspace.OpenConfiguration()\n configuration.createsNewApplicationInstance = false\n configuration.addsToRecentItems = false\n configuration.activates = false\n return configuration\n }()\n ) { app, error in\n if let error = error {\n Logger.communicationBridge.error(\n \"Failed to launch extension service app: \\(error)\"\n )\n } else {\n Logger.communicationBridge.info(\n \"Finished launching extension service app.\"\n )\n }\n\n self.application = app\n self.isLaunching = false\n }\n }\n}\n\n" }, { "path": "CommunicationBridge/main.swift", "content": "import AppKit\nimport Foundation\n\nclass AppDelegate: NSObject, NSApplicationDelegate {}\n\nlet bundleIdentifierBase = Bundle(url: Bundle.main.bundleURL.appendingPathComponent(\n \"CopilotForXcodeExtensionService.app\"\n))?.object(forInfoDictionaryKey: \"BUNDLE_IDENTIFIER_BASE\") as? String ?? \"com.intii.CopilotForXcode\"\n\nlet serviceIdentifier = bundleIdentifierBase + \".CommunicationBridge\"\nlet appDelegate = AppDelegate()\nlet delegate = ServiceDelegate()\nlet listener = NSXPCListener(machServiceName: serviceIdentifier)\nlistener.delegate = delegate\nlistener.resume()\nlet app = NSApplication.shared\napp.delegate = appDelegate\napp.run()\n\n" }, { "path": "Config.debug.xcconfig", "content": "#include \"Version.xcconfig\"\nSLASH = /\n\nHOST_APP_NAME = Copilot for Xcode Dev\nBUNDLE_IDENTIFIER_BASE = dev.com.intii.CopilotForXcode\nSPARKLE_FEED_URL = http:$(SLASH)$(SLASH)127.0.0.1:9433/appcast.xml\nSPARKLE_PUBLIC_KEY = WDzm5GHnc6c8kjeJEgX5GuGiPpW6Lc/ovGjLnrrZvPY=\nAPPLICATION_SUPPORT_FOLDER = dev.com.intii.CopilotForXcode\nEXTENSION_BUNDLE_NAME = Copilot Dev\nEXTENSION_BUNDLE_DISPLAY_NAME = Copilot Dev\nEXTENSION_SERVICE_NAME = CopilotForXcodeExtensionService\n\n// see also target Configs\n" }, { "path": "Config.xcconfig", "content": "#include \"Version.xcconfig\"\nSLASH = /\n\nHOST_APP_NAME = Copilot for Xcode\nBUNDLE_IDENTIFIER_BASE = com.intii.CopilotForXcode\nSPARKLE_FEED_URL = https:$(SLASH)$(SLASH)copilotforxcode.intii.com/appcast.xml\nSPARKLE_PUBLIC_KEY = WDzm5GHnc6c8kjeJEgX5GuGiPpW6Lc/ovGjLnrrZvPY=\nAPPLICATION_SUPPORT_FOLDER = com.intii.CopilotForXcode\nEXTENSION_BUNDLE_NAME = Copilot\nEXTENSION_BUNDLE_DISPLAY_NAME = Copilot\nEXTENSION_SERVICE_NAME = CopilotForXcodeExtensionService\n\n// see also target Configs\n" }, { "path": "Copilot for Xcode/App.swift", "content": "import Client\nimport HostApp\nimport LaunchAgentManager\nimport SwiftUI\nimport UpdateChecker\nimport XPCShared\n\nstruct VisualEffect: NSViewRepresentable {\n func makeNSView(context: Self.Context) -> NSView { return NSVisualEffectView() }\n func updateNSView(_ nsView: NSView, context: Context) {}\n}\n\nclass TheUpdateCheckerDelegate: UpdateCheckerDelegate {\n func prepareForRelaunch(finish: @escaping () -> Void) {\n Task {\n let service = try? getService()\n try? await service?.quitService()\n finish()\n }\n }\n}\n\nlet updateCheckerDelegate = TheUpdateCheckerDelegate()\n\n@main\nstruct CopilotForXcodeApp: App {\n var body: some Scene {\n WindowGroup {\n TabContainer()\n .frame(minWidth: 800, minHeight: 600)\n .background(VisualEffect().ignoresSafeArea())\n .onAppear {\n UserDefaults.setupDefaultSettings()\n }\n .environment(\n \\.updateChecker,\n {\n let checker = UpdateChecker(\n hostBundle: Bundle.main,\n shouldAutomaticallyCheckForUpdate: false\n )\n checker.updateCheckerDelegate = updateCheckerDelegate\n return checker\n }()\n )\n }\n }\n}\n\nvar isPreview: Bool { ProcessInfo.processInfo.environment[\"XCODE_RUNNING_FOR_PREVIEWS\"] == \"1\" }\n\n" }, { "path": "Copilot for Xcode/Assets.xcassets/AccentColor.colorset/Contents.json", "content": "{\n \"colors\" : [\n {\n \"idiom\" : \"universal\"\n }\n ],\n \"info\" : {\n \"author\" : \"xcode\",\n \"version\" : 1\n }\n}\n" }, { "path": "Copilot for Xcode/Assets.xcassets/AppIcon.appiconset/Contents.json", "content": "{\n \"images\" : [\n {\n \"filename\" : \"app-icon@16w.png\",\n \"idiom\" : \"mac\",\n \"scale\" : \"1x\",\n \"size\" : \"16x16\"\n },\n {\n \"filename\" : \"app-icon@32w.png\",\n \"idiom\" : \"mac\",\n \"scale\" : \"2x\",\n \"size\" : \"16x16\"\n },\n {\n \"filename\" : \"app-icon@32w.png\",\n \"idiom\" : \"mac\",\n \"scale\" : \"1x\",\n \"size\" : \"32x32\"\n },\n {\n \"filename\" : \"app-icon@64w.png\",\n \"idiom\" : \"mac\",\n \"scale\" : \"2x\",\n \"size\" : \"32x32\"\n },\n {\n \"filename\" : \"app-icon@128w.png\",\n \"idiom\" : \"mac\",\n \"scale\" : \"1x\",\n \"size\" : \"128x128\"\n },\n {\n \"filename\" : \"app-icon@256w.png\",\n \"idiom\" : \"mac\",\n \"scale\" : \"2x\",\n \"size\" : \"128x128\"\n },\n {\n \"filename\" : \"app-icon@256w.png\",\n \"idiom\" : \"mac\",\n \"scale\" : \"1x\",\n \"size\" : \"256x256\"\n },\n {\n \"filename\" : \"app-icon@512w.png\",\n \"idiom\" : \"mac\",\n \"scale\" : \"2x\",\n \"size\" : \"256x256\"\n },\n {\n \"filename\" : \"app-icon@512w.png\",\n \"idiom\" : \"mac\",\n \"scale\" : \"1x\",\n \"size\" : \"512x512\"\n },\n {\n \"filename\" : \"app-icon.png\",\n \"idiom\" : \"mac\",\n \"scale\" : \"2x\",\n \"size\" : \"512x512\"\n }\n ],\n \"info\" : {\n \"author\" : \"xcode\",\n \"version\" : 1\n }\n}\n" }, { "path": "Copilot for Xcode/Assets.xcassets/BackgroundColor.colorset/Contents.json", "content": "{\n \"colors\" : [\n {\n \"color\" : {\n \"color-space\" : \"srgb\",\n \"components\" : {\n \"alpha\" : \"1.000\",\n \"blue\" : \"40\",\n \"green\" : \"23\",\n \"red\" : \"25\"\n }\n },\n \"idiom\" : \"universal\"\n }\n ],\n \"info\" : {\n \"author\" : \"xcode\",\n \"version\" : 1\n }\n}\n" }, { "path": "Copilot for Xcode/Assets.xcassets/BackgroundColorTop.colorset/Contents.json", "content": "{\n \"colors\" : [\n {\n \"color\" : {\n \"color-space\" : \"srgb\",\n \"components\" : {\n \"alpha\" : \"1.000\",\n \"blue\" : \"54\",\n \"green\" : \"25\",\n \"red\" : \"30\"\n }\n },\n \"idiom\" : \"universal\"\n }\n ],\n \"info\" : {\n \"author\" : \"xcode\",\n \"version\" : 1\n }\n}\n" }, { "path": "Copilot for Xcode/Assets.xcassets/ButtonBackgroundColorDefault.colorset/Contents.json", "content": "{\n \"colors\" : [\n {\n \"color\" : {\n \"color-space\" : \"srgb\",\n \"components\" : {\n \"alpha\" : \"1.000\",\n \"blue\" : \"0x46\",\n \"green\" : \"0x24\",\n \"red\" : \"0x2C\"\n }\n },\n \"idiom\" : \"universal\"\n }\n ],\n \"info\" : {\n \"author\" : \"xcode\",\n \"version\" : 1\n }\n}\n" }, { "path": "Copilot for Xcode/Assets.xcassets/ButtonBackgroundColorPressed.colorset/Contents.json", "content": "{\n \"colors\" : [\n {\n \"color\" : {\n \"color-space\" : \"srgb\",\n \"components\" : {\n \"alpha\" : \"1.000\",\n \"blue\" : \"0.275\",\n \"green\" : \"0.141\",\n \"red\" : \"0.290\"\n }\n },\n \"idiom\" : \"universal\"\n }\n ],\n \"info\" : {\n \"author\" : \"xcode\",\n \"version\" : 1\n }\n}\n" }, { "path": "Copilot for Xcode/Assets.xcassets/Contents.json", "content": "{\n \"info\" : {\n \"author\" : \"xcode\",\n \"version\" : 1\n }\n}\n" }, { "path": "Copilot for Xcode/Copilot_for_Xcode.entitlements", "content": "\n\n\n\n\tcom.apple.security.app-sandbox\n\t\n\tcom.apple.security.application-groups\n\t\n\t\t$(TeamIdentifierPrefix)group.$(BUNDLE_IDENTIFIER_BASE)\n\t\n\tcom.apple.security.automation.apple-events\n\t\n\tcom.apple.security.files.user-selected.read-only\n\t\n\tkeychain-access-groups\n\t\n\t\t$(AppIdentifierPrefix)$(BUNDLE_IDENTIFIER_BASE).Shared\n\t\n\n\n" }, { "path": "Copilot for Xcode/Preview Content/Preview Assets.xcassets/Contents.json", "content": "{\n \"info\" : {\n \"author\" : \"xcode\",\n \"version\" : 1\n }\n}\n" }, { "path": "Copilot for Xcode.xcodeproj/project.pbxproj", "content": "// !$*UTF8*$!\n{\n\tarchiveVersion = 1;\n\tclasses = {\n\t};\n\tobjectVersion = 56;\n\tobjects = {\n\n/* Begin PBXBuildFile section */\n\t\tC8009BFF2941C551007AA7E8 /* ToggleRealtimeSuggestionsCommand.swift in Sources */ = {isa = PBXBuildFile; fileRef = C8009BFE2941C551007AA7E8 /* ToggleRealtimeSuggestionsCommand.swift */; };\n\t\tC8009C032941C576007AA7E8 /* RealtimeSuggestionCommand.swift in Sources */ = {isa = PBXBuildFile; fileRef = C8009C022941C576007AA7E8 /* RealtimeSuggestionCommand.swift */; };\n\t\tC800DBB1294C624D00B04CAC /* PrefetchSuggestionsCommand.swift in Sources */ = {isa = PBXBuildFile; fileRef = C800DBB0294C624D00B04CAC /* PrefetchSuggestionsCommand.swift */; };\n\t\tC80FFB962A95F58200704A25 /* AcceptPromptToCodeCommand.swift in Sources */ = {isa = PBXBuildFile; fileRef = C80FFB952A95F58200704A25 /* AcceptPromptToCodeCommand.swift */; };\n\t\tC81291D72994FE6900196E12 /* Main.storyboard in Resources */ = {isa = PBXBuildFile; fileRef = C81291D52994FE6900196E12 /* Main.storyboard */; };\n\t\tC814588F2939EFDC00135263 /* Cocoa.framework in Frameworks */ = {isa = PBXBuildFile; fileRef = C814588E2939EFDC00135263 /* Cocoa.framework */; };\n\t\tC81458942939EFDC00135263 /* SourceEditorExtension.swift in Sources */ = {isa = PBXBuildFile; fileRef = C81458932939EFDC00135263 /* SourceEditorExtension.swift */; };\n\t\tC81458962939EFDC00135263 /* GetSuggestionsCommand.swift in Sources */ = {isa = PBXBuildFile; fileRef = C81458952939EFDC00135263 /* GetSuggestionsCommand.swift */; };\n\t\tC814589B2939EFDC00135263 /* Copilot.appex in Embed Foundation Extensions */ = {isa = PBXBuildFile; fileRef = C814588C2939EFDC00135263 /* Copilot.appex */; settings = {ATTRIBUTES = (RemoveHeadersOnCopy, ); }; };\n\t\tC8189B1A2938972F00C9DCDA /* App.swift in Sources */ = {isa = PBXBuildFile; fileRef = C8189B192938972F00C9DCDA /* App.swift */; };\n\t\tC8189B1E2938973000C9DCDA /* Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = C8189B1D2938973000C9DCDA /* Assets.xcassets */; };\n\t\tC8189B212938973000C9DCDA /* Preview Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = C8189B202938973000C9DCDA /* Preview Assets.xcassets */; };\n\t\tC8216B73298036EC00AD38C7 /* main.swift in Sources */ = {isa = PBXBuildFile; fileRef = C8216B72298036EC00AD38C7 /* main.swift */; };\n\t\tC8216B782980370100AD38C7 /* ReloadLaunchAgent.swift in Sources */ = {isa = PBXBuildFile; fileRef = C8216B772980370100AD38C7 /* ReloadLaunchAgent.swift */; };\n\t\tC8216B7D2980374300AD38C7 /* ArgumentParser in Frameworks */ = {isa = PBXBuildFile; productRef = C8216B7C2980374300AD38C7 /* ArgumentParser */; };\n\t\tC8216B802980378300AD38C7 /* Helper in Embed XPCService */ = {isa = PBXBuildFile; fileRef = C8216B70298036EC00AD38C7 /* Helper */; settings = {ATTRIBUTES = (CodeSignOnCopy, ); }; };\n\t\tC828B27F2B1F7B4F00E7612A /* ExtensionPoint.appextensionpoint in Copy Extension Point */ = {isa = PBXBuildFile; fileRef = C828B27D2B1F241500E7612A /* ExtensionPoint.appextensionpoint */; };\n\t\tC8520301293C4D9000460097 /* Helpers.swift in Sources */ = {isa = PBXBuildFile; fileRef = C8520300293C4D9000460097 /* Helpers.swift */; };\n\t\tC861A6A329E5503F005C41A3 /* PromptToCodeCommand.swift in Sources */ = {isa = PBXBuildFile; fileRef = C861A6A229E5503F005C41A3 /* PromptToCodeCommand.swift */; };\n\t\tC861E6112994F6070056CB02 /* AppDelegate.swift in Sources */ = {isa = PBXBuildFile; fileRef = C861E6102994F6070056CB02 /* AppDelegate.swift */; };\n\t\tC861E6152994F6080056CB02 /* Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = C861E6142994F6080056CB02 /* Assets.xcassets */; };\n\t\tC861E61E2994F6150056CB02 /* Service in Frameworks */ = {isa = PBXBuildFile; productRef = C861E61D2994F6150056CB02 /* Service */; };\n\t\tC861E6202994F63A0056CB02 /* ServiceDelegate.swift in Sources */ = {isa = PBXBuildFile; fileRef = C861E61F2994F6390056CB02 /* ServiceDelegate.swift */; };\n\t\tC86612F82A06AF74009197D9 /* HostApp in Frameworks */ = {isa = PBXBuildFile; productRef = C86612F72A06AF74009197D9 /* HostApp */; };\n\t\tC8738B662BE4D4B900609E7F /* main.swift in Sources */ = {isa = PBXBuildFile; fileRef = C8738B652BE4D4B900609E7F /* main.swift */; };\n\t\tC8738B6B2BE4D56F00609E7F /* ServiceDelegate.swift in Sources */ = {isa = PBXBuildFile; fileRef = C8738B6A2BE4D56F00609E7F /* ServiceDelegate.swift */; };\n\t\tC8738B6F2BE4F7A600609E7F /* XPCShared in Frameworks */ = {isa = PBXBuildFile; productRef = C8738B6E2BE4F7A600609E7F /* XPCShared */; };\n\t\tC8738B712BE4F8B700609E7F /* XPCController.swift in Sources */ = {isa = PBXBuildFile; fileRef = C8738B702BE4F8B700609E7F /* XPCController.swift */; };\n\t\tC8738B7B2BE5363800609E7F /* SandboxedClientTesterApp.swift in Sources */ = {isa = PBXBuildFile; fileRef = C8738B7A2BE5363800609E7F /* SandboxedClientTesterApp.swift */; };\n\t\tC8738B7D2BE5363800609E7F /* ContentView.swift in Sources */ = {isa = PBXBuildFile; fileRef = C8738B7C2BE5363800609E7F /* ContentView.swift */; };\n\t\tC8738B7F2BE5363900609E7F /* Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = C8738B7E2BE5363900609E7F /* Assets.xcassets */; };\n\t\tC8738B822BE5363900609E7F /* Preview Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = C8738B812BE5363900609E7F /* Preview Assets.xcassets */; };\n\t\tC8738B882BE5365000609E7F /* Client in Frameworks */ = {isa = PBXBuildFile; productRef = C8738B872BE5365000609E7F /* Client */; };\n\t\tC8738B8A2BE540D000609E7F /* bridgeLaunchAgent.plist in Copy Launch Agent */ = {isa = PBXBuildFile; fileRef = C8738B6D2BE4F3E800609E7F /* bridgeLaunchAgent.plist */; };\n\t\tC8738B8B2BE540DD00609E7F /* CommunicationBridge in Embed XPCService */ = {isa = PBXBuildFile; fileRef = C8738B632BE4D4B900609E7F /* CommunicationBridge */; settings = {ATTRIBUTES = (CodeSignOnCopy, ); }; };\n\t\tC8758E7029F04BFF00D29C1C /* CustomCommand.swift in Sources */ = {isa = PBXBuildFile; fileRef = C8758E6F29F04BFF00D29C1C /* CustomCommand.swift */; };\n\t\tC8758E7229F04CF100D29C1C /* SeparatorCommand.swift in Sources */ = {isa = PBXBuildFile; fileRef = C8758E7129F04CF100D29C1C /* SeparatorCommand.swift */; };\n\t\tC87B03A5293B261200C77EAE /* AcceptSuggestionCommand.swift in Sources */ = {isa = PBXBuildFile; fileRef = C87B03A4293B261200C77EAE /* AcceptSuggestionCommand.swift */; };\n\t\tC87B03A7293B261900C77EAE /* RejectSuggestionCommand.swift in Sources */ = {isa = PBXBuildFile; fileRef = C87B03A6293B261900C77EAE /* RejectSuggestionCommand.swift */; };\n\t\tC87B03A9293B262600C77EAE /* NextSuggestionCommand.swift in Sources */ = {isa = PBXBuildFile; fileRef = C87B03A8293B262600C77EAE /* NextSuggestionCommand.swift */; };\n\t\tC87B03AB293B262E00C77EAE /* PreviousSuggestionCommand.swift in Sources */ = {isa = PBXBuildFile; fileRef = C87B03AA293B262E00C77EAE /* PreviousSuggestionCommand.swift */; };\n\t\tC87B03AC293B2CF300C77EAE /* XcodeKit.framework in Frameworks */ = {isa = PBXBuildFile; fileRef = C81458902939EFDC00135263 /* XcodeKit.framework */; };\n\t\tC87B03AD293B2CF300C77EAE /* XcodeKit.framework in Embed Frameworks */ = {isa = PBXBuildFile; fileRef = C81458902939EFDC00135263 /* XcodeKit.framework */; settings = {ATTRIBUTES = (CodeSignOnCopy, RemoveHeadersOnCopy, ); }; };\n\t\tC882175C294187EF00A22FD3 /* Client in Frameworks */ = {isa = PBXBuildFile; productRef = C882175B294187EF00A22FD3 /* Client */; };\n\t\tC89E75C32A46FB32000DD64F /* AppDelegate+Menu.swift in Sources */ = {isa = PBXBuildFile; fileRef = C89E75C22A46FB32000DD64F /* AppDelegate+Menu.swift */; };\n\t\tC8C8B60929AFA35F00034BEE /* CopilotForXcodeExtensionService.app in Embed XPCService */ = {isa = PBXBuildFile; fileRef = C861E60E2994F6070056CB02 /* CopilotForXcodeExtensionService.app */; settings = {ATTRIBUTES = (RemoveHeadersOnCopy, ); }; };\n\t\tC8DCF00029CE11D500FDDDD7 /* OpenChat.swift in Sources */ = {isa = PBXBuildFile; fileRef = C8DCEFFF29CE11D500FDDDD7 /* OpenChat.swift */; };\n\t\tC8DD9CB12BC673F80036641C /* CloseIdleTabsCommand.swift in Sources */ = {isa = PBXBuildFile; fileRef = C8DD9CB02BC673F80036641C /* CloseIdleTabsCommand.swift */; };\n/* End PBXBuildFile section */\n\n/* Begin PBXContainerItemProxy section */\n\t\tC81291AF2994F92700196E12 /* PBXContainerItemProxy */ = {\n\t\t\tisa = PBXContainerItemProxy;\n\t\t\tcontainerPortal = C8189B0E2938972F00C9DCDA /* Project object */;\n\t\t\tproxyType = 1;\n\t\t\tremoteGlobalIDString = C861E60D2994F6070056CB02;\n\t\t\tremoteInfo = ExtensionService;\n\t\t};\n\t\tC81458992939EFDC00135263 /* PBXContainerItemProxy */ = {\n\t\t\tisa = PBXContainerItemProxy;\n\t\t\tcontainerPortal = C8189B0E2938972F00C9DCDA /* Project object */;\n\t\t\tproxyType = 1;\n\t\t\tremoteGlobalIDString = C814588B2939EFDC00135263;\n\t\t\tremoteInfo = EditorExtension;\n\t\t};\n\t\tC8216B7E2980377E00AD38C7 /* PBXContainerItemProxy */ = {\n\t\t\tisa = PBXContainerItemProxy;\n\t\t\tcontainerPortal = C8189B0E2938972F00C9DCDA /* Project object */;\n\t\t\tproxyType = 1;\n\t\t\tremoteGlobalIDString = C8216B6F298036EC00AD38C7;\n\t\t\tremoteInfo = Helper;\n\t\t};\n\t\tC8738B8C2BE540F900609E7F /* PBXContainerItemProxy */ = {\n\t\t\tisa = PBXContainerItemProxy;\n\t\t\tcontainerPortal = C8189B0E2938972F00C9DCDA /* Project object */;\n\t\t\tproxyType = 1;\n\t\t\tremoteGlobalIDString = C8738B622BE4D4B900609E7F;\n\t\t\tremoteInfo = CommunicationBridge;\n\t\t};\n/* End PBXContainerItemProxy section */\n\n/* Begin PBXCopyFilesBuildPhase section */\n\t\tC814589F2939EFDC00135263 /* Embed Foundation Extensions */ = {\n\t\t\tisa = PBXCopyFilesBuildPhase;\n\t\t\tbuildActionMask = 2147483647;\n\t\t\tdstPath = \"\";\n\t\t\tdstSubfolderSpec = 13;\n\t\t\tfiles = (\n\t\t\t\tC814589B2939EFDC00135263 /* Copilot.appex in Embed Foundation Extensions */,\n\t\t\t);\n\t\t\tname = \"Embed Foundation Extensions\";\n\t\t\trunOnlyForDeploymentPostprocessing = 0;\n\t\t};\n\t\tC8216B6E298036EC00AD38C7 /* CopyFiles */ = {\n\t\t\tisa = PBXCopyFilesBuildPhase;\n\t\t\tbuildActionMask = 2147483647;\n\t\t\tdstPath = /usr/share/man/man1/;\n\t\t\tdstSubfolderSpec = 0;\n\t\t\tfiles = (\n\t\t\t);\n\t\t\trunOnlyForDeploymentPostprocessing = 1;\n\t\t};\n\t\tC828B27E2B1F7B3C00E7612A /* Copy Extension Point */ = {\n\t\t\tisa = PBXCopyFilesBuildPhase;\n\t\t\tbuildActionMask = 2147483647;\n\t\t\tdstPath = \"$(EXTENSIONS_FOLDER_PATH)\";\n\t\t\tdstSubfolderSpec = 16;\n\t\t\tfiles = (\n\t\t\t\tC828B27F2B1F7B4F00E7612A /* ExtensionPoint.appextensionpoint in Copy Extension Point */,\n\t\t\t);\n\t\t\tname = \"Copy Extension Point\";\n\t\t\trunOnlyForDeploymentPostprocessing = 0;\n\t\t};\n\t\tC8520306293CF0EF00460097 /* Embed XPCService */ = {\n\t\t\tisa = PBXCopyFilesBuildPhase;\n\t\t\tbuildActionMask = 2147483647;\n\t\t\tdstPath = ../Applications;\n\t\t\tdstSubfolderSpec = 6;\n\t\t\tfiles = (\n\t\t\t\tC8738B8B2BE540DD00609E7F /* CommunicationBridge in Embed XPCService */,\n\t\t\t\tC8216B802980378300AD38C7 /* Helper in Embed XPCService */,\n\t\t\t\tC8C8B60929AFA35F00034BEE /* CopilotForXcodeExtensionService.app in Embed XPCService */,\n\t\t\t);\n\t\t\tname = \"Embed XPCService\";\n\t\t\trunOnlyForDeploymentPostprocessing = 0;\n\t\t};\n\t\tC8738B612BE4D4B900609E7F /* CopyFiles */ = {\n\t\t\tisa = PBXCopyFilesBuildPhase;\n\t\t\tbuildActionMask = 12;\n\t\t\tdstPath = \"\";\n\t\t\tdstSubfolderSpec = 16;\n\t\t\tfiles = (\n\t\t\t);\n\t\t\trunOnlyForDeploymentPostprocessing = 0;\n\t\t};\n\t\tC87B03AE293B2CF300C77EAE /* Embed Frameworks */ = {\n\t\t\tisa = PBXCopyFilesBuildPhase;\n\t\t\tbuildActionMask = 2147483647;\n\t\t\tdstPath = \"\";\n\t\t\tdstSubfolderSpec = 10;\n\t\t\tfiles = (\n\t\t\t\tC87B03AD293B2CF300C77EAE /* XcodeKit.framework in Embed Frameworks */,\n\t\t\t);\n\t\t\tname = \"Embed Frameworks\";\n\t\t\trunOnlyForDeploymentPostprocessing = 0;\n\t\t};\n\t\tC8C8B60829AFA32800034BEE /* Embed Service */ = {\n\t\t\tisa = PBXCopyFilesBuildPhase;\n\t\t\tbuildActionMask = 2147483647;\n\t\t\tdstPath = \"$(CONTENTS_FOLDER_PATH)/XPCServices\";\n\t\t\tdstSubfolderSpec = 16;\n\t\t\tfiles = (\n\t\t\t);\n\t\t\tname = \"Embed Service\";\n\t\t\trunOnlyForDeploymentPostprocessing = 0;\n\t\t};\n\t\tC8F1032A2A7A38D200D28F4F /* Copy Launch Agent */ = {\n\t\t\tisa = PBXCopyFilesBuildPhase;\n\t\t\tbuildActionMask = 12;\n\t\t\tdstPath = Contents/Library/LaunchAgents;\n\t\t\tdstSubfolderSpec = 1;\n\t\t\tfiles = (\n\t\t\t\tC8738B8A2BE540D000609E7F /* bridgeLaunchAgent.plist in Copy Launch Agent */,\n\t\t\t);\n\t\t\tname = \"Copy Launch Agent\";\n\t\t\trunOnlyForDeploymentPostprocessing = 0;\n\t\t};\n/* End PBXCopyFilesBuildPhase section */\n\n/* Begin PBXFileReference section */\n\t\tC8009BFE2941C551007AA7E8 /* ToggleRealtimeSuggestionsCommand.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ToggleRealtimeSuggestionsCommand.swift; sourceTree = \"\"; };\n\t\tC8009C022941C576007AA7E8 /* RealtimeSuggestionCommand.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = RealtimeSuggestionCommand.swift; sourceTree = \"\"; };\n\t\tC800DBB0294C624D00B04CAC /* PrefetchSuggestionsCommand.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = PrefetchSuggestionsCommand.swift; sourceTree = \"\"; };\n\t\tC80FFB952A95F58200704A25 /* AcceptPromptToCodeCommand.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = AcceptPromptToCodeCommand.swift; sourceTree = \"\"; };\n\t\tC81291D52994FE6900196E12 /* Main.storyboard */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = file.storyboard; path = Main.storyboard; sourceTree = \"\"; };\n\t\tC81291D92994FE7900196E12 /* Info.plist */ = {isa = PBXFileReference; lastKnownFileType = text.plist; path = Info.plist; sourceTree = \"\"; };\n\t\tC814588C2939EFDC00135263 /* Copilot.appex */ = {isa = PBXFileReference; explicitFileType = \"wrapper.app-extension\"; includeInIndex = 0; path = Copilot.appex; sourceTree = BUILT_PRODUCTS_DIR; };\n\t\tC814588E2939EFDC00135263 /* Cocoa.framework */ = {isa = PBXFileReference; lastKnownFileType = wrapper.framework; name = Cocoa.framework; path = System/Library/Frameworks/Cocoa.framework; sourceTree = SDKROOT; };\n\t\tC81458902939EFDC00135263 /* XcodeKit.framework */ = {isa = PBXFileReference; lastKnownFileType = wrapper.framework; name = XcodeKit.framework; path = Library/Frameworks/XcodeKit.framework; sourceTree = DEVELOPER_DIR; };\n\t\tC81458932939EFDC00135263 /* SourceEditorExtension.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = SourceEditorExtension.swift; sourceTree = \"\"; };\n\t\tC81458952939EFDC00135263 /* GetSuggestionsCommand.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = GetSuggestionsCommand.swift; sourceTree = \"\"; };\n\t\tC81458972939EFDC00135263 /* Info.plist */ = {isa = PBXFileReference; lastKnownFileType = text.plist.xml; path = Info.plist; sourceTree = \"\"; };\n\t\tC81458982939EFDC00135263 /* EditorExtension.entitlements */ = {isa = PBXFileReference; lastKnownFileType = text.plist.entitlements; path = EditorExtension.entitlements; sourceTree = \"\"; };\n\t\tC81458AD293A009600135263 /* Config.xcconfig */ = {isa = PBXFileReference; lastKnownFileType = text.xcconfig; path = Config.xcconfig; sourceTree = \"\"; };\n\t\tC81458AE293A009800135263 /* Config.debug.xcconfig */ = {isa = PBXFileReference; lastKnownFileType = text.xcconfig; path = Config.debug.xcconfig; sourceTree = \"\"; };\n\t\tC8189B162938972F00C9DCDA /* Copilot for Xcode Dev.app */ = {isa = PBXFileReference; explicitFileType = wrapper.application; includeInIndex = 0; path = \"Copilot for Xcode Dev.app\"; sourceTree = BUILT_PRODUCTS_DIR; };\n\t\tC8189B192938972F00C9DCDA /* App.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = App.swift; sourceTree = \"\"; };\n\t\tC8189B1D2938973000C9DCDA /* Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = Assets.xcassets; sourceTree = \"\"; };\n\t\tC8189B202938973000C9DCDA /* Preview Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = \"Preview Assets.xcassets\"; sourceTree = \"\"; };\n\t\tC8189B222938973000C9DCDA /* Copilot_for_Xcode.entitlements */ = {isa = PBXFileReference; lastKnownFileType = text.plist.entitlements; path = Copilot_for_Xcode.entitlements; sourceTree = \"\"; };\n\t\tC8189B282938979000C9DCDA /* Core */ = {isa = PBXFileReference; lastKnownFileType = wrapper; path = Core; sourceTree = \"\"; };\n\t\tC81D181E2A1B509B006C1B70 /* Tool */ = {isa = PBXFileReference; lastKnownFileType = wrapper; path = Tool; sourceTree = \"\"; };\n\t\tC81E867D296FE4420026E908 /* Version.xcconfig */ = {isa = PBXFileReference; lastKnownFileType = text.xcconfig; path = Version.xcconfig; sourceTree = \"\"; };\n\t\tC8216B70298036EC00AD38C7 /* Helper */ = {isa = PBXFileReference; explicitFileType = \"compiled.mach-o.executable\"; includeInIndex = 0; path = Helper; sourceTree = BUILT_PRODUCTS_DIR; };\n\t\tC8216B72298036EC00AD38C7 /* main.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = main.swift; sourceTree = \"\"; };\n\t\tC8216B772980370100AD38C7 /* ReloadLaunchAgent.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ReloadLaunchAgent.swift; sourceTree = \"\"; };\n\t\tC828B27D2B1F241500E7612A /* ExtensionPoint.appextensionpoint */ = {isa = PBXFileReference; lastKnownFileType = text.xml; path = ExtensionPoint.appextensionpoint; sourceTree = \"\"; };\n\t\tC82E38492A1F025F00D4EADF /* LICENSE */ = {isa = PBXFileReference; lastKnownFileType = text; path = LICENSE; sourceTree = \"\"; };\n\t\tC84FD9D72CC671C600BE5093 /* ChatPlugins */ = {isa = PBXFileReference; lastKnownFileType = wrapper; path = ChatPlugins; sourceTree = \"\"; };\n\t\tC8520300293C4D9000460097 /* Helpers.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = Helpers.swift; sourceTree = \"\"; };\n\t\tC8520308293D805800460097 /* README.md */ = {isa = PBXFileReference; lastKnownFileType = net.daringfireball.markdown; path = README.md; sourceTree = \"\"; };\n\t\tC861A6A229E5503F005C41A3 /* PromptToCodeCommand.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = PromptToCodeCommand.swift; sourceTree = \"\"; };\n\t\tC861E60E2994F6070056CB02 /* CopilotForXcodeExtensionService.app */ = {isa = PBXFileReference; explicitFileType = wrapper.application; includeInIndex = 0; path = CopilotForXcodeExtensionService.app; sourceTree = BUILT_PRODUCTS_DIR; };\n\t\tC861E6102994F6070056CB02 /* AppDelegate.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = AppDelegate.swift; sourceTree = \"\"; };\n\t\tC861E6142994F6080056CB02 /* Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = Assets.xcassets; sourceTree = \"\"; };\n\t\tC861E6192994F6080056CB02 /* ExtensionService.entitlements */ = {isa = PBXFileReference; lastKnownFileType = text.plist.entitlements; path = ExtensionService.entitlements; sourceTree = \"\"; };\n\t\tC861E61F2994F6390056CB02 /* ServiceDelegate.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ServiceDelegate.swift; sourceTree = \"\"; };\n\t\tC8738B632BE4D4B900609E7F /* CommunicationBridge */ = {isa = PBXFileReference; explicitFileType = \"compiled.mach-o.executable\"; includeInIndex = 0; path = CommunicationBridge; sourceTree = BUILT_PRODUCTS_DIR; };\n\t\tC8738B652BE4D4B900609E7F /* main.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = main.swift; sourceTree = \"\"; };\n\t\tC8738B6A2BE4D56F00609E7F /* ServiceDelegate.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ServiceDelegate.swift; sourceTree = \"\"; };\n\t\tC8738B6D2BE4F3E800609E7F /* bridgeLaunchAgent.plist */ = {isa = PBXFileReference; lastKnownFileType = text.plist.xml; path = bridgeLaunchAgent.plist; sourceTree = \"\"; };\n\t\tC8738B702BE4F8B700609E7F /* XPCController.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = XPCController.swift; sourceTree = \"\"; };\n\t\tC8738B782BE5363800609E7F /* SandboxedClientTester.app */ = {isa = PBXFileReference; explicitFileType = wrapper.application; includeInIndex = 0; path = SandboxedClientTester.app; sourceTree = BUILT_PRODUCTS_DIR; };\n\t\tC8738B7A2BE5363800609E7F /* SandboxedClientTesterApp.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = SandboxedClientTesterApp.swift; sourceTree = \"\"; };\n\t\tC8738B7C2BE5363800609E7F /* ContentView.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ContentView.swift; sourceTree = \"\"; };\n\t\tC8738B7E2BE5363900609E7F /* Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = Assets.xcassets; sourceTree = \"\"; };\n\t\tC8738B812BE5363900609E7F /* Preview Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = \"Preview Assets.xcassets\"; sourceTree = \"\"; };\n\t\tC8738B832BE5363900609E7F /* SandboxedClientTester.entitlements */ = {isa = PBXFileReference; lastKnownFileType = text.plist.entitlements; path = SandboxedClientTester.entitlements; sourceTree = \"\"; };\n\t\tC8738B892BE5379E00609E7F /* Info.plist */ = {isa = PBXFileReference; lastKnownFileType = text.plist; path = Info.plist; sourceTree = \"\"; };\n\t\tC8758E6F29F04BFF00D29C1C /* CustomCommand.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = CustomCommand.swift; sourceTree = \"\"; };\n\t\tC8758E7129F04CF100D29C1C /* SeparatorCommand.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = SeparatorCommand.swift; sourceTree = \"\"; };\n\t\tC87B03A3293B24AB00C77EAE /* Copilot-for-Xcode-Info.plist */ = {isa = PBXFileReference; lastKnownFileType = text.plist; path = \"Copilot-for-Xcode-Info.plist\"; sourceTree = SOURCE_ROOT; };\n\t\tC87B03A4293B261200C77EAE /* AcceptSuggestionCommand.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = AcceptSuggestionCommand.swift; sourceTree = \"\"; };\n\t\tC87B03A6293B261900C77EAE /* RejectSuggestionCommand.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = RejectSuggestionCommand.swift; sourceTree = \"\"; };\n\t\tC87B03A8293B262600C77EAE /* NextSuggestionCommand.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = NextSuggestionCommand.swift; sourceTree = \"\"; };\n\t\tC87B03AA293B262E00C77EAE /* PreviousSuggestionCommand.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = PreviousSuggestionCommand.swift; sourceTree = \"\"; };\n\t\tC887BC832965D96000931567 /* DEVELOPMENT.md */ = {isa = PBXFileReference; lastKnownFileType = net.daringfireball.markdown; path = DEVELOPMENT.md; sourceTree = \"\"; };\n\t\tC89E75C22A46FB32000DD64F /* AppDelegate+Menu.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = \"AppDelegate+Menu.swift\"; sourceTree = \"\"; };\n\t\tC8BE64922EB9B42E00EDB2D7 /* OverlayWindow */ = {isa = PBXFileReference; lastKnownFileType = wrapper; path = OverlayWindow; sourceTree = \"\"; };\n\t\tC8CD828229B88006008D044D /* TestPlan.xctestplan */ = {isa = PBXFileReference; lastKnownFileType = text; path = TestPlan.xctestplan; sourceTree = \"\"; };\n\t\tC8DCEFFF29CE11D500FDDDD7 /* OpenChat.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = OpenChat.swift; sourceTree = \"\"; };\n\t\tC8DD9CB02BC673F80036641C /* CloseIdleTabsCommand.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = CloseIdleTabsCommand.swift; sourceTree = \"\"; };\n\t\tC8F103292A7A365000D28F4F /* launchAgent.plist */ = {isa = PBXFileReference; lastKnownFileType = text.plist.xml; path = launchAgent.plist; sourceTree = \"\"; };\n/* End PBXFileReference section */\n\n/* Begin PBXFrameworksBuildPhase section */\n\t\tC81458892939EFDC00135263 /* Frameworks */ = {\n\t\t\tisa = PBXFrameworksBuildPhase;\n\t\t\tbuildActionMask = 2147483647;\n\t\t\tfiles = (\n\t\t\t\tC87B03AC293B2CF300C77EAE /* XcodeKit.framework in Frameworks */,\n\t\t\t\tC814588F2939EFDC00135263 /* Cocoa.framework in Frameworks */,\n\t\t\t\tC882175C294187EF00A22FD3 /* Client in Frameworks */,\n\t\t\t);\n\t\t\trunOnlyForDeploymentPostprocessing = 0;\n\t\t};\n\t\tC8189B132938972F00C9DCDA /* Frameworks */ = {\n\t\t\tisa = PBXFrameworksBuildPhase;\n\t\t\tbuildActionMask = 2147483647;\n\t\t\tfiles = (\n\t\t\t\tC86612F82A06AF74009197D9 /* HostApp in Frameworks */,\n\t\t\t);\n\t\t\trunOnlyForDeploymentPostprocessing = 0;\n\t\t};\n\t\tC8216B6D298036EC00AD38C7 /* Frameworks */ = {\n\t\t\tisa = PBXFrameworksBuildPhase;\n\t\t\tbuildActionMask = 2147483647;\n\t\t\tfiles = (\n\t\t\t\tC8216B7D2980374300AD38C7 /* ArgumentParser in Frameworks */,\n\t\t\t);\n\t\t\trunOnlyForDeploymentPostprocessing = 0;\n\t\t};\n\t\tC861E60B2994F6070056CB02 /* Frameworks */ = {\n\t\t\tisa = PBXFrameworksBuildPhase;\n\t\t\tbuildActionMask = 2147483647;\n\t\t\tfiles = (\n\t\t\t\tC861E61E2994F6150056CB02 /* Service in Frameworks */,\n\t\t\t);\n\t\t\trunOnlyForDeploymentPostprocessing = 0;\n\t\t};\n\t\tC8738B602BE4D4B900609E7F /* Frameworks */ = {\n\t\t\tisa = PBXFrameworksBuildPhase;\n\t\t\tbuildActionMask = 2147483647;\n\t\t\tfiles = (\n\t\t\t\tC8738B6F2BE4F7A600609E7F /* XPCShared in Frameworks */,\n\t\t\t);\n\t\t\trunOnlyForDeploymentPostprocessing = 0;\n\t\t};\n\t\tC8738B752BE5363800609E7F /* Frameworks */ = {\n\t\t\tisa = PBXFrameworksBuildPhase;\n\t\t\tbuildActionMask = 2147483647;\n\t\t\tfiles = (\n\t\t\t\tC8738B882BE5365000609E7F /* Client in Frameworks */,\n\t\t\t);\n\t\t\trunOnlyForDeploymentPostprocessing = 0;\n\t\t};\n/* End PBXFrameworksBuildPhase section */\n\n/* Begin PBXGroup section */\n\t\tC814588D2939EFDC00135263 /* Frameworks */ = {\n\t\t\tisa = PBXGroup;\n\t\t\tchildren = (\n\t\t\t\tC814588E2939EFDC00135263 /* Cocoa.framework */,\n\t\t\t\tC81458902939EFDC00135263 /* XcodeKit.framework */,\n\t\t\t);\n\t\t\tname = Frameworks;\n\t\t\tsourceTree = \"\";\n\t\t};\n\t\tC81458922939EFDC00135263 /* EditorExtension */ = {\n\t\t\tisa = PBXGroup;\n\t\t\tchildren = (\n\t\t\t\tC81458932939EFDC00135263 /* SourceEditorExtension.swift */,\n\t\t\t\tC8758E7129F04CF100D29C1C /* SeparatorCommand.swift */,\n\t\t\t\tC8520300293C4D9000460097 /* Helpers.swift */,\n\t\t\t\tC81458952939EFDC00135263 /* GetSuggestionsCommand.swift */,\n\t\t\t\tC87B03A4293B261200C77EAE /* AcceptSuggestionCommand.swift */,\n\t\t\t\tC80FFB952A95F58200704A25 /* AcceptPromptToCodeCommand.swift */,\n\t\t\t\tC87B03A6293B261900C77EAE /* RejectSuggestionCommand.swift */,\n\t\t\t\tC87B03A8293B262600C77EAE /* NextSuggestionCommand.swift */,\n\t\t\t\tC87B03AA293B262E00C77EAE /* PreviousSuggestionCommand.swift */,\n\t\t\t\tC8009BFE2941C551007AA7E8 /* ToggleRealtimeSuggestionsCommand.swift */,\n\t\t\t\tC8009C022941C576007AA7E8 /* RealtimeSuggestionCommand.swift */,\n\t\t\t\tC800DBB0294C624D00B04CAC /* PrefetchSuggestionsCommand.swift */,\n\t\t\t\tC8758E6F29F04BFF00D29C1C /* CustomCommand.swift */,\n\t\t\t\tC8DCEFFF29CE11D500FDDDD7 /* OpenChat.swift */,\n\t\t\t\tC8DD9CB02BC673F80036641C /* CloseIdleTabsCommand.swift */,\n\t\t\t\tC861A6A229E5503F005C41A3 /* PromptToCodeCommand.swift */,\n\t\t\t\tC81458972939EFDC00135263 /* Info.plist */,\n\t\t\t\tC81458982939EFDC00135263 /* EditorExtension.entitlements */,\n\t\t\t);\n\t\t\tpath = EditorExtension;\n\t\t\tsourceTree = \"\";\n\t\t};\n\t\tC8189B0D2938972F00C9DCDA = {\n\t\t\tisa = PBXGroup;\n\t\t\tchildren = (\n\t\t\t\tC887BC832965D96000931567 /* DEVELOPMENT.md */,\n\t\t\t\tC8520308293D805800460097 /* README.md */,\n\t\t\t\tC82E38492A1F025F00D4EADF /* LICENSE */,\n\t\t\t\tC8F103292A7A365000D28F4F /* launchAgent.plist */,\n\t\t\t\tC8738B6D2BE4F3E800609E7F /* bridgeLaunchAgent.plist */,\n\t\t\t\tC81E867D296FE4420026E908 /* Version.xcconfig */,\n\t\t\t\tC81458AD293A009600135263 /* Config.xcconfig */,\n\t\t\t\tC81458AE293A009800135263 /* Config.debug.xcconfig */,\n\t\t\t\tC8CD828229B88006008D044D /* TestPlan.xctestplan */,\n\t\t\t\tC828B27D2B1F241500E7612A /* ExtensionPoint.appextensionpoint */,\n\t\t\t\tC8BE64922EB9B42E00EDB2D7 /* OverlayWindow */,\n\t\t\t\tC84FD9D72CC671C600BE5093 /* ChatPlugins */,\n\t\t\t\tC81D181E2A1B509B006C1B70 /* Tool */,\n\t\t\t\tC8189B282938979000C9DCDA /* Core */,\n\t\t\t\tC8189B182938972F00C9DCDA /* Copilot for Xcode */,\n\t\t\t\tC81458922939EFDC00135263 /* EditorExtension */,\n\t\t\t\tC8216B71298036EC00AD38C7 /* Helper */,\n\t\t\t\tC861E60F2994F6070056CB02 /* ExtensionService */,\n\t\t\t\tC8738B642BE4D4B900609E7F /* CommunicationBridge */,\n\t\t\t\tC8738B792BE5363800609E7F /* SandboxedClientTester */,\n\t\t\t\tC814588D2939EFDC00135263 /* Frameworks */,\n\t\t\t\tC8189B172938972F00C9DCDA /* Products */,\n\t\t\t);\n\t\t\tsourceTree = \"\";\n\t\t};\n\t\tC8189B172938972F00C9DCDA /* Products */ = {\n\t\t\tisa = PBXGroup;\n\t\t\tchildren = (\n\t\t\t\tC8189B162938972F00C9DCDA /* Copilot for Xcode Dev.app */,\n\t\t\t\tC814588C2939EFDC00135263 /* Copilot.appex */,\n\t\t\t\tC8216B70298036EC00AD38C7 /* Helper */,\n\t\t\t\tC861E60E2994F6070056CB02 /* CopilotForXcodeExtensionService.app */,\n\t\t\t\tC8738B632BE4D4B900609E7F /* CommunicationBridge */,\n\t\t\t\tC8738B782BE5363800609E7F /* SandboxedClientTester.app */,\n\t\t\t);\n\t\t\tname = Products;\n\t\t\tsourceTree = \"\";\n\t\t};\n\t\tC8189B182938972F00C9DCDA /* Copilot for Xcode */ = {\n\t\t\tisa = PBXGroup;\n\t\t\tchildren = (\n\t\t\t\tC87B03A3293B24AB00C77EAE /* Copilot-for-Xcode-Info.plist */,\n\t\t\t\tC8189B192938972F00C9DCDA /* App.swift */,\n\t\t\t\tC8189B1D2938973000C9DCDA /* Assets.xcassets */,\n\t\t\t\tC8189B222938973000C9DCDA /* Copilot_for_Xcode.entitlements */,\n\t\t\t\tC8189B1F2938973000C9DCDA /* Preview Content */,\n\t\t\t);\n\t\t\tpath = \"Copilot for Xcode\";\n\t\t\tsourceTree = \"\";\n\t\t};\n\t\tC8189B1F2938973000C9DCDA /* Preview Content */ = {\n\t\t\tisa = PBXGroup;\n\t\t\tchildren = (\n\t\t\t\tC8189B202938973000C9DCDA /* Preview Assets.xcassets */,\n\t\t\t);\n\t\t\tpath = \"Preview Content\";\n\t\t\tsourceTree = \"\";\n\t\t};\n\t\tC8216B71298036EC00AD38C7 /* Helper */ = {\n\t\t\tisa = PBXGroup;\n\t\t\tchildren = (\n\t\t\t\tC8216B72298036EC00AD38C7 /* main.swift */,\n\t\t\t\tC8216B772980370100AD38C7 /* ReloadLaunchAgent.swift */,\n\t\t\t);\n\t\t\tpath = Helper;\n\t\t\tsourceTree = \"\";\n\t\t};\n\t\tC861E60F2994F6070056CB02 /* ExtensionService */ = {\n\t\t\tisa = PBXGroup;\n\t\t\tchildren = (\n\t\t\t\tC81291D92994FE7900196E12 /* Info.plist */,\n\t\t\t\tC861E61F2994F6390056CB02 /* ServiceDelegate.swift */,\n\t\t\t\tC861E6102994F6070056CB02 /* AppDelegate.swift */,\n\t\t\t\tC89E75C22A46FB32000DD64F /* AppDelegate+Menu.swift */,\n\t\t\t\tC8738B702BE4F8B700609E7F /* XPCController.swift */,\n\t\t\t\tC81291D52994FE6900196E12 /* Main.storyboard */,\n\t\t\t\tC861E6142994F6080056CB02 /* Assets.xcassets */,\n\t\t\t\tC861E6192994F6080056CB02 /* ExtensionService.entitlements */,\n\t\t\t);\n\t\t\tpath = ExtensionService;\n\t\t\tsourceTree = \"\";\n\t\t};\n\t\tC8738B642BE4D4B900609E7F /* CommunicationBridge */ = {\n\t\t\tisa = PBXGroup;\n\t\t\tchildren = (\n\t\t\t\tC8738B652BE4D4B900609E7F /* main.swift */,\n\t\t\t\tC8738B6A2BE4D56F00609E7F /* ServiceDelegate.swift */,\n\t\t\t);\n\t\t\tpath = CommunicationBridge;\n\t\t\tsourceTree = \"\";\n\t\t};\n\t\tC8738B792BE5363800609E7F /* SandboxedClientTester */ = {\n\t\t\tisa = PBXGroup;\n\t\t\tchildren = (\n\t\t\t\tC8738B892BE5379E00609E7F /* Info.plist */,\n\t\t\t\tC8738B7A2BE5363800609E7F /* SandboxedClientTesterApp.swift */,\n\t\t\t\tC8738B7C2BE5363800609E7F /* ContentView.swift */,\n\t\t\t\tC8738B7E2BE5363900609E7F /* Assets.xcassets */,\n\t\t\t\tC8738B832BE5363900609E7F /* SandboxedClientTester.entitlements */,\n\t\t\t\tC8738B802BE5363900609E7F /* Preview Content */,\n\t\t\t);\n\t\t\tpath = SandboxedClientTester;\n\t\t\tsourceTree = \"\";\n\t\t};\n\t\tC8738B802BE5363900609E7F /* Preview Content */ = {\n\t\t\tisa = PBXGroup;\n\t\t\tchildren = (\n\t\t\t\tC8738B812BE5363900609E7F /* Preview Assets.xcassets */,\n\t\t\t);\n\t\t\tpath = \"Preview Content\";\n\t\t\tsourceTree = \"\";\n\t\t};\n/* End PBXGroup section */\n\n/* Begin PBXNativeTarget section */\n\t\tC814588B2939EFDC00135263 /* EditorExtension */ = {\n\t\t\tisa = PBXNativeTarget;\n\t\t\tbuildConfigurationList = C814589C2939EFDC00135263 /* Build configuration list for PBXNativeTarget \"EditorExtension\" */;\n\t\t\tbuildPhases = (\n\t\t\t\tC81458882939EFDC00135263 /* Sources */,\n\t\t\t\tC81458892939EFDC00135263 /* Frameworks */,\n\t\t\t\tC814588A2939EFDC00135263 /* Resources */,\n\t\t\t\tC87B03AE293B2CF300C77EAE /* Embed Frameworks */,\n\t\t\t);\n\t\t\tbuildRules = (\n\t\t\t);\n\t\t\tdependencies = (\n\t\t\t);\n\t\t\tname = EditorExtension;\n\t\t\tpackageProductDependencies = (\n\t\t\t\tC882175B294187EF00A22FD3 /* Client */,\n\t\t\t);\n\t\t\tproductName = EditorExtension;\n\t\t\tproductReference = C814588C2939EFDC00135263 /* Copilot.appex */;\n\t\t\tproductType = \"com.apple.product-type.xcode-extension\";\n\t\t};\n\t\tC8189B152938972F00C9DCDA /* Copilot for Xcode */ = {\n\t\t\tisa = PBXNativeTarget;\n\t\t\tbuildConfigurationList = C8189B252938973000C9DCDA /* Build configuration list for PBXNativeTarget \"Copilot for Xcode\" */;\n\t\t\tbuildPhases = (\n\t\t\t\tC8189B122938972F00C9DCDA /* Sources */,\n\t\t\t\tC8189B132938972F00C9DCDA /* Frameworks */,\n\t\t\t\tC8189B142938972F00C9DCDA /* Resources */,\n\t\t\t\tC814589F2939EFDC00135263 /* Embed Foundation Extensions */,\n\t\t\t\tC8520306293CF0EF00460097 /* Embed XPCService */,\n\t\t\t\tC8C8B60829AFA32800034BEE /* Embed Service */,\n\t\t\t\tC8F1032A2A7A38D200D28F4F /* Copy Launch Agent */,\n\t\t\t);\n\t\t\tbuildRules = (\n\t\t\t);\n\t\t\tdependencies = (\n\t\t\t\tC8738B8D2BE540F900609E7F /* PBXTargetDependency */,\n\t\t\t\tC81291B02994F92700196E12 /* PBXTargetDependency */,\n\t\t\t\tC8216B7F2980377E00AD38C7 /* PBXTargetDependency */,\n\t\t\t\tC814589A2939EFDC00135263 /* PBXTargetDependency */,\n\t\t\t);\n\t\t\tname = \"Copilot for Xcode\";\n\t\t\tpackageProductDependencies = (\n\t\t\t\tC86612F72A06AF74009197D9 /* HostApp */,\n\t\t\t);\n\t\t\tproductName = \"Copilot for Xcode\";\n\t\t\tproductReference = C8189B162938972F00C9DCDA /* Copilot for Xcode Dev.app */;\n\t\t\tproductType = \"com.apple.product-type.application\";\n\t\t};\n\t\tC8216B6F298036EC00AD38C7 /* Helper */ = {\n\t\t\tisa = PBXNativeTarget;\n\t\t\tbuildConfigurationList = C8216B74298036EC00AD38C7 /* Build configuration list for PBXNativeTarget \"Helper\" */;\n\t\t\tbuildPhases = (\n\t\t\t\tC8216B6C298036EC00AD38C7 /* Sources */,\n\t\t\t\tC8216B6D298036EC00AD38C7 /* Frameworks */,\n\t\t\t\tC8216B6E298036EC00AD38C7 /* CopyFiles */,\n\t\t\t);\n\t\t\tbuildRules = (\n\t\t\t);\n\t\t\tdependencies = (\n\t\t\t);\n\t\t\tname = Helper;\n\t\t\tpackageProductDependencies = (\n\t\t\t\tC8216B7C2980374300AD38C7 /* ArgumentParser */,\n\t\t\t);\n\t\t\tproductName = Helper;\n\t\t\tproductReference = C8216B70298036EC00AD38C7 /* Helper */;\n\t\t\tproductType = \"com.apple.product-type.tool\";\n\t\t};\n\t\tC861E60D2994F6070056CB02 /* ExtensionService */ = {\n\t\t\tisa = PBXNativeTarget;\n\t\t\tbuildConfigurationList = C861E61A2994F6080056CB02 /* Build configuration list for PBXNativeTarget \"ExtensionService\" */;\n\t\t\tbuildPhases = (\n\t\t\t\tC861E60A2994F6070056CB02 /* Sources */,\n\t\t\t\tC861E60B2994F6070056CB02 /* Frameworks */,\n\t\t\t\tC861E60C2994F6070056CB02 /* Resources */,\n\t\t\t\tC828B27E2B1F7B3C00E7612A /* Copy Extension Point */,\n\t\t\t);\n\t\t\tbuildRules = (\n\t\t\t);\n\t\t\tdependencies = (\n\t\t\t);\n\t\t\tname = ExtensionService;\n\t\t\tpackageProductDependencies = (\n\t\t\t\tC861E61D2994F6150056CB02 /* Service */,\n\t\t\t);\n\t\t\tproductName = ExtensionService;\n\t\t\tproductReference = C861E60E2994F6070056CB02 /* CopilotForXcodeExtensionService.app */;\n\t\t\tproductType = \"com.apple.product-type.application\";\n\t\t};\n\t\tC8738B622BE4D4B900609E7F /* CommunicationBridge */ = {\n\t\t\tisa = PBXNativeTarget;\n\t\t\tbuildConfigurationList = C8738B672BE4D4B900609E7F /* Build configuration list for PBXNativeTarget \"CommunicationBridge\" */;\n\t\t\tbuildPhases = (\n\t\t\t\tC8738B5F2BE4D4B900609E7F /* Sources */,\n\t\t\t\tC8738B602BE4D4B900609E7F /* Frameworks */,\n\t\t\t\tC8738B612BE4D4B900609E7F /* CopyFiles */,\n\t\t\t);\n\t\t\tbuildRules = (\n\t\t\t);\n\t\t\tdependencies = (\n\t\t\t);\n\t\t\tname = CommunicationBridge;\n\t\t\tpackageProductDependencies = (\n\t\t\t\tC8738B6E2BE4F7A600609E7F /* XPCShared */,\n\t\t\t);\n\t\t\tproductName = CommunicationBridge;\n\t\t\tproductReference = C8738B632BE4D4B900609E7F /* CommunicationBridge */;\n\t\t\tproductType = \"com.apple.product-type.tool\";\n\t\t};\n\t\tC8738B772BE5363800609E7F /* SandboxedClientTester */ = {\n\t\t\tisa = PBXNativeTarget;\n\t\t\tbuildConfigurationList = C8738B842BE5363900609E7F /* Build configuration list for PBXNativeTarget \"SandboxedClientTester\" */;\n\t\t\tbuildPhases = (\n\t\t\t\tC8738B742BE5363800609E7F /* Sources */,\n\t\t\t\tC8738B752BE5363800609E7F /* Frameworks */,\n\t\t\t\tC8738B762BE5363800609E7F /* Resources */,\n\t\t\t);\n\t\t\tbuildRules = (\n\t\t\t);\n\t\t\tdependencies = (\n\t\t\t);\n\t\t\tname = SandboxedClientTester;\n\t\t\tpackageProductDependencies = (\n\t\t\t\tC8738B872BE5365000609E7F /* Client */,\n\t\t\t);\n\t\t\tproductName = SandboxedClientTester;\n\t\t\tproductReference = C8738B782BE5363800609E7F /* SandboxedClientTester.app */;\n\t\t\tproductType = \"com.apple.product-type.application\";\n\t\t};\n/* End PBXNativeTarget section */\n\n/* Begin PBXProject section */\n\t\tC8189B0E2938972F00C9DCDA /* Project object */ = {\n\t\t\tisa = PBXProject;\n\t\t\tattributes = {\n\t\t\t\tBuildIndependentTargetsInParallel = 1;\n\t\t\t\tLastSwiftUpdateCheck = 1520;\n\t\t\t\tLastUpgradeCheck = 1410;\n\t\t\t\tTargetAttributes = {\n\t\t\t\t\tC814588B2939EFDC00135263 = {\n\t\t\t\t\t\tCreatedOnToolsVersion = 14.1;\n\t\t\t\t\t};\n\t\t\t\t\tC8189B152938972F00C9DCDA = {\n\t\t\t\t\t\tCreatedOnToolsVersion = 14.1;\n\t\t\t\t\t};\n\t\t\t\t\tC8216B6F298036EC00AD38C7 = {\n\t\t\t\t\t\tCreatedOnToolsVersion = 14.1;\n\t\t\t\t\t};\n\t\t\t\t\tC861E60D2994F6070056CB02 = {\n\t\t\t\t\t\tCreatedOnToolsVersion = 14.2;\n\t\t\t\t\t};\n\t\t\t\t\tC8738B622BE4D4B900609E7F = {\n\t\t\t\t\t\tCreatedOnToolsVersion = 15.2;\n\t\t\t\t\t};\n\t\t\t\t\tC8738B772BE5363800609E7F = {\n\t\t\t\t\t\tCreatedOnToolsVersion = 15.2;\n\t\t\t\t\t};\n\t\t\t\t};\n\t\t\t};\n\t\t\tbuildConfigurationList = C8189B112938972F00C9DCDA /* Build configuration list for PBXProject \"Copilot for Xcode\" */;\n\t\t\tcompatibilityVersion = \"Xcode 14.0\";\n\t\t\tdevelopmentRegion = en;\n\t\t\thasScannedForEncodings = 0;\n\t\t\tknownRegions = (\n\t\t\t\ten,\n\t\t\t\tBase,\n\t\t\t);\n\t\t\tmainGroup = C8189B0D2938972F00C9DCDA;\n\t\t\tpackageReferences = (\n\t\t\t\tC8216B792980373800AD38C7 /* XCRemoteSwiftPackageReference \"swift-argument-parser\" */,\n\t\t\t\tC80C91742A588DD800B5EADA /* XCRemoteSwiftPackageReference \"usearch\" */,\n\t\t\t);\n\t\t\tproductRefGroup = C8189B172938972F00C9DCDA /* Products */;\n\t\t\tprojectDirPath = \"\";\n\t\t\tprojectRoot = \"\";\n\t\t\ttargets = (\n\t\t\t\tC8189B152938972F00C9DCDA /* Copilot for Xcode */,\n\t\t\t\tC814588B2939EFDC00135263 /* EditorExtension */,\n\t\t\t\tC8216B6F298036EC00AD38C7 /* Helper */,\n\t\t\t\tC861E60D2994F6070056CB02 /* ExtensionService */,\n\t\t\t\tC8738B622BE4D4B900609E7F /* CommunicationBridge */,\n\t\t\t\tC8738B772BE5363800609E7F /* SandboxedClientTester */,\n\t\t\t);\n\t\t};\n/* End PBXProject section */\n\n/* Begin PBXResourcesBuildPhase section */\n\t\tC814588A2939EFDC00135263 /* Resources */ = {\n\t\t\tisa = PBXResourcesBuildPhase;\n\t\t\tbuildActionMask = 2147483647;\n\t\t\tfiles = (\n\t\t\t);\n\t\t\trunOnlyForDeploymentPostprocessing = 0;\n\t\t};\n\t\tC8189B142938972F00C9DCDA /* Resources */ = {\n\t\t\tisa = PBXResourcesBuildPhase;\n\t\t\tbuildActionMask = 2147483647;\n\t\t\tfiles = (\n\t\t\t\tC8189B212938973000C9DCDA /* Preview Assets.xcassets in Resources */,\n\t\t\t\tC8189B1E2938973000C9DCDA /* Assets.xcassets in Resources */,\n\t\t\t);\n\t\t\trunOnlyForDeploymentPostprocessing = 0;\n\t\t};\n\t\tC861E60C2994F6070056CB02 /* Resources */ = {\n\t\t\tisa = PBXResourcesBuildPhase;\n\t\t\tbuildActionMask = 2147483647;\n\t\t\tfiles = (\n\t\t\t\tC861E6152994F6080056CB02 /* Assets.xcassets in Resources */,\n\t\t\t\tC81291D72994FE6900196E12 /* Main.storyboard in Resources */,\n\t\t\t);\n\t\t\trunOnlyForDeploymentPostprocessing = 0;\n\t\t};\n\t\tC8738B762BE5363800609E7F /* Resources */ = {\n\t\t\tisa = PBXResourcesBuildPhase;\n\t\t\tbuildActionMask = 2147483647;\n\t\t\tfiles = (\n\t\t\t\tC8738B822BE5363900609E7F /* Preview Assets.xcassets in Resources */,\n\t\t\t\tC8738B7F2BE5363900609E7F /* Assets.xcassets in Resources */,\n\t\t\t);\n\t\t\trunOnlyForDeploymentPostprocessing = 0;\n\t\t};\n/* End PBXResourcesBuildPhase section */\n\n/* Begin PBXSourcesBuildPhase section */\n\t\tC81458882939EFDC00135263 /* Sources */ = {\n\t\t\tisa = PBXSourcesBuildPhase;\n\t\t\tbuildActionMask = 2147483647;\n\t\t\tfiles = (\n\t\t\t\tC8DCF00029CE11D500FDDDD7 /* OpenChat.swift in Sources */,\n\t\t\t\tC81458942939EFDC00135263 /* SourceEditorExtension.swift in Sources */,\n\t\t\t\tC8DD9CB12BC673F80036641C /* CloseIdleTabsCommand.swift in Sources */,\n\t\t\t\tC8758E7029F04BFF00D29C1C /* CustomCommand.swift in Sources */,\n\t\t\t\tC8758E7229F04CF100D29C1C /* SeparatorCommand.swift in Sources */,\n\t\t\t\tC861A6A329E5503F005C41A3 /* PromptToCodeCommand.swift in Sources */,\n\t\t\t\tC8520301293C4D9000460097 /* Helpers.swift in Sources */,\n\t\t\t\tC8009BFF2941C551007AA7E8 /* ToggleRealtimeSuggestionsCommand.swift in Sources */,\n\t\t\t\tC80FFB962A95F58200704A25 /* AcceptPromptToCodeCommand.swift in Sources */,\n\t\t\t\tC87B03A5293B261200C77EAE /* AcceptSuggestionCommand.swift in Sources */,\n\t\t\t\tC87B03A9293B262600C77EAE /* NextSuggestionCommand.swift in Sources */,\n\t\t\t\tC87B03AB293B262E00C77EAE /* PreviousSuggestionCommand.swift in Sources */,\n\t\t\t\tC87B03A7293B261900C77EAE /* RejectSuggestionCommand.swift in Sources */,\n\t\t\t\tC8009C032941C576007AA7E8 /* RealtimeSuggestionCommand.swift in Sources */,\n\t\t\t\tC800DBB1294C624D00B04CAC /* PrefetchSuggestionsCommand.swift in Sources */,\n\t\t\t\tC81458962939EFDC00135263 /* GetSuggestionsCommand.swift in Sources */,\n\t\t\t);\n\t\t\trunOnlyForDeploymentPostprocessing = 0;\n\t\t};\n\t\tC8189B122938972F00C9DCDA /* Sources */ = {\n\t\t\tisa = PBXSourcesBuildPhase;\n\t\t\tbuildActionMask = 2147483647;\n\t\t\tfiles = (\n\t\t\t\tC8189B1A2938972F00C9DCDA /* App.swift in Sources */,\n\t\t\t);\n\t\t\trunOnlyForDeploymentPostprocessing = 0;\n\t\t};\n\t\tC8216B6C298036EC00AD38C7 /* Sources */ = {\n\t\t\tisa = PBXSourcesBuildPhase;\n\t\t\tbuildActionMask = 2147483647;\n\t\t\tfiles = (\n\t\t\t\tC8216B73298036EC00AD38C7 /* main.swift in Sources */,\n\t\t\t\tC8216B782980370100AD38C7 /* ReloadLaunchAgent.swift in Sources */,\n\t\t\t);\n\t\t\trunOnlyForDeploymentPostprocessing = 0;\n\t\t};\n\t\tC861E60A2994F6070056CB02 /* Sources */ = {\n\t\t\tisa = PBXSourcesBuildPhase;\n\t\t\tbuildActionMask = 2147483647;\n\t\t\tfiles = (\n\t\t\t\tC89E75C32A46FB32000DD64F /* AppDelegate+Menu.swift in Sources */,\n\t\t\t\tC8738B712BE4F8B700609E7F /* XPCController.swift in Sources */,\n\t\t\t\tC861E6202994F63A0056CB02 /* ServiceDelegate.swift in Sources */,\n\t\t\t\tC861E6112994F6070056CB02 /* AppDelegate.swift in Sources */,\n\t\t\t);\n\t\t\trunOnlyForDeploymentPostprocessing = 0;\n\t\t};\n\t\tC8738B5F2BE4D4B900609E7F /* Sources */ = {\n\t\t\tisa = PBXSourcesBuildPhase;\n\t\t\tbuildActionMask = 2147483647;\n\t\t\tfiles = (\n\t\t\t\tC8738B6B2BE4D56F00609E7F /* ServiceDelegate.swift in Sources */,\n\t\t\t\tC8738B662BE4D4B900609E7F /* main.swift in Sources */,\n\t\t\t);\n\t\t\trunOnlyForDeploymentPostprocessing = 0;\n\t\t};\n\t\tC8738B742BE5363800609E7F /* Sources */ = {\n\t\t\tisa = PBXSourcesBuildPhase;\n\t\t\tbuildActionMask = 2147483647;\n\t\t\tfiles = (\n\t\t\t\tC8738B7D2BE5363800609E7F /* ContentView.swift in Sources */,\n\t\t\t\tC8738B7B2BE5363800609E7F /* SandboxedClientTesterApp.swift in Sources */,\n\t\t\t);\n\t\t\trunOnlyForDeploymentPostprocessing = 0;\n\t\t};\n/* End PBXSourcesBuildPhase section */\n\n/* Begin PBXTargetDependency section */\n\t\tC81291B02994F92700196E12 /* PBXTargetDependency */ = {\n\t\t\tisa = PBXTargetDependency;\n\t\t\ttarget = C861E60D2994F6070056CB02 /* ExtensionService */;\n\t\t\ttargetProxy = C81291AF2994F92700196E12 /* PBXContainerItemProxy */;\n\t\t};\n\t\tC814589A2939EFDC00135263 /* PBXTargetDependency */ = {\n\t\t\tisa = PBXTargetDependency;\n\t\t\ttarget = C814588B2939EFDC00135263 /* EditorExtension */;\n\t\t\ttargetProxy = C81458992939EFDC00135263 /* PBXContainerItemProxy */;\n\t\t};\n\t\tC8216B7F2980377E00AD38C7 /* PBXTargetDependency */ = {\n\t\t\tisa = PBXTargetDependency;\n\t\t\ttarget = C8216B6F298036EC00AD38C7 /* Helper */;\n\t\t\ttargetProxy = C8216B7E2980377E00AD38C7 /* PBXContainerItemProxy */;\n\t\t};\n\t\tC8738B8D2BE540F900609E7F /* PBXTargetDependency */ = {\n\t\t\tisa = PBXTargetDependency;\n\t\t\ttarget = C8738B622BE4D4B900609E7F /* CommunicationBridge */;\n\t\t\ttargetProxy = C8738B8C2BE540F900609E7F /* PBXContainerItemProxy */;\n\t\t};\n/* End PBXTargetDependency section */\n\n/* Begin XCBuildConfiguration section */\n\t\tC814589D2939EFDC00135263 /* Debug */ = {\n\t\t\tisa = XCBuildConfiguration;\n\t\t\tbuildSettings = {\n\t\t\t\tCODE_SIGN_ENTITLEMENTS = EditorExtension/EditorExtension.entitlements;\n\t\t\t\tCODE_SIGN_STYLE = Automatic;\n\t\t\t\tCOMBINE_HIDPI_IMAGES = YES;\n\t\t\t\tCURRENT_PROJECT_VERSION = \"$(APP_BUILD)\";\n\t\t\t\tDEAD_CODE_STRIPPING = YES;\n\t\t\t\tDEVELOPMENT_TEAM = 5YKZ4Y3DAW;\n\t\t\t\tENABLE_HARDENED_RUNTIME = YES;\n\t\t\t\tINFOPLIST_FILE = EditorExtension/Info.plist;\n\t\t\t\tINFOPLIST_KEY_CFBundleDisplayName = \"$(EXTENSION_BUNDLE_NAME)\";\n\t\t\t\tINFOPLIST_KEY_NSHumanReadableCopyright = \"\";\n\t\t\t\tLD_RUNPATH_SEARCH_PATHS = (\n\t\t\t\t\t\"$(inherited)\",\n\t\t\t\t\t\"@executable_path/../Frameworks\",\n\t\t\t\t\t\"@executable_path/../../../../Frameworks\",\n\t\t\t\t);\n\t\t\t\tMACOSX_DEPLOYMENT_TARGET = 13.0;\n\t\t\t\tMARKETING_VERSION = \"$(APP_VERSION)\";\n\t\t\t\tPRODUCT_BUNDLE_IDENTIFIER = \"$(BUNDLE_IDENTIFIER_BASE).EditorExtension\";\n\t\t\t\tPRODUCT_NAME = Copilot;\n\t\t\t\tSKIP_INSTALL = YES;\n\t\t\t\tSWIFT_EMIT_LOC_STRINGS = YES;\n\t\t\t\tSWIFT_VERSION = 5.0;\n\t\t\t};\n\t\t\tname = Debug;\n\t\t};\n\t\tC814589E2939EFDC00135263 /* Release */ = {\n\t\t\tisa = XCBuildConfiguration;\n\t\t\tbuildSettings = {\n\t\t\t\tCODE_SIGN_ENTITLEMENTS = EditorExtension/EditorExtension.entitlements;\n\t\t\t\tCODE_SIGN_STYLE = Automatic;\n\t\t\t\tCOMBINE_HIDPI_IMAGES = YES;\n\t\t\t\tCURRENT_PROJECT_VERSION = \"$(APP_BUILD)\";\n\t\t\t\tDEAD_CODE_STRIPPING = YES;\n\t\t\t\tDEVELOPMENT_TEAM = 5YKZ4Y3DAW;\n\t\t\t\tENABLE_HARDENED_RUNTIME = YES;\n\t\t\t\tINFOPLIST_FILE = EditorExtension/Info.plist;\n\t\t\t\tINFOPLIST_KEY_CFBundleDisplayName = \"$(EXTENSION_BUNDLE_NAME)\";\n\t\t\t\tINFOPLIST_KEY_NSHumanReadableCopyright = \"\";\n\t\t\t\tLD_RUNPATH_SEARCH_PATHS = (\n\t\t\t\t\t\"$(inherited)\",\n\t\t\t\t\t\"@executable_path/../Frameworks\",\n\t\t\t\t\t\"@executable_path/../../../../Frameworks\",\n\t\t\t\t);\n\t\t\t\tMACOSX_DEPLOYMENT_TARGET = 13.0;\n\t\t\t\tMARKETING_VERSION = \"$(APP_VERSION)\";\n\t\t\t\tPRODUCT_BUNDLE_IDENTIFIER = \"$(BUNDLE_IDENTIFIER_BASE).EditorExtension\";\n\t\t\t\tPRODUCT_NAME = Copilot;\n\t\t\t\tSKIP_INSTALL = YES;\n\t\t\t\tSWIFT_EMIT_LOC_STRINGS = YES;\n\t\t\t\tSWIFT_VERSION = 5.0;\n\t\t\t};\n\t\t\tname = Release;\n\t\t};\n\t\tC8189B232938973000C9DCDA /* Debug */ = {\n\t\t\tisa = XCBuildConfiguration;\n\t\t\tbaseConfigurationReference = C81458AE293A009800135263 /* Config.debug.xcconfig */;\n\t\t\tbuildSettings = {\n\t\t\t\tALWAYS_SEARCH_USER_PATHS = NO;\n\t\t\t\tCLANG_ANALYZER_NONNULL = YES;\n\t\t\t\tCLANG_ANALYZER_NUMBER_OBJECT_CONVERSION = YES_AGGRESSIVE;\n\t\t\t\tCLANG_CXX_LANGUAGE_STANDARD = \"gnu++20\";\n\t\t\t\tCLANG_ENABLE_MODULES = YES;\n\t\t\t\tCLANG_ENABLE_OBJC_ARC = YES;\n\t\t\t\tCLANG_ENABLE_OBJC_WEAK = YES;\n\t\t\t\tCLANG_WARN_BLOCK_CAPTURE_AUTORELEASING = YES;\n\t\t\t\tCLANG_WARN_BOOL_CONVERSION = YES;\n\t\t\t\tCLANG_WARN_COMMA = YES;\n\t\t\t\tCLANG_WARN_CONSTANT_CONVERSION = YES;\n\t\t\t\tCLANG_WARN_DEPRECATED_OBJC_IMPLEMENTATIONS = YES;\n\t\t\t\tCLANG_WARN_DIRECT_OBJC_ISA_USAGE = YES_ERROR;\n\t\t\t\tCLANG_WARN_DOCUMENTATION_COMMENTS = YES;\n\t\t\t\tCLANG_WARN_EMPTY_BODY = YES;\n\t\t\t\tCLANG_WARN_ENUM_CONVERSION = YES;\n\t\t\t\tCLANG_WARN_INFINITE_RECURSION = YES;\n\t\t\t\tCLANG_WARN_INT_CONVERSION = YES;\n\t\t\t\tCLANG_WARN_NON_LITERAL_NULL_CONVERSION = YES;\n\t\t\t\tCLANG_WARN_OBJC_IMPLICIT_RETAIN_SELF = YES;\n\t\t\t\tCLANG_WARN_OBJC_LITERAL_CONVERSION = YES;\n\t\t\t\tCLANG_WARN_OBJC_ROOT_CLASS = YES_ERROR;\n\t\t\t\tCLANG_WARN_QUOTED_INCLUDE_IN_FRAMEWORK_HEADER = YES;\n\t\t\t\tCLANG_WARN_RANGE_LOOP_ANALYSIS = YES;\n\t\t\t\tCLANG_WARN_STRICT_PROTOTYPES = YES;\n\t\t\t\tCLANG_WARN_SUSPICIOUS_MOVE = YES;\n\t\t\t\tCLANG_WARN_UNGUARDED_AVAILABILITY = YES_AGGRESSIVE;\n\t\t\t\tCLANG_WARN_UNREACHABLE_CODE = YES;\n\t\t\t\tCLANG_WARN__DUPLICATE_METHOD_MATCH = YES;\n\t\t\t\tCOPY_PHASE_STRIP = NO;\n\t\t\t\tDEAD_CODE_STRIPPING = YES;\n\t\t\t\tDEBUG_INFORMATION_FORMAT = dwarf;\n\t\t\t\tENABLE_STRICT_OBJC_MSGSEND = YES;\n\t\t\t\tENABLE_TESTABILITY = YES;\n\t\t\t\tENABLE_USER_SCRIPT_SANDBOXING = YES;\n\t\t\t\tGCC_C_LANGUAGE_STANDARD = gnu11;\n\t\t\t\tGCC_DYNAMIC_NO_PIC = NO;\n\t\t\t\tGCC_NO_COMMON_BLOCKS = YES;\n\t\t\t\tGCC_OPTIMIZATION_LEVEL = 0;\n\t\t\t\tGCC_PREPROCESSOR_DEFINITIONS = (\n\t\t\t\t\t\"DEBUG=1\",\n\t\t\t\t\t\"$(inherited)\",\n\t\t\t\t);\n\t\t\t\tGCC_WARN_64_TO_32_BIT_CONVERSION = YES;\n\t\t\t\tGCC_WARN_ABOUT_RETURN_TYPE = YES_ERROR;\n\t\t\t\tGCC_WARN_UNDECLARED_SELECTOR = YES;\n\t\t\t\tGCC_WARN_UNINITIALIZED_AUTOS = YES_AGGRESSIVE;\n\t\t\t\tGCC_WARN_UNUSED_FUNCTION = YES;\n\t\t\t\tGCC_WARN_UNUSED_VARIABLE = YES;\n\t\t\t\tMACOSX_DEPLOYMENT_TARGET = 12.0;\n\t\t\t\tMTL_ENABLE_DEBUG_INFO = INCLUDE_SOURCE;\n\t\t\t\tMTL_FAST_MATH = YES;\n\t\t\t\tONLY_ACTIVE_ARCH = YES;\n\t\t\t\tSDKROOT = macosx;\n\t\t\t\tSWIFT_ACTIVE_COMPILATION_CONDITIONS = DEBUG;\n\t\t\t\tSWIFT_OPTIMIZATION_LEVEL = \"-Onone\";\n\t\t\t};\n\t\t\tname = Debug;\n\t\t};\n\t\tC8189B242938973000C9DCDA /* Release */ = {\n\t\t\tisa = XCBuildConfiguration;\n\t\t\tbaseConfigurationReference = C81458AD293A009600135263 /* Config.xcconfig */;\n\t\t\tbuildSettings = {\n\t\t\t\tALWAYS_SEARCH_USER_PATHS = NO;\n\t\t\t\tCLANG_ANALYZER_NONNULL = YES;\n\t\t\t\tCLANG_ANALYZER_NUMBER_OBJECT_CONVERSION = YES_AGGRESSIVE;\n\t\t\t\tCLANG_CXX_LANGUAGE_STANDARD = \"gnu++20\";\n\t\t\t\tCLANG_ENABLE_MODULES = YES;\n\t\t\t\tCLANG_ENABLE_OBJC_ARC = YES;\n\t\t\t\tCLANG_ENABLE_OBJC_WEAK = YES;\n\t\t\t\tCLANG_WARN_BLOCK_CAPTURE_AUTORELEASING = YES;\n\t\t\t\tCLANG_WARN_BOOL_CONVERSION = YES;\n\t\t\t\tCLANG_WARN_COMMA = YES;\n\t\t\t\tCLANG_WARN_CONSTANT_CONVERSION = YES;\n\t\t\t\tCLANG_WARN_DEPRECATED_OBJC_IMPLEMENTATIONS = YES;\n\t\t\t\tCLANG_WARN_DIRECT_OBJC_ISA_USAGE = YES_ERROR;\n\t\t\t\tCLANG_WARN_DOCUMENTATION_COMMENTS = YES;\n\t\t\t\tCLANG_WARN_EMPTY_BODY = YES;\n\t\t\t\tCLANG_WARN_ENUM_CONVERSION = YES;\n\t\t\t\tCLANG_WARN_INFINITE_RECURSION = YES;\n\t\t\t\tCLANG_WARN_INT_CONVERSION = YES;\n\t\t\t\tCLANG_WARN_NON_LITERAL_NULL_CONVERSION = YES;\n\t\t\t\tCLANG_WARN_OBJC_IMPLICIT_RETAIN_SELF = YES;\n\t\t\t\tCLANG_WARN_OBJC_LITERAL_CONVERSION = YES;\n\t\t\t\tCLANG_WARN_OBJC_ROOT_CLASS = YES_ERROR;\n\t\t\t\tCLANG_WARN_QUOTED_INCLUDE_IN_FRAMEWORK_HEADER = YES;\n\t\t\t\tCLANG_WARN_RANGE_LOOP_ANALYSIS = YES;\n\t\t\t\tCLANG_WARN_STRICT_PROTOTYPES = YES;\n\t\t\t\tCLANG_WARN_SUSPICIOUS_MOVE = YES;\n\t\t\t\tCLANG_WARN_UNGUARDED_AVAILABILITY = YES_AGGRESSIVE;\n\t\t\t\tCLANG_WARN_UNREACHABLE_CODE = YES;\n\t\t\t\tCLANG_WARN__DUPLICATE_METHOD_MATCH = YES;\n\t\t\t\tCOPY_PHASE_STRIP = NO;\n\t\t\t\tDEAD_CODE_STRIPPING = YES;\n\t\t\t\tDEBUG_INFORMATION_FORMAT = \"dwarf-with-dsym\";\n\t\t\t\tENABLE_NS_ASSERTIONS = NO;\n\t\t\t\tENABLE_STRICT_OBJC_MSGSEND = YES;\n\t\t\t\tENABLE_USER_SCRIPT_SANDBOXING = YES;\n\t\t\t\tGCC_C_LANGUAGE_STANDARD = gnu11;\n\t\t\t\tGCC_NO_COMMON_BLOCKS = YES;\n\t\t\t\tGCC_WARN_64_TO_32_BIT_CONVERSION = YES;\n\t\t\t\tGCC_WARN_ABOUT_RETURN_TYPE = YES_ERROR;\n\t\t\t\tGCC_WARN_UNDECLARED_SELECTOR = YES;\n\t\t\t\tGCC_WARN_UNINITIALIZED_AUTOS = YES_AGGRESSIVE;\n\t\t\t\tGCC_WARN_UNUSED_FUNCTION = YES;\n\t\t\t\tGCC_WARN_UNUSED_VARIABLE = YES;\n\t\t\t\tMACOSX_DEPLOYMENT_TARGET = 12.0;\n\t\t\t\tMTL_ENABLE_DEBUG_INFO = NO;\n\t\t\t\tMTL_FAST_MATH = YES;\n\t\t\t\tSDKROOT = macosx;\n\t\t\t\tSWIFT_COMPILATION_MODE = wholemodule;\n\t\t\t\tSWIFT_OPTIMIZATION_LEVEL = \"-O\";\n\t\t\t};\n\t\t\tname = Release;\n\t\t};\n\t\tC8189B262938973000C9DCDA /* Debug */ = {\n\t\t\tisa = XCBuildConfiguration;\n\t\t\tbuildSettings = {\n\t\t\t\tALWAYS_EMBED_SWIFT_STANDARD_LIBRARIES = YES;\n\t\t\t\tASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;\n\t\t\t\tASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;\n\t\t\t\tCODE_SIGN_ENTITLEMENTS = \"Copilot for Xcode/Copilot_for_Xcode.entitlements\";\n\t\t\t\tCODE_SIGN_STYLE = Automatic;\n\t\t\t\tCOMBINE_HIDPI_IMAGES = YES;\n\t\t\t\tCURRENT_PROJECT_VERSION = \"$(APP_BUILD)\";\n\t\t\t\tDEAD_CODE_STRIPPING = YES;\n\t\t\t\tDEVELOPMENT_ASSET_PATHS = \"\\\"Copilot for Xcode/Preview Content\\\"\";\n\t\t\t\tDEVELOPMENT_TEAM = 5YKZ4Y3DAW;\n\t\t\t\tENABLE_HARDENED_RUNTIME = YES;\n\t\t\t\tENABLE_PREVIEWS = YES;\n\t\t\t\tGENERATE_INFOPLIST_FILE = YES;\n\t\t\t\tINFOPLIST_FILE = \"Copilot-for-Xcode-Info.plist\";\n\t\t\t\tINFOPLIST_KEY_CFBundleDisplayName = \"\";\n\t\t\t\tINFOPLIST_KEY_LSApplicationCategoryType = \"public.app-category.developer-tools\";\n\t\t\t\tINFOPLIST_KEY_NSHumanReadableCopyright = \"\";\n\t\t\t\tLD_RUNPATH_SEARCH_PATHS = (\n\t\t\t\t\t\"$(inherited)\",\n\t\t\t\t\t\"@executable_path/../Frameworks\",\n\t\t\t\t);\n\t\t\t\tMACOSX_DEPLOYMENT_TARGET = 13.0;\n\t\t\t\tMARKETING_VERSION = \"$(APP_VERSION)\";\n\t\t\t\tPRODUCT_BUNDLE_IDENTIFIER = \"$(BUNDLE_IDENTIFIER_BASE)\";\n\t\t\t\tPRODUCT_MODULE_NAME = Copilot_for_Xcode;\n\t\t\t\tPRODUCT_NAME = \"$(HOST_APP_NAME)\";\n\t\t\t\tSWIFT_EMIT_LOC_STRINGS = YES;\n\t\t\t\tSWIFT_VERSION = 5.0;\n\t\t\t};\n\t\t\tname = Debug;\n\t\t};\n\t\tC8189B272938973000C9DCDA /* Release */ = {\n\t\t\tisa = XCBuildConfiguration;\n\t\t\tbuildSettings = {\n\t\t\t\tALWAYS_EMBED_SWIFT_STANDARD_LIBRARIES = YES;\n\t\t\t\tASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;\n\t\t\t\tASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;\n\t\t\t\tCODE_SIGN_ENTITLEMENTS = \"Copilot for Xcode/Copilot_for_Xcode.entitlements\";\n\t\t\t\tCODE_SIGN_STYLE = Automatic;\n\t\t\t\tCOMBINE_HIDPI_IMAGES = YES;\n\t\t\t\tCURRENT_PROJECT_VERSION = \"$(APP_BUILD)\";\n\t\t\t\tDEAD_CODE_STRIPPING = YES;\n\t\t\t\tDEVELOPMENT_ASSET_PATHS = \"\\\"Copilot for Xcode/Preview Content\\\"\";\n\t\t\t\tDEVELOPMENT_TEAM = 5YKZ4Y3DAW;\n\t\t\t\tENABLE_HARDENED_RUNTIME = YES;\n\t\t\t\tENABLE_PREVIEWS = YES;\n\t\t\t\tGENERATE_INFOPLIST_FILE = YES;\n\t\t\t\tINFOPLIST_FILE = \"Copilot-for-Xcode-Info.plist\";\n\t\t\t\tINFOPLIST_KEY_CFBundleDisplayName = \"\";\n\t\t\t\tINFOPLIST_KEY_LSApplicationCategoryType = \"public.app-category.developer-tools\";\n\t\t\t\tINFOPLIST_KEY_NSHumanReadableCopyright = \"\";\n\t\t\t\tLD_RUNPATH_SEARCH_PATHS = (\n\t\t\t\t\t\"$(inherited)\",\n\t\t\t\t\t\"@executable_path/../Frameworks\",\n\t\t\t\t);\n\t\t\t\tMACOSX_DEPLOYMENT_TARGET = 13.0;\n\t\t\t\tMARKETING_VERSION = \"$(APP_VERSION)\";\n\t\t\t\tPRODUCT_BUNDLE_IDENTIFIER = \"$(BUNDLE_IDENTIFIER_BASE)\";\n\t\t\t\tPRODUCT_NAME = \"$(HOST_APP_NAME)\";\n\t\t\t\tSWIFT_EMIT_LOC_STRINGS = YES;\n\t\t\t\tSWIFT_VERSION = 5.0;\n\t\t\t};\n\t\t\tname = Release;\n\t\t};\n\t\tC8216B75298036EC00AD38C7 /* Debug */ = {\n\t\t\tisa = XCBuildConfiguration;\n\t\t\tbuildSettings = {\n\t\t\t\tCODE_SIGN_STYLE = Automatic;\n\t\t\t\tDEAD_CODE_STRIPPING = YES;\n\t\t\t\tDEVELOPMENT_TEAM = 5YKZ4Y3DAW;\n\t\t\t\tENABLE_HARDENED_RUNTIME = YES;\n\t\t\t\tMACOSX_DEPLOYMENT_TARGET = 13.0;\n\t\t\t\tPRODUCT_NAME = \"$(TARGET_NAME)\";\n\t\t\t\tSKIP_INSTALL = YES;\n\t\t\t\tSWIFT_VERSION = 5.0;\n\t\t\t};\n\t\t\tname = Debug;\n\t\t};\n\t\tC8216B76298036EC00AD38C7 /* Release */ = {\n\t\t\tisa = XCBuildConfiguration;\n\t\t\tbuildSettings = {\n\t\t\t\tCODE_SIGN_STYLE = Automatic;\n\t\t\t\tDEAD_CODE_STRIPPING = YES;\n\t\t\t\tDEVELOPMENT_TEAM = 5YKZ4Y3DAW;\n\t\t\t\tENABLE_HARDENED_RUNTIME = YES;\n\t\t\t\tMACOSX_DEPLOYMENT_TARGET = 13.0;\n\t\t\t\tPRODUCT_NAME = \"$(TARGET_NAME)\";\n\t\t\t\tSKIP_INSTALL = YES;\n\t\t\t\tSWIFT_VERSION = 5.0;\n\t\t\t};\n\t\t\tname = Release;\n\t\t};\n\t\tC861E61B2994F6080056CB02 /* Debug */ = {\n\t\t\tisa = XCBuildConfiguration;\n\t\t\tbuildSettings = {\n\t\t\t\tASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;\n\t\t\t\tASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;\n\t\t\t\tCODE_SIGN_ENTITLEMENTS = ExtensionService/ExtensionService.entitlements;\n\t\t\t\tCODE_SIGN_STYLE = Automatic;\n\t\t\t\tCOMBINE_HIDPI_IMAGES = YES;\n\t\t\t\tCURRENT_PROJECT_VERSION = \"$(APP_BUILD)\";\n\t\t\t\tDEAD_CODE_STRIPPING = YES;\n\t\t\t\tDEVELOPMENT_TEAM = 5YKZ4Y3DAW;\n\t\t\t\tENABLE_HARDENED_RUNTIME = YES;\n\t\t\t\tENABLE_PREVIEWS = YES;\n\t\t\t\tGENERATE_INFOPLIST_FILE = YES;\n\t\t\t\tINFOPLIST_FILE = ExtensionService/Info.plist;\n\t\t\t\tINFOPLIST_KEY_LSApplicationCategoryType = \"public.app-category.developer-tools\";\n\t\t\t\tINFOPLIST_KEY_NSHumanReadableCopyright = \"\";\n\t\t\t\tINFOPLIST_KEY_NSMainStoryboardFile = Main;\n\t\t\t\tINFOPLIST_KEY_NSPrincipalClass = NSApplication;\n\t\t\t\tLD_RUNPATH_SEARCH_PATHS = (\n\t\t\t\t\t\"$(inherited)\",\n\t\t\t\t\t\"@executable_path/../Frameworks\",\n\t\t\t\t);\n\t\t\t\tMACOSX_DEPLOYMENT_TARGET = 13.0;\n\t\t\t\tMARKETING_VERSION = \"$(APP_VERSION)\";\n\t\t\t\tPRODUCT_BUNDLE_IDENTIFIER = \"$(BUNDLE_IDENTIFIER_BASE).ExtensionService\";\n\t\t\t\tPRODUCT_NAME = \"$(EXTENSION_SERVICE_NAME)\";\n\t\t\t\tSKIP_INSTALL = YES;\n\t\t\t\tSWIFT_EMIT_LOC_STRINGS = YES;\n\t\t\t\tSWIFT_VERSION = 5.0;\n\t\t\t};\n\t\t\tname = Debug;\n\t\t};\n\t\tC861E61C2994F6080056CB02 /* Release */ = {\n\t\t\tisa = XCBuildConfiguration;\n\t\t\tbuildSettings = {\n\t\t\t\tASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;\n\t\t\t\tASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;\n\t\t\t\tCODE_SIGN_ENTITLEMENTS = ExtensionService/ExtensionService.entitlements;\n\t\t\t\tCODE_SIGN_STYLE = Automatic;\n\t\t\t\tCOMBINE_HIDPI_IMAGES = YES;\n\t\t\t\tCURRENT_PROJECT_VERSION = \"$(APP_BUILD)\";\n\t\t\t\tDEAD_CODE_STRIPPING = YES;\n\t\t\t\tDEVELOPMENT_TEAM = 5YKZ4Y3DAW;\n\t\t\t\tENABLE_HARDENED_RUNTIME = YES;\n\t\t\t\tENABLE_PREVIEWS = YES;\n\t\t\t\tGENERATE_INFOPLIST_FILE = YES;\n\t\t\t\tINFOPLIST_FILE = ExtensionService/Info.plist;\n\t\t\t\tINFOPLIST_KEY_LSApplicationCategoryType = \"public.app-category.developer-tools\";\n\t\t\t\tINFOPLIST_KEY_NSHumanReadableCopyright = \"\";\n\t\t\t\tINFOPLIST_KEY_NSMainStoryboardFile = Main;\n\t\t\t\tINFOPLIST_KEY_NSPrincipalClass = NSApplication;\n\t\t\t\tLD_RUNPATH_SEARCH_PATHS = (\n\t\t\t\t\t\"$(inherited)\",\n\t\t\t\t\t\"@executable_path/../Frameworks\",\n\t\t\t\t);\n\t\t\t\tMACOSX_DEPLOYMENT_TARGET = 13.0;\n\t\t\t\tMARKETING_VERSION = \"$(APP_VERSION)\";\n\t\t\t\tPRODUCT_BUNDLE_IDENTIFIER = \"$(BUNDLE_IDENTIFIER_BASE).ExtensionService\";\n\t\t\t\tPRODUCT_NAME = \"$(EXTENSION_SERVICE_NAME)\";\n\t\t\t\tSKIP_INSTALL = YES;\n\t\t\t\tSWIFT_EMIT_LOC_STRINGS = YES;\n\t\t\t\tSWIFT_VERSION = 5.0;\n\t\t\t};\n\t\t\tname = Release;\n\t\t};\n\t\tC8738B682BE4D4B900609E7F /* Debug */ = {\n\t\t\tisa = XCBuildConfiguration;\n\t\t\tbuildSettings = {\n\t\t\t\tASSETCATALOG_COMPILER_GENERATE_SWIFT_ASSET_SYMBOL_EXTENSIONS = YES;\n\t\t\t\tCODE_SIGN_STYLE = Automatic;\n\t\t\t\tDEVELOPMENT_TEAM = 5YKZ4Y3DAW;\n\t\t\t\tENABLE_HARDENED_RUNTIME = YES;\n\t\t\t\tENABLE_USER_SCRIPT_SANDBOXING = YES;\n\t\t\t\tGCC_C_LANGUAGE_STANDARD = gnu17;\n\t\t\t\tLOCALIZATION_PREFERS_STRING_CATALOGS = YES;\n\t\t\t\tMACOSX_DEPLOYMENT_TARGET = 13.0;\n\t\t\t\tPRODUCT_BUNDLE_IDENTIFIER = \"$(BUNDLE_IDENTIFIER_BASE).CommunicationBridge\";\n\t\t\t\tPRODUCT_NAME = \"$(TARGET_NAME)\";\n\t\t\t\tSKIP_INSTALL = YES;\n\t\t\t\tSWIFT_ACTIVE_COMPILATION_CONDITIONS = \"DEBUG $(inherited)\";\n\t\t\t\tSWIFT_VERSION = 5.0;\n\t\t\t};\n\t\t\tname = Debug;\n\t\t};\n\t\tC8738B692BE4D4B900609E7F /* Release */ = {\n\t\t\tisa = XCBuildConfiguration;\n\t\t\tbuildSettings = {\n\t\t\t\tASSETCATALOG_COMPILER_GENERATE_SWIFT_ASSET_SYMBOL_EXTENSIONS = YES;\n\t\t\t\tCODE_SIGN_STYLE = Automatic;\n\t\t\t\tDEVELOPMENT_TEAM = 5YKZ4Y3DAW;\n\t\t\t\tENABLE_HARDENED_RUNTIME = YES;\n\t\t\t\tENABLE_USER_SCRIPT_SANDBOXING = YES;\n\t\t\t\tGCC_C_LANGUAGE_STANDARD = gnu17;\n\t\t\t\tLOCALIZATION_PREFERS_STRING_CATALOGS = YES;\n\t\t\t\tMACOSX_DEPLOYMENT_TARGET = 13.0;\n\t\t\t\tPRODUCT_BUNDLE_IDENTIFIER = \"$(BUNDLE_IDENTIFIER_BASE).CommunicationBridge\";\n\t\t\t\tPRODUCT_NAME = \"$(TARGET_NAME)\";\n\t\t\t\tSKIP_INSTALL = YES;\n\t\t\t\tSWIFT_VERSION = 5.0;\n\t\t\t};\n\t\t\tname = Release;\n\t\t};\n\t\tC8738B852BE5363900609E7F /* Debug */ = {\n\t\t\tisa = XCBuildConfiguration;\n\t\t\tbuildSettings = {\n\t\t\t\tASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;\n\t\t\t\tASSETCATALOG_COMPILER_GENERATE_SWIFT_ASSET_SYMBOL_EXTENSIONS = YES;\n\t\t\t\tASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;\n\t\t\t\tCODE_SIGN_ENTITLEMENTS = SandboxedClientTester/SandboxedClientTester.entitlements;\n\t\t\t\tCODE_SIGN_STYLE = Automatic;\n\t\t\t\tCOMBINE_HIDPI_IMAGES = YES;\n\t\t\t\tCURRENT_PROJECT_VERSION = 1;\n\t\t\t\tDEVELOPMENT_ASSET_PATHS = \"\\\"SandboxedClientTester/Preview Content\\\"\";\n\t\t\t\tDEVELOPMENT_TEAM = 5YKZ4Y3DAW;\n\t\t\t\tENABLE_HARDENED_RUNTIME = YES;\n\t\t\t\tENABLE_PREVIEWS = YES;\n\t\t\t\tENABLE_USER_SCRIPT_SANDBOXING = YES;\n\t\t\t\tGCC_C_LANGUAGE_STANDARD = gnu17;\n\t\t\t\tGENERATE_INFOPLIST_FILE = YES;\n\t\t\t\tINFOPLIST_FILE = SandboxedClientTester/Info.plist;\n\t\t\t\tINFOPLIST_KEY_NSHumanReadableCopyright = \"\";\n\t\t\t\tLD_RUNPATH_SEARCH_PATHS = (\n\t\t\t\t\t\"$(inherited)\",\n\t\t\t\t\t\"@executable_path/../Frameworks\",\n\t\t\t\t);\n\t\t\t\tLOCALIZATION_PREFERS_STRING_CATALOGS = YES;\n\t\t\t\tMACOSX_DEPLOYMENT_TARGET = 14.2;\n\t\t\t\tMARKETING_VERSION = 1.0;\n\t\t\t\tPRODUCT_BUNDLE_IDENTIFIER = com.intii.CopilotForXcode.SandboxedClientTester;\n\t\t\t\tPRODUCT_NAME = \"$(TARGET_NAME)\";\n\t\t\t\tSWIFT_ACTIVE_COMPILATION_CONDITIONS = \"DEBUG $(inherited)\";\n\t\t\t\tSWIFT_EMIT_LOC_STRINGS = YES;\n\t\t\t\tSWIFT_VERSION = 5.0;\n\t\t\t};\n\t\t\tname = Debug;\n\t\t};\n\t\tC8738B862BE5363900609E7F /* Release */ = {\n\t\t\tisa = XCBuildConfiguration;\n\t\t\tbuildSettings = {\n\t\t\t\tASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;\n\t\t\t\tASSETCATALOG_COMPILER_GENERATE_SWIFT_ASSET_SYMBOL_EXTENSIONS = YES;\n\t\t\t\tASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;\n\t\t\t\tCODE_SIGN_ENTITLEMENTS = SandboxedClientTester/SandboxedClientTester.entitlements;\n\t\t\t\tCODE_SIGN_STYLE = Automatic;\n\t\t\t\tCOMBINE_HIDPI_IMAGES = YES;\n\t\t\t\tCURRENT_PROJECT_VERSION = 1;\n\t\t\t\tDEVELOPMENT_ASSET_PATHS = \"\\\"SandboxedClientTester/Preview Content\\\"\";\n\t\t\t\tDEVELOPMENT_TEAM = 5YKZ4Y3DAW;\n\t\t\t\tENABLE_HARDENED_RUNTIME = YES;\n\t\t\t\tENABLE_PREVIEWS = YES;\n\t\t\t\tENABLE_USER_SCRIPT_SANDBOXING = YES;\n\t\t\t\tGCC_C_LANGUAGE_STANDARD = gnu17;\n\t\t\t\tGENERATE_INFOPLIST_FILE = YES;\n\t\t\t\tINFOPLIST_FILE = SandboxedClientTester/Info.plist;\n\t\t\t\tINFOPLIST_KEY_NSHumanReadableCopyright = \"\";\n\t\t\t\tLD_RUNPATH_SEARCH_PATHS = (\n\t\t\t\t\t\"$(inherited)\",\n\t\t\t\t\t\"@executable_path/../Frameworks\",\n\t\t\t\t);\n\t\t\t\tLOCALIZATION_PREFERS_STRING_CATALOGS = YES;\n\t\t\t\tMACOSX_DEPLOYMENT_TARGET = 14.2;\n\t\t\t\tMARKETING_VERSION = 1.0;\n\t\t\t\tPRODUCT_BUNDLE_IDENTIFIER = com.intii.CopilotForXcode.SandboxedClientTester;\n\t\t\t\tPRODUCT_NAME = \"$(TARGET_NAME)\";\n\t\t\t\tSWIFT_EMIT_LOC_STRINGS = YES;\n\t\t\t\tSWIFT_VERSION = 5.0;\n\t\t\t};\n\t\t\tname = Release;\n\t\t};\n/* End XCBuildConfiguration section */\n\n/* Begin XCConfigurationList section */\n\t\tC814589C2939EFDC00135263 /* Build configuration list for PBXNativeTarget \"EditorExtension\" */ = {\n\t\t\tisa = XCConfigurationList;\n\t\t\tbuildConfigurations = (\n\t\t\t\tC814589D2939EFDC00135263 /* Debug */,\n\t\t\t\tC814589E2939EFDC00135263 /* Release */,\n\t\t\t);\n\t\t\tdefaultConfigurationIsVisible = 0;\n\t\t\tdefaultConfigurationName = Release;\n\t\t};\n\t\tC8189B112938972F00C9DCDA /* Build configuration list for PBXProject \"Copilot for Xcode\" */ = {\n\t\t\tisa = XCConfigurationList;\n\t\t\tbuildConfigurations = (\n\t\t\t\tC8189B232938973000C9DCDA /* Debug */,\n\t\t\t\tC8189B242938973000C9DCDA /* Release */,\n\t\t\t);\n\t\t\tdefaultConfigurationIsVisible = 0;\n\t\t\tdefaultConfigurationName = Release;\n\t\t};\n\t\tC8189B252938973000C9DCDA /* Build configuration list for PBXNativeTarget \"Copilot for Xcode\" */ = {\n\t\t\tisa = XCConfigurationList;\n\t\t\tbuildConfigurations = (\n\t\t\t\tC8189B262938973000C9DCDA /* Debug */,\n\t\t\t\tC8189B272938973000C9DCDA /* Release */,\n\t\t\t);\n\t\t\tdefaultConfigurationIsVisible = 0;\n\t\t\tdefaultConfigurationName = Release;\n\t\t};\n\t\tC8216B74298036EC00AD38C7 /* Build configuration list for PBXNativeTarget \"Helper\" */ = {\n\t\t\tisa = XCConfigurationList;\n\t\t\tbuildConfigurations = (\n\t\t\t\tC8216B75298036EC00AD38C7 /* Debug */,\n\t\t\t\tC8216B76298036EC00AD38C7 /* Release */,\n\t\t\t);\n\t\t\tdefaultConfigurationIsVisible = 0;\n\t\t\tdefaultConfigurationName = Release;\n\t\t};\n\t\tC861E61A2994F6080056CB02 /* Build configuration list for PBXNativeTarget \"ExtensionService\" */ = {\n\t\t\tisa = XCConfigurationList;\n\t\t\tbuildConfigurations = (\n\t\t\t\tC861E61B2994F6080056CB02 /* Debug */,\n\t\t\t\tC861E61C2994F6080056CB02 /* Release */,\n\t\t\t);\n\t\t\tdefaultConfigurationIsVisible = 0;\n\t\t\tdefaultConfigurationName = Release;\n\t\t};\n\t\tC8738B672BE4D4B900609E7F /* Build configuration list for PBXNativeTarget \"CommunicationBridge\" */ = {\n\t\t\tisa = XCConfigurationList;\n\t\t\tbuildConfigurations = (\n\t\t\t\tC8738B682BE4D4B900609E7F /* Debug */,\n\t\t\t\tC8738B692BE4D4B900609E7F /* Release */,\n\t\t\t);\n\t\t\tdefaultConfigurationIsVisible = 0;\n\t\t\tdefaultConfigurationName = Release;\n\t\t};\n\t\tC8738B842BE5363900609E7F /* Build configuration list for PBXNativeTarget \"SandboxedClientTester\" */ = {\n\t\t\tisa = XCConfigurationList;\n\t\t\tbuildConfigurations = (\n\t\t\t\tC8738B852BE5363900609E7F /* Debug */,\n\t\t\t\tC8738B862BE5363900609E7F /* Release */,\n\t\t\t);\n\t\t\tdefaultConfigurationIsVisible = 0;\n\t\t\tdefaultConfigurationName = Release;\n\t\t};\n/* End XCConfigurationList section */\n\n/* Begin XCRemoteSwiftPackageReference section */\n\t\tC80C91742A588DD800B5EADA /* XCRemoteSwiftPackageReference \"usearch\" */ = {\n\t\t\tisa = XCRemoteSwiftPackageReference;\n\t\t\trepositoryURL = \"https://github.com/unum-cloud/usearch\";\n\t\t\trequirement = {\n\t\t\t\tkind = upToNextMajorVersion;\n\t\t\t\tminimumVersion = 0.19.1;\n\t\t\t};\n\t\t};\n\t\tC8216B792980373800AD38C7 /* XCRemoteSwiftPackageReference \"swift-argument-parser\" */ = {\n\t\t\tisa = XCRemoteSwiftPackageReference;\n\t\t\trepositoryURL = \"https://github.com/apple/swift-argument-parser.git\";\n\t\t\trequirement = {\n\t\t\t\tkind = upToNextMajorVersion;\n\t\t\t\tminimumVersion = 1.0.0;\n\t\t\t};\n\t\t};\n/* End XCRemoteSwiftPackageReference section */\n\n/* Begin XCSwiftPackageProductDependency section */\n\t\tC8216B7C2980374300AD38C7 /* ArgumentParser */ = {\n\t\t\tisa = XCSwiftPackageProductDependency;\n\t\t\tpackage = C8216B792980373800AD38C7 /* XCRemoteSwiftPackageReference \"swift-argument-parser\" */;\n\t\t\tproductName = ArgumentParser;\n\t\t};\n\t\tC861E61D2994F6150056CB02 /* Service */ = {\n\t\t\tisa = XCSwiftPackageProductDependency;\n\t\t\tproductName = Service;\n\t\t};\n\t\tC86612F72A06AF74009197D9 /* HostApp */ = {\n\t\t\tisa = XCSwiftPackageProductDependency;\n\t\t\tproductName = HostApp;\n\t\t};\n\t\tC8738B6E2BE4F7A600609E7F /* XPCShared */ = {\n\t\t\tisa = XCSwiftPackageProductDependency;\n\t\t\tproductName = XPCShared;\n\t\t};\n\t\tC8738B872BE5365000609E7F /* Client */ = {\n\t\t\tisa = XCSwiftPackageProductDependency;\n\t\t\tproductName = Client;\n\t\t};\n\t\tC882175B294187EF00A22FD3 /* Client */ = {\n\t\t\tisa = XCSwiftPackageProductDependency;\n\t\t\tproductName = Client;\n\t\t};\n/* End XCSwiftPackageProductDependency section */\n\t};\n\trootObject = C8189B0E2938972F00C9DCDA /* Project object */;\n}\n" }, { "path": "Copilot for Xcode.xcodeproj/project.xcworkspace/contents.xcworkspacedata", "content": "\n\n \n \n\n" }, { "path": "Copilot for Xcode.xcodeproj/project.xcworkspace/xcshareddata/IDEWorkspaceChecks.plist", "content": "\n\n\n\n\tIDEDidComputeMac32BitWarning\n\t\n\n\n" }, { "path": "Copilot for Xcode.xcodeproj/project.xcworkspace/xcshareddata/swiftpm/Package.resolved", "content": "{\n \"pins\" : [\n {\n \"identity\" : \"combine-schedulers\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/combine-schedulers\",\n \"state\" : {\n \"revision\" : \"0625932976b3ae23949f6b816d13bd97f3b40b7c\",\n \"version\" : \"0.10.0\"\n }\n },\n {\n \"identity\" : \"fseventswrapper\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/Frizlab/FSEventsWrapper\",\n \"state\" : {\n \"revision\" : \"e0c59a2ce2775e5f6642da6d19207445f10112d0\",\n \"version\" : \"1.0.2\"\n }\n },\n {\n \"identity\" : \"glob\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/Bouke/Glob\",\n \"state\" : {\n \"revision\" : \"deda6e163d2ff2a8d7e138e2c3326dbd71157faf\",\n \"version\" : \"1.0.5\"\n }\n },\n {\n \"identity\" : \"highlightr\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/raspu/Highlightr\",\n \"state\" : {\n \"revision\" : \"93199b9e434f04bda956a613af8f571933f9f037\",\n \"version\" : \"2.1.2\"\n }\n },\n {\n \"identity\" : \"jsonrpc\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/ChimeHQ/JSONRPC\",\n \"state\" : {\n \"revision\" : \"5da978702aece6ba5c7879b0d253c180d61e4ef3\",\n \"version\" : \"0.6.0\"\n }\n },\n {\n \"identity\" : \"keychainaccess\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/kishikawakatsumi/KeychainAccess\",\n \"state\" : {\n \"revision\" : \"84e546727d66f1adc5439debad16270d0fdd04e7\",\n \"version\" : \"4.2.2\"\n }\n },\n {\n \"identity\" : \"languageclient\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/ChimeHQ/LanguageClient\",\n \"state\" : {\n \"revision\" : \"f0198ee0a102d266078f7d9c28f086f2989f988a\",\n \"version\" : \"0.3.1\"\n }\n },\n {\n \"identity\" : \"languageserverprotocol\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/ChimeHQ/LanguageServerProtocol\",\n \"state\" : {\n \"revision\" : \"6e97f943dc024307c5524a80bd33cdbd1cc621de\",\n \"version\" : \"0.8.0\"\n }\n },\n {\n \"identity\" : \"operationplus\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/ChimeHQ/OperationPlus\",\n \"state\" : {\n \"revision\" : \"1340f95dce3e93d742497d88db18f8676f4badf4\",\n \"version\" : \"1.6.0\"\n }\n },\n {\n \"identity\" : \"processenv\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/ChimeHQ/ProcessEnv\",\n \"state\" : {\n \"revision\" : \"29487b6581bb785c372c611c943541ef4309d051\",\n \"version\" : \"0.3.1\"\n }\n },\n {\n \"identity\" : \"sparkle\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/sparkle-project/Sparkle\",\n \"state\" : {\n \"revision\" : \"631846cc829f0f0cae327df9bafe5a32b7ddadce\",\n \"version\" : \"2.4.0\"\n }\n },\n {\n \"identity\" : \"splash\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/JohnSundell/Splash\",\n \"state\" : {\n \"branch\" : \"master\",\n \"revision\" : \"2e3f17c2d09689c8bf175c4a84ff7f2ad3353301\"\n }\n },\n {\n \"identity\" : \"swift-argument-parser\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/apple/swift-argument-parser.git\",\n \"state\" : {\n \"revision\" : \"fee6933f37fde9a5e12a1e4aeaa93fe60116ff2a\",\n \"version\" : \"1.2.2\"\n }\n },\n {\n \"identity\" : \"swift-async-algorithms\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/apple/swift-async-algorithms\",\n \"state\" : {\n \"revision\" : \"9cfed92b026c524674ed869a4ff2dcfdeedf8a2a\",\n \"version\" : \"0.1.0\"\n }\n },\n {\n \"identity\" : \"swift-case-paths\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-case-paths\",\n \"state\" : {\n \"revision\" : \"fc45e7b2cfece9dd80b5a45e6469ffe67fe67984\",\n \"version\" : \"0.14.1\"\n }\n },\n {\n \"identity\" : \"swift-clocks\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-clocks\",\n \"state\" : {\n \"revision\" : \"f9acfa1a45f4483fe0f2c434a74e6f68f865d12d\",\n \"version\" : \"0.3.0\"\n }\n },\n {\n \"identity\" : \"swift-collections\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/apple/swift-collections.git\",\n \"state\" : {\n \"revision\" : \"937e904258d22af6e447a0b72c0bc67583ef64a2\",\n \"version\" : \"1.0.4\"\n }\n },\n {\n \"identity\" : \"swift-composable-architecture\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-composable-architecture\",\n \"state\" : {\n \"revision\" : \"89f80fe2400d21a853abc9556a060a2fa50eb2cb\",\n \"version\" : \"0.55.0\"\n }\n },\n {\n \"identity\" : \"swift-custom-dump\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-custom-dump\",\n \"state\" : {\n \"revision\" : \"3a35f7892e7cf6ba28a78cd46a703c0be4e0c6dc\",\n \"version\" : \"0.11.0\"\n }\n },\n {\n \"identity\" : \"swift-dependencies\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-dependencies\",\n \"state\" : {\n \"revision\" : \"de1a984a71e51f6e488e98ce3652035563eb8acb\",\n \"version\" : \"0.5.1\"\n }\n },\n {\n \"identity\" : \"swift-identified-collections\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-identified-collections\",\n \"state\" : {\n \"revision\" : \"d01446a78fb768adc9a78cbb6df07767c8ccfc29\",\n \"version\" : \"0.8.0\"\n }\n },\n {\n \"identity\" : \"swift-markdown-ui\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/gonzalezreal/swift-markdown-ui\",\n \"state\" : {\n \"revision\" : \"12b351a75201a8124c2f2e1f9fc6ef5cd812c0b9\",\n \"version\" : \"2.1.0\"\n }\n },\n {\n \"identity\" : \"swift-parsing\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-parsing\",\n \"state\" : {\n \"revision\" : \"27c941bbd22a4bbc53005a15a0440443fd892f70\",\n \"version\" : \"0.12.1\"\n }\n },\n {\n \"identity\" : \"swiftsoup\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/scinfu/SwiftSoup.git\",\n \"state\" : {\n \"revision\" : \"0e96a20ffd37a515c5c963952d4335c89bed50a6\",\n \"version\" : \"2.6.0\"\n }\n },\n {\n \"identity\" : \"swiftui-navigation\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swiftui-navigation\",\n \"state\" : {\n \"revision\" : \"2aa885e719087ee19df251c08a5980ad3e787f12\",\n \"version\" : \"0.8.0\"\n }\n },\n {\n \"identity\" : \"tiktoken\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/intitni/Tiktoken\",\n \"state\" : {\n \"branch\" : \"main\",\n \"revision\" : \"6a2ac324c6daec167ca95268d5a487e6de6a1cea\"\n }\n },\n {\n \"identity\" : \"usearch\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/unum-cloud/usearch\",\n \"state\" : {\n \"revision\" : \"f2ab884d50902c3ad63f07a3a20bc34008a17449\",\n \"version\" : \"0.19.3\"\n }\n },\n {\n \"identity\" : \"xctest-dynamic-overlay\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/xctest-dynamic-overlay\",\n \"state\" : {\n \"revision\" : \"4af50b38daf0037cfbab15514a241224c3f62f98\",\n \"version\" : \"0.8.5\"\n }\n }\n ],\n \"version\" : 2\n}\n" }, { "path": "Copilot for Xcode.xcodeproj/xcshareddata/xcschemes/CommunicationBridge.xcscheme", "content": "\n\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n\n" }, { "path": "Copilot for Xcode.xcodeproj/xcshareddata/xcschemes/Copilot for Xcode Debug.xcscheme", "content": "\n\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n\n" }, { "path": "Copilot for Xcode.xcodeproj/xcshareddata/xcschemes/Copilot for Xcode.xcscheme", "content": "\n\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n\n" }, { "path": "Copilot for Xcode.xcodeproj/xcshareddata/xcschemes/EditorExtension.xcscheme", "content": "\n\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n\n" }, { "path": "Copilot for Xcode.xcodeproj/xcshareddata/xcschemes/ExtensionService.xcscheme", "content": "\n\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n\n" }, { "path": "Copilot for Xcode.xcodeproj/xcshareddata/xcschemes/SandboxedClientTester.xcscheme", "content": "\n\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n\n" }, { "path": "Copilot for Xcode.xcworkspace/contents.xcworkspacedata", "content": "\n\n \n \n \n \n\n" }, { "path": "Copilot for Xcode.xcworkspace/xcshareddata/IDEWorkspaceChecks.plist", "content": "\n\n\n\n\tIDEDidComputeMac32BitWarning\n\t\n\n\n" }, { "path": "Copilot for Xcode.xcworkspace/xcshareddata/swiftpm/Package.resolved", "content": "{\n \"pins\" : [\n {\n \"identity\" : \"aexml\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/tadija/AEXML.git\",\n \"state\" : {\n \"revision\" : \"db806756c989760b35108146381535aec231092b\",\n \"version\" : \"4.7.0\"\n }\n },\n {\n \"identity\" : \"cgeventoverride\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/intitni/CGEventOverride\",\n \"state\" : {\n \"revision\" : \"571d36d63e68fac30e4a350600cd186697936f74\",\n \"version\" : \"1.2.3\"\n }\n },\n {\n \"identity\" : \"codablewrappers\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/GottaGetSwifty/CodableWrappers\",\n \"state\" : {\n \"revision\" : \"4eb46a4c656333e8514db8aad204445741de7d40\",\n \"version\" : \"2.0.7\"\n }\n },\n {\n \"identity\" : \"combine-schedulers\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/combine-schedulers\",\n \"state\" : {\n \"revision\" : \"5928286acce13def418ec36d05a001a9641086f2\",\n \"version\" : \"1.0.3\"\n }\n },\n {\n \"identity\" : \"copilotforxcodekit\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/intitni/CopilotForXcodeKit\",\n \"state\" : {\n \"branch\" : \"feature/custom-chat-tab\",\n \"revision\" : \"63915ee1f8aba5375bc0f0166c8645fe81fe5b88\"\n }\n },\n {\n \"identity\" : \"fseventswrapper\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/Frizlab/FSEventsWrapper\",\n \"state\" : {\n \"revision\" : \"e0c59a2ce2775e5f6642da6d19207445f10112d0\",\n \"version\" : \"1.0.2\"\n }\n },\n {\n \"identity\" : \"generative-ai-swift\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/intitni/generative-ai-swift\",\n \"state\" : {\n \"branch\" : \"support-setting-base-url\",\n \"revision\" : \"12d7b30b566a64cc0dd628130bfb99a07368fea7\"\n }\n },\n {\n \"identity\" : \"glob\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/Bouke/Glob\",\n \"state\" : {\n \"revision\" : \"deda6e163d2ff2a8d7e138e2c3326dbd71157faf\",\n \"version\" : \"1.0.5\"\n }\n },\n {\n \"identity\" : \"highlightr\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/intitni/Highlightr\",\n \"state\" : {\n \"branch\" : \"master\",\n \"revision\" : \"81d8c8b3733939bf5d9e52cd6318f944cc033bd2\"\n }\n },\n {\n \"identity\" : \"indexstore-db\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/apple/indexstore-db.git\",\n \"state\" : {\n \"branch\" : \"release/6.1\",\n \"revision\" : \"54212fce1aecb199070808bdb265e7f17e396015\"\n }\n },\n {\n \"identity\" : \"jsonrpc\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/ChimeHQ/JSONRPC\",\n \"state\" : {\n \"revision\" : \"5da978702aece6ba5c7879b0d253c180d61e4ef3\",\n \"version\" : \"0.6.0\"\n }\n },\n {\n \"identity\" : \"keyboardshortcuts\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/intitni/KeyboardShortcuts\",\n \"state\" : {\n \"branch\" : \"main\",\n \"revision\" : \"65fb410b0c6d3ed96623b460bab31ffce5f48b4d\"\n }\n },\n {\n \"identity\" : \"languageclient\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/ChimeHQ/LanguageClient\",\n \"state\" : {\n \"revision\" : \"f0198ee0a102d266078f7d9c28f086f2989f988a\",\n \"version\" : \"0.3.1\"\n }\n },\n {\n \"identity\" : \"languageserverprotocol\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/ChimeHQ/LanguageServerProtocol\",\n \"state\" : {\n \"revision\" : \"6e97f943dc024307c5524a80bd33cdbd1cc621de\",\n \"version\" : \"0.8.0\"\n }\n },\n {\n \"identity\" : \"messagepacker\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/hirotakan/MessagePacker.git\",\n \"state\" : {\n \"revision\" : \"4d8346c6bc579347e4df0429493760691c5aeca2\",\n \"version\" : \"0.4.7\"\n }\n },\n {\n \"identity\" : \"networkimage\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/gonzalezreal/NetworkImage\",\n \"state\" : {\n \"revision\" : \"2849f5323265386e200484b0d0f896e73c3411b9\",\n \"version\" : \"6.0.1\"\n }\n },\n {\n \"identity\" : \"operationplus\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/ChimeHQ/OperationPlus\",\n \"state\" : {\n \"revision\" : \"1340f95dce3e93d742497d88db18f8676f4badf4\",\n \"version\" : \"1.6.0\"\n }\n },\n {\n \"identity\" : \"pathkit\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/kylef/PathKit.git\",\n \"state\" : {\n \"revision\" : \"3bfd2737b700b9a36565a8c94f4ad2b050a5e574\",\n \"version\" : \"1.0.1\"\n }\n },\n {\n \"identity\" : \"processenv\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/ChimeHQ/ProcessEnv\",\n \"state\" : {\n \"revision\" : \"29487b6581bb785c372c611c943541ef4309d051\",\n \"version\" : \"0.3.1\"\n }\n },\n {\n \"identity\" : \"sourcekitten\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/jpsim/SourceKitten\",\n \"state\" : {\n \"revision\" : \"eb6656ed26bdef967ad8d07c27e2eab34dc582f2\",\n \"version\" : \"0.37.0\"\n }\n },\n {\n \"identity\" : \"sparkle\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/sparkle-project/Sparkle\",\n \"state\" : {\n \"revision\" : \"0ca3004e98712ea2b39dd881d28448630cce1c99\",\n \"version\" : \"2.7.0\"\n }\n },\n {\n \"identity\" : \"spectre\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/kylef/Spectre.git\",\n \"state\" : {\n \"revision\" : \"26cc5e9ae0947092c7139ef7ba612e34646086c7\",\n \"version\" : \"0.10.1\"\n }\n },\n {\n \"identity\" : \"swift-argument-parser\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/apple/swift-argument-parser.git\",\n \"state\" : {\n \"revision\" : \"fee6933f37fde9a5e12a1e4aeaa93fe60116ff2a\",\n \"version\" : \"1.2.2\"\n }\n },\n {\n \"identity\" : \"swift-async-algorithms\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/apple/swift-async-algorithms\",\n \"state\" : {\n \"revision\" : \"042e1c4d9d19748c9c228f8d4ebc97bb1e339b0b\",\n \"version\" : \"1.0.4\"\n }\n },\n {\n \"identity\" : \"swift-case-paths\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-case-paths\",\n \"state\" : {\n \"revision\" : \"41b89b8b68d8c56c622dbb7132258f1a3e638b25\",\n \"version\" : \"1.7.0\"\n }\n },\n {\n \"identity\" : \"swift-clocks\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-clocks\",\n \"state\" : {\n \"revision\" : \"cc46202b53476d64e824e0b6612da09d84ffde8e\",\n \"version\" : \"1.0.6\"\n }\n },\n {\n \"identity\" : \"swift-cmark\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/swiftlang/swift-cmark\",\n \"state\" : {\n \"revision\" : \"b022b08312decdc46585e0b3440d97f6f22ef703\",\n \"version\" : \"0.6.0\"\n }\n },\n {\n \"identity\" : \"swift-collections\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/apple/swift-collections\",\n \"state\" : {\n \"revision\" : \"671108c96644956dddcd89dd59c203dcdb36cec7\",\n \"version\" : \"1.1.4\"\n }\n },\n {\n \"identity\" : \"swift-composable-architecture\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-composable-architecture\",\n \"state\" : {\n \"revision\" : \"69247baf7be2fd6f5820192caef0082d01849cd0\",\n \"version\" : \"1.16.1\"\n }\n },\n {\n \"identity\" : \"swift-concurrency-extras\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-concurrency-extras\",\n \"state\" : {\n \"revision\" : \"82a4ae7170d98d8538ec77238b7eb8e7199ef2e8\",\n \"version\" : \"1.3.1\"\n }\n },\n {\n \"identity\" : \"swift-custom-dump\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-custom-dump\",\n \"state\" : {\n \"revision\" : \"82645ec760917961cfa08c9c0c7104a57a0fa4b1\",\n \"version\" : \"1.3.3\"\n }\n },\n {\n \"identity\" : \"swift-dependencies\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-dependencies\",\n \"state\" : {\n \"revision\" : \"4c90d6b2b9bf0911af87b103bb40f41771891596\",\n \"version\" : \"1.9.2\"\n }\n },\n {\n \"identity\" : \"swift-identified-collections\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-identified-collections\",\n \"state\" : {\n \"revision\" : \"322d9ffeeba85c9f7c4984b39422ec7cc3c56597\",\n \"version\" : \"1.1.1\"\n }\n },\n {\n \"identity\" : \"swift-markdown-ui\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/gonzalezreal/swift-markdown-ui\",\n \"state\" : {\n \"revision\" : \"5f613358148239d0292c0cef674a3c2314737f9e\",\n \"version\" : \"2.4.1\"\n }\n },\n {\n \"identity\" : \"swift-navigation\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-navigation\",\n \"state\" : {\n \"revision\" : \"db6bc9dbfed001f21e6728fd36413d9342c235b4\",\n \"version\" : \"2.3.0\"\n }\n },\n {\n \"identity\" : \"swift-parsing\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-parsing\",\n \"state\" : {\n \"revision\" : \"3432cb81164dd3d69a75d0d63205be5fbae2c34b\",\n \"version\" : \"0.14.1\"\n }\n },\n {\n \"identity\" : \"swift-perception\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-perception\",\n \"state\" : {\n \"revision\" : \"d924c62a70fca5f43872f286dbd7cef0957f1c01\",\n \"version\" : \"1.6.0\"\n }\n },\n {\n \"identity\" : \"swift-syntax\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/apple/swift-syntax.git\",\n \"state\" : {\n \"revision\" : \"0687f71944021d616d34d922343dcef086855920\",\n \"version\" : \"600.0.1\"\n }\n },\n {\n \"identity\" : \"swiftsoup\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/scinfu/SwiftSoup.git\",\n \"state\" : {\n \"revision\" : \"8b6cf29eead8841a1fa7822481cb3af4ddaadba6\",\n \"version\" : \"2.6.1\"\n }\n },\n {\n \"identity\" : \"swiftterm\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/migueldeicaza/SwiftTerm\",\n \"state\" : {\n \"revision\" : \"e2b431dbf73f775fb4807a33e4572ffd3dc6933a\",\n \"version\" : \"1.2.5\"\n }\n },\n {\n \"identity\" : \"swifttreesitter\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/intitni/SwiftTreeSitter.git\",\n \"state\" : {\n \"branch\" : \"main\",\n \"revision\" : \"fd499bfafcccfae12a1a579dc922d8418025a35d\"\n }\n },\n {\n \"identity\" : \"swiftui-introspect\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/siteline/swiftui-introspect\",\n \"state\" : {\n \"revision\" : \"807f73ce09a9b9723f12385e592b4e0aaebd3336\",\n \"version\" : \"1.3.0\"\n }\n },\n {\n \"identity\" : \"swxmlhash\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/drmohundro/SWXMLHash.git\",\n \"state\" : {\n \"revision\" : \"a853604c9e9a83ad9954c7e3d2a565273982471f\",\n \"version\" : \"7.0.2\"\n }\n },\n {\n \"identity\" : \"tiktoken\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/intitni/Tiktoken\",\n \"state\" : {\n \"branch\" : \"main\",\n \"revision\" : \"6a2ac324c6daec167ca95268d5a487e6de6a1cea\"\n }\n },\n {\n \"identity\" : \"tree-sitter-objc\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/lukepistrol/tree-sitter-objc\",\n \"state\" : {\n \"branch\" : \"feature/spm\",\n \"revision\" : \"1b54ef0b5efddddf393b45e173788499cc572048\"\n }\n },\n {\n \"identity\" : \"usearch\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/unum-cloud/usearch\",\n \"state\" : {\n \"revision\" : \"f2ab884d50902c3ad63f07a3a20bc34008a17449\",\n \"version\" : \"0.19.3\"\n }\n },\n {\n \"identity\" : \"xcodeproj\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/tuist/XcodeProj.git\",\n \"state\" : {\n \"revision\" : \"b1caa062d4aaab3e3d2bed5fe0ac5f8ce9bf84f4\",\n \"version\" : \"8.27.7\"\n }\n },\n {\n \"identity\" : \"xctest-dynamic-overlay\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/xctest-dynamic-overlay\",\n \"state\" : {\n \"revision\" : \"39de59b2d47f7ef3ca88a039dff3084688fe27f4\",\n \"version\" : \"1.5.2\"\n }\n },\n {\n \"identity\" : \"yams\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/jpsim/Yams.git\",\n \"state\" : {\n \"revision\" : \"b4b8042411dc7bbb696300a34a4bf3ba1b7ad19b\",\n \"version\" : \"5.3.1\"\n }\n }\n ],\n \"version\" : 2\n}\n" }, { "path": "Copilot-for-Xcode-Info.plist", "content": "\n\n\n\n\tAPPLICATION_SUPPORT_FOLDER\n\t$(APPLICATION_SUPPORT_FOLDER)\n\tAPP_ID_PREFIX\n\t$(AppIdentifierPrefix)\n\tBUNDLE_IDENTIFIER_BASE\n\t$(BUNDLE_IDENTIFIER_BASE)\n\tEXTENSION_BUNDLE_NAME\n\t$(EXTENSION_BUNDLE_NAME)\n\tHOST_APP_NAME\n\t$(HOST_APP_NAME)\n\tNSAppTransportSecurity\n\t\n\t\tNSAllowsArbitraryLoads\n\t\t\n\t\n\tSUEnableJavaScript\n\tYES\n\tSUFeedURL\n\t$(SPARKLE_FEED_URL)\n\tSUPublicEDKey\n\t$(SPARKLE_PUBLIC_KEY)\n\tTEAM_ID_PREFIX\n\t$(TeamIdentifierPrefix)\n\n\n" }, { "path": "Core/.gitignore", "content": ".DS_Store\n/.build\n/Packages\n/*.xcodeproj\nxcuserdata/\nDerivedData/\n.swiftpm/config/registries.json\n.swiftpm/xcode/package.xcworkspace/contents.xcworkspacedata\n.netrc\n" }, { "path": "Core/.swiftpm/xcode/xcshareddata/xcschemes/Client.xcscheme", "content": "\n\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n\n" }, { "path": "Core/.swiftpm/xcode/xcshareddata/xcschemes/HostApp.xcscheme", "content": "\n\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n\n" }, { "path": "Core/.swiftpm/xcode/xcshareddata/xcschemes/Service.xcscheme", "content": "\n\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n\n" }, { "path": "Core/.swiftpm/xcode/xcshareddata/xcschemes/SuggestionInjector.xcscheme", "content": "\n\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n\n" }, { "path": "Core/Package.resolved", "content": "{\n \"pins\" : [\n {\n \"identity\" : \"codablewrappers\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/GottaGetSwifty/CodableWrappers\",\n \"state\" : {\n \"revision\" : \"4eb46a4c656333e8514db8aad204445741de7d40\",\n \"version\" : \"2.0.7\"\n }\n },\n {\n \"identity\" : \"combine-schedulers\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/combine-schedulers\",\n \"state\" : {\n \"revision\" : \"ec62f32d21584214a4b27c8cee2b2ad70ab2c38a\",\n \"version\" : \"0.11.0\"\n }\n },\n {\n \"identity\" : \"fseventswrapper\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/Frizlab/FSEventsWrapper\",\n \"state\" : {\n \"revision\" : \"e0c59a2ce2775e5f6642da6d19207445f10112d0\",\n \"version\" : \"1.0.2\"\n }\n },\n {\n \"identity\" : \"glob\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/Bouke/Glob\",\n \"state\" : {\n \"revision\" : \"deda6e163d2ff2a8d7e138e2c3326dbd71157faf\",\n \"version\" : \"1.0.5\"\n }\n },\n {\n \"identity\" : \"highlightr\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/intitni/Highlightr\",\n \"state\" : {\n \"branch\" : \"bump-highlight-js-version\",\n \"revision\" : \"4ffbb1b0b721378263297cafea6f2838044eb1eb\"\n }\n },\n {\n \"identity\" : \"jsonrpc\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/ChimeHQ/JSONRPC\",\n \"state\" : {\n \"revision\" : \"5da978702aece6ba5c7879b0d253c180d61e4ef3\",\n \"version\" : \"0.6.0\"\n }\n },\n {\n \"identity\" : \"languageclient\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/ChimeHQ/LanguageClient\",\n \"state\" : {\n \"revision\" : \"f0198ee0a102d266078f7d9c28f086f2989f988a\",\n \"version\" : \"0.3.1\"\n }\n },\n {\n \"identity\" : \"languageserverprotocol\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/ChimeHQ/LanguageServerProtocol\",\n \"state\" : {\n \"revision\" : \"6e97f943dc024307c5524a80bd33cdbd1cc621de\",\n \"version\" : \"0.8.0\"\n }\n },\n {\n \"identity\" : \"operationplus\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/ChimeHQ/OperationPlus\",\n \"state\" : {\n \"revision\" : \"1340f95dce3e93d742497d88db18f8676f4badf4\",\n \"version\" : \"1.6.0\"\n }\n },\n {\n \"identity\" : \"processenv\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/ChimeHQ/ProcessEnv\",\n \"state\" : {\n \"revision\" : \"29487b6581bb785c372c611c943541ef4309d051\",\n \"version\" : \"0.3.1\"\n }\n },\n {\n \"identity\" : \"sparkle\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/sparkle-project/Sparkle\",\n \"state\" : {\n \"revision\" : \"631846cc829f0f0cae327df9bafe5a32b7ddadce\",\n \"version\" : \"2.4.0\"\n }\n },\n {\n \"identity\" : \"swift-async-algorithms\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/apple/swift-async-algorithms\",\n \"state\" : {\n \"revision\" : \"9cfed92b026c524674ed869a4ff2dcfdeedf8a2a\",\n \"version\" : \"0.1.0\"\n }\n },\n {\n \"identity\" : \"swift-case-paths\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-case-paths\",\n \"state\" : {\n \"revision\" : \"fc45e7b2cfece9dd80b5a45e6469ffe67fe67984\",\n \"version\" : \"0.14.1\"\n }\n },\n {\n \"identity\" : \"swift-clocks\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-clocks\",\n \"state\" : {\n \"revision\" : \"0fbaebfc013715dab44d715a4d350ba37f297e4d\",\n \"version\" : \"0.4.0\"\n }\n },\n {\n \"identity\" : \"swift-collections\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/apple/swift-collections.git\",\n \"state\" : {\n \"revision\" : \"937e904258d22af6e447a0b72c0bc67583ef64a2\",\n \"version\" : \"1.0.4\"\n }\n },\n {\n \"identity\" : \"swift-composable-architecture\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-composable-architecture\",\n \"state\" : {\n \"revision\" : \"9f4202ab5b8422aa90f0ed983bf7652c3af7abf0\",\n \"version\" : \"0.59.0\"\n }\n },\n {\n \"identity\" : \"swift-concurrency-extras\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-concurrency-extras\",\n \"state\" : {\n \"revision\" : \"479750bd98fac2e813fffcf2af0728b5b0085795\",\n \"version\" : \"0.1.1\"\n }\n },\n {\n \"identity\" : \"swift-custom-dump\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-custom-dump\",\n \"state\" : {\n \"revision\" : \"4a87bb75be70c983a9548597e8783236feb3401e\",\n \"version\" : \"0.11.1\"\n }\n },\n {\n \"identity\" : \"swift-dependencies\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-dependencies\",\n \"state\" : {\n \"revision\" : \"16fd42ae04c6e7f74a6a86395d04722c641cccee\",\n \"version\" : \"0.6.0\"\n }\n },\n {\n \"identity\" : \"swift-identified-collections\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-identified-collections\",\n \"state\" : {\n \"revision\" : \"d01446a78fb768adc9a78cbb6df07767c8ccfc29\",\n \"version\" : \"0.8.0\"\n }\n },\n {\n \"identity\" : \"swift-markdown-ui\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/gonzalezreal/swift-markdown-ui\",\n \"state\" : {\n \"revision\" : \"12b351a75201a8124c2f2e1f9fc6ef5cd812c0b9\",\n \"version\" : \"2.1.0\"\n }\n },\n {\n \"identity\" : \"swift-parsing\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-parsing\",\n \"state\" : {\n \"revision\" : \"27c941bbd22a4bbc53005a15a0440443fd892f70\",\n \"version\" : \"0.12.1\"\n }\n },\n {\n \"identity\" : \"swift-syntax\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/apple/swift-syntax.git\",\n \"state\" : {\n \"branch\" : \"main\",\n \"revision\" : \"e149b01cfd3e96240e102729697e2095c19157ef\"\n }\n },\n {\n \"identity\" : \"swiftsoup\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/scinfu/SwiftSoup.git\",\n \"state\" : {\n \"revision\" : \"8b6cf29eead8841a1fa7822481cb3af4ddaadba6\",\n \"version\" : \"2.6.1\"\n }\n },\n {\n \"identity\" : \"swifttreesitter\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/ChimeHQ/SwiftTreeSitter\",\n \"state\" : {\n \"revision\" : \"a9b1335d5151b62b11f07599bd07d07dc5965de3\",\n \"version\" : \"0.7.2\"\n }\n },\n {\n \"identity\" : \"swiftui-navigation\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swiftui-navigation\",\n \"state\" : {\n \"revision\" : \"2aa885e719087ee19df251c08a5980ad3e787f12\",\n \"version\" : \"0.8.0\"\n }\n },\n {\n \"identity\" : \"tiktoken\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/intitni/Tiktoken\",\n \"state\" : {\n \"branch\" : \"main\",\n \"revision\" : \"6a2ac324c6daec167ca95268d5a487e6de6a1cea\"\n }\n },\n {\n \"identity\" : \"tree-sitter-objc\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/lukepistrol/tree-sitter-objc\",\n \"state\" : {\n \"branch\" : \"feature/spm\",\n \"revision\" : \"1b54ef0b5efddddf393b45e173788499cc572048\"\n }\n },\n {\n \"identity\" : \"tree-sitter-swift\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/alex-pinkus/tree-sitter-swift\",\n \"state\" : {\n \"branch\" : \"with-generated-files\",\n \"revision\" : \"eda05af7ac41adb4eb19c346883c0fa32fe3bdd8\"\n }\n },\n {\n \"identity\" : \"usearch\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/unum-cloud/usearch\",\n \"state\" : {\n \"revision\" : \"33c53288b44ccb55de77776820676132a6e4c42a\",\n \"version\" : \"0.23.0\"\n }\n },\n {\n \"identity\" : \"xctest-dynamic-overlay\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/xctest-dynamic-overlay\",\n \"state\" : {\n \"revision\" : \"50843cbb8551db836adec2290bb4bc6bac5c1865\",\n \"version\" : \"0.9.0\"\n }\n }\n ],\n \"version\" : 2\n}\n" }, { "path": "Core/Package.swift", "content": "// swift-tools-version: 5.7\n// The swift-tools-version declares the minimum version of Swift required to build this package.\n\nimport Foundation\nimport PackageDescription\n\n// MARK: - Package\n\nlet package = Package(\n name: \"Core\",\n platforms: [.macOS(.v13)],\n products: [\n .library(\n name: \"Service\",\n targets: [\n \"Service\",\n \"FileChangeChecker\",\n \"LaunchAgentManager\",\n \"UpdateChecker\",\n ]\n ),\n .library(\n name: \"Client\",\n targets: [\n \"Client\",\n ]\n ),\n .library(\n name: \"HostApp\",\n targets: [\n \"HostApp\",\n \"Client\",\n \"LaunchAgentManager\",\n \"UpdateChecker\",\n ]\n ),\n ],\n dependencies: [\n .package(path: \"../Tool\"),\n .package(path: \"../ChatPlugins\"),\n .package(path: \"../OverlayWindow\"),\n .package(url: \"https://github.com/apple/swift-async-algorithms\", from: \"1.0.0\"),\n .package(url: \"https://github.com/gonzalezreal/swift-markdown-ui\", from: \"2.4.1\"),\n .package(url: \"https://github.com/sparkle-project/Sparkle\", from: \"2.6.4\"),\n .package(url: \"https://github.com/pointfreeco/swift-parsing\", from: \"0.12.1\"),\n .package(url: \"https://github.com/pointfreeco/swift-dependencies\", from: \"1.0.0\"),\n .package(\n url: \"https://github.com/pointfreeco/swift-composable-architecture\",\n exact: \"1.16.1\"\n ),\n // quick hack to support custom UserDefaults\n // https://github.com/sindresorhus/KeyboardShortcuts\n .package(url: \"https://github.com/intitni/KeyboardShortcuts\", branch: \"main\"),\n .package(url: \"https://github.com/intitni/CGEventOverride\", from: \"1.2.1\"),\n .package(url: \"https://github.com/intitni/Highlightr\", branch: \"master\"),\n ].pro,\n targets: [\n // MARK: - Main\n\n .target(\n name: \"Client\",\n dependencies: [\n .product(name: \"XPCShared\", package: \"Tool\"),\n .product(name: \"SuggestionProvider\", package: \"Tool\"),\n .product(name: \"SuggestionBasic\", package: \"Tool\"),\n .product(name: \"Logger\", package: \"Tool\"),\n .product(name: \"Preferences\", package: \"Tool\"),\n ].proCore([\n \"LicenseManagement\",\n ])\n ),\n .target(\n name: \"Service\",\n dependencies: [\n \"SuggestionWidget\",\n \"SuggestionService\",\n \"ChatService\",\n \"PromptToCodeService\",\n \"ServiceUpdateMigration\",\n \"ChatGPTChatTab\",\n \"PlusFeatureFlag\",\n \"KeyBindingManager\",\n \"XcodeThemeController\",\n .product(name: \"XPCShared\", package: \"Tool\"),\n .product(name: \"SuggestionProvider\", package: \"Tool\"),\n .product(name: \"Workspace\", package: \"Tool\"),\n .product(name: \"WorkspaceSuggestionService\", package: \"Tool\"),\n .product(name: \"UserDefaultsObserver\", package: \"Tool\"),\n .product(name: \"AppMonitoring\", package: \"Tool\"),\n .product(name: \"SuggestionBasic\", package: \"Tool\"),\n .product(name: \"PromptToCode\", package: \"Tool\"),\n .product(name: \"ChatTab\", package: \"Tool\"),\n .product(name: \"Logger\", package: \"Tool\"),\n .product(name: \"OpenAIService\", package: \"Tool\"),\n .product(name: \"Preferences\", package: \"Tool\"),\n .product(name: \"CommandHandler\", package: \"Tool\"),\n .product(name: \"OverlayWindow\", package: \"OverlayWindow\"),\n .product(name: \"AsyncAlgorithms\", package: \"swift-async-algorithms\"),\n .product(name: \"ComposableArchitecture\", package: \"swift-composable-architecture\"),\n .product(name: \"Dependencies\", package: \"swift-dependencies\"),\n .product(name: \"KeyboardShortcuts\", package: \"KeyboardShortcuts\"),\n .product(name: \"CustomCommandTemplateProcessor\", package: \"Tool\"),\n ].pro([\n \"ProService\",\n ])\n ),\n .testTarget(\n name: \"ServiceTests\",\n dependencies: [\n \"Service\",\n \"Client\",\n .product(name: \"XPCShared\", package: \"Tool\"),\n .product(name: \"SuggestionProvider\", package: \"Tool\"),\n .product(name: \"SuggestionBasic\", package: \"Tool\"),\n .product(name: \"Preferences\", package: \"Tool\"),\n ]\n ),\n\n // MARK: - Host App\n\n .target(\n name: \"HostApp\",\n dependencies: [\n \"Client\",\n \"LaunchAgentManager\",\n \"PlusFeatureFlag\",\n .product(name: \"SuggestionProvider\", package: \"Tool\"),\n .product(name: \"Toast\", package: \"Tool\"),\n .product(name: \"SharedUIComponents\", package: \"Tool\"),\n .product(name: \"SuggestionBasic\", package: \"Tool\"),\n .product(name: \"WebSearchService\", package: \"Tool\"),\n .product(name: \"MarkdownUI\", package: \"swift-markdown-ui\"),\n .product(name: \"OpenAIService\", package: \"Tool\"),\n .product(name: \"Preferences\", package: \"Tool\"),\n .product(name: \"ComposableArchitecture\", package: \"swift-composable-architecture\"),\n .product(name: \"KeyboardShortcuts\", package: \"KeyboardShortcuts\"),\n ].pro([\n \"ProHostApp\",\n ])\n ),\n\n // MARK: - Suggestion Service\n\n .target(\n name: \"SuggestionService\",\n dependencies: [\n .product(name: \"UserDefaultsObserver\", package: \"Tool\"),\n .product(name: \"Preferences\", package: \"Tool\"),\n .product(name: \"SuggestionBasic\", package: \"Tool\"),\n .product(name: \"SuggestionProvider\", package: \"Tool\"),\n ].pro([\n \"ProExtension\",\n ])\n ),\n\n // MARK: - Prompt To Code\n\n .target(\n name: \"PromptToCodeService\",\n dependencies: [\n .product(name: \"PromptToCode\", package: \"Tool\"),\n .product(name: \"FocusedCodeFinder\", package: \"Tool\"),\n .product(name: \"SuggestionBasic\", package: \"Tool\"),\n .product(name: \"OpenAIService\", package: \"Tool\"),\n .product(name: \"AppMonitoring\", package: \"Tool\"),\n .product(name: \"ComposableArchitecture\", package: \"swift-composable-architecture\"),\n ].pro([\n \"ProService\",\n ])\n ),\n .testTarget(name: \"PromptToCodeServiceTests\", dependencies: [\"PromptToCodeService\"]),\n\n // MARK: - Chat\n\n .target(\n name: \"ChatService\",\n dependencies: [\n \"LegacyChatPlugin\",\n\n // context collectors\n \"WebChatContextCollector\",\n \"SystemInfoChatContextCollector\",\n\n .product(name: \"ChatContextCollector\", package: \"Tool\"),\n .product(name: \"PromptToCode\", package: \"Tool\"),\n .product(name: \"AppMonitoring\", package: \"Tool\"),\n .product(name: \"OpenAIService\", package: \"Tool\"),\n .product(name: \"Preferences\", package: \"Tool\"),\n .product(name: \"CustomCommandTemplateProcessor\", package: \"Tool\"),\n .product(name: \"ChatPlugins\", package: \"ChatPlugins\"),\n .product(name: \"Parsing\", package: \"swift-parsing\"),\n ].pro([\n \"ProService\",\n ])\n ),\n .testTarget(name: \"ChatServiceTests\", dependencies: [\"ChatService\"]),\n .target(\n name: \"LegacyChatPlugin\",\n dependencies: [\n .product(name: \"AppMonitoring\", package: \"Tool\"),\n .product(name: \"OpenAIService\", package: \"Tool\"),\n .product(name: \"Terminal\", package: \"Tool\"),\n ]\n ),\n\n .target(\n name: \"ChatGPTChatTab\",\n dependencies: [\n \"ChatService\",\n .product(name: \"SharedUIComponents\", package: \"Tool\"),\n .product(name: \"OpenAIService\", package: \"Tool\"),\n .product(name: \"Logger\", package: \"Tool\"),\n .product(name: \"ChatTab\", package: \"Tool\"),\n .product(name: \"Terminal\", package: \"Tool\"),\n .product(name: \"MarkdownUI\", package: \"swift-markdown-ui\"),\n .product(name: \"ComposableArchitecture\", package: \"swift-composable-architecture\"),\n ]\n ),\n\n // MARK: - UI\n\n .target(\n name: \"SuggestionWidget\",\n dependencies: [\n \"PromptToCodeService\",\n \"ChatGPTChatTab\",\n .product(name: \"PromptToCode\", package: \"Tool\"),\n .product(name: \"Toast\", package: \"Tool\"),\n .product(name: \"UserDefaultsObserver\", package: \"Tool\"),\n .product(name: \"SharedUIComponents\", package: \"Tool\"),\n .product(name: \"AppMonitoring\", package: \"Tool\"),\n .product(name: \"ChatTab\", package: \"Tool\"),\n .product(name: \"Logger\", package: \"Tool\"),\n .product(name: \"CustomAsyncAlgorithms\", package: \"Tool\"),\n .product(name: \"CodeDiff\", package: \"Tool\"),\n .product(name: \"AsyncAlgorithms\", package: \"swift-async-algorithms\"),\n .product(name: \"MarkdownUI\", package: \"swift-markdown-ui\"),\n .product(name: \"ComposableArchitecture\", package: \"swift-composable-architecture\"),\n ]\n ),\n .testTarget(name: \"SuggestionWidgetTests\", dependencies: [\"SuggestionWidget\"]),\n\n // MARK: - Helpers\n\n .target(name: \"FileChangeChecker\"),\n .target(name: \"LaunchAgentManager\"),\n .target(\n name: \"UpdateChecker\",\n dependencies: [\n \"Sparkle\",\n .product(name: \"Preferences\", package: \"Tool\"),\n .product(name: \"Logger\", package: \"Tool\"),\n ]\n ),\n .target(\n name: \"ServiceUpdateMigration\",\n dependencies: [\n .product(name: \"SuggestionProvider\", package: \"Tool\"),\n .product(name: \"Preferences\", package: \"Tool\"),\n .product(name: \"Keychain\", package: \"Tool\"),\n ]\n ),\n .testTarget(\n name: \"ServiceUpdateMigrationTests\",\n dependencies: [\n \"ServiceUpdateMigration\",\n ]\n ),\n .target(\n name: \"PlusFeatureFlag\",\n dependencies: [\n ].pro([\n \"LicenseManagement\",\n ])\n ),\n\n // MAKR: - Chat Context Collector\n\n .target(\n name: \"WebChatContextCollector\",\n dependencies: [\n .product(name: \"ChatContextCollector\", package: \"Tool\"),\n .product(name: \"LangChain\", package: \"Tool\"),\n .product(name: \"OpenAIService\", package: \"Tool\"),\n .product(name: \"ExternalServices\", package: \"Tool\"),\n .product(name: \"Preferences\", package: \"Tool\"),\n ],\n path: \"Sources/ChatContextCollectors/WebChatContextCollector\"\n ),\n\n .target(\n name: \"SystemInfoChatContextCollector\",\n dependencies: [\n .product(name: \"ChatContextCollector\", package: \"Tool\"),\n .product(name: \"OpenAIService\", package: \"Tool\"),\n ],\n path: \"Sources/ChatContextCollectors/SystemInfoChatContextCollector\"\n ),\n\n // MARK: Key Binding\n\n .target(\n name: \"KeyBindingManager\",\n dependencies: [\n .product(name: \"CommandHandler\", package: \"Tool\"),\n .product(name: \"Workspace\", package: \"Tool\"),\n .product(name: \"Preferences\", package: \"Tool\"),\n .product(name: \"Logger\", package: \"Tool\"),\n .product(name: \"AppMonitoring\", package: \"Tool\"),\n .product(name: \"UserDefaultsObserver\", package: \"Tool\"),\n .product(name: \"CGEventOverride\", package: \"CGEventOverride\"),\n .product(name: \"ComposableArchitecture\", package: \"swift-composable-architecture\"),\n ]\n ),\n .testTarget(\n name: \"KeyBindingManagerTests\",\n dependencies: [\"KeyBindingManager\"]\n ),\n\n // MARK: Theming\n\n .target(\n name: \"XcodeThemeController\",\n dependencies: [\n .product(name: \"Preferences\", package: \"Tool\"),\n .product(name: \"AppMonitoring\", package: \"Tool\"),\n .product(name: \"Highlightr\", package: \"Highlightr\"),\n ]\n ),\n ]\n)\n\nextension [Target.Dependency] {\n func pro(_ targetNames: [String]) -> [Target.Dependency] {\n if isProIncluded {\n return self + targetNames.map { Target.Dependency.product(name: $0, package: \"Pro\") }\n }\n return self\n }\n\n func proCore(_ targetNames: [String]) -> [Target.Dependency] {\n if isProIncluded {\n return self + targetNames\n .map { Target.Dependency.product(name: $0, package: \"ProCore\") }\n }\n return self\n }\n}\n\nextension [Package.Dependency] {\n var pro: [Package.Dependency] {\n if isProIncluded {\n return self + [.package(path: \"../../Pro\"), .package(path: \"../../Pro/ProCore\")]\n }\n return self\n }\n}\n\nvar isProIncluded: Bool {\n func isProIncluded(file: StaticString = #file) -> Bool {\n let filePath = \"\\(file)\"\n let fileURL = URL(fileURLWithPath: filePath)\n let rootURL = fileURL\n .deletingLastPathComponent()\n .deletingLastPathComponent()\n .deletingLastPathComponent()\n let confURL = rootURL.appendingPathComponent(\"PLUS\")\n return FileManager.default.fileExists(atPath: confURL.path)\n }\n\n return isProIncluded()\n}\n\n" }, { "path": "Core/README.md", "content": "# Core\n\nA description of this package.\n" }, { "path": "Core/Sources/ChatContextCollectors/SystemInfoChatContextCollector/SystemInfoChatContextCollector.swift", "content": "import ChatContextCollector\nimport Foundation\nimport OpenAIService\n\npublic final class SystemInfoChatContextCollector: ChatContextCollector {\n static let dateFormatter: DateFormatter = {\n let formatter = DateFormatter()\n formatter.dateFormat = \"EEEE, yyyy-MM-dd HH:mm:ssZ\"\n return formatter\n }()\n\n public init() {}\n\n public func generateContext(\n history: [ChatMessage],\n scopes: Set,\n content: String,\n configuration: ChatGPTConfiguration\n ) -> ChatContext {\n return .init(\n systemPrompt: \"\"\"\n ## System Info\n \n Current Time: \\(\n Self.dateFormatter.string(from: Date())\n ) (You can use it to calculate time in another time zone)\n \"\"\",\n retrievedContent: [],\n functions: []\n )\n }\n}\n\n" }, { "path": "Core/Sources/ChatContextCollectors/WebChatContextCollector/QueryWebsiteFunction.swift", "content": "import ChatBasic\nimport Foundation\nimport LangChain\nimport OpenAIService\nimport Preferences\n\nstruct QueryWebsiteFunction: ChatGPTFunction {\n struct Arguments: Codable {\n var query: String\n var urls: [String]\n }\n\n struct Result: ChatGPTFunctionResult {\n var answers: [String]\n\n var botReadableContent: String {\n return answers.joined(separator: \"\\n\")\n }\n \n var userReadableContent: ChatGPTFunctionResultUserReadableContent {\n .text(botReadableContent)\n }\n }\n\n var name: String {\n \"queryWebsite\"\n }\n\n var description: String {\n \"Useful for when you need to answer a question using information from a website.\"\n }\n\n var argumentSchema: JSONSchemaValue {\n return [\n .type: \"object\",\n .properties: [\n \"query\": [\n .type: \"string\",\n .description: \"things you want to know about the website\",\n ],\n \"urls\": [\n .type: \"array\",\n .description: \"urls of the website, you can use urls appearing in the conversation\",\n .items: [\n .type: \"string\",\n ],\n ],\n ],\n .required: [\"query\", \"urls\"],\n ]\n }\n\n func prepare(reportProgress: @escaping (String) async -> Void) async {\n await reportProgress(\"Reading..\")\n }\n\n func call(\n arguments: Arguments,\n reportProgress: @escaping (String) async -> Void\n ) async throws -> Result {\n do {\n let configuration = UserPreferenceEmbeddingConfiguration()\n let embedding = OpenAIEmbedding(configuration: configuration)\n let dimensions = configuration.dimensions\n let modelName = configuration.model?.name ?? \"model\"\n\n let result = try await withThrowingTaskGroup(of: String.self) { group in\n for urlString in arguments.urls {\n let storeIdentifier = \"\\(urlString)-\\(modelName)-\\(dimensions)\"\n guard let url = URL(string: urlString) else { continue }\n group.addTask {\n // 1. grab the website content\n await reportProgress(\"Loading \\(url)..\")\n\n if let database = await TemporaryUSearch.view(\n identifier: storeIdentifier,\n dimensions: dimensions\n ) {\n await reportProgress(\"Getting relevant information..\")\n let qa = QAInformationRetrievalChain(\n vectorStore: database,\n embedding: embedding\n )\n return try await qa.call(.init(arguments.query)).information\n }\n let loader = WebLoader(urls: [url])\n let documents = try await loader.load()\n await reportProgress(\"Processing \\(url)..\")\n // 2. split the content\n let splitter = RecursiveCharacterTextSplitter(\n chunkSize: 1500,\n chunkOverlap: 100\n )\n let splitDocuments = try await splitter.transformDocuments(documents)\n // 3. embedding and store in db\n await reportProgress(\"Embedding \\(url)..\")\n let embeddedDocuments = try await embedding.embed(documents: splitDocuments)\n let database = TemporaryUSearch(\n identifier: storeIdentifier,\n dimensions: dimensions\n )\n try await database.set(embeddedDocuments)\n // 4. generate answer\n await reportProgress(\"Getting relevant information..\")\n let qa = QAInformationRetrievalChain(\n vectorStore: database,\n embedding: embedding\n )\n let result = try await qa.call(.init(arguments.query))\n return result.information\n }\n }\n\n var all = [String]()\n for try await result in group {\n all.append(result)\n }\n await reportProgress(\"\"\"\n Finish reading websites.\n \\(\n arguments.urls\n .map { \"- [\\($0)](\\($0))\" }\n .joined(separator: \"\\n\")\n )\n \"\"\")\n\n return all\n }\n\n return .init(answers: result)\n } catch {\n await reportProgress(\"Failed reading websites.\")\n throw error\n }\n }\n}\n\n" }, { "path": "Core/Sources/ChatContextCollectors/WebChatContextCollector/SearchFunction.swift", "content": "import ChatBasic\nimport Foundation\nimport OpenAIService\nimport Preferences\nimport WebSearchService\n\nstruct SearchFunction: ChatGPTFunction {\n static let dateFormatter = {\n let it = DateFormatter()\n it.dateFormat = \"yyyy-MM-dd\"\n return it\n }()\n\n struct Arguments: Codable {\n var query: String\n var freshness: String?\n }\n\n struct Result: ChatGPTFunctionResult {\n var result: WebSearchResult\n\n var botReadableContent: String {\n result.webPages.enumerated().map {\n let (index, page) = $0\n return \"\"\"\n \\(index + 1). \\(page.title) \\(page.urlString)\n \\(page.snippet)\n \"\"\"\n }.joined(separator: \"\\n\")\n }\n \n var userReadableContent: ChatGPTFunctionResultUserReadableContent {\n .text(botReadableContent)\n }\n }\n\n let maxTokens: Int\n\n var name: String {\n \"searchWeb\"\n }\n\n var description: String {\n \"Useful for when you need to answer questions about latest information.\"\n }\n\n var argumentSchema: JSONSchemaValue {\n let today = Self.dateFormatter.string(from: Date())\n return [\n .type: \"object\",\n .properties: [\n \"query\": [\n .type: \"string\",\n .description: \"the search query\",\n ],\n \"freshness\": [\n .type: \"string\",\n .description: .string(\n \"limit the search result to a specific range, use only when I ask the question about current events. Today is \\(today). Format: yyyy-MM-dd..yyyy-MM-dd\"\n ),\n .examples: [\"1919-10-20..1988-10-20\"],\n ],\n ],\n .required: [\"query\"],\n ]\n }\n\n func prepare(reportProgress: @escaping ReportProgress) async {\n await reportProgress(\"Searching..\")\n }\n\n func call(\n arguments: Arguments,\n reportProgress: @escaping ReportProgress\n ) async throws -> Result {\n await reportProgress(\"Searching \\(arguments.query)\")\n\n do {\n let search = WebSearchService(provider: .userPreferred)\n\n let result = try await search.search(query: arguments.query)\n\n await reportProgress(\"\"\"\n Finish searching \\(arguments.query)\n \\(\n result.webPages\n .map { \"- [\\($0.title)](\\($0.urlString))\" }\n .joined(separator: \"\\n\")\n )\n \"\"\")\n\n return .init(result: result)\n } catch {\n await reportProgress(\"Failed searching: \\(error.localizedDescription)\")\n throw error\n }\n }\n}\n\n" }, { "path": "Core/Sources/ChatContextCollectors/WebChatContextCollector/WebChatContextCollector.swift", "content": "import ChatBasic\nimport ChatContextCollector\nimport Foundation\nimport OpenAIService\n\npublic final class WebChatContextCollector: ChatContextCollector {\n var recentLinks = [String]()\n\n public init() {}\n\n public func generateContext(\n history: [ChatMessage],\n scopes: Set,\n content: String,\n configuration: ChatGPTConfiguration\n ) -> ChatContext {\n guard scopes.contains(.web) else { return .empty }\n let links = Self.detectLinks(from: history) + Self.detectLinks(from: content)\n let functions: [(any ChatGPTFunction)?] = [\n SearchFunction(maxTokens: configuration.maxTokens),\n // allow this function only when there is a link in the memory.\n links.isEmpty ? nil : QueryWebsiteFunction(),\n ]\n return .init(\n systemPrompt: \"You prefer to answer questions with latest content on the internet.\",\n retrievedContent: [],\n functions: functions.compactMap { $0 }\n )\n }\n}\n\nextension WebChatContextCollector {\n static func detectLinks(from messages: [ChatMessage]) -> [String] {\n return messages.lazy\n .compactMap {\n $0.content ?? $0.toolCalls?.map(\\.function.arguments).joined(separator: \" \") ?? \"\"\n }\n .map(detectLinks(from:))\n .flatMap { $0 }\n }\n\n static func detectLinks(from content: String) -> [String] {\n var links = [String]()\n let detector = try? NSDataDetector(types: NSTextCheckingResult.CheckingType.link.rawValue)\n let matches = detector?.matches(\n in: content,\n options: [],\n range: NSRange(content.startIndex..., in: content)\n )\n\n for match in matches ?? [] {\n guard let range = Range(match.range, in: content) else { continue }\n let url = content[range]\n links.append(String(url))\n }\n return links\n }\n}\n\n" }, { "path": "Core/Sources/ChatGPTChatTab/Chat.swift", "content": "import AppKit\nimport ChatBasic\nimport ChatService\nimport ComposableArchitecture\nimport Foundation\nimport MarkdownUI\nimport OpenAIService\nimport Preferences\nimport Terminal\n\npublic struct DisplayedChatMessage: Equatable {\n public enum Role: Equatable {\n case user\n case assistant\n case tool\n case ignored\n }\n\n public struct Reference: Equatable {\n public typealias Kind = ChatMessage.Reference.Kind\n\n public var title: String\n public var subtitle: String\n public var uri: String\n public var startLine: Int?\n public var kind: Kind\n\n public init(\n title: String,\n subtitle: String,\n uri: String,\n startLine: Int?,\n kind: Kind\n ) {\n self.title = title\n self.subtitle = subtitle\n self.uri = uri\n self.startLine = startLine\n self.kind = kind\n }\n }\n\n public var id: String\n public var role: Role\n public var text: String\n public var markdownContent: MarkdownContent\n public var references: [Reference] = []\n\n public init(id: String, role: Role, text: String, references: [Reference]) {\n self.id = id\n self.role = role\n self.text = text\n markdownContent = .init(text)\n self.references = references\n }\n}\n\nprivate var isPreview: Bool {\n ProcessInfo.processInfo.environment[\"XCODE_RUNNING_FOR_PREVIEWS\"] == \"1\"\n}\n\n@Reducer\nstruct Chat {\n public typealias MessageID = String\n\n @ObservableState\n struct State: Equatable {\n var title: String = \"Chat\"\n var typedMessage = \"\"\n var history: [DisplayedChatMessage] = []\n var isReceivingMessage = false\n var chatMenu = ChatMenu.State()\n var focusedField: Field?\n var isEnabled = true\n var isPinnedToBottom = true\n\n enum Field: String, Hashable {\n case textField\n }\n }\n\n enum Action: Equatable, BindableAction {\n case binding(BindingAction)\n\n case appear\n case refresh\n case setIsEnabled(Bool)\n case sendButtonTapped\n case returnButtonTapped\n case stopRespondingButtonTapped\n case clearButtonTap\n case deleteMessageButtonTapped(MessageID)\n case resendMessageButtonTapped(MessageID)\n case setAsExtraPromptButtonTapped(MessageID)\n case manuallyScrolledUp\n case scrollToBottomButtonTapped\n case focusOnTextField\n case referenceClicked(DisplayedChatMessage.Reference)\n\n case observeChatService\n case observeHistoryChange\n case observeIsReceivingMessageChange\n case observeSystemPromptChange\n case observeExtraSystemPromptChange\n case observeDefaultScopesChange\n\n case historyChanged\n case isReceivingMessageChanged\n case systemPromptChanged\n case extraSystemPromptChanged\n case defaultScopesChanged\n\n case chatMenu(ChatMenu.Action)\n }\n\n let service: ChatService\n let id = UUID()\n\n enum CancelID: Hashable {\n case observeHistoryChange(UUID)\n case observeIsReceivingMessageChange(UUID)\n case observeSystemPromptChange(UUID)\n case observeExtraSystemPromptChange(UUID)\n case observeDefaultScopesChange(UUID)\n case sendMessage(UUID)\n }\n\n var body: some ReducerOf {\n BindingReducer()\n\n Scope(state: \\.chatMenu, action: \\.chatMenu) {\n ChatMenu(service: service)\n }\n\n Reduce { state, action in\n switch action {\n case .appear:\n return .run { send in\n if isPreview { return }\n await send(.observeChatService)\n await send(.historyChanged)\n await send(.isReceivingMessageChanged)\n await send(.systemPromptChanged)\n await send(.extraSystemPromptChanged)\n await send(.focusOnTextField)\n await send(.refresh)\n }\n\n case .refresh:\n return .run { send in\n await send(.chatMenu(.refresh))\n }\n\n case let .setIsEnabled(isEnabled):\n state.isEnabled = isEnabled\n return .none\n\n case .sendButtonTapped:\n guard !state.typedMessage.isEmpty else { return .none }\n let message = state.typedMessage\n state.typedMessage = \"\"\n return .run { _ in\n try await service.send(content: message)\n }.cancellable(id: CancelID.sendMessage(id))\n\n case .returnButtonTapped:\n state.typedMessage += \"\\n\"\n return .none\n\n case .stopRespondingButtonTapped:\n return .merge(\n .run { _ in\n await service.stopReceivingMessage()\n },\n .cancel(id: CancelID.sendMessage(id))\n )\n\n case .clearButtonTap:\n return .run { _ in\n await service.clearHistory()\n }\n\n case let .deleteMessageButtonTapped(id):\n return .run { _ in\n await service.deleteMessage(id: id)\n }\n\n case let .resendMessageButtonTapped(id):\n return .run { _ in\n try await service.resendMessage(id: id)\n }\n\n case let .setAsExtraPromptButtonTapped(id):\n return .run { _ in\n await service.setMessageAsExtraPrompt(id: id)\n }\n\n case let .referenceClicked(reference):\n let fileURL = URL(fileURLWithPath: reference.uri)\n return .run { _ in\n if FileManager.default.fileExists(atPath: fileURL.path) {\n let terminal = Terminal()\n do {\n _ = try await terminal.runCommand(\n \"/bin/bash\",\n arguments: [\n \"-c\",\n \"xed -l \\(reference.startLine ?? 0) ${TARGET_FILE}\",\n ],\n environment: [\"TARGET_FILE\": reference.uri]\n )\n } catch {\n print(error)\n }\n } else if let url = URL(string: reference.uri), url.scheme != nil {\n NSWorkspace.shared.open(url)\n }\n }\n\n case .manuallyScrolledUp:\n state.isPinnedToBottom = false\n return .none\n\n case .scrollToBottomButtonTapped:\n state.isPinnedToBottom = true\n return .none\n\n case .focusOnTextField:\n state.focusedField = .textField\n return .none\n\n case .observeChatService:\n return .run { send in\n await send(.observeHistoryChange)\n await send(.observeIsReceivingMessageChange)\n await send(.observeSystemPromptChange)\n await send(.observeExtraSystemPromptChange)\n await send(.observeDefaultScopesChange)\n }\n\n case .observeHistoryChange:\n return .run { send in\n let stream = AsyncStream { continuation in\n let cancellable = service.$chatHistory.sink { _ in\n continuation.yield()\n }\n continuation.onTermination = { _ in\n cancellable.cancel()\n }\n }\n let debouncedHistoryChange = TimedDebounceFunction(duration: 0.2) {\n await send(.historyChanged)\n }\n\n for await _ in stream {\n await debouncedHistoryChange()\n }\n }.cancellable(id: CancelID.observeHistoryChange(id), cancelInFlight: true)\n\n case .observeIsReceivingMessageChange:\n return .run { send in\n let stream = AsyncStream { continuation in\n let cancellable = service.$isReceivingMessage\n .sink { _ in\n continuation.yield()\n }\n continuation.onTermination = { _ in\n cancellable.cancel()\n }\n }\n for await _ in stream {\n await send(.isReceivingMessageChanged)\n }\n }.cancellable(\n id: CancelID.observeIsReceivingMessageChange(id),\n cancelInFlight: true\n )\n\n case .observeSystemPromptChange:\n return .run { send in\n let stream = AsyncStream { continuation in\n let cancellable = service.$systemPrompt.sink { _ in\n continuation.yield()\n }\n continuation.onTermination = { _ in\n cancellable.cancel()\n }\n }\n for await _ in stream {\n await send(.systemPromptChanged)\n }\n }.cancellable(id: CancelID.observeSystemPromptChange(id), cancelInFlight: true)\n\n case .observeExtraSystemPromptChange:\n return .run { send in\n let stream = AsyncStream { continuation in\n let cancellable = service.$extraSystemPrompt\n .sink { _ in\n continuation.yield()\n }\n continuation.onTermination = { _ in\n cancellable.cancel()\n }\n }\n for await _ in stream {\n await send(.extraSystemPromptChanged)\n }\n }.cancellable(id: CancelID.observeExtraSystemPromptChange(id), cancelInFlight: true)\n\n case .observeDefaultScopesChange:\n return .run { send in\n let stream = AsyncStream { continuation in\n let cancellable = service.$defaultScopes\n .sink { _ in\n continuation.yield()\n }\n continuation.onTermination = { _ in\n cancellable.cancel()\n }\n }\n for await _ in stream {\n await send(.defaultScopesChanged)\n }\n }.cancellable(id: CancelID.observeDefaultScopesChange(id), cancelInFlight: true)\n\n case .historyChanged:\n state.history = service.chatHistory.flatMap { message in\n var all = [DisplayedChatMessage]()\n all.append(.init(\n id: message.id,\n role: {\n switch message.role {\n case .system: return .ignored\n case .user: return .user\n case .assistant:\n if let text = message.summary ?? message.content,\n !text.isEmpty\n {\n return .assistant\n }\n return .ignored\n }\n }(),\n text: message.summary ?? message.content ?? \"\",\n references: message.references.map(convertReference)\n ))\n\n for call in message.toolCalls ?? [] {\n all.append(.init(\n id: message.id + call.id,\n role: .tool,\n text: call.response.summary ?? call.response.content,\n references: []\n ))\n }\n\n return all\n }\n\n state.title = {\n let defaultTitle = \"Chat\"\n guard let lastMessageText = state.history\n .filter({ $0.role == .assistant || $0.role == .user })\n .last?\n .text else { return defaultTitle }\n if lastMessageText.isEmpty { return defaultTitle }\n let trimmed = lastMessageText\n .trimmingCharacters(in: .punctuationCharacters)\n .trimmingCharacters(in: .whitespacesAndNewlines)\n if trimmed.starts(with: \"```\") {\n return \"Code Block\"\n } else {\n return trimmed\n }\n }()\n return .none\n\n case .isReceivingMessageChanged:\n state.isReceivingMessage = service.isReceivingMessage\n if service.isReceivingMessage {\n state.isPinnedToBottom = true\n }\n return .none\n\n case .systemPromptChanged:\n state.chatMenu.systemPrompt = service.systemPrompt\n return .none\n\n case .extraSystemPromptChanged:\n state.chatMenu.extraSystemPrompt = service.extraSystemPrompt\n return .none\n\n case .defaultScopesChanged:\n state.chatMenu.defaultScopes = service.defaultScopes\n return .none\n\n case .binding:\n return .none\n\n case .chatMenu:\n return .none\n }\n }\n }\n}\n\n@Reducer\nstruct ChatMenu {\n @ObservableState\n struct State: Equatable {\n var systemPrompt: String = \"\"\n var extraSystemPrompt: String = \"\"\n var temperatureOverride: Double? = nil\n var chatModelIdOverride: String? = nil\n var defaultScopes: Set = []\n }\n\n enum Action: Equatable {\n case appear\n case refresh\n case resetPromptButtonTapped\n case temperatureOverrideSelected(Double?)\n case chatModelIdOverrideSelected(String?)\n case customCommandButtonTapped(CustomCommand)\n case resetDefaultScopesButtonTapped\n case toggleScope(ChatService.Scope)\n }\n\n let service: ChatService\n\n var body: some ReducerOf {\n Reduce { state, action in\n switch action {\n case .appear:\n return .run {\n await $0(.refresh)\n }\n\n case .refresh:\n state.temperatureOverride = service.configuration.overriding.temperature\n state.chatModelIdOverride = service.configuration.overriding.modelId\n return .none\n\n case .resetPromptButtonTapped:\n return .run { _ in\n await service.resetPrompt()\n }\n case let .temperatureOverrideSelected(temperature):\n state.temperatureOverride = temperature\n return .run { _ in\n service.configuration.overriding.temperature = temperature\n }\n case let .chatModelIdOverrideSelected(chatModelId):\n state.chatModelIdOverride = chatModelId\n return .run { _ in\n service.configuration.overriding.modelId = chatModelId\n }\n case let .customCommandButtonTapped(command):\n return .run { _ in\n try await service.handleCustomCommand(command)\n }\n\n case .resetDefaultScopesButtonTapped:\n return .run { _ in\n service.resetDefaultScopes()\n }\n case let .toggleScope(scope):\n return .run { _ in\n service.defaultScopes.formSymmetricDifference([scope])\n }\n }\n }\n }\n}\n\nprivate actor TimedDebounceFunction {\n let duration: TimeInterval\n let block: () async -> Void\n\n var task: Task?\n var lastFireTime: Date = .init(timeIntervalSince1970: 0)\n\n init(duration: TimeInterval, block: @escaping () async -> Void) {\n self.duration = duration\n self.block = block\n }\n\n func callAsFunction() async {\n task?.cancel()\n if lastFireTime.timeIntervalSinceNow < -duration {\n await fire()\n task = nil\n } else {\n task = Task.detached { [weak self, duration] in\n try await Task.sleep(nanoseconds: UInt64(duration * 1_000_000_000))\n await self?.fire()\n }\n }\n }\n\n func fire() async {\n lastFireTime = Date()\n await block()\n }\n}\n\nprivate func convertReference(\n _ reference: ChatMessage.Reference\n) -> DisplayedChatMessage.Reference {\n .init(\n title: reference.title,\n subtitle: {\n switch reference.kind {\n case let .symbol(_, uri, _, _):\n return uri\n case let .webpage(uri):\n return uri\n case let .textFile(uri):\n return uri\n case let .other(kind):\n return kind\n case .text:\n return reference.content\n case .error:\n return reference.content\n }\n }(),\n uri: {\n switch reference.kind {\n case let .symbol(_, uri, _, _):\n return uri\n case let .webpage(uri):\n return uri\n case let .textFile(uri):\n return uri\n case .other:\n return \"\"\n case .text:\n return \"\"\n case .error:\n return \"\"\n }\n }(),\n startLine: {\n switch reference.kind {\n case let .symbol(_, _, startLine, _):\n return startLine\n default:\n return nil\n }\n }(),\n kind: reference.kind\n )\n}\n\n" }, { "path": "Core/Sources/ChatGPTChatTab/ChatContextMenu.swift", "content": "import AppKit\nimport ChatService\nimport ComposableArchitecture\nimport SharedUIComponents\nimport SwiftUI\n\nstruct ChatTabItemView: View {\n let chat: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n Text(chat.title)\n }\n }\n}\n\nstruct ChatContextMenu: View {\n let store: StoreOf\n @AppStorage(\\.customCommands) var customCommands\n @AppStorage(\\.chatModels) var chatModels\n @AppStorage(\\.defaultChatFeatureChatModelId) var defaultChatModelId\n @AppStorage(\\.chatGPTTemperature) var defaultTemperature\n\n var body: some View {\n WithPerceptionTracking {\n currentSystemPrompt\n .onAppear { store.send(.appear) }\n currentExtraSystemPrompt\n resetPrompt\n\n Divider()\n\n chatModel\n temperature\n defaultScopes\n\n Divider()\n\n customCommandMenu\n }\n }\n\n @ViewBuilder\n var currentSystemPrompt: some View {\n Text(\"System Prompt:\")\n Text({\n var text = store.systemPrompt\n if text.isEmpty { text = \"N/A\" }\n if text.count > 30 { text = String(text.prefix(30)) + \"...\" }\n return text\n }() as String)\n }\n\n @ViewBuilder\n var currentExtraSystemPrompt: some View {\n Text(\"Extra Prompt:\")\n Text({\n var text = store.extraSystemPrompt\n if text.isEmpty { text = \"N/A\" }\n if text.count > 30 { text = String(text.prefix(30)) + \"...\" }\n return text\n }() as String)\n }\n\n var resetPrompt: some View {\n Button(\"Reset System Prompt\") {\n store.send(.resetPromptButtonTapped)\n }\n }\n\n @ViewBuilder\n var chatModel: some View {\n let allModels = chatModels + [.init(\n id: \"com.github.copilot\",\n name: \"GitHub Copilot Language Server\",\n format: .openAI,\n info: .init()\n )]\n \n Menu(\"Chat Model\") {\n Button(action: {\n store.send(.chatModelIdOverrideSelected(nil))\n }) {\n HStack {\n if let defaultModel = allModels\n .first(where: { $0.id == defaultChatModelId })\n {\n Text(\"Default (\\(defaultModel.name))\")\n if store.chatModelIdOverride == nil {\n Image(systemName: \"checkmark\")\n }\n } else {\n Text(\"No Model Available\")\n }\n }\n }\n\n if let id = store.chatModelIdOverride, !allModels.map(\\.id).contains(id) {\n Button(action: {\n store.send(.chatModelIdOverrideSelected(nil))\n }) {\n HStack {\n Text(\"Default (Selected Model Not Found)\")\n Image(systemName: \"checkmark\")\n }\n }\n }\n\n Divider()\n\n ForEach(allModels, id: \\.id) { model in\n Button(action: {\n store.send(.chatModelIdOverrideSelected(model.id))\n }) {\n HStack {\n Text(model.name)\n if model.id == store.chatModelIdOverride {\n Image(systemName: \"checkmark\")\n }\n }\n }\n }\n }\n }\n\n @ViewBuilder\n var temperature: some View {\n Menu(\"Temperature\") {\n Button(action: {\n store.send(.temperatureOverrideSelected(nil))\n }) {\n HStack {\n Text(\n \"Default (\\(defaultTemperature.formatted(.number.precision(.fractionLength(1)))))\"\n )\n if store.temperatureOverride == nil {\n Image(systemName: \"checkmark\")\n }\n }\n }\n\n Divider()\n\n ForEach(Array(stride(from: 0.0, through: 2.0, by: 0.1)), id: \\.self) { value in\n Button(action: {\n store.send(.temperatureOverrideSelected(value))\n }) {\n HStack {\n Text(\"\\(value.formatted(.number.precision(.fractionLength(1))))\")\n if value == store.temperatureOverride {\n Image(systemName: \"checkmark\")\n }\n }\n }\n }\n }\n }\n\n @ViewBuilder\n var defaultScopes: some View {\n Menu(\"Default Scopes\") {\n Button(action: {\n store.send(.resetDefaultScopesButtonTapped)\n }) {\n Text(\"Reset Default Scopes\")\n }\n\n Divider()\n\n ForEach(ChatService.Scope.allCases, id: \\.rawValue) { value in\n Button(action: {\n store.send(.toggleScope(value))\n }) {\n HStack {\n Text(\"@\" + value.rawValue)\n if store.defaultScopes.contains(value) {\n Image(systemName: \"checkmark\")\n }\n }\n }\n }\n }\n }\n\n var customCommandMenu: some View {\n Menu(\"Custom Commands\") {\n ForEach(\n customCommands.filter {\n switch $0.feature {\n case .chatWithSelection, .customChat: return true\n case .promptToCode: return false\n case .singleRoundDialog: return false\n }\n },\n id: \\.name\n ) { command in\n Button(action: {\n store.send(.customCommandButtonTapped(command))\n }) {\n Text(command.name)\n }\n }\n }\n }\n}\n\n" }, { "path": "Core/Sources/ChatGPTChatTab/ChatGPTChatTab.swift", "content": "import ChatContextCollector\nimport ChatService\nimport ChatTab\nimport CodableWrappers\nimport Combine\nimport ComposableArchitecture\nimport DebounceFunction\nimport Foundation\nimport OpenAIService\nimport Preferences\nimport SwiftUI\n\n/// A chat tab that provides a context aware chat bot, powered by ChatGPT.\npublic class ChatGPTChatTab: ChatTab {\n public static var name: String { \"Chat\" }\n\n public let service: ChatService\n let chat: StoreOf\n private var cancellable = Set()\n private var observer = NSObject()\n private let updateContentDebounce = DebounceRunner(duration: 0.5)\n\n struct RestorableState: Codable {\n var history: [OpenAIService.ChatMessage]\n var configuration: OverridingChatGPTConfiguration.Overriding\n var systemPrompt: String\n var extraSystemPrompt: String\n var defaultScopes: Set?\n }\n\n struct Builder: ChatTabBuilder {\n var title: String\n var customCommand: CustomCommand?\n var afterBuild: (ChatGPTChatTab) async -> Void = { _ in }\n\n func build(store: StoreOf) async -> (any ChatTab)? {\n let tab = await ChatGPTChatTab(store: store)\n if let customCommand {\n try? await tab.service.handleCustomCommand(customCommand)\n }\n await afterBuild(tab)\n return tab\n }\n }\n\n public func buildView() -> any View {\n ChatPanel(chat: chat)\n }\n\n public func buildTabItem() -> any View {\n ChatTabItemView(chat: chat)\n }\n\n public func buildIcon() -> any View {\n WithPerceptionTracking {\n if self.chat.isReceivingMessage {\n Image(systemName: \"ellipsis.message\")\n } else {\n Image(systemName: \"message\")\n }\n }\n }\n\n public func buildMenu() -> any View {\n ChatContextMenu(store: chat.scope(state: \\.chatMenu, action: \\.chatMenu))\n }\n\n public func restorableState() async -> Data {\n let state = RestorableState(\n history: await service.memory.history,\n configuration: service.configuration.overriding,\n systemPrompt: service.systemPrompt,\n extraSystemPrompt: service.extraSystemPrompt,\n defaultScopes: service.defaultScopes\n )\n return (try? JSONEncoder().encode(state)) ?? Data()\n }\n\n public static func restore(from data: Data) async throws -> any ChatTabBuilder {\n let state = try JSONDecoder().decode(RestorableState.self, from: data)\n let builder = Builder(title: \"Chat\") { @MainActor tab in\n tab.service.configuration.overriding = state.configuration\n tab.service.mutateSystemPrompt(state.systemPrompt)\n tab.service.mutateExtraSystemPrompt(state.extraSystemPrompt)\n if let scopes = state.defaultScopes {\n tab.service.defaultScopes = scopes\n }\n await tab.service.memory.mutateHistory { history in\n history = state.history\n }\n tab.chat.send(.refresh)\n }\n return builder\n }\n\n public static func chatBuilders() -> [ChatTabBuilder] {\n let customCommands = UserDefaults.shared.value(for: \\.customCommands).compactMap {\n command in\n if case .customChat = command.feature {\n return Builder(title: command.name, customCommand: command)\n }\n return nil\n }\n\n return [Builder(title: \"Legacy Chat\", customCommand: nil)] + customCommands\n }\n\n public static func defaultBuilder() -> ChatTabBuilder {\n Builder(title: \"Legacy Chat\", customCommand: nil)\n }\n\n @MainActor\n public init(service: ChatService = .init(), store: StoreOf) {\n self.service = service\n chat = .init(initialState: .init(), reducer: { Chat(service: service) })\n super.init(store: store)\n }\n\n public func start() {\n observer = .init()\n cancellable = []\n\n chatTabStore.send(.updateTitle(\"Chat\"))\n\n service.$systemPrompt.removeDuplicates().sink { [weak self] _ in\n Task { @MainActor [weak self] in\n self?.chatTabStore.send(.tabContentUpdated)\n }\n }.store(in: &cancellable)\n\n service.$extraSystemPrompt.removeDuplicates().sink { [weak self] _ in\n Task { @MainActor [weak self] in\n self?.chatTabStore.send(.tabContentUpdated)\n }\n }.store(in: &cancellable)\n\n Task { @MainActor in\n var lastTrigger = -1\n observer.observe { [weak self] in\n guard let self else { return }\n let trigger = chatTabStore.focusTrigger\n guard lastTrigger != trigger else { return }\n lastTrigger = trigger\n Task { @MainActor [weak self] in\n self?.chat.send(.focusOnTextField)\n }\n }\n }\n\n Task { @MainActor in\n var lastTitle = \"\"\n observer.observe { [weak self] in\n guard let self else { return }\n let title = self.chatTabStore.state.title\n guard lastTitle != title else { return }\n lastTitle = title\n Task { @MainActor [weak self] in\n self?.chatTabStore.send(.updateTitle(title))\n }\n }\n }\n\n Task { @MainActor in\n observer.observe { [weak self] in\n guard let self else { return }\n _ = chat.history\n _ = chat.title\n _ = chat.isReceivingMessage\n Task {\n await self.updateContentDebounce.debounce { @MainActor [weak self] in\n self?.chatTabStore.send(.tabContentUpdated)\n }\n }\n }\n }\n }\n\n public func handleCustomCommand(_ customCommand: CustomCommand) -> Bool {\n Task {\n if service.isReceivingMessage {\n await service.stopReceivingMessage()\n }\n try? await service.handleCustomCommand(customCommand)\n }\n return true\n }\n}\n\n" }, { "path": "Core/Sources/ChatGPTChatTab/ChatPanel.swift", "content": "import AppKit\nimport Combine\nimport ComposableArchitecture\nimport MarkdownUI\nimport OpenAIService\nimport SharedUIComponents\nimport SwiftUI\n\nprivate let r: Double = 8\n\npublic struct ChatPanel: View {\n let chat: StoreOf\n @Namespace var inputAreaNamespace\n\n public var body: some View {\n VStack(spacing: 0) {\n ChatPanelMessages(chat: chat)\n Divider()\n ChatPanelInputArea(chat: chat)\n }\n .background(Color(nsColor: .windowBackgroundColor))\n .onAppear { chat.send(.appear) }\n }\n}\n\nprivate struct ScrollViewOffsetPreferenceKey: PreferenceKey {\n static var defaultValue = CGFloat.zero\n\n static func reduce(value: inout CGFloat, nextValue: () -> CGFloat) {\n value += nextValue()\n }\n}\n\nprivate struct ListHeightPreferenceKey: PreferenceKey {\n static var defaultValue = CGFloat.zero\n\n static func reduce(value: inout CGFloat, nextValue: () -> CGFloat) {\n value += nextValue()\n }\n}\n\nstruct ChatPanelMessages: View {\n let chat: StoreOf\n @State var cancellable = Set()\n @State var isScrollToBottomButtonDisplayed = true\n @Namespace var bottomID\n @Namespace var topID\n @Namespace var scrollSpace\n @State var scrollOffset: Double = 0\n @State var listHeight: Double = 0\n @State var didScrollToBottomOnAppearOnce = false\n @State var isBottomHidden = true\n @Environment(\\.isEnabled) var isEnabled\n\n var body: some View {\n WithPerceptionTracking {\n ScrollViewReader { proxy in\n GeometryReader { listGeo in\n List {\n Group {\n Spacer(minLength: 12)\n .id(topID)\n\n Instruction(chat: chat)\n\n ChatHistory(chat: chat)\n .listItemTint(.clear)\n\n ExtraSpacingInResponding(chat: chat)\n\n Spacer(minLength: 12)\n .id(bottomID)\n .onAppear {\n isBottomHidden = false\n if !didScrollToBottomOnAppearOnce {\n proxy.scrollTo(bottomID, anchor: .bottom)\n didScrollToBottomOnAppearOnce = true\n }\n }\n .onDisappear {\n isBottomHidden = true\n }\n .background(GeometryReader { geo in\n let offset = geo.frame(in: .named(scrollSpace)).minY\n Color.clear.preference(\n key: ScrollViewOffsetPreferenceKey.self,\n value: offset\n )\n })\n }\n .modify { view in\n if #available(macOS 13.0, *) {\n view\n .listRowSeparator(.hidden)\n .listSectionSeparator(.hidden)\n } else {\n view\n }\n }\n }\n .listStyle(.plain)\n .listRowBackground(EmptyView())\n .modify { view in\n if #available(macOS 13.0, *) {\n view.scrollContentBackground(.hidden)\n } else {\n view\n }\n }\n .coordinateSpace(name: scrollSpace)\n .preference(\n key: ListHeightPreferenceKey.self,\n value: listGeo.size.height\n )\n .onPreferenceChange(ListHeightPreferenceKey.self) { value in\n listHeight = value\n updatePinningState()\n }\n .onPreferenceChange(ScrollViewOffsetPreferenceKey.self) { value in\n scrollOffset = value\n updatePinningState()\n }\n .overlay(alignment: .bottom) {\n StopRespondingButton(chat: chat)\n }\n .overlay(alignment: .bottomTrailing) {\n scrollToBottomButton(proxy: proxy)\n }\n .background {\n PinToBottomHandler(chat: chat, isBottomHidden: isBottomHidden) {\n proxy.scrollTo(bottomID, anchor: .bottom)\n }\n }\n .onAppear {\n proxy.scrollTo(bottomID, anchor: .bottom)\n }\n .task {\n proxy.scrollTo(bottomID, anchor: .bottom)\n }\n }\n }\n .onAppear {\n trackScrollWheel()\n }\n .onDisappear {\n cancellable.forEach { $0.cancel() }\n cancellable = []\n }\n .onChange(of: isEnabled) { isEnabled in\n chat.send(.setIsEnabled(isEnabled))\n }\n }\n }\n\n func trackScrollWheel() {\n NSApplication.shared.publisher(for: \\.currentEvent)\n .receive(on: DispatchQueue.main)\n .filter { [chat] in\n guard chat.withState(\\.isEnabled) else { return false }\n return $0?.type == .scrollWheel\n }\n .compactMap { $0 }\n .sink { event in\n guard chat.withState(\\.isPinnedToBottom) else { return }\n let delta = event.deltaY\n let scrollUp = delta > 0\n if scrollUp {\n chat.send(.manuallyScrolledUp)\n }\n }\n .store(in: &cancellable)\n }\n\n @MainActor\n func updatePinningState() {\n // where does the 32 come from?\n withAnimation(.linear(duration: 0.1)) {\n isScrollToBottomButtonDisplayed = scrollOffset > listHeight + 32 + 20\n || scrollOffset <= 0\n }\n }\n\n @ViewBuilder\n func scrollToBottomButton(proxy: ScrollViewProxy) -> some View {\n Button(action: {\n chat.send(.scrollToBottomButtonTapped)\n withAnimation(.easeInOut(duration: 0.1)) {\n proxy.scrollTo(bottomID, anchor: .bottom)\n }\n }) {\n Image(systemName: \"arrow.down\")\n .padding(4)\n .background {\n Circle()\n .fill(.thickMaterial)\n .shadow(color: .black.opacity(0.2), radius: 2)\n }\n .overlay {\n Circle().stroke(Color(nsColor: .separatorColor), lineWidth: 1)\n }\n .foregroundStyle(.secondary)\n .padding(4)\n }\n .keyboardShortcut(.downArrow, modifiers: [.command])\n .opacity(isScrollToBottomButtonDisplayed ? 1 : 0)\n .buttonStyle(.plain)\n }\n\n struct ExtraSpacingInResponding: View {\n let chat: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n if chat.isReceivingMessage {\n Spacer(minLength: 12)\n }\n }\n }\n }\n\n struct PinToBottomHandler: View {\n let chat: StoreOf\n let isBottomHidden: Bool\n let scrollToBottom: () -> Void\n\n @State var isInitialLoad = true\n\n var body: some View {\n WithPerceptionTracking {\n EmptyView()\n .onChange(of: chat.isReceivingMessage) { isReceiving in\n if isReceiving {\n Task {\n await Task.yield()\n withAnimation(.easeInOut(duration: 0.1)) {\n scrollToBottom()\n }\n }\n }\n }\n .onChange(of: chat.history.last) { _ in\n if chat.withState(\\.isPinnedToBottom) || isInitialLoad {\n if isInitialLoad {\n isInitialLoad = false\n }\n Task {\n await Task.yield()\n withAnimation(.easeInOut(duration: 0.1)) {\n scrollToBottom()\n }\n }\n }\n }\n .onChange(of: isBottomHidden) { value in\n // This is important to prevent it from jumping to the top!\n if value, chat.withState(\\.isPinnedToBottom) {\n scrollToBottom()\n }\n }\n }\n }\n }\n}\n\nstruct ChatHistory: View {\n let chat: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n ForEach(chat.history, id: \\.id) { message in\n WithPerceptionTracking {\n ChatHistoryItem(chat: chat, message: message).id(message.id)\n }\n }\n }\n }\n}\n\nstruct ChatHistoryItem: View {\n let chat: StoreOf\n let message: DisplayedChatMessage\n\n var body: some View {\n WithPerceptionTracking {\n let text = message.text\n let markdownContent = message.markdownContent\n switch message.role {\n case .user:\n UserMessage(\n id: message.id,\n text: text,\n markdownContent: markdownContent,\n chat: chat\n )\n .listRowInsets(EdgeInsets(\n top: 0,\n leading: -8,\n bottom: 0,\n trailing: -8\n ))\n .padding(.vertical, 4)\n case .assistant:\n BotMessage(\n id: message.id,\n text: text,\n markdownContent: markdownContent,\n references: message.references,\n chat: chat\n )\n .listRowInsets(EdgeInsets(\n top: 0,\n leading: -8,\n bottom: 0,\n trailing: -8\n ))\n .padding(.vertical, 4)\n case .tool:\n FunctionMessage(id: message.id, text: text)\n case .ignored:\n EmptyView()\n }\n }\n }\n}\n\nprivate struct StopRespondingButton: View {\n let chat: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n if chat.isReceivingMessage {\n Button(action: {\n chat.send(.stopRespondingButtonTapped)\n }) {\n HStack(spacing: 4) {\n Image(systemName: \"stop.fill\")\n Text(\"Stop Responding\")\n }\n .padding(8)\n .background(\n .regularMaterial,\n in: RoundedRectangle(cornerRadius: r, style: .continuous)\n )\n .overlay {\n RoundedRectangle(cornerRadius: r, style: .continuous)\n .stroke(Color(nsColor: .separatorColor), lineWidth: 1)\n }\n }\n .buttonStyle(.borderless)\n .frame(maxWidth: .infinity, alignment: .center)\n .padding(.bottom, 8)\n .opacity(chat.isReceivingMessage ? 1 : 0)\n .disabled(!chat.isReceivingMessage)\n .transformEffect(.init(\n translationX: 0,\n y: chat.isReceivingMessage ? 0 : 20\n ))\n }\n }\n }\n}\n\nstruct ChatPanelInputArea: View {\n let chat: StoreOf\n @FocusState var focusedField: Chat.State.Field?\n\n var body: some View {\n HStack {\n clearButton\n InputAreaTextEditor(chat: chat, focusedField: $focusedField)\n }\n .padding(8)\n .background(.ultraThickMaterial)\n }\n\n @MainActor\n var clearButton: some View {\n Button(action: {\n chat.send(.clearButtonTap)\n }) {\n Group {\n if #available(macOS 13.0, *) {\n Image(systemName: \"eraser.line.dashed.fill\")\n } else {\n Image(systemName: \"trash.fill\")\n }\n }\n .padding(6)\n .background {\n Circle().fill(Color(nsColor: .controlBackgroundColor))\n }\n .overlay {\n Circle().stroke(Color(nsColor: .controlColor), lineWidth: 1)\n }\n }\n .buttonStyle(.plain)\n }\n\n struct InputAreaTextEditor: View {\n @Perception.Bindable var chat: StoreOf\n var focusedField: FocusState.Binding\n\n var body: some View {\n WithPerceptionTracking {\n HStack(spacing: 0) {\n AutoresizingCustomTextEditor(\n text: $chat.typedMessage,\n font: .systemFont(ofSize: 14),\n isEditable: true,\n maxHeight: 400,\n onSubmit: { chat.send(.sendButtonTapped) },\n completions: chatAutoCompletion\n )\n .focused(focusedField, equals: .textField)\n .bind($chat.focusedField, to: focusedField)\n .padding(8)\n .fixedSize(horizontal: false, vertical: true)\n\n Button(action: {\n chat.send(.sendButtonTapped)\n }) {\n Image(systemName: \"paperplane.fill\")\n .padding(8)\n }\n .buttonStyle(.plain)\n .disabled(chat.isReceivingMessage)\n .keyboardShortcut(KeyEquivalent.return, modifiers: [])\n }\n .frame(maxWidth: .infinity)\n .background {\n RoundedRectangle(cornerRadius: 6)\n .fill(Color(nsColor: .controlBackgroundColor))\n }\n .overlay {\n RoundedRectangle(cornerRadius: 6)\n .stroke(Color(nsColor: .controlColor), lineWidth: 1)\n }\n .background {\n Button(action: {\n chat.send(.returnButtonTapped)\n }) {\n EmptyView()\n }\n .keyboardShortcut(KeyEquivalent.return, modifiers: [.shift])\n\n Button(action: {\n focusedField.wrappedValue = .textField\n }) {\n EmptyView()\n }\n .keyboardShortcut(\"l\", modifiers: [.command])\n }\n }\n }\n\n func chatAutoCompletion(text: String, proposed: [String], range: NSRange) -> [String] {\n guard text.count == 1 else { return [] }\n let plugins = [String]() // chat.pluginIdentifiers.map { \"/\\($0)\" }\n let availableFeatures = plugins + [\n \"/exit\",\n \"@code\",\n \"@sense\",\n \"@project\",\n \"@web\",\n ]\n\n let result: [String] = availableFeatures\n .filter { $0.hasPrefix(text) && $0 != text }\n .compactMap {\n guard let index = $0.index(\n $0.startIndex,\n offsetBy: range.location,\n limitedBy: $0.endIndex\n ) else { return nil }\n return String($0[index...])\n }\n return result\n }\n }\n}\n\n// MARK: - Previews\n\nstruct ChatPanel_Preview: PreviewProvider {\n static let history: [DisplayedChatMessage] = [\n .init(\n id: \"1\",\n role: .user,\n text: \"**Hello**\",\n references: []\n ),\n .init(\n id: \"2\",\n role: .assistant,\n text: \"\"\"\n ```swift\n func foo() {}\n ```\n **Hey**! What can I do for you?**Hey**! What can I do for you?**Hey**! What can I do for you?**Hey**! What can I do for you?\n \"\"\",\n references: [\n .init(\n title: \"Hello Hello Hello Hello\",\n subtitle: \"Hi Hi Hi Hi\",\n uri: \"https://google.com\",\n startLine: nil,\n kind: .symbol(.class, uri: \"https://google.com\", startLine: nil, endLine: nil)\n ),\n ]\n ),\n .init(\n id: \"7\",\n role: .ignored,\n text: \"Ignored\",\n references: []\n ),\n .init(\n id: \"6\",\n role: .tool,\n text: \"\"\"\n Searching for something...\n - abc\n - [def](https://1.com)\n > hello\n > hi\n \"\"\",\n references: []\n ),\n .init(\n id: \"5\",\n role: .assistant,\n text: \"Yooo\",\n references: []\n ),\n .init(\n id: \"4\",\n role: .user,\n text: \"Yeeeehh\",\n references: []\n ),\n .init(\n id: \"3\",\n role: .user,\n text: #\"\"\"\n Please buy me a coffee!\n | Coffee | Milk |\n |--------|------|\n | Espresso | No |\n | Latte | Yes |\n\n ```swift\n func foo() {}\n ```\n ```objectivec\n - (void)bar {}\n ```\n \"\"\"#,\n references: []\n ),\n ]\n\n static var previews: some View {\n ChatPanel(chat: .init(\n initialState: .init(history: ChatPanel_Preview.history, isReceivingMessage: true),\n reducer: { Chat(service: .init()) }\n ))\n .frame(width: 450, height: 1200)\n .colorScheme(.dark)\n }\n}\n\nstruct ChatPanel_EmptyChat_Preview: PreviewProvider {\n static var previews: some View {\n ChatPanel(chat: .init(\n initialState: .init(history: [DisplayedChatMessage](), isReceivingMessage: false),\n reducer: { Chat(service: .init()) }\n ))\n .padding()\n .frame(width: 450, height: 600)\n .colorScheme(.dark)\n }\n}\n\nstruct ChatPanel_InputText_Preview: PreviewProvider {\n static var previews: some View {\n ChatPanel(chat: .init(\n initialState: .init(history: ChatPanel_Preview.history, isReceivingMessage: false),\n reducer: { Chat(service: .init()) }\n ))\n .padding()\n .frame(width: 450, height: 600)\n .colorScheme(.dark)\n }\n}\n\nstruct ChatPanel_InputMultilineText_Preview: PreviewProvider {\n static var previews: some View {\n ChatPanel(\n chat: .init(\n initialState: .init(\n typedMessage: \"Lorem ipsum dolor sit amet, consectetur adipiscing elit. Fusce turpis dolor, malesuada quis fringilla sit amet, placerat at nunc. Suspendisse orci tortor, tempor nec blandit a, malesuada vel tellus. Nunc sed leo ligula. Ut at ligula eget turpis pharetra tristique. Integer luctus leo non elit rhoncus fermentum.\",\n\n history: ChatPanel_Preview.history,\n isReceivingMessage: false\n ),\n reducer: { Chat(service: .init()) }\n )\n )\n .padding()\n .frame(width: 450, height: 600)\n .colorScheme(.dark)\n }\n}\n\nstruct ChatPanel_Light_Preview: PreviewProvider {\n static var previews: some View {\n ChatPanel(chat: .init(\n initialState: .init(history: ChatPanel_Preview.history, isReceivingMessage: true),\n reducer: { Chat(service: .init()) }\n ))\n .padding()\n .frame(width: 450, height: 600)\n .colorScheme(.light)\n }\n}\n\n" }, { "path": "Core/Sources/ChatGPTChatTab/CodeBlockHighlighter.swift", "content": "import Combine\nimport ComposableArchitecture\nimport DebounceFunction\nimport Foundation\nimport MarkdownUI\nimport Perception\nimport SharedUIComponents\nimport SwiftUI\n\n/// Use this instead of the built in ``CodeBlockView`` to highlight code blocks asynchronously,\n/// so that the UI doesn't freeze when rendering large code blocks.\nstruct AsyncCodeBlockView: View {\n @Perceptible\n class Storage {\n static let queue = DispatchQueue(\n label: \"chat-code-block-highlight\",\n qos: .userInteractive,\n attributes: .concurrent\n )\n\n var highlighted: AttributedString?\n @PerceptionIgnored var debounceFunction: DebounceFunction?\n @PerceptionIgnored private var highlightTask: Task?\n\n init() {\n debounceFunction = .init(duration: 0.5, block: { [weak self] view in\n self?.highlight(for: view)\n })\n }\n\n func highlight(debounce: Bool, for view: AsyncCodeBlockView) {\n if debounce {\n Task { await debounceFunction?(view) }\n } else {\n highlight(for: view)\n }\n }\n\n func highlight(for view: AsyncCodeBlockView) {\n highlightTask?.cancel()\n let content = view.content\n let language = view.fenceInfo ?? \"\"\n let brightMode = view.colorScheme != .dark\n let font = CodeHighlighting.SendableFont(font: view.font)\n highlightTask = Task {\n let string = await withUnsafeContinuation { continuation in\n Self.queue.async {\n let content = CodeHighlighting.highlightedCodeBlock(\n code: content,\n language: language,\n scenario: \"chat\",\n brightMode: brightMode,\n font:font\n )\n continuation.resume(returning: AttributedString(content))\n }\n }\n try Task.checkCancellation()\n await MainActor.run {\n self.highlighted = string\n }\n }\n }\n }\n\n let fenceInfo: String?\n let content: String\n let font: NSFont\n\n @Environment(\\.colorScheme) var colorScheme\n @State var storage = Storage()\n @AppStorage(\\.syncChatCodeHighlightTheme) var syncCodeHighlightTheme\n @AppStorage(\\.codeForegroundColorLight) var codeForegroundColorLight\n @AppStorage(\\.codeBackgroundColorLight) var codeBackgroundColorLight\n @AppStorage(\\.codeForegroundColorDark) var codeForegroundColorDark\n @AppStorage(\\.codeBackgroundColorDark) var codeBackgroundColorDark\n\n init(fenceInfo: String?, content: String, font: NSFont) {\n self.fenceInfo = fenceInfo\n self.content = content.hasSuffix(\"\\n\") ? String(content.dropLast()) : content\n self.font = font\n }\n\n var body: some View {\n WithPerceptionTracking {\n Group {\n if let highlighted = storage.highlighted {\n Text(highlighted)\n } else {\n Text(content).font(.init(font))\n }\n }\n .onAppear {\n storage.highlight(debounce: false, for: self)\n }\n .onChange(of: colorScheme) { _ in\n storage.highlight(debounce: false, for: self)\n }\n .onChange(of: syncCodeHighlightTheme) { _ in\n storage.highlight(debounce: true, for: self)\n }\n .onChange(of: codeForegroundColorLight) { _ in\n storage.highlight(debounce: true, for: self)\n }\n .onChange(of: codeBackgroundColorLight) { _ in\n storage.highlight(debounce: true, for: self)\n }\n .onChange(of: codeForegroundColorDark) { _ in\n storage.highlight(debounce: true, for: self)\n }\n .onChange(of: codeBackgroundColorDark) { _ in\n storage.highlight(debounce: true, for: self)\n }\n }\n }\n}\n\n" }, { "path": "Core/Sources/ChatGPTChatTab/Styles.swift", "content": "import AppKit\nimport MarkdownUI\nimport SharedUIComponents\nimport SwiftUI\n\nextension Color {\n static var contentBackground: Color {\n Color(nsColor: NSColor(name: nil, dynamicProvider: { appearance in\n if appearance.isDarkMode {\n return #colorLiteral(red: 0.1580096483, green: 0.1730263829, blue: 0.2026666105, alpha: 1)\n }\n return #colorLiteral(red: 0.9896564803, green: 0.9896564803, blue: 0.9896564803, alpha: 1)\n }))\n }\n\n static var userChatContentBackground: Color {\n Color(nsColor: NSColor(name: nil, dynamicProvider: { appearance in\n if appearance.isDarkMode {\n return #colorLiteral(red: 0.2284317913, green: 0.2145925438, blue: 0.3214019983, alpha: 1)\n }\n return #colorLiteral(red: 0.9458052187, green: 0.9311983998, blue: 0.9906365955, alpha: 1)\n }))\n }\n}\n\nextension NSAppearance {\n var isDarkMode: Bool {\n if bestMatch(from: [.darkAqua, .aqua]) == .darkAqua {\n return true\n } else {\n return false\n }\n }\n}\n\nextension View {\n var messageBubbleCornerRadius: Double { 8 }\n\n func codeBlockLabelStyle() -> some View {\n relativeLineSpacing(.em(0.225))\n .markdownTextStyle {\n FontFamilyVariant(.monospaced)\n FontSize(.em(0.85))\n }\n .padding(16)\n .padding(.top, 14)\n }\n\n func codeBlockStyle(\n _ configuration: CodeBlockConfiguration,\n backgroundColor: Color,\n labelColor: Color\n ) -> some View {\n background(backgroundColor)\n .clipShape(RoundedRectangle(cornerRadius: 6))\n .overlay(alignment: .top) {\n HStack(alignment: .center) {\n Text(configuration.language ?? \"code\")\n .foregroundStyle(labelColor)\n .font(.callout.bold())\n .padding(.leading, 8)\n .lineLimit(1)\n Spacer()\n CopyButton {\n NSPasteboard.general.clearContents()\n NSPasteboard.general.setString(configuration.content, forType: .string)\n }\n }\n }\n .overlay {\n RoundedRectangle(cornerRadius: 6).stroke(Color.primary.opacity(0.05), lineWidth: 1)\n }\n .markdownMargin(top: 4, bottom: 16)\n }\n}\n\nfinal class VerticalScrollingFixHostingView: NSHostingView where Content: View {\n override func wantsForwardedScrollEvents(for axis: NSEvent.GestureAxis) -> Bool {\n return axis == .vertical\n }\n}\n\nstruct VerticalScrollingFixViewRepresentable: NSViewRepresentable where Content: View {\n let content: Content\n\n func makeNSView(context: Context) -> NSHostingView {\n return VerticalScrollingFixHostingView(rootView: content)\n }\n\n func updateNSView(_ nsView: NSHostingView, context: Context) {}\n}\n\nstruct VerticalScrollingFixWrapper: View where Content: View {\n let content: () -> Content\n\n init(@ViewBuilder content: @escaping () -> Content) {\n self.content = content\n }\n\n var body: some View {\n VerticalScrollingFixViewRepresentable(content: self.content())\n }\n}\n\nextension View {\n /// https://stackoverflow.com/questions/64920744/swiftui-nested-scrollviews-problem-on-macos\n @ViewBuilder func workaroundForVerticalScrollingBugInMacOS() -> some View {\n VerticalScrollingFixWrapper { self }\n }\n}\n\nstruct RoundedCorners: Shape {\n var tl: CGFloat = 0.0\n var tr: CGFloat = 0.0\n var bl: CGFloat = 0.0\n var br: CGFloat = 0.0\n\n func path(in rect: CGRect) -> Path {\n Path { path in\n\n let w = rect.size.width\n let h = rect.size.height\n\n // Make sure we do not exceed the size of the rectangle\n let tr = min(min(self.tr, h / 2), w / 2)\n let tl = min(min(self.tl, h / 2), w / 2)\n let bl = min(min(self.bl, h / 2), w / 2)\n let br = min(min(self.br, h / 2), w / 2)\n\n path.move(to: CGPoint(x: w / 2.0, y: 0))\n path.addLine(to: CGPoint(x: w - tr, y: 0))\n path.addArc(\n center: CGPoint(x: w - tr, y: tr),\n radius: tr,\n startAngle: Angle(degrees: -90),\n endAngle: Angle(degrees: 0),\n clockwise: false\n )\n path.addLine(to: CGPoint(x: w, y: h - br))\n path.addArc(\n center: CGPoint(x: w - br, y: h - br),\n radius: br,\n startAngle: Angle(degrees: 0),\n endAngle: Angle(degrees: 90),\n clockwise: false\n )\n path.addLine(to: CGPoint(x: bl, y: h))\n path.addArc(\n center: CGPoint(x: bl, y: h - bl),\n radius: bl,\n startAngle: Angle(degrees: 90),\n endAngle: Angle(degrees: 180),\n clockwise: false\n )\n path.addLine(to: CGPoint(x: 0, y: tl))\n path.addArc(\n center: CGPoint(x: tl, y: tl),\n radius: tl,\n startAngle: Angle(degrees: 180),\n endAngle: Angle(degrees: 270),\n clockwise: false\n )\n path.closeSubpath()\n }\n }\n}\n\n" }, { "path": "Core/Sources/ChatGPTChatTab/Views/BotMessage.swift", "content": "import ComposableArchitecture\nimport Foundation\nimport MarkdownUI\nimport SharedUIComponents\nimport SwiftUI\n\nstruct BotMessage: View {\n var r: Double { messageBubbleCornerRadius }\n let id: String\n let text: String\n let markdownContent: MarkdownContent\n let references: [DisplayedChatMessage.Reference]\n let chat: StoreOf\n @Environment(\\.colorScheme) var colorScheme\n\n @State var isReferencesPresented = false\n @State var isReferencesHovered = false\n\n var body: some View {\n HStack(alignment: .bottom, spacing: 2) {\n VStack(alignment: .leading, spacing: 16) {\n if !references.isEmpty {\n Button(action: {\n isReferencesPresented.toggle()\n }, label: {\n HStack(spacing: 4) {\n Image(systemName: \"plus.circle\")\n Text(\"Used \\(references.count) references\")\n }\n .padding(8)\n .background {\n RoundedRectangle(cornerRadius: r - 4)\n .foregroundStyle(Color(isReferencesHovered ? .black : .clear))\n }\n .overlay {\n RoundedRectangle(cornerRadius: r - 4)\n .stroke(Color(nsColor: .separatorColor), lineWidth: 1)\n }\n .foregroundStyle(.secondary)\n })\n .buttonStyle(.plain)\n .popover(isPresented: $isReferencesPresented, arrowEdge: .trailing) {\n ReferenceList(references: references, chat: chat)\n }\n }\n\n ThemedMarkdownText(markdownContent)\n }\n .frame(alignment: .trailing)\n .padding()\n .background {\n RoundedCorners(tl: r, tr: r, bl: 0, br: r)\n .fill(Color.contentBackground)\n }\n .overlay {\n RoundedCorners(tl: r, tr: r, bl: 0, br: r)\n .stroke(Color(nsColor: .separatorColor), lineWidth: 1)\n }\n .padding(.leading, 8)\n .shadow(color: .black.opacity(0.05), radius: 6)\n .contextMenu {\n Button(\"Copy\") {\n NSPasteboard.general.clearContents()\n NSPasteboard.general.setString(text, forType: .string)\n }\n\n Button(\"Set as Extra System Prompt\") {\n chat.send(.setAsExtraPromptButtonTapped(id))\n }\n\n Divider()\n\n Button(\"Delete\") {\n chat.send(.deleteMessageButtonTapped(id))\n }\n }\n\n CopyButton {\n NSPasteboard.general.clearContents()\n NSPasteboard.general.setString(text, forType: .string)\n }\n }\n .frame(maxWidth: .infinity, alignment: .leading)\n .padding(.trailing, 2)\n }\n}\n\nstruct ReferenceList: View {\n let references: [DisplayedChatMessage.Reference]\n let chat: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n ScrollView {\n VStack(alignment: .leading, spacing: 8) {\n ForEach(0.. MarkdownUI.Theme {\n .gitHub.text {\n ForegroundColor(.secondary)\n BackgroundColor(Color.clear)\n FontSize(fontSize - 1)\n }\n .list { configuration in\n configuration.label\n .markdownMargin(top: 4, bottom: 4)\n }\n .paragraph { configuration in\n configuration.label\n .markdownMargin(top: 0, bottom: 4)\n }\n .codeBlock { configuration in\n configuration.label\n .relativeLineSpacing(.em(0.225))\n .markdownTextStyle {\n FontFamilyVariant(.monospaced)\n FontSize(.em(0.85))\n }\n .padding(16)\n .background(Color(nsColor: .textBackgroundColor).opacity(0.7))\n .clipShape(RoundedRectangle(cornerRadius: 6))\n .markdownMargin(top: 4, bottom: 4)\n }\n }\n}\n" }, { "path": "Core/Sources/ChatGPTChatTab/Views/FunctionMessage.swift", "content": "import Foundation\nimport MarkdownUI\nimport SwiftUI\n\nstruct FunctionMessage: View {\n let id: String\n let text: String\n @AppStorage(\\.chatFontSize) var chatFontSize\n\n var body: some View {\n Markdown(text)\n .textSelection(.enabled)\n .markdownTheme(.functionCall(fontSize: chatFontSize))\n .padding(.vertical, 2)\n .padding(.trailing, 2)\n }\n}\n\n#Preview {\n FunctionMessage(id: \"1\", text: \"\"\"\n Searching for something...\n - abc\n - [def](https://1.com)\n > hello\n > hi\n \"\"\")\n .padding()\n .fixedSize()\n}\n\n" }, { "path": "Core/Sources/ChatGPTChatTab/Views/InstructionMarkdownTheme.swift", "content": "import Foundation\nimport MarkdownUI\nimport SwiftUI\n\nextension MarkdownUI.Theme {\n static func instruction(fontSize: Double) -> MarkdownUI.Theme {\n .gitHub.text {\n ForegroundColor(.primary)\n BackgroundColor(Color.clear)\n FontSize(fontSize)\n }\n .code {\n FontFamilyVariant(.monospaced)\n FontSize(.em(0.85))\n BackgroundColor(Color.secondary.opacity(0.2))\n }\n .codeBlock { configuration in\n let wrapCode = UserDefaults.shared.value(for: \\.wrapCodeInChatCodeBlock)\n\n if wrapCode {\n configuration.label\n .codeBlockLabelStyle()\n .codeBlockStyle(\n configuration,\n backgroundColor: Color(nsColor: .textBackgroundColor).opacity(0.7),\n labelColor: Color.secondary.opacity(0.7)\n )\n } else {\n ScrollView(.horizontal) {\n configuration.label\n .codeBlockLabelStyle()\n }\n .workaroundForVerticalScrollingBugInMacOS()\n .codeBlockStyle(\n configuration,\n backgroundColor: Color(nsColor: .textBackgroundColor).opacity(0.7),\n labelColor: Color.secondary.opacity(0.7)\n )\n }\n }\n .table { configuration in\n configuration.label\n .fixedSize(horizontal: false, vertical: true)\n .markdownTableBorderStyle(.init(\n color: .init(nsColor: .separatorColor),\n strokeStyle: .init(lineWidth: 1)\n ))\n .markdownTableBackgroundStyle(\n .alternatingRows(Color.secondary.opacity(0.1), Color.secondary.opacity(0.2))\n )\n .markdownMargin(top: 0, bottom: 16)\n }\n .tableCell { configuration in\n configuration.label\n .markdownTextStyle {\n if configuration.row == 0 {\n FontWeight(.semibold)\n }\n BackgroundColor(nil)\n }\n .fixedSize(horizontal: false, vertical: true)\n .padding(.vertical, 6)\n .padding(.horizontal, 13)\n .relativeLineSpacing(.em(0.25))\n }\n }\n}\n\n" }, { "path": "Core/Sources/ChatGPTChatTab/Views/Instructions.swift", "content": "import ComposableArchitecture\nimport Foundation\nimport MarkdownUI\nimport SwiftUI\n\nstruct Instruction: View {\n let chat: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n Group {\n Markdown(\n \"\"\"\n You can use plugins to perform various tasks.\n \n | Plugin Name | Description |\n | --- | --- |\n | `/shell` | Runs a command under the project root |\n | `/shortcut(name)` | Runs a shortcut from the Shortcuts.app, with the previous message as input |\n \n To use plugins, you can prefix a message with `/pluginName`.\n \"\"\"\n )\n .modifier(InstructionModifier())\n \n Markdown(\n \"\"\"\n You can use scopes to give the bot extra abilities.\n \n | Scope Name | Abilities |\n | --- | --- |\n | `@file` | Read the metadata of the editing file |\n | `@code` | Read the code and metadata in the editing file |\n | `@sense`| Experimental. Read the relevant code of the focused editor |\n | `@project` | Experimental. Access content of the project |\n | `@web` (beta) | Search on Bing or query from a web page |\n \n To use scopes, you can prefix a message with `@code`.\n \n You can use shorthand to represent a scope, such as `@c`, and enable multiple scopes with `@c+web`.\n \"\"\"\n )\n .modifier(InstructionModifier())\n \n let scopes = chat.chatMenu.defaultScopes\n Markdown(\n \"\"\"\n Hello, I am your AI programming assistant. I can identify issues, explain and even improve code.\n \n \\({\n if scopes.isEmpty {\n return \"No scope is enabled by default\"\n } else {\n let scopes = scopes.map(\\.rawValue).sorted()\n .joined(separator: \", \")\n return \"Default scopes: `\\(scopes)`\"\n }\n }())\n \"\"\"\n )\n .modifier(InstructionModifier())\n }\n }\n }\n\n struct InstructionModifier: ViewModifier {\n @AppStorage(\\.chatFontSize) var chatFontSize\n\n func body(content: Content) -> some View {\n content\n .textSelection(.enabled)\n .markdownTheme(.instruction(fontSize: chatFontSize))\n .opacity(0.8)\n .frame(maxWidth: .infinity, alignment: .leading)\n .padding()\n .overlay {\n RoundedRectangle(cornerRadius: 8)\n .stroke(Color(nsColor: .separatorColor), lineWidth: 1)\n }\n }\n }\n}\n\n" }, { "path": "Core/Sources/ChatGPTChatTab/Views/ThemedMarkdownText.swift", "content": "import Foundation\nimport MarkdownUI\nimport SwiftUI\n\nstruct ThemedMarkdownText: View {\n @AppStorage(\\.syncChatCodeHighlightTheme) var syncCodeHighlightTheme\n @AppStorage(\\.codeForegroundColorLight) var codeForegroundColorLight\n @AppStorage(\\.codeBackgroundColorLight) var codeBackgroundColorLight\n @AppStorage(\\.codeForegroundColorDark) var codeForegroundColorDark\n @AppStorage(\\.codeBackgroundColorDark) var codeBackgroundColorDark\n @AppStorage(\\.chatFontSize) var chatFontSize\n @AppStorage(\\.chatCodeFont) var chatCodeFont\n @Environment(\\.colorScheme) var colorScheme\n\n let content: MarkdownContent\n\n init(_ text: String) {\n content = .init(text)\n }\n\n init(_ content: MarkdownContent) {\n self.content = content\n }\n\n var body: some View {\n Markdown(content)\n .textSelection(.enabled)\n .markdownTheme(.custom(\n fontSize: chatFontSize,\n codeFont: chatCodeFont.value.nsFont,\n codeBlockBackgroundColor: {\n if syncCodeHighlightTheme {\n if colorScheme == .light, let color = codeBackgroundColorLight.value {\n return color.swiftUIColor\n } else if let color = codeBackgroundColorDark.value {\n return color.swiftUIColor\n }\n }\n\n return Color(nsColor: .textBackgroundColor).opacity(0.7)\n }(),\n codeBlockLabelColor: {\n if syncCodeHighlightTheme {\n if colorScheme == .light,\n let color = codeForegroundColorLight.value\n {\n return color.swiftUIColor.opacity(0.5)\n } else if let color = codeForegroundColorDark.value {\n return color.swiftUIColor.opacity(0.5)\n }\n }\n return Color.secondary.opacity(0.7)\n }()\n ))\n }\n}\n\n// MARK: - Theme\n\nextension MarkdownUI.Theme {\n static func custom(\n fontSize: Double,\n codeFont: NSFont,\n codeBlockBackgroundColor: Color,\n codeBlockLabelColor: Color\n ) -> MarkdownUI.Theme {\n .gitHub.text {\n ForegroundColor(.primary)\n BackgroundColor(Color.clear)\n FontSize(fontSize)\n }\n .codeBlock { configuration in\n let wrapCode = UserDefaults.shared.value(for: \\.wrapCodeInChatCodeBlock)\n || [\n \"plaintext\", \"text\", \"markdown\", \"sh\", \"console\", \"bash\", \"shell\", \"latex\",\n \"tex\"\n ]\n .contains(configuration.language)\n\n if wrapCode {\n AsyncCodeBlockView(\n fenceInfo: configuration.language,\n content: configuration.content,\n font: codeFont\n )\n .codeBlockLabelStyle()\n .codeBlockStyle(\n configuration,\n backgroundColor: codeBlockBackgroundColor,\n labelColor: codeBlockLabelColor\n )\n } else {\n ScrollView(.horizontal) {\n AsyncCodeBlockView(\n fenceInfo: configuration.language,\n content: configuration.content,\n font: codeFont\n )\n .codeBlockLabelStyle()\n }\n .workaroundForVerticalScrollingBugInMacOS()\n .codeBlockStyle(\n configuration,\n backgroundColor: codeBlockBackgroundColor,\n labelColor: codeBlockLabelColor\n )\n }\n }\n }\n}\n\n" }, { "path": "Core/Sources/ChatGPTChatTab/Views/UserMessage.swift", "content": "import ComposableArchitecture\nimport Foundation\nimport MarkdownUI\nimport SwiftUI\n\nstruct UserMessage: View {\n var r: Double { messageBubbleCornerRadius }\n let id: String\n let text: String\n let markdownContent: MarkdownContent\n let chat: StoreOf\n @Environment(\\.colorScheme) var colorScheme\n\n var body: some View {\n ThemedMarkdownText(markdownContent)\n .frame(alignment: .leading)\n .padding()\n .background {\n RoundedCorners(tl: r, tr: r, bl: r, br: 0)\n .fill(Color.userChatContentBackground)\n }\n .overlay {\n RoundedCorners(tl: r, tr: r, bl: r, br: 0)\n .stroke(Color(nsColor: .separatorColor), lineWidth: 1)\n }\n .padding(.leading)\n .padding(.trailing, 8)\n .shadow(color: .black.opacity(0.05), radius: 6)\n .frame(maxWidth: .infinity, alignment: .trailing)\n .contextMenu {\n Button(\"Copy\") {\n NSPasteboard.general.clearContents()\n NSPasteboard.general.setString(text, forType: .string)\n }\n\n Button(\"Send Again\") {\n chat.send(.resendMessageButtonTapped(id))\n }\n\n Button(\"Set as Extra System Prompt\") {\n chat.send(.setAsExtraPromptButtonTapped(id))\n }\n\n Divider()\n\n Button(\"Delete\") {\n chat.send(.deleteMessageButtonTapped(id))\n }\n }\n }\n}\n\n#Preview {\n let text = #\"\"\"\n Please buy me a coffee!\n | Coffee | Milk |\n |--------|------|\n | Espresso | No |\n | Latte | Yes |\n ```swift\n func foo() {}\n ```\n ```objectivec\n - (void)bar {}\n ```\n \"\"\"#\n\n return UserMessage(\n id: \"A\",\n text: text,\n markdownContent: .init(text),\n chat: .init(\n initialState: .init(history: [] as [DisplayedChatMessage], isReceivingMessage: false),\n reducer: { Chat(service: .init()) }\n )\n )\n .padding()\n .fixedSize(horizontal: true, vertical: true)\n}\n\n" }, { "path": "Core/Sources/ChatService/AllContextCollector.swift", "content": "import ActiveDocumentChatContextCollector\nimport ChatContextCollector\nimport SystemInfoChatContextCollector\nimport WebChatContextCollector\n#if canImport(ProChatContextCollectors)\nimport ProChatContextCollectors\nlet allContextCollectors: [any ChatContextCollector] = [\n SystemInfoChatContextCollector(),\n WebChatContextCollector(),\n ProChatContextCollectors(),\n]\n#else\nlet allContextCollectors: [any ChatContextCollector] = [\n SystemInfoChatContextCollector(),\n ActiveDocumentChatContextCollector(),\n WebChatContextCollector(),\n]\n#endif\n\n" }, { "path": "Core/Sources/ChatService/AllPlugins.swift", "content": "import ChatBasic\nimport Foundation\nimport OpenAIService\nimport ShortcutChatPlugin\nimport TerminalChatPlugin\n\nlet allPlugins: [LegacyChatPlugin.Type] = [\n LegacyChatPluginWrapper.self,\n LegacyChatPluginWrapper.self,\n]\n\nprotocol LegacyChatPlugin: AnyObject {\n static var command: String { get }\n var name: String { get }\n\n init(inside chatGPTService: any LegacyChatGPTServiceType, delegate: LegacyChatPluginDelegate)\n func send(content: String, originalMessage: String) async\n func cancel() async\n func stopResponding() async\n}\n\nprotocol LegacyChatPluginDelegate: AnyObject {\n func pluginDidStart(_ plugin: LegacyChatPlugin)\n func pluginDidEnd(_ plugin: LegacyChatPlugin)\n func pluginDidStartResponding(_ plugin: LegacyChatPlugin)\n func pluginDidEndResponding(_ plugin: LegacyChatPlugin)\n func shouldStartAnotherPlugin(_ type: LegacyChatPlugin.Type, withContent: String)\n}\n\nfinal class LegacyChatPluginWrapper: LegacyChatPlugin {\n static var command: String { Plugin.command }\n var name: String { Plugin.name }\n\n let chatGPTService: any LegacyChatGPTServiceType\n weak var delegate: LegacyChatPluginDelegate?\n var isCancelled = false\n\n required init(\n inside chatGPTService: any LegacyChatGPTServiceType,\n delegate: any LegacyChatPluginDelegate\n ) {\n self.chatGPTService = chatGPTService\n self.delegate = delegate\n }\n\n func send(content: String, originalMessage: String) async {\n delegate?.pluginDidStart(self)\n delegate?.pluginDidStartResponding(self)\n\n let id = \"\\(Self.command)-\\(UUID().uuidString)\"\n var reply = ChatMessage(id: id, role: .assistant, content: \"\")\n\n await chatGPTService.memory.mutateHistory { history in\n history.append(.init(role: .user, content: originalMessage))\n }\n\n let plugin = Plugin()\n\n let stream = await plugin.sendForComplicatedResponse(.init(\n text: content,\n arguments: [],\n history: chatGPTService.memory.history\n ))\n\n do {\n var actions = [(id: String, name: String)]()\n var actionResults = [String: String]()\n var message = \"\"\n\n for try await response in stream {\n guard !isCancelled else { break }\n if Task.isCancelled { break }\n\n switch response {\n case .status:\n break\n case let .content(content):\n switch content {\n case let .text(token):\n message.append(token)\n }\n case .attachments:\n break\n case let .startAction(id, task):\n actions.append((id: id, name: task))\n case let .finishAction(id, result):\n actionResults[id] = switch result {\n case let .failure(error):\n error\n case let .success(result):\n result\n }\n case .references:\n break\n case .startNewMessage:\n break\n case .reasoning:\n break\n }\n\n await chatGPTService.memory.mutateHistory { history in\n if history.last?.id == id {\n history.removeLast()\n }\n\n let actionString = actions.map {\n \"> \\($0.name): \\(actionResults[$0.id] ?? \"...\")\"\n }.joined(separator: \"\\n>\\n\")\n\n if message.isEmpty {\n reply.content = actionString\n } else {\n reply.content = \"\"\"\n \\(actionString)\n\n \\(message)\n \"\"\"\n }\n history.append(reply)\n }\n }\n } catch {\n await chatGPTService.memory.mutateHistory { history in\n if history.last?.id == id {\n history.removeLast()\n }\n reply.content = error.localizedDescription\n history.append(reply)\n }\n }\n\n delegate?.pluginDidEndResponding(self)\n delegate?.pluginDidEnd(self)\n }\n\n func cancel() async {\n isCancelled = true\n }\n\n func stopResponding() async {\n isCancelled = true\n }\n}\n\n" }, { "path": "Core/Sources/ChatService/ChatFunctionProvider.swift", "content": "import ChatBasic\nimport Foundation\nimport OpenAIService\n\nfinal class ChatFunctionProvider {\n var functions: [any ChatGPTFunction] = []\n \n init() {}\n \n func removeAll() {\n functions = []\n }\n \n func append(functions others: [any ChatGPTFunction]) {\n functions.append(contentsOf: others)\n }\n}\n\nextension ChatFunctionProvider: ChatGPTFunctionProvider {\n var functionCallStrategy: OpenAIService.FunctionCallStrategy? {\n nil\n }\n}\n\n" }, { "path": "Core/Sources/ChatService/ChatPluginController.swift", "content": "import Combine\nimport Foundation\nimport LegacyChatPlugin\nimport OpenAIService\n\nfinal class ChatPluginController {\n let chatGPTService: any LegacyChatGPTServiceType\n let plugins: [String: LegacyChatPlugin.Type]\n var runningPlugin: LegacyChatPlugin?\n weak var chatService: ChatService?\n\n init(chatGPTService: any LegacyChatGPTServiceType, plugins: [LegacyChatPlugin.Type]) {\n self.chatGPTService = chatGPTService\n var all = [String: LegacyChatPlugin.Type]()\n for plugin in plugins {\n all[plugin.command.lowercased()] = plugin\n }\n self.plugins = all\n }\n\n convenience init(\n chatGPTService: any LegacyChatGPTServiceType,\n plugins: LegacyChatPlugin.Type...\n ) {\n self.init(chatGPTService: chatGPTService, plugins: plugins)\n }\n\n /// Handle the message in a plugin if required. Return false if no plugin handles the message.\n func handleContent(_ content: String) async throws -> Bool {\n // look for the prefix of content, see if there is something like /command.\n // If there is, then we need to find the plugin that can handle this command.\n // If there is no such plugin, then we just send the message to the GPT service.\n let regex = try NSRegularExpression(pattern: #\"^\\/([a-zA-Z0-9]+)\"#)\n let matches = regex.matches(in: content, range: NSRange(content.startIndex..., in: content))\n if let match = matches.first {\n let command = String(content[Range(match.range(at: 1), in: content)!]).lowercased()\n // handle exit plugin\n if command == \"exit\" {\n if let plugin = runningPlugin {\n runningPlugin = nil\n _ = await chatGPTService.memory.mutateHistory { history in\n history.append(.init(\n role: .user,\n content: \"\",\n summary: \"Exit plugin \\(plugin.name).\"\n ))\n history.append(.init(\n role: .system,\n content: \"\",\n summary: \"Exited plugin \\(plugin.name).\"\n ))\n }\n } else {\n _ = await chatGPTService.memory.mutateHistory { history in\n history.append(.init(\n role: .system,\n content: \"\",\n summary: \"No plugin running.\"\n ))\n }\n }\n return true\n }\n\n // pass message to running plugin\n if let runningPlugin {\n await runningPlugin.send(content: content, originalMessage: content)\n return true\n }\n\n // pass message to new plugin\n if let pluginType = plugins[command] {\n let plugin = pluginType.init(inside: chatGPTService, delegate: self)\n if #available(macOS 13.0, *) {\n await plugin.send(\n content: String(\n content.dropFirst(command.count + 1)\n .trimmingPrefix(while: { $0 == \" \" })\n ),\n originalMessage: content\n )\n } else {\n await plugin.send(\n content: String(content.dropFirst(command.count + 1)),\n originalMessage: content\n )\n }\n return true\n }\n\n return false\n } else if let runningPlugin {\n // pass message to running plugin\n await runningPlugin.send(content: content, originalMessage: content)\n return true\n } else {\n return false\n }\n }\n\n func stopResponding() async {\n await runningPlugin?.stopResponding()\n }\n\n func cancel() async {\n await runningPlugin?.cancel()\n }\n}\n\n// MARK: - ChatPluginDelegate\n\nextension ChatPluginController: LegacyChatPluginDelegate {\n public func pluginDidStartResponding(_: LegacyChatPlugin) {\n chatService?.isReceivingMessage = true\n }\n\n public func pluginDidEndResponding(_: LegacyChatPlugin) {\n chatService?.isReceivingMessage = false\n }\n\n public func pluginDidStart(_ plugin: LegacyChatPlugin) {\n runningPlugin = plugin\n }\n\n public func pluginDidEnd(_ plugin: LegacyChatPlugin) {\n if runningPlugin === plugin {\n runningPlugin = nil\n }\n }\n\n public func shouldStartAnotherPlugin(\n _ type: LegacyChatPlugin.Type,\n withContent content: String\n ) {\n let plugin = type.init(inside: chatGPTService, delegate: self)\n Task {\n await plugin.send(content: content, originalMessage: content)\n }\n }\n}\n\n" }, { "path": "Core/Sources/ChatService/ChatService.swift", "content": "import ChatContextCollector\nimport LegacyChatPlugin\nimport Combine\nimport CustomCommandTemplateProcessor\nimport Foundation\nimport OpenAIService\nimport Preferences\n\npublic final class ChatService: ObservableObject {\n public typealias Scope = ChatContext.Scope\n\n public let memory: ContextAwareAutoManagedChatGPTMemory\n public let configuration: OverridingChatGPTConfiguration\n public let chatGPTService: any LegacyChatGPTServiceType\n public var allPluginCommands: [String] { allPlugins.map { $0.command } }\n @Published public internal(set) var chatHistory: [ChatMessage] = []\n @Published public internal(set) var isReceivingMessage = false\n @Published public internal(set) var systemPrompt = UserDefaults.shared\n .value(for: \\.defaultChatSystemPrompt)\n @Published public internal(set) var extraSystemPrompt = \"\"\n @Published public var defaultScopes = Set()\n\n let pluginController: ChatPluginController\n var cancellable = Set()\n\n init(\n memory: ContextAwareAutoManagedChatGPTMemory,\n configuration: OverridingChatGPTConfiguration,\n chatGPTService: T\n ) {\n self.memory = memory\n self.configuration = configuration\n self.chatGPTService = chatGPTService\n pluginController = ChatPluginController(\n chatGPTService: chatGPTService,\n plugins: allPlugins\n )\n\n pluginController.chatService = self\n }\n\n public convenience init() {\n let configuration = UserPreferenceChatGPTConfiguration().overriding()\n /// Used by context collector\n let extraConfiguration = configuration.overriding()\n extraConfiguration.textWindowTerminator = {\n guard let last = $0.last else { return false }\n return last.isNewline || last.isPunctuation\n }\n let memory = ContextAwareAutoManagedChatGPTMemory(\n configuration: extraConfiguration,\n functionProvider: ChatFunctionProvider()\n )\n self.init(\n memory: memory,\n configuration: configuration,\n chatGPTService: LegacyChatGPTService(\n memory: memory,\n configuration: extraConfiguration,\n functionProvider: memory.functionProvider\n )\n )\n\n resetDefaultScopes()\n\n memory.chatService = self\n memory.observeHistoryChange { [weak self] in\n Task { [weak self] in\n self?.chatHistory = await memory.history\n }\n }\n }\n\n public func resetDefaultScopes() {\n var scopes = Set()\n if UserDefaults.shared.value(for: \\.enableFileScopeByDefaultInChatContext) {\n scopes.insert(.file)\n }\n\n if UserDefaults.shared.value(for: \\.enableCodeScopeByDefaultInChatContext) {\n scopes.insert(.code)\n }\n\n if UserDefaults.shared.value(for: \\.enableProjectScopeByDefaultInChatContext) {\n scopes.insert(.project)\n }\n\n if UserDefaults.shared.value(for: \\.enableSenseScopeByDefaultInChatContext) {\n scopes.insert(.sense)\n }\n\n if UserDefaults.shared.value(for: \\.enableWebScopeByDefaultInChatContext) {\n scopes.insert(.web)\n }\n\n defaultScopes = scopes\n }\n\n public func send(content: String) async throws {\n memory.contextController.defaultScopes = defaultScopes\n guard !isReceivingMessage else { throw CancellationError() }\n let handledInPlugin = try await pluginController.handleContent(content)\n if handledInPlugin { return }\n isReceivingMessage = true\n defer { isReceivingMessage = false }\n\n let stream = try await chatGPTService.send(content: content, summary: nil)\n do {\n for try await _ in stream {\n try Task.checkCancellation()\n }\n } catch {}\n }\n\n public func sendAndWait(content: String) async throws -> String {\n try await send(content: content)\n if let reply = await memory.history.last(where: { $0.role == .assistant })?.content {\n return reply\n }\n return \"\"\n }\n\n public func stopReceivingMessage() async {\n await pluginController.stopResponding()\n await chatGPTService.stopReceivingMessage()\n isReceivingMessage = false\n\n // if it's stopped before the tool calls finish, remove the message.\n await memory.mutateHistory { history in\n if history.last?.role == .assistant, history.last?.toolCalls != nil {\n history.removeLast()\n }\n }\n }\n\n public func clearHistory() async {\n await pluginController.cancel()\n await memory.clearHistory()\n await chatGPTService.stopReceivingMessage()\n isReceivingMessage = false\n }\n\n public func resetPrompt() async {\n systemPrompt = UserDefaults.shared.value(for: \\.defaultChatSystemPrompt)\n extraSystemPrompt = \"\"\n }\n\n public func deleteMessage(id: String) async {\n await memory.removeMessage(id)\n }\n\n public func resendMessage(id: String) async throws {\n if let message = (await memory.history).first(where: { $0.id == id }),\n let content = message.content\n {\n try await send(content: content)\n }\n }\n\n public func setMessageAsExtraPrompt(id: String) async {\n if let message = (await memory.history).first(where: { $0.id == id }),\n let content = message.content\n {\n mutateExtraSystemPrompt(content)\n await mutateHistory { history in\n history.append(.init(\n role: .assistant,\n content: \"\",\n summary: \"System prompt updated.\"\n ))\n }\n }\n }\n\n /// Setting it to `nil` to reset the system prompt\n public func mutateSystemPrompt(_ newPrompt: String?) {\n systemPrompt = newPrompt ?? UserDefaults.shared.value(for: \\.defaultChatSystemPrompt)\n }\n\n public func mutateExtraSystemPrompt(_ newPrompt: String) {\n extraSystemPrompt = newPrompt\n }\n\n public func mutateHistory(_ mutator: @escaping (inout [ChatMessage]) -> Void) async {\n await memory.mutateHistory(mutator)\n }\n\n public func handleCustomCommand(_ command: CustomCommand) async throws {\n struct CustomCommandInfo {\n var specifiedSystemPrompt: String?\n var extraSystemPrompt: String?\n var sendingMessageImmediately: String?\n var name: String?\n }\n\n let info: CustomCommandInfo? = {\n switch command.feature {\n case let .chatWithSelection(extraSystemPrompt, prompt, useExtraSystemPrompt):\n let updatePrompt = useExtraSystemPrompt ?? true\n return .init(\n extraSystemPrompt: updatePrompt ? extraSystemPrompt : nil,\n sendingMessageImmediately: prompt,\n name: command.name\n )\n case let .customChat(systemPrompt, prompt):\n memory.contextController.defaultScopes = []\n return .init(\n specifiedSystemPrompt: systemPrompt,\n extraSystemPrompt: \"\",\n sendingMessageImmediately: prompt,\n name: command.name\n )\n case .promptToCode: return nil\n case .singleRoundDialog: return nil\n }\n }()\n\n guard let info else { return }\n\n let templateProcessor = CustomCommandTemplateProcessor()\n if let specifiedSystemPrompt = info.specifiedSystemPrompt {\n await mutateSystemPrompt(templateProcessor.process(specifiedSystemPrompt))\n }\n if let extraSystemPrompt = info.extraSystemPrompt {\n await mutateExtraSystemPrompt(templateProcessor.process(extraSystemPrompt))\n } else {\n mutateExtraSystemPrompt(\"\")\n }\n\n let customCommandPrefix = {\n if let name = info.name { return \"[\\(name)] \" }\n return \"\"\n }()\n\n if info.specifiedSystemPrompt != nil || info.extraSystemPrompt != nil {\n await mutateHistory { history in\n history.append(.init(\n role: .assistant,\n content: \"\",\n summary: \"\\(customCommandPrefix)System prompt is updated.\"\n ))\n }\n }\n\n if let sendingMessageImmediately = info.sendingMessageImmediately,\n !sendingMessageImmediately.isEmpty\n {\n try await send(content: templateProcessor.process(sendingMessageImmediately))\n }\n }\n\n public func handleSingleRoundDialogCommand(\n systemPrompt: String?,\n overwriteSystemPrompt: Bool,\n prompt: String\n ) async throws -> String {\n let templateProcessor = CustomCommandTemplateProcessor()\n if let systemPrompt {\n if overwriteSystemPrompt {\n await mutateSystemPrompt(templateProcessor.process(systemPrompt))\n } else {\n await mutateExtraSystemPrompt(templateProcessor.process(systemPrompt))\n }\n }\n return try await sendAndWait(content: templateProcessor.process(prompt))\n }\n\n public func processMessage(\n systemPrompt: String?,\n extraSystemPrompt: String?,\n prompt: String\n ) async throws -> String {\n let templateProcessor = CustomCommandTemplateProcessor()\n if let systemPrompt {\n await mutateSystemPrompt(templateProcessor.process(systemPrompt))\n }\n if let extraSystemPrompt {\n await mutateExtraSystemPrompt(templateProcessor.process(extraSystemPrompt))\n }\n return try await sendAndWait(content: templateProcessor.process(prompt))\n }\n}\n\n" }, { "path": "Core/Sources/ChatService/ContextAwareAutoManagedChatGPTMemory.swift", "content": "import Foundation\nimport OpenAIService\n\npublic final class ContextAwareAutoManagedChatGPTMemory: ChatGPTMemory {\n private let memory: AutoManagedChatGPTMemory\n let contextController: DynamicContextController\n let functionProvider: ChatFunctionProvider\n weak var chatService: ChatService?\n\n public var history: [ChatMessage] {\n get async { await memory.history }\n }\n\n func observeHistoryChange(_ observer: @escaping () -> Void) {\n memory.observeHistoryChange(observer)\n }\n\n init(\n configuration: OverridingChatGPTConfiguration,\n functionProvider: ChatFunctionProvider\n ) {\n memory = AutoManagedChatGPTMemory(\n systemPrompt: \"\",\n configuration: configuration,\n functionProvider: functionProvider,\n maxNumberOfMessages: UserDefaults.shared.value(for: \\.chatGPTMaxMessageCount)\n )\n contextController = DynamicContextController(\n memory: memory,\n functionProvider: functionProvider,\n configuration: configuration,\n contextCollectors: allContextCollectors\n )\n self.functionProvider = functionProvider\n }\n\n public func mutateHistory(_ update: (inout [ChatMessage]) -> Void) async {\n await memory.mutateHistory(update)\n }\n\n public func generatePrompt() async -> ChatGPTPrompt {\n let content = (await memory.history)\n .last(where: { $0.role == .user })?.content\n try? await contextController.collectContextInformation(\n systemPrompt: \"\"\"\n \\(chatService?.systemPrompt ?? \"\")\n \\(chatService?.extraSystemPrompt ?? \"\")\n \"\"\".trimmingCharacters(in: .whitespacesAndNewlines),\n content: content ?? \"\"\n )\n return await memory.generatePrompt()\n }\n}\n\n" }, { "path": "Core/Sources/ChatService/DynamicContextController.swift", "content": "import ChatContextCollector\nimport Foundation\nimport OpenAIService\nimport Preferences\nimport XcodeInspector\n\nfinal class DynamicContextController {\n let contextCollectors: [ChatContextCollector]\n let memory: AutoManagedChatGPTMemory\n let functionProvider: ChatFunctionProvider\n let configuration: OverridingChatGPTConfiguration\n var defaultScopes = [] as Set\n\n convenience init(\n memory: AutoManagedChatGPTMemory,\n functionProvider: ChatFunctionProvider,\n configuration: OverridingChatGPTConfiguration,\n contextCollectors: ChatContextCollector...\n ) {\n self.init(\n memory: memory,\n functionProvider: functionProvider,\n configuration: configuration,\n contextCollectors: contextCollectors\n )\n }\n\n init(\n memory: AutoManagedChatGPTMemory,\n functionProvider: ChatFunctionProvider,\n configuration: OverridingChatGPTConfiguration,\n contextCollectors: [ChatContextCollector]\n ) {\n self.memory = memory\n self.functionProvider = functionProvider\n self.configuration = configuration\n self.contextCollectors = contextCollectors\n }\n\n func collectContextInformation(systemPrompt: String, content: String) async throws {\n var content = content\n var scopes = Self.parseScopes(&content)\n scopes.formUnion(defaultScopes)\n\n let overridingChatModelId = {\n var ids = [String]()\n if scopes.contains(.sense) {\n ids.append(UserDefaults.shared.value(for: \\.preferredChatModelIdForSenseScope))\n }\n\n if scopes.contains(.project) {\n ids.append(UserDefaults.shared.value(for: \\.preferredChatModelIdForProjectScope))\n }\n\n if scopes.contains(.web) {\n ids.append(UserDefaults.shared.value(for: \\.preferredChatModelIdForWebScope))\n }\n\n let chatModels = UserDefaults.shared.value(for: \\.chatModels)\n let idIndexMap = chatModels.enumerated().reduce(into: [String: Int]()) {\n $0[$1.element.id] = $1.offset\n }\n return ids.filter { !$0.isEmpty }.sorted(by: {\n let lhs = idIndexMap[$0] ?? Int.max\n let rhs = idIndexMap[$1] ?? Int.max\n return lhs < rhs\n }).first\n }()\n\n configuration.overriding.modelId = overridingChatModelId\n\n functionProvider.removeAll()\n let language = UserDefaults.shared.value(for: \\.chatGPTLanguage)\n let oldMessages = await memory.history\n let contexts = await withTaskGroup(\n of: ChatContext.self\n ) { [scopes, content, configuration] group in\n for collector in contextCollectors {\n group.addTask {\n await collector.generateContext(\n history: oldMessages,\n scopes: scopes,\n content: content,\n configuration: configuration\n )\n }\n }\n var contexts = [ChatContext]()\n for await context in group {\n contexts.append(context)\n }\n return contexts\n }\n\n let contextSystemPrompt = contexts\n .map(\\.systemPrompt)\n .filter { !$0.isEmpty }\n .joined(separator: \"\\n\\n\")\n\n let retrievedContent = contexts\n .flatMap(\\.retrievedContent)\n .filter { !$0.document.content.isEmpty }\n .sorted { $0.priority > $1.priority }\n .prefix(15)\n\n let contextualSystemPrompt = \"\"\"\n \\(language.isEmpty ? \"\" : \"You must always reply in \\(language)\")\n \\(systemPrompt)\n \"\"\".trimmingCharacters(in: .whitespacesAndNewlines)\n await memory.mutateSystemPrompt(contextualSystemPrompt)\n await memory.mutateContextSystemPrompt(contextSystemPrompt)\n await memory.mutateRetrievedContent(retrievedContent.map(\\.document))\n functionProvider.append(functions: contexts.flatMap(\\.functions))\n }\n}\n\nextension DynamicContextController {\n static func parseScopes(_ prompt: inout String) -> Set {\n let parser = MessageScopeParser()\n return parser(&prompt)\n }\n}\n\n" }, { "path": "Core/Sources/Client/XPCService.swift", "content": "import Foundation\nimport Logger\nimport os.log\nimport XPCShared\n\nlet shared = XPCExtensionService(logger: .client)\n\npublic func getService() throws -> XPCExtensionService {\n if ProcessInfo.processInfo.environment[\"XCODE_RUNNING_FOR_PREVIEWS\"] == \"1\" {\n struct RunningInPreview: Error {}\n throw RunningInPreview()\n }\n return shared\n}\n" }, { "path": "Core/Sources/FileChangeChecker/FileChangeChecker.swift", "content": "import CryptoKit\nimport Dispatch\nimport Foundation\n\n/// Check that a file is changed.\npublic actor FileChangeChecker {\n let url: URL\n var checksum: Data?\n\n public init(fileURL: URL) async {\n url = fileURL\n checksum = getChecksum()\n }\n\n public func checkIfChanged() -> Bool {\n guard let newChecksum = getChecksum() else { return false }\n return newChecksum != checksum\n }\n\n func getChecksum() -> Data? {\n let bufferSize = 16 * 1024\n guard let file = try? FileHandle(forReadingFrom: url) else { return nil }\n defer { try? file.close() }\n var md5 = CryptoKit.Insecure.MD5()\n while autoreleasepool(invoking: {\n let data = file.readData(ofLength: bufferSize)\n if !data.isEmpty {\n md5.update(data: data)\n return true // Continue\n } else {\n return false // End of file\n }\n }) {}\n\n let data = Data(md5.finalize())\n\n return data\n }\n}\n" }, { "path": "Core/Sources/HostApp/AccountSettings/APIKeyManagement/APIKeyManagementView.swift", "content": "import ComposableArchitecture\nimport SharedUIComponents\nimport SwiftUI\n\nstruct APIKeyManagementView: View {\n @Perception.Bindable var store: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n VStack(spacing: 0) {\n HStack {\n Button(action: {\n store.send(.closeButtonClicked)\n }) {\n Image(systemName: \"xmark.circle.fill\")\n .foregroundStyle(.secondary)\n .padding()\n }\n .buttonStyle(.plain)\n Text(\"API Keys\")\n Spacer()\n Button(action: {\n store.send(.addButtonClicked)\n }) {\n Image(systemName: \"plus.circle.fill\")\n .foregroundStyle(.secondary)\n .padding()\n }\n .buttonStyle(.plain)\n }\n .background(Color(nsColor: .separatorColor))\n\n List {\n ForEach(store.availableAPIKeyNames, id: \\.self) { name in\n WithPerceptionTracking {\n HStack {\n Text(name)\n .contextMenu {\n Button(\"Remove\") {\n store.send(.deleteButtonClicked(name: name))\n }\n }\n Spacer()\n\n Button(action: {\n store.send(.deleteButtonClicked(name: name))\n }) {\n Image(systemName: \"trash.fill\")\n .foregroundStyle(.secondary)\n }\n .buttonStyle(.plain)\n }\n }\n }\n .modify { view in\n if #available(macOS 13.0, *) {\n view.listRowSeparator(.hidden).listSectionSeparator(.hidden)\n } else {\n view\n }\n }\n }\n .removeBackground()\n .overlay {\n if store.availableAPIKeyNames.isEmpty {\n Text(\"\"\"\n Empty\n Add a new key by clicking the add button\n \"\"\")\n .multilineTextAlignment(.center)\n .padding()\n }\n }\n }\n .focusable(false)\n .frame(width: 300, height: 400)\n .background(.thickMaterial)\n .onAppear {\n store.send(.appear)\n }\n .sheet(item: $store.scope(\n state: \\.apiKeySubmission,\n action: \\.apiKeySubmission\n )) { store in\n WithPerceptionTracking {\n APIKeySubmissionView(store: store)\n .frame(minWidth: 400)\n }\n }\n }\n }\n}\n\nstruct APIKeySubmissionView: View {\n @Perception.Bindable var store: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n ScrollView {\n VStack(spacing: 0) {\n Form {\n TextField(\"Name\", text: $store.name)\n SecureField(\"Key\", text: $store.key)\n }\n .padding()\n\n Divider()\n\n HStack {\n Spacer()\n\n Button(\"Cancel\") { store.send(.cancelButtonClicked) }\n .keyboardShortcut(.cancelAction)\n\n Button(\"Save\", action: { store.send(.saveButtonClicked) })\n .keyboardShortcut(.defaultAction)\n }.padding()\n }\n }\n .textFieldStyle(.roundedBorder)\n }\n }\n}\n\nclass APIKeyManagementView_Preview: PreviewProvider {\n static var previews: some View {\n APIKeyManagementView(\n store: .init(\n initialState: .init(\n availableAPIKeyNames: [\"test1\", \"test2\"]\n ),\n reducer: { APIKeyManagement() }\n )\n )\n }\n}\n\nclass APIKeySubmissionView_Preview: PreviewProvider {\n static var previews: some View {\n APIKeySubmissionView(\n store: .init(\n initialState: .init(),\n reducer: { APIKeySubmission() }\n )\n )\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/AccountSettings/APIKeyManagement/APIKeyManangement.swift", "content": "import ComposableArchitecture\nimport Foundation\n\n@Reducer\nstruct APIKeyManagement {\n @ObservableState\n struct State: Equatable {\n var availableAPIKeyNames: [String] = []\n @Presents var apiKeySubmission: APIKeySubmission.State?\n }\n\n enum Action: Equatable {\n case appear\n case closeButtonClicked\n case addButtonClicked\n case deleteButtonClicked(name: String)\n case refreshAvailableAPIKeyNames\n\n case apiKeySubmission(PresentationAction)\n }\n\n @Dependency(\\.toast) var toast\n @Dependency(\\.apiKeyKeychain) var keychain\n\n var body: some ReducerOf {\n Reduce { state, action in\n switch action {\n case .appear:\n if isPreview { return .none }\n \n return .run { send in\n await send(.refreshAvailableAPIKeyNames)\n }\n case .closeButtonClicked:\n return .none\n \n case .addButtonClicked:\n state.apiKeySubmission = .init()\n \n return .none\n\n case let .deleteButtonClicked(name):\n do {\n try keychain.remove(name)\n return .run { send in\n await send(.refreshAvailableAPIKeyNames)\n }\n } catch {\n toast(error.localizedDescription, .error)\n return .none\n }\n\n case .refreshAvailableAPIKeyNames:\n do {\n let pairs = try keychain.getAll()\n state.availableAPIKeyNames = Array(pairs.keys).sorted()\n } catch {\n toast(error.localizedDescription, .error)\n }\n\n return .none\n\n case .apiKeySubmission(.presented(.saveFinished)):\n state.apiKeySubmission = nil\n return .run { send in\n await send(.refreshAvailableAPIKeyNames)\n }\n\n case .apiKeySubmission(.presented(.cancelButtonClicked)):\n state.apiKeySubmission = nil\n return .none\n\n case .apiKeySubmission:\n return .none\n }\n }\n .ifLet(\\.$apiKeySubmission, action: \\.apiKeySubmission) {\n APIKeySubmission()\n }\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/AccountSettings/APIKeyManagement/APIKeyPicker.swift", "content": "import ComposableArchitecture\nimport SwiftUI\n\nstruct APIKeyPicker: View {\n @Perception.Bindable var store: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n HStack {\n Picker(\n selection: $store.apiKeyName,\n content: {\n Text(\"No API Key\").tag(\"\")\n if store.availableAPIKeyNames.isEmpty {\n Text(\"No API key found, please add a new one โ†’\")\n }\n\n if !store.availableAPIKeyNames.contains(store.apiKeyName),\n !store.apiKeyName.isEmpty\n {\n Text(\"Key not found: \\(store.apiKeyName)\")\n .tag(store.apiKeyName)\n }\n\n ForEach(store.availableAPIKeyNames, id: \\.self) { name in\n Text(name).tag(name)\n }\n\n },\n label: { Text(\"API Key\") }\n )\n\n Button(action: { store.send(.manageAPIKeysButtonClicked) }) {\n Text(Image(systemName: \"key\"))\n }\n }.sheet(isPresented: $store.isAPIKeyManagementPresented) {\n WithPerceptionTracking {\n APIKeyManagementView(store: store.scope(\n state: \\.apiKeyManagement,\n action: \\.apiKeyManagement\n ))\n }\n }\n .onAppear {\n store.send(.appear)\n }\n }\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/AccountSettings/APIKeyManagement/APIKeySelection.swift", "content": "import Foundation\nimport SwiftUI\nimport ComposableArchitecture\n\n@Reducer\nstruct APIKeySelection {\n @ObservableState\n struct State: Equatable {\n var apiKeyName: String = \"\"\n var availableAPIKeyNames: [String] {\n apiKeyManagement.availableAPIKeyNames\n }\n var apiKeyManagement: APIKeyManagement.State = .init()\n var isAPIKeyManagementPresented: Bool = false\n }\n \n enum Action: Equatable, BindableAction {\n case appear\n case manageAPIKeysButtonClicked\n \n case binding(BindingAction)\n case apiKeyManagement(APIKeyManagement.Action)\n }\n \n @Dependency(\\.toast) var toast\n @Dependency(\\.apiKeyKeychain) var keychain\n \n var body: some ReducerOf {\n BindingReducer()\n \n Scope(state: \\.apiKeyManagement, action: \\.apiKeyManagement) {\n APIKeyManagement()\n }\n \n Reduce { state, action in\n switch action {\n case .appear:\n return .run { send in\n await send(.apiKeyManagement(.refreshAvailableAPIKeyNames))\n }\n \n case .manageAPIKeysButtonClicked:\n state.isAPIKeyManagementPresented = true\n return .none\n \n case .binding:\n return .none\n\n case .apiKeyManagement(.closeButtonClicked):\n state.isAPIKeyManagementPresented = false\n return .none\n \n case .apiKeyManagement:\n return .none\n }\n }\n }\n}\n" }, { "path": "Core/Sources/HostApp/AccountSettings/APIKeyManagement/APIKeySubmission.swift", "content": "import ComposableArchitecture\nimport Foundation\n\n@Reducer\nstruct APIKeySubmission {\n @ObservableState\n struct State: Equatable {\n var name: String = \"\"\n var key: String = \"\"\n }\n\n enum Action: Equatable, BindableAction {\n case binding(BindingAction)\n case saveButtonClicked\n case cancelButtonClicked\n case saveFinished\n }\n\n @Dependency(\\.toast) var toast\n @Dependency(\\.apiKeyKeychain) var keychain\n\n enum E: Error, LocalizedError {\n case nameIsEmpty\n case keyIsEmpty\n }\n\n var body: some ReducerOf {\n BindingReducer()\n\n Reduce { state, action in\n switch action {\n case .saveButtonClicked:\n do {\n guard !state.name.isEmpty else { throw E.nameIsEmpty }\n guard !state.key.isEmpty else { throw E.keyIsEmpty }\n\n try keychain.update(\n state.key,\n key: state.name.trimmingCharacters(in: .whitespacesAndNewlines)\n )\n return .run { send in\n await send(.saveFinished)\n }\n } catch {\n toast(error.localizedDescription, .error)\n return .none\n }\n\n case .cancelButtonClicked:\n return .none\n\n case .saveFinished:\n return .none\n\n case .binding:\n return .none\n }\n }\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/AccountSettings/ChatModelManagement/ChatModelEdit.swift", "content": "import AIModel\nimport ComposableArchitecture\nimport Dependencies\nimport Keychain\nimport OpenAIService\nimport Preferences\nimport SwiftUI\nimport Toast\n\n@Reducer\nstruct ChatModelEdit {\n @ObservableState\n struct State: Equatable, Identifiable {\n var id: String\n var name: String\n var format: ChatModel.Format\n var maxTokens: Int = 4000\n var supportsFunctionCalling: Bool = true\n var modelName: String = \"\"\n var ollamaKeepAlive: String = \"\"\n var apiVersion: String = \"\"\n var apiKeyName: String { apiKeySelection.apiKeyName }\n var baseURL: String { baseURLSelection.baseURL }\n var isFullURL: Bool { baseURLSelection.isFullURL }\n var availableModelNames: [String] = []\n var availableAPIKeys: [String] = []\n var isTesting = false\n var suggestedMaxTokens: Int?\n var apiKeySelection: APIKeySelection.State = .init()\n var baseURLSelection: BaseURLSelection.State = .init()\n var enforceMessageOrder: Bool = false\n var openAIOrganizationID: String = \"\"\n var openAIProjectID: String = \"\"\n var customHeaders: [ChatModel.Info.CustomHeaderInfo.HeaderField] = []\n var openAICompatibleSupportsMultipartMessageContent = true\n var requiresBeginWithUserMessage = false\n var customBody: String = \"\"\n var supportsImages: Bool = true\n }\n\n enum Action: Equatable, BindableAction {\n case binding(BindingAction)\n case appear\n case saveButtonClicked\n case cancelButtonClicked\n case refreshAvailableModelNames\n case testButtonClicked\n case testSucceeded(String)\n case testFailed(String)\n case checkSuggestedMaxTokens\n case selectModelFormat(ModelFormat)\n case apiKeySelection(APIKeySelection.Action)\n case baseURLSelection(BaseURLSelection.Action)\n }\n\n enum ModelFormat: CaseIterable {\n case openAI\n case azureOpenAI\n case googleAI\n case ollama\n case claude\n case gitHubCopilot\n case openAICompatible\n case deepSeekOpenAICompatible\n case openRouterOpenAICompatible\n case grokOpenAICompatible\n case mistralOpenAICompatible\n\n init(_ format: ChatModel.Format) {\n switch format {\n case .openAI:\n self = .openAI\n case .azureOpenAI:\n self = .azureOpenAI\n case .googleAI:\n self = .googleAI\n case .ollama:\n self = .ollama\n case .claude:\n self = .claude\n case .openAICompatible:\n self = .openAICompatible\n case .gitHubCopilot:\n self = .gitHubCopilot\n }\n }\n }\n\n var toast: (String, ToastType) -> Void {\n @Dependency(\\.namespacedToast) var toast\n return {\n toast($0, $1, \"ChatModelEdit\")\n }\n }\n\n @Dependency(\\.apiKeyKeychain) var keychain\n\n var body: some ReducerOf {\n BindingReducer()\n\n Scope(state: \\.apiKeySelection, action: \\.apiKeySelection) {\n APIKeySelection()\n }\n\n Scope(state: \\.baseURLSelection, action: \\.baseURLSelection) {\n BaseURLSelection()\n }\n\n Reduce { state, action in\n switch action {\n case .appear:\n return .run { send in\n await send(.refreshAvailableModelNames)\n await send(.checkSuggestedMaxTokens)\n }\n\n case .saveButtonClicked:\n return .none\n\n case .cancelButtonClicked:\n return .none\n\n case .testButtonClicked:\n guard !state.isTesting else { return .none }\n state.isTesting = true\n let model = ChatModel(state: state)\n return .run { send in\n do {\n let configuration = UserPreferenceChatGPTConfiguration().overriding {\n $0.model = model\n }\n let service = ChatGPTService(configuration: configuration)\n let stream = service.send(TemplateChatGPTMemory(\n memoryTemplate: .init(messages: [\n .init(chatMessage: .init(\n role: .system,\n content: \"You are a bot. Just do what is told.\"\n )),\n .init(chatMessage: .init(\n role: .assistant,\n content: \"Hello\"\n )),\n .init(chatMessage: .init(\n role: .user,\n content: \"Respond with \\\"Test succeeded.\\\"\"\n )),\n .init(chatMessage: .init(\n role: .user,\n content: \"Respond with \\\"Test succeeded.\\\"\"\n )),\n ]),\n configuration: configuration,\n functionProvider: NoChatGPTFunctionProvider()\n ))\n let streamReply = try await stream.asText()\n await send(.testSucceeded(streamReply))\n } catch {\n await send(.testFailed(error.localizedDescription))\n }\n }\n\n case let .testSucceeded(message):\n state.isTesting = false\n toast(message.trimmingCharacters(in: .whitespacesAndNewlines), .info)\n return .none\n\n case let .testFailed(message):\n state.isTesting = false\n toast(message.trimmingCharacters(in: .whitespacesAndNewlines), .error)\n return .none\n\n case .refreshAvailableModelNames:\n if state.format == .openAI {\n state.availableModelNames = ChatGPTModel.allCases.map(\\.rawValue)\n }\n\n return .none\n\n case .checkSuggestedMaxTokens:\n switch state.format {\n case .openAI:\n if let knownModel = ChatGPTModel(rawValue: state.modelName) {\n state.suggestedMaxTokens = knownModel.maxToken\n } else {\n state.suggestedMaxTokens = nil\n }\n return .none\n case .googleAI:\n if let knownModel = GoogleGenerativeAIModel(rawValue: state.modelName) {\n state.suggestedMaxTokens = knownModel.maxToken\n } else {\n state.suggestedMaxTokens = nil\n }\n return .none\n case .claude:\n if let knownModel = ClaudeChatCompletionsService\n .KnownModel(rawValue: state.modelName)\n {\n state.suggestedMaxTokens = knownModel.contextWindow\n } else {\n state.suggestedMaxTokens = nil\n }\n return .none\n case .gitHubCopilot:\n if let knownModel = AvailableGitHubCopilotModel(rawValue: state.modelName) {\n state.suggestedMaxTokens = knownModel.contextWindow\n } else {\n state.suggestedMaxTokens = nil\n }\n return .none\n default:\n state.suggestedMaxTokens = nil\n return .none\n }\n\n case let .selectModelFormat(format):\n switch format {\n case .openAI:\n state.format = .openAI\n case .azureOpenAI:\n state.format = .azureOpenAI\n case .googleAI:\n state.format = .googleAI\n case .ollama:\n state.format = .ollama\n case .claude:\n state.format = .claude\n case .gitHubCopilot:\n state.format = .gitHubCopilot\n case .openAICompatible:\n state.format = .openAICompatible\n case .deepSeekOpenAICompatible:\n state.format = .openAICompatible\n state.baseURLSelection.baseURL = \"https://api.deepseek.com\"\n state.baseURLSelection.isFullURL = false\n case .openRouterOpenAICompatible:\n state.format = .openAICompatible\n state.baseURLSelection.baseURL = \"https://openrouter.ai\"\n state.baseURLSelection.isFullURL = false\n case .grokOpenAICompatible:\n state.format = .openAICompatible\n state.baseURLSelection.baseURL = \"https://api.x.ai\"\n state.baseURLSelection.isFullURL = false\n case .mistralOpenAICompatible:\n state.format = .openAICompatible\n state.baseURLSelection.baseURL = \"https://api.mistral.ai\"\n state.baseURLSelection.isFullURL = false\n }\n return .none\n\n case .apiKeySelection:\n return .none\n\n case .baseURLSelection:\n return .none\n\n case .binding(\\.format):\n return .run { send in\n await send(.refreshAvailableModelNames)\n await send(.checkSuggestedMaxTokens)\n }\n\n case .binding(\\.modelName):\n return .run { send in\n await send(.checkSuggestedMaxTokens)\n }\n\n case .binding:\n return .none\n }\n }\n }\n}\n\nextension ChatModel {\n init(state: ChatModelEdit.State) {\n self.init(\n id: state.id,\n name: state.name,\n format: state.format,\n info: .init(\n apiKeyName: state.apiKeyName,\n baseURL: state.baseURL.trimmingCharacters(in: .whitespacesAndNewlines),\n isFullURL: state.isFullURL,\n maxTokens: state.maxTokens,\n supportsFunctionCalling: {\n switch state.format {\n case .googleAI, .ollama, .claude:\n return false\n case .azureOpenAI, .openAI, .openAICompatible, .gitHubCopilot:\n return state.supportsFunctionCalling\n }\n }(),\n supportsImage: state.supportsImages,\n modelName: state.modelName\n .trimmingCharacters(in: .whitespacesAndNewlines),\n openAIInfo: .init(\n organizationID: state.openAIOrganizationID,\n projectID: state.openAIProjectID\n ),\n ollamaInfo: .init(keepAlive: state.ollamaKeepAlive),\n googleGenerativeAIInfo: .init(apiVersion: state.apiVersion),\n openAICompatibleInfo: .init(\n enforceMessageOrder: state.enforceMessageOrder,\n supportsMultipartMessageContent: state\n .openAICompatibleSupportsMultipartMessageContent,\n requiresBeginWithUserMessage: state.requiresBeginWithUserMessage\n ),\n customHeaderInfo: .init(headers: state.customHeaders),\n customBodyInfo: .init(jsonBody: state.customBody)\n )\n )\n }\n\n func toState() -> ChatModelEdit.State {\n .init(\n id: id,\n name: name,\n format: format,\n maxTokens: info.maxTokens,\n supportsFunctionCalling: info.supportsFunctionCalling,\n modelName: info.modelName,\n ollamaKeepAlive: info.ollamaInfo.keepAlive,\n apiVersion: info.googleGenerativeAIInfo.apiVersion,\n apiKeySelection: .init(\n apiKeyName: info.apiKeyName,\n apiKeyManagement: .init(availableAPIKeyNames: [info.apiKeyName])\n ),\n baseURLSelection: .init(baseURL: info.baseURL, isFullURL: info.isFullURL),\n enforceMessageOrder: info.openAICompatibleInfo.enforceMessageOrder,\n openAIOrganizationID: info.openAIInfo.organizationID,\n openAIProjectID: info.openAIInfo.projectID,\n customHeaders: info.customHeaderInfo.headers,\n openAICompatibleSupportsMultipartMessageContent: info.openAICompatibleInfo\n .supportsMultipartMessageContent,\n requiresBeginWithUserMessage: info.openAICompatibleInfo.requiresBeginWithUserMessage,\n customBody: info.customBodyInfo.jsonBody,\n supportsImages: info.supportsImage\n )\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/AccountSettings/ChatModelManagement/ChatModelEditView.swift", "content": "import AIModel\nimport ComposableArchitecture\nimport OpenAIService\nimport Preferences\nimport SwiftUI\n\n@MainActor\nstruct ChatModelEditView: View {\n @Perception.Bindable var store: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n ScrollView {\n VStack(spacing: 0) {\n Form {\n NameTextField(store: store)\n FormatPicker(store: store)\n\n switch store.format {\n case .openAI:\n OpenAIForm(store: store)\n case .azureOpenAI:\n AzureOpenAIForm(store: store)\n case .openAICompatible:\n OpenAICompatibleForm(store: store)\n case .googleAI:\n GoogleAIForm(store: store)\n case .ollama:\n OllamaForm(store: store)\n case .claude:\n ClaudeForm(store: store)\n case .gitHubCopilot:\n GitHubCopilotForm(store: store)\n }\n }\n .padding()\n\n Divider()\n\n HStack {\n HStack(spacing: 8) {\n Button(\"Test\") {\n store.send(.testButtonClicked)\n }\n .disabled(store.isTesting)\n\n if store.isTesting {\n ProgressView()\n .controlSize(.small)\n }\n }\n\n CustomBodyEdit(store: store)\n .disabled({\n switch store.format {\n case .openAI, .openAICompatible, .claude:\n return false\n default:\n return true\n }\n }())\n CustomHeaderEdit(store: store)\n .disabled({\n switch store.format {\n case .openAI, .openAICompatible, .ollama, .gitHubCopilot, .claude:\n return false\n default:\n return true\n }\n }())\n\n Spacer()\n\n Button(\"Cancel\") {\n store.send(.cancelButtonClicked)\n }\n .keyboardShortcut(.cancelAction)\n\n Button(action: { store.send(.saveButtonClicked) }) {\n Text(\"Save\")\n }\n .keyboardShortcut(.defaultAction)\n }\n .padding()\n }\n }\n .textFieldStyle(.roundedBorder)\n .onAppear {\n store.send(.appear)\n }\n .fixedSize(horizontal: false, vertical: true)\n .handleToast(namespace: \"ChatModelEdit\")\n }\n }\n\n struct NameTextField: View {\n @Perception.Bindable var store: StoreOf\n var body: some View {\n WithPerceptionTracking {\n TextField(\"Name\", text: $store.name)\n }\n }\n }\n\n struct FormatPicker: View {\n @Perception.Bindable var store: StoreOf\n var body: some View {\n WithPerceptionTracking {\n Picker(\n selection: Binding(\n get: { .init(store.format) },\n set: { store.send(.selectModelFormat($0)) }\n ),\n content: {\n ForEach(\n ChatModelEdit.ModelFormat.allCases,\n id: \\.self\n ) { format in\n switch format {\n case .openAI:\n Text(\"OpenAI\")\n case .azureOpenAI:\n Text(\"Azure OpenAI\")\n case .openAICompatible:\n Text(\"OpenAI Compatible\")\n case .googleAI:\n Text(\"Google AI\")\n case .ollama:\n Text(\"Ollama\")\n case .claude:\n Text(\"Claude\")\n case .gitHubCopilot:\n Text(\"GitHub Copilot\")\n case .deepSeekOpenAICompatible:\n Text(\"DeepSeek (OpenAI Compatible)\")\n case .openRouterOpenAICompatible:\n Text(\"OpenRouter (OpenAI Compatible)\")\n case .grokOpenAICompatible:\n Text(\"Grok (OpenAI Compatible)\")\n case .mistralOpenAICompatible:\n Text(\"Mistral (OpenAI Compatible)\")\n }\n }\n },\n label: { Text(\"Format\") }\n )\n .pickerStyle(.menu)\n }\n }\n }\n\n struct BaseURLTextField: View {\n let store: StoreOf\n var title: String = \"Base URL\"\n let prompt: Text?\n @ViewBuilder var trailingContent: () -> V\n\n var body: some View {\n WithPerceptionTracking {\n BaseURLPicker(\n title: title,\n prompt: prompt,\n store: store.scope(\n state: \\.baseURLSelection,\n action: \\.baseURLSelection\n ),\n trailingContent: trailingContent\n )\n }\n }\n }\n\n struct SupportsFunctionCallingToggle: View {\n @Perception.Bindable var store: StoreOf\n var body: some View {\n WithPerceptionTracking {\n Toggle(\n \"Supports Function Calling\",\n isOn: $store.supportsFunctionCalling\n )\n\n Text(\n \"Function calling is required by some features, if this model doesn't support function calling, you should turn it off to avoid undefined behaviors.\"\n )\n .foregroundColor(.secondary)\n .font(.callout)\n .dynamicHeightTextInFormWorkaround()\n }\n }\n }\n\n struct MaxTokensTextField: View {\n @Perception.Bindable var store: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n HStack {\n let textFieldBinding = Binding(\n get: { String(store.maxTokens) },\n set: {\n if let selectionMaxToken = Int($0) {\n $store.maxTokens.wrappedValue = selectionMaxToken\n } else {\n $store.maxTokens.wrappedValue = 0\n }\n }\n )\n\n TextField(text: textFieldBinding) {\n Text(\"Context Window\")\n .multilineTextAlignment(.trailing)\n }\n .overlay(alignment: .trailing) {\n Stepper(\n value: $store.maxTokens,\n in: 0...Int.max,\n step: 100\n ) {\n EmptyView()\n }\n }\n .foregroundColor({\n guard let max = store.suggestedMaxTokens else {\n return .primary\n }\n if store.maxTokens > max {\n return .red\n }\n return .primary\n }() as Color)\n\n if let max = store.suggestedMaxTokens {\n Text(\"Max: \\(max)\")\n }\n }\n }\n }\n }\n\n struct ApiKeyNamePicker: View {\n let store: StoreOf\n var body: some View {\n WithPerceptionTracking {\n APIKeyPicker(store: store.scope(\n state: \\.apiKeySelection,\n action: \\.apiKeySelection\n ))\n }\n }\n }\n\n struct CustomBodyEdit: View {\n @Perception.Bindable var store: StoreOf\n @State private var isEditing = false\n @Dependency(\\.namespacedToast) var toast\n\n var body: some View {\n Button(\"Custom Body\") {\n isEditing = true\n }\n .sheet(isPresented: $isEditing) {\n WithPerceptionTracking {\n VStack {\n TextEditor(text: $store.customBody)\n .font(Font.system(.body, design: .monospaced))\n .padding(4)\n .frame(minHeight: 120)\n .multilineTextAlignment(.leading)\n .overlay(\n RoundedRectangle(cornerRadius: 4)\n .stroke(Color(nsColor: .separatorColor), lineWidth: 1)\n )\n .handleToast(namespace: \"CustomBodyEdit\")\n\n Text(\n \"The custom body will be added to the request body. Please use it to add parameters that are not yet available in the form. It should be a valid JSON object.\"\n )\n .foregroundColor(.secondary)\n .font(.callout)\n .padding(.bottom)\n\n Button(action: {\n if store.customBody.trimmingCharacters(in: .whitespacesAndNewlines)\n .isEmpty\n {\n isEditing = false\n return\n }\n guard let _ = try? JSONSerialization\n .jsonObject(with: store.customBody.data(using: .utf8) ?? Data())\n else {\n toast(\"Invalid JSON object\", .error, \"CustomBodyEdit\")\n return\n }\n isEditing = false\n }) {\n Text(\"Done\")\n }\n .keyboardShortcut(.defaultAction)\n }\n .padding()\n .frame(width: 600, height: 500)\n .background(Color(nsColor: .windowBackgroundColor))\n }\n }\n }\n }\n\n struct CustomHeaderEdit: View {\n @Perception.Bindable var store: StoreOf\n @State private var isEditing = false\n\n var body: some View {\n Button(\"Custom Headers\") {\n isEditing = true\n }\n .sheet(isPresented: $isEditing) {\n WithPerceptionTracking {\n CustomHeaderSettingsView(headers: $store.customHeaders)\n }\n }\n }\n }\n\n struct OpenAIForm: View {\n @Perception.Bindable var store: StoreOf\n var body: some View {\n WithPerceptionTracking {\n BaseURLTextField(store: store, prompt: Text(\"https://api.openai.com\")) {\n Text(\"/v1/chat/completions\")\n }\n ApiKeyNamePicker(store: store)\n\n TextField(\"Model Name\", text: $store.modelName)\n .overlay(alignment: .trailing) {\n Picker(\n \"\",\n selection: $store.modelName,\n content: {\n if ChatGPTModel(rawValue: store.modelName) == nil {\n Text(\"Custom Model\").tag(store.modelName)\n }\n ForEach(ChatGPTModel.allCases, id: \\.self) { model in\n Text(model.rawValue).tag(model.rawValue)\n }\n }\n )\n .frame(width: 20)\n }\n\n MaxTokensTextField(store: store)\n SupportsFunctionCallingToggle(store: store)\n\n TextField(text: $store.openAIOrganizationID, prompt: Text(\"Optional\")) {\n Text(\"Organization ID\")\n }\n\n TextField(text: $store.openAIProjectID, prompt: Text(\"Optional\")) {\n Text(\"Project ID\")\n }\n \n Toggle(isOn: $store.supportsImages) {\n Text(\"Supports Images\")\n }\n\n VStack(alignment: .leading, spacing: 8) {\n Text(Image(systemName: \"exclamationmark.triangle.fill\")) + Text(\n \" To get an API key, please visit [https://platform.openai.com/api-keys](https://platform.openai.com/api-keys)\"\n )\n\n Text(Image(systemName: \"exclamationmark.triangle.fill\")) + Text(\n \" If you don't have access to GPT-4, you may need to visit [https://platform.openai.com/account/billing/overview](https://platform.openai.com/account/billing/overview) to buy some credits. A ChatGPT Plus subscription is not enough to access GPT-4 through API.\"\n )\n }\n .padding(.vertical)\n }\n }\n }\n\n struct AzureOpenAIForm: View {\n @Perception.Bindable var store: StoreOf\n var body: some View {\n WithPerceptionTracking {\n BaseURLTextField(store: store, prompt: Text(\"https://xxxx.openai.azure.com\")) {\n EmptyView()\n }\n ApiKeyNamePicker(store: store)\n\n TextField(\"Deployment Name\", text: $store.modelName)\n\n MaxTokensTextField(store: store)\n SupportsFunctionCallingToggle(store: store)\n \n Toggle(isOn: $store.supportsImages) {\n Text(\"Supports Images\")\n }\n }\n }\n }\n\n struct OpenAICompatibleForm: View {\n @Perception.Bindable var store: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n Picker(\n selection: $store.baseURLSelection.isFullURL,\n content: {\n Text(\"Base URL\").tag(false)\n Text(\"Full URL\").tag(true)\n },\n label: { Text(\"URL\") }\n )\n .pickerStyle(.segmented)\n\n BaseURLTextField(\n store: store,\n title: \"\",\n prompt: store.isFullURL\n ? Text(\"https://api.openai.com/v1/chat/completions\")\n : Text(\"https://api.openai.com\")\n ) {\n if !store.isFullURL {\n Text(\"/v1/chat/completions\")\n }\n }\n\n ApiKeyNamePicker(store: store)\n\n TextField(\"Model Name\", text: $store.modelName)\n\n MaxTokensTextField(store: store)\n SupportsFunctionCallingToggle(store: store)\n\n Toggle(isOn: $store.enforceMessageOrder) {\n Text(\"Enforce message order to be user/assistant alternated\")\n }\n\n Toggle(isOn: $store.openAICompatibleSupportsMultipartMessageContent) {\n Text(\"Support multi-part message content\")\n }\n\n Toggle(isOn: $store.requiresBeginWithUserMessage) {\n Text(\"Requires the first message to be from the user\")\n }\n \n Toggle(isOn: $store.supportsImages) {\n Text(\"Supports Images\")\n }\n }\n }\n }\n\n struct GoogleAIForm: View {\n @Perception.Bindable var store: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n BaseURLTextField(\n store: store,\n prompt: Text(\"https://generativelanguage.googleapis.com\")\n ) {\n Text(\"/v1\")\n }\n\n ApiKeyNamePicker(store: store)\n\n TextField(\"Model Name\", text: $store.modelName)\n .overlay(alignment: .trailing) {\n Picker(\n \"\",\n selection: $store.modelName,\n content: {\n if GoogleGenerativeAIModel(rawValue: store.modelName) == nil {\n Text(\"Custom Model\").tag(store.modelName)\n }\n ForEach(GoogleGenerativeAIModel.allCases, id: \\.self) { model in\n Text(model.rawValue).tag(model.rawValue)\n }\n }\n )\n .frame(width: 20)\n }\n\n MaxTokensTextField(store: store)\n\n TextField(\"API Version\", text: $store.apiVersion, prompt: Text(\"v1\"))\n \n Toggle(isOn: $store.supportsImages) {\n Text(\"Supports Images\")\n }\n }\n }\n }\n\n struct OllamaForm: View {\n @Perception.Bindable var store: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n BaseURLTextField(store: store, prompt: Text(\"http://127.0.0.1:11434\")) {\n Text(\"/api/chat\")\n }\n\n ApiKeyNamePicker(store: store)\n\n TextField(\"Model Name\", text: $store.modelName)\n\n MaxTokensTextField(store: store)\n\n TextField(text: $store.ollamaKeepAlive, prompt: Text(\"Default Value\")) {\n Text(\"Keep Alive\")\n }\n\n Toggle(isOn: $store.supportsImages) {\n Text(\"Supports Images\")\n }\n \n VStack(alignment: .leading, spacing: 8) {\n Text(Image(systemName: \"exclamationmark.triangle.fill\")) + Text(\n \" For more details, please visit [https://ollama.com](https://ollama.com).\"\n )\n }\n .padding(.vertical)\n }\n }\n }\n\n struct ClaudeForm: View {\n @Perception.Bindable var store: StoreOf\n var body: some View {\n WithPerceptionTracking {\n BaseURLTextField(store: store, prompt: Text(\"https://api.anthropic.com\")) {\n Text(\"/v1/messages\")\n }\n\n ApiKeyNamePicker(store: store)\n\n TextField(\"Model Name\", text: $store.modelName)\n .overlay(alignment: .trailing) {\n Picker(\n \"\",\n selection: $store.modelName,\n content: {\n if ClaudeChatCompletionsService\n .KnownModel(rawValue: store.modelName) == nil\n {\n Text(\"Custom Model\").tag(store.modelName)\n }\n ForEach(\n ClaudeChatCompletionsService.KnownModel.allCases,\n id: \\.self\n ) { model in\n Text(model.rawValue).tag(model.rawValue)\n }\n }\n )\n .frame(width: 20)\n }\n\n MaxTokensTextField(store: store)\n \n Toggle(isOn: $store.supportsImages) {\n Text(\"Supports Images\")\n }\n\n VStack(alignment: .leading, spacing: 8) {\n Text(Image(systemName: \"exclamationmark.triangle.fill\")) + Text(\n \" For more details, please visit [https://anthropic.com](https://anthropic.com).\"\n )\n }\n .padding(.vertical)\n }\n }\n }\n\n struct GitHubCopilotForm: View {\n @Perception.Bindable var store: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n #warning(\"Todo: use the old picker and update the context window limit.\")\n GitHubCopilotModelPicker(\n title: \"Model Name\",\n hasDefaultModel: false,\n gitHubCopilotModelId: $store.modelName\n )\n\n MaxTokensTextField(store: store)\n SupportsFunctionCallingToggle(store: store)\n\n Toggle(isOn: $store.enforceMessageOrder) {\n Text(\"Enforce message order to be user/assistant alternated\")\n }\n\n Toggle(isOn: $store.openAICompatibleSupportsMultipartMessageContent) {\n Text(\"Support multi-part message content\")\n }\n \n Toggle(isOn: $store.supportsImages) {\n Text(\"Supports Images\")\n }\n\n VStack(alignment: .leading, spacing: 8) {\n Text(Image(systemName: \"exclamationmark.triangle.fill\")) + Text(\n \" Please login in the GitHub Copilot settings to use the model.\"\n )\n\n Text(Image(systemName: \"exclamationmark.triangle.fill\")) + Text(\n \" This will call the APIs directly, which may not be allowed by GitHub. But it's used in other popular apps like Zed.\"\n )\n }\n .dynamicHeightTextInFormWorkaround()\n .padding(.vertical)\n }\n }\n }\n}\n\n#Preview(\"OpenAI\") {\n ChatModelEditView(\n store: .init(\n initialState: ChatModel(\n id: \"3\",\n name: \"Test Model 3\",\n format: .openAI,\n info: .init(\n apiKeyName: \"key\",\n baseURL: \"apple.com\",\n maxTokens: 3000,\n supportsFunctionCalling: false,\n modelName: \"gpt-3.5-turbo\"\n )\n ).toState(),\n reducer: { ChatModelEdit() }\n )\n )\n}\n\n#Preview(\"OpenAI Compatible\") {\n ChatModelEditView(\n store: .init(\n initialState: ChatModel(\n id: \"3\",\n name: \"Test Model 3\",\n format: .openAICompatible,\n info: .init(\n apiKeyName: \"key\",\n baseURL: \"apple.com\",\n isFullURL: false,\n maxTokens: 3000,\n supportsFunctionCalling: false,\n modelName: \"gpt-3.5-turbo\"\n )\n ).toState(),\n reducer: { ChatModelEdit() }\n )\n )\n}\n\n" }, { "path": "Core/Sources/HostApp/AccountSettings/ChatModelManagement/ChatModelManagement.swift", "content": "import AIModel\nimport ComposableArchitecture\nimport Keychain\nimport Preferences\nimport SwiftUI\n\nextension ChatModel: ManageableAIModel {\n var formatName: String {\n switch format {\n case .openAI: return \"OpenAI\"\n case .azureOpenAI: return \"Azure OpenAI\"\n case .openAICompatible: return \"OpenAI Compatible\"\n case .googleAI: return \"Google Generative AI\"\n case .ollama: return \"Ollama\"\n case .claude: return \"Claude\"\n case .gitHubCopilot: return \"GitHub Copilot\"\n }\n }\n\n @ViewBuilder\n var infoDescriptors: some View {\n Text(info.modelName)\n\n if !info.baseURL.isEmpty {\n Image(systemName: \"line.diagonal\")\n Text(info.baseURL)\n }\n\n Image(systemName: \"line.diagonal\")\n\n Text(\"\\(info.maxTokens) tokens\")\n\n Image(systemName: \"line.diagonal\")\n\n Text(\n \"function calling \\(info.supportsFunctionCalling ? Image(systemName: \"checkmark.square\") : Image(systemName: \"xmark.square\"))\"\n )\n }\n}\n\n@Reducer\nstruct ChatModelManagement: AIModelManagement {\n typealias Model = ChatModel\n\n @ObservableState\n struct State: Equatable, AIModelManagementState {\n typealias Model = ChatModel\n var models: IdentifiedArrayOf = []\n @Presents var editingModel: ChatModelEdit.State?\n var selectedModelId: String? { editingModel?.id }\n }\n\n enum Action: Equatable, AIModelManagementAction {\n typealias Model = ChatModel\n case appear\n case createModel\n case removeModel(id: Model.ID)\n case selectModel(id: Model.ID)\n case duplicateModel(id: Model.ID)\n case moveModel(from: IndexSet, to: Int)\n case chatModelItem(PresentationAction)\n }\n\n @Dependency(\\.toast) var toast\n @Dependency(\\.userDefaults) var userDefaults\n\n var body: some ReducerOf {\n Reduce { state, action in\n switch action {\n case .appear:\n if isPreview { return .none }\n state.models = .init(\n userDefaults.value(for: \\.chatModels),\n id: \\.id,\n uniquingIDsWith: { a, _ in a }\n )\n\n return .none\n\n case .createModel:\n state.editingModel = .init(\n id: UUID().uuidString,\n name: \"New Model\",\n format: .openAI\n )\n return .none\n\n case let .removeModel(id):\n state.models.remove(id: id)\n persist(state)\n return .none\n\n case let .selectModel(id):\n guard let model = state.models[id: id] else { return .none }\n state.editingModel = model.toState()\n return .none\n\n case let .duplicateModel(id):\n guard var model = state.models[id: id] else { return .none }\n model.id = UUID().uuidString\n model.name += \" (Copy)\"\n\n if let index = state.models.index(id: id) {\n state.models.insert(model, at: index + 1)\n } else {\n state.models.append(model)\n }\n persist(state)\n return .none\n\n case let .moveModel(from, to):\n state.models.move(fromOffsets: from, toOffset: to)\n persist(state)\n return .none\n\n case .chatModelItem(.presented(.saveButtonClicked)):\n guard let editingModel = state.editingModel, validateModel(editingModel)\n else { return .none }\n\n if let index = state.models\n .firstIndex(where: { $0.id == editingModel.id })\n {\n state.models[index] = .init(state: editingModel)\n } else {\n state.models.append(.init(state: editingModel))\n }\n persist(state)\n return .run { send in\n await send(.chatModelItem(.dismiss))\n }\n\n case .chatModelItem(.presented(.cancelButtonClicked)):\n return .run { send in\n await send(.chatModelItem(.dismiss))\n }\n\n case .chatModelItem:\n return .none\n }\n }.ifLet(\\.$editingModel, action: \\.chatModelItem) {\n ChatModelEdit()\n }\n }\n\n func persist(_ state: State) {\n let models = state.models\n userDefaults.set(Array(models), for: \\.chatModels)\n }\n\n func validateModel(_ chatModel: ChatModelEdit.State) -> Bool {\n guard !chatModel.name.isEmpty else {\n toast(\"Model name cannot be empty\", .error)\n return false\n }\n guard !chatModel.id.isEmpty else {\n toast(\"Model ID cannot be empty\", .error)\n return false\n }\n\n guard !chatModel.modelName.isEmpty else {\n toast(\"Model name cannot be empty\", .error)\n return false\n }\n return true\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/AccountSettings/ChatModelManagement/ChatModelManagementView.swift", "content": "import AIModel\nimport ComposableArchitecture\nimport SwiftUI\n\nstruct ChatModelManagementView: View {\n @Perception.Bindable var store: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n AIModelManagementView(store: store)\n .sheet(item: $store.scope(\n state: \\.editingModel,\n action: \\.chatModelItem\n )) { store in\n ChatModelEditView(store: store)\n .frame(width: 800)\n }\n }\n }\n}\n\n// MARK: - Previews\n\nclass ChatModelManagementView_Previews: PreviewProvider {\n static var previews: some View {\n ChatModelManagementView(\n store: .init(\n initialState: .init(\n models: IdentifiedArray(uniqueElements: [\n ChatModel(\n id: \"1\",\n name: \"Test Model\",\n format: .openAI,\n info: .init(\n apiKeyName: \"key\",\n baseURL: \"google.com\",\n maxTokens: 3000,\n supportsFunctionCalling: true,\n modelName: \"gpt-3.5-turbo\"\n )\n ),\n ChatModel(\n id: \"2\",\n name: \"Test Model 2\",\n format: .azureOpenAI,\n info: .init(\n apiKeyName: \"key\",\n baseURL: \"apple.com\",\n maxTokens: 3000,\n supportsFunctionCalling: false,\n modelName: \"gpt-3.5-turbo\"\n )\n ),\n ChatModel(\n id: \"3\",\n name: \"Test Model 3\",\n format: .openAICompatible,\n info: .init(\n apiKeyName: \"key\",\n baseURL: \"apple.com\",\n maxTokens: 3000,\n supportsFunctionCalling: false,\n modelName: \"gpt-3.5-turbo\"\n )\n ),\n ]),\n editingModel: ChatModel(\n id: \"3\",\n name: \"Test Model 3\",\n format: .openAICompatible,\n info: .init(\n apiKeyName: \"key\",\n baseURL: \"apple.com\",\n maxTokens: 3000,\n supportsFunctionCalling: false,\n modelName: \"gpt-3.5-turbo\"\n )\n ).toState()\n ),\n reducer: { ChatModelManagement() }\n )\n )\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/AccountSettings/CodeiumView.swift", "content": "import CodeiumService\nimport Foundation\nimport SharedUIComponents\nimport SwiftUI\n\nstruct CodeiumView: View {\n class ViewModel: ObservableObject {\n let codeiumAuthService = CodeiumAuthService()\n let installationManager = CodeiumInstallationManager()\n @Published var isSignedIn: Bool\n @Published var installationStatus: CodeiumInstallationManager.InstallationStatus\n @Published var installationStep: CodeiumInstallationManager.InstallationStep?\n @AppStorage(\\.codeiumVerboseLog) var codeiumVerboseLog\n @AppStorage(\\.codeiumEnterpriseMode) var codeiumEnterpriseMode\n @AppStorage(\\.codeiumPortalUrl) var codeiumPortalUrl\n @AppStorage(\\.codeiumApiUrl) var codeiumApiUrl\n @AppStorage(\\.codeiumIndexEnabled) var indexEnabled\n\n init() {\n isSignedIn = codeiumAuthService.isSignedIn\n installationStatus = .notInstalled\n Task { @MainActor in\n installationStatus = await installationManager.checkInstallation()\n }\n }\n\n init(\n isSignedIn: Bool,\n installationStatus: CodeiumInstallationManager.InstallationStatus,\n installationStep: CodeiumInstallationManager.InstallationStep?\n ) {\n assert(isPreview)\n self.isSignedIn = isSignedIn\n self.installationStatus = installationStatus\n self.installationStep = installationStep\n }\n\n func generateAuthURL() -> URL {\n if codeiumEnterpriseMode && (codeiumPortalUrl != \"\") {\n return URL(\n string: codeiumPortalUrl +\n \"/profile?response_type=token&redirect_uri=show-auth-token&state=\\(UUID().uuidString)&scope=openid%20profile%20email&redirect_parameters_type=query\"\n )!\n }\n\n return URL(\n string: \"https://www.codeium.com/profile?response_type=token&redirect_uri=show-auth-token&state=\\(UUID().uuidString)&scope=openid%20profile%20email&redirect_parameters_type=query\"\n )!\n }\n\n func signIn(token: String) async throws {\n try await codeiumAuthService.signIn(token: token)\n Task { @MainActor in isSignedIn = true }\n }\n\n func signOut() async throws {\n try await codeiumAuthService.signOut()\n Task { @MainActor in isSignedIn = false }\n }\n\n func refreshInstallationStatus() {\n Task { @MainActor in\n installationStatus = await installationManager.checkInstallation()\n }\n }\n\n func install() async throws {\n defer { refreshInstallationStatus() }\n do {\n for try await step in installationManager.installLatestVersion() {\n Task { @MainActor in\n self.installationStep = step\n }\n }\n Task {\n try await Task.sleep(nanoseconds: 1_000_000_000)\n Task { @MainActor in\n self.installationStep = nil\n }\n }\n } catch {\n Task { @MainActor in\n installationStep = nil\n }\n throw error\n }\n }\n\n func uninstall() {\n Task {\n defer { refreshInstallationStatus() }\n try await installationManager.uninstall()\n }\n }\n }\n\n @StateObject var viewModel = ViewModel()\n @Environment(\\.toast) var toast\n @State var isSignInPanelPresented = false\n\n var installButton: some View {\n Button(action: {\n Task {\n do {\n try await viewModel.install()\n } catch {\n toast(error.localizedDescription, .error)\n }\n }\n }) {\n Text(\"Install\")\n }\n .disabled(viewModel.installationStep != nil)\n }\n\n var updateButton: some View {\n Button(action: {\n Task {\n do {\n try await viewModel.install()\n } catch {\n toast(error.localizedDescription, .error)\n }\n }\n }) {\n Text(\"Update\")\n }\n .disabled(viewModel.installationStep != nil)\n }\n\n var uninstallButton: some View {\n Button(action: {\n viewModel.uninstall()\n }) {\n Text(\"Uninstall\")\n }\n .disabled(viewModel.installationStep != nil)\n }\n\n var body: some View {\n VStack(alignment: .leading) {\n SubSection(title: Text(\"Codeium Language Server\")) {\n switch viewModel.installationStatus {\n case .notInstalled:\n HStack {\n Text(\"Language Server Version: Not Installed\")\n installButton\n }\n case let .installed(version):\n HStack {\n Text(\"Language Server Version: \\(version)\")\n uninstallButton\n }\n case let .outdated(current: current, latest: latest, _):\n HStack {\n Text(\"Language Server Version: \\(current) (Update Available: \\(latest))\")\n uninstallButton\n updateButton\n }\n case let .unsupported(current: current, latest: latest):\n HStack {\n Text(\"Language Server Version: \\(current) (Supported Version: \\(latest))\")\n uninstallButton\n updateButton\n }\n }\n\n if viewModel.isSignedIn {\n Text(\"Status: Signed In\")\n\n Button(action: {\n Task {\n do {\n try await viewModel.signOut()\n } catch {\n toast(error.localizedDescription, .error)\n }\n }\n }) {\n Text(\"Sign Out\")\n }\n } else {\n Text(\"Status: Not Signed In\")\n\n Button(action: {\n isSignInPanelPresented = true\n }) {\n Text(\"Sign In\")\n }\n }\n }\n .sheet(isPresented: $isSignInPanelPresented) {\n CodeiumSignInView(viewModel: viewModel, isPresented: $isSignInPanelPresented)\n }\n .onChange(of: viewModel.installationStep) { newValue in\n if let step = newValue {\n switch step {\n case .downloading:\n toast(\"Downloading..\", .info)\n case .uninstalling:\n toast(\"Uninstalling old version..\", .info)\n case .decompressing:\n toast(\"Decompressing..\", .info)\n case .done:\n toast(\"Done!\", .info)\n }\n }\n }\n \n SubSection(title: Text(\"Indexing\")) {\n Form {\n Toggle(\"Enable Indexing\", isOn: $viewModel.indexEnabled)\n }\n }\n\n SubSection(title: Text(\"Enterprise\")) {\n Form {\n Toggle(\"Codeium Enterprise Mode\", isOn: $viewModel.codeiumEnterpriseMode)\n TextField(\"Codeium Portal URL\", text: $viewModel.codeiumPortalUrl)\n TextField(\"Codeium API URL\", text: $viewModel.codeiumApiUrl)\n }\n }\n\n SettingsDivider(\"Advanced\")\n\n Form {\n Toggle(\"Verbose Log\", isOn: $viewModel.codeiumVerboseLog)\n }\n }\n }\n}\n\nstruct CodeiumSignInView: View {\n let viewModel: CodeiumView.ViewModel\n @Binding var isPresented: Bool\n @Environment(\\.openURL) var openURL\n @Environment(\\.toast) var toast\n @State var isGeneratingKey = false\n @State var token = \"\"\n\n var body: some View {\n VStack {\n Text(\n \"You will be redirected to codeium.com. Please paste the generated token below and click the \\\"Sign In\\\" button.\"\n )\n\n TextEditor(text: $token)\n .font(Font.system(.body, design: .monospaced))\n .padding(4)\n .frame(minHeight: 120)\n .multilineTextAlignment(.leading)\n .overlay(\n RoundedRectangle(cornerRadius: 4)\n .stroke(Color(nsColor: .separatorColor), lineWidth: 1)\n )\n\n HStack {\n Spacer()\n\n Button(action: {\n isPresented = false\n }) {\n Text(\"Cancel\")\n }\n\n Button(action: {\n isGeneratingKey = true\n Task {\n do {\n try await viewModel.signIn(token: token)\n isGeneratingKey = false\n isPresented = false\n } catch {\n isGeneratingKey = false\n toast(error.localizedDescription, .error)\n }\n }\n }) {\n Text(isGeneratingKey ? \"Signing In..\" : \"Sign In\")\n }\n .disabled(isGeneratingKey)\n .keyboardShortcut(.defaultAction)\n }\n }\n .padding()\n .onAppear {\n openURL(viewModel.generateAuthURL())\n }\n }\n}\n\nstruct CodeiumView_Previews: PreviewProvider {\n class TestViewModel: CodeiumView.ViewModel {\n override func generateAuthURL() -> URL {\n return URL(string: \"about:blank\")!\n }\n\n override func signIn(token: String) async throws {}\n\n override func signOut() async throws {}\n\n override func refreshInstallationStatus() {}\n\n override func install() async throws {}\n\n override func uninstall() {}\n }\n\n static var previews: some View {\n ScrollView {\n VStack(alignment: .leading, spacing: 8) {\n CodeiumView(viewModel: TestViewModel(\n isSignedIn: false,\n installationStatus: .notInstalled,\n installationStep: nil\n ))\n \n CodeiumView(viewModel: TestViewModel(\n isSignedIn: true,\n installationStatus: .installed(\"1.2.9\"),\n installationStep: nil\n ))\n \n CodeiumView(viewModel: TestViewModel(\n isSignedIn: true,\n installationStatus: .outdated(current: \"1.2.9\", latest: \"1.3.0\", mandatory: true),\n installationStep: .downloading\n ))\n \n CodeiumView(viewModel: TestViewModel(\n isSignedIn: true,\n installationStatus: .unsupported(current: \"1.5.9\", latest: \"1.3.0\"),\n installationStep: .downloading\n ))\n }\n .padding(8)\n }\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/AccountSettings/CustomHeaderSettingsView.swift", "content": "import AIModel\nimport Foundation\nimport SwiftUI\n\nstruct CustomHeaderSettingsView: View {\n @Binding var headers: [ChatModel.Info.CustomHeaderInfo.HeaderField]\n @Environment(\\.dismiss) var dismiss\n @State private var newKey = \"\"\n @State private var newValue = \"\"\n\n var body: some View {\n VStack {\n List {\n ForEach(headers.indices, id: \\.self) { index in\n HStack {\n TextField(\"Key\", text: Binding(\n get: { headers[index].key },\n set: { newKey in\n headers[index].key = newKey\n }\n ))\n TextField(\"Value\", text: Binding(\n get: { headers[index].value },\n set: { headers[index].value = $0 }\n ))\n Button(action: {\n headers.remove(at: index)\n }) {\n Image(systemName: \"trash\")\n .foregroundColor(.red)\n }\n }\n }\n\n HStack {\n TextField(\"New Key\", text: $newKey)\n TextField(\"New Value\", text: $newValue)\n Button(action: {\n if !newKey.isEmpty {\n headers.append(ChatModel.Info.CustomHeaderInfo.HeaderField(\n key: newKey,\n value: newValue\n ))\n newKey = \"\"\n newValue = \"\"\n }\n }) {\n Image(systemName: \"plus.circle.fill\")\n .foregroundColor(.green)\n }\n }\n }\n\n HStack {\n Spacer()\n Button(\"Done\") {\n dismiss()\n }\n }.padding()\n }\n .frame(height: 500)\n }\n}\n\n#Preview {\n struct V: View {\n @State var headers: [ChatModel.Info.CustomHeaderInfo.HeaderField] = [\n .init(key: \"key\", value: \"value\"),\n .init(key: \"key2\", value: \"value2\"),\n ]\n var body: some View {\n CustomHeaderSettingsView(headers: $headers)\n }\n }\n\n return V()\n}\n\n" }, { "path": "Core/Sources/HostApp/AccountSettings/EmbeddingModel.swift", "content": "import SwiftUI\nimport Keychain\n" }, { "path": "Core/Sources/HostApp/AccountSettings/EmbeddingModelManagement/EmbeddingModelEdit.swift", "content": "import AIModel\nimport ComposableArchitecture\nimport Dependencies\nimport Keychain\nimport OpenAIService\nimport Preferences\nimport SwiftUI\nimport Toast\n\n@Reducer\nstruct EmbeddingModelEdit {\n @ObservableState\n struct State: Equatable, Identifiable {\n var id: String\n var name: String\n var format: EmbeddingModel.Format\n var maxTokens: Int = 8191\n var dimensions: Int = 1536\n var modelName: String = \"\"\n var ollamaKeepAlive: String = \"\"\n var apiKeyName: String { apiKeySelection.apiKeyName }\n var baseURL: String { baseURLSelection.baseURL }\n var isFullURL: Bool { baseURLSelection.isFullURL }\n var availableModelNames: [String] = []\n var availableAPIKeys: [String] = []\n var isTesting = false\n var suggestedMaxTokens: Int?\n var apiKeySelection: APIKeySelection.State = .init()\n var baseURLSelection: BaseURLSelection.State = .init()\n var customHeaders: [ChatModel.Info.CustomHeaderInfo.HeaderField] = []\n }\n\n enum Action: Equatable, BindableAction {\n case binding(BindingAction)\n case appear\n case saveButtonClicked\n case cancelButtonClicked\n case refreshAvailableModelNames\n case testButtonClicked\n case testSucceeded(String)\n case testFailed(String)\n case fixDimensions(Int)\n case checkSuggestedMaxTokens\n case selectModelFormat(ModelFormat)\n case apiKeySelection(APIKeySelection.Action)\n case baseURLSelection(BaseURLSelection.Action)\n }\n \n enum ModelFormat: CaseIterable {\n case openAI\n case azureOpenAI\n case ollama\n case gitHubCopilot\n case openAICompatible\n case mistralOpenAICompatible\n case voyageAIOpenAICompatible\n \n init(_ format: EmbeddingModel.Format) {\n switch format {\n case .openAI:\n self = .openAI\n case .azureOpenAI:\n self = .azureOpenAI\n case .ollama:\n self = .ollama\n case .openAICompatible:\n self = .openAICompatible\n case .gitHubCopilot:\n self = .gitHubCopilot\n }\n }\n }\n\n var toast: (String, ToastType) -> Void {\n @Dependency(\\.namespacedToast) var toast\n return {\n toast($0, $1, \"EmbeddingModelEdit\")\n }\n }\n\n @Dependency(\\.apiKeyKeychain) var keychain\n\n var body: some ReducerOf {\n BindingReducer()\n\n Scope(state: \\.apiKeySelection, action: \\.apiKeySelection) {\n APIKeySelection()\n }\n\n Scope(state: \\.baseURLSelection, action: \\.baseURLSelection) {\n BaseURLSelection()\n }\n\n Reduce { state, action in\n switch action {\n case .appear:\n return .run { send in\n await send(.refreshAvailableModelNames)\n await send(.checkSuggestedMaxTokens)\n }\n\n case .saveButtonClicked:\n return .none\n\n case .cancelButtonClicked:\n return .none\n\n case .testButtonClicked:\n guard !state.isTesting else { return .none }\n state.isTesting = true\n let dimensions = state.dimensions\n let model = EmbeddingModel(\n id: state.id,\n name: state.name,\n format: state.format,\n info: .init(\n apiKeyName: state.apiKeyName,\n baseURL: state.baseURL,\n isFullURL: state.isFullURL,\n maxTokens: state.maxTokens,\n dimensions: dimensions,\n modelName: state.modelName\n )\n )\n return .run { send in\n do {\n let result = try await EmbeddingService(\n configuration: UserPreferenceEmbeddingConfiguration()\n .overriding {\n $0.model = model\n }\n ).embed(text: \"Hello\")\n if result.data.isEmpty {\n await send(.testFailed(\"No data returned\"))\n return\n }\n let actualDimensions = result.data.first?.embedding.count ?? 0\n if actualDimensions != dimensions {\n await send(\n .testFailed(\"Invalid dimension, should be \\(actualDimensions)\")\n )\n await send(.fixDimensions(actualDimensions))\n } else {\n await send(\n .testSucceeded(\"Succeeded! (Dimensions: \\(actualDimensions))\")\n )\n }\n } catch {\n await send(.testFailed(error.localizedDescription))\n }\n }\n\n case let .testSucceeded(message):\n state.isTesting = false\n toast(message, .info)\n return .none\n\n case let .testFailed(message):\n state.isTesting = false\n toast(message, .error)\n return .none\n\n case .refreshAvailableModelNames:\n if state.format == .openAI {\n state.availableModelNames = ChatGPTModel.allCases.map(\\.rawValue)\n }\n\n return .none\n\n case .checkSuggestedMaxTokens:\n guard state.format == .openAI,\n let knownModel = OpenAIEmbeddingModel(rawValue: state.modelName)\n else {\n state.suggestedMaxTokens = nil\n return .none\n }\n state.suggestedMaxTokens = knownModel.maxToken\n state.dimensions = knownModel.dimensions\n return .none\n\n case let .fixDimensions(value):\n state.dimensions = value\n return .none\n \n case let .selectModelFormat(format):\n switch format {\n case .openAI:\n state.format = .openAI\n case .azureOpenAI:\n state.format = .azureOpenAI\n case .ollama:\n state.format = .ollama\n case .openAICompatible:\n state.format = .openAICompatible\n case .gitHubCopilot:\n state.format = .gitHubCopilot\n case .mistralOpenAICompatible:\n state.format = .openAICompatible\n state.baseURLSelection.baseURL = \"https://api.mistral.ai\"\n state.baseURLSelection.isFullURL = false\n case .voyageAIOpenAICompatible:\n state.format = .openAICompatible\n state.baseURLSelection.baseURL = \"https://api.voyage.ai\"\n state.baseURLSelection.isFullURL = false\n }\n return .none\n\n case .apiKeySelection:\n return .none\n\n case .baseURLSelection:\n return .none\n\n case .binding(\\.format):\n return .run { send in\n await send(.refreshAvailableModelNames)\n await send(.checkSuggestedMaxTokens)\n }\n\n case .binding(\\.modelName):\n return .run { send in\n await send(.checkSuggestedMaxTokens)\n }\n\n case .binding:\n return .none\n }\n }\n }\n}\n\nextension EmbeddingModel {\n init(state: EmbeddingModelEdit.State) {\n self.init(\n id: state.id,\n name: state.name,\n format: state.format,\n info: .init(\n apiKeyName: state.apiKeyName,\n baseURL: state.baseURL.trimmingCharacters(in: .whitespacesAndNewlines),\n isFullURL: state.isFullURL,\n maxTokens: state.maxTokens,\n dimensions: state.dimensions,\n modelName: state.modelName.trimmingCharacters(in: .whitespacesAndNewlines),\n ollamaInfo: .init(keepAlive: state.ollamaKeepAlive),\n customHeaderInfo: .init(headers: state.customHeaders)\n )\n )\n }\n\n func toState() -> EmbeddingModelEdit.State {\n .init(\n id: id,\n name: name,\n format: format,\n maxTokens: info.maxTokens,\n dimensions: info.dimensions,\n modelName: info.modelName,\n ollamaKeepAlive: info.ollamaInfo.keepAlive,\n apiKeySelection: .init(\n apiKeyName: info.apiKeyName,\n apiKeyManagement: .init(availableAPIKeyNames: [info.apiKeyName])\n ),\n baseURLSelection: .init(\n baseURL: info.baseURL,\n isFullURL: info.isFullURL\n ),\n customHeaders: info.customHeaderInfo.headers\n )\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/AccountSettings/EmbeddingModelManagement/EmbeddingModelEditView.swift", "content": "import AIModel\nimport ComposableArchitecture\nimport Preferences\nimport SwiftUI\n\n@MainActor\nstruct EmbeddingModelEditView: View {\n @Perception.Bindable var store: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n ScrollView {\n VStack(spacing: 0) {\n Form {\n NameTextField(store: store)\n FormatPicker(store: store)\n\n switch store.format {\n case .openAI:\n OpenAIForm(store: store)\n case .azureOpenAI:\n AzureOpenAIForm(store: store)\n case .openAICompatible:\n OpenAICompatibleForm(store: store)\n case .ollama:\n OllamaForm(store: store)\n case .gitHubCopilot:\n GitHubCopilotForm(store: store)\n }\n }\n .padding()\n\n Divider()\n\n HStack {\n HStack(spacing: 8) {\n Button(\"Test\") {\n store.send(.testButtonClicked)\n }\n .disabled(store.isTesting)\n\n if store.isTesting {\n ProgressView()\n .controlSize(.small)\n }\n }\n\n Spacer()\n\n Button(\"Cancel\") {\n store.send(.cancelButtonClicked)\n }\n .keyboardShortcut(.cancelAction)\n\n Button(action: { store.send(.saveButtonClicked) }) {\n Text(\"Save\")\n }\n .keyboardShortcut(.defaultAction)\n }\n .padding()\n }\n }\n .textFieldStyle(.roundedBorder)\n .onAppear {\n store.send(.appear)\n }\n .fixedSize(horizontal: false, vertical: true)\n .handleToast(namespace: \"EmbeddingModelEdit\")\n }\n }\n\n struct NameTextField: View {\n @Perception.Bindable var store: StoreOf\n var body: some View {\n WithPerceptionTracking {\n TextField(\"Name\", text: $store.name)\n }\n }\n }\n\n struct FormatPicker: View {\n @Perception.Bindable var store: StoreOf\n var body: some View {\n WithPerceptionTracking {\n Picker(\n selection: Binding(\n get: { .init(store.format) },\n set: { store.send(.selectModelFormat($0)) }\n ),\n content: {\n ForEach(\n EmbeddingModelEdit.ModelFormat.allCases,\n id: \\.self\n ) { format in\n switch format {\n case .openAI:\n Text(\"OpenAI\")\n case .azureOpenAI:\n Text(\"Azure OpenAI\")\n case .ollama:\n Text(\"Ollama\")\n case .openAICompatible:\n Text(\"OpenAI Compatible\")\n case .mistralOpenAICompatible:\n Text(\"Mistral (OpenAI Compatible)\")\n case .voyageAIOpenAICompatible:\n Text(\"Voyage (OpenAI Compatible)\")\n case .gitHubCopilot:\n Text(\"GitHub Copilot\")\n }\n }\n },\n label: { Text(\"Format\") }\n )\n .pickerStyle(.menu)\n }\n }\n }\n\n struct BaseURLTextField: View {\n let store: StoreOf\n var title: String = \"Base URL\"\n let prompt: Text?\n @ViewBuilder var trailingContent: () -> V\n\n var body: some View {\n WithPerceptionTracking {\n BaseURLPicker(\n title: title,\n prompt: prompt,\n store: store.scope(\n state: \\.baseURLSelection,\n action: \\.baseURLSelection\n ),\n trailingContent: trailingContent\n )\n }\n }\n }\n\n struct MaxTokensTextField: View {\n @Perception.Bindable var store: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n HStack {\n let textFieldBinding = Binding(\n get: { String(store.maxTokens) },\n set: {\n if let selectionMaxToken = Int($0) {\n $store.maxTokens.wrappedValue = selectionMaxToken\n } else {\n $store.maxTokens.wrappedValue = 0\n }\n }\n )\n\n TextField(text: textFieldBinding) {\n Text(\"Max Input Tokens\")\n .multilineTextAlignment(.trailing)\n }\n .overlay(alignment: .trailing) {\n Stepper(\n value: $store.maxTokens,\n in: 0...Int.max,\n step: 100\n ) {\n EmptyView()\n }\n }\n .foregroundColor({\n guard let max = store.suggestedMaxTokens else {\n return .primary\n }\n if store.maxTokens > max {\n return .red\n }\n return .primary\n }() as Color)\n\n if let max = store.suggestedMaxTokens {\n Text(\"Max: \\(max)\")\n }\n }\n }\n }\n }\n\n struct DimensionsTextField: View {\n @Perception.Bindable var store: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n HStack {\n let textFieldBinding = Binding(\n get: { String(store.dimensions) },\n set: {\n if let selectionDimensions = Int($0) {\n $store.dimensions.wrappedValue = selectionDimensions\n } else {\n $store.dimensions.wrappedValue = 0\n }\n }\n )\n\n TextField(text: textFieldBinding) {\n Text(\"Dimensions\")\n .multilineTextAlignment(.trailing)\n }\n .overlay(alignment: .trailing) {\n Stepper(\n value: $store.dimensions,\n in: 0...Int.max,\n step: 100\n ) {\n EmptyView()\n }\n }\n .foregroundColor({\n if store.dimensions <= 0 {\n return .red\n }\n return .primary\n }() as Color)\n }\n\n Text(\"If you are not sure, run test to get the correct value.\")\n .font(.caption)\n .dynamicHeightTextInFormWorkaround()\n }\n }\n }\n\n struct ApiKeyNamePicker: View {\n let store: StoreOf\n var body: some View {\n WithPerceptionTracking {\n APIKeyPicker(store: store.scope(\n state: \\.apiKeySelection,\n action: \\.apiKeySelection\n ))\n }\n }\n }\n\n struct OpenAIForm: View {\n @Perception.Bindable var store: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n BaseURLTextField(store: store, prompt: Text(\"https://api.openai.com\")) {\n Text(\"/v1/embeddings\")\n }\n ApiKeyNamePicker(store: store)\n\n TextField(\"Model Name\", text: $store.modelName)\n .overlay(alignment: .trailing) {\n Picker(\n \"\",\n selection: $store.modelName,\n content: {\n if OpenAIEmbeddingModel(rawValue: store.modelName) == nil {\n Text(\"Custom Model\").tag(store.modelName)\n }\n ForEach(OpenAIEmbeddingModel.allCases, id: \\.self) { model in\n Text(model.rawValue).tag(model.rawValue)\n }\n }\n )\n .frame(width: 20)\n }\n\n MaxTokensTextField(store: store)\n DimensionsTextField(store: store)\n\n VStack(alignment: .leading, spacing: 8) {\n Text(Image(systemName: \"exclamationmark.triangle.fill\")) + Text(\n \" To get an API key, please visit [https://platform.openai.com/api-keys](https://platform.openai.com/api-keys)\"\n )\n\n Text(Image(systemName: \"exclamationmark.triangle.fill\")) + Text(\n \" If you don't have access to GPT-4, you may need to visit [https://platform.openai.com/account/billing/overview](https://platform.openai.com/account/billing/overview) to buy some credits. A ChatGPT Plus subscription is not enough to access GPT-4 through API.\"\n )\n }\n .padding(.vertical)\n }\n }\n }\n\n struct AzureOpenAIForm: View {\n @Perception.Bindable var store: StoreOf\n var body: some View {\n WithPerceptionTracking {\n BaseURLTextField(store: store, prompt: Text(\"https://xxxx.openai.azure.com\")) {\n EmptyView()\n }\n ApiKeyNamePicker(store: store)\n\n TextField(\"Deployment Name\", text: $store.modelName)\n\n MaxTokensTextField(store: store)\n DimensionsTextField(store: store)\n }\n }\n }\n\n struct OpenAICompatibleForm: View {\n @Perception.Bindable var store: StoreOf\n @State var isEditingCustomHeader = false\n\n var body: some View {\n WithPerceptionTracking {\n Picker(\n selection: $store.baseURLSelection.isFullURL,\n content: {\n Text(\"Base URL\").tag(false)\n Text(\"Full URL\").tag(true)\n },\n label: { Text(\"URL\") }\n )\n .pickerStyle(.segmented)\n\n BaseURLTextField(\n store: store,\n title: \"\",\n prompt: store.isFullURL\n ? Text(\"https://api.openai.com/v1/embeddings\")\n : Text(\"https://api.openai.com\")\n ) {\n if !store.isFullURL {\n Text(\"/v1/embeddings\")\n }\n }\n\n ApiKeyNamePicker(store: store)\n\n TextField(\"Model Name\", text: $store.modelName)\n\n MaxTokensTextField(store: store)\n DimensionsTextField(store: store)\n\n Button(\"Custom Headers\") {\n isEditingCustomHeader.toggle()\n }\n }.sheet(isPresented: $isEditingCustomHeader) {\n CustomHeaderSettingsView(headers: $store.customHeaders)\n }\n }\n }\n\n struct OllamaForm: View {\n @Perception.Bindable var store: StoreOf\n @State var isEditingCustomHeader = false\n\n var body: some View {\n WithPerceptionTracking {\n BaseURLTextField(store: store, prompt: Text(\"http://127.0.0.1:11434\")) {\n Text(\"/api/embeddings\")\n }\n\n ApiKeyNamePicker(store: store)\n\n TextField(\"Model Name\", text: $store.modelName)\n\n MaxTokensTextField(store: store)\n DimensionsTextField(store: store)\n\n WithPerceptionTracking {\n TextField(text: $store.ollamaKeepAlive, prompt: Text(\"Default Value\")) {\n Text(\"Keep Alive\")\n }\n }\n\n Button(\"Custom Headers\") {\n isEditingCustomHeader.toggle()\n }\n\n VStack(alignment: .leading, spacing: 8) {\n Text(Image(systemName: \"exclamationmark.triangle.fill\")) + Text(\n \" For more details, please visit [https://ollama.com](https://ollama.com).\"\n )\n }\n .padding(.vertical)\n\n }.sheet(isPresented: $isEditingCustomHeader) {\n CustomHeaderSettingsView(headers: $store.customHeaders)\n }\n }\n }\n\n struct GitHubCopilotForm: View {\n @Perception.Bindable var store: StoreOf\n @State var isEditingCustomHeader = false\n\n var body: some View {\n WithPerceptionTracking {\n TextField(\"Model Name\", text: $store.modelName)\n .overlay(alignment: .trailing) {\n Picker(\n \"\",\n selection: $store.modelName,\n content: {\n if OpenAIEmbeddingModel(rawValue: store.modelName) == nil {\n Text(\"Custom Model\").tag(store.modelName)\n }\n ForEach(OpenAIEmbeddingModel.allCases, id: \\.self) { model in\n Text(model.rawValue).tag(model.rawValue)\n }\n }\n )\n .frame(width: 20)\n }\n\n MaxTokensTextField(store: store)\n DimensionsTextField(store: store)\n\n Button(\"Custom Headers\") {\n isEditingCustomHeader.toggle()\n }\n\n VStack(alignment: .leading, spacing: 8) {\n Text(Image(systemName: \"exclamationmark.triangle.fill\")) + Text(\n \" Please login in the GitHub Copilot settings to use the model.\"\n )\n\n Text(Image(systemName: \"exclamationmark.triangle.fill\")) + Text(\n \" This will call the APIs directly, which may not be allowed by GitHub. But it's used in other popular apps like Zed.\"\n )\n }\n .dynamicHeightTextInFormWorkaround()\n .padding(.vertical)\n }.sheet(isPresented: $isEditingCustomHeader) {\n CustomHeaderSettingsView(headers: $store.customHeaders)\n }\n }\n }\n}\n\nclass EmbeddingModelManagementView_Editing_Previews: PreviewProvider {\n static var previews: some View {\n EmbeddingModelEditView(\n store: .init(\n initialState: EmbeddingModel(\n id: \"3\",\n name: \"Test Model 3\",\n format: .openAICompatible,\n info: .init(\n apiKeyName: \"key\",\n baseURL: \"apple.com\",\n maxTokens: 3000,\n modelName: \"gpt-3.5-turbo\"\n )\n ).toState(),\n reducer: { EmbeddingModelEdit() }\n )\n )\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/AccountSettings/EmbeddingModelManagement/EmbeddingModelManagement.swift", "content": "import AIModel\nimport ComposableArchitecture\nimport Keychain\nimport Preferences\nimport SwiftUI\n\nextension EmbeddingModel: ManageableAIModel {\n var formatName: String {\n switch format {\n case .openAI: return \"OpenAI\"\n case .azureOpenAI: return \"Azure OpenAI\"\n case .openAICompatible: return \"OpenAI Compatible\"\n case .ollama: return \"Ollama\"\n case .gitHubCopilot: return \"GitHub Copilot\"\n }\n }\n\n @ViewBuilder\n var infoDescriptors: some View {\n Text(info.modelName)\n\n if !info.baseURL.isEmpty {\n Image(systemName: \"line.diagonal\")\n Text(info.baseURL)\n }\n\n Image(systemName: \"line.diagonal\")\n\n Text(\"\\(info.maxTokens) tokens\")\n }\n}\n\n@Reducer\nstruct EmbeddingModelManagement: AIModelManagement {\n typealias Model = EmbeddingModel\n\n @ObservableState\n struct State: Equatable, AIModelManagementState {\n typealias Model = EmbeddingModel\n var models: IdentifiedArrayOf = []\n @Presents var editingModel: EmbeddingModelEdit.State?\n var selectedModelId: Model.ID? { editingModel?.id }\n }\n\n enum Action: Equatable, AIModelManagementAction {\n typealias Model = EmbeddingModel\n case appear\n case createModel\n case removeModel(id: Model.ID)\n case selectModel(id: Model.ID)\n case duplicateModel(id: Model.ID)\n case moveModel(from: IndexSet, to: Int)\n case embeddingModelItem(PresentationAction)\n }\n\n @Dependency(\\.toast) var toast\n @Dependency(\\.userDefaults) var userDefaults\n\n var body: some ReducerOf {\n Reduce { state, action in\n switch action {\n case .appear:\n if isPreview { return .none }\n state.models = .init(\n userDefaults.value(for: \\.embeddingModels),\n id: \\.id,\n uniquingIDsWith: { a, _ in a }\n )\n\n return .none\n\n case .createModel:\n state.editingModel = .init(\n id: UUID().uuidString,\n name: \"New Model\",\n format: .openAI\n )\n return .none\n\n case let .removeModel(id):\n state.models.remove(id: id)\n persist(state)\n return .none\n\n case let .selectModel(id):\n guard let model = state.models[id: id] else { return .none }\n state.editingModel = model.toState()\n return .none\n\n case let .duplicateModel(id):\n guard var model = state.models[id: id] else { return .none }\n model.id = UUID().uuidString\n model.name += \" (Copy)\"\n\n if let index = state.models.index(id: id) {\n state.models.insert(model, at: index + 1)\n } else {\n state.models.append(model)\n }\n persist(state)\n return .none\n\n case let .moveModel(from, to):\n state.models.move(fromOffsets: from, toOffset: to)\n persist(state)\n return .none\n\n case .embeddingModelItem(.presented(.saveButtonClicked)):\n guard let editingModel = state.editingModel, validateModel(editingModel)\n else { return .none }\n\n if let index = state.models\n .firstIndex(where: { $0.id == editingModel.id })\n {\n state.models[index] = .init(state: editingModel)\n } else {\n state.models.append(.init(state: editingModel))\n }\n persist(state)\n return .run { send in\n await send(.embeddingModelItem(.dismiss))\n }\n\n case .embeddingModelItem(.presented(.cancelButtonClicked)):\n return .run { send in\n await send(.embeddingModelItem(.dismiss))\n }\n\n case .embeddingModelItem:\n return .none\n }\n }.ifLet(\\.$editingModel, action: \\.embeddingModelItem) {\n EmbeddingModelEdit()\n }\n }\n\n func persist(_ state: State) {\n let models = state.models\n userDefaults.set(Array(models), for: \\.embeddingModels)\n }\n\n func validateModel(_ chatModel: EmbeddingModelEdit.State) -> Bool {\n guard !chatModel.name.isEmpty else {\n toast(\"Model name cannot be empty\", .error)\n return false\n }\n guard !chatModel.id.isEmpty else {\n toast(\"Model ID cannot be empty\", .error)\n return false\n }\n\n guard !chatModel.modelName.isEmpty else {\n toast(\"Model name cannot be empty\", .error)\n return false\n }\n return true\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/AccountSettings/EmbeddingModelManagement/EmbeddingModelManagementView.swift", "content": "import AIModel\nimport ComposableArchitecture\nimport SwiftUI\n\nstruct EmbeddingModelManagementView: View {\n @Perception.Bindable var store: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n AIModelManagementView(store: store)\n .sheet(item: $store.scope(\n state: \\.editingModel,\n action: \\.embeddingModelItem\n )) { store in\n EmbeddingModelEditView(store: store)\n .frame(width: 800)\n }\n }\n }\n}\n\n// MARK: - Previews\n\nclass EmbeddingModelManagementView_Previews: PreviewProvider {\n static var previews: some View {\n EmbeddingModelManagementView(\n store: .init(\n initialState: .init(\n models: IdentifiedArray(uniqueElements: [\n EmbeddingModel(\n id: \"1\",\n name: \"Test Model\",\n format: .openAI,\n info: .init(\n apiKeyName: \"key\",\n baseURL: \"google.com\",\n maxTokens: 3000,\n modelName: \"gpt-3.5-turbo\"\n )\n ),\n EmbeddingModel(\n id: \"2\",\n name: \"Test Model 2\",\n format: .azureOpenAI,\n info: .init(\n apiKeyName: \"key\",\n baseURL: \"apple.com\",\n maxTokens: 3000,\n modelName: \"gpt-3.5-turbo\"\n )\n ),\n EmbeddingModel(\n id: \"3\",\n name: \"Test Model 3\",\n format: .openAICompatible,\n info: .init(\n apiKeyName: \"key\",\n baseURL: \"apple.com\",\n maxTokens: 3000,\n modelName: \"gpt-3.5-turbo\"\n )\n ),\n ]),\n editingModel: EmbeddingModel(\n id: \"3\",\n name: \"Test Model 3\",\n format: .openAICompatible,\n info: .init(\n apiKeyName: \"key\",\n baseURL: \"apple.com\",\n maxTokens: 3000,\n modelName: \"gpt-3.5-turbo\"\n )\n ).toState()\n ),\n reducer: { EmbeddingModelManagement() }\n )\n )\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/AccountSettings/GitHubCopilotModelPicker.swift", "content": "import Dependencies\nimport Foundation\nimport GitHubCopilotService\nimport Perception\nimport SwiftUI\nimport Toast\n\npublic struct GitHubCopilotModelPicker: View {\n @Perceptible\n final class ViewModel {\n var availableModels: [GitHubCopilotLLMModel] = []\n @PerceptionIgnored @Dependency(\\.toast) var toast\n\n init() {}\n\n func appear() {\n reloadAvailableModels()\n }\n\n func disappear() {}\n\n func reloadAvailableModels() {\n Task { @MainActor in\n do {\n availableModels = try await GitHubCopilotExtension.fetchLLMModels()\n } catch {\n toast(\"Failed to fetch GitHub Copilot models: \\(error)\", .error)\n }\n }\n }\n }\n\n let title: String\n let hasDefaultModel: Bool\n @Binding var gitHubCopilotModelId: String\n @State var viewModel: ViewModel\n\n init(\n title: String,\n hasDefaultModel: Bool = true,\n gitHubCopilotModelId: Binding\n ) {\n self.title = title\n _gitHubCopilotModelId = gitHubCopilotModelId\n self.hasDefaultModel = hasDefaultModel\n viewModel = .init()\n }\n\n public var body: some View {\n WithPerceptionTracking {\n TextField(title, text: $gitHubCopilotModelId)\n .overlay(alignment: .trailing) {\n Picker(\n \"\",\n selection: $gitHubCopilotModelId,\n content: {\n if hasDefaultModel {\n Text(\"Default\").tag(\"\")\n }\n\n if !gitHubCopilotModelId.isEmpty,\n !viewModel.availableModels.contains(where: {\n $0.modelId == gitHubCopilotModelId\n })\n {\n Text(gitHubCopilotModelId).tag(gitHubCopilotModelId)\n }\n if viewModel.availableModels.isEmpty {\n Text({\n viewModel.reloadAvailableModels()\n return \"Loading...\"\n }()).tag(\"Loading...\")\n }\n ForEach(viewModel.availableModels) { model in\n Text(model.modelId)\n .tag(model.modelId)\n }\n }\n )\n .frame(width: 20)\n }\n .onAppear {\n viewModel.appear()\n }\n .onDisappear {\n viewModel.disappear()\n }\n }\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/AccountSettings/GitHubCopilotView.swift", "content": "import AppKit\nimport Client\nimport GitHubCopilotService\nimport Preferences\nimport SharedUIComponents\nimport SuggestionBasic\nimport SwiftUI\n\nstruct GitHubCopilotView: View {\n static var copilotAuthService: GitHubCopilotAuthServiceType?\n\n class Settings: ObservableObject {\n @AppStorage(\\.nodePath) var nodePath: String\n @AppStorage(\\.runNodeWith) var runNodeWith\n @AppStorage(\"username\") var username: String = \"\"\n @AppStorage(\\.gitHubCopilotVerboseLog) var gitHubCopilotVerboseLog\n @AppStorage(\\.gitHubCopilotProxyHost) var gitHubCopilotProxyHost\n @AppStorage(\\.gitHubCopilotProxyPort) var gitHubCopilotProxyPort\n @AppStorage(\\.gitHubCopilotProxyUsername) var gitHubCopilotProxyUsername\n @AppStorage(\\.gitHubCopilotProxyPassword) var gitHubCopilotProxyPassword\n @AppStorage(\\.gitHubCopilotUseStrictSSL) var gitHubCopilotUseStrictSSL\n @AppStorage(\\.gitHubCopilotEnterpriseURI) var gitHubCopilotEnterpriseURI\n @AppStorage(\\.gitHubCopilotPretendIDEToBeVSCode) var pretendIDEToBeVSCode\n @AppStorage(\\.disableGitHubCopilotSettingsAutoRefreshOnAppear)\n var disableGitHubCopilotSettingsAutoRefreshOnAppear\n @AppStorage(\\.gitHubCopilotLoadKeyChainCertificates)\n var gitHubCopilotLoadKeyChainCertificates\n @AppStorage(\\.gitHubCopilotModelId) var gitHubCopilotModelId\n init() {}\n }\n\n class ViewModel: ObservableObject {\n let installationManager = GitHubCopilotInstallationManager()\n\n @Published var installationStatus: GitHubCopilotInstallationManager.InstallationStatus?\n @Published var installationStep: GitHubCopilotInstallationManager.InstallationStep?\n\n init() {}\n\n init(\n installationStatus: GitHubCopilotInstallationManager.InstallationStatus,\n installationStep: GitHubCopilotInstallationManager.InstallationStep?\n ) {\n assert(isPreview)\n self.installationStatus = installationStatus\n self.installationStep = installationStep\n }\n\n func refreshInstallationStatus() {\n Task { @MainActor in\n installationStatus = installationManager.checkInstallation()\n }\n }\n\n func install() async throws {\n defer { refreshInstallationStatus() }\n do {\n for try await step in installationManager.installLatestVersion() {\n Task { @MainActor in\n self.installationStep = step\n }\n }\n Task {\n try await Task.sleep(nanoseconds: 1_000_000_000)\n Task { @MainActor in\n self.installationStep = nil\n }\n }\n } catch {\n Task { @MainActor in\n installationStep = nil\n }\n throw error\n }\n }\n\n func uninstall() {\n Task {\n defer { refreshInstallationStatus() }\n try await installationManager.uninstall()\n Task { @MainActor in\n GitHubCopilotView.copilotAuthService = nil\n }\n }\n }\n }\n\n @Environment(\\.openURL) var openURL\n @Environment(\\.toast) var toast\n @StateObject var settings = Settings()\n @StateObject var viewModel = ViewModel()\n\n @State var status: GitHubCopilotAccountStatus?\n @State var userCode: String?\n @State var version: String?\n @State var isRunningAction: Bool = false\n @State var isUserCodeCopiedAlertPresented = false\n\n func getGitHubCopilotAuthService() throws -> GitHubCopilotAuthServiceType {\n if let service = Self.copilotAuthService { return service }\n let service = try GitHubCopilotAuthService()\n Self.copilotAuthService = service\n return service\n }\n\n var installButton: some View {\n Button(action: {\n Task {\n do {\n try await viewModel.install()\n } catch {\n toast(error.localizedDescription, .error)\n }\n }\n }) {\n Text(\"Install\")\n }\n .disabled(viewModel.installationStep != nil)\n }\n\n var updateButton: some View {\n Button(action: {\n Task {\n do {\n try await viewModel.install()\n } catch {\n toast(error.localizedDescription, .error)\n }\n }\n }) {\n Text(\"Update\")\n }\n .disabled(viewModel.installationStep != nil)\n }\n\n var uninstallButton: some View {\n Button(action: {\n viewModel.uninstall()\n }) {\n Text(\"Uninstall\")\n }\n .disabled(viewModel.installationStep != nil)\n }\n\n var body: some View {\n HStack {\n VStack(alignment: .leading, spacing: 8) {\n SubSection(\n title: Text(\"Node Settings\"),\n description: \"\"\"\n You may have to restart the extension app to apply the changes. To do so, simply close the helper app by clicking on the menu bar icon that looks like a tentacle, it will automatically restart as needed.\n \"\"\"\n ) {\n Form {\n TextField(\n text: $settings.nodePath,\n prompt: Text(\n \"node\"\n )\n ) {\n Text(\"Path to Node (v22.0+)\")\n }\n\n Text(\n \"Provide the path to the executable if it can't be found by the app, shim executable is not supported\"\n )\n .lineLimit(10)\n .foregroundColor(.secondary)\n .font(.callout)\n .dynamicHeightTextInFormWorkaround()\n\n Picker(selection: $settings.runNodeWith) {\n ForEach(NodeRunner.allCases, id: \\.rawValue) { runner in\n switch runner {\n case .env:\n Text(\"/usr/bin/env\").tag(runner)\n case .bash:\n Text(\"/bin/bash -i -l\").tag(runner)\n case .shell:\n Text(\"$SHELL -i -l\").tag(runner)\n }\n }\n } label: {\n Text(\"Run Node with\")\n }\n\n Group {\n switch settings.runNodeWith {\n case .env:\n Text(\n \"PATH: `/opt/homebrew/bin:/opt/homebrew/sbin:/usr/local/bin:/usr/bin:/bin:/usr/sbin:/sbin`\"\n )\n case .bash:\n Text(\"PATH inherited from bash configurations.\")\n case .shell:\n Text(\"PATH inherited from $SHELL configurations.\")\n }\n }\n .lineLimit(10)\n .foregroundColor(.secondary)\n .font(.callout)\n .dynamicHeightTextInFormWorkaround()\n\n Toggle(isOn: $settings.gitHubCopilotLoadKeyChainCertificates) {\n Text(\"Load certificates in keychain\")\n }\n }\n }\n\n SubSection(\n title: Text(\"GitHub Copilot Language Server\")\n ) {\n HStack {\n switch viewModel.installationStatus {\n case .none:\n Text(\"Copilot.Vim Version: Loading..\")\n case .notInstalled:\n Text(\"Copilot.Vim Version: Not Installed\")\n installButton\n case let .installed(version):\n Text(\"Copilot.Vim Version: \\(version)\")\n uninstallButton\n case let .outdated(version, latest, _):\n Text(\"Copilot.Vim Version: \\(version) (Update Available: \\(latest))\")\n updateButton\n uninstallButton\n case let .unsupported(version, latest):\n Text(\"Copilot.Vim Version: \\(version) (Supported Version: \\(latest))\")\n updateButton\n uninstallButton\n }\n }\n\n Text(\"Language Server Version: \\(version ?? \"Loading..\")\")\n\n Text(\"Status: \\(status?.description ?? \"Loading..\")\")\n\n HStack(alignment: .center) {\n Button(\"Refresh\") {\n viewModel.refreshInstallationStatus()\n checkStatus()\n }\n if status == .notSignedIn {\n Button(\"Sign In\") { signIn() }\n .alert(isPresented: $isUserCodeCopiedAlertPresented) {\n Alert(\n title: Text(userCode ?? \"\"),\n message: Text(\n \"The user code is pasted into your clipboard, please paste it in the opened website to login.\\nAfter that, click \\\"Confirm Sign-in\\\" to finish.\"\n ),\n dismissButton: .default(Text(\"OK\"))\n )\n }\n Button(\"Confirm Sign-in\") { confirmSignIn() }\n }\n if status == .ok || status == .alreadySignedIn ||\n status == .notAuthorized\n {\n Button(\"Sign Out\") { signOut() }\n }\n if isRunningAction {\n ActivityIndicatorView()\n }\n } \n .opacity(isRunningAction ? 0.8 : 1)\n .disabled(isRunningAction)\n\n Button(\"Refresh configurations\") {\n refreshConfiguration()\n }\n\n Form {\n GitHubCopilotModelPicker(\n title: \"Chat Model Name\",\n gitHubCopilotModelId: $settings.gitHubCopilotModelId\n )\n }\n }\n\n SettingsDivider(\"Advanced\")\n\n Form {\n Toggle(\"Verbose log\", isOn: $settings.gitHubCopilotVerboseLog)\n Toggle(\"Pretend IDE to be VSCode\", isOn: $settings.pretendIDEToBeVSCode)\n }\n\n SettingsDivider(\"Enterprise\")\n\n Form {\n TextField(\n text: $settings.gitHubCopilotEnterpriseURI,\n prompt: Text(\"Leave it blank if non is available.\")\n ) {\n Text(\"Auth provider URL\")\n }\n }\n\n SettingsDivider(\"Proxy\")\n\n Form {\n TextField(\n text: $settings.gitHubCopilotProxyHost,\n prompt: Text(\"xxx.xxx.xxx.xxx, leave it blank to disable proxy.\")\n ) {\n Text(\"Proxy host\")\n }\n TextField(text: $settings.gitHubCopilotProxyPort, prompt: Text(\"80\")) {\n Text(\"Proxy port\")\n }\n TextField(text: $settings.gitHubCopilotProxyUsername) {\n Text(\"Proxy username\")\n }\n SecureField(text: $settings.gitHubCopilotProxyPassword) {\n Text(\"Proxy password\")\n }\n Toggle(\"Proxy strict SSL\", isOn: $settings.gitHubCopilotUseStrictSSL)\n }\n }\n Spacer()\n }.onAppear {\n if isPreview { return }\n if settings.disableGitHubCopilotSettingsAutoRefreshOnAppear { return }\n viewModel.refreshInstallationStatus()\n checkStatus()\n }.onChange(of: settings.runNodeWith) { _ in\n Self.copilotAuthService = nil\n }.onChange(of: settings.nodePath) { _ in\n Self.copilotAuthService = nil\n }.onChange(of: viewModel.installationStep) { newValue in\n if let step = newValue {\n switch step {\n case .downloading:\n toast(\"Downloading..\", .info)\n case .uninstalling:\n toast(\"Uninstalling old version..\", .info)\n case .decompressing:\n toast(\"Decompressing..\", .info)\n case .done:\n toast(\"Done!\", .info)\n checkStatus()\n }\n }\n }\n .textFieldStyle(.roundedBorder)\n }\n\n func checkStatus() {\n Task {\n isRunningAction = true\n defer { isRunningAction = false }\n do {\n let service = try getGitHubCopilotAuthService()\n status = try await service.checkStatus()\n version = try await service.version()\n isRunningAction = false\n\n if status != .ok, status != .notSignedIn {\n toast(\n \"GitHub Copilot status is not \\\"ok\\\". Please check if you have a valid GitHub Copilot subscription.\",\n .error\n )\n }\n } catch {\n toast(error.localizedDescription, .error)\n }\n }\n }\n\n func signIn() {\n Task {\n isRunningAction = true\n defer { isRunningAction = false }\n do {\n let service = try getGitHubCopilotAuthService()\n let (uri, userCode) = try await service.signInInitiate()\n self.userCode = userCode\n guard let url = URL(string: uri) else {\n toast(\"Verification URI is incorrect.\", .error)\n return\n }\n let pasteboard = NSPasteboard.general\n pasteboard.declareTypes([NSPasteboard.PasteboardType.string], owner: nil)\n pasteboard.setString(userCode, forType: NSPasteboard.PasteboardType.string)\n toast(\"Usercode \\(userCode) already copied!\", .info)\n openURL(url)\n isUserCodeCopiedAlertPresented = true\n } catch {\n toast(error.localizedDescription, .error)\n }\n }\n }\n\n func confirmSignIn() {\n Task {\n isRunningAction = true\n defer { isRunningAction = false }\n do {\n let service = try getGitHubCopilotAuthService()\n guard let userCode else {\n toast(\"Usercode is empty.\", .error)\n return\n }\n let (username, status) = try await service.signInConfirm(userCode: userCode)\n self.settings.username = username\n self.status = status\n } catch {\n toast(error.localizedDescription, .error)\n }\n }\n }\n\n func signOut() {\n Task {\n isRunningAction = true\n defer { isRunningAction = false }\n do {\n let service = try getGitHubCopilotAuthService()\n status = try await service.signOut()\n } catch {\n toast(error.localizedDescription, .error)\n }\n }\n }\n\n func refreshConfiguration() {\n NotificationCenter.default.post(\n name: .gitHubCopilotShouldRefreshEditorInformation,\n object: nil\n )\n\n Task {\n let service = try getService()\n do {\n try await service.postNotification(\n name: Notification.Name\n .gitHubCopilotShouldRefreshEditorInformation.rawValue\n )\n } catch {\n toast(error.localizedDescription, .error)\n }\n }\n }\n}\n\nstruct ActivityIndicatorView: NSViewRepresentable {\n func makeNSView(context _: Context) -> NSProgressIndicator {\n let progressIndicator = NSProgressIndicator()\n progressIndicator.style = .spinning\n progressIndicator.appearance = NSAppearance(named: .vibrantLight)\n progressIndicator.controlSize = .small\n progressIndicator.startAnimation(nil)\n return progressIndicator\n }\n\n func updateNSView(_: NSProgressIndicator, context _: Context) {\n // No-op\n }\n}\n\nstruct CopilotView_Previews: PreviewProvider {\n static var previews: some View {\n VStack(alignment: .leading, spacing: 8) {\n GitHubCopilotView(status: .notSignedIn, version: \"1.0.0\")\n GitHubCopilotView(status: .alreadySignedIn, isRunningAction: true)\n }\n .padding(.all, 8)\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/AccountSettings/OtherSuggestionServicesView.swift", "content": "import Foundation\nimport SwiftUI\n\nstruct OtherSuggestionServicesView: View {\n @Environment(\\.openURL) var openURL\n var body: some View {\n VStack(alignment: .leading) {\n Text(\n \"You can use other locally run services (Tabby, Ollma, etc.) to generate suggestions with the Custom Suggestion Service extension.\"\n )\n .lineLimit(nil)\n .multilineTextAlignment(.leading)\n\n Button(action: {\n if let url = URL(\n string: \"https://github.com/intitni/CustomSuggestionServiceForCopilotForXcode\"\n ) {\n openURL(url)\n }\n }) {\n Text(\"Get It Now\")\n }\n }\n }\n}\n\n#Preview {\n OtherSuggestionServicesView()\n .frame(width: 200)\n}\n\n" }, { "path": "Core/Sources/HostApp/AccountSettings/SharedModelManagement/AIModelManagementVIew.swift", "content": "import AIModel\nimport ComposableArchitecture\nimport PlusFeatureFlag\nimport SharedUIComponents\nimport SwiftUI\n\nprotocol AIModelManagementAction {\n associatedtype Model: ManageableAIModel\n static var appear: Self { get }\n static var createModel: Self { get }\n static func removeModel(id: Model.ID) -> Self\n static func selectModel(id: Model.ID) -> Self\n static func duplicateModel(id: Model.ID) -> Self\n static func moveModel(from: IndexSet, to: Int) -> Self\n}\n\nprotocol AIModelManagementState: Equatable {\n associatedtype Model: ManageableAIModel\n var models: IdentifiedArrayOf { get }\n var selectedModelId: Model.ID? { get }\n}\n\nprotocol AIModelManagement: Reducer where\n Action: AIModelManagementAction,\n State: AIModelManagementState & ObservableState,\n Action.Model == Self.Model,\n State.Model == Self.Model\n{\n associatedtype Model: ManageableAIModel\n}\n\nprotocol ManageableAIModel: Identifiable {\n associatedtype V: View\n var name: String { get }\n var formatName: String { get }\n var infoDescriptors: V { get }\n}\n\nstruct AIModelManagementView: View\n where Management.Model == Model\n{\n @Perception.Bindable var store: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n VStack(spacing: 0) {\n HStack {\n Spacer()\n if isFeatureAvailable(\\.unlimitedChatAndEmbeddingModels) {\n Button(\"Add Model\") {\n store.send(.createModel)\n }\n } else {\n Text(\"\\(store.models.count) / 2\")\n .foregroundColor(.secondary)\n\n let disabled = store.models.count >= 2\n\n Button(disabled ? \"Add More Model (Plus)\" : \"Add Model\") {\n store.send(.createModel)\n }.disabled(disabled)\n }\n }.padding(4)\n\n Divider()\n\n ModelList(store: store)\n }\n .onAppear {\n store.send(.appear)\n }\n }\n }\n\n struct ModelList: View {\n @Perception.Bindable var store: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n List {\n ForEach(store.models) { model in\n WithPerceptionTracking {\n let isSelected = store.selectedModelId == model.id\n HStack(spacing: 4) {\n Image(systemName: \"line.3.horizontal\")\n\n Button(action: {\n store.send(.selectModel(id: model.id))\n }) {\n Cell(model: model, isSelected: isSelected)\n .contentShape(Rectangle())\n }\n .buttonStyle(.plain)\n .contextMenu {\n Button(\"Duplicate\") {\n store.send(.duplicateModel(id: model.id))\n }\n Button(\"Remove\") {\n store.send(.removeModel(id: model.id))\n }\n }\n }\n }\n }\n .onMove(perform: { indices, newOffset in\n store.send(.moveModel(from: indices, to: newOffset))\n })\n .modify { view in\n if #available(macOS 13.0, *) {\n view.listRowSeparator(.hidden).listSectionSeparator(.hidden)\n } else {\n view\n }\n }\n }\n .removeBackground()\n .listStyle(.plain)\n .listRowInsets(EdgeInsets())\n .overlay {\n if store.models.isEmpty {\n Text(\"No model found, please add a new one.\")\n .foregroundColor(.secondary)\n }\n }\n }\n }\n }\n\n struct Cell: View {\n let model: Model\n let isSelected: Bool\n @State var isHovered: Bool = false\n\n var body: some View {\n HStack {\n VStack(alignment: .leading, spacing: 2) {\n HStack {\n Text(model.formatName)\n .foregroundColor(isSelected ? .white : .primary)\n .font(.subheadline.bold())\n .padding(.vertical, 2)\n .padding(.horizontal, 4)\n .background {\n RoundedRectangle(cornerRadius: 4)\n .fill(\n isSelected\n ? .white.opacity(0.2)\n : Color.primary.opacity(0.1)\n )\n }\n\n Text(model.name)\n .font(.headline)\n }\n\n HStack(spacing: 4) {\n model.infoDescriptors\n }\n .font(.subheadline)\n .opacity(0.7)\n .padding(.leading, 2)\n }\n Spacer()\n }\n .onHover(perform: {\n isHovered = $0\n })\n .padding(.vertical, 8)\n .padding(.horizontal, 8)\n .background {\n RoundedRectangle(cornerRadius: 8, style: .continuous)\n .fill({\n switch (isSelected, isHovered) {\n case (true, _):\n return Color.accentColor\n case (_, true):\n return Color.primary.opacity(0.1)\n case (_, false):\n return Color.clear\n }\n }() as Color)\n }\n .foregroundColor(isSelected ? .white : .primary)\n .animation(.easeInOut(duration: 0.1), value: isSelected)\n .animation(.easeInOut(duration: 0.1), value: isHovered)\n }\n }\n}\n\n// MARK: - Previews\n\nclass AIModelManagement_Previews: PreviewProvider {\n static var previews: some View {\n AIModelManagementView(\n store: .init(\n initialState: .init(\n models: IdentifiedArray(uniqueElements: [\n ChatModel(\n id: \"1\",\n name: \"Test Model\",\n format: .openAI,\n info: .init(\n apiKeyName: \"key\",\n baseURL: \"google.com\",\n maxTokens: 3000,\n supportsFunctionCalling: true,\n modelName: \"gpt-3.5-turbo\"\n )\n ),\n ChatModel(\n id: \"2\",\n name: \"Test Model 2\",\n format: .azureOpenAI,\n info: .init(\n apiKeyName: \"key\",\n baseURL: \"apple.com\",\n maxTokens: 3000,\n supportsFunctionCalling: false,\n modelName: \"gpt-3.5-turbo\"\n )\n ),\n ChatModel(\n id: \"3\",\n name: \"Test Model 3\",\n format: .openAICompatible,\n info: .init(\n apiKeyName: \"key\",\n baseURL: \"apple.com\",\n maxTokens: 3000,\n supportsFunctionCalling: false,\n modelName: \"gpt-3.5-turbo\"\n )\n ),\n ]),\n editingModel: ChatModel(\n id: \"3\",\n name: \"Test Model 3\",\n format: .openAICompatible,\n info: .init(\n apiKeyName: \"key\",\n baseURL: \"apple.com\",\n maxTokens: 3000,\n supportsFunctionCalling: false,\n modelName: \"gpt-3.5-turbo\"\n )\n ).toState()\n ),\n reducer: { ChatModelManagement() }\n )\n )\n }\n}\n\nclass AIModelManagement_Empty_Previews: PreviewProvider {\n static var previews: some View {\n AIModelManagementView(\n store: .init(\n initialState: .init(models: [] as IdentifiedArrayOf),\n reducer: { ChatModelManagement() }\n )\n )\n }\n}\n\nclass AIModelManagement_Cell_Previews: PreviewProvider {\n static var previews: some View {\n AIModelManagementView.Cell(model: ChatModel(\n id: \"1\",\n name: \"Test Model\",\n format: .openAI,\n info: .init(\n apiKeyName: \"key\",\n baseURL: \"google.com\",\n maxTokens: 3000,\n supportsFunctionCalling: true,\n modelName: \"gpt-3.5-turbo\"\n )\n ), isSelected: false)\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/AccountSettings/SharedModelManagement/BaseURLPicker.swift", "content": "import ComposableArchitecture\nimport SwiftUI\n\nstruct BaseURLPicker: View {\n let title: String\n let prompt: Text?\n @Perception.Bindable var store: StoreOf\n @ViewBuilder let trailingContent: () -> TrailingContent\n\n var body: some View {\n WithPerceptionTracking {\n HStack {\n TextField(title, text: $store.baseURL, prompt: prompt)\n .overlay(alignment: .trailing) {\n Picker(\n \"\",\n selection: $store.baseURL,\n content: {\n if !store.availableBaseURLs\n .contains(store.baseURL),\n !store.baseURL.isEmpty\n {\n Text(\"Custom Value\").tag(store.baseURL)\n }\n\n Text(\"Empty (Default Value)\").tag(\"\")\n\n ForEach(store.availableBaseURLs, id: \\.self) { baseURL in\n Text(baseURL).tag(baseURL)\n }\n }\n )\n .frame(width: 20)\n }\n\n trailingContent()\n .foregroundStyle(.secondary)\n }\n .onAppear {\n store.send(.appear)\n }\n }\n }\n}\n\nextension BaseURLPicker where TrailingContent == EmptyView {\n init(\n title: String,\n prompt: Text? = nil,\n store: StoreOf\n ) {\n self.init(\n title: title,\n prompt: prompt,\n store: store,\n trailingContent: { EmptyView() }\n )\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/AccountSettings/SharedModelManagement/BaseURLSelection.swift", "content": "import ComposableArchitecture\nimport Foundation\nimport Preferences\nimport SwiftUI\n\n@Reducer\nstruct BaseURLSelection {\n @ObservableState\n struct State: Equatable {\n var baseURL: String = \"\"\n var isFullURL: Bool = false\n var availableBaseURLs: [String] = []\n }\n\n enum Action: Equatable, BindableAction {\n case appear\n case refreshAvailableBaseURLNames\n case binding(BindingAction)\n }\n\n @Dependency(\\.toast) var toast\n @Dependency(\\.userDefaults) var userDefaults\n\n var body: some ReducerOf {\n BindingReducer()\n\n Reduce { state, action in\n switch action {\n case .appear:\n return .run { send in\n await send(.refreshAvailableBaseURLNames)\n }\n\n case .refreshAvailableBaseURLNames:\n let chatModels = userDefaults.value(for: \\.chatModels)\n let embeddingModels = userDefaults.value(for: \\.embeddingModels)\n var allBaseURLs = Set(\n chatModels.map(\\.info.baseURL)\n .map { $0.trimmingCharacters(in: .whitespacesAndNewlines) }\n + embeddingModels.map(\\.info.baseURL)\n .map { $0.trimmingCharacters(in: .whitespacesAndNewlines) }\n )\n allBaseURLs.remove(\"\")\n state.availableBaseURLs = Array(allBaseURLs).sorted()\n return .none\n\n case .binding:\n return .none\n }\n }\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/AccountSettings/WebSearchView.swift", "content": "import AppKit\nimport Client\nimport ComposableArchitecture\nimport OpenAIService\nimport Preferences\nimport SuggestionBasic\nimport SwiftUI\nimport WebSearchService\nimport SharedUIComponents\n\n@Reducer\nstruct WebSearchSettings {\n struct TestResult: Identifiable, Equatable {\n let id = UUID()\n var duration: TimeInterval\n var result: Result?\n\n static func == (lhs: Self, rhs: Self) -> Bool {\n lhs.id == rhs.id\n }\n }\n\n @ObservableState\n struct State: Equatable {\n var apiKeySelection: APIKeySelection.State = .init()\n var testResult: TestResult?\n }\n\n enum Action: BindableAction {\n case binding(BindingAction)\n case appear\n case test\n case bringUpTestResult\n case updateTestResult(TimeInterval, Result)\n case apiKeySelection(APIKeySelection.Action)\n }\n\n var body: some ReducerOf {\n BindingReducer()\n\n Scope(state: \\.apiKeySelection, action: \\.apiKeySelection) {\n APIKeySelection()\n }\n\n Reduce { state, action in\n switch action {\n case .binding:\n return .none\n case .appear:\n state.testResult = nil\n state.apiKeySelection.apiKeyName = UserDefaults.shared.value(for: \\.serpAPIKeyName)\n return .none\n case .test:\n return .run { send in\n let searchService = WebSearchService(provider: .userPreferred)\n await send(.bringUpTestResult)\n let start = Date()\n do {\n let result = try await searchService.search(query: \"Swift\")\n let duration = Date().timeIntervalSince(start)\n await send(.updateTestResult(duration, .success(result)))\n } catch {\n let duration = Date().timeIntervalSince(start)\n await send(.updateTestResult(duration, .failure(error)))\n }\n }\n case .bringUpTestResult:\n state.testResult = .init(duration: 0)\n return .none\n case let .updateTestResult(duration, result):\n state.testResult?.duration = duration\n state.testResult?.result = result\n return .none\n case let .apiKeySelection(action):\n switch action {\n case .binding(\\APIKeySelection.State.apiKeyName):\n UserDefaults.shared.set(state.apiKeySelection.apiKeyName, for: \\.serpAPIKeyName)\n return .none\n default:\n return .none\n }\n }\n }\n }\n}\n\nfinal class WebSearchViewSettings: ObservableObject {\n @AppStorage(\\.serpAPIEngine) var serpAPIEngine\n @AppStorage(\\.headlessBrowserEngine) var headlessBrowserEngine\n @AppStorage(\\.searchProvider) var searchProvider\n init() {}\n}\n\nstruct WebSearchView: View {\n @Perception.Bindable var store: StoreOf\n @Environment(\\.openURL) var openURL\n @StateObject var settings = WebSearchViewSettings()\n\n var body: some View {\n WithPerceptionTracking {\n ScrollView {\n VStack(alignment: .leading) {\n Form {\n Picker(\"Search Provider\", selection: $settings.searchProvider) {\n ForEach(UserDefaultPreferenceKeys.SearchProvider.allCases, id: \\.self) {\n provider in\n switch provider {\n case .serpAPI:\n Text(\"Serp API\").tag(provider)\n case .headlessBrowser:\n Text(\"Headless Browser\").tag(provider)\n }\n \n }\n }\n .pickerStyle(.segmented)\n }\n\n switch settings.searchProvider {\n case .serpAPI:\n serpAPIForm()\n case .headlessBrowser:\n headlessBrowserForm()\n }\n }\n .padding()\n }\n .safeAreaInset(edge: .bottom) {\n VStack(spacing: 0) {\n Divider()\n HStack {\n Button(\"Test Search\") {\n store.send(.test)\n }\n Spacer()\n }\n .padding()\n }\n .background(.regularMaterial)\n }\n .sheet(item: $store.testResult) { testResult in\n testResultView(testResult: testResult)\n }\n .onAppear {\n store.send(.appear)\n }\n }\n }\n\n @ViewBuilder\n func serpAPIForm() -> some View {\n SubSection(\n title: Text(\"Serp API Settings\"),\n description: \"\"\"\n Use Serp API to do web search. Serp API is more reliable and faster than headless browser. But you need to provide an API key for it.\n \"\"\"\n ) {\n Picker(\"Engine\", selection: $settings.serpAPIEngine) {\n ForEach(\n UserDefaultPreferenceKeys.SerpAPIEngine.allCases,\n id: \\.self\n ) { engine in\n Text(engine.rawValue).tag(engine)\n }\n }\n \n WithPerceptionTracking {\n APIKeyPicker(store: store.scope(\n state: \\.apiKeySelection,\n action: \\.apiKeySelection\n ))\n }\n }\n }\n\n @ViewBuilder\n func headlessBrowserForm() -> some View {\n SubSection(\n title: Text(\"Headless Browser Settings\"),\n description: \"\"\"\n The app will open a webview in the background to do web search. This method uses a set of rules to extract information from the web page, if you notice that it stops working, please submit an issue to the developer.\n \"\"\"\n ) {\n Picker(\"Engine\", selection: $settings.headlessBrowserEngine) {\n ForEach(\n UserDefaultPreferenceKeys.HeadlessBrowserEngine.allCases,\n id: \\.self\n ) { engine in\n Text(engine.rawValue).tag(engine)\n }\n }\n }\n }\n\n @ViewBuilder\n func testResultView(testResult: WebSearchSettings.TestResult) -> some View {\n VStack {\n Text(\"Test Result\")\n .padding(.top)\n .font(.headline)\n\n if let result = testResult.result {\n switch result {\n case let .success(webSearchResult):\n VStack(alignment: .leading) {\n Text(\"Success (Completed in \\(testResult.duration, specifier: \"%.2f\")s)\")\n .foregroundColor(.green)\n\n Text(\"Found \\(webSearchResult.webPages.count) results:\")\n\n ScrollView {\n ForEach(webSearchResult.webPages, id: \\.urlString) { page in\n HStack {\n VStack(alignment: .leading) {\n Text(page.title)\n .font(.headline)\n Text(page.urlString)\n .font(.caption)\n .foregroundColor(.blue)\n Text(page.snippet)\n .padding(.top, 2)\n }\n Spacer(minLength: 0)\n }\n .padding(.vertical, 4)\n Divider()\n }\n }\n }\n .padding()\n case let .failure(error):\n VStack(alignment: .leading) {\n Text(\"Error (Completed in \\(testResult.duration, specifier: \"%.2f\")s)\")\n .foregroundColor(.red)\n Text(error.localizedDescription)\n }\n }\n } else {\n ProgressView().padding()\n }\n\n Spacer()\n \n VStack(spacing: 0) {\n Divider()\n \n HStack {\n Spacer()\n \n Button(\"Close\") {\n store.testResult = nil\n }\n .keyboardShortcut(.cancelAction)\n }\n .padding()\n }\n }\n .frame(minWidth: 400, minHeight: 300)\n }\n}\n\n// Helper struct to make TestResult identifiable for sheet presentation\nprivate struct TestResultWrapper: Identifiable {\n var id: UUID = .init()\n var testResult: WebSearchSettings.TestResult\n}\n\nstruct WebSearchView_Previews: PreviewProvider {\n static var previews: some View {\n VStack(alignment: .leading, spacing: 8) {\n WebSearchView(store: .init(initialState: .init(), reducer: { WebSearchSettings() }))\n }\n .frame(height: 800)\n .padding(.all, 8)\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/CustomCommandSettings/CustomCommand.swift", "content": "import ComposableArchitecture\nimport Foundation\nimport PlusFeatureFlag\nimport Preferences\nimport SwiftUI\nimport Toast\n\n@Reducer\nstruct CustomCommandFeature {\n @ObservableState\n struct State: Equatable {\n var editCustomCommand: EditCustomCommand.State?\n }\n\n let settings: CustomCommandView.Settings\n\n enum Action: Equatable {\n case createNewCommand\n case editCommand(CustomCommand)\n case editCustomCommand(EditCustomCommand.Action)\n case deleteCommand(CustomCommand)\n case exportCommand(CustomCommand)\n case importCommand(at: URL)\n case importCommandClicked\n }\n\n @Dependency(\\.toast) var toast\n\n var body: some ReducerOf {\n Reduce { state, action in\n switch action {\n case .createNewCommand:\n if !isFeatureAvailable(\\.unlimitedCustomCommands),\n settings.customCommands.count >= 10\n {\n toast(\"Upgrade to Plus to add more commands\", .info)\n return .none\n }\n state.editCustomCommand = EditCustomCommand.State(nil)\n return .none\n case let .editCommand(command):\n state.editCustomCommand = EditCustomCommand.State(command)\n return .none\n case .editCustomCommand(.close):\n state.editCustomCommand = nil\n return .none\n case let .deleteCommand(command):\n settings.customCommands.removeAll(\n where: { $0.id == command.id }\n )\n if state.editCustomCommand?.commandId == command.id {\n state.editCustomCommand = nil\n }\n return .none\n case .editCustomCommand:\n return .none\n case let .exportCommand(command):\n return .run { _ in\n do {\n let data = try JSONEncoder().encode(command)\n let filename = \"CustomCommand-\\(command.name).json\"\n\n let url = await withCheckedContinuation { continuation in\n Task { @MainActor in\n let panel = NSSavePanel()\n panel.canCreateDirectories = true\n panel.nameFieldStringValue = filename\n let result = await panel.begin()\n switch result {\n case .OK:\n continuation.resume(returning: panel.url)\n default:\n continuation.resume(returning: nil)\n }\n }\n }\n\n if let url {\n try data.write(to: url)\n toast(\"Saved!\", .info)\n }\n\n } catch {\n toast(error.localizedDescription, .error)\n }\n }\n\n case let .importCommand(url):\n if !isFeatureAvailable(\\.unlimitedCustomCommands),\n settings.customCommands.count >= 10\n {\n toast(\"Upgrade to Plus to add more commands\", .info)\n return .none\n }\n \n do {\n let data = try Data(contentsOf: url)\n var command = try JSONDecoder().decode(CustomCommand.self, from: data)\n command.commandId = UUID().uuidString\n settings.customCommands.append(command)\n toast(\"Imported custom command \\(command.name)!\", .info)\n } catch {\n toast(\"Failed to import command: \\(error.localizedDescription)\", .error)\n }\n return .none\n\n case .importCommandClicked:\n return .run { send in\n let url = await withCheckedContinuation { continuation in\n Task { @MainActor in\n let panel = NSOpenPanel()\n panel.allowedContentTypes = [.json]\n let result = await panel.begin()\n if result == .OK {\n continuation.resume(returning: panel.url)\n } else {\n continuation.resume(returning: nil)\n }\n }\n }\n \n if let url {\n await send(.importCommand(at: url))\n }\n }\n }\n }.ifLet(\\.editCustomCommand, action: \\.editCustomCommand) {\n EditCustomCommand(settings: settings)\n }\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/CustomCommandSettings/CustomCommandView.swift", "content": "import ComposableArchitecture\nimport MarkdownUI\nimport PlusFeatureFlag\nimport Preferences\nimport SharedUIComponents\nimport SwiftUI\nimport Toast\n\nextension List {\n @ViewBuilder\n func removeBackground() -> some View {\n if #available(macOS 13.0, *) {\n scrollContentBackground(.hidden)\n .listRowBackground(EmptyView())\n } else {\n background(Color.clear)\n .listRowBackground(EmptyView())\n }\n }\n}\n\nlet customCommandStore = StoreOf(\n initialState: .init(),\n reducer: { CustomCommandFeature(settings: .init()) }\n)\n\nstruct CustomCommandView: View {\n final class Settings: ObservableObject {\n @AppStorage(\\.customCommands) var customCommands\n\n init(customCommands: AppStorage<[CustomCommand]>? = nil) {\n if let list = customCommands {\n _customCommands = list\n }\n }\n }\n\n var store: StoreOf\n @StateObject var settings = Settings()\n @Environment(\\.toast) var toast\n\n var body: some View {\n HStack(spacing: 0) {\n LeftPanel(store: store, settings: settings)\n Divider()\n RightPanel(store: store)\n }\n }\n\n struct LeftPanel: View {\n let store: StoreOf\n @ObservedObject var settings: Settings\n @Environment(\\.toast) var toast\n\n var body: some View {\n WithPerceptionTracking {\n List {\n ForEach(settings.customCommands, id: \\.commandId) { command in\n CommandButton(store: store, command: command)\n }\n .onMove(perform: { indices, newOffset in\n settings.customCommands.move(fromOffsets: indices, toOffset: newOffset)\n })\n .modify { view in\n if #available(macOS 13.0, *) {\n view.listRowSeparator(.hidden).listSectionSeparator(.hidden)\n } else {\n view\n }\n }\n }\n .removeBackground()\n .padding(.vertical, 4)\n .listStyle(.plain)\n .frame(width: 200)\n .background(Color.primary.opacity(0.05))\n .overlay {\n if settings.customCommands.isEmpty {\n Text(\"\"\"\n Empty\n Add command with \"+\" button\n \"\"\")\n .multilineTextAlignment(.center)\n }\n }\n .safeAreaInset(edge: .bottom) {\n Button(action: {\n store.send(.createNewCommand)\n }) {\n if isFeatureAvailable(\\.unlimitedCustomCommands) {\n Text(Image(systemName: \"plus.circle.fill\")) + Text(\" New Command\")\n } else {\n Text(Image(systemName: \"plus.circle.fill\")) +\n Text(\" New Command (\\(settings.customCommands.count)/10)\")\n }\n }\n .buttonStyle(.plain)\n .padding()\n .contextMenu {\n Button(\"Import\") {\n store.send(.importCommandClicked)\n }\n }\n }\n .onDrop(of: [.json], delegate: FileDropDelegate(store: store, toast: toast))\n }\n }\n }\n\n struct FileDropDelegate: DropDelegate {\n let store: StoreOf\n let toast: (String, ToastType) -> Void\n func performDrop(info: DropInfo) -> Bool {\n let jsonFiles = info.itemProviders(for: [.json])\n for file in jsonFiles {\n file\n .loadInPlaceFileRepresentation(forTypeIdentifier: \"public.json\") { url, _, error in\n Task { @MainActor in\n if let url {\n store.send(.importCommand(at: url))\n } else if let error {\n toast(error.localizedDescription, .error)\n }\n }\n }\n }\n\n return !jsonFiles.isEmpty\n }\n }\n\n struct CommandButton: View {\n @Perception.Bindable var store: StoreOf\n let command: CustomCommand\n\n var body: some View {\n WithPerceptionTracking {\n HStack(spacing: 4) {\n Image(systemName: \"line.3.horizontal\")\n\n VStack(alignment: .leading) {\n Text(command.name)\n .foregroundStyle(.primary)\n\n Group {\n switch command.feature {\n case .chatWithSelection:\n Text(\"Send Message\")\n case .customChat:\n Text(\"Custom Chat\")\n case .promptToCode:\n Text(\"Modification\")\n case .singleRoundDialog:\n Text(\"Single Round Dialog\")\n }\n }\n .font(.caption)\n .foregroundStyle(.tertiary)\n }\n .frame(maxWidth: .infinity, alignment: .leading)\n .contentShape(Rectangle())\n .onTapGesture {\n store.send(.editCommand(command))\n }\n }\n .padding(4)\n .background {\n RoundedRectangle(cornerRadius: 4)\n .fill(\n store.editCustomCommand?.commandId == command.id\n ? Color.primary.opacity(0.05)\n : Color.clear\n )\n }\n .contextMenu {\n Button(\"Remove\") {\n store.send(.deleteCommand(command))\n }\n\n Button(\"Export\") {\n store.send(.exportCommand(command))\n }\n }\n }\n }\n }\n\n struct RightPanel: View {\n let store: StoreOf\n var body: some View {\n WithPerceptionTracking {\n if let store = store.scope(\n state: \\.editCustomCommand,\n action: \\.editCustomCommand\n ) {\n EditCustomCommandView(store: store)\n } else {\n VStack {\n SubSection(title: Text(\"Send Message\")) {\n Text(\n \"This command sends a message to the active chat tab. You can provide additional context as well. The additional context will be removed once a message is sent. If the message provided is empty, you can manually type the message in the chat.\"\n )\n }\n SubSection(title: Text(\"Modification\")) {\n Text(\n \"This command opens the prompt-to-code panel and executes the provided requirements on the selected code. You can provide additional context through the \\\"Extra Context\\\" as well.\"\n )\n }\n SubSection(title: Text(\"Custom Chat\")) {\n Text(\n \"This command will overwrite the context of the chat. You can use it to switch to different contexts in the chat. If a message is provided, it will be sent to the chat as well.\"\n )\n }\n SubSection(title: Text(\"Single Round Dialog\")) {\n Text(\n \"This command allows you to send a message to a temporary chat without opening the chat panel. It is particularly useful for one-time commands, such as running a terminal command with `/shell`. For example, you can set the prompt to `/shell open .` to open the project in Finder.\"\n )\n }\n }\n .padding()\n }\n }\n }\n }\n}\n\nstruct CustomCommandTypeDescription: View {\n let text: String\n var body: some View {\n ScrollView {\n Markdown(text)\n .lineLimit(nil)\n .markdownTheme(\n .gitHub\n .text {\n ForegroundColor(.secondary)\n BackgroundColor(.clear)\n FontSize(14)\n }\n .heading1 { conf in\n VStack(alignment: .leading, spacing: 0) {\n conf.label\n .relativePadding(.bottom, length: .em(0.3))\n .relativeLineSpacing(.em(0.125))\n .markdownMargin(top: 24, bottom: 16)\n .markdownTextStyle {\n FontWeight(.semibold)\n FontSize(.em(1.25))\n }\n Divider()\n }\n }\n )\n .padding()\n .background(Color.primary.opacity(0.02), in: RoundedRectangle(cornerRadius: 8))\n .overlay {\n RoundedRectangle(cornerRadius: 8)\n .stroke(style: .init(lineWidth: 1))\n .foregroundColor(Color(nsColor: .separatorColor))\n }\n .padding()\n }\n }\n}\n\n// MARK: - Previews\n\nstruct CustomCommandView_Preview: PreviewProvider {\n static var previews: some View {\n let settings = CustomCommandView.Settings(customCommands: .init(wrappedValue: [\n .init(\n commandId: \"1\",\n name: \"Explain Code\",\n feature: .chatWithSelection(\n extraSystemPrompt: nil,\n prompt: \"Hello\",\n useExtraSystemPrompt: false\n ),\n ignoreExistingAttachments: false,\n attachments: []\n ),\n .init(\n commandId: \"2\",\n name: \"Refactor Code\",\n feature: .promptToCode(\n extraSystemPrompt: nil,\n prompt: \"Refactor\",\n continuousMode: false,\n generateDescription: true\n ),\n ignoreExistingAttachments: false,\n attachments: []\n ),\n ], \"CustomCommandView_Preview\"))\n\n return CustomCommandView(\n store: .init(\n initialState: .init(\n editCustomCommand: .init(.init(.init(\n commandId: \"1\",\n name: \"Explain Code\",\n feature: .chatWithSelection(\n extraSystemPrompt: nil,\n prompt: \"Hello\",\n useExtraSystemPrompt: false\n ),\n ignoreExistingAttachments: false,\n attachments: [] as [CustomCommand.Attachment]\n )))\n ),\n reducer: { CustomCommandFeature(settings: settings) }\n ),\n settings: settings\n )\n }\n}\n\nstruct CustomCommandView_NoEditing_Preview: PreviewProvider {\n static var previews: some View {\n let settings = CustomCommandView.Settings(customCommands: .init(wrappedValue: [\n .init(\n commandId: \"1\",\n name: \"Explain Code\",\n feature: .chatWithSelection(\n extraSystemPrompt: nil,\n prompt: \"Hello\",\n useExtraSystemPrompt: false\n ),\n ignoreExistingAttachments: false,\n attachments: []\n ),\n .init(\n commandId: \"2\",\n name: \"Refactor Code\",\n feature: .promptToCode(\n extraSystemPrompt: nil,\n prompt: \"Refactor\",\n continuousMode: false,\n generateDescription: true\n ),\n ignoreExistingAttachments: false,\n attachments: []\n ),\n ], \"CustomCommandView_Preview\"))\n\n return CustomCommandView(\n store: .init(\n initialState: .init(\n editCustomCommand: nil\n ),\n reducer: { CustomCommandFeature(settings: settings) }\n ),\n settings: settings\n )\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/CustomCommandSettings/EditCustomCommand.swift", "content": "import ComposableArchitecture\nimport Foundation\nimport Preferences\nimport SwiftUI\n\n@Reducer\nstruct EditCustomCommand {\n enum CommandType: Int, CaseIterable, Equatable {\n case sendMessage\n case promptToCode\n case customChat\n case singleRoundDialog\n }\n\n @ObservableState\n struct State: Equatable {\n var name: String = \"\"\n var commandType: CommandType = .sendMessage\n var isNewCommand: Bool = false\n let commandId: String\n\n var sendMessage = EditSendMessageCommand.State()\n var promptToCode = EditPromptToCodeCommand.State()\n var customChat = EditCustomChatCommand.State()\n var singleRoundDialog = EditSingleRoundDialogCommand.State()\n var attachments = EditCustomCommandAttachment.State()\n\n init(_ command: CustomCommand?) {\n isNewCommand = command == nil\n commandId = command?.id ?? UUID().uuidString\n name = command?.name ?? \"New Command\"\n attachments = .init(\n attachments: command?.attachments ?? [],\n ignoreExistingAttachments: command?.ignoreExistingAttachments ?? false\n )\n\n switch command?.feature {\n case let .chatWithSelection(extraSystemPrompt, prompt, useExtraSystemPrompt):\n commandType = .sendMessage\n sendMessage = .init(\n extraSystemPrompt: extraSystemPrompt ?? \"\",\n useExtraSystemPrompt: useExtraSystemPrompt ?? false,\n prompt: prompt ?? \"\"\n )\n case .none:\n commandType = .sendMessage\n sendMessage = .init(\n extraSystemPrompt: \"\",\n useExtraSystemPrompt: false,\n prompt: \"Hello\"\n )\n case let .customChat(systemPrompt, prompt):\n commandType = .customChat\n customChat = .init(\n systemPrompt: systemPrompt ?? \"\",\n prompt: prompt ?? \"\"\n )\n case let .singleRoundDialog(\n systemPrompt,\n overwriteSystemPrompt,\n prompt,\n receiveReplyInNotification\n ):\n commandType = .singleRoundDialog\n singleRoundDialog = .init(\n systemPrompt: systemPrompt ?? \"\",\n overwriteSystemPrompt: overwriteSystemPrompt ?? false,\n prompt: prompt ?? \"\",\n receiveReplyInNotification: receiveReplyInNotification ?? false\n )\n case let .promptToCode(extraSystemPrompt, prompt, continuousMode, generateDescription):\n commandType = .promptToCode\n promptToCode = .init(\n extraSystemPrompt: extraSystemPrompt ?? \"\",\n prompt: prompt ?? \"\",\n continuousMode: continuousMode ?? false,\n generateDescription: generateDescription ?? true\n )\n }\n }\n }\n\n enum Action: BindableAction, Equatable {\n case saveCommand\n case close\n case binding(BindingAction)\n case sendMessage(EditSendMessageCommand.Action)\n case promptToCode(EditPromptToCodeCommand.Action)\n case customChat(EditCustomChatCommand.Action)\n case singleRoundDialog(EditSingleRoundDialogCommand.Action)\n case attachments(EditCustomCommandAttachment.Action)\n }\n\n let settings: CustomCommandView.Settings\n\n @Dependency(\\.toast) var toast\n\n var body: some ReducerOf {\n Scope(state: \\.sendMessage, action: \\.sendMessage) {\n EditSendMessageCommand()\n }\n\n Scope(state: \\.promptToCode, action: \\.promptToCode) {\n EditPromptToCodeCommand()\n }\n\n Scope(state: \\.customChat, action: \\.customChat) {\n EditCustomChatCommand()\n }\n\n Scope(state: \\.singleRoundDialog, action: \\.singleRoundDialog) {\n EditSingleRoundDialogCommand()\n }\n\n Scope(state: \\.attachments, action: \\.attachments) {\n EditCustomCommandAttachment()\n }\n\n BindingReducer()\n\n Reduce { state, action in\n switch action {\n case .saveCommand:\n guard !state.name.isEmpty else {\n toast(\"Command name cannot be empty.\", .error)\n return .none\n }\n\n let newCommand = CustomCommand(\n commandId: state.commandId,\n name: state.name,\n feature: {\n switch state.commandType {\n case .sendMessage:\n let state = state.sendMessage\n return .chatWithSelection(\n extraSystemPrompt: state.extraSystemPrompt,\n prompt: state.prompt,\n useExtraSystemPrompt: state.useExtraSystemPrompt\n )\n case .promptToCode:\n let state = state.promptToCode\n return .promptToCode(\n extraSystemPrompt: state.extraSystemPrompt,\n prompt: state.prompt,\n continuousMode: state.continuousMode,\n generateDescription: state.generateDescription\n )\n case .customChat:\n let state = state.customChat\n return .customChat(\n systemPrompt: state.systemPrompt,\n prompt: state.prompt\n )\n case .singleRoundDialog:\n let state = state.singleRoundDialog\n return .singleRoundDialog(\n systemPrompt: state.systemPrompt,\n overwriteSystemPrompt: state.overwriteSystemPrompt,\n prompt: state.prompt,\n receiveReplyInNotification: state.receiveReplyInNotification\n )\n }\n }(),\n ignoreExistingAttachments: state.attachments.ignoreExistingAttachments,\n attachments: state.attachments.attachments\n )\n\n if state.isNewCommand {\n settings.customCommands.append(newCommand)\n state.isNewCommand = false\n toast(\"The command is created.\", .info)\n } else {\n if let index = settings.customCommands.firstIndex(where: {\n $0.id == newCommand.id\n }) {\n settings.customCommands[index] = newCommand\n } else {\n settings.customCommands.append(newCommand)\n }\n toast(\"The command is updated.\", .info)\n }\n\n return .none\n\n case .close:\n return .none\n\n case .binding:\n return .none\n case .sendMessage:\n return .none\n case .promptToCode:\n return .none\n case .customChat:\n return .none\n case .singleRoundDialog:\n return .none\n case .attachments:\n return .none\n }\n }\n }\n}\n\n@Reducer\nstruct EditCustomCommandAttachment {\n @ObservableState\n struct State: Equatable {\n var attachments: [CustomCommand.Attachment] = []\n var ignoreExistingAttachments: Bool = false\n }\n\n enum Action: BindableAction, Equatable {\n case binding(BindingAction)\n }\n\n var body: some ReducerOf {\n BindingReducer()\n\n Reduce { _, action in\n switch action {\n case .binding:\n return .none\n }\n }\n }\n}\n\n@Reducer\nstruct EditSendMessageCommand {\n @ObservableState\n struct State: Equatable {\n var extraSystemPrompt: String = \"\"\n var useExtraSystemPrompt: Bool = false\n var prompt: String = \"\"\n }\n\n enum Action: BindableAction, Equatable {\n case binding(BindingAction)\n }\n\n var body: some ReducerOf {\n BindingReducer()\n\n Reduce { _, action in\n switch action {\n case .binding:\n return .none\n }\n }\n }\n}\n\n@Reducer\nstruct EditPromptToCodeCommand {\n @ObservableState\n struct State: Equatable {\n var extraSystemPrompt: String = \"\"\n var prompt: String = \"\"\n var continuousMode: Bool = false\n var generateDescription: Bool = false\n }\n\n enum Action: BindableAction, Equatable {\n case binding(BindingAction)\n }\n\n var body: some ReducerOf {\n BindingReducer()\n }\n}\n\n@Reducer\nstruct EditCustomChatCommand {\n @ObservableState\n struct State: Equatable {\n var systemPrompt: String = \"\"\n var prompt: String = \"\"\n }\n\n enum Action: BindableAction, Equatable {\n case binding(BindingAction)\n }\n\n var body: some ReducerOf {\n BindingReducer()\n }\n}\n\n@Reducer\nstruct EditSingleRoundDialogCommand {\n @ObservableState\n struct State: Equatable {\n var systemPrompt: String = \"\"\n var overwriteSystemPrompt: Bool = false\n var prompt: String = \"\"\n var receiveReplyInNotification: Bool = false\n }\n\n enum Action: BindableAction, Equatable {\n case binding(BindingAction)\n }\n\n var body: some ReducerOf {\n BindingReducer()\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/CustomCommandSettings/EditCustomCommandView.swift", "content": "import ComposableArchitecture\nimport MarkdownUI\nimport Preferences\nimport SwiftUI\n\n@MainActor\nstruct EditCustomCommandView: View {\n @Environment(\\.toast) var toast\n @Perception.Bindable var store: StoreOf\n\n init(store: StoreOf) {\n self.store = store\n }\n\n var body: some View {\n ScrollView {\n Form {\n sharedForm\n featureSpecificForm\n }.padding()\n }.safeAreaInset(edge: .bottom) {\n bottomBar\n }\n }\n\n @ViewBuilder var sharedForm: some View {\n WithPerceptionTracking {\n TextField(\"Name\", text: $store.name)\n\n Picker(\"Command Type\", selection: $store.commandType) {\n ForEach(\n EditCustomCommand.CommandType.allCases,\n id: \\.rawValue\n ) { commandType in\n Text({\n switch commandType {\n case .sendMessage:\n return \"Send Message\"\n case .promptToCode:\n return \"Modification\"\n case .customChat:\n return \"Custom Chat\"\n case .singleRoundDialog:\n return \"Single Round Dialog\"\n }\n }() as String).tag(commandType)\n }\n }\n }\n }\n\n @ViewBuilder var featureSpecificForm: some View {\n WithPerceptionTracking {\n switch store.commandType {\n case .sendMessage:\n EditSendMessageCommandView(\n store: store.scope(\n state: \\.sendMessage,\n action: \\.sendMessage\n ),\n attachmentStore: store.scope(\n state: \\.attachments,\n action: \\.attachments\n )\n )\n case .promptToCode:\n EditPromptToCodeCommandView(\n store: store.scope(\n state: \\.promptToCode,\n action: \\.promptToCode\n )\n )\n case .customChat:\n EditCustomChatCommandView(\n store: store.scope(\n state: \\.customChat,\n action: \\.customChat\n ),\n attachmentStore: store.scope(\n state: \\.attachments,\n action: \\.attachments\n )\n )\n case .singleRoundDialog:\n EditSingleRoundDialogCommandView(\n store: store.scope(\n state: \\.singleRoundDialog,\n action: \\.singleRoundDialog\n )\n )\n }\n }\n }\n\n @ViewBuilder var bottomBar: some View {\n WithPerceptionTracking {\n VStack {\n Divider()\n\n VStack(alignment: .trailing) {\n Text(\n \"After renaming or adding a custom command, please restart Xcode to refresh the menu.\"\n )\n .foregroundStyle(.secondary)\n\n HStack {\n Spacer()\n Button(\"Close\") {\n store.send(.close)\n }\n\n if store.isNewCommand {\n Button(\"Add\") {\n store.send(.saveCommand)\n }\n } else {\n Button(\"Save\") {\n store.send(.saveCommand)\n }\n }\n }\n }\n .padding(.horizontal)\n }\n .padding(.bottom)\n .background(.regularMaterial)\n }\n }\n}\n\nstruct CustomCommandAttachmentPickerView: View {\n @Perception.Bindable var store: StoreOf\n @State private var isFileInputPresented = false\n @State private var filePath = \"\"\n\n #if canImport(ProHostApp)\n var body: some View {\n WithPerceptionTracking {\n VStack(alignment: .leading) {\n Text(\"Contexts\")\n\n HStack(alignment: .top) {\n VStack(alignment: .leading, spacing: 8) {\n if store.attachments.isEmpty {\n Text(\"No context\")\n .foregroundStyle(.secondary)\n } else {\n ForEach(store.attachments, id: \\.kind) { attachment in\n HStack {\n switch attachment.kind {\n case let .file(path: path):\n HStack {\n Text(\"File:\")\n Text(path).foregroundStyle(.secondary)\n }\n default:\n Text(attachment.kind.description)\n }\n Spacer()\n Button {\n store.attachments.removeAll { $0.kind == attachment.kind }\n } label: {\n Image(systemName: \"trash\")\n }\n .buttonStyle(.plain)\n }\n }\n }\n }\n .frame(minWidth: 240)\n .padding(.vertical, 8)\n .padding(.horizontal, 8)\n .background {\n RoundedRectangle(cornerRadius: 8)\n .strokeBorder(.separator, lineWidth: 1)\n }\n\n Form {\n Menu {\n ForEach(CustomCommand.Attachment.Kind.allCases.filter { kind in\n !store.attachments.contains { $0.kind == kind }\n }, id: \\.self) { kind in\n if kind == .file(path: \"\") {\n Button {\n isFileInputPresented = true\n } label: {\n Text(\"File...\")\n }\n } else {\n Button {\n store.attachments.append(.init(kind: kind))\n } label: {\n Text(kind.description)\n }\n }\n }\n } label: {\n Label(\"Add context\", systemImage: \"plus\")\n }\n\n Toggle(\n \"Ignore existing contexts\",\n isOn: $store.ignoreExistingAttachments\n )\n }\n }\n }\n .sheet(isPresented: $isFileInputPresented) {\n VStack(alignment: .leading, spacing: 16) {\n Text(\"Enter file path:\")\n .font(.headline)\n Text(\n \"You can enter either an absolute path or a path relative to the project root.\"\n )\n .font(.caption)\n .foregroundStyle(.secondary)\n TextField(\"File path\", text: $filePath)\n .textFieldStyle(.roundedBorder)\n HStack {\n Spacer()\n Button(\"Cancel\") {\n isFileInputPresented = false\n filePath = \"\"\n }\n Button(\"Add\") {\n store.attachments.append(.init(kind: .file(path: filePath)))\n isFileInputPresented = false\n filePath = \"\"\n }\n .disabled(filePath.isEmpty)\n }\n }\n .padding()\n .frame(minWidth: 400)\n }\n }\n }\n #else\n var body: some View { EmptyView() }\n #endif\n}\n\nextension CustomCommand.Attachment.Kind {\n public static var allCases: [CustomCommand.Attachment.Kind] {\n [\n .activeDocument,\n .debugArea,\n .clipboard,\n .senseScope,\n .projectScope,\n .webScope,\n .gitStatus,\n .gitLog,\n .file(path: \"\"),\n ]\n }\n\n var description: String {\n switch self {\n case .activeDocument: return \"Active Document\"\n case .debugArea: return \"Debug Area\"\n case .clipboard: return \"Clipboard\"\n case .senseScope: return \"Sense Scope\"\n case .projectScope: return \"Project Scope\"\n case .webScope: return \"Web Scope\"\n case .gitStatus: return \"Git Status and Diff\"\n case .gitLog: return \"Git Log\"\n case .file: return \"File\"\n }\n }\n}\n\nstruct EditSendMessageCommandView: View {\n @Perception.Bindable var store: StoreOf\n var attachmentStore: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n VStack(alignment: .leading, spacing: 4) {\n Toggle(\"Extra Context\", isOn: $store.useExtraSystemPrompt)\n EditableText(text: $store.extraSystemPrompt)\n }\n .padding(.vertical, 4)\n\n VStack(alignment: .leading, spacing: 4) {\n Text(\"Send immediately\")\n EditableText(text: $store.prompt)\n }\n .padding(.vertical, 4)\n\n CustomCommandAttachmentPickerView(store: attachmentStore)\n .padding(.vertical, 4)\n }\n }\n}\n\nstruct EditPromptToCodeCommandView: View {\n @Perception.Bindable var store: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n Toggle(\"Continuous Mode\", isOn: $store.continuousMode)\n\n VStack(alignment: .leading, spacing: 4) {\n Text(\"Extra Context\")\n EditableText(text: $store.extraSystemPrompt)\n }\n .padding(.vertical, 4)\n\n VStack(alignment: .leading, spacing: 4) {\n Text(\"Instruction\")\n EditableText(text: $store.prompt)\n }\n .padding(.vertical, 4)\n }\n }\n}\n\nstruct EditCustomChatCommandView: View {\n @Perception.Bindable var store: StoreOf\n var attachmentStore: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n VStack(alignment: .leading, spacing: 4) {\n Text(\"Topic\")\n EditableText(text: $store.systemPrompt)\n }\n .padding(.vertical, 4)\n\n VStack(alignment: .leading, spacing: 4) {\n Text(\"Send immediately\")\n EditableText(text: $store.prompt)\n }\n .padding(.vertical, 4)\n\n CustomCommandAttachmentPickerView(store: attachmentStore)\n .padding(.vertical, 4)\n }\n }\n}\n\nstruct EditSingleRoundDialogCommandView: View {\n @Perception.Bindable var store: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n VStack(alignment: .leading, spacing: 4) {\n Text(\"System Prompt\")\n EditableText(text: $store.systemPrompt)\n }\n .padding(.vertical, 4)\n\n Picker(selection: $store.overwriteSystemPrompt) {\n Text(\"Append to default system prompt\").tag(false)\n Text(\"Overwrite default system prompt\").tag(true)\n } label: {\n Text(\"Mode\")\n }\n .pickerStyle(.radioGroup)\n\n VStack(alignment: .leading, spacing: 4) {\n Text(\"Prompt\")\n EditableText(text: $store.prompt)\n }\n .padding(.vertical, 4)\n\n Toggle(\"Receive response in notification\", isOn: $store.receiveReplyInNotification)\n Text(\n \"You will be prompted to grant the app permission to send notifications for the first time.\"\n )\n .font(.footnote)\n .foregroundColor(.secondary)\n }\n }\n}\n\n// MARK: - Preview\n\nstruct EditCustomCommandView_Preview: PreviewProvider {\n static var previews: some View {\n EditCustomCommandView(\n store: .init(\n initialState: .init(.init(\n commandId: \"4\",\n name: \"Explain Code\",\n feature: .promptToCode(\n extraSystemPrompt: nil,\n prompt: \"Hello\",\n continuousMode: false,\n generateDescription: true\n ),\n ignoreExistingAttachments: false,\n attachments: [] as [CustomCommand.Attachment]\n )),\n reducer: {\n EditCustomCommand(\n settings: .init(customCommands: .init(\n wrappedValue: [],\n \"CustomCommandView_Preview\"\n ))\n )\n }\n )\n )\n .frame(width: 800)\n }\n}\n\nstruct EditSingleRoundDialogCommandView_Preview: PreviewProvider {\n static var previews: some View {\n EditSingleRoundDialogCommandView(store: .init(\n initialState: .init(),\n reducer: { EditSingleRoundDialogCommand() }\n ))\n .frame(width: 800, height: 600)\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/DebugView.swift", "content": "import Preferences\nimport SwiftUI\n\nfinal class DebugSettings: ObservableObject {\n @AppStorage(\\.animationACrashSuggestion) var animationACrashSuggestion\n @AppStorage(\\.animationBCrashSuggestion) var animationBCrashSuggestion\n @AppStorage(\\.animationCCrashSuggestion) var animationCCrashSuggestion\n @AppStorage(\\.preCacheOnFileOpen) var preCacheOnFileOpen\n @AppStorage(\\.useCustomScrollViewWorkaround) var useCustomScrollViewWorkaround\n @AppStorage(\\.triggerActionWithAccessibilityAPI) var triggerActionWithAccessibilityAPI\n @AppStorage(\\.alwaysAcceptSuggestionWithAccessibilityAPI)\n var alwaysAcceptSuggestionWithAccessibilityAPI\n @AppStorage(\\.enableXcodeInspectorDebugMenu) var enableXcodeInspectorDebugMenu\n @AppStorage(\\.disableFunctionCalling) var disableFunctionCalling\n @AppStorage(\\.disableGitHubCopilotSettingsAutoRefreshOnAppear)\n var disableGitHubCopilotSettingsAutoRefreshOnAppear\n @AppStorage(\\.useUserDefaultsBaseAPIKeychain) var useUserDefaultsBaseAPIKeychain\n @AppStorage(\\.disableEnhancedWorkspace) var disableEnhancedWorkspace\n @AppStorage(\\.disableGitIgnoreCheck) var disableGitIgnoreCheck\n @AppStorage(\\.disableFileContentManipulationByCheatsheet)\n var disableFileContentManipulationByCheatsheet\n @AppStorage(\\.restartXcodeInspectorIfAccessibilityAPIIsMalfunctioning)\n var restartXcodeInspectorIfAccessibilityAPIIsMalfunctioning\n @AppStorage(\\.restartXcodeInspectorIfAccessibilityAPIIsMalfunctioningNoTimer)\n var restartXcodeInspectorIfAccessibilityAPIIsMalfunctioningNoTimer\n @AppStorage(\\.toastForTheReasonWhyXcodeInspectorNeedsToBeRestarted)\n var toastForTheReasonWhyXcodeInspectorNeedsToBeRestarted\n @AppStorage(\\.observeToAXNotificationWithDefaultMode)\n var observeToAXNotificationWithDefaultMode\n @AppStorage(\\.useCloudflareDomainNameForLicenseCheck)\n var useCloudflareDomainNameForLicenseCheck\n @AppStorage(\\.doNotInstallLaunchAgentAutomatically)\n var doNotInstallLaunchAgentAutomatically\n init() {}\n}\n\nstruct DebugSettingsView: View {\n @StateObject var settings = DebugSettings()\n @Environment(\\.updateChecker) var updateChecker\n\n var body: some View {\n ScrollView {\n Form {\n Toggle(isOn: $settings.animationACrashSuggestion) {\n Text(\"Enable animation A\")\n }\n Toggle(isOn: $settings.animationBCrashSuggestion) {\n Text(\"Enable animation B\")\n }\n Toggle(isOn: $settings.animationCCrashSuggestion) {\n Text(\"Enable widget breathing animation\")\n }\n Toggle(isOn: $settings.preCacheOnFileOpen) {\n Text(\"Cache editor information on file open\")\n }\n Toggle(isOn: $settings.useCustomScrollViewWorkaround) {\n Text(\"Use custom scroll view workaround for smooth scrolling\")\n }\n Toggle(isOn: $settings.triggerActionWithAccessibilityAPI) {\n Text(\"Trigger command with Accessibility API\")\n }\n Group {\n Toggle(isOn: $settings.alwaysAcceptSuggestionWithAccessibilityAPI) {\n Text(\"Always accept suggestion with Accessibility API\")\n }\n Toggle(isOn: $settings.enableXcodeInspectorDebugMenu) {\n Text(\"Enable Xcode inspector debug menu\")\n }\n Toggle(isOn: $settings.disableFunctionCalling) {\n Text(\"Disable function calling for chat feature\")\n }\n Toggle(isOn: $settings.disableGitHubCopilotSettingsAutoRefreshOnAppear) {\n Text(\"Disable GitHub Copilot settings auto refresh status on appear\")\n }\n Toggle(isOn: $settings.useUserDefaultsBaseAPIKeychain) {\n Text(\"Store API keys in UserDefaults\")\n }\n\n Toggle(isOn: $settings.disableEnhancedWorkspace) {\n Text(\"Disable enhanced workspace\")\n }\n\n Toggle(isOn: $settings.disableGitIgnoreCheck) {\n Text(\"Disable git ignore check\")\n }\n\n Toggle(isOn: $settings.disableFileContentManipulationByCheatsheet) {\n Text(\"Disable file content manipulation by cheatsheet\")\n }\n\n Group {\n Toggle(\n isOn: $settings\n .restartXcodeInspectorIfAccessibilityAPIIsMalfunctioning\n ) {\n Text(\n \"Re-activate Xcode Inspector when Accessibility API malfunctioning detected\"\n )\n }\n\n Toggle(\n isOn: $settings\n .restartXcodeInspectorIfAccessibilityAPIIsMalfunctioningNoTimer\n ) {\n Text(\"Trigger malfunctioning detection only with events\")\n }\n\n Toggle(\n isOn: $settings\n .toastForTheReasonWhyXcodeInspectorNeedsToBeRestarted\n ) {\n Text(\"Toast for the reason of re-activation of Xcode Inspector\")\n }\n }\n\n Button(\"Reset migration version to 0\") {\n UserDefaults.shared.set(nil, forKey: \"OldMigrationVersion\")\n }\n\n Button(\"Reset 0.23.0 migration\") {\n UserDefaults.shared.set(\"239\", forKey: \"OldMigrationVersion\")\n UserDefaults.shared.set(nil, forKey: \"MigrateTo240Finished\")\n UserDefaults.shared.set(nil, forKey: \"ChatModels\")\n UserDefaults.shared.set(nil, forKey: \"EmbeddingModels\")\n }\n\n Group {\n Toggle(\n isOn: $settings.observeToAXNotificationWithDefaultMode\n ) {\n Text(\"Observe to AXNotification with default mode\")\n }\n }\n \n Toggle(\n isOn: $settings.useCloudflareDomainNameForLicenseCheck\n ) {\n Text(\"Use Cloudflare domain name for license check\")\n }\n \n Toggle(\n isOn: $settings.doNotInstallLaunchAgentAutomatically\n ) {\n Text(\"Don't install launch agent automatically\")\n }\n \n Button(\"Reset update cycle\") {\n updateChecker.resetUpdateCycle()\n }\n }\n }\n .frame(maxWidth: .infinity)\n .padding()\n }\n }\n}\n\nstruct DebugSettingsView_Preview: PreviewProvider {\n static var previews: some View {\n DebugSettingsView()\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/FeatureSettings/Chat/ChatSettingsGeneralSectionView.swift", "content": "import Client\nimport Preferences\nimport SharedUIComponents\nimport SwiftUI\nimport XPCShared\n\n#if canImport(ProHostApp)\nimport ProHostApp\n#endif\n\nstruct ChatSettingsGeneralSectionView: View {\n class Settings: ObservableObject {\n static let availableLocalizedLocales = Locale.availableLocalizedLocales\n @AppStorage(\\.chatGPTLanguage) var chatGPTLanguage\n @AppStorage(\\.chatGPTTemperature) var chatGPTTemperature\n @AppStorage(\\.chatGPTMaxMessageCount) var chatGPTMaxMessageCount\n @AppStorage(\\.chatFontSize) var chatFontSize\n @AppStorage(\\.chatCodeFont) var chatCodeFont\n\n @AppStorage(\\.defaultChatFeatureChatModelId) var defaultChatFeatureChatModelId\n @AppStorage(\\.preferredChatModelIdForUtilities) var utilityChatModelId\n @AppStorage(\\.defaultChatSystemPrompt) var defaultChatSystemPrompt\n @AppStorage(\\.chatSearchPluginMaxIterations) var chatSearchPluginMaxIterations\n @AppStorage(\\.defaultChatFeatureEmbeddingModelId) var defaultChatFeatureEmbeddingModelId\n @AppStorage(\\.chatModels) var chatModels\n @AppStorage(\\.embeddingModels) var embeddingModels\n @AppStorage(\\.wrapCodeInChatCodeBlock) var wrapCodeInCodeBlock\n @AppStorage(\\.chatPanelFloatOnTopOption) var chatPanelFloatOnTopOption\n @AppStorage(\n \\.keepFloatOnTopIfChatPanelAndXcodeOverlaps\n ) var keepFloatOnTopIfChatPanelAndXcodeOverlaps\n @AppStorage(\n \\.disableFloatOnTopWhenTheChatPanelIsDetached\n ) var disableFloatOnTopWhenTheChatPanelIsDetached\n @AppStorage(\\.openChatMode) var openChatMode\n @AppStorage(\\.openChatInBrowserURL) var openChatInBrowserURL\n @AppStorage(\\.openChatInBrowserInInAppBrowser) var openChatInBrowserInInAppBrowser\n\n var refreshExtensionExtensionOpenChatHandlerTask: Task?\n\n @MainActor\n @Published\n var openChatOptions = [OpenChatMode]()\n\n init() {\n Task { @MainActor in\n refreshExtensionOpenChatHandlers()\n }\n refreshExtensionExtensionOpenChatHandlerTask = Task { [weak self] in\n let sequence = NotificationCenter.default\n .notifications(named: NSApplication.didBecomeActiveNotification)\n for await _ in sequence {\n guard let self else { return }\n await MainActor.run {\n self.refreshExtensionOpenChatHandlers()\n }\n }\n }\n }\n\n @MainActor\n func refreshExtensionOpenChatHandlers() {\n guard let service = try? getService() else { return }\n Task { @MainActor in\n let handlers = try await service\n .send(requestBody: ExtensionServiceRequests.GetExtensionOpenChatHandlers())\n openChatOptions = handlers.map {\n if $0.isBuiltIn {\n return .builtinExtension(\n extensionIdentifier: $0.bundleIdentifier,\n id: $0.id,\n tabName: $0.tabName\n )\n } else {\n return .externalExtension(\n extensionIdentifier: $0.bundleIdentifier,\n id: $0.id,\n tabName: $0.tabName\n )\n }\n }\n }\n }\n }\n\n @Environment(\\.openURL) var openURL\n @Environment(\\.toast) var toast\n @StateObject var settings = Settings()\n @State var maxTokenOverLimit = false\n\n var body: some View {\n VStack {\n openChatSettingsForm\n SettingsDivider(\"Conversation\")\n chatSettingsForm\n SettingsDivider(\"UI\")\n uiForm\n SettingsDivider(\"Plugin\")\n pluginForm\n }\n }\n\n @ViewBuilder\n var openChatSettingsForm: some View {\n Form {\n Picker(\n \"Open Chat Mode\",\n selection: .init(get: {\n settings.openChatMode.value\n }, set: {\n settings.openChatMode = .init($0)\n })\n ) {\n Text(\"Open chat panel\").tag(OpenChatMode.chatPanel)\n Text(\"Open web page in browser\").tag(OpenChatMode.browser)\n ForEach(settings.openChatOptions) { mode in\n switch mode {\n case let .builtinExtension(_, _, name):\n Text(\"Open \\(name) tab\").tag(mode)\n case let .externalExtension(_, _, name):\n Text(\"Open \\(name) tab\").tag(mode)\n default:\n EmptyView()\n }\n }\n }\n\n if settings.openChatMode.value == .browser {\n TextField(\n \"Chat web page URL\",\n text: $settings.openChatInBrowserURL,\n prompt: Text(\"https://\")\n )\n .textFieldStyle(.roundedBorder)\n .disableAutocorrection(true)\n .autocorrectionDisabled(true)\n\n #if canImport(ProHostApp)\n WithFeatureEnabled(\\.browserTab) {\n Toggle(\n \"Open web page in chat panel\",\n isOn: $settings.openChatInBrowserInInAppBrowser\n )\n }\n #endif\n }\n }\n }\n\n @ViewBuilder\n var chatSettingsForm: some View {\n Form {\n Picker(\n \"Chat model\",\n selection: $settings.defaultChatFeatureChatModelId\n ) {\n let allModels = settings.chatModels + [.init(\n id: \"com.github.copilot\",\n name: \"GitHub Copilot Language Server\",\n format: .openAI,\n info: .init()\n )]\n\n if !allModels.contains(where: { $0.id == settings.defaultChatFeatureChatModelId }) {\n Text(\n (allModels.first?.name).map { \"\\($0) (Default)\" } ?? \"No model found\"\n )\n .tag(settings.defaultChatFeatureChatModelId)\n }\n\n ForEach(allModels, id: \\.id) { chatModel in\n Text(chatModel.name).tag(chatModel.id)\n }\n }\n\n Picker(\n \"Utility chat model\",\n selection: $settings.utilityChatModelId\n ) {\n Text(\"Use the default model\").tag(\"\")\n\n if !settings.chatModels.contains(where: { $0.id == settings.utilityChatModelId }),\n !settings.utilityChatModelId.isEmpty\n {\n Text(\n (settings.chatModels.first?.name).map { \"\\($0) (Default)\" }\n ?? \"No Model Found\"\n )\n .tag(settings.utilityChatModelId)\n }\n\n ForEach(settings.chatModels, id: \\.id) { chatModel in\n Text(chatModel.name).tag(chatModel.id)\n }\n }\n\n Picker(\n \"Embedding model\",\n selection: $settings.defaultChatFeatureEmbeddingModelId\n ) {\n if !settings.embeddingModels\n .contains(where: { $0.id == settings.defaultChatFeatureEmbeddingModelId })\n {\n Text(\n (settings.embeddingModels.first?.name).map { \"\\($0) (Default)\" }\n ?? \"No model found\"\n )\n .tag(settings.defaultChatFeatureEmbeddingModelId)\n }\n\n ForEach(settings.embeddingModels, id: \\.id) { embeddingModel in\n Text(embeddingModel.name).tag(embeddingModel.id)\n }\n }\n\n if #available(macOS 13.0, *) {\n LabeledContent(\"Reply in language\") {\n languagePicker\n }\n } else {\n HStack {\n Text(\"Reply in language\")\n languagePicker\n }\n }\n\n HStack {\n Slider(value: $settings.chatGPTTemperature, in: 0...2, step: 0.1) {\n Text(\"Temperature\")\n }\n\n Text(\n \"\\(settings.chatGPTTemperature.formatted(.number.precision(.fractionLength(1))))\"\n )\n .font(.body)\n .foregroundColor(settings.chatGPTTemperature >= 1 ? .red : .secondary)\n .monospacedDigit()\n .padding(.vertical, 2)\n .padding(.horizontal, 6)\n .background(\n RoundedRectangle(cornerRadius: 4, style: .continuous)\n .fill(Color.primary.opacity(0.1))\n )\n }\n\n Picker(\n \"Memory\",\n selection: $settings.chatGPTMaxMessageCount\n ) {\n Text(\"No Limit\").tag(0)\n Text(\"3 Messages\").tag(3)\n Text(\"5 Messages\").tag(5)\n Text(\"7 Messages\").tag(7)\n Text(\"9 Messages\").tag(9)\n Text(\"11 Messages\").tag(11)\n Text(\"21 Messages\").tag(21)\n Text(\"31 Messages\").tag(31)\n Text(\"41 Messages\").tag(41)\n Text(\"51 Messages\").tag(51)\n Text(\"71 Messages\").tag(71)\n Text(\"91 Messages\").tag(91)\n Text(\"111 Messages\").tag(111)\n Text(\"151 Messages\").tag(151)\n Text(\"201 Messages\").tag(201)\n }\n\n VStack(alignment: .leading, spacing: 4) {\n Text(\"Additional system prompt\")\n EditableText(text: $settings.defaultChatSystemPrompt)\n .lineLimit(6)\n }\n .padding(.vertical, 4)\n }\n }\n\n @ViewBuilder\n var uiForm: some View {\n Form {\n HStack {\n TextField(text: .init(get: {\n \"\\(Int(settings.chatFontSize))\"\n }, set: {\n settings.chatFontSize = Double(Int($0) ?? 0)\n })) {\n Text(\"Font size of message\")\n }\n .textFieldStyle(.roundedBorder)\n\n Text(\"pt\")\n }\n\n FontPicker(font: $settings.chatCodeFont) {\n Text(\"Font of code\")\n }\n\n Toggle(isOn: $settings.wrapCodeInCodeBlock) {\n Text(\"Wrap text in code block\")\n }\n\n CodeHighlightThemePicker(scenario: .chat)\n\n Picker(\"Always-on-top behavior\", selection: $settings.chatPanelFloatOnTopOption) {\n Text(\"Always\").tag(UserDefaultPreferenceKeys.ChatPanelFloatOnTopOption.alwaysOnTop)\n Text(\"When Xcode is active\")\n .tag(UserDefaultPreferenceKeys.ChatPanelFloatOnTopOption.onTopWhenXcodeIsActive)\n Text(\"Never\").tag(UserDefaultPreferenceKeys.ChatPanelFloatOnTopOption.never)\n }\n\n Toggle(isOn: $settings.disableFloatOnTopWhenTheChatPanelIsDetached) {\n Text(\"Disable always-on-top when the chat panel is detached\")\n }.disabled(settings.chatPanelFloatOnTopOption == .never)\n\n Toggle(isOn: $settings.keepFloatOnTopIfChatPanelAndXcodeOverlaps) {\n Text(\"Keep always-on-top if the chat panel and Xcode overlaps and Xcode is active\")\n }\n .disabled(\n !settings.disableFloatOnTopWhenTheChatPanelIsDetached\n || settings.chatPanelFloatOnTopOption == .never\n )\n }\n }\n\n @ViewBuilder\n var pluginForm: some View {\n Form {\n TextField(text: .init(get: {\n \"\\(Int(settings.chatSearchPluginMaxIterations))\"\n }, set: {\n settings.chatSearchPluginMaxIterations = Int($0) ?? 0\n })) {\n Text(\"Search plugin max iterations\")\n }\n .textFieldStyle(.roundedBorder)\n }\n }\n\n var languagePicker: some View {\n Menu {\n if !settings.chatGPTLanguage.isEmpty,\n !Settings.availableLocalizedLocales\n .contains(settings.chatGPTLanguage)\n {\n Button(\n settings.chatGPTLanguage,\n action: { self.settings.chatGPTLanguage = settings.chatGPTLanguage }\n )\n }\n Button(\n \"Auto-detected by LLM\",\n action: { self.settings.chatGPTLanguage = \"\" }\n )\n ForEach(\n Settings.availableLocalizedLocales,\n id: \\.self\n ) { localizedLocales in\n Button(\n localizedLocales,\n action: { self.settings.chatGPTLanguage = localizedLocales }\n )\n }\n } label: {\n Text(\n settings.chatGPTLanguage.isEmpty\n ? \"Auto-detected by LLM\"\n : settings.chatGPTLanguage\n )\n }\n }\n}\n\n// MARK: - Preview\n\n//\n// #Preview {\n// ScrollView {\n// ChatSettingsView()\n// .padding()\n// }\n// .frame(height: 800)\n// .environment(\\.overrideFeatureFlag, \\.never)\n// }\n//\n\n" }, { "path": "Core/Sources/HostApp/FeatureSettings/Chat/ChatSettingsView.swift", "content": "import Preferences\nimport SharedUIComponents\nimport SwiftUI\n\nstruct ChatSettingsView: View {\n enum Tab {\n case general\n }\n \n @State var tabSelection: Tab = .general\n \n var body: some View {\n VStack(spacing: 0) {\n Picker(\"\", selection: $tabSelection) {\n Text(\"General\").tag(Tab.general)\n }\n .pickerStyle(.segmented)\n .padding(8)\n \n Divider()\n .shadow(radius: 10)\n \n ScrollView {\n Group {\n switch tabSelection {\n case .general:\n ChatSettingsGeneralSectionView()\n }\n }.padding()\n }\n }\n }\n}\n\n#Preview {\n ChatSettingsView()\n .frame(width: 600, height: 500)\n}\n" }, { "path": "Core/Sources/HostApp/FeatureSettings/PromptToCodeSettingsView.swift", "content": "import Preferences\nimport SharedUIComponents\nimport SwiftUI\n\nstruct PromptToCodeSettingsView: View {\n final class Settings: ObservableObject {\n @AppStorage(\\.hideCommonPrecedingSpacesInPromptToCode)\n var hideCommonPrecedingSpaces\n @AppStorage(\\.promptToCodeCodeFont)\n var font\n @AppStorage(\\.promptToCodeGenerateDescription)\n var promptToCodeGenerateDescription\n @AppStorage(\\.promptToCodeGenerateDescriptionInUserPreferredLanguage)\n var promptToCodeGenerateDescriptionInUserPreferredLanguage\n @AppStorage(\\.promptToCodeChatModelId)\n var promptToCodeChatModelId\n @AppStorage(\\.promptToCodeEmbeddingModelId)\n var promptToCodeEmbeddingModelId\n @AppStorage(\\.wrapCodeInPromptToCode) var wrapCode\n @AppStorage(\\.chatModels) var chatModels\n @AppStorage(\\.embeddingModels) var embeddingModels\n init() {}\n }\n\n @StateObject var settings = Settings()\n\n var body: some View {\n VStack(alignment: .center) {\n Form {\n Picker(\n \"Chat model\",\n selection: $settings.promptToCodeChatModelId\n ) {\n Text(\"Same as chat feature\").tag(\"\")\n\n if !settings.chatModels\n .contains(where: { $0.id == settings.promptToCodeChatModelId }),\n !settings.promptToCodeChatModelId.isEmpty\n {\n Text(\n (settings.chatModels.first?.name).map { \"\\($0) (Default)\" }\n ?? \"No model found\"\n )\n .tag(settings.promptToCodeChatModelId)\n }\n\n ForEach(settings.chatModels, id: \\.id) { chatModel in\n Text(chatModel.name).tag(chatModel.id)\n }\n }\n }\n\n SettingsDivider(\"UI\")\n\n Form {\n Toggle(isOn: $settings.hideCommonPrecedingSpaces) {\n Text(\"Hide common preceding spaces\")\n }\n\n Toggle(isOn: $settings.wrapCode) {\n Text(\"Wrap code\")\n }\n\n CodeHighlightThemePicker(scenario: .promptToCode)\n\n FontPicker(font: $settings.font) {\n Text(\"Font\")\n }\n }\n }\n }\n}\n\n#Preview {\n PromptToCodeSettingsView()\n .padding()\n}\n\n" }, { "path": "Core/Sources/HostApp/FeatureSettings/Suggestion/SuggestionFeatureDisabledLanguageListView.swift", "content": "import SuggestionBasic\nimport SwiftUI\nimport SharedUIComponents\n\nstruct SuggestionFeatureDisabledLanguageListView: View {\n final class Settings: ObservableObject {\n @AppStorage(\\.suggestionFeatureDisabledLanguageList)\n var suggestionFeatureDisabledLanguageList: [String]\n\n init(suggestionFeatureDisabledLanguageList: AppStorage<[String]>? = nil) {\n if let list = suggestionFeatureDisabledLanguageList {\n _suggestionFeatureDisabledLanguageList = list\n }\n }\n }\n\n var isOpen: Binding\n @State var isAddingNewProject = false\n @StateObject var settings = Settings()\n\n var body: some View {\n VStack(spacing: 0) {\n HStack {\n Button(action: {\n self.isOpen.wrappedValue = false\n }) {\n Image(systemName: \"xmark.circle.fill\")\n .foregroundStyle(.secondary)\n .padding()\n }\n .buttonStyle(.plain)\n Text(\"Disabled Languages\")\n Spacer()\n }\n .background(Color(nsColor: .separatorColor))\n\n List {\n ForEach(\n settings.suggestionFeatureDisabledLanguageList,\n id: \\.self\n ) { language in\n HStack {\n Text(language.capitalized)\n .contextMenu {\n Button(\"Remove\") {\n settings.suggestionFeatureDisabledLanguageList.removeAll(\n where: { $0 == language }\n )\n }\n }\n Spacer()\n\n Button(action: {\n settings.suggestionFeatureDisabledLanguageList.removeAll(\n where: { $0 == language }\n )\n }) {\n Image(systemName: \"trash.fill\")\n .foregroundStyle(.secondary)\n }\n .buttonStyle(.plain)\n }\n }\n .modify { view in\n if #available(macOS 13.0, *) {\n view.listRowSeparator(.hidden).listSectionSeparator(.hidden)\n } else {\n view\n }\n }\n }\n .removeBackground()\n .overlay {\n if settings.suggestionFeatureDisabledLanguageList.isEmpty {\n Text(\"\"\"\n Empty\n Disable the language of a file by right clicking the indicator widget.\n \"\"\")\n .multilineTextAlignment(.center)\n .padding()\n }\n }\n }\n .focusable(false)\n .frame(width: 300, height: 400)\n .background(Color(nsColor: .windowBackgroundColor))\n }\n}\n\nstruct SuggestionFeatureDisabledLanguageListView_Preview: PreviewProvider {\n static var previews: some View {\n SuggestionFeatureDisabledLanguageListView(\n isOpen: .constant(true),\n settings: .init(suggestionFeatureDisabledLanguageList: .init(wrappedValue: [\n \"hello/2\",\n \"hello/3\",\n \"hello/4\",\n ], \"SuggestionFeatureDisabledLanguageListView_Preview\"))\n )\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/FeatureSettings/Suggestion/SuggestionFeatureEnabledProjectListView.swift", "content": "import SharedUIComponents\nimport SwiftUI\n\nstruct SuggestionFeatureEnabledProjectListView: View {\n final class Settings: ObservableObject {\n @AppStorage(\\.suggestionFeatureEnabledProjectList)\n var suggestionFeatureEnabledProjectList: [String]\n\n init(suggestionFeatureEnabledProjectList: AppStorage<[String]>? = nil) {\n if let list = suggestionFeatureEnabledProjectList {\n _suggestionFeatureEnabledProjectList = list\n }\n }\n }\n\n var isOpen: Binding\n @State var isAddingNewProject = false\n @StateObject var settings = Settings()\n\n var body: some View {\n VStack(spacing: 0) {\n HStack {\n Button(action: {\n self.isOpen.wrappedValue = false\n }) {\n Image(systemName: \"xmark.circle.fill\")\n .foregroundStyle(.secondary)\n .padding()\n }\n .buttonStyle(.plain)\n Text(\"Enabled Projects\")\n Spacer()\n Button(action: {\n isAddingNewProject = true\n }) {\n Image(systemName: \"plus.circle.fill\")\n .foregroundStyle(.secondary)\n .padding()\n }\n .buttonStyle(.plain)\n }\n .background(Color(nsColor: .separatorColor))\n\n List {\n ForEach(\n settings.suggestionFeatureEnabledProjectList,\n id: \\.self\n ) { project in\n HStack {\n Text(project)\n .contextMenu {\n Button(\"Remove\") {\n settings.suggestionFeatureEnabledProjectList.removeAll(\n where: { $0 == project }\n )\n }\n }\n Spacer()\n\n Button(action: {\n settings.suggestionFeatureEnabledProjectList.removeAll(\n where: { $0 == project }\n )\n }) {\n Image(systemName: \"trash.fill\")\n .foregroundStyle(.secondary)\n }\n .buttonStyle(.plain)\n }\n }\n .modify { view in\n if #available(macOS 13.0, *) {\n view.listRowSeparator(.hidden).listSectionSeparator(.hidden)\n } else {\n view\n }\n }\n }\n .removeBackground()\n .overlay {\n if settings.suggestionFeatureEnabledProjectList.isEmpty {\n Text(\"\"\"\n Empty\n Add project with \"+\" button\n Or right clicking the indicator widget\n \"\"\")\n .multilineTextAlignment(.center)\n }\n }\n }\n .focusable(false)\n .frame(width: 300, height: 400)\n .background(Color(nsColor: .windowBackgroundColor))\n .sheet(isPresented: $isAddingNewProject) {\n SuggestionFeatureAddEnabledProjectView(isOpen: $isAddingNewProject, settings: settings)\n }\n }\n}\n\nstruct SuggestionFeatureAddEnabledProjectView: View {\n var isOpen: Binding\n var settings: SuggestionFeatureEnabledProjectListView.Settings\n @State var rootPath = \"\"\n\n var body: some View {\n VStack {\n Text(\n \"Enter the root path of the project. Do not use `~` to replace /Users/yourUserName.\"\n )\n TextField(\"Root path\", text: $rootPath)\n HStack {\n Spacer()\n Button(\"Cancel\") {\n isOpen.wrappedValue = false\n }\n Button(\"Add\") {\n settings.suggestionFeatureEnabledProjectList.append(rootPath)\n isOpen.wrappedValue = false\n }\n }\n }\n .padding()\n .frame(minWidth: 500)\n .background(Color(nsColor: .windowBackgroundColor))\n }\n}\n\nstruct SuggestionFeatureEnabledProjectListView_Preview: PreviewProvider {\n static var previews: some View {\n SuggestionFeatureEnabledProjectListView(\n isOpen: .constant(true),\n settings: .init(suggestionFeatureEnabledProjectList: .init(wrappedValue: [\n \"hello/2\",\n \"hello/3\",\n \"hello/4\",\n ], \"SuggestionFeatureEnabledProjectListView_Preview\"))\n )\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/FeatureSettings/Suggestion/SuggestionSettingsGeneralSectionView.swift", "content": "import Client\nimport Preferences\nimport SharedUIComponents\nimport SwiftUI\nimport XPCShared\n\n#if canImport(ProHostApp)\nimport ProHostApp\n#endif\n\nstruct SuggestionSettingsGeneralSectionView: View {\n struct SuggestionFeatureProviderOption: Identifiable, Hashable {\n var id: String {\n (builtInProvider?.rawValue).map(String.init) ?? bundleIdentifier ?? \"n/A\"\n }\n\n var name: String\n var builtInProvider: BuiltInSuggestionFeatureProvider?\n var bundleIdentifier: String?\n\n func hash(into hasher: inout Hasher) {\n id.hash(into: &hasher)\n }\n\n init(\n name: String,\n builtInProvider: BuiltInSuggestionFeatureProvider? = nil,\n bundleIdentifier: String? = nil\n ) {\n self.name = name\n self.builtInProvider = builtInProvider\n self.bundleIdentifier = bundleIdentifier\n }\n }\n\n final class Settings: ObservableObject {\n @AppStorage(\\.realtimeSuggestionToggle)\n var realtimeSuggestionToggle\n @AppStorage(\\.realtimeSuggestionDebounce)\n var realtimeSuggestionDebounce\n @AppStorage(\\.suggestionPresentationMode)\n var suggestionPresentationMode\n @AppStorage(\\.disableSuggestionFeatureGlobally)\n var disableSuggestionFeatureGlobally\n @AppStorage(\\.suggestionFeatureEnabledProjectList)\n var suggestionFeatureEnabledProjectList\n @AppStorage(\\.hideCommonPrecedingSpacesInSuggestion)\n var hideCommonPrecedingSpacesInSuggestion\n @AppStorage(\\.suggestionCodeFont)\n var font\n @AppStorage(\\.suggestionFeatureProvider)\n var suggestionFeatureProvider\n @AppStorage(\\.suggestionDisplayCompactMode)\n var suggestionDisplayCompactMode\n @AppStorage(\\.acceptSuggestionWithTab)\n var acceptSuggestionWithTab\n @AppStorage(\\.dismissSuggestionWithEsc)\n var dismissSuggestionWithEsc\n\n var refreshExtensionSuggestionFeatureProvidersTask: Task?\n\n @MainActor\n @Published\n var extensionSuggestionFeatureProviderOptions = [SuggestionFeatureProviderOption]()\n\n init() {\n Task { @MainActor in\n refreshExtensionSuggestionFeatureProviders()\n }\n refreshExtensionSuggestionFeatureProvidersTask = Task { [weak self] in\n let sequence = NotificationCenter.default\n .notifications(named: NSApplication.didBecomeActiveNotification)\n for await _ in sequence {\n guard let self else { return }\n await MainActor.run {\n self.refreshExtensionSuggestionFeatureProviders()\n }\n }\n }\n }\n\n @MainActor\n func refreshExtensionSuggestionFeatureProviders() {\n guard let service = try? getService() else { return }\n Task { @MainActor in\n let services = try await service\n .send(requestBody: ExtensionServiceRequests.GetExtensionSuggestionServices())\n extensionSuggestionFeatureProviderOptions = services.map {\n .init(name: $0.name, bundleIdentifier: $0.bundleIdentifier)\n }\n }\n }\n }\n\n @StateObject var settings = Settings()\n @State var isSuggestionFeatureEnabledListPickerOpen = false\n @State var isSuggestionFeatureDisabledLanguageListViewOpen = false\n @State var isTabToAcceptSuggestionModifierViewOpen = false\n\n var body: some View {\n Form {\n Picker(selection: $settings.suggestionPresentationMode) {\n ForEach(PresentationMode.allCases, id: \\.rawValue) {\n switch $0 {\n case .nearbyTextCursor:\n Text(\"Nearby text cursor\").tag($0)\n case .floatingWidget:\n Text(\"Floating widget\").tag($0)\n }\n }\n } label: {\n Text(\"Presentation\")\n }\n\n Picker(selection: Binding(get: {\n switch settings.suggestionFeatureProvider {\n case let .builtIn(provider):\n return SuggestionFeatureProviderOption(\n name: \"\",\n builtInProvider: provider\n )\n case let .extension(name, identifier):\n return SuggestionFeatureProviderOption(\n name: name,\n bundleIdentifier: identifier\n )\n }\n }, set: { (option: SuggestionFeatureProviderOption) in\n if let provider = option.builtInProvider {\n settings.suggestionFeatureProvider = .builtIn(provider)\n } else {\n settings.suggestionFeatureProvider = .extension(\n name: option.name,\n bundleIdentifier: option.bundleIdentifier ?? \"\"\n )\n }\n })) {\n ForEach(BuiltInSuggestionFeatureProvider.allCases, id: \\.rawValue) {\n switch $0 {\n case .gitHubCopilot:\n Text(\"GitHub Copilot\")\n .tag(SuggestionFeatureProviderOption(name: \"\", builtInProvider: $0))\n case .codeium:\n Text(\"Codeium\")\n .tag(SuggestionFeatureProviderOption(name: \"\", builtInProvider: $0))\n }\n }\n\n ForEach(settings.extensionSuggestionFeatureProviderOptions, id: \\.self) { item in\n Text(item.name).tag(item)\n }\n\n if case let .extension(name, identifier) = settings.suggestionFeatureProvider {\n if !settings.extensionSuggestionFeatureProviderOptions.contains(where: {\n $0.bundleIdentifier == identifier\n }) {\n Text(\"\\(name) (Not found)\").tag(\n SuggestionFeatureProviderOption(\n name: name,\n bundleIdentifier: identifier\n )\n )\n }\n }\n } label: {\n Text(\"Feature provider\")\n }\n\n Toggle(isOn: $settings.realtimeSuggestionToggle) {\n Text(\"Real-time suggestion\")\n }\n\n Toggle(isOn: $settings.acceptSuggestionWithTab) {\n HStack {\n Text(\"Accept suggestion with Tab\")\n\n Button(action: {\n isTabToAcceptSuggestionModifierViewOpen = true\n }) {\n Image(systemName: \"gearshape.fill\")\n }\n .buttonStyle(.plain)\n }\n }.sheet(isPresented: $isTabToAcceptSuggestionModifierViewOpen) {\n TabToAcceptSuggestionModifierView()\n }\n\n Toggle(isOn: $settings.dismissSuggestionWithEsc) {\n Text(\"Dismiss suggestion with ESC\")\n }\n\n HStack {\n Toggle(isOn: $settings.disableSuggestionFeatureGlobally) {\n Text(\"Disable suggestion feature globally\")\n }\n\n Button(\"Exception list\") {\n isSuggestionFeatureEnabledListPickerOpen = true\n }\n }.sheet(isPresented: $isSuggestionFeatureEnabledListPickerOpen) {\n SuggestionFeatureEnabledProjectListView(\n isOpen: $isSuggestionFeatureEnabledListPickerOpen\n )\n }\n\n HStack {\n Button(\"Disabled language list\") {\n isSuggestionFeatureDisabledLanguageListViewOpen = true\n }\n }.sheet(isPresented: $isSuggestionFeatureDisabledLanguageListViewOpen) {\n SuggestionFeatureDisabledLanguageListView(\n isOpen: $isSuggestionFeatureDisabledLanguageListViewOpen\n )\n }\n\n HStack {\n Slider(value: $settings.realtimeSuggestionDebounce, in: 0.1...2, step: 0.1) {\n Text(\"Real-time suggestion debounce\")\n }\n\n Text(\n \"\\(settings.realtimeSuggestionDebounce.formatted(.number.precision(.fractionLength(2))))s\"\n )\n .font(.body)\n .monospacedDigit()\n .padding(.vertical, 2)\n .padding(.horizontal, 6)\n .background(\n RoundedRectangle(cornerRadius: 4, style: .continuous)\n .fill(Color.primary.opacity(0.1))\n )\n }\n }\n\n SettingsDivider(\"UI\")\n\n Form {\n Toggle(isOn: $settings.suggestionDisplayCompactMode) {\n Text(\"Hide buttons\")\n }\n\n Toggle(isOn: $settings.hideCommonPrecedingSpacesInSuggestion) {\n Text(\"Hide common preceding spaces\")\n }\n\n CodeHighlightThemePicker(scenario: .suggestion)\n\n FontPicker(font: $settings.font) {\n Text(\"Font\")\n }\n }\n }\n\n struct TabToAcceptSuggestionModifierView: View {\n final class Settings: ObservableObject {\n @AppStorage(\\.acceptSuggestionWithModifierCommand)\n var needCommand\n @AppStorage(\\.acceptSuggestionWithModifierOption)\n var needOption\n @AppStorage(\\.acceptSuggestionWithModifierShift)\n var needShift\n @AppStorage(\\.acceptSuggestionWithModifierControl)\n var needControl\n @AppStorage(\\.acceptSuggestionWithModifierOnlyForSwift)\n var onlyForSwift\n @AppStorage(\\.acceptSuggestionLineWithModifierControl)\n var acceptLineWithControl\n }\n\n @StateObject var settings = Settings()\n @Environment(\\.dismiss) var dismiss\n\n var body: some View {\n VStack(spacing: 0) {\n Form {\n Text(\"Accept suggestion with modifier\")\n .font(.headline)\n HStack {\n Toggle(isOn: $settings.needCommand) {\n Text(\"Command\")\n }\n Toggle(isOn: $settings.needOption) {\n Text(\"Option\")\n }\n Toggle(isOn: $settings.needShift) {\n Text(\"Shift\")\n }\n Toggle(isOn: $settings.needControl) {\n Text(\"Control\")\n }\n }\n Toggle(isOn: $settings.onlyForSwift) {\n Text(\"Only require modifiers for Swift\")\n }\n \n Divider()\n \n Toggle(isOn: $settings.acceptLineWithControl) {\n Text(\"Accept suggestion first line with Control\")\n }\n }\n .padding()\n\n Divider()\n\n HStack {\n Spacer()\n Button(action: { dismiss() }) {\n Text(\"Done\")\n }\n .keyboardShortcut(.defaultAction)\n }\n .padding()\n }\n }\n }\n}\n\n#Preview {\n SuggestionSettingsGeneralSectionView()\n .padding()\n}\n\n#Preview {\n SuggestionSettingsGeneralSectionView.TabToAcceptSuggestionModifierView()\n}\n\n" }, { "path": "Core/Sources/HostApp/FeatureSettings/Suggestion/SuggestionSettingsView.swift", "content": "import Client\nimport Preferences\nimport SharedUIComponents\nimport SwiftUI\nimport XPCShared\n\nstruct SuggestionSettingsView: View {\n var tabContainer: ExternalTabContainer {\n ExternalTabContainer.tabContainer(for: \"SuggestionSettings\")\n }\n \n enum Tab: Hashable {\n case general\n case other(String)\n }\n \n @State var tabSelection: Tab = .general\n \n var body: some View {\n VStack(spacing: 0) {\n Picker(\"\", selection: $tabSelection) {\n Text(\"General\").tag(Tab.general)\n ForEach(tabContainer.tabs, id: \\.id) { tab in\n Text(tab.title).tag(Tab.other(tab.id))\n }\n }\n .pickerStyle(.segmented)\n .padding(8)\n \n Divider()\n .shadow(radius: 10)\n \n ScrollView {\n Group {\n switch tabSelection {\n case .general:\n SuggestionSettingsGeneralSectionView()\n case let .other(id):\n tabContainer.tabView(for: id)\n }\n }.padding()\n }\n }\n }\n}\n\nstruct SuggestionSettingsView_Previews: PreviewProvider {\n static var previews: some View {\n SuggestionSettingsView()\n .frame(width: 600, height: 500)\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/FeatureSettings/TerminalSettingsView.swift", "content": "import Preferences\nimport SharedUIComponents\nimport SwiftUI\n\n#if canImport(ProHostApp)\nimport ProHostApp\n#endif\n\nstruct TerminalSettingsView: View {\n class Settings: ObservableObject {\n @AppStorage(\\.terminalFont) var terminalFont\n init() {}\n }\n \n @StateObject var settings = Settings()\n \n var body: some View {\n ScrollView {\n Form {\n FontPicker(font: $settings.terminalFont) {\n Text(\"Font of code\")\n }\n }\n }\n \n }\n}\n" }, { "path": "Core/Sources/HostApp/FeatureSettings/XcodeSettingsView.swift", "content": "import Foundation\nimport SharedUIComponents\nimport SwiftUI\n\n#if canImport(ProHostApp)\nimport ProHostApp\n#endif\n\nstruct XcodeSettingsView: View {\n var body: some View {\n VStack {\n #if canImport(ProHostApp)\n CloseXcodeIdleTabsSettingsView()\n #endif\n\n EmptyView()\n }\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/FeatureSettingsView.swift", "content": "import SwiftUI\nimport SharedUIComponents\n\nstruct FeatureSettingsView: View {\n var tabContainer: ExternalTabContainer {\n ExternalTabContainer.tabContainer(for: \"Features\")\n }\n \n @State var tag = 0\n\n var body: some View {\n SidebarTabView(tag: $tag) {\n SuggestionSettingsView()\n .sidebarItem(\n tag: 0,\n title: \"Suggestion\",\n subtitle: \"Generate suggestions for your code\",\n image: \"lightbulb\"\n )\n\n ChatSettingsView()\n .sidebarItem(\n tag: 1,\n title: \"Chat\",\n subtitle: \"Chat about your code\",\n image: \"character.bubble\"\n )\n\n ScrollView {\n PromptToCodeSettingsView().padding()\n }\n .sidebarItem(\n tag: 2,\n title: \"Modification\",\n subtitle: \"Write or modify code with natural language\",\n image: \"paintbrush\"\n )\n\n ScrollView {\n XcodeSettingsView().padding()\n }\n .sidebarItem(\n tag: 3,\n title: \"Xcode\",\n subtitle: \"Xcode related features\",\n image: \"hammer.circle\"\n )\n \n ForEach(Array(tabContainer.tabs.enumerated()), id: \\.1.id) { index, tab in\n ScrollView {\n tab.viewBuilder().padding()\n }\n .sidebarItem(\n tag: 4 + index,\n title: tab.title,\n subtitle: tab.description,\n image: tab.image\n )\n }\n }\n }\n}\n\nstruct FeatureSettingsView_Previews: PreviewProvider {\n static var previews: some View {\n FeatureSettingsView()\n .frame(width: 800)\n }\n}\n" }, { "path": "Core/Sources/HostApp/General.swift", "content": "import Client\nimport ComposableArchitecture\nimport Foundation\nimport LaunchAgentManager\nimport SwiftUI\nimport XPCShared\n\n@Reducer\nstruct General {\n @ObservableState\n struct State: Equatable {\n var xpcServiceVersion: String?\n var isAccessibilityPermissionGranted: Bool?\n var isReloading = false\n @Presents var alert: AlertState?\n }\n\n enum Action {\n case appear\n case setupLaunchAgentIfNeeded\n case setupLaunchAgentClicked\n case removeLaunchAgentClicked\n case reloadLaunchAgentClicked\n case openExtensionManager\n case reloadStatus\n case finishReloading(xpcServiceVersion: String, permissionGranted: Bool)\n case failedReloading\n case alert(PresentationAction)\n\n case setupLaunchAgent\n case finishSetupLaunchAgent\n case finishRemoveLaunchAgent\n case finishReloadLaunchAgent\n\n @CasePathable\n enum Alert: Equatable {\n case moveToApplications\n case moveTo(URL)\n case install\n }\n }\n\n @Dependency(\\.toast) var toast\n\n struct ReloadStatusCancellableId: Hashable {}\n\n static var didWarnInstallationPosition: Bool {\n get { UserDefaults.standard.bool(forKey: \"didWarnInstallationPosition\") }\n set { UserDefaults.standard.set(newValue, forKey: \"didWarnInstallationPosition\") }\n }\n\n static var bundleIsInApplicationsFolder: Bool {\n Bundle.main.bundleURL.path.hasPrefix(\"/Applications\")\n }\n\n var body: some ReducerOf {\n Reduce { state, action in\n switch action {\n case .appear:\n if Self.bundleIsInApplicationsFolder {\n return .run { send in\n await send(.setupLaunchAgentIfNeeded)\n }\n }\n\n if !Self.didWarnInstallationPosition {\n Self.didWarnInstallationPosition = true\n state.alert = .init {\n TextState(\"Move to Applications Folder?\")\n } actions: {\n ButtonState(action: .moveToApplications) {\n TextState(\"Move\")\n }\n ButtonState(role: .cancel) {\n TextState(\"Not Now\")\n }\n } message: {\n TextState(\n \"To ensure the best experience, please move the app to the Applications folder. If the app is not inside the Applications folder, please set up the launch agent manually by clicking the button.\"\n )\n }\n }\n\n return .none\n\n case .setupLaunchAgentIfNeeded:\n return .run { send in\n #if DEBUG\n // do not auto install on debug build\n #else\n do {\n try await LaunchAgentManager()\n .setupLaunchAgentForTheFirstTimeIfNeeded()\n } catch {\n toast(error.localizedDescription, .error)\n }\n #endif\n await send(.reloadStatus)\n }\n\n case .setupLaunchAgentClicked:\n if Self.bundleIsInApplicationsFolder {\n return .run { send in\n await send(.setupLaunchAgent)\n }\n }\n\n state.alert = .init {\n TextState(\"Setup Launch Agent\")\n } actions: {\n ButtonState(action: .install) {\n TextState(\"Setup\")\n }\n\n ButtonState(action: .moveToApplications) {\n TextState(\"Move to Applications Folder\")\n }\n\n ButtonState(role: .cancel) {\n TextState(\"Cancel\")\n }\n } message: {\n TextState(\n \"It's recommended to move the app into the Applications folder. But you can still keep it in the current folder and install the launch agent to ~/Library/LaunchAgents.\"\n )\n }\n\n return .none\n\n case .removeLaunchAgentClicked:\n return .run { send in\n do {\n try await LaunchAgentManager().removeLaunchAgent()\n await send(.finishRemoveLaunchAgent)\n } catch {\n toast(error.localizedDescription, .error)\n }\n await send(.reloadStatus)\n }\n\n case .reloadLaunchAgentClicked:\n return .run { send in\n do {\n try await LaunchAgentManager().reloadLaunchAgent()\n await send(.finishReloadLaunchAgent)\n } catch {\n toast(error.localizedDescription, .error)\n }\n await send(.reloadStatus)\n }\n\n case .setupLaunchAgent:\n return .run { send in\n do {\n try await LaunchAgentManager().setupLaunchAgent()\n await send(.finishSetupLaunchAgent)\n } catch {\n toast(error.localizedDescription, .error)\n }\n await send(.reloadStatus)\n }\n\n case .finishSetupLaunchAgent:\n state.alert = .init {\n TextState(\"Launch Agent Installed\")\n } actions: {\n ButtonState {\n TextState(\"OK\")\n }\n } message: {\n TextState(\n \"The launch agent has been installed. Please restart the app.\"\n )\n }\n return .none\n\n case .finishRemoveLaunchAgent:\n state.alert = .init {\n TextState(\"Launch Agent Removed\")\n } actions: {\n ButtonState {\n TextState(\"OK\")\n }\n } message: {\n TextState(\n \"The launch agent has been removed.\"\n )\n }\n return .none\n\n case .finishReloadLaunchAgent:\n state.alert = .init {\n TextState(\"Launch Agent Reloaded\")\n } actions: {\n ButtonState {\n TextState(\"OK\")\n }\n } message: {\n TextState(\n \"The launch agent has been reloaded.\"\n )\n }\n return .none\n\n case .openExtensionManager:\n return .run { send in\n let service = try getService()\n do {\n _ = try await service\n .send(requestBody: ExtensionServiceRequests.OpenExtensionManager())\n } catch {\n toast(error.localizedDescription, .error)\n await send(.failedReloading)\n }\n }\n\n case .reloadStatus:\n state.isReloading = true\n return .run { send in\n let service = try getService()\n do {\n let isCommunicationReady = try await service.launchIfNeeded()\n if isCommunicationReady {\n let xpcServiceVersion = try await service.getXPCServiceVersion().version\n let isAccessibilityPermissionGranted = try await service\n .getXPCServiceAccessibilityPermission()\n await send(.finishReloading(\n xpcServiceVersion: xpcServiceVersion,\n permissionGranted: isAccessibilityPermissionGranted\n ))\n } else {\n toast(\"Launching service app.\", .info)\n try await Task.sleep(nanoseconds: 5_000_000_000)\n await send(.reloadStatus)\n }\n } catch let error as XPCCommunicationBridgeError {\n toast(\n \"Failed to reach communication bridge. \\(error.localizedDescription)\",\n .error\n )\n await send(.failedReloading)\n } catch {\n toast(error.localizedDescription, .error)\n await send(.failedReloading)\n }\n }.cancellable(id: ReloadStatusCancellableId(), cancelInFlight: true)\n\n case let .finishReloading(version, granted):\n state.xpcServiceVersion = version\n state.isAccessibilityPermissionGranted = granted\n state.isReloading = false\n return .none\n\n case .failedReloading:\n state.isReloading = false\n return .none\n\n case let .alert(.presented(action)):\n switch action {\n case .moveToApplications:\n return .run { send in\n let appURL = URL(fileURLWithPath: \"/Applications\")\n await send(.alert(.presented(.moveTo(appURL))))\n }\n\n case let .moveTo(url):\n return .run { _ in\n do {\n try FileManager.default.moveItem(\n at: Bundle.main.bundleURL,\n to: url.appendingPathComponent(\n Bundle.main.bundleURL.lastPathComponent\n )\n )\n await NSApplication.shared.terminate(nil)\n } catch {\n toast(error.localizedDescription, .error)\n }\n }\n case .install:\n return .run { send in\n await send(.setupLaunchAgent)\n }\n }\n\n case .alert(.dismiss):\n state.alert = nil\n return .none\n }\n }\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/GeneralView.swift", "content": "import Client\nimport ComposableArchitecture\nimport KeyboardShortcuts\nimport LaunchAgentManager\nimport Preferences\nimport SharedUIComponents\nimport SwiftUI\n\nstruct GeneralView: View {\n let store: StoreOf\n\n var body: some View {\n ScrollView {\n VStack(alignment: .leading, spacing: 0) {\n AppInfoView(store: store)\n SettingsDivider()\n ExtensionServiceView(store: store)\n SettingsDivider()\n LaunchAgentView(store: store)\n SettingsDivider()\n GeneralSettingsView()\n }\n }\n .onAppear {\n store.send(.appear)\n }\n }\n}\n\nstruct AppInfoView: View {\n @State var appVersion = Bundle.main.infoDictionary?[\"CFBundleShortVersionString\"] as? String\n @Environment(\\.updateChecker) var updateChecker\n @Perception.Bindable var store: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n VStack(alignment: .leading) {\n HStack(alignment: .top) {\n Text(\n Bundle.main\n .object(forInfoDictionaryKey: \"HOST_APP_NAME\") as? String\n ?? \"Copilot for Xcode\"\n )\n .font(.title)\n Text(appVersion ?? \"\")\n .font(.footnote)\n .foregroundColor(.secondary)\n\n Spacer()\n\n Button(action: {\n store.send(.openExtensionManager)\n }) {\n HStack(spacing: 2) {\n Image(systemName: \"puzzlepiece.extension.fill\")\n Text(\"Extensions\")\n }\n }\n\n Button(action: {\n updateChecker.checkForUpdates()\n }) {\n HStack(spacing: 2) {\n Image(systemName: \"arrow.up.right.circle.fill\")\n Text(\"Check for Updates\")\n }\n }\n }\n\n HStack(spacing: 16) {\n Link(\n destination: URL(string: \"https://github.com/intitni/CopilotForXcode\")!\n ) {\n HStack(spacing: 2) {\n Image(systemName: \"link\")\n Text(\"GitHub\")\n }\n }\n .focusable(false)\n .foregroundColor(.accentColor)\n\n Link(destination: URL(string: \"https://www.buymeacoffee.com/intitni\")!) {\n HStack(spacing: 2) {\n Image(systemName: \"cup.and.saucer.fill\")\n Text(\"Buy Me A Coffee\")\n }\n }\n .foregroundColor(.accentColor)\n .focusable(false)\n }\n }\n .padding()\n .alert($store.scope(state: \\.alert, action: \\.alert))\n }\n }\n}\n\nstruct ExtensionServiceView: View {\n @Perception.Bindable var store: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n VStack(alignment: .leading) {\n Text(\"Extension Service Version: \\(store.xpcServiceVersion ?? \"Loading..\")\")\n\n let grantedStatus: String = {\n guard let granted = store.isAccessibilityPermissionGranted\n else { return \"Loading..\" }\n return granted ? \"Granted\" : \"Not Granted\"\n }()\n Text(\"Accessibility Permission: \\(grantedStatus)\")\n\n HStack {\n Button(action: { store.send(.reloadStatus) }) {\n Text(\"Refresh\")\n }.disabled(store.isReloading)\n\n Button(action: {\n Task {\n let workspace = NSWorkspace.shared\n let url = Bundle.main.bundleURL\n .appendingPathComponent(\"Contents\")\n .appendingPathComponent(\"Applications\")\n .appendingPathComponent(\"CopilotForXcodeExtensionService.app\")\n workspace.activateFileViewerSelecting([url])\n }\n }) {\n Text(\"Reveal Extension Service in Finder\")\n }\n\n Button(action: {\n let url = URL(\n string: \"x-apple.systempreferences:com.apple.preference.security?Privacy_Accessibility\"\n )!\n NSWorkspace.shared.open(url)\n }) {\n Text(\"Accessibility Settings\")\n }\n\n Button(action: {\n let url = URL(\n string: \"x-apple.systempreferences:com.apple.ExtensionsPreferences\"\n )!\n NSWorkspace.shared.open(url)\n }) {\n Text(\"Extensions Settings\")\n }\n }\n }\n }\n .padding()\n }\n}\n\nstruct LaunchAgentView: View {\n @Perception.Bindable var store: StoreOf\n @Environment(\\.toast) var toast\n\n var body: some View {\n WithPerceptionTracking {\n VStack(alignment: .leading) {\n HStack {\n Button(action: {\n store.send(.setupLaunchAgentClicked)\n }) {\n Text(\"Setup Launch Agent\")\n }\n\n Button(action: {\n store.send(.removeLaunchAgentClicked)\n }) {\n Text(\"Remove Launch Agent\")\n }\n\n Button(action: {\n store.send(.reloadLaunchAgentClicked)\n }) {\n Text(\"Reload Launch Agent\")\n }\n }\n }\n .padding()\n }\n }\n}\n\nstruct GeneralSettingsView: View {\n class Settings: ObservableObject {\n @AppStorage(\\.quitXPCServiceOnXcodeAndAppQuit)\n var quitXPCServiceOnXcodeAndAppQuit\n @AppStorage(\\.suggestionWidgetPositionMode)\n var suggestionWidgetPositionMode\n @AppStorage(\\.widgetColorScheme)\n var widgetColorScheme\n @AppStorage(\\.preferWidgetToStayInsideEditorWhenWidthGreaterThan)\n var preferWidgetToStayInsideEditorWhenWidthGreaterThan\n @AppStorage(\\.hideCircularWidget)\n var hideCircularWidget\n @AppStorage(\\.showHideWidgetShortcutGlobally)\n var showHideWidgetShortcutGlobally\n @AppStorage(\\.installBetaBuilds)\n var installBetaBuilds\n }\n\n @StateObject var settings = Settings()\n @Environment(\\.updateChecker) var updateChecker\n @State var automaticallyCheckForUpdate: Bool?\n\n var body: some View {\n Form {\n Toggle(isOn: $settings.quitXPCServiceOnXcodeAndAppQuit) {\n Text(\"Quit service when Xcode and host app are terminated\")\n }\n\n Toggle(isOn: .init(\n get: { automaticallyCheckForUpdate ?? updateChecker.automaticallyChecksForUpdates },\n set: {\n updateChecker.automaticallyChecksForUpdates = $0\n automaticallyCheckForUpdate = $0\n }\n )) {\n Text(\"Automatically Check for Update\")\n }\n\n Toggle(isOn: $settings.installBetaBuilds) {\n Text(\"Install beta builds\")\n }\n\n Picker(selection: $settings.suggestionWidgetPositionMode) {\n ForEach(SuggestionWidgetPositionMode.allCases, id: \\.rawValue) {\n switch $0 {\n case .fixedToBottom:\n Text(\"Fixed to Bottom\").tag($0)\n case .alignToTextCursor:\n Text(\"Follow Text Cursor\").tag($0)\n }\n }\n } label: {\n Text(\"Widget position\")\n }\n\n Picker(selection: $settings.widgetColorScheme) {\n ForEach(WidgetColorScheme.allCases, id: \\.rawValue) {\n switch $0 {\n case .system:\n Text(\"System\").tag($0)\n case .light:\n Text(\"Light\").tag($0)\n case .dark:\n Text(\"Dark\").tag($0)\n }\n }\n } label: {\n Text(\"Widget color scheme\")\n }\n\n HStack(alignment: .firstTextBaseline) {\n TextField(text: .init(get: {\n \"\\(Int(settings.preferWidgetToStayInsideEditorWhenWidthGreaterThan))\"\n }, set: {\n settings\n .preferWidgetToStayInsideEditorWhenWidthGreaterThan =\n Double(Int($0) ?? 0)\n })) {\n Text(\"Prefer widget to be inside editor\\nwhen width greater than\")\n .multilineTextAlignment(.trailing)\n }\n .textFieldStyle(.roundedBorder)\n\n Text(\"pt\")\n }\n\n KeyboardShortcuts.Recorder(\"Hotkey to Toggle Widgets\", name: .showHideWidget) { _ in\n // It's not used in this app!\n KeyboardShortcuts.disable(.showHideWidget)\n }\n\n Toggle(isOn: $settings.showHideWidgetShortcutGlobally) {\n Text(\"Enable the Hotkey Globally\")\n }\n\n Toggle(isOn: $settings.hideCircularWidget) {\n Text(\"Hide indicator widget\")\n }\n }.padding()\n }\n}\n\nstruct WidgetPositionIcon: View {\n var position: SuggestionWidgetPositionMode\n var isSelected: Bool\n\n var body: some View {\n ZStack {\n Rectangle()\n .fill(Color(nsColor: .textBackgroundColor))\n Rectangle()\n .fill(Color.accentColor.opacity(0.2))\n .frame(width: 120, height: 20)\n }\n .frame(width: 120, height: 80)\n }\n}\n\nstruct LargeIconPicker<\n Data: RandomAccessCollection,\n ID: Hashable,\n Content: View,\n Label: View\n>: View {\n @Binding var selection: Data.Element\n var data: Data\n var id: KeyPath\n var builder: (Data.Element, _ isSelected: Bool) -> Content\n var label: () -> Label\n\n @ViewBuilder\n var content: some View {\n HStack {\n ForEach(data, id: id) { item in\n let isSelected = selection[keyPath: id] == item[keyPath: id]\n Button(action: {\n selection = item\n }) {\n builder(item, isSelected)\n .clipShape(RoundedRectangle(cornerRadius: 8))\n .overlay {\n RoundedRectangle(cornerRadius: 8)\n .stroke(\n isSelected ? Color.accentColor : Color.primary.opacity(0.1),\n style: .init(lineWidth: 2)\n )\n }\n }.buttonStyle(.plain)\n }\n }\n }\n\n var body: some View {\n if #available(macOS 13.0, *) {\n LabeledContent {\n content\n } label: {\n label()\n }\n } else {\n VStack {\n label()\n content\n }\n }\n }\n}\n\nstruct GeneralView_Previews: PreviewProvider {\n static var previews: some View {\n GeneralView(store: .init(initialState: .init(), reducer: { General() }))\n .frame(height: 800)\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/HandleToast.swift", "content": "import Dependencies\nimport SwiftUI\nimport Toast\n\nstruct ToastHandler: View {\n @ObservedObject var toastController: ToastController\n let namespace: String?\n\n init(toastController: ToastController, namespace: String?) {\n _toastController = .init(wrappedValue: toastController)\n self.namespace = namespace\n }\n\n var body: some View {\n VStack(spacing: 4) {\n ForEach(toastController.messages) { message in\n if let n = message.namespace, n != namespace {\n EmptyView()\n } else {\n message.content\n .foregroundColor(.white)\n .padding(8)\n .background({\n switch message.type {\n case .info: return Color.accentColor\n case .error: return Color(nsColor: .systemRed)\n case .warning: return Color(nsColor: .systemOrange)\n }\n }() as Color, in: RoundedRectangle(cornerRadius: 8))\n .shadow(color: Color.black.opacity(0.2), radius: 4)\n }\n }\n }\n .padding()\n .allowsHitTesting(false)\n }\n}\n\nextension View {\n func handleToast(namespace: String? = nil) -> some View {\n @Dependency(\\.toastController) var toastController\n return overlay(alignment: .bottom) {\n ToastHandler(toastController: toastController, namespace: namespace)\n }.environment(\\.toast) { [toastController] content, type in\n toastController.toast(content: content, type: type, namespace: namespace)\n }\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/HostApp.swift", "content": "import Client\nimport ComposableArchitecture\nimport Foundation\nimport KeyboardShortcuts\n\n#if canImport(LicenseManagement)\nimport ProHostApp\n#endif\n\nextension KeyboardShortcuts.Name {\n static let showHideWidget = Self(\"ShowHideWidget\")\n}\n\n@Reducer\nstruct HostApp {\n @ObservableState\n struct State: Equatable {\n var general = General.State()\n var chatModelManagement = ChatModelManagement.State()\n var embeddingModelManagement = EmbeddingModelManagement.State()\n var webSearchSettings = WebSearchSettings.State()\n }\n\n enum Action {\n case appear\n case general(General.Action)\n case chatModelManagement(ChatModelManagement.Action)\n case embeddingModelManagement(EmbeddingModelManagement.Action)\n case webSearchSettings(WebSearchSettings.Action)\n }\n\n @Dependency(\\.toast) var toast\n \n init() {\n KeyboardShortcuts.userDefaults = .shared\n }\n\n var body: some ReducerOf {\n Scope(state: \\.general, action: \\.general) {\n General()\n }\n\n Scope(state: \\.chatModelManagement, action: \\.chatModelManagement) {\n ChatModelManagement()\n }\n\n Scope(state: \\.embeddingModelManagement, action: \\.embeddingModelManagement) {\n EmbeddingModelManagement()\n }\n \n Scope(state: \\.webSearchSettings, action: \\.webSearchSettings) {\n WebSearchSettings()\n }\n\n Reduce { _, action in\n switch action {\n case .appear:\n #if canImport(ProHostApp)\n ProHostApp.start()\n #endif\n return .none\n\n case .general:\n return .none\n\n case .chatModelManagement:\n return .none\n\n case .embeddingModelManagement:\n return .none\n \n case .webSearchSettings:\n return .none\n }\n }\n }\n}\n\nimport Dependencies\nimport Keychain\nimport Preferences\n\nstruct UserDefaultsDependencyKey: DependencyKey {\n static var liveValue: UserDefaultsType = UserDefaults.shared\n static var previewValue: UserDefaultsType = {\n let it = UserDefaults(suiteName: \"HostAppPreview\")!\n it.removePersistentDomain(forName: \"HostAppPreview\")\n return it\n }()\n\n static var testValue: UserDefaultsType = {\n let it = UserDefaults(suiteName: \"HostAppTest\")!\n it.removePersistentDomain(forName: \"HostAppTest\")\n return it\n }()\n}\n\nextension DependencyValues {\n var userDefaults: UserDefaultsType {\n get { self[UserDefaultsDependencyKey.self] }\n set { self[UserDefaultsDependencyKey.self] = newValue }\n }\n}\n\nstruct APIKeyKeychainDependencyKey: DependencyKey {\n static var liveValue: KeychainType = Keychain.apiKey\n static var previewValue: KeychainType = FakeKeyChain()\n static var testValue: KeychainType = FakeKeyChain()\n}\n\nextension DependencyValues {\n var apiKeyKeychain: KeychainType {\n get { self[APIKeyKeychainDependencyKey.self] }\n set { self[APIKeyKeychainDependencyKey.self] = newValue }\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/IsPreview.swift", "content": "import Foundation\n\nvar isPreview: Bool { ProcessInfo.processInfo.environment[\"XCODE_RUNNING_FOR_PREVIEWS\"] == \"1\" }\n" }, { "path": "Core/Sources/HostApp/LaunchAgentManager.swift", "content": "import Foundation\nimport LaunchAgentManager\n\nextension LaunchAgentManager {\n init() {\n self.init(\n serviceIdentifier: Bundle.main\n .object(forInfoDictionaryKey: \"BUNDLE_IDENTIFIER_BASE\") as! String +\n \".CommunicationBridge\",\n executableURL: Bundle.main.bundleURL\n .appendingPathComponent(\"Contents\")\n .appendingPathComponent(\"Applications\")\n .appendingPathComponent(\"CommunicationBridge\"),\n bundleIdentifier: Bundle.main\n .object(forInfoDictionaryKey: \"BUNDLE_IDENTIFIER_BASE\") as! String\n )\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/ServiceView.swift", "content": "import ComposableArchitecture\nimport SwiftUI\n\nstruct ServiceView: View {\n let store: StoreOf\n @State var tag = 0\n\n var body: some View {\n WithPerceptionTracking {\n SidebarTabView(tag: $tag) {\n WithPerceptionTracking {\n ScrollView {\n GitHubCopilotView().padding()\n }.sidebarItem(\n tag: 0,\n title: \"GitHub Copilot\",\n subtitle: \"Suggestion\",\n image: \"globe\"\n )\n\n ScrollView {\n CodeiumView().padding()\n }.sidebarItem(\n tag: 1,\n title: \"Codeium\",\n subtitle: \"Suggestion\",\n image: \"globe\"\n )\n\n ChatModelManagementView(store: store.scope(\n state: \\.chatModelManagement,\n action: \\.chatModelManagement\n )).sidebarItem(\n tag: 2,\n title: \"Chat Models\",\n subtitle: \"Chat, Modification\",\n image: \"globe\"\n )\n\n EmbeddingModelManagementView(store: store.scope(\n state: \\.embeddingModelManagement,\n action: \\.embeddingModelManagement\n )).sidebarItem(\n tag: 3,\n title: \"Embedding Models\",\n subtitle: \"Chat, Modification\",\n image: \"globe\"\n )\n\n WebSearchView(store: store.scope(\n state: \\.webSearchSettings,\n action: \\.webSearchSettings\n )).sidebarItem(\n tag: 4,\n title: \"Web Search\",\n subtitle: \"Chat, Modification\",\n image: \"globe\"\n )\n\n ScrollView {\n OtherSuggestionServicesView().padding()\n }.sidebarItem(\n tag: 5,\n title: \"Other Suggestion Services\",\n subtitle: \"Suggestion\",\n image: \"globe\"\n )\n }\n }\n }\n }\n}\n\nstruct AccountView_Previews: PreviewProvider {\n static var previews: some View {\n ServiceView(store: .init(initialState: .init(), reducer: { HostApp() }))\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/SharedComponents/CodeHighlightThemePicker.swift", "content": "import Foundation\nimport Preferences\nimport SwiftUI\n\npublic struct CodeHighlightThemePicker: View {\n public enum Scenario {\n case suggestion\n case promptToCode\n case chat\n }\n\n let scenario: Scenario\n\n public init(scenario: Scenario) {\n self.scenario = scenario\n }\n\n public var body: some View {\n switch scenario {\n case .suggestion:\n SuggestionThemePicker()\n case .promptToCode:\n PromptToCodeThemePicker()\n case .chat:\n ChatThemePicker()\n }\n }\n\n struct SuggestionThemePicker: View {\n @AppStorage(\\.syncSuggestionHighlightTheme) var sync: Bool\n var body: some View {\n SyncToggle(sync: $sync)\n }\n }\n\n struct PromptToCodeThemePicker: View {\n @AppStorage(\\.syncPromptToCodeHighlightTheme) var sync: Bool\n var body: some View {\n SyncToggle(sync: $sync)\n }\n }\n\n struct ChatThemePicker: View {\n @AppStorage(\\.syncChatCodeHighlightTheme) var sync: Bool\n var body: some View {\n SyncToggle(sync: $sync)\n }\n }\n\n struct SyncToggle: View {\n @Binding var sync: Bool\n\n var body: some View {\n VStack(alignment: .leading) {\n Toggle(isOn: $sync) {\n Text(\"Sync color scheme with Xcode\")\n }\n\n Text(\"To refresh the theme, you must activate the extension service app once.\")\n .font(.footnote)\n .foregroundColor(.secondary)\n }\n }\n }\n}\n\n#Preview {\n CodeHighlightThemePicker.SyncToggle(sync: .constant(true))\n CodeHighlightThemePicker.SyncToggle(sync: .constant(false))\n}\n\n" }, { "path": "Core/Sources/HostApp/SharedComponents/EditableText.swift", "content": "import Foundation\nimport SwiftUI\n\n// Hack to disable smart quotes and dashes in TextEditor\nextension NSTextView {\n open override var frame: CGRect {\n didSet {\n self.isAutomaticQuoteSubstitutionEnabled = false\n self.isAutomaticDashSubstitutionEnabled = false\n }\n }\n}\n\nstruct EditableText: View {\n var text: Binding\n @State var isEditing: Bool = false\n\n var body: some View {\n Button(action: {\n isEditing = true\n }) {\n HStack(alignment: .top) {\n Text(text.wrappedValue)\n .font(Font.system(.body, design: .monospaced))\n .padding(4)\n .multilineTextAlignment(.leading)\n .frame(maxWidth: .infinity, alignment: .leading)\n .background {\n RoundedRectangle(cornerRadius: 4)\n .fill(Color(nsColor: .textBackgroundColor))\n }\n .overlay {\n RoundedRectangle(cornerRadius: 4)\n .stroke(Color(nsColor: .separatorColor), style: .init(lineWidth: 1))\n }\n Image(systemName: \"square.and.pencil\")\n .resizable()\n .scaledToFit()\n .frame(width: 14)\n .padding(4)\n .background(\n Color.primary.opacity(0.1),\n in: RoundedRectangle(cornerRadius: 4)\n )\n }\n }\n .buttonStyle(.plain)\n .sheet(isPresented: $isEditing) {\n VStack {\n TextEditor(text: text)\n .font(Font.system(.body, design: .monospaced))\n .padding(4)\n .frame(minHeight: 120)\n .multilineTextAlignment(.leading)\n .overlay(\n RoundedRectangle(cornerRadius: 4)\n .stroke(Color(nsColor: .separatorColor), lineWidth: 1)\n )\n\n Button(action: {\n isEditing = false\n }) {\n Text(\"Done\")\n }\n }\n .padding()\n .frame(width: 600, height: 500)\n .background(Color(nsColor: .windowBackgroundColor))\n }\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/SidebarTabView.swift", "content": "import SwiftUI\n\nprivate struct SidebarItem: Identifiable, Equatable {\n var id: Int { tag }\n var tag: Int\n var title: String\n var subtitle: String? = nil\n var image: String? = nil\n}\n\nprivate struct SidebarItemPreferenceKey: PreferenceKey {\n static var defaultValue: [SidebarItem] = []\n static func reduce(value: inout [SidebarItem], nextValue: () -> [SidebarItem]) {\n value.append(contentsOf: nextValue())\n }\n}\n\nprivate struct SidebarTabTagKey: EnvironmentKey {\n static var defaultValue: Int = 0\n}\n\nprivate extension EnvironmentValues {\n var sidebarTabTag: Int {\n get { self[SidebarTabTagKey.self] }\n set { self[SidebarTabTagKey.self] = newValue }\n }\n}\n\nprivate struct SidebarTabViewWrapper: View {\n @Environment(\\.sidebarTabTag) var sidebarTabTag\n var tag: Int\n var title: String\n var subtitle: String? = nil\n var image: String? = nil\n var content: () -> Content\n\n var body: some View {\n Group {\n if tag == sidebarTabTag {\n content()\n } else {\n Color.clear\n }\n }\n .preference(\n key: SidebarItemPreferenceKey.self,\n value: [.init(tag: tag, title: title, subtitle: subtitle, image: image)]\n )\n }\n}\n\nextension View {\n func sidebarItem(\n tag: Int,\n title: String,\n subtitle: String? = nil,\n image: String? = nil\n ) -> some View {\n SidebarTabViewWrapper(\n tag: tag,\n title: title,\n subtitle: subtitle,\n image: image,\n content: { self }\n )\n }\n}\n\nstruct SidebarTabView: View {\n @State private var sidebarItems = [SidebarItem]()\n @Binding var tag: Int\n @ViewBuilder var views: () -> Content\n var body: some View {\n HStack(spacing: 0) {\n ScrollView {\n VStack(alignment: .leading) {\n ForEach(sidebarItems) { item in\n Button(action: {\n tag = item.tag\n }) {\n HStack {\n if let image = item.image {\n Image(systemName: image)\n .resizable()\n .scaledToFit()\n .frame(width: 20, height: 20)\n }\n VStack(alignment: .leading, spacing: 2) {\n Text(item.title)\n .foregroundStyle(.primary)\n if let subtitle = item.subtitle {\n Text(subtitle)\n .lineSpacing(0)\n .font(.caption)\n .foregroundStyle(.secondary)\n .opacity(0.5)\n .multilineTextAlignment(.leading)\n }\n }\n }\n .frame(maxWidth: .infinity, alignment: .leading)\n .padding(.horizontal, 8)\n .padding(.vertical, 4)\n .background(\n Color.primary.opacity(tag == item.tag ? 0.1 : 0),\n in: RoundedRectangle(cornerRadius: 4)\n )\n .padding(.horizontal, 8)\n .contentShape(Rectangle())\n }\n .buttonStyle(.plain)\n }\n }\n .frame(width: 200)\n .padding(.vertical, 8)\n }\n .background(Color.primary.opacity(0.05))\n\n Divider()\n\n ZStack(alignment: .topLeading) {\n views()\n }\n }\n .environment(\\.sidebarTabTag, tag)\n .onPreferenceChange(SidebarItemPreferenceKey.self) { items in\n sidebarItems = items\n }\n }\n}\n\nstruct SidebarTabView_Previews: PreviewProvider {\n static var previews: some View {\n SidebarTabView(tag: .constant(0)) {\n Color.red.sidebarItem(\n tag: 0,\n title: \"Hello\",\n subtitle: \"Meow\\nMeow\",\n image: \"person.circle.fill\"\n )\n Color.blue.sidebarItem(\n tag: 1,\n title: \"World\",\n image: \"person.circle.fill\"\n )\n Color.blue.sidebarItem(\n tag: 3,\n title: \"Pikachu\",\n image: \"person.circle.fill\"\n )\n }\n }\n}\n\n" }, { "path": "Core/Sources/HostApp/TabContainer.swift", "content": "import ComposableArchitecture\nimport Dependencies\nimport Foundation\nimport LaunchAgentManager\nimport SharedUIComponents\nimport SwiftUI\nimport Toast\nimport UpdateChecker\n\n@MainActor\nlet hostAppStore: StoreOf = .init(initialState: .init(), reducer: { HostApp() })\n\npublic struct TabContainer: View {\n let store: StoreOf\n @ObservedObject var toastController: ToastController\n @State private var tabBarItems = [TabBarItem]()\n @State var tag: Int = 0\n\n var externalTabContainer: ExternalTabContainer {\n ExternalTabContainer.tabContainer(for: \"TabContainer\")\n }\n\n public init() {\n toastController = ToastControllerDependencyKey.liveValue\n store = hostAppStore\n }\n\n init(store: StoreOf, toastController: ToastController) {\n self.store = store\n self.toastController = toastController\n }\n\n public var body: some View {\n WithPerceptionTracking {\n VStack(spacing: 0) {\n TabBar(tag: $tag, tabBarItems: tabBarItems)\n .padding(.bottom, 8)\n\n Divider()\n\n ZStack(alignment: .center) {\n GeneralView(store: store.scope(state: \\.general, action: \\.general))\n .tabBarItem(\n tag: 0,\n title: \"General\",\n image: \"app.gift\"\n )\n ServiceView(store: store).tabBarItem(\n tag: 1,\n title: \"Service\",\n image: \"globe\"\n )\n FeatureSettingsView().tabBarItem(\n tag: 2,\n title: \"Feature\",\n image: \"star.square\"\n )\n CustomCommandView(store: customCommandStore).tabBarItem(\n tag: 3,\n title: \"Custom Command\",\n image: \"command.square\"\n )\n \n ForEach(0..: View {\n @Environment(\\.tabBarTabTag) var tabBarTabTag\n var tag: Int\n var title: String\n var image: String\n var content: () -> Content\n\n var body: some View {\n Group {\n if tag == tabBarTabTag {\n content()\n } else {\n Color.clear\n }\n }\n .preference(\n key: TabBarItemPreferenceKey.self,\n value: [.init(tag: tag, title: title, image: image)]\n )\n }\n}\n\nprivate extension View {\n func tabBarItem(\n tag: Int,\n title: String,\n image: String\n ) -> some View {\n TabBarTabViewWrapper(\n tag: tag,\n title: title,\n image: image,\n content: { self }\n )\n }\n}\n\nprivate struct TabBarItem: Identifiable, Equatable {\n var id: Int { tag }\n var tag: Int\n var title: String\n var image: String\n}\n\nprivate struct TabBarItemPreferenceKey: PreferenceKey {\n static var defaultValue: [TabBarItem] = []\n static func reduce(value: inout [TabBarItem], nextValue: () -> [TabBarItem]) {\n value.append(contentsOf: nextValue())\n }\n}\n\nprivate struct TabBarTabTagKey: EnvironmentKey {\n static var defaultValue: Int = 0\n}\n\nprivate extension EnvironmentValues {\n var tabBarTabTag: Int {\n get { self[TabBarTabTagKey.self] }\n set { self[TabBarTabTagKey.self] = newValue }\n }\n}\n\nstruct UpdateCheckerKey: EnvironmentKey {\n static var defaultValue: UpdateChecker = .init(\n hostBundle: nil,\n shouldAutomaticallyCheckForUpdate: false\n )\n}\n\npublic extension EnvironmentValues {\n var updateChecker: UpdateChecker {\n get { self[UpdateCheckerKey.self] }\n set { self[UpdateCheckerKey.self] = newValue }\n }\n}\n\n// MARK: - Previews\n\nstruct TabContainer_Previews: PreviewProvider {\n static var previews: some View {\n TabContainer()\n .frame(width: 800)\n }\n}\n\nstruct TabContainer_Toasts_Previews: PreviewProvider {\n static var previews: some View {\n TabContainer(\n store: .init(initialState: .init(), reducer: { HostApp() }),\n toastController: .init(messages: [\n .init(id: UUID(), type: .info, content: Text(\"info\")),\n .init(id: UUID(), type: .error, content: Text(\"error\")),\n .init(id: UUID(), type: .warning, content: Text(\"warning\")),\n ])\n )\n .frame(width: 800)\n }\n}\n\n" }, { "path": "Core/Sources/KeyBindingManager/KeyBindingManager.swift", "content": "import Foundation\nimport Workspace\n\n\npublic final class KeyBindingManager {\n let tabToAcceptSuggestion: TabToAcceptSuggestion\n\n public init() {\n tabToAcceptSuggestion = .init()\n }\n \n public func start() {\n tabToAcceptSuggestion.start()\n }\n \n @MainActor\n public func stopForExit() {\n tabToAcceptSuggestion.stopForExit()\n }\n}\n\n \n" }, { "path": "Core/Sources/KeyBindingManager/TabToAcceptSuggestion.swift", "content": "import ActiveApplicationMonitor\nimport AppKit\nimport CGEventOverride\nimport CommandHandler\nimport Dependencies\nimport Foundation\nimport Logger\nimport Preferences\nimport SuggestionBasic\nimport UserDefaultsObserver\nimport Workspace\nimport XcodeInspector\n\nfinal class TabToAcceptSuggestion {\n let hook: CGEventHookType = CGEventHook(eventsOfInterest: [.keyDown]) { message in\n Logger.service.debug(\"TabToAcceptSuggestion: \\(message)\")\n }\n\n @Dependency(\\.workspacePool) var workspacePool\n @Dependency(\\.commandHandler) var commandHandler\n\n private var CGEventObservationTask: Task?\n private var isObserving: Bool { CGEventObservationTask != nil }\n private let userDefaultsObserver = UserDefaultsObserver(\n object: UserDefaults.shared, forKeyPaths: [\n UserDefaultPreferenceKeys().acceptSuggestionWithTab.key,\n UserDefaultPreferenceKeys().dismissSuggestionWithEsc.key,\n ], context: nil\n )\n private var stoppedForExit = false\n\n struct ObservationKey: Hashable {}\n\n var canTapToAcceptSuggestion: Bool {\n UserDefaults.shared.value(for: \\.acceptSuggestionWithTab)\n }\n\n var canEscToDismissSuggestion: Bool {\n UserDefaults.shared.value(for: \\.dismissSuggestionWithEsc)\n }\n\n @MainActor\n func stopForExit() {\n stoppedForExit = true\n stopObservation()\n }\n\n init() {\n _ = ThreadSafeAccessToXcodeInspector.shared\n\n hook.add(\n .init(\n eventsOfInterest: [.keyDown],\n convert: { [weak self] _, _, event in\n self?.handleEvent(event) ?? .unchanged\n }\n ),\n forKey: ObservationKey()\n )\n }\n\n func start() {\n Task { [weak self] in\n for await _ in ActiveApplicationMonitor.shared.createInfoStream() {\n guard let self else { return }\n try Task.checkCancellation()\n Task { @MainActor in\n if ActiveApplicationMonitor.shared.activeXcode != nil {\n self.startObservation()\n } else {\n self.stopObservation()\n }\n }\n }\n }\n\n userDefaultsObserver.onChange = { [weak self] in\n guard let self else { return }\n Task { @MainActor in\n if self.canTapToAcceptSuggestion || self.canEscToDismissSuggestion {\n self.startObservation()\n } else {\n self.stopObservation()\n }\n }\n }\n }\n\n @MainActor\n func startObservation() {\n guard !stoppedForExit else { return }\n guard canTapToAcceptSuggestion || canEscToDismissSuggestion else { return }\n hook.activateIfPossible()\n }\n\n @MainActor\n func stopObservation() {\n hook.deactivate()\n }\n\n func handleEvent(_ event: CGEvent) -> CGEventManipulation.Result {\n let keycode = Int(event.getIntegerValueField(.keyboardEventKeycode))\n let tab = 48\n let esc = 53\n\n switch keycode {\n case tab:\n return handleTab(event.flags)\n case esc:\n return handleEsc(event.flags)\n default:\n return .unchanged\n }\n }\n\n func handleTab(_ flags: CGEventFlags) -> CGEventManipulation.Result {\n Logger.service.info(\"TabToAcceptSuggestion: Tab\")\n\n guard let fileURL = ThreadSafeAccessToXcodeInspector.shared.activeDocumentURL\n else {\n Logger.service.info(\"TabToAcceptSuggestion: No active document\")\n return .unchanged\n }\n\n let language = languageIdentifierFromFileURL(fileURL)\n\n if flags.contains(.maskHelp) { return .unchanged }\n\n let requiredFlagsToTrigger: CGEventFlags = {\n var all = CGEventFlags()\n if UserDefaults.shared.value(for: \\.acceptSuggestionWithModifierShift) {\n all.insert(.maskShift)\n }\n if UserDefaults.shared.value(for: \\.acceptSuggestionWithModifierControl) {\n all.insert(.maskControl)\n }\n if UserDefaults.shared.value(for: \\.acceptSuggestionWithModifierOption) {\n all.insert(.maskAlternate)\n }\n if UserDefaults.shared.value(for: \\.acceptSuggestionWithModifierCommand) {\n all.insert(.maskCommand)\n }\n if UserDefaults.shared.value(for: \\.acceptSuggestionWithModifierOnlyForSwift) {\n if language == .builtIn(.swift) {\n return all\n } else {\n return []\n }\n } else {\n return all\n }\n }()\n\n let flagsToAvoidWhenNotRequired: [CGEventFlags] = [\n .maskShift, .maskCommand, .maskHelp, .maskSecondaryFn,\n ]\n\n guard flags.contains(requiredFlagsToTrigger) else {\n Logger.service.info(\"TabToAcceptSuggestion: Modifier not found\")\n return .unchanged\n }\n\n for flag in flagsToAvoidWhenNotRequired {\n if flags.contains(flag), !requiredFlagsToTrigger.contains(flag) {\n return .unchanged\n }\n }\n\n guard canTapToAcceptSuggestion else {\n Logger.service.info(\"TabToAcceptSuggestion: Feature not available\")\n return .unchanged\n }\n\n guard ThreadSafeAccessToXcodeInspector.shared.activeXcode != nil\n else {\n Logger.service.info(\"TabToAcceptSuggestion: Xcode not found\")\n return .unchanged\n }\n guard let editor = ThreadSafeAccessToXcodeInspector.shared.focusedEditor\n else {\n Logger.service.info(\"TabToAcceptSuggestion: No editor found\")\n return .unchanged\n }\n guard let filespace = workspacePool.fetchFilespaceIfExisted(fileURL: fileURL)\n else {\n Logger.service.info(\"TabToAcceptSuggestion: No file found\")\n return .unchanged\n }\n guard let presentingSuggestion = filespace.presentingSuggestion\n else {\n Logger.service.info(\n \"TabToAcceptSuggestion: No presenting found for \\(filespace.fileURL.lastPathComponent), found \\(filespace.suggestions.count) suggestion, index \\(filespace.suggestionIndex).\"\n )\n return .unchanged\n }\n\n let editorContent = editor.getContent()\n\n let shouldAcceptSuggestion = Self.checkIfAcceptSuggestion(\n lines: editorContent.lines,\n cursorPosition: editorContent.cursorPosition,\n codeMetadata: filespace.codeMetadata,\n presentingSuggestionText: presentingSuggestion.text\n )\n\n if shouldAcceptSuggestion {\n Logger.service.info(\"TabToAcceptSuggestion: Accept\")\n if flags.contains(.maskControl),\n !requiredFlagsToTrigger.contains(.maskControl)\n {\n Task { await commandHandler.acceptActiveSuggestionLineInGroup(atIndex: nil)\n }\n } else {\n Task { await commandHandler.acceptSuggestion() }\n }\n return .discarded\n } else {\n Logger.service.info(\"TabToAcceptSuggestion: Should not accept\")\n return .unchanged\n }\n }\n\n func handleEsc(_ flags: CGEventFlags) -> CGEventManipulation.Result {\n guard\n !flags.contains(.maskShift),\n !flags.contains(.maskControl),\n !flags.contains(.maskAlternate),\n !flags.contains(.maskCommand),\n !flags.contains(.maskHelp),\n canEscToDismissSuggestion\n else { return .unchanged }\n\n guard\n let fileURL = ThreadSafeAccessToXcodeInspector.shared.activeDocumentURL,\n ThreadSafeAccessToXcodeInspector.shared.activeXcode != nil,\n let filespace = workspacePool.fetchFilespaceIfExisted(fileURL: fileURL),\n filespace.presentingSuggestion != nil\n else { return .unchanged }\n\n Task { await commandHandler.dismissSuggestion() }\n return .discarded\n }\n}\n\nextension TabToAcceptSuggestion {\n static func checkIfAcceptSuggestion(\n lines: [String],\n cursorPosition: CursorPosition,\n codeMetadata: FilespaceCodeMetadata,\n presentingSuggestionText: String\n ) -> Bool {\n let line = cursorPosition.line\n guard line >= 0, line < lines.endIndex else {\n return true\n }\n let col = cursorPosition.character\n let prefixEndIndex = lines[line].utf16.index(\n lines[line].utf16.startIndex,\n offsetBy: col,\n limitedBy: lines[line].utf16.endIndex\n ) ?? lines[line].utf16.endIndex\n let prefix = String(lines[line][..\n \n \n \n Label\n \\(serviceIdentifier)\n Program\n \\(executableURL.path)\n MachServices\n \n \\(serviceIdentifier)\n \n \n AssociatedBundleIdentifiers\n \n \\(bundleIdentifier)\n \\(serviceIdentifier)\n \n \n \n \"\"\"\n if !FileManager.default.fileExists(atPath: launchAgentDirURL.path) {\n try FileManager.default.createDirectory(\n at: launchAgentDirURL,\n withIntermediateDirectories: false\n )\n }\n FileManager.default.createFile(\n atPath: launchAgentPath,\n contents: content.data(using: .utf8)\n )\n #if DEBUG\n #else\n try await launchctl(\"load\", launchAgentPath)\n #endif\n }\n}\n\nprivate func process(_ launchPath: String, _ args: [String]) async throws {\n let task = Process()\n task.launchPath = launchPath\n task.arguments = args\n task.environment = [\n \"PATH\": \"/usr/bin\",\n ]\n let outpipe = Pipe()\n task.standardOutput = outpipe\n\n return try await withUnsafeThrowingContinuation { continuation in\n do {\n task.terminationHandler = { process in\n do {\n if process.terminationStatus == 0 {\n continuation.resume(returning: ())\n } else {\n if let data = try? outpipe.fileHandleForReading.readToEnd(),\n let content = String(data: data, encoding: .utf8)\n {\n continuation.resume(throwing: E(errorDescription: content))\n } else {\n continuation.resume(\n throwing: E(\n errorDescription: \"Unknown error.\"\n )\n )\n }\n }\n }\n }\n try task.run()\n } catch {\n continuation.resume(throwing: error)\n }\n }\n}\n\nprivate func helper(_ args: String...) async throws {\n // TODO: A more robust way to locate the executable.\n guard let url = Bundle.main.executableURL?\n .deletingLastPathComponent()\n .deletingLastPathComponent()\n .appendingPathComponent(\"Applications\")\n .appendingPathComponent(\"Helper\")\n else { throw E(errorDescription: \"Unable to locate Helper.\") }\n return try await process(url.path, args)\n}\n\nprivate func launchctl(_ args: String...) async throws {\n return try await process(\"/bin/launchctl\", args)\n}\n\nprivate struct E: Error, LocalizedError {\n var errorDescription: String?\n}\n\n" }, { "path": "Core/Sources/LegacyChatPlugin/AskChatGPT.swift", "content": "import Foundation\nimport OpenAIService\n\n/// Quickly ask a question to ChatGPT.\npublic func askChatGPT(\n systemPrompt: String,\n question: String,\n temperature: Double? = nil\n) async throws -> String? {\n let configuration = UserPreferenceChatGPTConfiguration()\n .overriding(.init(temperature: temperature))\n let memory = AutoManagedChatGPTMemory(\n systemPrompt: systemPrompt,\n configuration: configuration,\n functionProvider: NoChatGPTFunctionProvider(), \n maxNumberOfMessages: .max\n )\n let service = LegacyChatGPTService(\n memory: memory,\n configuration: configuration\n )\n return try await service.sendAndWait(content: question)\n}\n\n" }, { "path": "Core/Sources/LegacyChatPlugin/CallAIFunction.swift", "content": "import Foundation\nimport OpenAIService\n\n/// This is a magic function that can do anything with no-code. See\n/// https://github.com/Torantulino/AI-Functions for more info.\nfunc callAIFunction(\n function: String,\n args: [Any?],\n description: String\n) async throws -> String? {\n let args = args.map { arg -> String in\n if let arg = arg {\n return String(describing: arg)\n } else {\n return \"None\"\n }\n }\n let argsString = args.joined(separator: \", \")\n let configuration = UserPreferenceChatGPTConfiguration()\n .overriding(.init(temperature: 0))\n let service = LegacyChatGPTService(\n memory: AutoManagedChatGPTMemory(\n systemPrompt: \"You are now the following python function: ```# \\(description)\\n\\(function)```\\n\\nOnly respond with your `return` value.\",\n configuration: configuration,\n functionProvider: NoChatGPTFunctionProvider(),\n maxNumberOfMessages: .max\n ),\n configuration: configuration\n )\n return try await service.sendAndWait(content: argsString)\n}\n\n" }, { "path": "Core/Sources/LegacyChatPlugin/LegacyChatPlugin.swift", "content": "import Foundation\nimport OpenAIService\n\npublic protocol LegacyChatPlugin: AnyObject {\n /// Should be [a-zA-Z0-9]+\n static var command: String { get }\n var name: String { get }\n\n init(inside chatGPTService: any LegacyChatGPTServiceType, delegate: LegacyChatPluginDelegate)\n func send(content: String, originalMessage: String) async\n func cancel() async\n func stopResponding() async\n}\n\npublic protocol LegacyChatPluginDelegate: AnyObject {\n func pluginDidStart(_ plugin: LegacyChatPlugin)\n func pluginDidEnd(_ plugin: LegacyChatPlugin)\n func pluginDidStartResponding(_ plugin: LegacyChatPlugin)\n func pluginDidEndResponding(_ plugin: LegacyChatPlugin)\n func shouldStartAnotherPlugin(_ type: LegacyChatPlugin.Type, withContent: String)\n}\n" }, { "path": "Core/Sources/LegacyChatPlugin/TerminalChatPlugin.swift", "content": "import Foundation\nimport OpenAIService\nimport Terminal\nimport XcodeInspector\n\npublic actor TerminalChatPlugin: LegacyChatPlugin {\n public static var command: String { \"run\" }\n public nonisolated var name: String { \"Terminal\" }\n\n let chatGPTService: any LegacyChatGPTServiceType\n var terminal: TerminalType = Terminal()\n var isCancelled = false\n weak var delegate: LegacyChatPluginDelegate?\n\n public init(inside chatGPTService: any LegacyChatGPTServiceType, delegate: LegacyChatPluginDelegate) {\n self.chatGPTService = chatGPTService\n self.delegate = delegate\n }\n\n public func send(content: String, originalMessage: String) async {\n delegate?.pluginDidStart(self)\n delegate?.pluginDidStartResponding(self)\n\n let id = \"\\(Self.command)-\\(UUID().uuidString)\"\n var message = ChatMessage(id: id, role: .assistant, content: \"\")\n var outputContent = \"\" {\n didSet {\n message.content = \"\"\"\n ```\n \\(outputContent)\n ```\n \"\"\"\n }\n }\n\n do {\n let fileURL = XcodeInspector.shared.realtimeActiveDocumentURL\n let projectURL = XcodeInspector.shared.realtimeActiveProjectURL\n \n var environment = [String: String]()\n if let fileURL {\n environment[\"FILE_PATH\"] = fileURL.path\n }\n if let projectURL {\n environment[\"PROJECT_ROOT\"] = projectURL.path\n }\n\n await chatGPTService.memory.mutateHistory { history in\n history.append(\n .init(\n role: .user,\n content: originalMessage\n )\n )\n }\n\n if isCancelled { throw CancellationError() }\n\n let env = ProcessInfo.processInfo.environment\n let shell = env[\"SHELL\"] ?? \"/bin/bash\"\n\n let output = terminal.streamCommand(\n shell,\n arguments: [\"-i\", \"-l\", \"-c\", content],\n currentDirectoryURL: projectURL,\n environment: environment\n )\n\n for try await content in output {\n if isCancelled { throw CancellationError() }\n await chatGPTService.memory.mutateHistory { history in\n if history.last?.id == id {\n history.removeLast()\n }\n outputContent += content\n history.append(message)\n }\n }\n outputContent += \"\\n[finished]\"\n await chatGPTService.memory.mutateHistory { history in\n if history.last?.id == id {\n history.removeLast()\n }\n history.append(message)\n }\n } catch let error as Terminal.TerminationError {\n outputContent += \"\\n[error: \\(error.status)]\"\n await chatGPTService.memory.mutateHistory { history in\n if history.last?.id == id {\n history.removeLast()\n }\n history.append(message)\n }\n } catch {\n outputContent += \"\\n[error: \\(error.localizedDescription)]\"\n await chatGPTService.memory.mutateHistory { history in\n if history.last?.id == id {\n history.removeLast()\n }\n history.append(message)\n }\n }\n\n delegate?.pluginDidEndResponding(self)\n delegate?.pluginDidEnd(self)\n }\n\n public func cancel() async {\n isCancelled = true\n await terminal.terminate()\n }\n\n public func stopResponding() async {\n isCancelled = true\n await terminal.terminate()\n }\n}\n\n" }, { "path": "Core/Sources/LegacyChatPlugin/Translate.swift", "content": "import Foundation\nimport Preferences\n\n@MainActor\nvar translationCache = [String: String]()\n\npublic func translate(text: String, cache: Bool = true) async -> String {\n let language = UserDefaults.shared.value(for: \\.chatGPTLanguage)\n if language.isEmpty { return text }\n\n let key = \"\\(language)-\\(text)\"\n if cache, let cached = await translationCache[key] {\n return cached\n }\n\n if let translated = try? await askChatGPT(\n systemPrompt: \"\"\"\n You are a translator. Your job is to translate the message into \\(language). The reply should only contain the translated content.\n User: ###${{some text}}###\n Assistant: ${{translated text}}\n \"\"\",\n question: \"###\\(text)###\"\n ) {\n if cache {\n let storeTask = Task { @MainActor in\n translationCache[key] = translated\n }\n _ = await storeTask.result\n }\n return translated\n }\n return text\n}\n\n" }, { "path": "Core/Sources/PlusFeatureFlag/PlusFeatureFlag.swift", "content": "import Foundation\nimport SwiftUI\n\n#if canImport(LicenseManagement)\n\nimport LicenseManagement\n\n#else\n\npublic typealias PlusFeatureFlag = Int\n\n@dynamicMemberLookup\npublic struct PlusFeatureFlags {\n public subscript(dynamicMember dynamicMember: String) -> PlusFeatureFlag { return 0 }\n init() {}\n}\n\n#endif\n\npublic func withFeatureEnabled(\n _ flag: KeyPath,\n then: () throws -> Void\n) rethrows {\n #if canImport(LicenseManagement)\n try LicenseManagement.withFeatureEnabled(flag, then: then)\n #endif\n}\n\npublic func withFeatureEnabled(\n _ flag: KeyPath,\n then: () async throws -> Void\n) async rethrows {\n #if canImport(LicenseManagement)\n try await LicenseManagement.withFeatureEnabled(flag, then: then)\n #endif\n}\n\npublic func isFeatureAvailable(_ flag: KeyPath) -> Bool {\n #if canImport(LicenseManagement)\n return LicenseManagement.isFeatureAvailable(flag)\n #else\n return false\n #endif\n}\n\n" }, { "path": "Core/Sources/PromptToCodeService/OpenAIPromptToCodeService.swift", "content": "import Foundation\nimport ModificationBasic\nimport OpenAIService\nimport Preferences\nimport SuggestionBasic\nimport XcodeInspector\n\npublic final class SimpleModificationAgent: ModificationAgent {\n public func send(_ request: Request) -> AsyncThrowingStream {\n AsyncThrowingStream { continuation in\n let task = Task {\n do {\n let stream = try await modifyCode(\n code: request.code,\n requirement: request.requirement,\n source: .init(\n language: request.source.language,\n documentURL: request.source.documentURL,\n projectRootURL: request.source.projectRootURL,\n content: request.source.content,\n lines: request.source.lines,\n range: request.range\n ),\n isDetached: request.isDetached,\n extraSystemPrompt: request.extraSystemPrompt,\n generateDescriptionRequirement: false\n )\n\n for try await response in stream {\n continuation.yield(response)\n }\n\n continuation.finish()\n } catch {\n continuation.finish(throwing: error)\n }\n }\n\n continuation.onTermination = { _ in\n task.cancel()\n }\n }\n }\n\n public init() {}\n\n func modifyCode(\n code: String,\n requirement: String,\n source: PromptToCodeSource,\n isDetached: Bool,\n extraSystemPrompt: String?,\n generateDescriptionRequirement: Bool?\n ) async throws -> AsyncThrowingStream {\n let userPreferredLanguage = UserDefaults.shared.value(for: \\.chatGPTLanguage)\n let textLanguage = {\n if !UserDefaults.shared\n .value(for: \\.promptToCodeGenerateDescriptionInUserPreferredLanguage)\n {\n return \"\"\n }\n return userPreferredLanguage.isEmpty ? \"\" : \" in \\(userPreferredLanguage)\"\n }()\n\n let editor: EditorInformation = await XcodeInspector.shared.getFocusedEditorContent()\n ?? .init(\n editorContent: .init(\n content: source.content,\n lines: source.lines,\n selections: [source.range],\n cursorPosition: .outOfScope,\n cursorOffset: -1,\n lineAnnotations: []\n ),\n selectedContent: code,\n selectedLines: [],\n documentURL: source.documentURL,\n workspaceURL: source.projectRootURL,\n projectRootURL: source.projectRootURL,\n relativePath: \"\",\n language: source.language\n )\n\n let rule: String = {\n func generateDescription(index: Int) -> String {\n let generateDescription = generateDescriptionRequirement ?? UserDefaults.shared\n .value(for: \\.promptToCodeGenerateDescription)\n return generateDescription\n ? \"\"\"\n \\(index). After the code block, write a clear and concise description \\\n in 1-3 sentences about what you did in step 1\\(textLanguage).\n \\(index + 1). Reply with the result.\n \"\"\"\n : \"\\(index). Reply with the result.\"\n }\n switch editor.language {\n case .builtIn(.markdown), .plaintext:\n if code.isEmpty {\n return \"\"\"\n 1. Write the content that meets my requirements.\n 2. Embed the new content in a markdown code block.\n \\(generateDescription(index: 3))\n \"\"\"\n } else {\n return \"\"\"\n 1. Do what I required.\n 2. Format the updated content to use the original indentation. Especially the first line.\n 3. Embed the updated content in a markdown code block.\n 4. You MUST never translate the content in the code block if it's not requested in the requirements.\n \\(generateDescription(index: 5))\n \"\"\"\n }\n default:\n if code.isEmpty {\n return \"\"\"\n 1. Write the code that meets my requirements.\n 2. Embed the code in a markdown code block.\n \\(generateDescription(index: 3))\n \"\"\"\n } else {\n return \"\"\"\n 1. Do what I required.\n 2. Format the updated code to use the original indentation. Especially the first line.\n 3. Embed the updated code in a markdown code block.\n \\(generateDescription(index: 4))\n \"\"\"\n }\n }\n }()\n\n let systemPrompt = {\n switch editor.language {\n case .builtIn(.markdown), .plaintext:\n if code.isEmpty {\n return \"\"\"\n You are good at writing in \\(editor.language.rawValue).\n The active file is: \\(editor.documentURL.lastPathComponent).\n \\(extraSystemPrompt ?? \"\")\n\n \\(rule)\n \"\"\"\n } else {\n return \"\"\"\n You are good at writing in \\(editor.language.rawValue).\n The active file is: \\(editor.documentURL.lastPathComponent).\n \\(extraSystemPrompt ?? \"\")\n\n \\(rule)\n \"\"\"\n }\n default:\n if code.isEmpty {\n return \"\"\"\n You are a senior programer in writing in \\(editor.language.rawValue).\n The active file is: \\(editor.documentURL.lastPathComponent).\n \\(extraSystemPrompt ?? \"\")\n\n \\(rule)\n \"\"\"\n } else {\n return \"\"\"\n You are a senior programer in writing in \\(editor.language.rawValue).\n The active file is: \\(editor.documentURL.lastPathComponent).\n \\(extraSystemPrompt ?? \"\")\n\n \\(rule)\n \"\"\"\n }\n }\n }()\n\n let annotations = isDetached\n ? \"\"\n : extractAnnotations(editorInformation: editor, source: source)\n\n let firstMessage: String? = {\n if code.isEmpty { return nil }\n switch editor.language {\n case .builtIn(.markdown), .plaintext:\n return \"\"\"\n ```\n \\(code)\n ```\n\n \\(annotations)\n \"\"\"\n default:\n return \"\"\"\n ```\n \\(code)\n ```\n\n \\(annotations)\n \"\"\"\n }\n }()\n\n let indentation = getCommonLeadingSpaceCount(code)\n\n let secondMessage = \"\"\"\n I will update the code you just provided.\n Every line has an indentation of \\(indentation) spaces, I will keep that.\n\n What is your requirement?\n \"\"\"\n\n let configuration =\n UserPreferenceChatGPTConfiguration(chatModelKey: \\.promptToCodeChatModelId)\n .overriding(.init(temperature: 0))\n\n let memory = AutoManagedChatGPTMemory(\n systemPrompt: systemPrompt,\n configuration: configuration,\n functionProvider: NoChatGPTFunctionProvider(),\n maxNumberOfMessages: .max\n )\n let chatGPTService = ChatGPTService(\n configuration: configuration,\n functionProvider: NoChatGPTFunctionProvider()\n )\n\n if let firstMessage {\n await memory.mutateHistory { history in\n history.append(.init(role: .user, content: firstMessage))\n history.append(.init(role: .assistant, content: secondMessage))\n history.append(.init(role: .user, content: requirement))\n }\n }\n let stream = chatGPTService.send(memory)\n\n return .init { continuation in\n let task = Task {\n let parser = ExplanationThenCodeStreamParser()\n do {\n func yield(fragments: [ExplanationThenCodeStreamParser.Fragment]) {\n for fragment in fragments {\n switch fragment {\n case let .code(code):\n continuation.yield(.code(code))\n case let .explanation(explanation):\n continuation.yield(.explanation(explanation))\n }\n }\n }\n\n for try await response in stream {\n guard case let .partialText(fragment) = response else { continue }\n try Task.checkCancellation()\n await yield(fragments: parser.yield(fragment))\n }\n await yield(fragments: parser.finish())\n continuation.finish()\n } catch {\n continuation.finish(throwing: error)\n }\n }\n\n continuation.onTermination = { _ in\n task.cancel()\n }\n }\n }\n}\n\n// MAKR: - Internal\n\nextension SimpleModificationAgent {\n func extractCodeAndDescription(from content: String)\n -> (code: String, description: String)\n {\n func extractCodeFromMarkdown(\n _ markdown: String\n ) -> (code: String, endIndex: String.Index)? {\n let codeBlockRegex = try! NSRegularExpression(\n pattern: #\"```(?:\\w+)?\\R([\\s\\S]+?)\\R```\"#,\n options: .dotMatchesLineSeparators\n )\n let range = NSRange(markdown.startIndex.. String {\n guard startIndex < markdown.endIndex else { return \"\" }\n let range = startIndex.. Int {\n let lines = code.split(whereSeparator: \\.isNewline)\n guard !lines.isEmpty else { return 0 }\n var commonCount = Int.max\n for line in lines {\n let count = line.prefix(while: { $0 == \" \" }).count\n commonCount = min(commonCount, count)\n if commonCount == 0 { break }\n }\n return commonCount\n }\n\n func extractAnnotations(\n editorInformation: EditorInformation,\n source: PromptToCodeSource\n ) -> String {\n guard let annotations = editorInformation.editorContent?.lineAnnotations\n else { return \"\" }\n let all = annotations\n .lazy\n .filter { annotation in\n annotation.line >= source.range.start.line + 1\n && annotation.line <= source.range.end.line + 1\n }.map { annotation in\n let relativeLine = annotation.line - source.range.start.line\n return \"line \\(relativeLine): \\(annotation.type) \\(annotation.message)\"\n }\n guard !all.isEmpty else { return \"\" }\n return \"\"\"\n line annotations found:\n \\(annotations.map { \"- \\($0)\" }.joined(separator: \"\\n\"))\n \"\"\"\n }\n}\n\n" }, { "path": "Core/Sources/PromptToCodeService/PreviewPromptToCodeService.swift", "content": "import Foundation\nimport ModificationBasic\nimport SuggestionBasic\n\npublic final class PreviewModificationAgent: ModificationAgent {\n public func send(_ request: Request) -> AsyncThrowingStream {\n AsyncThrowingStream { continuation in\n let task = Task {\n do {\n let stream = try await modifyCode(\n code: request.code,\n requirement: request.requirement,\n source: .init(\n language: request.source.language,\n documentURL: request.source.documentURL,\n projectRootURL: request.source.projectRootURL,\n content: request.source.content,\n lines: request.source.lines,\n range: request.range\n ),\n isDetached: request.isDetached,\n extraSystemPrompt: request.extraSystemPrompt,\n generateDescriptionRequirement: false\n )\n\n for try await (code, description) in stream {\n continuation.yield(.code(code))\n }\n\n continuation.finish()\n } catch {\n continuation.finish(throwing: error)\n }\n }\n\n continuation.onTermination = { _ in\n task.cancel()\n }\n }\n }\n\n public init() {}\n\n func modifyCode(\n code: String,\n requirement: String,\n source: PromptToCodeSource,\n isDetached: Bool,\n extraSystemPrompt: String?,\n generateDescriptionRequirement: Bool?\n ) async throws -> AsyncThrowingStream<(code: String, description: String), Error> {\n return AsyncThrowingStream { continuation in\n Task {\n let code = \"\"\"\n struct Cat {\n var name: String\n }\n\n print(\"Hello world!\")\n \"\"\"\n let description = \"I have created a struct `Cat`.\"\n var resultCode = \"\"\n var resultDescription = \"\"\n do {\n for character in code {\n try await Task.sleep(nanoseconds: 50_000_000)\n resultCode.append(character)\n continuation.yield((resultCode, resultDescription))\n }\n for character in description {\n try await Task.sleep(nanoseconds: 50_000_000)\n resultDescription.append(character)\n continuation.yield((resultCode, resultDescription))\n }\n continuation.finish()\n } catch {\n continuation.finish(throwing: error)\n }\n }\n }\n }\n\n public func stopResponding() {}\n}\n\n" }, { "path": "Core/Sources/PromptToCodeService/PromptToCodeServiceType.swift", "content": "import Dependencies\nimport Foundation\nimport SuggestionBasic\n\npublic struct PromptToCodeSource {\n public var language: CodeLanguage\n public var documentURL: URL\n public var projectRootURL: URL\n public var content: String\n public var lines: [String]\n public var range: CursorRange\n\n public init(\n language: CodeLanguage,\n documentURL: URL,\n projectRootURL: URL,\n content: String,\n lines: [String],\n range: CursorRange\n ) {\n self.language = language\n self.documentURL = documentURL\n self.projectRootURL = projectRootURL\n self.content = content\n self.lines = lines\n self.range = range\n }\n}\n" }, { "path": "Core/Sources/Service/DependencyUpdater.swift", "content": "import CodeiumService\nimport Foundation\nimport GitHubCopilotService\nimport Logger\n\nstruct DependencyUpdater {\n init() {}\n\n func update() {\n Task {\n await withTaskGroup(of: Void.self) { taskGroup in\n let gitHubCopilot = GitHubCopilotInstallationManager()\n switch gitHubCopilot.checkInstallation() {\n case .notInstalled: break\n case .installed: break\n case .unsupported: break\n case .outdated:\n taskGroup.addTask {\n do {\n for try await step in gitHubCopilot.installLatestVersion() {\n let state = {\n switch step {\n case .downloading:\n return \"Downloading\"\n case .uninstalling:\n return \"Uninstalling old version\"\n case .decompressing:\n return \"Decompressing\"\n case .done:\n return \"Done\"\n }\n }()\n Logger.service\n .error(\"Update GitHub Copilot language server: \\(state)\")\n }\n } catch {\n Logger.service.error(\n \"Update GitHub Copilot language server: \\(error.localizedDescription)\"\n )\n }\n }\n }\n\n let codeium = CodeiumInstallationManager()\n\n switch await codeium.checkInstallation() {\n case .notInstalled: break\n case .installed: break\n case .unsupported: break\n case .outdated:\n taskGroup.addTask {\n do {\n for try await step in codeium.installLatestVersion() {\n let state = {\n switch step {\n case .downloading:\n return \"Downloading\"\n case .uninstalling:\n return \"Uninstalling old version\"\n case .decompressing:\n return \"Decompressing\"\n case .done:\n return \"Done\"\n }\n }()\n Logger.service.error(\"Update Codeium language server: \\(state)\")\n }\n } catch {\n Logger.service.error(\n \"Update Codeium language server: \\(error.localizedDescription)\"\n )\n }\n }\n }\n }\n }\n }\n}\n\n" }, { "path": "Core/Sources/Service/GUI/ChatTabFactory.swift", "content": "import BuiltinExtension\nimport ChatGPTChatTab\nimport ChatService\nimport ChatTab\nimport Foundation\nimport PromptToCodeService\nimport SuggestionBasic\nimport SuggestionWidget\nimport XcodeInspector\n\nenum ChatTabFactory {\n static func chatTabBuilderCollection() -> [ChatTabBuilderCollection] {\n let chatGPTChatTab = folderIfNeeded(\n ChatGPTChatTab.chatBuilders(),\n title: ChatGPTChatTab.name\n )\n \n let (defaultChatTab, othersChatTabs) = chatTabsFromExtensions()\n \n if let defaultChatTab {\n return [defaultChatTab] + othersChatTabs + [chatGPTChatTab].compactMap(\\.self)\n } else {\n return [chatGPTChatTab].compactMap(\\.self) + othersChatTabs\n }\n }\n\n private static func folderIfNeeded(\n _ builders: [any ChatTabBuilder],\n title: String\n ) -> ChatTabBuilderCollection? {\n if builders.count > 1 {\n return .folder(title: title, kinds: builders.map(ChatTabKind.init))\n }\n if let first = builders.first { return .kind(ChatTabKind(first)) }\n return nil\n }\n\n static func chatTabsFromExtensions()\n -> (default: ChatTabBuilderCollection?, others: [ChatTabBuilderCollection])\n {\n let extensions = BuiltinExtensionManager.shared.extensions\n let chatTabTypes = extensions.flatMap(\\.chatTabTypes)\n var defaultChatTab: ChatTabBuilderCollection?\n var otherChatTabs = [ChatTabBuilderCollection]()\n for chatTabType in chatTabTypes {\n if chatTabType.isDefaultChatTabReplacement {\n defaultChatTab = folderIfNeeded(chatTabType.chatBuilders(), title: chatTabType.name)\n } else if let tab = folderIfNeeded(\n chatTabType.chatBuilders(),\n title: chatTabType.name\n ) {\n otherChatTabs.append(tab)\n }\n }\n return (defaultChatTab, otherChatTabs)\n }\n}\n" }, { "path": "Core/Sources/Service/GUI/GraphicalUserInterfaceController.swift", "content": "import ActiveApplicationMonitor\nimport AppActivator\nimport AppKit\nimport BuiltinExtension\nimport ChatGPTChatTab\nimport ChatTab\nimport ComposableArchitecture\nimport Dependencies\nimport Logger\nimport Preferences\nimport SuggestionBasic\nimport SuggestionWidget\n\n#if canImport(ChatTabPersistent)\nimport ChatTabPersistent\n#endif\n\n@Reducer\nstruct GUI {\n @ObservableState\n struct State {\n var suggestionWidgetState = Widget.State()\n\n var chatTabGroup: SuggestionWidget.ChatPanel.ChatTabGroup {\n get { suggestionWidgetState.chatPanelState.chatTabGroup }\n set { suggestionWidgetState.chatPanelState.chatTabGroup = newValue }\n }\n\n var promptToCodeGroup: PromptToCodeGroup.State {\n get { suggestionWidgetState.panelState.content.promptToCodeGroup }\n set { suggestionWidgetState.panelState.content.promptToCodeGroup = newValue }\n }\n\n #if canImport(ChatTabPersistent)\n var isChatTabRestoreFinished: Bool = false\n var persistentState: ChatTabPersistent.State {\n get {\n .init(\n chatTabInfo: chatTabGroup.tabInfo,\n isRestoreFinished: isChatTabRestoreFinished,\n selectedChatTapId: chatTabGroup.selectedTabId\n )\n }\n set {\n chatTabGroup.tabInfo = newValue.chatTabInfo\n isChatTabRestoreFinished = newValue.isRestoreFinished\n chatTabGroup.selectedTabId = newValue.selectedChatTapId\n }\n }\n #endif\n }\n\n enum Action {\n case start\n case openChatPanel(forceDetach: Bool, activateThisApp: Bool)\n case createAndSwitchToChatGPTChatTabIfNeeded\n case createAndSwitchToChatTabIfNeededMatching(\n check: (any ChatTab) -> Bool,\n kind: ChatTabKind?\n )\n case sendCustomCommandToActiveChat(CustomCommand)\n case toggleWidgetsHotkeyPressed\n\n case suggestionWidget(Widget.Action)\n\n static func promptToCodeGroup(_ action: PromptToCodeGroup.Action) -> Self {\n .suggestionWidget(.panel(.sharedPanel(.promptToCodeGroup(action))))\n }\n\n #if canImport(ChatTabPersistent)\n case persistent(ChatTabPersistent.Action)\n #endif\n }\n\n @Dependency(\\.chatTabPool) var chatTabPool\n @Dependency(\\.activateThisApp) var activateThisApp\n\n public enum Debounce: Hashable {\n case updateChatTabOrder\n }\n\n var body: some ReducerOf {\n CombineReducers {\n Scope(state: \\.suggestionWidgetState, action: \\.suggestionWidget) {\n Widget()\n }\n\n Scope(\n state: \\.chatTabGroup,\n action: \\.suggestionWidget.chatPanel\n ) {\n Reduce { _, action in\n switch action {\n case let .createNewTapButtonClicked(kind):\n return .run { send in\n if let (_, chatTabInfo) = await chatTabPool.createTab(for: kind) {\n await send(.appendAndSelectTab(chatTabInfo))\n }\n }\n\n case let .closeTabButtonClicked(id):\n return .run { _ in\n chatTabPool.removeTab(of: id)\n }\n\n case let .chatTab(.element(_, .openNewTab(builder))):\n return .run { send in\n if let (_, chatTabInfo) = await chatTabPool\n .createTab(from: builder.chatTabBuilder)\n {\n await send(.appendAndSelectTab(chatTabInfo))\n }\n }\n\n default:\n return .none\n }\n }\n }\n\n #if canImport(ChatTabPersistent)\n Scope(state: \\.persistentState, action: \\.persistent) {\n ChatTabPersistent()\n }\n #endif\n\n Reduce { state, action in\n switch action {\n case .start:\n #if canImport(ChatTabPersistent)\n return .run { send in\n await send(.persistent(.restoreChatTabs))\n }\n #else\n return .none\n #endif\n\n case let .openChatPanel(forceDetach, activate):\n return .run { send in\n await send(\n .suggestionWidget(\n .chatPanel(.presentChatPanel(forceDetach: forceDetach))\n )\n )\n await send(.suggestionWidget(.updateKeyWindow(.chatPanel)))\n\n if activate {\n activateThisApp()\n }\n }\n\n case .createAndSwitchToChatGPTChatTabIfNeeded:\n return .run { send in\n await send(.createAndSwitchToChatTabIfNeededMatching(\n check: { $0 is ChatGPTChatTab },\n kind: nil\n ))\n }\n\n case let .createAndSwitchToChatTabIfNeededMatching(check, kind):\n if let selectedTabInfo = state.chatTabGroup.selectedTabInfo,\n let tab = chatTabPool.getTab(of: selectedTabInfo.id),\n check(tab)\n {\n // Already in ChatGPT tab\n return .none\n }\n\n if let firstChatGPTTabInfo = state.chatTabGroup.tabInfo.first(where: {\n if let tab = chatTabPool.getTab(of: $0.id) {\n return check(tab)\n }\n return false\n }) {\n return .run { send in\n await send(.suggestionWidget(.chatPanel(.tabClicked(\n id: firstChatGPTTabInfo.id\n ))))\n }\n }\n return .run { send in\n if let (_, chatTabInfo) = await chatTabPool.createTab(for: kind) {\n await send(\n .suggestionWidget(.chatPanel(.appendAndSelectTab(chatTabInfo)))\n )\n }\n }\n\n case let .sendCustomCommandToActiveChat(command):\n if let info = state.chatTabGroup.selectedTabInfo,\n let tab = chatTabPool.getTab(of: info.id),\n tab.handleCustomCommand(command)\n {\n return .run { send in\n await send(.openChatPanel(forceDetach: false, activateThisApp: false))\n }\n }\n\n for info in state.chatTabGroup.tabInfo {\n if let chatTab = chatTabPool.getTab(of: info.id),\n chatTab.handleCustomCommand(command)\n {\n state.chatTabGroup.selectedTabId = chatTab.id\n return .run { send in\n await send(.openChatPanel(\n forceDetach: false,\n activateThisApp: false\n ))\n }\n }\n }\n\n return .run { send in\n guard let (chatTab, chatTabInfo) = await chatTabPool.createTab(for: nil)\n else { return }\n await send(.suggestionWidget(.chatPanel(.appendAndSelectTab(chatTabInfo))))\n await send(.openChatPanel(forceDetach: false, activateThisApp: false))\n _ = chatTab.handleCustomCommand(command)\n }\n\n case .toggleWidgetsHotkeyPressed:\n return .run { send in\n await send(.suggestionWidget(.circularWidget(.widgetClicked)))\n }\n\n case let .suggestionWidget(.chatPanel(.chatTab(.element(id, .tabContentUpdated)))):\n #if canImport(ChatTabPersistent)\n // when a tab is updated, persist it.\n return .run { send in\n await send(.persistent(.chatTabUpdated(id: id)))\n }\n #else\n return .none\n #endif\n\n case let .suggestionWidget(.chatPanel(.closeTabButtonClicked(id))):\n #if canImport(ChatTabPersistent)\n // when a tab is closed, remove it from persistence.\n return .run { send in\n await send(.persistent(.chatTabClosed(id: id)))\n }\n #else\n return .none\n #endif\n\n case .suggestionWidget:\n return .none\n\n #if canImport(ChatTabPersistent)\n case .persistent:\n return .none\n #endif\n }\n }\n }.onChange(of: \\.chatTabGroup.tabInfo) { old, new in\n Reduce { _, _ in\n guard old.map(\\.id) != new.map(\\.id) else {\n return .none\n }\n #if canImport(ChatTabPersistent)\n return .run { send in\n await send(.persistent(.chatOrderChanged))\n }.debounce(id: Debounce.updateChatTabOrder, for: 1, scheduler: DispatchQueue.main)\n #else\n return .none\n #endif\n }\n }\n }\n}\n\n@MainActor\npublic final class GraphicalUserInterfaceController {\n let store: StoreOf\n let widgetController: SuggestionWidgetController\n let widgetDataSource: WidgetDataSource\n let chatTabPool: ChatTabPool\n\n class WeakStoreHolder {\n weak var store: StoreOf?\n }\n\n init() {\n let chatTabPool = ChatTabPool()\n let suggestionDependency = SuggestionWidgetControllerDependency()\n let setupDependency: (inout DependencyValues) -> Void = { dependencies in\n dependencies.suggestionWidgetControllerDependency = suggestionDependency\n dependencies.suggestionWidgetUserDefaultsObservers = .init()\n dependencies.chatTabPool = chatTabPool\n dependencies.chatTabBuilderCollection = ChatTabFactory.chatTabBuilderCollection\n\n #if canImport(ChatTabPersistent) && canImport(ProChatTabs)\n dependencies.restoreChatTabInPool = {\n await chatTabPool.restore($0)\n }\n #endif\n }\n let store = StoreOf(\n initialState: .init(),\n reducer: { GUI() },\n withDependencies: setupDependency\n )\n self.store = store\n self.chatTabPool = chatTabPool\n widgetDataSource = .init()\n\n widgetController = SuggestionWidgetController(\n store: store.scope(\n state: \\.suggestionWidgetState,\n action: \\.suggestionWidget\n ),\n chatTabPool: chatTabPool,\n dependency: suggestionDependency\n )\n\n chatTabPool.createStore = { id in\n store.scope(\n state: { state in\n state.chatTabGroup.tabInfo[id: id] ?? .init(id: id, title: \"\")\n },\n action: { childAction in\n .suggestionWidget(.chatPanel(.chatTab(.element(\n id: id,\n action: childAction\n ))))\n }\n )\n }\n\n suggestionDependency.suggestionWidgetDataSource = widgetDataSource\n suggestionDependency.onOpenChatClicked = {\n Task {\n PseudoCommandHandler().openChat(forceDetach: false, activateThisApp: true)\n }\n }\n suggestionDependency.onOpenModificationButtonClicked = {\n Task {\n guard let content = await PseudoCommandHandler().getEditorContent(sourceEditor: nil)\n else { return }\n _ = try await WindowBaseCommandHandler().promptToCode(editor: content)\n }\n }\n suggestionDependency.onCustomCommandClicked = { command in\n Task {\n let commandHandler = PseudoCommandHandler()\n await commandHandler.handleCustomCommand(command)\n }\n }\n }\n\n func start() {\n store.send(.start)\n }\n\n public func openGlobalChat() {\n PseudoCommandHandler().openChat(forceDetach: true)\n }\n}\n\nextension ChatTabPool {\n @MainActor\n func createTab(\n id: String = UUID().uuidString,\n from builder: ChatTabBuilder\n ) async -> (any ChatTab, ChatTabInfo)? {\n let id = id\n let info = ChatTabInfo(id: id, title: \"\")\n guard let chatTap = await builder.build(store: createStore(id)) else { return nil }\n setTab(chatTap, forId: id)\n return (chatTap, info)\n }\n\n @MainActor\n func createTab(\n for kind: ChatTabKind?\n ) async -> (any ChatTab, ChatTabInfo)? {\n let id = UUID().uuidString\n let info = ChatTabInfo(id: id, title: \"\")\n let builder = kind?.builder ?? {\n for ext in BuiltinExtensionManager.shared.extensions {\n guard let tab = ext.chatTabTypes.first(where: { $0.isDefaultChatTabReplacement })\n else { continue }\n return tab.defaultChatBuilder()\n }\n return ChatGPTChatTab.defaultBuilder()\n }()\n guard let chatTap = await builder.build(store: createStore(id)) else { return nil }\n setTab(chatTap, forId: id)\n return (chatTap, info)\n }\n\n #if canImport(ChatTabPersistent)\n @MainActor\n func restore(\n _ data: ChatTabPersistent.RestorableTabData\n ) async -> (any ChatTab, ChatTabInfo)? {\n switch data.name {\n case ChatGPTChatTab.name:\n guard let builder = try? await ChatGPTChatTab.restore(from: data.data)\n else { fallthrough }\n return await createTab(id: data.id, from: builder)\n default:\n let chatTabTypes = BuiltinExtensionManager.shared.extensions.flatMap(\\.chatTabTypes)\n for type in chatTabTypes {\n if type.name == data.name {\n do {\n let builder = try await type.restore(from: data.data)\n return await createTab(id: data.id, from: builder)\n } catch {\n Logger.service.error(\"Failed to restore chat tab \\(data.name): \\(error)\")\n break\n }\n }\n }\n }\n\n guard let builder = try? await EmptyChatTab.restore(from: data.data) else { return nil }\n return await createTab(id: data.id, from: builder)\n }\n #endif\n}\n\n" }, { "path": "Core/Sources/Service/GUI/WidgetDataSource.swift", "content": "import ActiveApplicationMonitor\nimport AppActivator\nimport AppKit\nimport ChatService\nimport ComposableArchitecture\nimport Foundation\nimport GitHubCopilotService\nimport OpenAIService\nimport PromptToCodeService\nimport SuggestionBasic\nimport SuggestionWidget\n\n@MainActor\nfinal class WidgetDataSource {}\n\nextension WidgetDataSource: SuggestionWidgetDataSource {\n func suggestionForFile(at url: URL) async -> PresentingCodeSuggestion? {\n for workspace in Service.shared.workspacePool.workspaces.values {\n if let filespace = workspace.filespaces[url],\n let suggestion = filespace.presentingSuggestion\n {\n return .init(\n code: suggestion.text,\n language: filespace.language.rawValue,\n startLineIndex: suggestion.position.line,\n suggestionCount: filespace.suggestions.count,\n currentSuggestionIndex: filespace.suggestionIndex,\n replacingRange: suggestion.range,\n replacingLines: suggestion.replacingLines,\n descriptions: suggestion.descriptions\n )\n }\n }\n return nil\n }\n}\n\n" }, { "path": "Core/Sources/Service/GlobalShortcutManager.swift", "content": "import AppKit\nimport Combine\nimport Foundation\nimport KeyboardShortcuts\nimport XcodeInspector\n\nextension KeyboardShortcuts.Name {\n static let showHideWidget = Self(\"ShowHideWidget\")\n}\n\n@MainActor\nfinal class GlobalShortcutManager {\n let guiController: GraphicalUserInterfaceController\n private var activeAppChangeTask: Task?\n\n nonisolated init(guiController: GraphicalUserInterfaceController) {\n self.guiController = guiController\n }\n\n func start() {\n KeyboardShortcuts.userDefaults = .shared\n setupShortcutIfNeeded()\n\n KeyboardShortcuts.onKeyUp(for: .showHideWidget) { [guiController] in\n let isXCodeActive = XcodeInspector.shared.activeXcode != nil\n\n if !isXCodeActive,\n !guiController.store.state.suggestionWidgetState.chatPanelState.isPanelDisplayed,\n UserDefaults.shared.value(for: \\.showHideWidgetShortcutGlobally)\n {\n guiController.store.send(.openChatPanel(forceDetach: true, activateThisApp: true))\n } else {\n guiController.store.send(.toggleWidgetsHotkeyPressed)\n }\n }\n\n activeAppChangeTask?.cancel()\n activeAppChangeTask = Task.detached { [weak self] in\n let notifications = NotificationCenter.default\n .notifications(named: .activeApplicationDidChange)\n for await _ in notifications {\n guard let self else { return }\n try Task.checkCancellation()\n if !UserDefaults.shared.value(for: \\.showHideWidgetShortcutGlobally) {\n let app = await XcodeInspector.shared.activeApplication\n let shouldBeEnabled = if let app, app.isXcode || app.isExtensionService {\n true\n } else {\n false\n }\n if shouldBeEnabled {\n await self.setupShortcutIfNeeded()\n } else {\n await self.removeShortcutIfNeeded()\n }\n } else {\n await self.setupShortcutIfNeeded()\n }\n }\n }\n }\n\n func setupShortcutIfNeeded() {\n KeyboardShortcuts.enable(.showHideWidget)\n }\n\n func removeShortcutIfNeeded() {\n KeyboardShortcuts.disable(.showHideWidget)\n }\n}\n\n" }, { "path": "Core/Sources/Service/Helpers.swift", "content": "import Foundation\nimport LanguageServerProtocol\n\nextension NSError {\n static func from(_ error: Error) -> NSError {\n if let error = error as? ServerError {\n var message = \"Unknown\"\n switch error {\n case let .handlerUnavailable(handler):\n message = \"Handler unavailable: \\(handler).\"\n case let .unhandledMethod(method):\n message = \"Methond unhandled: \\(method).\"\n case let .notificationDispatchFailed(error):\n message = \"Notification dispatch failed: \\(error.localizedDescription).\"\n case let .requestDispatchFailed(error):\n message = \"Request dispatch failed: \\(error.localizedDescription).\"\n case let .clientDataUnavailable(error):\n message = \"Client data unavailable: \\(error.localizedDescription).\"\n case .serverUnavailable:\n message = \"Server unavailable, please make sure you have installed Node.\"\n case .missingExpectedParameter:\n message = \"Missing expected parameter.\"\n case .missingExpectedResult:\n message = \"Missing expected result.\"\n case let .unableToDecodeRequest(error):\n message = \"Unable to decode request: \\(error.localizedDescription).\"\n case let .unableToSendRequest(error):\n message = \"Unable to send request: \\(error.localizedDescription).\"\n case let .unableToSendNotification(error):\n message = \"Unable to send notification: \\(error.localizedDescription).\"\n case let .serverError(code, m, _):\n message = \"Server error: (\\(code)) \\(m).\"\n case let .invalidRequest(error):\n message = \"Invalid request: \\(error?.localizedDescription ?? \"Unknown\").\"\n case .timeout:\n message = \"Timeout.\"\n }\n return NSError(domain: \"com.intii.CopilotForXcode\", code: -1, userInfo: [\n NSLocalizedDescriptionKey: message,\n ])\n }\n if let error = error as? CancellationError {\n return NSError(domain: \"com.intii.CopilotForXcode\", code: -100, userInfo: [\n NSLocalizedDescriptionKey: error.localizedDescription,\n ])\n }\n return NSError(domain: \"com.intii.CopilotForXcode\", code: -1, userInfo: [\n NSLocalizedDescriptionKey: error.localizedDescription,\n ])\n }\n}\n" }, { "path": "Core/Sources/Service/RealtimeSuggestionController.swift", "content": "import ActiveApplicationMonitor\nimport AppKit\nimport AsyncAlgorithms\nimport AXExtension\nimport Foundation\nimport Logger\nimport Preferences\nimport QuartzCore\nimport Workspace\nimport XcodeInspector\n\npublic actor RealtimeSuggestionController {\n private var xcodeChangeObservationTask: Task?\n private var inflightPrefetchTask: Task?\n private var editorObservationTask: Task?\n private var sourceEditor: SourceEditor?\n\n init() {}\n\n deinit {\n inflightPrefetchTask?.cancel()\n editorObservationTask?.cancel()\n }\n\n nonisolated\n func start() {\n Task { await observeXcodeChange() }\n }\n\n private func observeXcodeChange() {\n xcodeChangeObservationTask?.cancel()\n\n xcodeChangeObservationTask = Task { [weak self] in\n for await _ in NotificationCenter.default\n .notifications(named: .focusedEditorDidChange)\n {\n guard let self else { return }\n try Task.checkCancellation()\n guard let editor = await XcodeInspector.shared.focusedEditor else { continue }\n await self.handleFocusElementChange(editor)\n }\n }\n }\n\n private func handleFocusElementChange(_ sourceEditor: SourceEditor) {\n self.sourceEditor = sourceEditor\n\n let notificationsFromEditor = sourceEditor.axNotifications\n\n editorObservationTask?.cancel()\n editorObservationTask = nil\n\n editorObservationTask = Task { [weak self] in\n if let fileURL = XcodeInspector.shared.realtimeActiveDocumentURL {\n await PseudoCommandHandler().invalidateRealtimeSuggestionsIfNeeded(\n fileURL: fileURL,\n sourceEditor: sourceEditor\n )\n }\n\n let valueChange = await notificationsFromEditor.notifications()\n .filter { $0.kind == .valueChanged }\n let selectedTextChanged = await notificationsFromEditor.notifications()\n .filter { $0.kind == .selectedTextChanged }\n\n await withTaskGroup(of: Void.self) { [weak self] group in\n group.addTask { [weak self] in\n let handler = { [weak self] in\n guard let self else { return }\n await cancelInFlightTasks()\n await self.triggerPrefetchDebounced()\n await self.notifyEditingFileChange(editor: sourceEditor.element)\n }\n\n if #available(macOS 13.0, *) {\n for await _ in valueChange._throttle(for: .milliseconds(200)) {\n if Task.isCancelled { return }\n await handler()\n }\n } else {\n for await _ in valueChange {\n if Task.isCancelled { return }\n await handler()\n }\n }\n }\n group.addTask {\n let handler = {\n guard let fileURL = await XcodeInspector.shared.activeDocumentURL\n else { return }\n await PseudoCommandHandler().invalidateRealtimeSuggestionsIfNeeded(\n fileURL: fileURL,\n sourceEditor: sourceEditor\n )\n }\n\n if #available(macOS 13.0, *) {\n for await _ in selectedTextChanged._throttle(for: .milliseconds(200)) {\n if Task.isCancelled { return }\n await handler()\n }\n } else {\n for await _ in selectedTextChanged {\n if Task.isCancelled { return }\n await handler()\n }\n }\n }\n\n await group.waitForAll()\n }\n }\n\n Task { @WorkspaceActor in // Get cache ready for real-time suggestions.\n guard UserDefaults.shared.value(for: \\.preCacheOnFileOpen) else { return }\n guard await XcodeInspector.shared.activeApplication?.isXcode ?? false else { return }\n guard let fileURL = XcodeInspector.shared.realtimeActiveDocumentURL\n else { return }\n let (_, filespace) = try await Service.shared.workspacePool\n .fetchOrCreateWorkspaceAndFilespace(fileURL: fileURL)\n\n if filespace.codeMetadata.uti == nil {\n Logger.service.info(\"Generate cache for file.\")\n // avoid the command get called twice\n filespace.codeMetadata.uti = \"\"\n do {\n try await XcodeInspector.shared.latestActiveXcode?\n .triggerCopilotCommand(name: \"Prepare for Real-time Suggestions\")\n } catch {\n if filespace.codeMetadata.uti?.isEmpty ?? true {\n filespace.codeMetadata.uti = nil\n }\n }\n }\n }\n }\n\n func triggerPrefetchDebounced(force: Bool = false) {\n inflightPrefetchTask = Task(priority: .utility) { @WorkspaceActor in\n try? await Task.sleep(nanoseconds: UInt64(\n max(UserDefaults.shared.value(for: \\.realtimeSuggestionDebounce), 0.15)\n * 1_000_000_000\n ))\n\n if Task.isCancelled { return }\n\n guard UserDefaults.shared.value(for: \\.realtimeSuggestionToggle)\n else { return }\n\n if UserDefaults.shared.value(for: \\.disableSuggestionFeatureGlobally),\n let fileURL = await XcodeInspector.shared.activeDocumentURL,\n let (workspace, _) = try? await Service.shared.workspacePool\n .fetchOrCreateWorkspaceAndFilespace(fileURL: fileURL)\n {\n let isEnabled = workspace.isSuggestionFeatureEnabled\n if !isEnabled { return }\n }\n if Task.isCancelled { return }\n\n // So the editor won't be blocked (after information are cached)!\n await PseudoCommandHandler().generateRealtimeSuggestions(sourceEditor: sourceEditor)\n }\n }\n\n func cancelInFlightTasks(excluding: Task? = nil) async {\n inflightPrefetchTask?.cancel()\n let workspaces = await Service.shared.workspacePool.workspaces\n // cancel in-flight tasks\n await withTaskGroup(of: Void.self) { group in\n for (_, workspace) in workspaces {\n group.addTask {\n await workspace.cancelInFlightRealtimeSuggestionRequests()\n }\n }\n }\n }\n\n /// This method will still return true if the completion panel is hidden by esc.\n /// Looks like the Xcode will keep the panel around until content is changed,\n /// not sure how to observe that it's hidden.\n func isCompletionPanelPresenting() -> Bool {\n guard let activeXcode = ActiveApplicationMonitor.shared.activeXcode else { return false }\n let application = AXUIElementCreateApplication(activeXcode.processIdentifier)\n return application.focusedWindow?.child(identifier: \"_XC_COMPLETION_TABLE_\") != nil\n }\n\n func notifyEditingFileChange(editor: AXUIElement) async {\n guard let fileURL = await XcodeInspector.shared.activeDocumentURL,\n let (workspace, _) = try? await Service.shared.workspacePool\n .fetchOrCreateWorkspaceAndFilespace(fileURL: fileURL)\n else { return }\n await workspace.didUpdateFilespace(fileURL: fileURL, content: editor.value)\n }\n}\n\n" }, { "path": "Core/Sources/Service/ScheduledCleaner.swift", "content": "import ActiveApplicationMonitor\nimport AppKit\nimport AXExtension\nimport BuiltinExtension\nimport Foundation\nimport Logger\nimport Workspace\nimport XcodeInspector\n\npublic final class ScheduledCleaner {\n weak var service: Service?\n\n init() {}\n\n func start() {\n Task { @ServiceActor in\n while !Task.isCancelled {\n try await Task.sleep(nanoseconds: 10 * 60 * 1_000_000_000)\n await cleanUp()\n }\n }\n\n Task { @ServiceActor in\n for await app in ActiveApplicationMonitor.shared.createInfoStream() {\n try Task.checkCancellation()\n if let app, !app.isXcode {\n await cleanUp()\n }\n }\n }\n }\n\n @ServiceActor\n func cleanUp() async {\n guard let service else { return }\n\n let workspaceInfos = await XcodeInspector.shared.xcodes.reduce(\n into: [\n XcodeAppInstanceInspector.WorkspaceIdentifier:\n XcodeAppInstanceInspector.WorkspaceInfo\n ]()\n ) { result, xcode in\n let infos = xcode.realtimeWorkspaces\n for (id, info) in infos {\n if let existed = result[id] {\n result[id] = existed.combined(with: info)\n } else {\n result[id] = info\n }\n }\n }\n for (url, workspace) in service.workspacePool.workspaces {\n if workspace.isExpired, workspaceInfos[.url(url)] == nil {\n Logger.service.info(\"Remove idle workspace\")\n _ = await Task { @MainActor in\n service.guiController.store.send(\n .promptToCodeGroup(.discardExpiredPromptToCode(documentURLs: Array(\n workspace.filespaces.keys\n )))\n )\n }.result\n await workspace.cleanUp(availableTabs: [])\n await service.workspacePool.removeWorkspace(url: url)\n } else {\n let tabs = (workspaceInfos[.url(url)]?.tabs ?? [])\n .union(workspaceInfos[.unknown]?.tabs ?? [])\n // cleanup chats for unused files\n let filespaces = workspace.filespaces\n for (url, _) in filespaces {\n if workspace.isFilespaceExpired(\n fileURL: url,\n availableTabs: tabs\n ) {\n Logger.service.info(\"Remove idle filespace\")\n _ = await Task { @MainActor in\n service.guiController.store.send(\n .promptToCodeGroup(.discardExpiredPromptToCode(documentURLs: [url]))\n )\n }.result\n }\n }\n // cleanup workspace\n await workspace.cleanUp(availableTabs: tabs)\n }\n }\n\n #if canImport(ProService)\n await service.proService.cleanUp(workspaceInfos: workspaceInfos)\n #endif\n }\n\n @ServiceActor\n public func closeAllChildProcesses() async {\n BuiltinExtensionManager.shared.terminate()\n }\n}\n\n" }, { "path": "Core/Sources/Service/Service.swift", "content": "import BuiltinExtension\nimport CodeiumService\nimport Combine\nimport CommandHandler\nimport Dependencies\nimport Foundation\nimport GitHubCopilotService\nimport KeyBindingManager\nimport Logger\nimport OverlayWindow\nimport SuggestionService\nimport Toast\nimport Workspace\nimport WorkspaceSuggestionService\nimport XcodeInspector\nimport XcodeThemeController\nimport XPCShared\n#if canImport(ProService)\nimport ProService\n#endif\n\n@globalActor public enum ServiceActor {\n public actor TheActor {}\n public static let shared = TheActor()\n}\n\n/// The running extension service.\npublic final class Service {\n @MainActor\n public static let shared = Service()\n\n @Dependency(\\.workspacePool) var workspacePool\n @MainActor\n public let guiController: GraphicalUserInterfaceController\n public let commandHandler: CommandHandler\n public let realtimeSuggestionController: RealtimeSuggestionController\n public let scheduledCleaner: ScheduledCleaner\n let globalShortcutManager: GlobalShortcutManager\n let keyBindingManager: KeyBindingManager\n let xcodeThemeController: XcodeThemeController = .init()\n let overlayWindowController: OverlayWindowController\n\n #if canImport(ProService)\n let proService: ProService\n #endif\n\n @Dependency(\\.toast) var toast\n var cancellable = Set()\n\n @MainActor\n private init() {\n @Dependency(\\.workspacePool) var workspacePool\n let commandHandler = PseudoCommandHandler()\n UniversalCommandHandler.shared.commandHandler = commandHandler\n self.commandHandler = commandHandler\n\n realtimeSuggestionController = .init()\n scheduledCleaner = .init()\n overlayWindowController = .init()\n\n #if canImport(ProService)\n proService = ProService()\n #endif\n\n BuiltinExtensionManager.shared.addExtensions([\n GitHubCopilotExtension(workspacePool: workspacePool),\n CodeiumExtension(workspacePool: workspacePool),\n ])\n\n let guiController = GraphicalUserInterfaceController()\n self.guiController = guiController\n globalShortcutManager = .init(guiController: guiController)\n keyBindingManager = .init()\n\n workspacePool.registerPlugin {\n SuggestionServiceWorkspacePlugin(workspace: $0) { SuggestionService.service() }\n }\n workspacePool.registerPlugin {\n GitHubCopilotWorkspacePlugin(workspace: $0)\n }\n workspacePool.registerPlugin {\n CodeiumWorkspacePlugin(workspace: $0)\n }\n workspacePool.registerPlugin {\n BuiltinExtensionWorkspacePlugin(workspace: $0)\n }\n\n scheduledCleaner.service = self\n }\n\n @MainActor\n public func start() {\n scheduledCleaner.start()\n realtimeSuggestionController.start()\n guiController.start()\n xcodeThemeController.start()\n #if canImport(ProService)\n proService.start()\n #endif\n overlayWindowController.start()\n DependencyUpdater().update()\n globalShortcutManager.start()\n keyBindingManager.start()\n\n Task.detached { [weak self] in\n let notifications = NotificationCenter.default\n .notifications(named: .activeDocumentURLDidChange)\n var previousURL: URL?\n for await _ in notifications {\n guard self != nil else { return }\n let url = await XcodeInspector.shared.activeDocumentURL\n if let url, url != previousURL, url != .init(fileURLWithPath: \"/\") {\n previousURL = url\n @Dependency(\\.workspacePool) var workspacePool\n _ = try await workspacePool\n .fetchOrCreateWorkspaceAndFilespace(fileURL: url)\n }\n }\n }\n }\n\n @MainActor\n public func prepareForExit() async {\n Logger.service.info(\"Prepare for exit.\")\n keyBindingManager.stopForExit()\n #if canImport(ProService)\n proService.prepareForExit()\n #endif\n await scheduledCleaner.closeAllChildProcesses()\n }\n}\n\npublic extension Service {\n func handleXPCServiceRequests(\n endpoint: String,\n requestBody: Data,\n reply: @escaping (Data?, Error?) -> Void\n ) {\n do {\n #if canImport(ProService)\n try proService.handleXPCServiceRequests(\n endpoint: endpoint,\n requestBody: requestBody,\n reply: reply\n )\n #endif\n\n try ExtensionServiceRequests.GetExtensionOpenChatHandlers.handle(\n endpoint: endpoint,\n requestBody: requestBody,\n reply: reply\n ) { _ in\n BuiltinExtensionManager.shared.extensions.reduce(into: []) { result, ext in\n let tabs = ext.chatTabTypes\n for tab in tabs {\n if tab.canHandleOpenChatCommand {\n result.append(.init(\n bundleIdentifier: ext.extensionIdentifier,\n id: tab.name,\n tabName: tab.name,\n isBuiltIn: true\n ))\n }\n }\n }\n }\n\n try ExtensionServiceRequests.GetSuggestionLineAcceptedCode.handle(\n endpoint: endpoint,\n requestBody: requestBody,\n reply: reply\n ) { request in\n let editor = request.editorContent\n let handler = WindowBaseCommandHandler()\n let updatedContent = try? await handler\n .acceptSuggestionLine(editor: editor)\n return updatedContent\n }\n } catch is XPCRequestHandlerHitError {\n return\n } catch {\n reply(nil, error)\n return\n }\n\n reply(nil, XPCRequestNotHandledError())\n }\n}\n\n" }, { "path": "Core/Sources/Service/SuggestionCommandHandler/PseudoCommandHandler.swift", "content": "import ActiveApplicationMonitor\nimport AppKit\nimport BuiltinExtension\nimport CodeiumService\nimport CommandHandler\nimport ComposableArchitecture\nimport enum CopilotForXcodeKit.SuggestionServiceError\nimport Dependencies\nimport Logger\nimport ModificationBasic\nimport PlusFeatureFlag\nimport Preferences\nimport PromptToCodeCustomization\nimport SuggestionBasic\nimport SuggestionInjector\nimport Terminal\nimport Toast\nimport Workspace\nimport WorkspaceSuggestionService\nimport XcodeInspector\nimport XPCShared\n\n#if canImport(BrowserChatTab)\nimport BrowserChatTab\n#endif\n\n/// It's used to run some commands without really triggering the menu bar item.\n///\n/// For example, we can use it to generate real-time suggestions without Apple Scripts.\nstruct PseudoCommandHandler: CommandHandler {\n static var lastTimeCommandFailedToTriggerWithAccessibilityAPI = Date(timeIntervalSince1970: 0)\n private var toast: ToastController { ToastControllerDependencyKey.liveValue }\n\n func presentPreviousSuggestion() async {\n let handler = WindowBaseCommandHandler()\n _ = try? await handler.presentPreviousSuggestion(editor: .init(\n content: \"\",\n lines: [],\n uti: \"\",\n cursorPosition: .outOfScope,\n cursorOffset: -1,\n selections: [],\n tabSize: 0,\n indentSize: 0,\n usesTabsForIndentation: false\n ))\n }\n\n func presentNextSuggestion() async {\n let handler = WindowBaseCommandHandler()\n _ = try? await handler.presentNextSuggestion(editor: .init(\n content: \"\",\n lines: [],\n uti: \"\",\n cursorPosition: .outOfScope,\n cursorOffset: -1,\n selections: [],\n tabSize: 0,\n indentSize: 0,\n usesTabsForIndentation: false\n ))\n }\n\n @WorkspaceActor\n func generateRealtimeSuggestions(sourceEditor: SourceEditor?) async {\n guard let filespace = await getFilespace(),\n let (workspace, _) = try? await Service.shared.workspacePool\n .fetchOrCreateWorkspaceAndFilespace(fileURL: filespace.fileURL) else { return }\n\n if Task.isCancelled { return }\n\n // Can't use handler if content is not available.\n guard let editor = await getEditorContent(sourceEditor: sourceEditor)\n else { return }\n\n let fileURL = filespace.fileURL\n let presenter = PresentInWindowSuggestionPresenter()\n\n presenter.markAsProcessing(true)\n defer { presenter.markAsProcessing(false) }\n\n if filespace.presentingSuggestion != nil {\n // Check if the current suggestion is still valid.\n if filespace.validateSuggestions(\n lines: editor.lines,\n cursorPosition: editor.cursorPosition\n ) {\n return\n } else {\n presenter.discardSuggestion(fileURL: filespace.fileURL)\n }\n }\n\n let snapshot = FilespaceSuggestionSnapshot(\n lines: editor.lines,\n cursorPosition: editor.cursorPosition\n )\n\n guard filespace.suggestionSourceSnapshot != snapshot else { return }\n\n do {\n try await workspace.generateSuggestions(\n forFileAt: fileURL,\n editor: editor\n )\n if let sourceEditor {\n let editorContent = sourceEditor.getContent()\n _ = filespace.validateSuggestions(\n lines: editorContent.lines,\n cursorPosition: editorContent.cursorPosition\n )\n }\n if filespace.presentingSuggestion != nil {\n presenter.presentSuggestion(fileURL: fileURL)\n } else {\n presenter.discardSuggestion(fileURL: fileURL)\n }\n } catch let error as SuggestionServiceError {\n switch error {\n case let .notice(error):\n presenter.presentErrorMessage(error.localizedDescription)\n case .silent:\n Logger.service.error(error.localizedDescription)\n return\n }\n } catch {\n Logger.service.error(error.localizedDescription)\n return\n }\n }\n\n @WorkspaceActor\n func invalidateRealtimeSuggestionsIfNeeded(fileURL: URL, sourceEditor: SourceEditor) async {\n guard let (_, filespace) = try? await Service.shared.workspacePool\n .fetchOrCreateWorkspaceAndFilespace(fileURL: fileURL) else { return }\n\n if filespace.presentingSuggestion == nil {\n return // skip if there's no suggestion presented.\n }\n\n let content = sourceEditor.getContent()\n if !filespace.validateSuggestions(\n lines: content.lines,\n cursorPosition: content.cursorPosition\n ) {\n PresentInWindowSuggestionPresenter().discardSuggestion(fileURL: fileURL)\n }\n }\n\n func rejectSuggestions() async {\n let handler = WindowBaseCommandHandler()\n _ = try? await handler.rejectSuggestion(editor: .init(\n content: \"\",\n lines: [],\n uti: \"\",\n cursorPosition: .outOfScope,\n cursorOffset: -1,\n selections: [],\n tabSize: 0,\n indentSize: 0,\n usesTabsForIndentation: false\n ))\n }\n\n func handleCustomCommand(_ command: CustomCommand) async {\n guard let editor = await {\n if let it = await getEditorContent(sourceEditor: nil) {\n return it\n }\n switch command.feature {\n // editor content is not required.\n case .customChat, .chatWithSelection, .singleRoundDialog:\n return .init(\n content: \"\",\n lines: [],\n uti: \"\",\n cursorPosition: .outOfScope,\n cursorOffset: -1,\n selections: [],\n tabSize: 0,\n indentSize: 0,\n usesTabsForIndentation: false\n )\n // editor content is required.\n case .promptToCode:\n return nil\n }\n }() else {\n do {\n try await XcodeInspector.shared.latestActiveXcode?\n .triggerCopilotCommand(name: command.name)\n } catch {\n let presenter = PresentInWindowSuggestionPresenter()\n presenter.presentError(error)\n }\n return\n }\n\n let handler = WindowBaseCommandHandler()\n do {\n try await handler.handleCustomCommand(id: command.id, editor: editor)\n } catch {\n let presenter = PresentInWindowSuggestionPresenter()\n presenter.presentError(error)\n }\n }\n\n func acceptModification() async {\n do {\n if UserDefaults.shared.value(for: \\.alwaysAcceptSuggestionWithAccessibilityAPI) {\n throw CancellationError()\n }\n do {\n try await XcodeInspector.shared.latestActiveXcode?\n .triggerCopilotCommand(name: \"Accept Modification\")\n } catch {\n do {\n try await XcodeInspector.shared.latestActiveXcode?\n .triggerCopilotCommand(name: \"Accept Prompt to Code\")\n } catch {\n let last = Self.lastTimeCommandFailedToTriggerWithAccessibilityAPI\n let now = Date()\n if now.timeIntervalSince(last) > 60 * 60 {\n Self.lastTimeCommandFailedToTriggerWithAccessibilityAPI = now\n toast.toast(content: \"\"\"\n The app is using a fallback solution to accept suggestions. \\\n For better experience, please restart Xcode to re-activate the Copilot \\\n menu item.\n \"\"\", type: .warning, duration: 10)\n }\n\n throw error\n }\n }\n } catch {\n guard let xcode = ActiveApplicationMonitor.shared.activeXcode\n ?? ActiveApplicationMonitor.shared.latestXcode else { return }\n let application = AXUIElementCreateApplication(xcode.processIdentifier)\n guard let focusElement = application.focusedElement,\n focusElement.description == \"Source Editor\"\n else { return }\n guard let (\n content,\n lines,\n _,\n cursorPosition,\n cursorOffset\n ) = await getFileContent(sourceEditor: nil)\n else {\n PresentInWindowSuggestionPresenter()\n .presentErrorMessage(\"Unable to get file content.\")\n return\n }\n let handler = WindowBaseCommandHandler()\n do {\n guard let result = try await handler.acceptPromptToCode(editor: .init(\n content: content,\n lines: lines,\n uti: \"\",\n cursorPosition: cursorPosition,\n cursorOffset: cursorOffset,\n selections: [],\n tabSize: 0,\n indentSize: 0,\n usesTabsForIndentation: false\n )) else { return }\n\n try injectUpdatedCodeWithAccessibilityAPI(result, focusElement: focusElement)\n } catch {\n PresentInWindowSuggestionPresenter().presentError(error)\n }\n }\n }\n\n func presentModification(state: Shared) async {\n let store = await Service.shared.guiController.store\n await store.send(.promptToCodeGroup(.createPromptToCode(.init(\n promptToCodeState: state,\n instruction: nil,\n commandName: nil,\n isContinuous: false\n ), sendImmediately: false)))\n }\n\n func acceptActiveSuggestionLineInGroup(atIndex index: Int?) async {\n do {\n if UserDefaults.shared.value(for: \\.alwaysAcceptSuggestionWithAccessibilityAPI) {\n throw CancellationError()\n }\n do {\n try await XcodeInspector.shared.latestActiveXcode?\n .triggerCopilotCommand(name: \"Accept Suggestion Line\")\n } catch {\n let last = Self.lastTimeCommandFailedToTriggerWithAccessibilityAPI\n let now = Date()\n if now.timeIntervalSince(last) > 60 * 60 {\n Self.lastTimeCommandFailedToTriggerWithAccessibilityAPI = now\n toast.toast(content: \"\"\"\n The app is using a fallback solution to accept suggestions. \\\n For better experience, please restart Xcode to re-activate the Copilot \\\n menu item.\n \"\"\", type: .warning, duration: 10)\n }\n\n throw error\n }\n } catch {\n guard let xcode = ActiveApplicationMonitor.shared.activeXcode\n ?? ActiveApplicationMonitor.shared.latestXcode else { return }\n let application = AXUIElementCreateApplication(xcode.processIdentifier)\n guard let focusElement = application.focusedElement,\n focusElement.description == \"Source Editor\"\n else { return }\n guard let (\n content,\n lines,\n _,\n cursorPosition,\n cursorOffset\n ) = await getFileContent(sourceEditor: nil)\n else {\n PresentInWindowSuggestionPresenter()\n .presentErrorMessage(\"Unable to get file content.\")\n return\n }\n let handler = WindowBaseCommandHandler()\n do {\n guard let result = try await handler.acceptSuggestion(editor: .init(\n content: content,\n lines: lines,\n uti: \"\",\n cursorPosition: cursorPosition,\n cursorOffset: cursorOffset,\n selections: [],\n tabSize: 0,\n indentSize: 0,\n usesTabsForIndentation: false\n )) else { return }\n\n try injectUpdatedCodeWithAccessibilityAPI(result, focusElement: focusElement)\n } catch {\n PresentInWindowSuggestionPresenter().presentError(error)\n }\n }\n }\n\n func acceptSuggestion() async {\n do {\n if UserDefaults.shared.value(for: \\.alwaysAcceptSuggestionWithAccessibilityAPI) {\n throw CancellationError()\n }\n do {\n try await XcodeInspector.shared.latestActiveXcode?\n .triggerCopilotCommand(name: \"Accept Suggestion\")\n } catch {\n let last = Self.lastTimeCommandFailedToTriggerWithAccessibilityAPI\n let now = Date()\n if now.timeIntervalSince(last) > 60 * 60 {\n Self.lastTimeCommandFailedToTriggerWithAccessibilityAPI = now\n toast.toast(content: \"\"\"\n The app is using a fallback solution to accept suggestions. \\\n For better experience, please restart Xcode to re-activate the Copilot \\\n menu item.\n \"\"\", type: .warning, duration: 10)\n }\n\n throw error\n }\n } catch {\n guard let xcode = ActiveApplicationMonitor.shared.activeXcode\n ?? ActiveApplicationMonitor.shared.latestXcode else { return }\n let application = AXUIElementCreateApplication(xcode.processIdentifier)\n guard let focusElement = application.focusedElement,\n focusElement.description == \"Source Editor\"\n else { return }\n guard let (\n content,\n lines,\n _,\n cursorPosition,\n cursorOffset\n ) = await getFileContent(sourceEditor: nil)\n else {\n PresentInWindowSuggestionPresenter()\n .presentErrorMessage(\"Unable to get file content.\")\n return\n }\n let handler = WindowBaseCommandHandler()\n do {\n guard let result = try await handler.acceptSuggestion(editor: .init(\n content: content,\n lines: lines,\n uti: \"\",\n cursorPosition: cursorPosition,\n cursorOffset: cursorOffset,\n selections: [],\n tabSize: 0,\n indentSize: 0,\n usesTabsForIndentation: false\n )) else { return }\n\n try injectUpdatedCodeWithAccessibilityAPI(result, focusElement: focusElement)\n } catch {\n PresentInWindowSuggestionPresenter().presentError(error)\n }\n }\n }\n\n func dismissSuggestion() async {\n guard let documentURL = await XcodeInspector.shared.activeDocumentURL else { return }\n PresentInWindowSuggestionPresenter().discardSuggestion(fileURL: documentURL)\n guard let (_, filespace) = try? await Service.shared.workspacePool\n .fetchOrCreateWorkspaceAndFilespace(fileURL: documentURL) else { return }\n await filespace.reset()\n }\n\n func openChat(forceDetach: Bool, activateThisApp: Bool = true) {\n switch UserDefaults.shared.value(for: \\.openChatMode) {\n case .chatPanel:\n for ext in BuiltinExtensionManager.shared.extensions {\n guard let tab = ext.chatTabTypes.first(where: { $0.isDefaultChatTabReplacement })\n else { continue }\n Task { @MainActor in\n let store = Service.shared.guiController.store\n await store.send(\n .createAndSwitchToChatTabIfNeededMatching(\n check: { $0.name == tab.name },\n kind: .init(tab.defaultChatBuilder())\n )\n ).finish()\n store.send(.openChatPanel(\n forceDetach: forceDetach,\n activateThisApp: activateThisApp\n ))\n }\n return\n }\n Task { @MainActor in\n let store = Service.shared.guiController.store\n await store.send(.createAndSwitchToChatGPTChatTabIfNeeded).finish()\n store.send(.openChatPanel(\n forceDetach: forceDetach,\n activateThisApp: activateThisApp\n ))\n }\n case .browser:\n let urlString = UserDefaults.shared.value(for: \\.openChatInBrowserURL)\n let openInApp = {\n if !UserDefaults.shared.value(for: \\.openChatInBrowserInInAppBrowser) {\n return false\n }\n return isFeatureAvailable(\\.browserTab)\n }()\n guard let url = URL(string: urlString) else {\n let alert = NSAlert()\n alert.messageText = \"Invalid URL\"\n alert.informativeText = \"The URL provided is not valid.\"\n alert.alertStyle = .warning\n alert.runModal()\n return\n }\n\n if openInApp {\n #if canImport(BrowserChatTab)\n Task { @MainActor in\n let store = Service.shared.guiController.store\n await store.send(.createAndSwitchToChatTabIfNeededMatching(\n check: {\n func match(_ tabURL: URL?) -> Bool {\n guard let tabURL else { return false }\n return tabURL == url\n || tabURL.absoluteString.hasPrefix(url.absoluteString)\n }\n\n guard let tab = $0 as? BrowserChatTab,\n match(tab.url) else { return false }\n return true\n },\n kind: .init(BrowserChatTab.urlChatBuilder(url: url))\n )).finish()\n store.send(.openChatPanel(\n forceDetach: forceDetach,\n activateThisApp: activateThisApp\n ))\n }\n #endif\n } else {\n Task {\n NSWorkspace.shared.open(url)\n }\n }\n case let .builtinExtension(extensionIdentifier, id, _):\n guard let ext = BuiltinExtensionManager.shared.extensions\n .first(where: { $0.extensionIdentifier == extensionIdentifier }),\n let tab = ext.chatTabTypes.first(where: { $0.name == id })\n else { return }\n Task { @MainActor in\n let store = Service.shared.guiController.store\n await store.send(\n .createAndSwitchToChatTabIfNeededMatching(\n check: { $0.name == id },\n kind: .init(tab.defaultChatBuilder())\n )\n ).finish()\n store.send(.openChatPanel(\n forceDetach: forceDetach,\n activateThisApp: activateThisApp\n ))\n }\n case let .externalExtension(extensionIdentifier, id, _):\n guard let ext = BuiltinExtensionManager.shared.extensions\n .first(where: { $0.extensionIdentifier == \"plus\" }),\n let tab = ext.chatTabTypes\n .first(where: { $0.name == \"\\(extensionIdentifier).\\(id)\" })\n else { return }\n Task { @MainActor in\n let store = Service.shared.guiController.store\n await store.send(\n .createAndSwitchToChatTabIfNeededMatching(\n check: { $0.name == \"\\(extensionIdentifier).\\(id)\" },\n kind: .init(tab.defaultChatBuilder())\n )\n ).finish()\n store.send(.openChatPanel(\n forceDetach: forceDetach,\n activateThisApp: activateThisApp\n ))\n }\n }\n }\n\n @MainActor\n func sendChatMessage(_ message: String) async {\n let store = Service.shared.guiController.store\n await store.send(.sendCustomCommandToActiveChat(CustomCommand(\n commandId: \"\",\n name: \"\",\n feature: .chatWithSelection(\n extraSystemPrompt: nil,\n prompt: message,\n useExtraSystemPrompt: nil\n ),\n ignoreExistingAttachments: false,\n attachments: []\n ))).finish()\n }\n\n @WorkspaceActor\n func presentSuggestions(_ suggestions: [SuggestionBasic.CodeSuggestion]) async {\n guard let filespace = await getFilespace() else { return }\n filespace.setSuggestions(suggestions)\n PresentInWindowSuggestionPresenter().presentSuggestion(fileURL: filespace.fileURL)\n }\n\n func toast(_ message: String, as type: ToastType) {\n Task { @MainActor in\n let store = Service.shared.guiController.store\n store.send(.suggestionWidget(.toastPanel(.toast(.toast(message, type, nil)))))\n }\n }\n\n func presentFile(at fileURL: URL, line: Int?) async {\n let terminal = Terminal()\n do {\n if let line {\n _ = try await terminal.runCommand(\n \"/bin/bash\",\n arguments: [\n \"-c\",\n \"xed -l \\(line) ${TARGET_FILE}\",\n ],\n environment: [\"TARGET_FILE\": fileURL.path],\n )\n } else {\n _ = try await terminal.runCommand(\n \"/bin/bash\",\n arguments: [\n \"-c\",\n \"xed ${TARGET_FILE}\",\n ],\n environment: [\"TARGET_FILE\": fileURL.path],\n )\n }\n } catch {\n print(error)\n }\n }\n}\n\nextension PseudoCommandHandler {\n /// When Xcode commands are not available, we can fallback to directly\n /// set the value of the editor with Accessibility API.\n func injectUpdatedCodeWithAccessibilityAPI(\n _ result: UpdatedContent,\n focusElement: AXUIElement\n ) throws {\n let oldPosition = focusElement.selectedTextRange\n let oldScrollPosition = focusElement.parent?.verticalScrollBar?.doubleValue\n\n let error = AXUIElementSetAttributeValue(\n focusElement,\n kAXValueAttribute as CFString,\n result.content as CFTypeRef\n )\n\n if error != AXError.success {\n PresentInWindowSuggestionPresenter()\n .presentErrorMessage(\"Fail to set editor content.\")\n }\n\n // recover selection range\n\n if let selection = result.newSelections.first {\n var range = SourceEditor.convertCursorRangeToRange(selection, in: result.content)\n if let value = AXValueCreate(.cfRange, &range) {\n AXUIElementSetAttributeValue(\n focusElement,\n kAXSelectedTextRangeAttribute as CFString,\n value\n )\n }\n } else if let oldPosition {\n var range = CFRange(\n location: oldPosition.lowerBound,\n length: 0\n )\n if let value = AXValueCreate(.cfRange, &range) {\n AXUIElementSetAttributeValue(\n focusElement,\n kAXSelectedTextRangeAttribute as CFString,\n value\n )\n }\n }\n\n // recover scroll position\n\n if let oldScrollPosition,\n let scrollBar = focusElement.parent?.verticalScrollBar\n {\n AXUIElementSetAttributeValue(\n scrollBar,\n kAXValueAttribute as CFString,\n oldScrollPosition as CFTypeRef\n )\n }\n }\n\n func getFileContent(sourceEditor: AXUIElement?) async\n -> (\n content: String,\n lines: [String],\n selections: [CursorRange],\n cursorPosition: CursorPosition,\n cursorOffset: Int\n )?\n {\n guard let xcode = ActiveApplicationMonitor.shared.activeXcode\n ?? ActiveApplicationMonitor.shared.latestXcode else { return nil }\n let application = AXUIElementCreateApplication(xcode.processIdentifier)\n guard let focusElement = sourceEditor ?? application.focusedElement,\n focusElement.description == \"Source Editor\"\n else { return nil }\n guard let selectionRange = focusElement.selectedTextRange else { return nil }\n let content = focusElement.value\n let split = content.breakLines(appendLineBreakToLastLine: false)\n let range = SourceEditor.convertRangeToCursorRange(selectionRange, in: content)\n return (content, split, [range], range.start, selectionRange.lowerBound)\n }\n\n func getFileURL() async -> URL? {\n XcodeInspector.shared.realtimeActiveDocumentURL\n }\n\n @WorkspaceActor\n func getFilespace() async -> Filespace? {\n guard\n let fileURL = await getFileURL(),\n let (_, filespace) = try? await Service.shared.workspacePool\n .fetchOrCreateWorkspaceAndFilespace(fileURL: fileURL)\n else { return nil }\n return filespace\n }\n\n @WorkspaceActor\n func getEditorContent(sourceEditor: SourceEditor?) async -> EditorContent? {\n guard let filespace = await getFilespace(),\n let sourceEditor = await {\n if let sourceEditor { sourceEditor }\n else { await XcodeInspector.shared.latestFocusedEditor }\n }()\n else { return nil }\n if Task.isCancelled { return nil }\n let content = sourceEditor.getContent()\n let uti = filespace.codeMetadata.uti ?? \"\"\n let tabSize = filespace.codeMetadata.tabSize ?? 4\n let indentSize = filespace.codeMetadata.indentSize ?? 4\n let usesTabsForIndentation = filespace.codeMetadata.usesTabsForIndentation ?? false\n return .init(\n content: content.content,\n lines: content.lines,\n uti: uti,\n cursorPosition: content.cursorPosition,\n cursorOffset: content.cursorOffset,\n selections: content.selections.map {\n .init(start: $0.start, end: $0.end)\n },\n tabSize: tabSize,\n indentSize: indentSize,\n usesTabsForIndentation: usesTabsForIndentation\n )\n }\n\n func handleAcceptSuggestionLineCommand(editor: EditorContent) async throws -> CodeSuggestion? {\n guard let _ = XcodeInspector.shared.realtimeActiveDocumentURL\n else { return nil }\n\n return try await acceptSuggestionLineInGroup(\n atIndex: 0,\n editor: editor\n )\n }\n\n func acceptSuggestionLineInGroup(\n atIndex index: Int?,\n editor: EditorContent\n ) async throws -> CodeSuggestion? {\n guard let fileURL = XcodeInspector.shared.realtimeActiveDocumentURL\n else { return nil }\n let (workspace, _) = try await Service.shared.workspacePool\n .fetchOrCreateWorkspaceAndFilespace(fileURL: fileURL)\n\n guard var acceptedSuggestion = await workspace.acceptSuggestion(\n forFileAt: fileURL,\n editor: editor\n ) else { return nil }\n\n let text = acceptedSuggestion.text\n acceptedSuggestion.text = String(text.splitByNewLine().first ?? \"\")\n return acceptedSuggestion\n }\n}\n\n" }, { "path": "Core/Sources/Service/SuggestionCommandHandler/SuggestionCommandHandler.swift", "content": "import SuggestionBasic\nimport XPCShared\n\nprotocol SuggestionCommandHandler {\n @ServiceActor\n func presentSuggestions(editor: EditorContent) async throws -> UpdatedContent?\n @ServiceActor\n func presentNextSuggestion(editor: EditorContent) async throws -> UpdatedContent?\n @ServiceActor\n func presentPreviousSuggestion(editor: EditorContent) async throws -> UpdatedContent?\n @ServiceActor\n func rejectSuggestion(editor: EditorContent) async throws -> UpdatedContent?\n @ServiceActor\n func acceptSuggestionLine(editor: EditorContent) async throws -> UpdatedContent?\n @ServiceActor\n func acceptSuggestion(editor: EditorContent) async throws -> UpdatedContent?\n @ServiceActor\n func acceptPromptToCode(editor: EditorContent) async throws -> UpdatedContent?\n @ServiceActor\n func presentRealtimeSuggestions(editor: EditorContent) async throws -> UpdatedContent?\n @ServiceActor\n func generateRealtimeSuggestions(editor: EditorContent) async throws -> UpdatedContent?\n @ServiceActor\n func promptToCode(editor: EditorContent) async throws -> UpdatedContent?\n @ServiceActor\n func customCommand(id: String, editor: EditorContent) async throws -> UpdatedContent?\n}\n\n" }, { "path": "Core/Sources/Service/SuggestionCommandHandler/WindowBaseCommandHandler.swift", "content": "import AppKit\nimport ChatService\nimport ComposableArchitecture\nimport CustomCommandTemplateProcessor\nimport Foundation\nimport GitHubCopilotService\nimport LanguageServerProtocol\nimport Logger\nimport ModificationBasic\nimport OpenAIService\nimport SuggestionBasic\nimport SuggestionInjector\nimport SuggestionWidget\nimport UserNotifications\nimport Workspace\nimport WorkspaceSuggestionService\nimport XcodeInspector\nimport XPCShared\n\nstruct WindowBaseCommandHandler: SuggestionCommandHandler {\n nonisolated init() {}\n\n let presenter = PresentInWindowSuggestionPresenter()\n\n func presentSuggestions(editor: EditorContent) async throws -> UpdatedContent? {\n Task {\n do {\n try await _presentSuggestions(editor: editor)\n } catch let error as ServerError {\n Logger.service.error(error)\n } catch {\n presenter.presentError(error)\n Logger.service.error(error)\n }\n }\n return nil\n }\n\n @WorkspaceActor\n private func _presentSuggestions(editor: EditorContent) async throws {\n presenter.markAsProcessing(true)\n defer {\n presenter.markAsProcessing(false)\n }\n guard let fileURL = XcodeInspector.shared.realtimeActiveDocumentURL\n else { return }\n let (workspace, filespace) = try await Service.shared.workspacePool\n .fetchOrCreateWorkspaceAndFilespace(fileURL: fileURL)\n\n try Task.checkCancellation()\n\n try await workspace.generateSuggestions(\n forFileAt: fileURL,\n editor: editor\n )\n\n try Task.checkCancellation()\n\n if filespace.presentingSuggestion != nil {\n presenter.presentSuggestion(fileURL: fileURL)\n } else {\n presenter.discardSuggestion(fileURL: fileURL)\n }\n }\n\n func presentNextSuggestion(editor: EditorContent) async throws -> UpdatedContent? {\n Task {\n do {\n try await _presentNextSuggestion(editor: editor)\n } catch {\n presenter.presentError(error)\n }\n }\n return nil\n }\n\n @WorkspaceActor\n private func _presentNextSuggestion(editor: EditorContent) async throws {\n guard let fileURL = XcodeInspector.shared.realtimeActiveDocumentURL\n else { return }\n let (workspace, filespace) = try await Service.shared.workspacePool\n .fetchOrCreateWorkspaceAndFilespace(fileURL: fileURL)\n workspace.selectNextSuggestion(forFileAt: fileURL)\n\n if filespace.presentingSuggestion != nil {\n presenter.presentSuggestion(fileURL: fileURL)\n } else {\n presenter.discardSuggestion(fileURL: fileURL)\n }\n }\n\n func presentPreviousSuggestion(editor: EditorContent) async throws -> UpdatedContent? {\n Task {\n do {\n try await _presentPreviousSuggestion(editor: editor)\n } catch {\n presenter.presentError(error)\n }\n }\n return nil\n }\n\n @WorkspaceActor\n private func _presentPreviousSuggestion(editor: EditorContent) async throws {\n guard let fileURL = XcodeInspector.shared.realtimeActiveDocumentURL\n else { return }\n let (workspace, filespace) = try await Service.shared.workspacePool\n .fetchOrCreateWorkspaceAndFilespace(fileURL: fileURL)\n workspace.selectPreviousSuggestion(forFileAt: fileURL)\n\n if filespace.presentingSuggestion != nil {\n presenter.presentSuggestion(fileURL: fileURL)\n } else {\n presenter.discardSuggestion(fileURL: fileURL)\n }\n }\n\n func rejectSuggestion(editor: EditorContent) async throws -> UpdatedContent? {\n Task {\n do {\n try await _rejectSuggestion(editor: editor)\n } catch {\n presenter.presentError(error)\n }\n }\n return nil\n }\n\n @WorkspaceActor\n private func _rejectSuggestion(editor: EditorContent) async throws {\n guard let fileURL = XcodeInspector.shared.realtimeActiveDocumentURL\n else { return }\n\n let (workspace, _) = try await Service.shared.workspacePool\n .fetchOrCreateWorkspaceAndFilespace(fileURL: fileURL)\n workspace.rejectSuggestion(forFileAt: fileURL, editor: editor)\n presenter.discardSuggestion(fileURL: fileURL)\n }\n\n @WorkspaceActor\n func acceptSuggestion(editor: EditorContent) async throws -> UpdatedContent? {\n guard let fileURL = XcodeInspector.shared.realtimeActiveDocumentURL\n else { return nil }\n let (workspace, _) = try await Service.shared.workspacePool\n .fetchOrCreateWorkspaceAndFilespace(fileURL: fileURL)\n\n let injector = SuggestionInjector()\n var lines = editor.lines\n var cursorPosition = editor.cursorPosition\n var extraInfo = SuggestionInjector.ExtraInfo()\n\n if let acceptedSuggestion = workspace.acceptSuggestion(\n forFileAt: fileURL,\n editor: editor\n ) {\n injector.acceptSuggestion(\n intoContentWithoutSuggestion: &lines,\n cursorPosition: &cursorPosition,\n completion: acceptedSuggestion,\n extraInfo: &extraInfo\n )\n\n presenter.discardSuggestion(fileURL: fileURL)\n\n return .init(\n content: String(lines.joined(separator: \"\")),\n newSelection: .cursor(cursorPosition),\n modifications: extraInfo.modifications\n )\n }\n\n return nil\n }\n\n func acceptSuggestionLine(editor: EditorContent) async throws -> UpdatedContent? {\n if let acceptedSuggestion = try await PseudoCommandHandler()\n .handleAcceptSuggestionLineCommand(editor: editor)\n {\n let injector = SuggestionInjector()\n var lines = editor.lines\n var cursorPosition = editor.cursorPosition\n var extraInfo = SuggestionInjector.ExtraInfo()\n\n injector.acceptSuggestion(\n intoContentWithoutSuggestion: &lines,\n cursorPosition: &cursorPosition,\n completion: acceptedSuggestion,\n extraInfo: &extraInfo\n )\n\n return .init(\n content: String(lines.joined(separator: \"\")),\n newSelection: .cursor(cursorPosition),\n modifications: extraInfo.modifications\n )\n }\n\n return nil\n }\n\n func acceptPromptToCode(editor: EditorContent) async throws -> UpdatedContent? {\n guard let fileURL = XcodeInspector.shared.realtimeActiveDocumentURL\n else { return nil }\n\n let injector = SuggestionInjector()\n var lines = editor.lines\n var cursorPosition = editor.cursorPosition\n var extraInfo = SuggestionInjector.ExtraInfo()\n\n let store = await Service.shared.guiController.store\n\n if let promptToCode = await MainActor\n .run(body: { store.state.promptToCodeGroup.activePromptToCode })\n {\n if promptToCode.promptToCodeState.isAttachedToTarget,\n promptToCode.promptToCodeState.source.documentURL != fileURL\n {\n return nil\n }\n\n let suggestions = promptToCode.promptToCodeState.snippets\n .map { snippet in\n let range = {\n if promptToCode.promptToCodeState.isAttachedToTarget {\n return snippet.attachedRange\n }\n return editor.selections.first.map {\n CursorRange(start: $0.start, end: $0.end)\n } ?? CursorRange(\n start: editor.cursorPosition,\n end: editor.cursorPosition\n )\n }()\n return CodeSuggestion(\n id: snippet.id.uuidString,\n text: snippet.modifiedCode,\n position: range.start,\n range: range\n )\n }\n\n injector.acceptSuggestions(\n intoContentWithoutSuggestion: &lines,\n cursorPosition: &cursorPosition,\n completions: suggestions,\n extraInfo: &extraInfo\n )\n\n for (id, range) in extraInfo.modificationRanges {\n _ = await MainActor.run {\n store.send(\n .promptToCodeGroup(.updatePromptToCodeRange(\n id: promptToCode.id,\n snippetId: .init(uuidString: id) ?? .init(),\n range: range\n ))\n )\n }\n }\n\n _ = await MainActor.run {\n store.send(\n .promptToCodeGroup(.discardAcceptedPromptToCodeIfNotContinuous(\n id: promptToCode.id\n ))\n )\n }\n\n return .init(\n content: String(lines.joined(separator: \"\")),\n newSelections: extraInfo.modificationRanges.values\n .sorted(by: { $0.start.line <= $1.start.line }),\n modifications: extraInfo.modifications\n )\n }\n\n return nil\n }\n\n func presentRealtimeSuggestions(editor: EditorContent) async throws -> UpdatedContent? {\n Task {\n try? await prepareCache(editor: editor)\n }\n return nil\n }\n\n @WorkspaceActor\n func prepareCache(editor: EditorContent) async throws -> UpdatedContent? {\n guard let fileURL = XcodeInspector.shared.realtimeActiveDocumentURL\n else { return nil }\n let (_, filespace) = try await Service.shared.workspacePool\n .fetchOrCreateWorkspaceAndFilespace(fileURL: fileURL)\n filespace.codeMetadata.uti = editor.uti\n filespace.codeMetadata.tabSize = editor.tabSize\n filespace.codeMetadata.indentSize = editor.indentSize\n filespace.codeMetadata.usesTabsForIndentation = editor.usesTabsForIndentation\n filespace.codeMetadata.guessLineEnding(from: editor.lines.first)\n return nil\n }\n\n func generateRealtimeSuggestions(editor: EditorContent) async throws -> UpdatedContent? {\n return try await presentSuggestions(editor: editor)\n }\n\n func promptToCode(editor: EditorContent) async throws -> UpdatedContent? {\n Task {\n do {\n try await presentPromptToCode(\n editor: editor,\n extraSystemPrompt: nil,\n prompt: nil,\n isContinuous: false,\n generateDescription: nil,\n name: nil\n )\n } catch {\n presenter.presentError(error)\n }\n }\n return nil\n }\n\n func customCommand(id: String, editor: EditorContent) async throws -> UpdatedContent? {\n Task {\n do {\n try await handleCustomCommand(id: id, editor: editor)\n } catch {\n presenter.presentError(error)\n }\n }\n return nil\n }\n}\n\nextension WindowBaseCommandHandler {\n func handleCustomCommand(id: String, editor: EditorContent) async throws {\n struct CommandNotFoundError: Error, LocalizedError {\n var errorDescription: String? { \"Command not found\" }\n }\n\n let availableCommands = UserDefaults.shared.value(for: \\.customCommands)\n guard let command = availableCommands.first(where: { $0.id == id })\n else { throw CommandNotFoundError() }\n\n switch command.feature {\n case .chatWithSelection, .customChat:\n Task { @MainActor in\n Service.shared.guiController.store\n .send(.sendCustomCommandToActiveChat(command))\n }\n case let .promptToCode(extraSystemPrompt, prompt, continuousMode, generateDescription):\n try await presentPromptToCode(\n editor: editor,\n extraSystemPrompt: extraSystemPrompt,\n prompt: prompt,\n isContinuous: continuousMode ?? false,\n generateDescription: generateDescription,\n name: command.name\n )\n case let .singleRoundDialog(\n systemPrompt,\n overwriteSystemPrompt,\n prompt,\n receiveReplyInNotification\n ):\n try await executeSingleRoundDialog(\n systemPrompt: systemPrompt,\n overwriteSystemPrompt: overwriteSystemPrompt ?? false,\n prompt: prompt ?? \"\",\n receiveReplyInNotification: receiveReplyInNotification ?? false\n )\n }\n }\n\n @WorkspaceActor\n func presentPromptToCode(\n editor: EditorContent,\n extraSystemPrompt: String?,\n prompt: String?,\n isContinuous: Bool,\n generateDescription: Bool?,\n name: String?\n ) async throws {\n guard let fileURL = XcodeInspector.shared.realtimeActiveDocumentURL\n else { return }\n let (workspace, _) = try await Service.shared.workspacePool\n .fetchOrCreateWorkspaceAndFilespace(fileURL: fileURL)\n guard workspace.suggestionPlugin?.isSuggestionFeatureEnabled ?? false else {\n presenter.presentErrorMessage(\"Prompt to code is disabled for this project\")\n return\n }\n\n let codeLanguage = languageIdentifierFromFileURL(fileURL)\n\n let selections: [CursorRange] = {\n if let firstSelection = editor.selections.first,\n let lastSelection = editor.selections.last\n {\n let range = CursorRange(\n start: firstSelection.start,\n end: lastSelection.end\n )\n return [range]\n }\n return []\n }()\n\n let snippets = selections.map { selection in\n guard selection.start != selection.end else {\n return ModificationSnippet(\n startLineIndex: selection.start.line,\n originalCode: \"\",\n modifiedCode: \"\",\n description: \"\",\n error: \"\",\n attachedRange: selection\n )\n }\n var selection = selection\n let isMultipleLine = selection.start.line != selection.end.line\n let isSpaceOnlyBeforeStartPositionOnTheSameLine = {\n guard selection.start.line >= 0, selection.start.line < editor.lines.count else {\n return false\n }\n let line = editor.lines[selection.start.line]\n guard selection.start.character > 0,\n selection.start.character < line.utf16.count\n else { return false }\n let substring = line[line.utf16.startIndex..<(line.index(\n line.utf16.startIndex,\n offsetBy: selection.start.character,\n limitedBy: line.utf16.endIndex\n ) ?? line.utf16.endIndex)]\n return substring.allSatisfy { $0.isWhitespace }\n }()\n\n if isMultipleLine || isSpaceOnlyBeforeStartPositionOnTheSameLine {\n // when there are multiple lines start from char 0 so that it can keep the\n // indentation.\n selection.start = .init(line: selection.start.line, character: 0)\n }\n let selectedCode = editor.selectedCode(in: .init(\n start: selection.start,\n end: selection.end\n ))\n return ModificationSnippet(\n startLineIndex: selection.start.line,\n originalCode: selectedCode,\n modifiedCode: selectedCode,\n description: \"\",\n error: \"\",\n attachedRange: .init(start: selection.start, end: selection.end)\n )\n }\n\n let store = await Service.shared.guiController.store\n\n let customCommandTemplateProcessor = CustomCommandTemplateProcessor()\n\n let newExtraSystemPrompt: String? = if let extraSystemPrompt {\n await customCommandTemplateProcessor.process(extraSystemPrompt)\n } else {\n nil\n }\n\n let newPrompt: String? = if let prompt {\n await customCommandTemplateProcessor.process(prompt)\n } else {\n nil\n }\n\n _ = await MainActor.run {\n store.send(.promptToCodeGroup(.activateOrCreatePromptToCode(.init(\n promptToCodeState: Shared(.init(\n source: .init(\n language: codeLanguage,\n documentURL: fileURL,\n projectRootURL: workspace.projectRootURL,\n content: editor.content,\n lines: editor.lines\n ),\n snippets: IdentifiedArray(uniqueElements: snippets),\n extraSystemPrompt: newExtraSystemPrompt ?? \"\",\n isAttachedToTarget: true\n )),\n instruction: newPrompt,\n commandName: name,\n isContinuous: isContinuous\n ))))\n }\n }\n\n func executeSingleRoundDialog(\n systemPrompt: String?,\n overwriteSystemPrompt: Bool,\n prompt: String,\n receiveReplyInNotification: Bool\n ) async throws {\n guard !prompt.isEmpty else { return }\n\n let service = ChatService()\n\n let result = try await service.handleSingleRoundDialogCommand(\n systemPrompt: systemPrompt,\n overwriteSystemPrompt: overwriteSystemPrompt,\n prompt: prompt\n )\n\n guard receiveReplyInNotification else { return }\n\n let granted = try await UNUserNotificationCenter.current()\n .requestAuthorization(options: [.alert])\n\n if granted {\n let content = UNMutableNotificationContent()\n content.title = \"Reply\"\n content.body = result\n let request = UNNotificationRequest(\n identifier: \"reply\",\n content: content,\n trigger: nil\n )\n do {\n try await UNUserNotificationCenter.current().add(request)\n } catch {\n presenter.presentError(error)\n }\n } else {\n presenter.presentErrorMessage(\"Notification permission is not granted.\")\n }\n }\n}\n\n" }, { "path": "Core/Sources/Service/SuggestionPresenter/PresentInWindowSuggestionPresenter.swift", "content": "import ChatService\nimport Foundation\nimport OpenAIService\nimport SuggestionBasic\nimport SuggestionWidget\n\nstruct PresentInWindowSuggestionPresenter {\n func presentSuggestion(fileURL: URL) {\n Task { @MainActor in\n let controller = Service.shared.guiController.widgetController\n controller.suggestCode()\n }\n }\n\n func discardSuggestion(fileURL: URL) {\n Task { @MainActor in\n let controller = Service.shared.guiController.widgetController\n controller.discardSuggestion()\n }\n }\n\n func markAsProcessing(_ isProcessing: Bool) {\n Task { @MainActor in\n let controller = Service.shared.guiController.widgetController\n controller.markAsProcessing(isProcessing)\n }\n }\n\n func presentError(_ error: Error) {\n if error is CancellationError { return }\n if let urlError = error as? URLError, urlError.code == URLError.cancelled { return }\n Task { @MainActor in\n let controller = Service.shared.guiController.widgetController\n controller.presentError(error.localizedDescription)\n }\n }\n\n func presentErrorMessage(_ message: String) {\n Task { @MainActor in\n let controller = Service.shared.guiController.widgetController\n controller.presentError(message)\n }\n }\n\n func presentChatRoom(fileURL: URL) {\n Task { @MainActor in\n let controller = Service.shared.guiController\n controller.store.send(.openChatPanel(forceDetach: false, activateThisApp: true))\n }\n }\n}\n\n" }, { "path": "Core/Sources/Service/WorkspaceExtension/Workspace+Cleanup.swift", "content": "import Foundation\nimport SuggestionProvider\nimport Workspace\nimport WorkspaceSuggestionService\n\nextension Workspace {\n @WorkspaceActor\n func cleanUp(availableTabs: Set) {\n for (fileURL, _) in filespaces {\n if isFilespaceExpired(fileURL: fileURL, availableTabs: availableTabs) {\n openedFileRecoverableStorage.closeFile(fileURL: fileURL)\n closeFilespace(fileURL: fileURL)\n }\n }\n }\n\n func isFilespaceExpired(fileURL: URL, availableTabs: Set) -> Bool {\n let filename = fileURL.lastPathComponent\n if availableTabs.contains(filename) { return false }\n guard let filespace = filespaces[fileURL] else { return true }\n return filespace.isExpired\n }\n\n func cancelInFlightRealtimeSuggestionRequests() async {\n guard let suggestionService else { return }\n await suggestionService.cancelRequest(workspaceInfo: .init(\n workspaceURL: workspaceURL,\n projectURL: projectRootURL\n ))\n }\n}\n\n" }, { "path": "Core/Sources/Service/XPCService.swift", "content": "import AppKit\nimport Foundation\nimport GitHubCopilotService\nimport LanguageServerProtocol\nimport Logger\nimport Preferences\nimport XPCShared\n\npublic class XPCService: NSObject, XPCServiceProtocol {\n // MARK: - Service\n\n public func getXPCServiceVersion(withReply reply: @escaping (String, String) -> Void) {\n reply(\n Bundle.main.infoDictionary?[\"CFBundleShortVersionString\"] as? String ?? \"N/A\",\n Bundle.main.infoDictionary?[\"CFBundleVersion\"] as? String ?? \"N/A\"\n )\n }\n\n public func getXPCServiceAccessibilityPermission(withReply reply: @escaping (Bool) -> Void) {\n reply(AXIsProcessTrusted())\n }\n\n // MARK: - Suggestion\n\n @discardableResult\n private func replyWithUpdatedContent(\n editorContent: Data,\n file: StaticString = #file,\n line: UInt = #line,\n isRealtimeSuggestionRelatedCommand: Bool = false,\n withReply reply: @escaping (Data?, Error?) -> Void,\n getUpdatedContent: @escaping @ServiceActor (\n SuggestionCommandHandler,\n EditorContent\n ) async throws -> UpdatedContent?\n ) -> Task {\n let task = Task {\n do {\n let editor = try JSONDecoder().decode(EditorContent.self, from: editorContent)\n let handler: SuggestionCommandHandler = WindowBaseCommandHandler()\n try Task.checkCancellation()\n guard let updatedContent = try await getUpdatedContent(handler, editor) else {\n reply(nil, nil)\n return\n }\n try Task.checkCancellation()\n try reply(JSONEncoder().encode(updatedContent), nil)\n } catch {\n Logger.service.error(\"\\(file):\\(line) \\(error.localizedDescription)\")\n reply(nil, NSError.from(error))\n }\n }\n\n Task {\n await Service.shared.realtimeSuggestionController.cancelInFlightTasks(excluding: task)\n }\n return task\n }\n\n public func getSuggestedCode(\n editorContent: Data,\n withReply reply: @escaping (Data?, Error?) -> Void\n ) {\n replyWithUpdatedContent(editorContent: editorContent, withReply: reply) { handler, editor in\n try await handler.presentSuggestions(editor: editor)\n }\n }\n\n public func getNextSuggestedCode(\n editorContent: Data,\n withReply reply: @escaping (Data?, Error?) -> Void\n ) {\n replyWithUpdatedContent(editorContent: editorContent, withReply: reply) { handler, editor in\n try await handler.presentNextSuggestion(editor: editor)\n }\n }\n\n public func getPreviousSuggestedCode(\n editorContent: Data,\n withReply reply: @escaping (Data?, Error?) -> Void\n ) {\n replyWithUpdatedContent(editorContent: editorContent, withReply: reply) { handler, editor in\n try await handler.presentPreviousSuggestion(editor: editor)\n }\n }\n\n public func getSuggestionRejectedCode(\n editorContent: Data,\n withReply reply: @escaping (Data?, Error?) -> Void\n ) {\n replyWithUpdatedContent(editorContent: editorContent, withReply: reply) { handler, editor in\n try await handler.rejectSuggestion(editor: editor)\n }\n }\n\n public func getSuggestionAcceptedCode(\n editorContent: Data,\n withReply reply: @escaping (Data?, Error?) -> Void\n ) {\n replyWithUpdatedContent(editorContent: editorContent, withReply: reply) { handler, editor in\n try await handler.acceptSuggestion(editor: editor)\n }\n }\n\n public func getPromptToCodeAcceptedCode(\n editorContent: Data,\n withReply reply: @escaping (Data?, Error?) -> Void\n ) {\n replyWithUpdatedContent(editorContent: editorContent, withReply: reply) { handler, editor in\n try await handler.acceptPromptToCode(editor: editor)\n }\n }\n\n public func getRealtimeSuggestedCode(\n editorContent: Data,\n withReply reply: @escaping (Data?, Error?) -> Void\n ) {\n replyWithUpdatedContent(\n editorContent: editorContent,\n isRealtimeSuggestionRelatedCommand: true,\n withReply: reply\n ) { handler, editor in\n try await handler.presentRealtimeSuggestions(editor: editor)\n }\n }\n\n public func prefetchRealtimeSuggestions(\n editorContent: Data,\n withReply reply: @escaping () -> Void\n ) {\n // We don't need to wait for this.\n reply()\n\n replyWithUpdatedContent(\n editorContent: editorContent,\n isRealtimeSuggestionRelatedCommand: true,\n withReply: { _, _ in }\n ) { handler, editor in\n try await handler.generateRealtimeSuggestions(editor: editor)\n }\n }\n\n public func openChat(\n editorContent: Data,\n withReply reply: @escaping (Data?, Error?) -> Void\n ) {\n let handler = PseudoCommandHandler()\n handler.openChat(forceDetach: false)\n reply(nil, nil)\n }\n\n public func promptToCode(\n editorContent: Data,\n withReply reply: @escaping (Data?, Error?) -> Void\n ) {\n replyWithUpdatedContent(editorContent: editorContent, withReply: reply) { handler, editor in\n try await handler.promptToCode(editor: editor)\n }\n }\n\n public func customCommand(\n id: String,\n editorContent: Data,\n withReply reply: @escaping (Data?, Error?) -> Void\n ) {\n replyWithUpdatedContent(editorContent: editorContent, withReply: reply) { handler, editor in\n try await handler.customCommand(id: id, editor: editor)\n }\n }\n\n // MARK: - Settings\n\n public func toggleRealtimeSuggestion(withReply reply: @escaping (Error?) -> Void) {\n guard AXIsProcessTrusted() else {\n reply(NoAccessToAccessibilityAPIError())\n return\n }\n Task { @ServiceActor in\n await Service.shared.realtimeSuggestionController.cancelInFlightTasks()\n let on = !UserDefaults.shared.value(for: \\.realtimeSuggestionToggle)\n UserDefaults.shared.set(on, for: \\.realtimeSuggestionToggle)\n Task { @MainActor in\n Service.shared.guiController.store\n .send(.suggestionWidget(.toastPanel(.toast(.toast(\n \"Real-time suggestion is turned \\(on ? \"on\" : \"off\")\",\n .info,\n nil\n )))))\n }\n reply(nil)\n }\n }\n\n public func postNotification(name: String, withReply reply: @escaping () -> Void) {\n reply()\n NotificationCenter.default.post(name: .init(name), object: nil)\n }\n \n public func quit(reply: @escaping () -> Void) {\n Task {\n await Service.shared.prepareForExit()\n reply()\n }\n }\n\n // MARK: - Requests\n\n public func send(\n endpoint: String,\n requestBody: Data,\n reply: @escaping (Data?, Error?) -> Void\n ) {\n Task {\n await Service.shared.handleXPCServiceRequests(\n endpoint: endpoint,\n requestBody: requestBody,\n reply: reply\n )\n }\n }\n}\n\nstruct NoAccessToAccessibilityAPIError: Error, LocalizedError {\n var errorDescription: String? {\n \"Accessibility API permission is not granted. Please enable in System Settings.app.\"\n }\n\n init() {}\n}\n\n" }, { "path": "Core/Sources/ServiceUpdateMigration/MigrateTo135.swift", "content": "import Foundation\nimport GitHubCopilotService\n\nfunc migrateFromLowerThanOrEqualToVersion135() throws {\n // 0. Create the application support folder if it doesn't exist\n\n let urls = try GitHubCopilotBaseService.createFoldersIfNeeded()\n\n // 1. Move the undefined folder in application support into a sub folder called `GitHub\n // Copilot/support`\n\n let undefinedFolderURL = urls.applicationSupportURL.appendingPathComponent(\"undefined\")\n var isUndefinedADirectory: ObjCBool = false\n let isUndefinedExisted = FileManager.default.fileExists(\n atPath: undefinedFolderURL.path,\n isDirectory: &isUndefinedADirectory\n )\n if isUndefinedExisted, isUndefinedADirectory.boolValue {\n try FileManager.default.moveItem(\n at: undefinedFolderURL,\n to: urls.supportURL.appendingPathComponent(\"undefined\")\n )\n }\n\n // 2. Copy the GitHub copilot language service to `GitHub Copilot/executable`\n\n let copilotFolderURL = urls.executableURL.appendingPathComponent(\"copilot\")\n var copilotIsFolder: ObjCBool = false\n let executable = Bundle.main.resourceURL?.appendingPathComponent(\"copilot\")\n if let executable,\n FileManager.default.fileExists(atPath: executable.path, isDirectory: &copilotIsFolder),\n !FileManager.default.fileExists(atPath: copilotFolderURL.path)\n {\n try FileManager.default.copyItem(\n at: executable,\n to: urls.executableURL.appendingPathComponent(\"copilot\")\n )\n }\n\n // 3. Use chmod to change the permission of the executable to 755\n\n try FileManager.default.setAttributes(\n [.posixPermissions: 0o755],\n ofItemAtPath: copilotFolderURL.path\n )\n}\n\n" }, { "path": "Core/Sources/ServiceUpdateMigration/MigrateTo240.swift", "content": "import AIModel\nimport Foundation\nimport Keychain\nimport Preferences\n\nfunc migrateTo240(\n defaults: UserDefaults = .shared,\n keychain: KeychainType = Keychain.apiKey\n) throws {\n let finishedMigrationKey = \"MigrateTo240Finished\"\n if defaults.bool(forKey: finishedMigrationKey) { return }\n\n do {\n let chatModelOpenAIId = UUID().uuidString\n let chatModelAzureOpenAIId = UUID().uuidString\n let embeddingModelOpenAIId = UUID().uuidString\n let embeddingModelAzureOpenAIId = UUID().uuidString\n \n let openAIAPIKeyName = \"OpenAI\"\n let openAIAPIKey = defaults.deprecatedValue(for: \\.openAIAPIKey)\n if !openAIAPIKey.isEmpty {\n try keychain.update(openAIAPIKey, key: openAIAPIKeyName)\n }\n \n let azureOpenAIAPIKeyName = \"Azure OpenAI\"\n let azureOpenAIAPIKey = defaults.deprecatedValue(for: \\.azureOpenAIAPIKey)\n if !azureOpenAIAPIKey.isEmpty {\n try keychain.update(azureOpenAIAPIKey, key: azureOpenAIAPIKeyName)\n }\n \n defaults.set({\n let openAIModel = ChatGPTModel(rawValue: defaults.deprecatedValue(for: \\.chatGPTModel))\n \n let openAI = ChatModel(\n id: chatModelOpenAIId,\n name: \"OpenAI\",\n format: .openAI,\n info: .init(\n apiKeyName: openAIAPIKeyName,\n baseURL: defaults.deprecatedValue(for: \\.openAIBaseURL),\n maxTokens: openAIModel?.maxToken ?? defaults\n .deprecatedValue(for: \\.chatGPTMaxToken),\n modelName: openAIModel?.rawValue ?? defaults\n .deprecatedValue(for: \\.chatGPTModel)\n )\n )\n let azureOpenAI = ChatModel(\n id: chatModelAzureOpenAIId,\n name: \"Azure OpenAI\",\n format: .azureOpenAI,\n info: .init(\n apiKeyName: azureOpenAIAPIKeyName,\n baseURL: defaults.deprecatedValue(for: \\.azureOpenAIBaseURL),\n maxTokens: defaults.deprecatedValue(for: \\.chatGPTMaxToken),\n modelName: defaults\n .deprecatedValue(for: \\.azureChatGPTDeployment)\n )\n )\n \n return [openAI, azureOpenAI]\n }(), for: \\.chatModels)\n \n defaults.set({\n if defaults.deprecatedValue(for: \\.chatFeatureProvider) == .azureOpenAI {\n return chatModelAzureOpenAIId\n }\n return chatModelOpenAIId\n }(), for: \\.defaultChatFeatureChatModelId)\n \n defaults.set({\n let openAIModel = OpenAIEmbeddingModel(\n rawValue: defaults.deprecatedValue(for: \\.embeddingModel)\n )\n \n let openAI = EmbeddingModel(\n id: embeddingModelOpenAIId,\n name: \"OpenAI\",\n format: .openAI,\n info: .init(\n apiKeyName: openAIAPIKeyName,\n baseURL: defaults.deprecatedValue(for: \\.openAIBaseURL),\n maxTokens: openAIModel?.maxToken ?? 8191,\n modelName: openAIModel?.rawValue ?? defaults.deprecatedValue(for: \\.embeddingModel)\n )\n )\n \n let azureOpenAI = EmbeddingModel(\n id: embeddingModelAzureOpenAIId,\n name: \"Azure OpenAI\",\n format: .azureOpenAI,\n info: .init(\n apiKeyName: azureOpenAIAPIKeyName,\n baseURL: defaults.deprecatedValue(for: \\.azureOpenAIBaseURL),\n maxTokens: 8191,\n modelName: defaults\n .deprecatedValue(for: \\.azureEmbeddingDeployment)\n )\n )\n \n return [openAI, azureOpenAI]\n }(), for: \\.embeddingModels)\n \n defaults.set({\n if defaults.deprecatedValue(for: \\.embeddingFeatureProvider) == .azureOpenAI {\n return embeddingModelAzureOpenAIId\n }\n return embeddingModelOpenAIId\n }(), for: \\.defaultChatFeatureEmbeddingModelId)\n \n defaults.set(true, forKey: finishedMigrationKey)\n } catch {\n print(error.localizedDescription)\n throw error\n }\n}\n\n" }, { "path": "Core/Sources/ServiceUpdateMigration/ServiceUpdateMigrator.swift", "content": "import Configs\nimport Foundation\nimport Preferences\n\nextension UserDefaultPreferenceKeys {\n struct OldMigrationVersion: UserDefaultPreferenceKey {\n var defaultValue: String = \"0\"\n let key = \"OldMigrationVersion\"\n }\n\n var oldMigrationVersion: OldMigrationVersion { .init() }\n}\n\npublic struct ServiceUpdateMigrator {\n public init() {}\n\n public func migrate() async throws {\n let version = Bundle.main.object(forInfoDictionaryKey: \"CFBundleVersion\") as? String ?? \"0\"\n\n try await migrate(from: UserDefaults.shared.value(for: \\.oldMigrationVersion), to: version)\n UserDefaults.shared.set(version, for: \\.oldMigrationVersion)\n }\n\n func migrate(from oldVersion: String, to currentVersion: String) async throws {\n guard let old = Int(oldVersion), old != 0 else { return }\n if old <= 135 {\n try migrateFromLowerThanOrEqualToVersion135()\n }\n if old < 240 {\n try migrateTo240()\n }\n }\n}\n" }, { "path": "Core/Sources/SuggestionService/SuggestionService.swift", "content": "import BuiltinExtension\nimport CodeiumService\nimport enum CopilotForXcodeKit.SuggestionServiceError\nimport struct CopilotForXcodeKit.WorkspaceInfo\nimport Foundation\nimport GitHubCopilotService\nimport Preferences\nimport SuggestionBasic\nimport SuggestionProvider\nimport UserDefaultsObserver\nimport Workspace\n\n#if canImport(ProExtension)\nimport ProExtension\n#endif\n\npublic protocol SuggestionServiceType: SuggestionServiceProvider {}\n\npublic actor SuggestionService: SuggestionServiceType {\n public typealias Middleware = SuggestionServiceMiddleware\n public typealias EventHandler = SuggestionServiceEventHandler\n public var configuration: SuggestionProvider.SuggestionServiceConfiguration {\n get async { await suggestionProvider.configuration }\n }\n\n let middlewares: [Middleware]\n let eventHandlers: [EventHandler]\n\n let suggestionProvider: SuggestionServiceProvider\n\n public init(\n provider: any SuggestionServiceProvider,\n middlewares: [Middleware] = SuggestionServiceMiddlewareContainer.middlewares,\n eventHandlers: [EventHandler] = SuggestionServiceEventHandlerContainer.handlers\n ) {\n suggestionProvider = provider\n self.middlewares = middlewares\n self.eventHandlers = eventHandlers\n }\n\n public static func service(\n for serviceType: SuggestionFeatureProvider = UserDefaults.shared\n .value(for: \\.suggestionFeatureProvider)\n ) -> SuggestionService {\n #if canImport(ProExtension)\n if let provider = ProExtension.suggestionProviderFactory(serviceType) {\n return SuggestionService(provider: provider)\n }\n #endif\n\n switch serviceType {\n case .builtIn(.codeium):\n let provider = BuiltinExtensionSuggestionServiceProvider(\n extension: CodeiumExtension.self\n )\n return SuggestionService(provider: provider)\n case .builtIn(.gitHubCopilot), .extension:\n let provider = BuiltinExtensionSuggestionServiceProvider(\n extension: GitHubCopilotExtension.self\n )\n return SuggestionService(provider: provider)\n }\n }\n}\n\npublic extension SuggestionService {\n func getSuggestions(\n _ request: SuggestionRequest,\n workspaceInfo: CopilotForXcodeKit.WorkspaceInfo\n ) async throws -> [SuggestionBasic.CodeSuggestion] {\n do {\n var getSuggestion = suggestionProvider.getSuggestions(_:workspaceInfo:)\n let configuration = await configuration\n\n for middleware in middlewares.reversed() {\n getSuggestion = { [getSuggestion] request, workspaceInfo in\n try await middleware.getSuggestion(\n request,\n configuration: configuration,\n next: { [getSuggestion] request in\n try await getSuggestion(request, workspaceInfo)\n }\n )\n }\n }\n\n return try await getSuggestion(request, workspaceInfo)\n } catch let error as SuggestionServiceError {\n throw error\n } catch {\n throw SuggestionServiceError.silent(error)\n }\n }\n\n func notifyAccepted(\n _ suggestion: SuggestionBasic.CodeSuggestion,\n workspaceInfo: CopilotForXcodeKit.WorkspaceInfo\n ) async {\n eventHandlers.forEach { $0.didAccept(suggestion, workspaceInfo: workspaceInfo) }\n await suggestionProvider.notifyAccepted(suggestion, workspaceInfo: workspaceInfo)\n }\n\n func notifyRejected(\n _ suggestions: [SuggestionBasic.CodeSuggestion],\n workspaceInfo: CopilotForXcodeKit.WorkspaceInfo\n ) async {\n eventHandlers.forEach { $0.didReject(suggestions, workspaceInfo: workspaceInfo) }\n await suggestionProvider.notifyRejected(suggestions, workspaceInfo: workspaceInfo)\n }\n\n func cancelRequest(workspaceInfo: CopilotForXcodeKit.WorkspaceInfo) async {\n await suggestionProvider.cancelRequest(workspaceInfo: workspaceInfo)\n }\n}\n\n" }, { "path": "Core/Sources/SuggestionWidget/ChatPanelWindow.swift", "content": "import AppKit\nimport ChatTab\nimport ComposableArchitecture\nimport Foundation\nimport SwiftUI\n\nfinal class ChatPanelWindow: WidgetWindow {\n override var canBecomeKey: Bool { true }\n override var canBecomeMain: Bool { true }\n\n private let storeObserver = NSObject()\n private let store: StoreOf\n\n var minimizeWindow: () -> Void = {}\n\n var isDetached: Bool {\n store.withState { $0.isDetached }\n }\n\n override var defaultCollectionBehavior: NSWindow.CollectionBehavior {\n [\n .fullScreenAuxiliary,\n .transient,\n .fullScreenPrimary,\n .fullScreenAllowsTiling,\n ]\n }\n\n init(\n store: StoreOf,\n chatTabPool: ChatTabPool,\n minimizeWindow: @escaping () -> Void\n ) {\n self.store = store\n self.minimizeWindow = minimizeWindow\n super.init(\n contentRect: .init(x: 0, y: 0, width: 300, height: 400),\n styleMask: [.resizable, .titled, .miniaturizable, .fullSizeContentView, .closable],\n backing: .buffered,\n defer: false\n )\n\n titleVisibility = .hidden\n addTitlebarAccessoryViewController({\n let controller = NSTitlebarAccessoryViewController()\n let view = NSHostingView(rootView: ChatTitleBar(store: store))\n controller.view = view\n view.frame = .init(x: 0, y: 0, width: 100, height: 40)\n controller.layoutAttribute = .right\n return controller\n }())\n titlebarAppearsTransparent = true\n isReleasedWhenClosed = false\n level = widgetLevel(1)\n hasShadow = true\n contentView = NSHostingView(\n rootView: ChatWindowView(\n store: store,\n toggleVisibility: { [weak self] isDisplayed in\n guard let self else { return }\n self.isPanelDisplayed = isDisplayed\n }\n )\n .environment(\\.chatTabPool, chatTabPool)\n )\n setIsVisible(true)\n isPanelDisplayed = false\n\n var wasDetached = false\n storeObserver.observe { [weak self] in\n guard let self else { return }\n let isDetached = store.isDetached\n guard isDetached != wasDetached else { return }\n wasDetached = isDetached\n Task { @MainActor in\n if UserDefaults.shared.value(for: \\.disableFloatOnTopWhenTheChatPanelIsDetached) {\n self.setFloatOnTop(!isDetached)\n } else {\n self.setFloatOnTop(true)\n }\n }\n }\n }\n\n func centerInActiveSpaceIfNeeded() {\n guard !isOnActiveSpace else { return }\n center()\n }\n\n var isWindowHidden: Bool = false {\n didSet {\n alphaValue = isPanelDisplayed && !isWindowHidden ? 1 : 0\n }\n }\n\n var isPanelDisplayed: Bool = false {\n didSet {\n alphaValue = isPanelDisplayed && !isWindowHidden ? 1 : 0\n }\n }\n\n override var alphaValue: CGFloat {\n didSet {\n ignoresMouseEvents = alphaValue <= 0\n }\n }\n\n override func miniaturize(_: Any?) {\n minimizeWindow()\n }\n\n override func close() {\n store.send(.closeActiveTabClicked)\n }\n}\n\n" }, { "path": "Core/Sources/SuggestionWidget/ChatWindowView.swift", "content": "import ActiveApplicationMonitor\nimport AppKit\nimport ChatGPTChatTab\nimport ChatTab\nimport ComposableArchitecture\nimport SwiftUI\nimport SharedUIComponents\n\nprivate let r: Double = 8\n\nstruct ChatWindowView: View {\n let store: StoreOf\n let toggleVisibility: (Bool) -> Void\n\n var body: some View {\n WithPerceptionTracking {\n let _ = store.chatTabGroup.selectedTabId // force re-evaluation\n VStack(spacing: 0) {\n Rectangle().fill(.regularMaterial).frame(height: 28)\n\n Divider()\n\n ChatTabBar(store: store)\n .frame(height: 26)\n .clipped()\n\n Divider()\n\n ChatTabContainer(store: store)\n .frame(maxWidth: .infinity, maxHeight: .infinity)\n }\n .xcodeStyleFrame()\n .ignoresSafeArea(edges: .top)\n .onChange(of: store.isPanelDisplayed) { isDisplayed in\n toggleVisibility(isDisplayed)\n }\n .preferredColorScheme(store.colorScheme)\n }\n }\n}\n\nstruct ChatTitleBar: View {\n let store: StoreOf\n @State var isHovering = false\n\n var body: some View {\n WithPerceptionTracking {\n HStack(spacing: 6) {\n Button(action: {\n store.send(.closeActiveTabClicked)\n }) {\n EmptyView()\n }\n .opacity(0)\n .keyboardShortcut(\"w\", modifiers: [.command])\n\n Button(\n action: {\n store.send(.hideButtonClicked)\n }\n ) {\n Image(systemName: \"minus\")\n .foregroundStyle(.black.opacity(0.5))\n .font(Font.system(size: 8).weight(.heavy))\n }\n .opacity(0)\n .keyboardShortcut(\"m\", modifiers: [.command])\n\n Spacer()\n\n TrafficLightButton(\n isHovering: isHovering,\n isActive: store.isDetached,\n color: Color(nsColor: .systemCyan),\n action: {\n store.send(.toggleChatPanelDetachedButtonClicked)\n }\n ) {\n Image(systemName: \"pin.fill\")\n .foregroundStyle(.black.opacity(0.5))\n .font(Font.system(size: 6).weight(.black))\n .transformEffect(.init(translationX: 0, y: 0.5))\n }\n }\n .buttonStyle(.plain)\n .padding(.trailing, 8)\n .onHover(perform: { hovering in\n isHovering = hovering\n })\n }\n }\n\n struct TrafficLightButton: View {\n let isHovering: Bool\n let isActive: Bool\n let color: Color\n let action: () -> Void\n let icon: () -> Icon\n\n @Environment(\\.controlActiveState) var controlActiveState\n\n var body: some View {\n Button(action: {\n action()\n }) {\n Circle()\n .fill(\n controlActiveState == .key && isActive\n ? color\n : Color(nsColor: .separatorColor)\n )\n .frame(\n width: Style.trafficLightButtonSize,\n height: Style.trafficLightButtonSize\n )\n .overlay {\n Circle().stroke(lineWidth: 0.5).foregroundColor(.black.opacity(0.2))\n }\n .overlay {\n if isHovering {\n icon()\n }\n }\n }\n .focusable(false)\n }\n }\n}\n\nextension View {\n func hideScrollIndicator() -> some View {\n if #available(macOS 13.0, *) {\n return scrollIndicators(.hidden)\n } else {\n return self\n }\n }\n}\n\nstruct ChatTabBar: View {\n let store: StoreOf\n\n struct TabBarState: Equatable {\n var tabInfo: IdentifiedArray\n var selectedTabId: String\n }\n\n var body: some View {\n HStack(spacing: 0) {\n Divider()\n Tabs(store: store)\n CreateButton(store: store)\n }\n .background {\n Button(action: { store.send(.switchToNextTab) }) { EmptyView() }\n .opacity(0)\n .keyboardShortcut(\"]\", modifiers: [.command, .shift])\n Button(action: { store.send(.switchToPreviousTab) }) { EmptyView() }\n .opacity(0)\n .keyboardShortcut(\"[\", modifiers: [.command, .shift])\n }\n }\n\n struct Tabs: View {\n let store: StoreOf\n @State var draggingTabId: String?\n @Environment(\\.chatTabPool) var chatTabPool\n\n var body: some View {\n WithPerceptionTracking {\n let tabInfo = store.chatTabGroup.tabInfo\n let selectedTabId = store.chatTabGroup.selectedTabId\n ?? store.chatTabGroup.tabInfo.first?.id\n ?? \"\"\n ScrollViewReader { proxy in\n ScrollView(.horizontal) {\n HStack(spacing: 0) {\n ForEach(tabInfo, id: \\.id) { info in\n WithPerceptionTracking {\n if let tab = chatTabPool.getTab(of: info.id) {\n ChatTabBarButton(\n store: store,\n info: info,\n content: { tab.tabItem },\n icon: { tab.icon },\n isSelected: info.id == selectedTabId\n )\n .contextMenu {\n tab.menu\n }\n .id(info.id)\n .onDrag {\n draggingTabId = info.id\n return NSItemProvider(object: info.id as NSString)\n }\n .onDrop(\n of: [.text],\n delegate: ChatTabBarDropDelegate(\n store: store,\n tabs: tabInfo,\n itemId: info.id,\n draggingTabId: $draggingTabId\n )\n )\n\n } else {\n ChatTabBarButton(\n store: store,\n info: info,\n content: { Text(\"Not Found\") },\n icon: { Image(systemName: \"questionmark.diamond\") },\n isSelected: info.id == selectedTabId\n )\n }\n }\n }\n }\n }\n .hideScrollIndicator()\n .onChange(of: selectedTabId) { id in\n withAnimation(.easeInOut(duration: 0.2)) {\n proxy.scrollTo(id)\n }\n }\n }\n }\n }\n }\n\n struct CreateButton: View {\n let store: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n let collection = store.chatTabGroup.tabCollection\n Menu {\n ForEach(0..\n let tabs: IdentifiedArray\n let itemId: String\n @Binding var draggingTabId: String?\n\n func dropUpdated(info: DropInfo) -> DropProposal? {\n return DropProposal(operation: .move)\n }\n\n func performDrop(info: DropInfo) -> Bool {\n draggingTabId = nil\n return true\n }\n\n func dropEntered(info: DropInfo) {\n guard itemId != draggingTabId else { return }\n let from = tabs.firstIndex { $0.id == draggingTabId }\n let to = tabs.firstIndex { $0.id == itemId }\n guard let from, let to, from != to else { return }\n store.send(.moveChatTab(from: from, to: to))\n }\n}\n\nstruct ChatTabBarButton: View {\n let store: StoreOf\n let info: ChatTabInfo\n let content: () -> Content\n let icon: () -> Icon\n let isSelected: Bool\n @State var isHovered: Bool = false\n\n var body: some View {\n HStack(spacing: 0) {\n HStack(spacing: 4) {\n icon().foregroundColor(.secondary)\n content()\n }\n .font(.callout)\n .lineLimit(1)\n .frame(maxWidth: 120)\n .padding(.horizontal, 28)\n .contentShape(Rectangle())\n .onTapGesture {\n store.send(.tabClicked(id: info.id))\n }\n .overlay(alignment: .leading) {\n Button(action: {\n store.send(.closeTabButtonClicked(id: info.id))\n }) {\n Image(systemName: \"xmark\")\n .foregroundColor(.secondary)\n }\n .buttonStyle(.plain)\n .padding(2)\n .padding(.leading, 8)\n .opacity(isHovered ? 1 : 0)\n }\n .onHover { isHovered = $0 }\n .animation(.linear(duration: 0.1), value: isHovered)\n .animation(.linear(duration: 0.1), value: isSelected)\n\n Divider().padding(.vertical, 6)\n }\n .background(isSelected ? Color(nsColor: .selectedControlColor) : Color.clear)\n .frame(maxHeight: .infinity)\n }\n}\n\nstruct ChatTabContainer: View {\n let store: StoreOf\n @Environment(\\.chatTabPool) var chatTabPool\n\n var body: some View {\n WithPerceptionTracking {\n let tabInfo = store.chatTabGroup.tabInfo\n let selectedTabId = store.chatTabGroup.selectedTabId\n ?? store.chatTabGroup.tabInfo.first?.id\n ?? \"\"\n\n ZStack {\n if tabInfo.isEmpty {\n Text(\"Empty\")\n } else {\n ForEach(tabInfo) { tabInfo in\n if let tab = chatTabPool.getTab(of: tabInfo.id) {\n let isActive = tab.id == selectedTabId\n tab.body\n .opacity(isActive ? 1 : 0)\n .disabled(!isActive)\n .allowsHitTesting(isActive)\n .frame(maxWidth: .infinity, maxHeight: .infinity)\n // move it out of window\n .rotationEffect(\n isActive ? .zero : .degrees(90),\n anchor: .topLeading\n )\n } else {\n Text(\"404 Not Found\")\n }\n }\n }\n }\n }\n }\n}\n\nstruct CreateOtherChatTabMenuStyle: MenuStyle {\n func makeBody(configuration: Configuration) -> some View {\n Image(systemName: \"chevron.down\")\n .resizable()\n .frame(width: 7, height: 4)\n .frame(maxHeight: .infinity)\n .padding(.leading, 4)\n .padding(.trailing, 8)\n .foregroundColor(.secondary)\n }\n}\n\nstruct ChatWindowView_Previews: PreviewProvider {\n static let pool = ChatTabPool([\n \"2\": EmptyChatTab(id: \"2\"),\n \"3\": EmptyChatTab(id: \"3\"),\n \"4\": EmptyChatTab(id: \"4\"),\n \"5\": EmptyChatTab(id: \"5\"),\n \"6\": EmptyChatTab(id: \"6\"),\n \"7\": EmptyChatTab(id: \"7\"),\n ])\n\n static func createStore() -> StoreOf {\n StoreOf(\n initialState: .init(\n chatTabGroup: .init(\n tabInfo: [\n .init(id: \"2\", title: \"Empty-2\"),\n .init(id: \"3\", title: \"Empty-3\"),\n .init(id: \"4\", title: \"Empty-4\"),\n .init(id: \"5\", title: \"Empty-5\"),\n .init(id: \"6\", title: \"Empty-6\"),\n .init(id: \"7\", title: \"Empty-7\"),\n ] as IdentifiedArray,\n selectedTabId: \"2\"\n ),\n isPanelDisplayed: true\n ),\n reducer: { ChatPanel() }\n )\n }\n\n static var previews: some View {\n ChatWindowView(store: createStore(), toggleVisibility: { _ in })\n .xcodeStyleFrame()\n .padding()\n .environment(\\.chatTabPool, pool)\n }\n}\n\n" }, { "path": "Core/Sources/SuggestionWidget/FeatureReducers/ChatPanel.swift", "content": "import ActiveApplicationMonitor\nimport AppKit\nimport ChatTab\nimport ComposableArchitecture\nimport SwiftUI\n\npublic enum ChatTabBuilderCollection: Equatable {\n case folder(title: String, kinds: [ChatTabKind])\n case kind(ChatTabKind)\n}\n\npublic struct ChatTabKind: Equatable {\n public var builder: any ChatTabBuilder\n var title: String { builder.title }\n\n public init(_ builder: any ChatTabBuilder) {\n self.builder = builder\n }\n\n public static func == (lhs: Self, rhs: Self) -> Bool {\n lhs.title == rhs.title\n }\n}\n\n@Reducer\npublic struct ChatPanel {\n public struct ChatTabGroup: Equatable {\n public var tabInfo: IdentifiedArray\n public var tabCollection: [ChatTabBuilderCollection]\n public var selectedTabId: String?\n\n public var selectedTabInfo: ChatTabInfo? {\n guard let id = selectedTabId else { return tabInfo.first }\n return tabInfo[id: id]\n }\n\n init(\n tabInfo: IdentifiedArray = [],\n tabCollection: [ChatTabBuilderCollection] = [],\n selectedTabId: String? = nil\n ) {\n self.tabInfo = tabInfo\n self.tabCollection = tabCollection\n self.selectedTabId = selectedTabId\n }\n }\n\n @ObservableState\n public struct State: Equatable {\n public var chatTabGroup = ChatTabGroup()\n var colorScheme: ColorScheme = .light\n public internal(set) var isPanelDisplayed = false\n var isDetached = false\n var isFullScreen = false\n }\n\n public enum Action: Equatable {\n // Window\n case hideButtonClicked\n case closeActiveTabClicked\n case toggleChatPanelDetachedButtonClicked\n case detachChatPanel\n case attachChatPanel\n case enterFullScreen\n case exitFullScreen\n case presentChatPanel(forceDetach: Bool)\n\n // Tabs\n case updateChatTabInfo(IdentifiedArray)\n case createNewTapButtonHovered\n case closeTabButtonClicked(id: String)\n case createNewTapButtonClicked(kind: ChatTabKind?)\n case tabClicked(id: String)\n case appendAndSelectTab(ChatTabInfo)\n case switchToNextTab\n case switchToPreviousTab\n case moveChatTab(from: Int, to: Int)\n case focusActiveChatTab\n\n case chatTab(IdentifiedActionOf)\n }\n\n @Dependency(\\.chatTabPool) var chatTabPool\n @Dependency(\\.suggestionWidgetControllerDependency) var suggestionWidgetControllerDependency\n @Dependency(\\.xcodeInspector) var xcodeInspector\n @Dependency(\\.activatePreviousActiveXcode) var activatePreviouslyActiveXcode\n @Dependency(\\.activateThisApp) var activateExtensionService\n @Dependency(\\.chatTabBuilderCollection) var chatTabBuilderCollection\n\n @MainActor func toggleFullScreen() {\n let window = suggestionWidgetControllerDependency.windowsController?.windows\n .chatPanelWindow\n window?.toggleFullScreen(nil)\n }\n\n public var body: some ReducerOf {\n Reduce { state, action in\n switch action {\n case .hideButtonClicked:\n state.isPanelDisplayed = false\n\n if state.isFullScreen {\n return .run { _ in\n await MainActor.run { toggleFullScreen() }\n activatePreviouslyActiveXcode()\n }\n }\n\n return .run { _ in\n activatePreviouslyActiveXcode()\n }\n\n case .closeActiveTabClicked:\n if let id = state.chatTabGroup.selectedTabId {\n return .run { send in\n await send(.closeTabButtonClicked(id: id))\n }\n }\n\n state.isPanelDisplayed = false\n return .none\n\n case .toggleChatPanelDetachedButtonClicked:\n if state.isFullScreen, state.isDetached {\n return .run { send in\n await send(.attachChatPanel)\n }\n }\n\n state.isDetached.toggle()\n return .none\n\n case .detachChatPanel:\n state.isDetached = true\n return .none\n\n case .attachChatPanel:\n if state.isFullScreen {\n return .run { send in\n await MainActor.run { toggleFullScreen() }\n try await Task.sleep(nanoseconds: 1_000_000_000)\n await send(.attachChatPanel)\n }\n }\n\n state.isDetached = false\n return .none\n\n case .enterFullScreen:\n state.isFullScreen = true\n return .run { send in\n await send(.detachChatPanel)\n }\n\n case .exitFullScreen:\n state.isFullScreen = false\n return .none\n\n case let .presentChatPanel(forceDetach):\n if forceDetach {\n state.isDetached = true\n }\n state.isPanelDisplayed = true\n return .run { send in\n if forceDetach {\n await suggestionWidgetControllerDependency.windowsController?.windows\n .chatPanelWindow\n .centerInActiveSpaceIfNeeded()\n }\n await send(.focusActiveChatTab)\n }\n\n case let .updateChatTabInfo(chatTabInfo):\n let previousSelectedIndex = state.chatTabGroup.tabInfo\n .firstIndex(where: { $0.id == state.chatTabGroup.selectedTabId })\n state.chatTabGroup.tabInfo = chatTabInfo\n if !chatTabInfo.contains(where: { $0.id == state.chatTabGroup.selectedTabId }) {\n if let previousSelectedIndex {\n let proposedSelectedIndex = previousSelectedIndex - 1\n if proposedSelectedIndex >= 0,\n proposedSelectedIndex < chatTabInfo.endIndex\n {\n state.chatTabGroup.selectedTabId = chatTabInfo[proposedSelectedIndex].id\n } else {\n state.chatTabGroup.selectedTabId = chatTabInfo.first?.id\n }\n } else {\n state.chatTabGroup.selectedTabId = nil\n }\n }\n return .none\n\n case let .closeTabButtonClicked(id):\n let firstIndex = state.chatTabGroup.tabInfo.firstIndex { $0.id == id }\n let nextIndex = {\n guard let firstIndex else { return 0 }\n let nextIndex = firstIndex - 1\n return max(nextIndex, 0)\n }()\n state.chatTabGroup.tabInfo.removeAll { $0.id == id }\n chatTabPool.getTab(of: id)?.close()\n if state.chatTabGroup.tabInfo.isEmpty {\n state.isPanelDisplayed = false\n }\n if nextIndex < state.chatTabGroup.tabInfo.count {\n state.chatTabGroup.selectedTabId = state.chatTabGroup.tabInfo[nextIndex].id\n } else {\n state.chatTabGroup.selectedTabId = nil\n }\n return .none\n\n case .createNewTapButtonHovered:\n state.chatTabGroup.tabCollection = chatTabBuilderCollection()\n return .none\n\n case .createNewTapButtonClicked:\n return .none // handled elsewhere\n\n case let .tabClicked(id):\n guard state.chatTabGroup.tabInfo.contains(where: { $0.id == id }) else {\n state.chatTabGroup.selectedTabId = nil\n return .none\n }\n state.chatTabGroup.selectedTabId = id\n return .run { send in\n await send(.focusActiveChatTab)\n }\n\n case let .appendAndSelectTab(tab):\n guard !state.chatTabGroup.tabInfo.contains(where: { $0.id == tab.id })\n else { return .none }\n state.chatTabGroup.tabInfo.append(tab)\n state.chatTabGroup.selectedTabId = tab.id\n return .run { send in\n await send(.focusActiveChatTab)\n }\n\n case .switchToNextTab:\n let selectedId = state.chatTabGroup.selectedTabId\n guard let index = state.chatTabGroup.tabInfo\n .firstIndex(where: { $0.id == selectedId })\n else { return .none }\n let nextIndex = index + 1\n if nextIndex >= state.chatTabGroup.tabInfo.endIndex {\n return .none\n }\n let targetId = state.chatTabGroup.tabInfo[nextIndex].id\n state.chatTabGroup.selectedTabId = targetId\n return .run { send in\n await send(.focusActiveChatTab)\n }\n\n case .switchToPreviousTab:\n let selectedId = state.chatTabGroup.selectedTabId\n guard let index = state.chatTabGroup.tabInfo\n .firstIndex(where: { $0.id == selectedId })\n else { return .none }\n let previousIndex = index - 1\n if previousIndex < 0 || previousIndex >= state.chatTabGroup.tabInfo.endIndex {\n return .none\n }\n let targetId = state.chatTabGroup.tabInfo[previousIndex].id\n state.chatTabGroup.selectedTabId = targetId\n return .run { send in\n await send(.focusActiveChatTab)\n }\n\n case let .moveChatTab(from, to):\n guard from >= 0, from < state.chatTabGroup.tabInfo.endIndex, to >= 0,\n to <= state.chatTabGroup.tabInfo.endIndex\n else {\n return .none\n }\n let tab = state.chatTabGroup.tabInfo[from]\n state.chatTabGroup.tabInfo.remove(at: from)\n state.chatTabGroup.tabInfo.insert(tab, at: to)\n return .none\n\n case .focusActiveChatTab:\n let id = state.chatTabGroup.selectedTabInfo?.id\n guard let id else { return .none }\n return .run { send in\n await send(.chatTab(.element(id: id, action: .focus)))\n }\n\n case let .chatTab(.element(id, .close)):\n return .run { send in\n await send(.closeTabButtonClicked(id: id))\n }\n\n case .chatTab:\n return .none\n }\n }.forEach(\\.chatTabGroup.tabInfo, action: \\.chatTab) {\n ChatTabItem()\n }\n }\n}\n\n" }, { "path": "Core/Sources/SuggestionWidget/FeatureReducers/CircularWidget.swift", "content": "import ActiveApplicationMonitor\nimport ComposableArchitecture\nimport Preferences\nimport SuggestionBasic\nimport SwiftUI\n\n@Reducer\npublic struct CircularWidget {\n public struct IsProcessingCounter: Equatable {\n var expirationDate: TimeInterval\n }\n\n @ObservableState\n public struct State: Equatable {\n var isProcessingCounters = [IsProcessingCounter]()\n var isProcessing: Bool\n var isDisplayingContent: Bool\n var isContentEmpty: Bool\n var isChatPanelDetached: Bool\n var isChatOpen: Bool\n }\n\n public enum Action: Equatable {\n case widgetClicked\n case detachChatPanelToggleClicked\n case openChatButtonClicked\n case openModificationButtonClicked\n case runCustomCommandButtonClicked(CustomCommand)\n case markIsProcessing\n case endIsProcessing\n case _forceEndIsProcessing\n }\n\n struct CancelAutoEndIsProcessKey: Hashable {}\n\n @Dependency(\\.suggestionWidgetControllerDependency) var suggestionWidgetControllerDependency\n \n public var body: some ReducerOf {\n Reduce { state, action in\n switch action {\n case .detachChatPanelToggleClicked:\n return .none // handled elsewhere\n \n case .openChatButtonClicked:\n return .run { _ in\n suggestionWidgetControllerDependency.onOpenChatClicked()\n }\n \n case .openModificationButtonClicked:\n return .run { _ in\n suggestionWidgetControllerDependency.onOpenModificationButtonClicked()\n }\n \n case let .runCustomCommandButtonClicked(command):\n return .run { _ in\n suggestionWidgetControllerDependency.onCustomCommandClicked(command)\n }\n \n case .widgetClicked:\n return .none // handled elsewhere\n \n case .markIsProcessing:\n let deadline = Date().timeIntervalSince1970 + 20\n state.isProcessingCounters.append(IsProcessingCounter(expirationDate: deadline))\n state.isProcessing = true\n return .run { send in\n try await Task.sleep(nanoseconds: 20 * 1_000_000_000)\n try Task.checkCancellation()\n await send(._forceEndIsProcessing)\n }.cancellable(id: CancelAutoEndIsProcessKey(), cancelInFlight: true)\n \n case .endIsProcessing:\n if !state.isProcessingCounters.isEmpty {\n state.isProcessingCounters.removeFirst()\n }\n state.isProcessingCounters\n .removeAll(where: { $0.expirationDate < Date().timeIntervalSince1970 })\n state.isProcessing = !state.isProcessingCounters.isEmpty\n return .none\n \n case ._forceEndIsProcessing:\n state.isProcessingCounters.removeAll()\n state.isProcessing = false\n return .none\n }\n }\n }\n}\n\n" }, { "path": "Core/Sources/SuggestionWidget/FeatureReducers/PromptToCodeGroup.swift", "content": "import ComposableArchitecture\nimport Foundation\nimport PromptToCodeService\nimport SuggestionBasic\nimport XcodeInspector\n\n@Reducer\npublic struct PromptToCodeGroup {\n @ObservableState\n public struct State {\n public var promptToCodes: IdentifiedArrayOf = []\n public var activeDocumentURL: PromptToCodePanel.State.ID? = XcodeInspector.shared\n .realtimeActiveDocumentURL\n public var selectedTabId: URL?\n public var activePromptToCode: PromptToCodePanel.State? {\n get {\n guard let selectedTabId else { return promptToCodes.first }\n return promptToCodes[id: selectedTabId] ?? promptToCodes.first\n }\n set {\n selectedTabId = newValue?.id\n if let id = selectedTabId {\n promptToCodes[id: id] = newValue\n }\n }\n }\n }\n\n public enum Action {\n /// Activate the prompt to code if it exists or create it if it doesn't\n case activateOrCreatePromptToCode(PromptToCodePanel.State)\n case createPromptToCode(PromptToCodePanel.State, sendImmediately: Bool)\n case updatePromptToCodeRange(\n id: PromptToCodePanel.State.ID,\n snippetId: UUID,\n range: CursorRange\n )\n case discardAcceptedPromptToCodeIfNotContinuous(id: PromptToCodePanel.State.ID)\n case updateActivePromptToCode(documentURL: URL)\n case discardExpiredPromptToCode(documentURLs: [URL])\n case tabClicked(id: URL)\n case closeTabButtonClicked(id: URL)\n case switchToNextTab\n case switchToPreviousTab\n case promptToCode(IdentifiedActionOf)\n case activePromptToCode(PromptToCodePanel.Action)\n }\n\n @Dependency(\\.activatePreviousActiveXcode) var activatePreviousActiveXcode\n\n public var body: some ReducerOf {\n Reduce { state, action in\n switch action {\n case let .activateOrCreatePromptToCode(s):\n if let promptToCode = state.activePromptToCode, s.id == promptToCode.id {\n state.selectedTabId = promptToCode.id\n return .run { send in\n await send(.promptToCode(.element(\n id: promptToCode.id,\n action: .focusOnTextField\n )))\n }\n }\n return .run { send in\n await send(.createPromptToCode(s, sendImmediately: false))\n }\n case let .createPromptToCode(newPromptToCode, sendImmediately):\n var newPromptToCode = newPromptToCode\n newPromptToCode.isActiveDocument = newPromptToCode.id == state.activeDocumentURL\n state.promptToCodes.append(newPromptToCode)\n state.selectedTabId = newPromptToCode.id\n return .run { [newPromptToCode] send in\n if sendImmediately,\n !newPromptToCode.contextInputController.instruction.string.isEmpty\n {\n await send(.promptToCode(.element(\n id: newPromptToCode.id,\n action: .modifyCodeButtonTapped\n )))\n }\n }.cancellable(\n id: PromptToCodePanel.CancellationKey.modifyCode(newPromptToCode.id),\n cancelInFlight: true\n )\n\n case let .updatePromptToCodeRange(id, snippetId, range):\n if let p = state.promptToCodes[id: id], p.promptToCodeState.isAttachedToTarget {\n state.promptToCodes[id: id]?.promptToCodeState.snippets[id: snippetId]?\n .attachedRange = range\n }\n return .none\n\n case let .discardAcceptedPromptToCodeIfNotContinuous(id):\n for itemId in state.promptToCodes.ids {\n if itemId == id, state.promptToCodes[id: itemId]?.clickedButton == .accept {\n state.promptToCodes.remove(id: itemId)\n } else {\n state.promptToCodes[id: itemId]?.clickedButton = nil\n }\n }\n return .none\n\n case let .updateActivePromptToCode(documentURL):\n state.activeDocumentURL = documentURL\n for index in state.promptToCodes.indices {\n state.promptToCodes[index].isActiveDocument =\n state.promptToCodes[index].id == documentURL\n }\n return .none\n\n case let .discardExpiredPromptToCode(documentURLs):\n for url in documentURLs {\n state.promptToCodes.remove(id: url)\n }\n return .none\n\n case let .tabClicked(id):\n state.selectedTabId = id\n return .none\n\n case let .closeTabButtonClicked(id):\n return .run { send in\n await send(.promptToCode(.element(\n id: id,\n action: .cancelButtonTapped\n )))\n }\n\n case .switchToNextTab:\n if let selectedTabId = state.selectedTabId,\n let index = state.promptToCodes.index(id: selectedTabId)\n {\n let nextIndex = (index + 1) % state.promptToCodes.count\n state.selectedTabId = state.promptToCodes[nextIndex].id\n }\n return .none\n\n case .switchToPreviousTab:\n if let selectedTabId = state.selectedTabId,\n let index = state.promptToCodes.index(id: selectedTabId)\n {\n let previousIndex = (index - 1 + state.promptToCodes.count) % state\n .promptToCodes.count\n state.selectedTabId = state.promptToCodes[previousIndex].id\n }\n return .none\n\n case .promptToCode:\n return .none\n\n case .activePromptToCode:\n return .none\n }\n }\n .ifLet(\\.activePromptToCode, action: \\.activePromptToCode) {\n PromptToCodePanel()\n }\n .forEach(\\.promptToCodes, action: \\.promptToCode, element: {\n PromptToCodePanel()\n })\n\n Reduce { state, action in\n switch action {\n case let .promptToCode(.element(id, .cancelButtonTapped)):\n state.promptToCodes.remove(id: id)\n let isEmpty = state.promptToCodes.isEmpty\n return .run { _ in\n if isEmpty {\n activatePreviousActiveXcode()\n }\n }\n case .activePromptToCode(.cancelButtonTapped):\n guard let id = state.selectedTabId else { return .none }\n state.promptToCodes.remove(id: id)\n let isEmpty = state.promptToCodes.isEmpty\n return .run { _ in\n if isEmpty {\n activatePreviousActiveXcode()\n }\n }\n default: return .none\n }\n }\n }\n}\n\n" }, { "path": "Core/Sources/SuggestionWidget/FeatureReducers/PromptToCodePanel.swift", "content": "import AppKit\nimport ChatBasic\nimport ComposableArchitecture\nimport CustomAsyncAlgorithms\nimport Dependencies\nimport Foundation\nimport ModificationBasic\nimport Preferences\nimport PromptToCodeCustomization\nimport PromptToCodeService\nimport SuggestionBasic\nimport XcodeInspector\n\n@Reducer\npublic struct PromptToCodePanel {\n @ObservableState\n public struct State: Identifiable {\n public enum FocusField: Equatable {\n case textField\n }\n\n public enum ClickedButton: Equatable {\n case accept\n case acceptAndContinue\n }\n\n @Shared public var promptToCodeState: ModificationState\n @ObservationStateIgnored\n public var contextInputController: PromptToCodeContextInputController\n\n public var id: URL { promptToCodeState.source.documentURL }\n\n public var commandName: String?\n public var isContinuous: Bool\n public var focusedField: FocusField? = .textField\n\n public var filename: String {\n promptToCodeState.source.documentURL.lastPathComponent\n }\n\n public var canRevert: Bool { !promptToCodeState.history.isEmpty }\n\n public var generateDescriptionRequirement: Bool\n\n public var clickedButton: ClickedButton?\n\n public var isActiveDocument: Bool = false\n\n public var snippetPanels: IdentifiedArrayOf {\n get {\n IdentifiedArrayOf(\n uniqueElements: promptToCodeState.snippets.map {\n PromptToCodeSnippetPanel.State(snippet: $0)\n }\n )\n }\n set {\n promptToCodeState.snippets = IdentifiedArrayOf(\n uniqueElements: newValue.map(\\.snippet)\n )\n }\n }\n\n public init(\n promptToCodeState: Shared,\n instruction: String?,\n commandName: String? = nil,\n isContinuous: Bool = false,\n generateDescriptionRequirement: Bool = UserDefaults.shared\n .value(for: \\.promptToCodeGenerateDescription)\n ) {\n _promptToCodeState = promptToCodeState\n self.isContinuous = isContinuous\n self.generateDescriptionRequirement = generateDescriptionRequirement\n self.commandName = commandName\n contextInputController = PromptToCodeCustomization\n .contextInputControllerFactory(promptToCodeState)\n focusedField = .textField\n contextInputController.instruction = instruction\n .map(NSAttributedString.init(string:)) ?? .init()\n }\n }\n\n public enum Action: BindableAction {\n case binding(BindingAction)\n case focusOnTextField\n case selectionRangeToggleTapped\n case modifyCodeButtonTapped\n case revertButtonTapped\n case stopRespondingButtonTapped\n case modifyCodeFinished\n case modifyCodeCancelled\n case cancelButtonTapped\n case acceptButtonTapped\n case acceptAndContinueButtonTapped\n case revealFileButtonClicked\n case statusUpdated([String])\n case referencesUpdated([ChatMessage.Reference])\n case snippetPanel(IdentifiedActionOf)\n }\n\n @Dependency(\\.commandHandler) var commandHandler\n @Dependency(\\.activateThisApp) var activateThisApp\n @Dependency(\\.activatePreviousActiveXcode) var activatePreviousActiveXcode\n\n enum CancellationKey: Hashable {\n case modifyCode(State.ID)\n }\n\n public var body: some ReducerOf {\n BindingReducer()\n\n Reduce { state, action in\n switch action {\n case .binding:\n return .none\n\n case .snippetPanel:\n return .none\n\n case .focusOnTextField:\n state.focusedField = .textField\n return .none\n\n case .selectionRangeToggleTapped:\n state.promptToCodeState.isAttachedToTarget.toggle()\n return .none\n\n case .modifyCodeButtonTapped:\n guard !state.promptToCodeState.isGenerating else { return .none }\n let copiedState = state\n let contextInputController = state.contextInputController\n state.promptToCodeState.isGenerating = true\n state.promptToCodeState.pushHistory(instruction: .init(\n attributedString: contextInputController.instruction\n ))\n state.promptToCodeState.references = []\n let snippets = state.promptToCodeState.snippets\n\n return .run { send in\n do {\n let context = await contextInputController.resolveContext(\n forDocumentURL: copiedState.promptToCodeState.source.documentURL,\n onStatusChange: { await send(.statusUpdated($0)) }\n )\n await send(.referencesUpdated(context.references))\n let agentFactory = context.agent ?? { SimpleModificationAgent() }\n _ = try await withThrowingTaskGroup(of: Void.self) { group in\n for (index, snippet) in snippets.enumerated() {\n if index > 3 { // at most 3 at a time\n _ = try await group.next()\n }\n group.addTask {\n try await Task\n .sleep(nanoseconds: UInt64.random(in: 0...1_000_000_000))\n let agent = agentFactory()\n let stream = agent.send(.init(\n code: snippet.originalCode,\n requirement: context.instruction,\n source: .init(\n language: copiedState.promptToCodeState.source.language,\n documentURL: copiedState.promptToCodeState.source\n .documentURL,\n projectRootURL: copiedState.promptToCodeState.source\n .projectRootURL,\n content: copiedState.promptToCodeState.source.content,\n lines: copiedState.promptToCodeState.source.lines\n ),\n isDetached: !copiedState.promptToCodeState\n .isAttachedToTarget,\n extraSystemPrompt: copiedState.promptToCodeState\n .extraSystemPrompt,\n range: snippet.attachedRange,\n references: context.references,\n topics: context.topics\n )).map {\n switch $0 {\n case let .code(code):\n return (code: code, description: \"\")\n case let .explanation(explanation):\n return (code: \"\", description: explanation)\n }\n }.timedDebounce(for: 0.4) { lhs, rhs in\n (\n code: lhs.code + rhs.code,\n description: lhs.description + rhs.description\n )\n }\n\n do {\n for try await response in stream {\n try Task.checkCancellation()\n await send(.snippetPanel(.element(\n id: snippet.id,\n action: .modifyCodeChunkReceived(\n code: response.code,\n description: response.description\n )\n )))\n }\n await send(.snippetPanel(.element(\n id: snippet.id,\n action: .modifyCodeFinished\n )))\n } catch is CancellationError {\n await send(.snippetPanel(.element(\n id: snippet.id,\n action: .modifyCodeFinished\n )))\n throw CancellationError()\n } catch {\n if (error as NSError).code == NSURLErrorCancelled {\n await send(.snippetPanel(.element(\n id: snippet.id,\n action: .modifyCodeFinished\n )))\n throw CancellationError()\n }\n await send(.snippetPanel(.element(\n id: snippet.id,\n action: .modifyCodeFailed(\n error: error.localizedDescription\n )\n )))\n }\n }\n }\n\n try await group.waitForAll()\n }\n\n await send(.modifyCodeFinished)\n } catch is CancellationError {\n try Task.checkCancellation()\n await send(.modifyCodeCancelled)\n } catch {\n await send(.modifyCodeFinished)\n }\n }.cancellable(id: CancellationKey.modifyCode(state.id), cancelInFlight: true)\n\n case .revertButtonTapped:\n if let instruction = state.promptToCodeState.popHistory() {\n state.contextInputController.instruction = instruction\n }\n return .none\n\n case .stopRespondingButtonTapped:\n state.promptToCodeState.isGenerating = false\n state.promptToCodeState.status = []\n return .cancel(id: CancellationKey.modifyCode(state.id))\n\n case .modifyCodeFinished:\n state.contextInputController.instruction = .init(\"\")\n state.promptToCodeState.isGenerating = false\n state.promptToCodeState.status = []\n\n if state.promptToCodeState.snippets.allSatisfy({ snippet in\n snippet.modifiedCode.isEmpty && snippet.description.isEmpty && snippet\n .error == nil\n }) {\n // if both code and description are empty, we treat it as failed\n return .run { send in\n await send(.revertButtonTapped)\n }\n }\n return .none\n\n case .modifyCodeCancelled:\n state.promptToCodeState.isGenerating = false\n return .none\n\n case .cancelButtonTapped:\n return .cancel(id: CancellationKey.modifyCode(state.id))\n\n case .acceptButtonTapped:\n state.clickedButton = .accept\n return .run { _ in\n await commandHandler.acceptModification()\n activatePreviousActiveXcode()\n }\n\n case .acceptAndContinueButtonTapped:\n state.clickedButton = .acceptAndContinue\n return .run { _ in\n await commandHandler.acceptModification()\n activateThisApp()\n }\n\n case .revealFileButtonClicked:\n let url = state.promptToCodeState.source.documentURL\n let startLine = state.snippetPanels.first?.snippet.attachedRange.start.line ?? 0\n return .run { _ in\n await commandHandler.presentFile(at: url, line: startLine)\n }\n\n case let .statusUpdated(status):\n state.promptToCodeState.status = status\n return .none\n\n case let .referencesUpdated(references):\n state.promptToCodeState.references = references\n return .none\n }\n }\n\n Reduce { _, _ in .none }.forEach(\\.snippetPanels, action: \\.snippetPanel) {\n PromptToCodeSnippetPanel()\n }\n }\n}\n\n@Reducer\npublic struct PromptToCodeSnippetPanel {\n @ObservableState\n public struct State: Identifiable {\n public var id: UUID { snippet.id }\n var snippet: ModificationSnippet\n }\n\n public enum Action {\n case modifyCodeFinished\n case modifyCodeChunkReceived(code: String, description: String)\n case modifyCodeFailed(error: String)\n case copyCodeButtonTapped\n }\n\n public var body: some ReducerOf {\n Reduce { state, action in\n switch action {\n case .modifyCodeFinished:\n return .none\n\n case let .modifyCodeChunkReceived(code, description):\n state.snippet.modifiedCode += code\n state.snippet.description += description\n return .none\n\n case let .modifyCodeFailed(error):\n state.snippet.error = error\n return .none\n\n case .copyCodeButtonTapped:\n NSPasteboard.general.clearContents()\n NSPasteboard.general.setString(state.snippet.modifiedCode, forType: .string)\n return .none\n }\n }\n }\n}\n\nfinal class DefaultPromptToCodeContextInputControllerDelegate: PromptToCodeContextInputControllerDelegate {\n let store: StoreOf\n\n init(store: StoreOf) {\n self.store = store\n }\n\n func modifyCodeButtonClicked() {\n Task {\n await store.send(.modifyCodeButtonTapped)\n }\n }\n}\n\n" }, { "path": "Core/Sources/SuggestionWidget/FeatureReducers/SharedPanel.swift", "content": "import ComposableArchitecture\nimport Preferences\nimport SwiftUI\n\n@Reducer\npublic struct SharedPanel {\n public struct Content {\n public var promptToCodeGroup = PromptToCodeGroup.State()\n var suggestion: PresentingCodeSuggestion?\n var error: String?\n }\n\n @ObservableState\n public struct State {\n var content: Content = .init()\n var colorScheme: ColorScheme = .light\n var alignTopToAnchor = false\n var isPanelDisplayed: Bool = false\n var isEmpty: Bool {\n if content.error != nil { return false }\n if !content.promptToCodeGroup.promptToCodes.isEmpty { return false }\n if content.suggestion != nil,\n UserDefaults.shared\n .value(for: \\.suggestionPresentationMode) == .floatingWidget { return false }\n return true\n }\n\n var opacity: Double {\n guard isPanelDisplayed else { return 0 }\n guard !isEmpty else { return 0 }\n return 1\n }\n }\n\n public enum Action {\n case errorMessageCloseButtonTapped\n case promptToCodeGroup(PromptToCodeGroup.Action)\n }\n\n public var body: some ReducerOf {\n Scope(state: \\.content.promptToCodeGroup, action: \\.promptToCodeGroup) {\n PromptToCodeGroup()\n }\n\n Reduce { state, action in\n switch action {\n case .errorMessageCloseButtonTapped:\n state.content.error = nil\n return .none\n case .promptToCodeGroup:\n return .none\n }\n }\n }\n}\n\n" }, { "path": "Core/Sources/SuggestionWidget/FeatureReducers/SuggestionPanel.swift", "content": "import ComposableArchitecture\nimport Foundation\nimport SwiftUI\n\n@Reducer\npublic struct SuggestionPanel {\n @ObservableState\n public struct State: Equatable {\n var content: PresentingCodeSuggestion?\n var colorScheme: ColorScheme = .light\n var alignTopToAnchor = false\n var isPanelDisplayed: Bool = false\n var isPanelOutOfFrame: Bool = false\n var opacity: Double {\n guard isPanelDisplayed else { return 0 }\n if isPanelOutOfFrame { return 0 }\n guard content != nil else { return 0 }\n return 1\n }\n }\n\n public enum Action: Equatable {\n case noAction\n }\n\n public var body: some ReducerOf {\n Reduce { _, _ in .none }\n }\n}\n" }, { "path": "Core/Sources/SuggestionWidget/FeatureReducers/ToastPanel.swift", "content": "import ComposableArchitecture\nimport Preferences\nimport SwiftUI\nimport Toast\n\n@Reducer\npublic struct ToastPanel {\n @ObservableState\n public struct State: Equatable {\n var toast: Toast.State = .init()\n var colorScheme: ColorScheme = .light\n var alignTopToAnchor = false\n }\n \n public enum Action: Equatable {\n case start\n case toast(Toast.Action)\n }\n \n public var body: some ReducerOf {\n Scope(state: \\.toast, action: \\.toast) {\n Toast()\n }\n \n Reduce { state, action in\n switch action {\n case .start:\n return .run { send in\n await send(.toast(.start))\n }\n case .toast:\n return .none\n }\n }\n }\n}\n" }, { "path": "Core/Sources/SuggestionWidget/FeatureReducers/Widget.swift", "content": "import ActiveApplicationMonitor\nimport AppActivator\nimport AsyncAlgorithms\nimport ComposableArchitecture\nimport Foundation\nimport Logger\nimport Preferences\nimport SwiftUI\nimport Toast\nimport XcodeInspector\n\n@Reducer\npublic struct Widget {\n public struct WindowState: Equatable {\n var alphaValue: Double = 0\n var frame: CGRect = .zero\n }\n\n public enum WindowCanBecomeKey: Equatable {\n case sharedPanel\n case chatPanel\n }\n\n @ObservableState\n public struct State {\n var focusingDocumentURL: URL?\n public var colorScheme: ColorScheme = .light\n\n var toastPanel = ToastPanel.State()\n\n // MARK: Panels\n\n public var panelState = WidgetPanel.State()\n\n // MARK: ChatPanel\n\n public var chatPanelState = ChatPanel.State()\n\n // MARK: CircularWidget\n\n public struct CircularWidgetState: Equatable {\n var isProcessingCounters = [CircularWidget.IsProcessingCounter]()\n var isProcessing: Bool = false\n }\n\n public var circularWidgetState = CircularWidgetState()\n var _internalCircularWidgetState: CircularWidget.State {\n get {\n .init(\n isProcessingCounters: circularWidgetState.isProcessingCounters,\n isProcessing: circularWidgetState.isProcessing,\n isDisplayingContent: {\n if chatPanelState.isPanelDisplayed {\n return true\n }\n if panelState.sharedPanelState.isPanelDisplayed,\n !panelState.sharedPanelState.isEmpty\n {\n return true\n }\n if panelState.suggestionPanelState.isPanelDisplayed,\n panelState.suggestionPanelState.content != nil\n {\n return true\n }\n return false\n }(),\n isContentEmpty: chatPanelState.chatTabGroup.tabInfo.isEmpty\n && panelState.sharedPanelState.isEmpty,\n isChatPanelDetached: chatPanelState.isDetached,\n isChatOpen: chatPanelState.isPanelDisplayed\n )\n }\n set {\n circularWidgetState = .init(\n isProcessingCounters: newValue.isProcessingCounters,\n isProcessing: newValue.isProcessing\n )\n }\n }\n\n public init() {}\n }\n\n private enum CancelID {\n case observeActiveApplicationChange\n case observeCompletionPanelChange\n case observeWindowChange\n case observeEditorChange\n case observeUserDefaults\n }\n\n public enum Action {\n case startup\n case observeActiveApplicationChange\n case observeColorSchemeChange\n\n case updateActiveApplication\n case updateColorScheme\n\n case updatePanelStateToMatch(WidgetLocation)\n case updateFocusingDocumentURL\n case setFocusingDocumentURL(to: URL?)\n case updateKeyWindow(WindowCanBecomeKey)\n\n case toastPanel(ToastPanel.Action)\n case panel(WidgetPanel.Action)\n case chatPanel(ChatPanel.Action)\n case circularWidget(CircularWidget.Action)\n }\n\n var windowsController: WidgetWindowsController? {\n suggestionWidgetControllerDependency.windowsController\n }\n\n @Dependency(\\.suggestionWidgetUserDefaultsObservers) var userDefaultsObservers\n @Dependency(\\.suggestionWidgetControllerDependency) var suggestionWidgetControllerDependency\n @Dependency(\\.xcodeInspector) var xcodeInspector\n @Dependency(\\.mainQueue) var mainQueue\n @Dependency(\\.activateThisApp) var activateThisApp\n @Dependency(\\.activatePreviousActiveApp) var activatePreviousActiveApp\n\n public enum DebounceKey: Hashable {\n case updateWindowOpacity\n }\n\n public init() {}\n\n public var body: some ReducerOf {\n Scope(state: \\.toastPanel, action: \\.toastPanel) {\n ToastPanel()\n }\n\n Scope(state: \\._internalCircularWidgetState, action: \\.circularWidget) {\n CircularWidget()\n }\n\n Reduce { state, action in\n switch action {\n case .circularWidget(.detachChatPanelToggleClicked):\n return .run { send in\n await send(.chatPanel(.toggleChatPanelDetachedButtonClicked))\n }\n\n case .circularWidget(.widgetClicked):\n let wasDisplayingContent = state._internalCircularWidgetState.isDisplayingContent\n if wasDisplayingContent {\n state.panelState.sharedPanelState.isPanelDisplayed = false\n state.panelState.suggestionPanelState.isPanelDisplayed = false\n state.chatPanelState.isPanelDisplayed = false\n } else {\n state.panelState.sharedPanelState.isPanelDisplayed = true\n state.panelState.suggestionPanelState.isPanelDisplayed = true\n state.chatPanelState.isPanelDisplayed = true\n }\n\n let isDisplayingContent = state._internalCircularWidgetState.isDisplayingContent\n let hasChat = state.chatPanelState.chatTabGroup.selectedTabInfo != nil\n let hasPromptToCode = state.panelState.sharedPanelState.content\n .promptToCodeGroup.activePromptToCode != nil\n\n return .run { send in\n if isDisplayingContent {\n if hasPromptToCode {\n await send(.updateKeyWindow(.sharedPanel))\n } else if hasChat {\n await send(.updateKeyWindow(.chatPanel))\n }\n await send(.chatPanel(.focusActiveChatTab))\n }\n\n if isDisplayingContent, !(await NSApplication.shared.isActive) {\n activateThisApp()\n } else if !isDisplayingContent {\n activatePreviousActiveApp()\n }\n }\n\n default: return .none\n }\n }\n\n Scope(state: \\.panelState, action: \\.panel) {\n WidgetPanel()\n }\n\n Scope(state: \\.chatPanelState, action: \\.chatPanel) {\n ChatPanel()\n }\n\n Reduce { state, action in\n switch action {\n case .chatPanel(.presentChatPanel):\n let isDetached = state.chatPanelState.isDetached\n return .run { _ in\n await windowsController?.updateWindowLocation(\n animated: false,\n immediately: false\n )\n await windowsController?.updateWindowOpacity(immediately: false)\n if isDetached {\n Task { @MainActor in\n windowsController?.windows.chatPanelWindow.isWindowHidden = false\n }\n }\n }\n\n case .chatPanel(.toggleChatPanelDetachedButtonClicked):\n let isDetached = state.chatPanelState.isDetached\n return .run { _ in\n await windowsController?.updateWindowLocation(\n animated: !isDetached,\n immediately: false\n )\n await windowsController?.updateWindowOpacity(immediately: false)\n }\n default: return .none\n }\n }\n\n Reduce { state, action in\n switch action {\n case .startup:\n return .merge(\n .run { send in\n await send(.toastPanel(.start))\n await send(.observeActiveApplicationChange)\n await send(.observeColorSchemeChange)\n }\n )\n\n case .observeActiveApplicationChange:\n return .run { send in\n let stream = AsyncStream { continuation in\n let task = Task {\n let notifications = NotificationCenter.default\n .notifications(named: .activeApplicationDidChange)\n for await _ in notifications {\n try Task.checkCancellation()\n if let app = await XcodeInspector.shared.activeApplication {\n continuation.yield(app)\n }\n }\n }\n continuation.onTermination = { _ in\n task.cancel()\n }\n }\n\n var previousAppIdentifier: pid_t?\n for await app in stream {\n try Task.checkCancellation()\n if app.processIdentifier != previousAppIdentifier {\n await send(.updateActiveApplication)\n }\n previousAppIdentifier = app.processIdentifier\n }\n }.cancellable(id: CancelID.observeActiveApplicationChange, cancelInFlight: true)\n\n case .observeColorSchemeChange:\n return .run { send in\n await send(.updateColorScheme)\n let stream = AsyncStream { continuation in\n userDefaultsObservers.colorSchemeChangeObserver.onChange = {\n continuation.yield()\n }\n\n userDefaultsObservers.systemColorSchemeChangeObserver.onChange = {\n continuation.yield()\n }\n\n continuation.onTermination = { _ in\n userDefaultsObservers.colorSchemeChangeObserver.onChange = {}\n userDefaultsObservers.systemColorSchemeChangeObserver.onChange = {}\n }\n }\n\n for await _ in stream {\n try Task.checkCancellation()\n await send(.updateColorScheme)\n }\n }.cancellable(id: CancelID.observeUserDefaults, cancelInFlight: true)\n\n case .updateActiveApplication:\n return .none\n\n case .updateColorScheme:\n let widgetColorScheme = UserDefaults.shared.value(for: \\.widgetColorScheme)\n let systemColorScheme: ColorScheme = NSApp.effectiveAppearance.name == .darkAqua\n ? .dark\n : .light\n\n let scheme: ColorScheme = {\n switch (widgetColorScheme, systemColorScheme) {\n case (.system, .dark), (.dark, _):\n return .dark\n case (.system, .light), (.light, _):\n return .light\n case (.system, _):\n return .light\n }\n }()\n\n state.colorScheme = scheme\n state.toastPanel.colorScheme = scheme\n state.panelState.sharedPanelState.colorScheme = scheme\n state.panelState.suggestionPanelState.colorScheme = scheme\n state.chatPanelState.colorScheme = scheme\n return .none\n\n case .updateFocusingDocumentURL:\n return .run { send in\n await send(.setFocusingDocumentURL(\n to: xcodeInspector.realtimeActiveDocumentURL\n ))\n }\n\n case let .setFocusingDocumentURL(url):\n state.focusingDocumentURL = url\n return .none\n\n case let .updatePanelStateToMatch(widgetLocation):\n state.panelState.sharedPanelState.alignTopToAnchor = widgetLocation\n .defaultPanelLocation\n .alignPanelTop\n\n if let suggestionPanelLocation = widgetLocation.suggestionPanelLocation {\n state.panelState.suggestionPanelState.isPanelOutOfFrame = false\n state.panelState.suggestionPanelState\n .alignTopToAnchor = suggestionPanelLocation\n .alignPanelTop\n } else {\n state.panelState.suggestionPanelState.isPanelOutOfFrame = true\n }\n\n state.toastPanel.alignTopToAnchor = widgetLocation\n .defaultPanelLocation\n .alignPanelTop\n\n return .none\n\n case let .updateKeyWindow(window):\n return .run { _ in\n await MainActor.run {\n switch window {\n case .chatPanel:\n windowsController?.windows.chatPanelWindow\n .makeKeyAndOrderFront(nil)\n case .sharedPanel:\n windowsController?.windows.sharedPanelWindow\n .makeKeyAndOrderFront(nil)\n }\n }\n }\n\n case .toastPanel:\n return .none\n\n case .circularWidget:\n return .none\n\n case .panel:\n return .none\n\n case .chatPanel:\n return .none\n }\n }\n }\n}\n\n" }, { "path": "Core/Sources/SuggestionWidget/FeatureReducers/WidgetPanel.swift", "content": "import AppKit\nimport ComposableArchitecture\nimport Foundation\n\n@Reducer\npublic struct WidgetPanel {\n @ObservableState\n public struct State {\n public var content: SharedPanel.Content {\n get { sharedPanelState.content }\n set {\n sharedPanelState.content = newValue\n suggestionPanelState.content = newValue.suggestion\n }\n }\n\n // MARK: SharedPanel\n\n var sharedPanelState = SharedPanel.State()\n\n // MARK: SuggestionPanel\n\n var suggestionPanelState = SuggestionPanel.State()\n }\n\n public enum Action {\n case presentSuggestion\n case presentSuggestionProvider(PresentingCodeSuggestion, displayContent: Bool)\n case presentError(String)\n case presentPromptToCode(PromptToCodePanel.State)\n case displayPanelContent\n case discardSuggestion\n case removeDisplayedContent\n case switchToAnotherEditorAndUpdateContent\n\n case sharedPanel(SharedPanel.Action)\n case suggestionPanel(SuggestionPanel.Action)\n }\n\n @Dependency(\\.suggestionWidgetControllerDependency) var suggestionWidgetControllerDependency\n @Dependency(\\.xcodeInspector) var xcodeInspector\n @Dependency(\\.activateThisApp) var activateThisApp\n var windows: WidgetWindows? { suggestionWidgetControllerDependency.windowsController?.windows }\n\n public var body: some ReducerOf {\n Scope(state: \\.suggestionPanelState, action: \\.suggestionPanel) {\n SuggestionPanel()\n }\n\n Scope(state: \\.sharedPanelState, action: \\.sharedPanel) {\n SharedPanel()\n }\n\n Reduce { state, action in\n switch action {\n case .presentSuggestion:\n return .run { send in\n guard let fileURL = await xcodeInspector.activeDocumentURL,\n let provider = await fetchSuggestionProvider(fileURL: fileURL)\n else { return }\n await send(.presentSuggestionProvider(provider, displayContent: true))\n }\n\n case let .presentSuggestionProvider(provider, displayContent):\n state.content.suggestion = provider\n if displayContent {\n return .run { send in\n await send(.displayPanelContent)\n }.animation(.easeInOut(duration: 0.2))\n }\n return .none\n\n case let .presentError(errorDescription):\n state.content.error = errorDescription\n return .run { send in\n await send(.displayPanelContent)\n }.animation(.easeInOut(duration: 0.2))\n\n case let .presentPromptToCode(initialState):\n return .run { send in\n await send(.sharedPanel(.promptToCodeGroup(.createPromptToCode(\n initialState,\n sendImmediately: true\n ))))\n }\n\n case .displayPanelContent:\n if !state.sharedPanelState.isEmpty {\n state.sharedPanelState.isPanelDisplayed = true\n }\n\n if state.suggestionPanelState.content != nil {\n state.suggestionPanelState.isPanelDisplayed = true\n }\n\n return .none\n\n case .discardSuggestion:\n state.content.suggestion = nil\n return .none\n\n case .switchToAnotherEditorAndUpdateContent:\n return .run { send in\n guard let fileURL = xcodeInspector.realtimeActiveDocumentURL\n else { return }\n\n await send(.sharedPanel(\n .promptToCodeGroup(\n .updateActivePromptToCode(documentURL: fileURL)\n )\n ))\n }\n\n case .removeDisplayedContent:\n state.content.error = nil\n state.content.suggestion = nil\n return .none\n\n case .sharedPanel(.promptToCodeGroup(.activateOrCreatePromptToCode)),\n .sharedPanel(.promptToCodeGroup(.createPromptToCode)):\n let hasPromptToCode = !state.content.promptToCodeGroup.promptToCodes.isEmpty\n return .run { send in\n await send(.displayPanelContent)\n\n if hasPromptToCode {\n activateThisApp()\n await MainActor.run {\n windows?.sharedPanelWindow.makeKey()\n }\n }\n }.animation(.easeInOut(duration: 0.2))\n\n case .sharedPanel:\n return .none\n\n case .suggestionPanel:\n return .none\n }\n }\n }\n\n func fetchSuggestionProvider(fileURL: URL) async -> PresentingCodeSuggestion? {\n guard let provider = await suggestionWidgetControllerDependency\n .suggestionWidgetDataSource?\n .suggestionForFile(at: fileURL) else { return nil }\n return provider\n }\n}\n\n" }, { "path": "Core/Sources/SuggestionWidget/ModuleDependency.swift", "content": "import ActiveApplicationMonitor\nimport AppKit\nimport ChatTab\nimport ComposableArchitecture\nimport Dependencies\nimport Foundation\nimport Preferences\nimport SwiftUI\nimport UserDefaultsObserver\nimport XcodeInspector\n\npublic final class SuggestionWidgetControllerDependency {\n public var suggestionWidgetDataSource: SuggestionWidgetDataSource?\n public var onOpenChatClicked: () -> Void = {}\n public var onOpenModificationButtonClicked: () -> Void = {}\n public var onCustomCommandClicked: (CustomCommand) -> Void = { _ in }\n var windowsController: WidgetWindowsController?\n\n public init() {}\n}\n\npublic final class WidgetUserDefaultsObservers {\n let presentationModeChangeObserver = UserDefaultsObserver(\n object: UserDefaults.shared,\n forKeyPaths: [\n UserDefaultPreferenceKeys().suggestionPresentationMode.key,\n ], context: nil\n )\n let colorSchemeChangeObserver = UserDefaultsObserver(\n object: UserDefaults.shared, forKeyPaths: [\n UserDefaultPreferenceKeys().widgetColorScheme.key,\n ], context: nil\n )\n let systemColorSchemeChangeObserver = UserDefaultsObserver(\n object: UserDefaults.standard, forKeyPaths: [\"AppleInterfaceStyle\"], context: nil\n )\n\n public init() {}\n}\n\nstruct SuggestionWidgetControllerDependencyKey: DependencyKey {\n static let liveValue = SuggestionWidgetControllerDependency()\n}\n\nstruct UserDefaultsDependencyKey: DependencyKey {\n static let liveValue = WidgetUserDefaultsObservers()\n}\n\nstruct XcodeInspectorKey: DependencyKey {\n static let liveValue = XcodeInspector.shared\n}\n\nstruct ActiveApplicationMonitorKey: DependencyKey {\n static let liveValue = ActiveApplicationMonitor.shared\n}\n\nstruct ChatTabBuilderCollectionKey: DependencyKey {\n static let liveValue: () -> [ChatTabBuilderCollection] = { [] }\n}\n\npublic extension DependencyValues {\n var suggestionWidgetControllerDependency: SuggestionWidgetControllerDependency {\n get { self[SuggestionWidgetControllerDependencyKey.self] }\n set { self[SuggestionWidgetControllerDependencyKey.self] = newValue }\n }\n\n var suggestionWidgetUserDefaultsObservers: WidgetUserDefaultsObservers {\n get { self[UserDefaultsDependencyKey.self] }\n set { self[UserDefaultsDependencyKey.self] = newValue }\n }\n\n var chatTabBuilderCollection: () -> [ChatTabBuilderCollection] {\n get { self[ChatTabBuilderCollectionKey.self] }\n set { self[ChatTabBuilderCollectionKey.self] = newValue }\n }\n}\n\nextension DependencyValues {\n var xcodeInspector: XcodeInspector {\n get { self[XcodeInspectorKey.self] }\n set { self[XcodeInspectorKey.self] = newValue }\n }\n\n var activeApplicationMonitor: ActiveApplicationMonitor {\n get { self[ActiveApplicationMonitorKey.self] }\n set { self[ActiveApplicationMonitorKey.self] = newValue }\n }\n}\n\n" }, { "path": "Core/Sources/SuggestionWidget/PromptToCodePanelGroupView.swift", "content": "import ComposableArchitecture\nimport Foundation\nimport SwiftUI\n\nstruct PromptToCodePanelGroupView: View {\n let store: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n VStack(spacing: 0) {\n PromptToCodeTabBar(store: store)\n .frame(height: 26)\n\n Divider()\n\n if let store = self.store.scope(\n state: \\.activePromptToCode,\n action: \\.activePromptToCode\n ) {\n PromptToCodePanelView(store: store)\n }\n }\n .background(.ultraThickMaterial)\n .xcodeStyleFrame()\n }\n }\n}\n\nstruct PromptToCodeTabBar: View {\n let store: StoreOf\n\n struct TabInfo: Equatable, Identifiable {\n var id: URL\n var tabTitle: String\n var isProcessing: Bool\n }\n\n var body: some View {\n HStack(spacing: 0) {\n Tabs(store: store)\n }\n .background {\n Button(action: { store.send(.switchToNextTab) }) { EmptyView() }\n .opacity(0)\n .keyboardShortcut(\"]\", modifiers: [.command, .shift])\n Button(action: { store.send(.switchToPreviousTab) }) { EmptyView() }\n .opacity(0)\n .keyboardShortcut(\"[\", modifiers: [.command, .shift])\n }\n }\n\n struct Tabs: View {\n let store: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n let tabInfo = store.promptToCodes.map {\n TabInfo(\n id: $0.id,\n tabTitle: $0.filename,\n isProcessing: $0.promptToCodeState.isGenerating\n )\n }\n let selectedTabId = store.selectedTabId\n ?? store.promptToCodes.first?.id\n\n ScrollViewReader { proxy in\n ScrollView(.horizontal) {\n HStack(spacing: 0) {\n ForEach(tabInfo) { info in\n WithPerceptionTracking {\n PromptToCodeTabBarButton(\n store: store,\n info: info,\n isSelected: info.id == store.selectedTabId\n )\n .id(info.id)\n }\n }\n }\n }\n .hideScrollIndicator()\n .onChange(of: selectedTabId) { id in\n withAnimation(.easeInOut(duration: 0.2)) {\n proxy.scrollTo(id)\n }\n }\n }\n }\n }\n }\n}\n\nstruct PromptToCodeTabBarButton: View {\n let store: StoreOf\n let info: PromptToCodeTabBar.TabInfo\n let isSelected: Bool\n @State var isHovered: Bool = false\n\n var body: some View {\n HStack(spacing: 0) {\n HStack(spacing: 4) {\n if info.isProcessing {\n ProgressView()\n .controlSize(.small)\n }\n Text(info.tabTitle)\n .truncationMode(.middle)\n .allowsTightening(true)\n }\n .font(.callout)\n .lineLimit(1)\n .frame(maxWidth: 120)\n .padding(.horizontal, 28)\n .contentShape(Rectangle())\n .onTapGesture {\n store.send(.tabClicked(id: info.id))\n }\n .overlay(alignment: .leading) {\n Button(action: {\n store.send(.closeTabButtonClicked(id: info.id))\n }) {\n Image(systemName: \"xmark\")\n .foregroundColor(.secondary)\n }\n .buttonStyle(.plain)\n .padding(2)\n .padding(.leading, 8)\n .opacity(isHovered ? 1 : 0)\n }\n .onHover { isHovered = $0 }\n .animation(.linear(duration: 0.1), value: isHovered)\n .animation(.linear(duration: 0.1), value: isSelected)\n\n Divider().padding(.vertical, 6)\n }\n .background(isSelected ? Color(nsColor: .selectedControlColor) : Color.clear)\n .frame(maxHeight: .infinity)\n }\n}\n\n" }, { "path": "Core/Sources/SuggestionWidget/SharedPanelView.swift", "content": "import ComposableArchitecture\nimport Preferences\nimport SwiftUI\n\nextension View {\n @ViewBuilder\n func animation(\n featureFlag: KeyPath,\n _ animation: Animation?,\n value: V\n ) -> some View {\n let isOn = UserDefaults.shared.value(for: featureFlag)\n if isOn {\n self.animation(animation, value: value)\n } else {\n self\n }\n }\n}\n\nstruct SharedPanelView: View {\n var store: StoreOf\n\n struct OverallState: Equatable {\n var isPanelDisplayed: Bool\n var opacity: Double\n var colorScheme: ColorScheme\n var alignTopToAnchor: Bool\n }\n\n var body: some View {\n GeometryReader { geometry in\n WithPerceptionTracking {\n VStack(spacing: 0) {\n if !store.alignTopToAnchor {\n Spacer()\n .frame(minHeight: 0, maxHeight: .infinity)\n .allowsHitTesting(false)\n }\n\n DynamicContent(store: store)\n .frame(maxWidth: .infinity, maxHeight: geometry.size.height)\n .fixedSize(horizontal: false, vertical: true)\n .allowsHitTesting(store.isPanelDisplayed)\n .layoutPriority(1)\n\n if store.alignTopToAnchor {\n Spacer()\n .frame(minHeight: 0, maxHeight: .infinity)\n .allowsHitTesting(false)\n }\n }\n .preferredColorScheme(store.colorScheme)\n .opacity(store.opacity)\n .animation(\n featureFlag: \\.animationBCrashSuggestion,\n .easeInOut(duration: 0.2),\n value: store.isPanelDisplayed\n )\n .frame(maxWidth: Style.panelWidth, maxHeight: .infinity)\n }\n }\n }\n\n struct DynamicContent: View {\n let store: StoreOf\n\n @AppStorage(\\.suggestionPresentationMode) var suggestionPresentationMode\n\n var body: some View {\n WithPerceptionTracking {\n ZStack(alignment: .topLeading) {\n if let errorMessage = store.content.error {\n error(errorMessage)\n } else if !store.content.promptToCodeGroup.promptToCodes.isEmpty {\n promptToCode()\n } else if let suggestionProvider = store.content.suggestion {\n suggestion(suggestionProvider)\n }\n }\n }\n }\n\n @ViewBuilder\n func error(_ error: String) -> some View {\n ErrorPanelView(description: error) {\n store.send(\n .errorMessageCloseButtonTapped,\n animation: .easeInOut(duration: 0.2)\n )\n }\n }\n\n @ViewBuilder\n func promptToCode() -> some View {\n PromptToCodePanelGroupView(store: store.scope(\n state: \\.content.promptToCodeGroup,\n action: \\.promptToCodeGroup\n ))\n }\n\n @ViewBuilder\n func suggestion(_ suggestion: PresentingCodeSuggestion) -> some View {\n switch suggestionPresentationMode {\n case .nearbyTextCursor:\n EmptyView()\n case .floatingWidget:\n CodeBlockSuggestionPanelView(suggestion: suggestion)\n }\n }\n }\n}\n\nstruct CommandButtonStyle: ButtonStyle {\n var color: Color\n var cornerRadius: Double = 4\n\n func makeBody(configuration: Configuration) -> some View {\n configuration.label\n .padding(.vertical, 4)\n .padding(.horizontal, 8)\n .foregroundColor(.white)\n .background(\n RoundedRectangle(cornerRadius: cornerRadius, style: .continuous)\n .fill(color.opacity(configuration.isPressed ? 0.8 : 1))\n .animation(.easeOut(duration: 0.1), value: configuration.isPressed)\n )\n .overlay {\n RoundedRectangle(cornerRadius: cornerRadius, style: .continuous)\n .stroke(Color.white.opacity(0.2), style: .init(lineWidth: 1))\n }\n }\n}\n\n// MARK: - Previews\n\nstruct SharedPanelView_Error_Preview: PreviewProvider {\n static var previews: some View {\n SharedPanelView(store: .init(\n initialState: .init(\n content: .init(error: \"This is an error\\nerror\"),\n colorScheme: .light,\n isPanelDisplayed: true\n ),\n reducer: { SharedPanel() }\n ))\n .frame(width: 450, height: 200)\n }\n}\n\nstruct SharedPanelView_Both_DisplayingSuggestion_Preview: PreviewProvider {\n static var previews: some View {\n SharedPanelView(store: .init(\n initialState: .init(\n content: .init(\n suggestion: .init(\n code: \"\"\"\n - (void)addSubview:(UIView *)view {\n [self addSubview:view];\n }\n \"\"\",\n language: \"objective-c\",\n startLineIndex: 8,\n suggestionCount: 2,\n currentSuggestionIndex: 0,\n replacingRange: .zero,\n replacingLines: [\"\"]\n )\n ),\n colorScheme: .dark,\n isPanelDisplayed: true\n ),\n reducer: { SharedPanel() }\n ))\n .frame(width: 450, height: 200)\n .background {\n HStack {\n Color.red\n Color.green\n Color.blue\n }\n }\n }\n}\n\n" }, { "path": "Core/Sources/SuggestionWidget/Styles.swift", "content": "import AppKit\nimport MarkdownUI\nimport SharedUIComponents\nimport SwiftUI\n\nenum Style {\n static let panelHeight: Double = 560\n static let panelWidth: Double = 454\n static let inlineSuggestionMinWidth: Double = 540\n static let inlineSuggestionMaxHeight: Double = 400\n static let widgetHeight: Double = 30\n static var widgetWidth: Double = 8\n static let widgetPadding: Double = 4\n static let indicatorBottomPadding: Double = 40\n static let chatWindowTitleBarHeight: Double = 24\n static let trafficLightButtonSize: Double = 12\n}\n\nextension Color {\n static var contentBackground: Color {\n Color(nsColor: NSColor(name: nil, dynamicProvider: { appearance in\n if appearance.isDarkMode {\n return #colorLiteral(red: 0.1580096483, green: 0.1730263829, blue: 0.2026666105, alpha: 1)\n }\n return .white\n }))\n }\n\n static var userChatContentBackground: Color {\n Color(nsColor: NSColor(name: nil, dynamicProvider: { appearance in\n if appearance.isDarkMode {\n return #colorLiteral(red: 0.2284317913, green: 0.2145925438, blue: 0.3214019983, alpha: 1)\n }\n return #colorLiteral(red: 0.9458052187, green: 0.9311983998, blue: 0.9906365955, alpha: 1)\n }))\n }\n}\n\nextension NSAppearance {\n var isDarkMode: Bool {\n if bestMatch(from: [.darkAqua, .aqua]) == .darkAqua {\n return true\n } else {\n return false\n }\n }\n}\n\nextension MarkdownUI.Theme {\n static func custom(fontSize: Double) -> MarkdownUI.Theme {\n .gitHub.text {\n ForegroundColor(.primary)\n BackgroundColor(Color.clear)\n FontSize(fontSize)\n }\n .codeBlock { configuration in\n configuration.label\n .relativeLineSpacing(.em(0.225))\n .markdownTextStyle {\n FontFamilyVariant(.monospaced)\n FontSize(.em(0.85))\n }\n .padding(16)\n .padding(.top, 14)\n .background(Color(nsColor: .textBackgroundColor).opacity(0.7))\n .clipShape(RoundedRectangle(cornerRadius: 6))\n .overlay(alignment: .top) {\n HStack(alignment: .center) {\n Text(configuration.language ?? \"code\")\n .foregroundStyle(.tertiary)\n .font(.callout)\n .padding(.leading, 8)\n .lineLimit(1)\n Spacer()\n CopyButton {\n NSPasteboard.general.clearContents()\n NSPasteboard.general.setString(configuration.content, forType: .string)\n }\n }\n }\n .markdownMargin(top: 4, bottom: 16)\n }\n }\n\n static func functionCall(fontSize: Double) -> MarkdownUI.Theme {\n .gitHub.text {\n ForegroundColor(.secondary)\n BackgroundColor(Color.clear)\n FontSize(fontSize - 1)\n }\n .list { configuration in\n configuration.label\n .markdownMargin(top: 4, bottom: 4)\n }\n .paragraph { configuration in\n configuration.label\n .markdownMargin(top: 0, bottom: 4)\n }\n .codeBlock { configuration in\n configuration.label\n .relativeLineSpacing(.em(0.225))\n .markdownTextStyle {\n FontFamilyVariant(.monospaced)\n FontSize(.em(0.85))\n }\n .padding(16)\n .background(Color(nsColor: .textBackgroundColor).opacity(0.7))\n .clipShape(RoundedRectangle(cornerRadius: 6))\n .markdownMargin(top: 4, bottom: 4)\n }\n }\n}\n\n" }, { "path": "Core/Sources/SuggestionWidget/SuggestionPanelContent/CodeBlockSuggestionPanelView.swift", "content": "import Combine\nimport CommandHandler\nimport Dependencies\nimport Perception\nimport SharedUIComponents\nimport SuggestionBasic\nimport SwiftUI\nimport XcodeInspector\n\npublic struct PresentingCodeSuggestion: Equatable {\n public var code: String\n public var language: String\n public var startLineIndex: Int\n public var suggestionCount: Int\n public var currentSuggestionIndex: Int\n public var replacingRange: CursorRange\n public var replacingLines: [String]\n public var descriptions: [CodeSuggestion.Description]\n\n public init(\n code: String,\n language: String,\n startLineIndex: Int,\n suggestionCount: Int,\n currentSuggestionIndex: Int,\n replacingRange: CursorRange,\n replacingLines: [String],\n descriptions: [CodeSuggestion.Description] = []\n ) {\n self.code = code\n self.language = language\n self.startLineIndex = startLineIndex\n self.suggestionCount = suggestionCount\n self.currentSuggestionIndex = currentSuggestionIndex\n self.replacingRange = replacingRange\n self.replacingLines = replacingLines\n self.descriptions = descriptions\n }\n}\n\nstruct CodeBlockSuggestionPanelView: View {\n let suggestion: PresentingCodeSuggestion\n @Environment(\\.textCursorTracker) var textCursorTracker\n @Environment(\\.colorScheme) var colorScheme\n @AppStorage(\\.suggestionCodeFont) var codeFont\n @AppStorage(\\.suggestionDisplayCompactMode) var suggestionDisplayCompactMode\n @AppStorage(\\.suggestionPresentationMode) var suggestionPresentationMode\n @AppStorage(\\.hideCommonPrecedingSpacesInSuggestion) var hideCommonPrecedingSpaces\n @AppStorage(\\.syncSuggestionHighlightTheme) var syncHighlightTheme\n @AppStorage(\\.codeForegroundColorLight) var codeForegroundColorLight\n @AppStorage(\\.codeForegroundColorDark) var codeForegroundColorDark\n @AppStorage(\\.codeBackgroundColorLight) var codeBackgroundColorLight\n @AppStorage(\\.codeBackgroundColorDark) var codeBackgroundColorDark\n\n struct ToolBar: View {\n @Dependency(\\.commandHandler) var commandHandler\n @Environment(\\.modifierFlags) var modifierFlags\n @AppStorage(\\.acceptSuggestionLineWithModifierControl) var acceptLineWithControl\n let suggestion: PresentingCodeSuggestion\n\n var body: some View {\n WithPerceptionTracking {\n HStack {\n Button(action: {\n Task {\n await commandHandler.presentPreviousSuggestion()\n }\n }) {\n Image(systemName: \"chevron.left\")\n }.buttonStyle(.plain)\n\n Text(\n \"\\(suggestion.currentSuggestionIndex + 1) / \\(suggestion.suggestionCount)\"\n )\n .monospacedDigit()\n\n Button(action: {\n Task { await commandHandler.presentNextSuggestion() }\n }) {\n Image(systemName: \"chevron.right\")\n }.buttonStyle(.plain)\n\n Spacer()\n\n Button(action: {\n Task {\n await commandHandler.dismissSuggestion()\n NSWorkspace.activatePreviousActiveXcode()\n }\n }) {\n Text(\"Dismiss\").foregroundStyle(.tertiary).padding(.trailing, 4)\n }.buttonStyle(.plain)\n\n Button(action: {\n Task {\n await commandHandler.rejectSuggestions()\n NSWorkspace.activatePreviousActiveXcode()\n }\n }) {\n Text(\"Reject\")\n }.buttonStyle(CommandButtonStyle(color: .gray))\n\n if modifierFlags.contains(.control) && acceptLineWithControl {\n Button(action: {\n Task {\n await commandHandler.acceptActiveSuggestionLineInGroup(atIndex: nil)\n NSWorkspace.activatePreviousActiveXcode()\n }\n }) {\n Text(\"Accept Line\")\n }.buttonStyle(CommandButtonStyle(color: .gray))\n } else {\n Button(action: {\n Task {\n await commandHandler.acceptSuggestion()\n NSWorkspace.activatePreviousActiveXcode()\n }\n }) {\n Text(\"Accept\")\n }.buttonStyle(CommandButtonStyle(color: .accentColor))\n }\n }\n .padding(6)\n .foregroundColor(.secondary)\n .background(.regularMaterial)\n }\n }\n }\n\n struct CompactToolBar: View {\n @Dependency(\\.commandHandler) var commandHandler\n @Environment(\\.modifierFlags) var modifierFlags\n @AppStorage(\\.acceptSuggestionLineWithModifierControl) var acceptLineWithControl\n let suggestion: PresentingCodeSuggestion\n\n var body: some View {\n WithPerceptionTracking {\n HStack {\n Button(action: {\n Task { await commandHandler.presentPreviousSuggestion() }\n }) {\n Image(systemName: \"chevron.left\")\n }.buttonStyle(.plain)\n\n Text(\n \"\\(suggestion.currentSuggestionIndex + 1) / \\(suggestion.suggestionCount)\"\n )\n .monospacedDigit()\n\n Button(action: {\n Task { await commandHandler.presentNextSuggestion() }\n }) {\n Image(systemName: \"chevron.right\")\n }.buttonStyle(.plain)\n\n Spacer()\n \n if modifierFlags.contains(.control) && acceptLineWithControl {\n Text(\"Accept Line\")\n .foregroundColor(.secondary)\n .padding(.trailing, 4)\n }\n\n Button(action: {\n Task {\n await commandHandler.dismissSuggestion()\n NSWorkspace.activatePreviousActiveXcode()\n }\n }) {\n Image(systemName: \"xmark\")\n }.buttonStyle(.plain)\n }\n .padding(4)\n .font(.caption)\n .foregroundColor(.secondary)\n .background(.regularMaterial)\n }\n }\n }\n\n struct Description: View {\n var descriptions: [CodeSuggestion.Description]\n\n var body: some View {\n VStack(spacing: 0) {\n ForEach(0.. Color in\n if syncHighlightTheme {\n if colorScheme == .light,\n let color = codeBackgroundColorLight.value?.swiftUIColor\n {\n return color\n } else if let color = codeBackgroundColorDark.value?.swiftUIColor {\n return color\n }\n }\n return Color.contentBackground\n }())\n }\n }\n\n Description(descriptions: suggestion.descriptions)\n\n Divider()\n \n if suggestionDisplayCompactMode {\n CompactToolBar(suggestion: suggestion)\n } else {\n ToolBar(suggestion: suggestion)\n }\n }\n .xcodeStyleFrame(cornerRadius: {\n switch suggestionPresentationMode {\n case .nearbyTextCursor:\n if #available(macOS 26.0, *) {\n return 8\n } else {\n return 6\n }\n case .floatingWidget: return nil\n }\n }())\n }\n }\n\n @MainActor\n func extractCode() -> (\n code: String,\n originalCode: String,\n dimmedCharacterCount: AsyncCodeBlock.DimmedCharacterCount\n ) {\n var range = suggestion.replacingRange\n range.end = .init(line: range.end.line - range.start.line, character: range.end.character)\n range.start = .init(line: 0, character: range.start.character)\n let codeInRange = EditorInformation.code(in: suggestion.replacingLines, inside: range)\n let leftover = {\n if range.end.line >= 0, range.end.line < suggestion.replacingLines.endIndex {\n let lastLine = suggestion.replacingLines[range.end.line]\n if range.end.character < lastLine.utf16.count {\n let startIndex = lastLine.utf16.index(\n lastLine.utf16.startIndex,\n offsetBy: range.end.character\n )\n var leftover = String(lastLine.utf16.suffix(from: startIndex))\n if leftover?.last?.isNewline ?? false {\n leftover?.removeLast(1)\n }\n return leftover ?? \"\"\n }\n }\n return \"\"\n }()\n\n let prefix = {\n if range.start.line >= 0, range.start.line < suggestion.replacingLines.endIndex {\n let firstLine = suggestion.replacingLines[range.start.line]\n if range.start.character < firstLine.utf16.count {\n let endIndex = firstLine.utf16.index(\n firstLine.utf16.startIndex,\n offsetBy: range.start.character\n )\n let prefix = String(firstLine.utf16.prefix(upTo: endIndex))\n return prefix ?? \"\"\n }\n }\n return \"\"\n }()\n\n let code = prefix + suggestion.code + leftover\n\n let typedCount = suggestion.startLineIndex == textCursorTracker.cursorPosition.line\n ? textCursorTracker.cursorPosition.character\n : 0\n\n return (\n code,\n codeInRange.code,\n .init(prefix: typedCount, suffix: leftover.utf16.count)\n )\n }\n}\n\n// MARK: - Previews\n\n#Preview(\"Code Block Suggestion Panel\") {\n CodeBlockSuggestionPanelView(suggestion: PresentingCodeSuggestion(\n code: \"\"\"\n LazyVGrid(columns: [GridItem(.fixed(30)), GridItem(.flexible())]) {\n ForEach(0.. Void\n\n var body: some View {\n ZStack(alignment: .topTrailing) {\n Text(description)\n .multilineTextAlignment(.leading)\n .frame(maxWidth: .infinity, alignment: .leading)\n .foregroundColor(.white)\n .padding()\n .background(Color.red)\n \n // close button\n Button(action: onCloseButtonTap) {\n Image(systemName: \"xmark\")\n .padding([.leading, .bottom], 16)\n .padding([.top, .trailing], 8)\n .foregroundColor(.white)\n }\n .buttonStyle(.plain)\n }\n .xcodeStyleFrame()\n }\n}\n" }, { "path": "Core/Sources/SuggestionWidget/SuggestionPanelContent/PromptToCodePanelView.swift", "content": "import ChatBasic\nimport Cocoa\nimport ComposableArchitecture\nimport MarkdownUI\nimport ModificationBasic\nimport PromptToCodeCustomization\nimport SharedUIComponents\nimport SuggestionBasic\nimport SwiftUI\n\nstruct PromptToCodePanelView: View {\n let store: StoreOf\n @FocusState var isTextFieldFocused: Bool\n\n var body: some View {\n WithPerceptionTracking {\n PromptToCodeCustomization.CustomizedUI(\n state: store.$promptToCodeState,\n delegate: DefaultPromptToCodeContextInputControllerDelegate(store: store),\n contextInputController: store.contextInputController,\n isInputFieldFocused: _isTextFieldFocused\n ) { customizedViews in\n VStack(spacing: 0) {\n TopBar(store: store)\n\n Content(store: store)\n .safeAreaInset(edge: .bottom) {\n VStack {\n StatusBar(store: store)\n\n ActionBar(store: store)\n\n if let inputField = customizedViews.contextInputField {\n inputField\n } else {\n Toolbar(store: store)\n }\n }\n }\n }\n }\n }\n .task {\n await MainActor.run {\n isTextFieldFocused = true\n }\n }\n }\n}\n\nextension PromptToCodePanelView {\n struct TopBar: View {\n let store: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n VStack(spacing: 0) {\n if let previousStep = store.promptToCodeState.history.last {\n Button(action: {\n store.send(.revertButtonTapped)\n }, label: {\n HStack(spacing: 4) {\n Text(Image(systemName: \"arrow.uturn.backward.circle.fill\"))\n .foregroundStyle(.secondary)\n Text(previousStep.instruction.string)\n .lineLimit(1)\n .truncationMode(.tail)\n .foregroundStyle(.secondary)\n Spacer()\n }\n .contentShape(Rectangle())\n })\n .buttonStyle(.plain)\n .disabled(store.promptToCodeState.isGenerating)\n .padding(6)\n\n Divider()\n }\n }\n .animation(.linear(duration: 0.1), value: store.promptToCodeState.history.count)\n }\n }\n\n struct SelectionRangeButton: View {\n let store: StoreOf\n var body: some View {\n WithPerceptionTracking {\n Button(action: {\n store.send(.selectionRangeToggleTapped, animation: .linear(duration: 0.1))\n }) {\n let attachedToFilename = store.filename\n let isAttached = store.promptToCodeState.isAttachedToTarget\n let color: Color = isAttached ? .accentColor : .secondary.opacity(0.6)\n HStack(spacing: 4) {\n Image(\n systemName: isAttached ? \"link\" : \"character.cursor.ibeam\"\n )\n .resizable()\n .aspectRatio(contentMode: .fit)\n .frame(width: 14, height: 14)\n .frame(width: 20, height: 20, alignment: .center)\n .foregroundColor(.white)\n .background(\n color,\n in: RoundedRectangle(\n cornerRadius: 4,\n style: .continuous\n )\n )\n\n if isAttached {\n HStack(spacing: 4) {\n Text(attachedToFilename)\n .lineLimit(1)\n .truncationMode(.middle)\n }.foregroundColor(.primary)\n } else {\n Text(\"current selection\").foregroundColor(.secondary)\n }\n }\n .padding(2)\n .padding(.trailing, 4)\n .overlay {\n RoundedRectangle(cornerRadius: 6, style: .continuous)\n .stroke(color, lineWidth: 1)\n }\n .background {\n RoundedRectangle(cornerRadius: 6, style: .continuous)\n .fill(color.opacity(0.2))\n }\n .padding(2)\n }\n .keyboardShortcut(\"j\", modifiers: [.command])\n .buttonStyle(.plain)\n }\n }\n }\n }\n\n struct StatusBar: View {\n let store: StoreOf\n @State var isAllStatusesPresented = false\n var body: some View {\n WithPerceptionTracking {\n if store.promptToCodeState.isGenerating, !store.promptToCodeState.status.isEmpty {\n if let firstStatus = store.promptToCodeState.status.first {\n let count = store.promptToCodeState.status.count\n Button(action: {\n isAllStatusesPresented = true\n }) {\n HStack {\n Text(firstStatus)\n .lineLimit(1)\n .font(.caption)\n\n Text(\"\\(count)\")\n .lineLimit(1)\n .font(.caption)\n .background(\n Circle()\n .fill(Color.secondary.opacity(0.3))\n .frame(width: 12, height: 12)\n )\n }\n .padding(8)\n .background(\n .regularMaterial,\n in: RoundedRectangle(cornerRadius: 6, style: .continuous)\n )\n .overlay {\n RoundedRectangle(cornerRadius: 6, style: .continuous)\n .stroke(Color(nsColor: .separatorColor), lineWidth: 1)\n }\n .contentShape(Rectangle())\n }\n .buttonStyle(.plain)\n .frame(maxWidth: 400)\n .popover(isPresented: $isAllStatusesPresented, arrowEdge: .top) {\n WithPerceptionTracking {\n VStack(alignment: .leading, spacing: 16) {\n ForEach(store.promptToCodeState.status, id: \\.self) { status in\n HStack {\n ProgressView()\n .progressViewStyle(CircularProgressViewStyle())\n .controlSize(.small)\n Text(status)\n }\n }\n }\n .padding()\n }\n }\n .onChange(of: store.promptToCodeState.isGenerating) { isGenerating in\n if !isGenerating {\n isAllStatusesPresented = false\n }\n }\n }\n }\n }\n }\n }\n\n struct ActionBar: View {\n let store: StoreOf\n\n var body: some View {\n HStack {\n ReferencesButton(store: store)\n StopRespondingButton(store: store)\n ActionButtons(store: store)\n }\n }\n\n struct StopRespondingButton: View {\n let store: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n if store.promptToCodeState.isGenerating {\n Button(action: {\n store.send(.stopRespondingButtonTapped)\n }) {\n HStack(spacing: 4) {\n Image(systemName: \"stop.fill\")\n Text(\"Stop\")\n }\n .padding(8)\n .background(\n .regularMaterial,\n in: RoundedRectangle(cornerRadius: 6, style: .continuous)\n )\n .overlay {\n RoundedRectangle(cornerRadius: 6, style: .continuous)\n .stroke(Color(nsColor: .separatorColor), lineWidth: 1)\n }\n }\n .buttonStyle(.plain)\n }\n }\n }\n }\n\n struct ReferencesButton: View {\n let store: StoreOf\n @State var isReferencesPresented = false\n @State var isReferencesHovered = false\n\n var body: some View {\n if !store.promptToCodeState.references.isEmpty {\n Button(action: {\n isReferencesPresented.toggle()\n }, label: {\n HStack(spacing: 4) {\n Image(systemName: \"doc.text.magnifyingglass\")\n Text(\"\\(store.promptToCodeState.references.count)\")\n }\n .padding(8)\n .background(\n .regularMaterial,\n in: RoundedRectangle(cornerRadius: 6, style: .continuous)\n )\n .overlay {\n RoundedRectangle(cornerRadius: 6, style: .continuous)\n .stroke(Color(nsColor: .separatorColor), lineWidth: 1)\n }\n })\n .buttonStyle(.plain)\n .popover(isPresented: $isReferencesPresented, arrowEdge: .trailing) {\n ReferenceList(store: store)\n }\n .onHover { hovering in\n withAnimation {\n isReferencesHovered = hovering\n }\n }\n }\n }\n }\n\n struct ActionButtons: View {\n @Perception.Bindable var store: StoreOf\n @AppStorage(\\.chatModels) var chatModels\n @AppStorage(\\.promptToCodeChatModelId) var defaultChatModelId\n\n var body: some View {\n WithPerceptionTracking {\n let isResponding = store.promptToCodeState.isGenerating\n let isCodeEmpty = store.promptToCodeState.snippets\n .allSatisfy(\\.modifiedCode.isEmpty)\n var isRespondingButCodeIsReady: Bool {\n isResponding && !isCodeEmpty\n }\n if !isResponding || isRespondingButCodeIsReady {\n HStack {\n Menu {\n WithPerceptionTracking {\n Toggle(\n \"Always accept and continue\",\n isOn: $store.isContinuous\n .animation(.easeInOut(duration: 0.1))\n )\n .toggleStyle(.checkbox)\n }\n\n chatModelMenu\n } label: {\n Image(systemName: \"gearshape.fill\")\n .resizable()\n .scaledToFit()\n .foregroundStyle(.secondary)\n .frame(width: 16)\n .frame(maxHeight: .infinity)\n .contentShape(Rectangle())\n }\n .buttonStyle(.plain)\n\n Button(action: {\n store.send(.cancelButtonTapped)\n }) {\n Text(\"Cancel\")\n }\n .buttonStyle(CommandButtonStyle(color: .gray))\n .keyboardShortcut(\"w\", modifiers: [.command])\n\n if store.isActiveDocument {\n if !isCodeEmpty {\n AcceptButton(store: store)\n }\n } else {\n RevealButton(store: store)\n }\n }\n .fixedSize()\n .padding(8)\n .background(\n .regularMaterial,\n in: RoundedRectangle(cornerRadius: 6, style: .continuous)\n )\n .overlay {\n RoundedRectangle(cornerRadius: 6, style: .continuous)\n .stroke(Color(nsColor: .separatorColor), lineWidth: 1)\n }\n .animation(\n .easeInOut(duration: 0.1),\n value: store.promptToCodeState.snippets\n )\n }\n }\n }\n\n @ViewBuilder\n var chatModelMenu: some View {\n let allModels = chatModels\n\n Menu(\"Chat Model\") {\n Button(action: {\n defaultChatModelId = \"\"\n }) {\n HStack {\n Text(\"Same as chat feature\")\n if defaultChatModelId.isEmpty {\n Image(systemName: \"checkmark\")\n }\n }\n }\n\n if !allModels.contains(where: { $0.id == defaultChatModelId }),\n !defaultChatModelId.isEmpty\n {\n Button(action: {\n defaultChatModelId = allModels.first?.id ?? \"\"\n }) {\n HStack {\n Text(\n (allModels.first?.name).map { \"\\($0) (Default)\" }\n ?? \"No model found\"\n )\n Image(systemName: \"checkmark\")\n }\n }\n }\n\n ForEach(allModels, id: \\.id) { model in\n Button(action: {\n defaultChatModelId = model.id\n }) {\n HStack {\n Text(model.name)\n if model.id == defaultChatModelId {\n Image(systemName: \"checkmark\")\n }\n }\n }\n }\n }\n }\n }\n\n struct RevealButton: View {\n let store: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n Button(action: {\n store.send(.revealFileButtonClicked)\n }) {\n Text(\"Jump to File(โŒ˜ + โŽ)\")\n }\n .buttonStyle(CommandButtonStyle(color: .accentColor))\n .keyboardShortcut(KeyEquivalent.return, modifiers: [.command])\n }\n }\n }\n\n struct AcceptButton: View {\n let store: StoreOf\n @Environment(\\.modifierFlags) var modifierFlags\n\n struct TheButtonStyle: ButtonStyle {\n func makeBody(configuration: Configuration) -> some View {\n configuration.label\n .background(\n Rectangle()\n .fill(Color.accentColor.opacity(configuration.isPressed ? 0.8 : 1))\n )\n }\n }\n\n var body: some View {\n WithPerceptionTracking {\n let defaultModeIsContinuous = store.isContinuous\n let isAttached = store.promptToCodeState.isAttachedToTarget\n\n HStack(spacing: 0) {\n Button(action: {\n switch (\n modifierFlags.contains(.option),\n defaultModeIsContinuous\n ) {\n case (true, true):\n store.send(.acceptButtonTapped)\n case (false, true):\n store.send(.acceptAndContinueButtonTapped)\n case (true, false):\n store.send(.acceptAndContinueButtonTapped)\n case (false, false):\n store.send(.acceptButtonTapped)\n }\n }) {\n Group {\n switch (\n isAttached,\n modifierFlags.contains(.option),\n defaultModeIsContinuous\n ) {\n case (true, true, true):\n Text(\"Accept(โŒฅ + โŒ˜ + โŽ)\")\n case (true, false, true):\n Text(\"Accept and Continue(โŒ˜ + โŽ)\")\n case (true, true, false):\n Text(\"Accept and Continue(โŒฅ + โŒ˜ + โŽ)\")\n case (true, false, false):\n Text(\"Accept(โŒ˜ + โŽ)\")\n case (false, true, true):\n Text(\"Replace(โŒฅ + โŒ˜ + โŽ)\")\n case (false, false, true):\n Text(\"Replace and Continue(โŒ˜ + โŽ)\")\n case (false, true, false):\n Text(\"Replace and Continue(โŒฅ + โŒ˜ + โŽ)\")\n case (false, false, false):\n Text(\"Replace(โŒ˜ + โŽ)\")\n }\n }\n .padding(.vertical, 4)\n .padding(.leading, 8)\n .padding(.trailing, 4)\n }\n .buttonStyle(TheButtonStyle())\n .keyboardShortcut(\n KeyEquivalent.return,\n modifiers: modifierFlags\n .contains(.option) ? [.command, .option] : [.command]\n )\n\n Divider()\n\n Menu {\n WithPerceptionTracking {\n if defaultModeIsContinuous {\n Button(action: {\n store.send(.acceptButtonTapped)\n }) {\n Text(\"Accept(โŒฅ + โŒ˜ + โŽ)\")\n }\n } else {\n Button(action: {\n store.send(.acceptAndContinueButtonTapped)\n }) {\n Text(\"Accept and Continue(โŒฅ + โŒ˜ + โŽ)\")\n }\n }\n }\n } label: {\n Text(Image(systemName: \"chevron.down\"))\n .font(.footnote.weight(.bold))\n .scaleEffect(0.8)\n .foregroundStyle(.white.opacity(0.8))\n .frame(maxHeight: .infinity)\n .padding(.leading, 1)\n .padding(.trailing, 2)\n .contentShape(Rectangle())\n }\n .buttonStyle(.plain)\n }\n .fixedSize()\n\n .foregroundColor(.white)\n .clipShape(RoundedRectangle(cornerRadius: 4, style: .continuous))\n .background(\n RoundedRectangle(cornerRadius: 4, style: .continuous)\n .fill(Color.accentColor)\n )\n .overlay {\n RoundedRectangle(cornerRadius: 4, style: .continuous)\n .stroke(Color.white.opacity(0.2), style: .init(lineWidth: 1))\n }\n }\n }\n }\n }\n\n struct Content: View {\n let store: StoreOf\n\n @Environment(\\.colorScheme) var colorScheme\n @AppStorage(\\.syncPromptToCodeHighlightTheme) var syncHighlightTheme\n @AppStorage(\\.codeForegroundColorLight) var codeForegroundColorLight\n @AppStorage(\\.codeForegroundColorDark) var codeForegroundColorDark\n @AppStorage(\\.codeBackgroundColorLight) var codeBackgroundColorLight\n @AppStorage(\\.codeBackgroundColorDark) var codeBackgroundColorDark\n\n var codeForegroundColor: Color? {\n if syncHighlightTheme {\n if colorScheme == .light,\n let color = codeForegroundColorLight.value?.swiftUIColor\n {\n return color\n } else if let color = codeForegroundColorDark.value?.swiftUIColor {\n return color\n }\n }\n return nil\n }\n\n var codeBackgroundColor: Color {\n if syncHighlightTheme {\n if colorScheme == .light,\n let color = codeBackgroundColorLight.value?.swiftUIColor\n {\n return color\n } else if let color = codeBackgroundColorDark.value?.swiftUIColor {\n return color\n }\n }\n return Color.contentBackground\n }\n\n var body: some View {\n WithPerceptionTracking {\n ScrollView {\n WithPerceptionTracking {\n VStack(spacing: 0) {\n VStack(spacing: 0) {\n let language = store.promptToCodeState.source.language\n let isAttached = store.promptToCodeState.isAttachedToTarget\n let lastId = store.promptToCodeState.snippets.last?.id\n let isGenerating = store.promptToCodeState.isGenerating\n ForEach(store.scope(\n state: \\.snippetPanels,\n action: \\.snippetPanel\n )) { snippetStore in\n WithPerceptionTracking {\n SnippetPanelView(\n store: snippetStore,\n language: language,\n codeForegroundColor: codeForegroundColor ?? .primary,\n codeBackgroundColor: codeBackgroundColor,\n isAttached: isAttached,\n isGenerating: isGenerating\n )\n\n if snippetStore.id != lastId {\n Divider()\n }\n }\n }\n }\n\n Spacer(minLength: 56)\n }\n }\n }\n .background(codeBackgroundColor)\n }\n }\n\n struct SnippetPanelView: View {\n let store: StoreOf\n let language: CodeLanguage\n let codeForegroundColor: Color\n let codeBackgroundColor: Color\n let isAttached: Bool\n let isGenerating: Bool\n\n var body: some View {\n WithPerceptionTracking {\n VStack(alignment: .leading, spacing: 0) {\n SnippetTitleBar(\n store: store,\n language: language,\n codeForegroundColor: codeForegroundColor,\n isAttached: isAttached\n )\n\n DescriptionContent(store: store, codeForegroundColor: codeForegroundColor)\n\n CodeContent(\n store: store,\n language: language,\n isGenerating: isGenerating,\n codeForegroundColor: codeForegroundColor\n )\n\n ErrorMessage(store: store)\n }\n }\n }\n }\n\n struct SnippetTitleBar: View {\n let store: StoreOf\n let language: CodeLanguage\n let codeForegroundColor: Color\n let isAttached: Bool\n var body: some View {\n WithPerceptionTracking {\n HStack {\n Text(language.rawValue)\n .foregroundStyle(codeForegroundColor)\n .font(.callout.bold())\n .lineLimit(1)\n if isAttached {\n Text(String(describing: store.snippet.attachedRange))\n .foregroundStyle(codeForegroundColor.opacity(0.5))\n .font(.callout)\n }\n Spacer()\n CopyCodeButton(store: store)\n }\n .padding(.leading, 8)\n }\n }\n }\n\n struct CopyCodeButton: View {\n let store: StoreOf\n var body: some View {\n WithPerceptionTracking {\n if !store.snippet.modifiedCode.isEmpty {\n DraggableCopyButton {\n store.withState {\n $0.snippet.modifiedCode\n }\n }\n }\n }\n }\n }\n\n struct ErrorMessage: View {\n let store: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n if let errorMessage = store.snippet.error, !errorMessage.isEmpty {\n (\n Text(Image(systemName: \"exclamationmark.triangle.fill\")) +\n Text(\" \") +\n Text(errorMessage)\n )\n .multilineTextAlignment(.leading)\n .foregroundColor(.red)\n .padding(.horizontal, 8)\n .padding(.vertical, 4)\n }\n }\n }\n }\n\n struct DescriptionContent: View {\n let store: StoreOf\n let codeForegroundColor: Color?\n\n var body: some View {\n WithPerceptionTracking {\n if !store.snippet.description.isEmpty {\n Markdown(store.snippet.description)\n .textSelection(.enabled)\n .markdownTheme(.gitHub.text {\n BackgroundColor(Color.clear)\n ForegroundColor(codeForegroundColor)\n })\n .padding(.horizontal)\n .padding(.vertical, 4)\n .frame(maxWidth: .infinity)\n }\n }\n }\n }\n\n struct CodeContent: View {\n let store: StoreOf\n let language: CodeLanguage\n let isGenerating: Bool\n let codeForegroundColor: Color?\n\n @AppStorage(\\.wrapCodeInPromptToCode) var wrapCode\n\n var body: some View {\n WithPerceptionTracking {\n if !store.snippet.modifiedCode.isEmpty {\n let wrapCode = wrapCode ||\n [CodeLanguage.plaintext, .builtIn(.markdown), .builtIn(.shellscript),\n .builtIn(.tex)].contains(language)\n if wrapCode {\n CodeBlockInContent(\n store: store,\n language: language,\n codeForegroundColor: codeForegroundColor,\n presentAllContent: !isGenerating\n )\n } else {\n MinScrollView {\n CodeBlockInContent(\n store: store,\n language: language,\n codeForegroundColor: codeForegroundColor,\n presentAllContent: !isGenerating\n )\n }\n .modify {\n if #available(macOS 13.0, *) {\n $0.scrollIndicators(.hidden)\n } else {\n $0\n }\n }\n }\n } else {\n if isGenerating {\n Text(\"Thinking...\")\n .foregroundStyle(.secondary)\n .padding(.horizontal, 8)\n .padding(.vertical, 4)\n } else {\n Text(\"Enter your requirements to generate code.\")\n .foregroundStyle(.secondary)\n .padding(.horizontal, 8)\n .padding(.vertical, 4)\n }\n }\n }\n }\n\n struct MinWidthPreferenceKey: PreferenceKey {\n static func reduce(value: inout CGFloat, nextValue: () -> CGFloat) {\n value = nextValue()\n }\n\n static var defaultValue: CGFloat = 0\n }\n\n struct MinScrollView: View {\n @ViewBuilder let content: Content\n @State var minWidth: CGFloat = 0\n\n var body: some View {\n ScrollView(.horizontal) {\n content\n .frame(minWidth: minWidth)\n }\n .overlay {\n GeometryReader { proxy in\n Color.clear.preference(\n key: MinWidthPreferenceKey.self,\n value: proxy.size.width\n )\n }\n }\n .onPreferenceChange(MinWidthPreferenceKey.self) {\n minWidth = $0\n }\n }\n }\n\n struct CodeBlockInContent: View {\n let store: StoreOf\n let language: CodeLanguage\n let codeForegroundColor: Color?\n let presentAllContent: Bool\n\n @Environment(\\.colorScheme) var colorScheme\n @AppStorage(\\.promptToCodeCodeFont) var codeFont\n @AppStorage(\\.hideCommonPrecedingSpacesInPromptToCode) var hideCommonPrecedingSpaces\n\n var body: some View {\n WithPerceptionTracking {\n let startLineIndex = store.snippet.attachedRange.start.line\n AsyncDiffCodeBlock(\n code: store.snippet.modifiedCode,\n originalCode: store.snippet.originalCode,\n language: language.rawValue,\n startLineIndex: startLineIndex,\n scenario: \"promptToCode\",\n font: codeFont.value.nsFont,\n droppingLeadingSpaces: hideCommonPrecedingSpaces,\n proposedForegroundColor: codeForegroundColor,\n skipLastOnlyRemovalSection: !presentAllContent\n )\n .frame(maxWidth: CGFloat.infinity)\n }\n }\n }\n }\n }\n\n struct Toolbar: View {\n let store: StoreOf\n @FocusState var focusField: PromptToCodePanel.State.FocusField?\n\n var body: some View {\n HStack {\n HStack(spacing: 0) {\n if let contextInputController = store.contextInputController\n as? DefaultPromptToCodeContextInputController\n {\n InputField(\n store: store,\n contextInputField: contextInputController,\n focusField: $focusField\n )\n SendButton(store: store)\n }\n }\n .frame(maxWidth: .infinity)\n .background {\n RoundedRectangle(cornerRadius: 6)\n .fill(Color(nsColor: .controlBackgroundColor))\n }\n .overlay {\n RoundedRectangle(cornerRadius: 6)\n .stroke(Color(nsColor: .controlColor), lineWidth: 1)\n }\n .background {\n Button(action: {\n (\n store.contextInputController\n as? DefaultPromptToCodeContextInputController\n )?.appendNewLineToPromptButtonTapped()\n }) {\n EmptyView()\n }\n .keyboardShortcut(KeyEquivalent.return, modifiers: [.shift])\n }\n .background {\n Button(action: { focusField = .textField }) {\n EmptyView()\n }\n .keyboardShortcut(\"l\", modifiers: [.command])\n }\n }\n .padding(8)\n .background(.ultraThickMaterial)\n }\n\n struct InputField: View {\n @Perception.Bindable var store: StoreOf\n @Perception.Bindable var contextInputField: DefaultPromptToCodeContextInputController\n var focusField: FocusState.Binding\n\n var body: some View {\n WithPerceptionTracking {\n AutoresizingCustomTextEditor(\n text: $contextInputField.instructionString,\n font: .systemFont(ofSize: 14),\n isEditable: !store.promptToCodeState.isGenerating,\n maxHeight: 400,\n onSubmit: { store.send(.modifyCodeButtonTapped) }\n )\n .opacity(store.promptToCodeState.isGenerating ? 0.5 : 1)\n .disabled(store.promptToCodeState.isGenerating)\n .focused(focusField, equals: PromptToCodePanel.State.FocusField.textField)\n .bind($store.focusedField, to: focusField)\n }\n .padding(8)\n .fixedSize(horizontal: false, vertical: true)\n }\n }\n\n struct SendButton: View {\n let store: StoreOf\n var body: some View {\n WithPerceptionTracking {\n Button(action: {\n store.send(.modifyCodeButtonTapped)\n }) {\n Image(systemName: \"paperplane.fill\")\n .padding(8)\n }\n .buttonStyle(.plain)\n .disabled(store.promptToCodeState.isGenerating)\n .keyboardShortcut(KeyEquivalent.return, modifiers: [])\n }\n }\n }\n }\n\n struct ReferenceList: View {\n @Perception.Bindable var store: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n ScrollView {\n VStack(alignment: .leading, spacing: 8) {\n ForEach(\n 0.. Void\n\n var body: some View {\n Button(action: onClick) {\n VStack(alignment: .leading, spacing: 4) {\n HStack(spacing: 4) {\n ReferenceIcon(kind: reference.kind)\n .layoutPriority(2)\n Text(reference.title)\n .truncationMode(.middle)\n .lineLimit(1)\n .layoutPriority(1)\n .foregroundStyle(isUsed ? .primary : .secondary)\n }\n Text(reference.content)\n .lineLimit(3)\n .truncationMode(.tail)\n .foregroundStyle(.tertiary)\n .foregroundStyle(isUsed ? .secondary : .tertiary)\n }\n .padding(.vertical, 4)\n .padding(.leading, 4)\n .padding(.trailing)\n .frame(maxWidth: .infinity, alignment: .leading)\n .overlay {\n RoundedRectangle(cornerRadius: 4)\n .stroke(Color(nsColor: .separatorColor), lineWidth: 1)\n }\n }\n .buttonStyle(.plain)\n }\n }\n }\n\n struct ReferenceIcon: View {\n let kind: ChatMessage.Reference.Kind\n\n var body: some View {\n RoundedRectangle(cornerRadius: 4)\n .fill({\n switch kind {\n case let .symbol(symbol, _, _, _):\n switch symbol {\n case .class:\n Color.purple\n case .struct:\n Color.purple\n case .enum:\n Color.purple\n case .actor:\n Color.purple\n case .protocol:\n Color.purple\n case .extension:\n Color.indigo\n case .case:\n Color.green\n case .property:\n Color.teal\n case .typealias:\n Color.orange\n case .function:\n Color.teal\n case .method:\n Color.blue\n }\n case .text:\n Color.gray\n case .webpage:\n Color.blue\n case .textFile:\n Color.gray\n case .other:\n Color.gray\n case .error:\n Color.red\n }\n }())\n .frame(width: 26, height: 14)\n .overlay(alignment: .center) {\n Group {\n switch kind {\n case let .symbol(symbol, _, _, _):\n switch symbol {\n case .class:\n Text(\"C\")\n case .struct:\n Text(\"S\")\n case .enum:\n Text(\"E\")\n case .actor:\n Text(\"A\")\n case .protocol:\n Text(\"Pr\")\n case .extension:\n Text(\"Ex\")\n case .case:\n Text(\"K\")\n case .property:\n Text(\"P\")\n case .typealias:\n Text(\"T\")\n case .function:\n Text(\"๐‘“\")\n case .method:\n Text(\"M\")\n }\n case .text:\n Text(\"Txt\")\n case .webpage:\n Text(\"Web\")\n case .other:\n Text(\"*\")\n case .textFile:\n Text(\"Txt\")\n case .error:\n Text(\"Err\")\n }\n }\n .font(.system(size: 10).monospaced())\n .foregroundColor(.white)\n }\n }\n }\n}\n\n// MARK: - Previews\n\n#Preview(\"Multiple Snippets\") {\n PromptToCodePanelView(store: .init(initialState: .init(\n promptToCodeState: Shared(ModificationState(\n source: .init(\n language: CodeLanguage.builtIn(.swift),\n documentURL: URL(\n fileURLWithPath: \"path/to/file-name-is-super-long-what-should-we-do-with-it-hah-longer-longer.txt\"\n ),\n projectRootURL: URL(fileURLWithPath: \"path/to/file.txt\"),\n content: \"\",\n lines: []\n ),\n history: [\n .init(snippets: [\n .init(\n startLineIndex: 8,\n originalCode: \"print(foo)\",\n modifiedCode: \"print(bar)\",\n description: \"\",\n error: \"Error\",\n attachedRange: CursorRange(\n start: .init(line: 8, character: 0),\n end: .init(line: 12, character: 2)\n )\n ),\n ], instruction: .init(\"Previous instruction\"), references: []),\n ],\n snippets: [\n .init(\n startLineIndex: 8,\n originalCode: \"print(foo)\",\n modifiedCode: \"print(bar)\\nprint(baz)\",\n description: \"\",\n error: \"Error\",\n attachedRange: CursorRange(\n start: .init(line: 8, character: 0),\n end: .init(line: 12, character: 2)\n )\n ),\n .init(\n startLineIndex: 13,\n originalCode: \"\"\"\n struct Foo {\n var foo: Int\n }\n \"\"\",\n modifiedCode: \"\"\"\n struct Bar {\n var bar: String\n }\n \"\"\",\n description: \"Cool\",\n error: nil,\n attachedRange: CursorRange(\n start: .init(line: 13, character: 0),\n end: .init(line: 12, character: 2)\n )\n ),\n ],\n extraSystemPrompt: \"\",\n isAttachedToTarget: true,\n references: [\n ChatMessage.Reference(\n title: \"Foo\",\n content: \"struct Foo { var foo: Int }\",\n kind: .symbol(\n .struct,\n uri: \"file:///path/to/file.txt\",\n startLine: 13,\n endLine: 13\n )\n ),\n ],\n )),\n instruction: nil,\n commandName: \"Generate Code\"\n ), reducer: { PromptToCodePanel() }))\n .frame(maxWidth: 450, maxHeight: Style.panelHeight)\n .fixedSize(horizontal: false, vertical: true)\n .frame(width: 500, height: 500, alignment: .center)\n}\n\n#Preview(\"Detached With Long File Name\") {\n PromptToCodePanelView(store: .init(initialState: .init(\n promptToCodeState: Shared(ModificationState(\n source: .init(\n language: CodeLanguage.builtIn(.swift),\n documentURL: URL(\n fileURLWithPath: \"path/to/file-name-is-super-long-what-should-we-do-with-it-hah.txt\"\n ),\n projectRootURL: URL(fileURLWithPath: \"path/to/file.txt\"),\n content: \"\",\n lines: []\n ),\n snippets: [\n .init(\n startLineIndex: 8,\n originalCode: \"print(foo)\",\n modifiedCode: \"print(bar)\",\n description: \"\",\n error: \"Error\",\n attachedRange: CursorRange(\n start: .init(line: 8, character: 0),\n end: .init(line: 12, character: 2)\n )\n ),\n .init(\n startLineIndex: 13,\n originalCode: \"\"\"\n struct Bar {\n var foo: Int\n }\n \"\"\",\n modifiedCode: \"\"\"\n struct Bar {\n var foo: String\n }\n \"\"\",\n description: \"Cool\",\n error: nil,\n attachedRange: CursorRange(\n start: .init(line: 13, character: 0),\n end: .init(line: 12, character: 2)\n )\n ),\n ],\n extraSystemPrompt: \"\",\n isAttachedToTarget: false\n )),\n instruction: nil,\n commandName: \"Generate Code\"\n ), reducer: { PromptToCodePanel() }))\n .frame(maxWidth: 450, maxHeight: Style.panelHeight)\n .fixedSize(horizontal: false, vertical: true)\n .frame(width: 500, height: 500, alignment: .center)\n}\n\n#Preview(\"Generating\") {\n PromptToCodePanelView(store: .init(initialState: .init(\n promptToCodeState: Shared(ModificationState(\n source: .init(\n language: CodeLanguage.builtIn(.swift),\n documentURL: URL(\n fileURLWithPath: \"path/to/file-name-is-super-long-what-should-we-do-with-it-hah.txt\"\n ),\n projectRootURL: URL(fileURLWithPath: \"path/to/file.txt\"),\n content: \"\",\n lines: []\n ),\n snippets: [\n .init(\n startLineIndex: 8,\n originalCode: \"print(foo)\",\n modifiedCode: \"print(bar)\",\n description: \"\",\n error: \"Error\",\n attachedRange: CursorRange(\n start: .init(line: 8, character: 0),\n end: .init(line: 12, character: 2)\n )\n ),\n .init(\n startLineIndex: 13,\n originalCode: \"\"\"\n struct Bar {\n var foo: Int\n }\n \"\"\",\n modifiedCode: \"\"\"\n struct Bar {\n var foo: String\n }\n \"\"\",\n description: \"Cool\",\n error: nil,\n attachedRange: CursorRange(\n start: .init(line: 13, character: 0),\n end: .init(line: 12, character: 2)\n )\n ),\n ],\n extraSystemPrompt: \"\",\n isAttachedToTarget: true,\n isGenerating: true,\n status: [\"Status 1\", \"Status 2\"]\n )),\n instruction: nil,\n commandName: \"Generate Code\"\n ), reducer: { PromptToCodePanel() }))\n .frame(maxWidth: 450, maxHeight: Style.panelHeight)\n .fixedSize(horizontal: false, vertical: true)\n .frame(width: 500, height: 500, alignment: .center)\n}\n\n" }, { "path": "Core/Sources/SuggestionWidget/SuggestionPanelContent/ToastPanelView.swift", "content": "import ComposableArchitecture\nimport Dependencies\nimport Foundation\nimport SwiftUI\nimport Toast\n\nstruct ToastPanelView: View {\n let store: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n VStack(spacing: 4) {\n if !store.alignTopToAnchor {\n Spacer()\n .allowsHitTesting(false)\n }\n\n ForEach(store.toast.messages) { message in\n HStack {\n message.content\n .foregroundColor(.white)\n .textSelection(.enabled)\n \n\n if !message.buttons.isEmpty {\n HStack {\n ForEach(\n Array(message.buttons.enumerated()),\n id: \\.offset\n ) { _, button in\n Button(action: button.action) {\n button.label\n .foregroundColor(.white)\n .padding(.horizontal, 6)\n .padding(.vertical, 2)\n .background {\n RoundedRectangle(cornerRadius: 4)\n .stroke(Color.white, lineWidth: 1)\n }\n .contentShape(Rectangle())\n }\n .buttonStyle(.plain)\n .allowsHitTesting(true)\n }\n }\n }\n }\n .padding(8)\n .frame(maxWidth: .infinity)\n .background({\n switch message.type {\n case .info: return Color.accentColor\n case .error: return Color(nsColor: .systemRed)\n case .warning: return Color(nsColor: .systemOrange)\n }\n }() as Color, in: RoundedRectangle(cornerRadius: 8))\n .overlay {\n RoundedRectangle(cornerRadius: 8)\n .stroke(Color.black.opacity(0.1), lineWidth: 1)\n }\n }\n\n if store.alignTopToAnchor {\n Spacer()\n .allowsHitTesting(false)\n }\n }\n .colorScheme(store.colorScheme)\n .frame(maxWidth: .infinity, maxHeight: .infinity)\n }\n }\n}\n\n#Preview {\n ToastPanelView(store: .init(initialState: .init(\n toast: .init(messages: [\n ToastController.Message(\n id: UUID(),\n type: .info,\n content: Text(\"Info message\"),\n buttons: [\n .init(label: Text(\"Dismiss\"), action: {}),\n .init(label: Text(\"More info\"), action: {}),\n ]\n ),\n ToastController.Message(\n id: UUID(),\n type: .error,\n content: Text(\"Error message\"),\n buttons: [.init(label: Text(\"Dismiss\"), action: {})]\n ),\n ToastController.Message(\n id: UUID(),\n type: .warning,\n content: Text(\"Warning message\"),\n buttons: [.init(label: Text(\"Dismiss\"), action: {})]\n ),\n ])\n ), reducer: {\n ToastPanel()\n }))\n}\n\n" }, { "path": "Core/Sources/SuggestionWidget/SuggestionPanelView.swift", "content": "import ComposableArchitecture\nimport Foundation\nimport SwiftUI\n\nstruct SuggestionPanelView: View {\n let store: StoreOf\n\n struct OverallState: Equatable {\n var isPanelDisplayed: Bool\n var opacity: Double\n var colorScheme: ColorScheme\n var isPanelOutOfFrame: Bool\n var alignTopToAnchor: Bool\n }\n\n var body: some View {\n WithPerceptionTracking {\n VStack(spacing: 0) {\n if !store.alignTopToAnchor {\n Spacer()\n .frame(minHeight: 0, maxHeight: .infinity)\n .allowsHitTesting(false)\n }\n\n Content(store: store)\n .allowsHitTesting(\n store.isPanelDisplayed && !store.isPanelOutOfFrame\n )\n .frame(maxWidth: .infinity)\n\n if store.alignTopToAnchor {\n Spacer()\n .frame(minHeight: 0, maxHeight: .infinity)\n .allowsHitTesting(false)\n }\n }\n .preferredColorScheme(store.colorScheme)\n .opacity(store.opacity)\n .animation(\n featureFlag: \\.animationBCrashSuggestion,\n .easeInOut(duration: 0.2),\n value: store.isPanelDisplayed\n )\n .animation(\n featureFlag: \\.animationBCrashSuggestion,\n .easeInOut(duration: 0.2),\n value: store.isPanelOutOfFrame\n )\n .frame(\n maxWidth: Style.inlineSuggestionMinWidth,\n maxHeight: Style.inlineSuggestionMaxHeight\n )\n }\n }\n\n struct Content: View {\n let store: StoreOf\n @AppStorage(\\.suggestionPresentationMode) var suggestionPresentationMode\n\n var body: some View {\n WithPerceptionTracking {\n if let content = store.content {\n ZStack(alignment: .topLeading) {\n switch suggestionPresentationMode {\n case .nearbyTextCursor:\n CodeBlockSuggestionPanelView(suggestion: content)\n case .floatingWidget:\n EmptyView()\n }\n }\n .frame(maxWidth: .infinity, maxHeight: Style.inlineSuggestionMaxHeight)\n .fixedSize(horizontal: false, vertical: true)\n }\n }\n }\n }\n}\n\n" }, { "path": "Core/Sources/SuggestionWidget/SuggestionWidgetController.swift", "content": "import ActiveApplicationMonitor\nimport AppKit\nimport AsyncAlgorithms\nimport ChatTab\nimport Combine\nimport ComposableArchitecture\nimport Preferences\nimport SwiftUI\nimport UserDefaultsObserver\nimport XcodeInspector\n\n@MainActor\npublic final class SuggestionWidgetController: NSObject {\n let store: StoreOf\n let chatTabPool: ChatTabPool\n let windowsController: WidgetWindowsController\n private var cancellable = Set()\n\n public let dependency: SuggestionWidgetControllerDependency\n\n public init(\n store: StoreOf,\n chatTabPool: ChatTabPool,\n dependency: SuggestionWidgetControllerDependency\n ) {\n self.dependency = dependency\n self.store = store\n self.chatTabPool = chatTabPool\n windowsController = .init(store: store, chatTabPool: chatTabPool)\n\n super.init()\n\n if ProcessInfo.processInfo.environment[\"IS_UNIT_TEST\"] == \"YES\" { return }\n\n dependency.windowsController = windowsController\n\n store.send(.startup)\n Task {\n await windowsController.start()\n }\n }\n}\n\n// MARK: - Handle Events\n\npublic extension SuggestionWidgetController {\n func suggestCode() {\n store.send(.panel(.presentSuggestion))\n }\n\n func discardSuggestion() {\n store.withState { state in\n if state.panelState.content.suggestion != nil {\n store.send(.panel(.discardSuggestion))\n }\n }\n }\n\n #warning(\"TODO: Make a progress controller that doesn't use TCA.\")\n func markAsProcessing(_ isProcessing: Bool) {\n store.withState { state in\n if isProcessing, !state.circularWidgetState.isProcessing {\n store.send(.circularWidget(.markIsProcessing))\n } else if !isProcessing, state.circularWidgetState.isProcessing {\n store.send(.circularWidget(.endIsProcessing))\n }\n }\n }\n\n func presentError(_ errorDescription: String) {\n store.send(.toastPanel(.toast(.toast(errorDescription, .error, nil))))\n }\n}\n\n" }, { "path": "Core/Sources/SuggestionWidget/SuggestionWidgetDataSource.swift", "content": "import Foundation\n\npublic protocol SuggestionWidgetDataSource {\n func suggestionForFile(at url: URL) async -> PresentingCodeSuggestion?\n}\n\nstruct MockWidgetDataSource: SuggestionWidgetDataSource {\n func suggestionForFile(at url: URL) async -> PresentingCodeSuggestion? {\n return PresentingCodeSuggestion(\n code: \"\"\"\n func test() {\n let x = 1\n let y = 2\n let z = x + y\n }\n \"\"\",\n language: \"swift\",\n startLineIndex: 1,\n suggestionCount: 3,\n currentSuggestionIndex: 0,\n replacingRange: .zero, \n replacingLines: []\n )\n }\n}\n\n" }, { "path": "Core/Sources/SuggestionWidget/TextCursorTracker.swift", "content": "import Foundation\nimport Perception\nimport SuggestionBasic\nimport SwiftUI\nimport XcodeInspector\n\n/// A passive tracker that observe the changes of the source editor content.\n@Perceptible\nfinal class TextCursorTracker {\n @MainActor\n var cursorPosition: CursorPosition { content.cursorPosition }\n @MainActor\n var currentLine: String {\n if content.cursorPosition.line >= 0, content.cursorPosition.line < content.lines.count {\n content.lines[content.cursorPosition.line]\n } else {\n \"\"\n }\n }\n\n @MainActor\n var content: SourceEditor.Content = .init(\n content: \"\",\n lines: [],\n selections: [],\n cursorPosition: .zero,\n cursorOffset: 0,\n lineAnnotations: []\n )\n\n @PerceptionIgnored var eventObservationTask: Task?\n\n init() {\n observeAppChange()\n }\n\n deinit {\n eventObservationTask?.cancel()\n }\n\n var isPreview: Bool {\n ProcessInfo.processInfo.environment[\"XCODE_RUNNING_FOR_PREVIEWS\"] == \"1\"\n }\n\n private func observeAppChange() {\n if isPreview { return }\n Task { [weak self] in\n let notifications = NotificationCenter.default\n .notifications(named: .focusedEditorDidChange)\n for await _ in notifications {\n guard let self else { return }\n guard let editor = await XcodeInspector.shared.focusedEditor else { continue }\n await self.observeAXNotifications(editor)\n }\n }\n }\n\n private func observeAXNotifications(_ editor: SourceEditor) async {\n if isPreview { return }\n eventObservationTask?.cancel()\n let content = editor.getLatestEvaluatedContent()\n await MainActor.run {\n self.content = content\n }\n eventObservationTask = Task { [weak self] in\n for await event in await editor.axNotifications.notifications() {\n try Task.checkCancellation()\n guard let self else { return }\n guard event.kind == .evaluatedContentChanged else { continue }\n let content = editor.getLatestEvaluatedContent()\n await MainActor.run {\n self.content = content\n }\n }\n }\n }\n}\n\nstruct TextCursorTrackerEnvironmentKey: EnvironmentKey {\n static var defaultValue: TextCursorTracker = .init()\n}\n\nextension EnvironmentValues {\n var textCursorTracker: TextCursorTracker {\n get { self[TextCursorTrackerEnvironmentKey.self] }\n set { self[TextCursorTrackerEnvironmentKey.self] = newValue }\n }\n}\n\n" }, { "path": "Core/Sources/SuggestionWidget/WidgetPositionStrategy.swift", "content": "import AppKit\nimport Foundation\n\npublic struct WidgetLocation: Equatable {\n struct PanelLocation: Equatable {\n var frame: CGRect\n var alignPanelTop: Bool\n }\n\n var widgetFrame: CGRect\n var tabFrame: CGRect\n var sharedPanelLocation: PanelLocation\n var defaultPanelLocation: PanelLocation\n var suggestionPanelLocation: PanelLocation?\n}\n\nenum UpdateLocationStrategy {\n struct AlignToTextCursor {\n func framesForWindows(\n windowFrame: CGRect,\n editorFrame: CGRect,\n mainScreen: NSScreen,\n activeScreen: NSScreen,\n editor: AXUIElement,\n hideCircularWidget: Bool = UserDefaults.shared.value(for: \\.hideCircularWidget),\n preferredInsideEditorMinWidth: Double = UserDefaults.shared\n .value(for: \\.preferWidgetToStayInsideEditorWhenWidthGreaterThan)\n ) -> WidgetLocation {\n guard let selectedRange: AXValue = try? editor\n .copyValue(key: kAXSelectedTextRangeAttribute),\n let rect: AXValue = try? editor.copyParameterizedValue(\n key: kAXBoundsForRangeParameterizedAttribute,\n parameters: selectedRange\n )\n else {\n return FixedToBottom().framesForWindows(\n windowFrame: windowFrame,\n editorFrame: editorFrame,\n mainScreen: mainScreen,\n activeScreen: activeScreen,\n hideCircularWidget: hideCircularWidget\n )\n }\n var frame: CGRect = .zero\n let found = AXValueGetValue(rect, .cgRect, &frame)\n guard found else {\n return FixedToBottom().framesForWindows(\n windowFrame: windowFrame,\n editorFrame: editorFrame,\n mainScreen: mainScreen,\n activeScreen: activeScreen,\n hideCircularWidget: hideCircularWidget\n )\n }\n return HorizontalMovable().framesForWindows(\n y: mainScreen.frame.height - frame.maxY,\n windowFrame: windowFrame,\n alignPanelTopToAnchor: nil,\n editorFrame: editorFrame,\n mainScreen: mainScreen,\n activeScreen: activeScreen,\n preferredInsideEditorMinWidth: preferredInsideEditorMinWidth,\n hideCircularWidget: hideCircularWidget\n )\n }\n }\n\n struct FixedToBottom {\n func framesForWindows(\n windowFrame: CGRect,\n editorFrame: CGRect,\n mainScreen: NSScreen,\n activeScreen: NSScreen,\n hideCircularWidget: Bool = UserDefaults.shared.value(for: \\.hideCircularWidget),\n preferredInsideEditorMinWidth: Double = UserDefaults.shared\n .value(for: \\.preferWidgetToStayInsideEditorWhenWidthGreaterThan),\n editorFrameExpendedSize: CGSize = .zero\n ) -> WidgetLocation {\n var frames = HorizontalMovable().framesForWindows(\n y: mainScreen.frame.height - editorFrame.maxY + Style.widgetPadding,\n windowFrame: windowFrame,\n alignPanelTopToAnchor: false,\n editorFrame: editorFrame,\n mainScreen: mainScreen,\n activeScreen: activeScreen,\n preferredInsideEditorMinWidth: preferredInsideEditorMinWidth,\n hideCircularWidget: hideCircularWidget,\n editorFrameExpendedSize: editorFrameExpendedSize\n )\n\n frames.sharedPanelLocation.frame.size.height = max(\n frames.defaultPanelLocation.frame.height,\n editorFrame.height - Style.widgetHeight\n )\n frames.defaultPanelLocation.frame.size.height = max(\n frames.defaultPanelLocation.frame.height,\n (editorFrame.height - Style.widgetHeight) / 2\n )\n return frames\n }\n }\n\n struct HorizontalMovable {\n func framesForWindows(\n y: CGFloat,\n windowFrame: CGRect,\n alignPanelTopToAnchor fixedAlignment: Bool?,\n editorFrame: CGRect,\n mainScreen: NSScreen,\n activeScreen: NSScreen,\n preferredInsideEditorMinWidth: Double,\n hideCircularWidget: Bool = UserDefaults.shared.value(for: \\.hideCircularWidget),\n editorFrameExpendedSize: CGSize = .zero\n ) -> WidgetLocation {\n let maxY = max(\n y,\n mainScreen.frame.height - editorFrame.maxY + Style.widgetPadding,\n 4 + activeScreen.frame.minY\n )\n let y = min(\n maxY,\n activeScreen.frame.maxY - 4,\n mainScreen.frame.height - editorFrame.minY - Style.widgetHeight - Style\n .widgetPadding\n )\n\n var proposedAnchorFrameOnTheRightSide = CGRect(\n x: editorFrame.maxX - Style.widgetPadding,\n y: y,\n width: 0,\n height: 0\n )\n\n let widgetFrameOnTheRightSide = CGRect(\n x: editorFrame.maxX - Style.widgetPadding - Style.widgetWidth,\n y: y,\n width: Style.widgetWidth,\n height: Style.widgetHeight\n )\n \n let widgetFrame = CGRect(\n x: windowFrame.minX,\n y: mainScreen.frame.height - windowFrame.maxY + Style.indicatorBottomPadding,\n width: Style.widgetWidth,\n height: Style.widgetHeight\n )\n\n if !hideCircularWidget {\n proposedAnchorFrameOnTheRightSide = widgetFrameOnTheRightSide\n }\n\n let proposedPanelX = proposedAnchorFrameOnTheRightSide.maxX\n + Style.widgetPadding * 2\n - editorFrameExpendedSize.width\n let putPanelToTheRight = {\n if editorFrame.size.width >= preferredInsideEditorMinWidth { return false }\n return activeScreen.frame.maxX > proposedPanelX + Style.panelWidth\n }()\n let alignPanelTopToAnchor = fixedAlignment ?? (y > activeScreen.frame.midY)\n\n if putPanelToTheRight {\n let anchorFrame = proposedAnchorFrameOnTheRightSide\n let panelFrame = CGRect(\n x: proposedPanelX,\n y: alignPanelTopToAnchor\n ? anchorFrame.maxY - Style.panelHeight\n : anchorFrame.minY,\n width: Style.panelWidth,\n height: Style.panelHeight\n )\n let tabFrame = CGRect(\n x: anchorFrame.origin.x,\n y: alignPanelTopToAnchor\n ? anchorFrame.minY - Style.widgetHeight - Style.widgetPadding\n : anchorFrame.maxY + Style.widgetPadding,\n width: Style.widgetWidth,\n height: Style.widgetHeight\n )\n\n return .init(\n widgetFrame: widgetFrame,\n tabFrame: tabFrame,\n sharedPanelLocation: .init(\n frame: panelFrame,\n alignPanelTop: alignPanelTopToAnchor\n ),\n defaultPanelLocation: .init(\n frame: panelFrame,\n alignPanelTop: alignPanelTopToAnchor\n ),\n suggestionPanelLocation: nil\n )\n } else {\n var proposedAnchorFrameOnTheLeftSide = CGRect(\n x: editorFrame.minX + Style.widgetPadding,\n y: proposedAnchorFrameOnTheRightSide.origin.y,\n width: 0,\n height: 0\n )\n\n let widgetFrameOnTheLeftSide = CGRect(\n x: editorFrame.minX + Style.widgetPadding,\n y: proposedAnchorFrameOnTheRightSide.origin.y,\n width: Style.widgetWidth,\n height: Style.widgetHeight\n )\n\n if !hideCircularWidget {\n proposedAnchorFrameOnTheLeftSide = widgetFrameOnTheLeftSide\n }\n\n let proposedPanelX = proposedAnchorFrameOnTheLeftSide.minX\n - Style.widgetPadding * 2\n - Style.panelWidth\n + editorFrameExpendedSize.width\n let putAnchorToTheLeft = {\n if editorFrame.size.width >= preferredInsideEditorMinWidth {\n if editorFrame.maxX <= activeScreen.frame.maxX {\n return false\n }\n }\n return proposedPanelX > activeScreen.frame.minX\n }()\n\n if putAnchorToTheLeft {\n let anchorFrame = proposedAnchorFrameOnTheLeftSide\n let panelFrame = CGRect(\n x: proposedPanelX,\n y: alignPanelTopToAnchor\n ? anchorFrame.maxY - Style.panelHeight\n : anchorFrame.minY - editorFrameExpendedSize.height,\n width: Style.panelWidth,\n height: Style.panelHeight\n )\n let tabFrame = CGRect(\n x: anchorFrame.origin.x,\n y: alignPanelTopToAnchor\n ? anchorFrame.minY - Style.widgetHeight - Style.widgetPadding\n : anchorFrame.maxY + Style.widgetPadding,\n width: Style.widgetWidth,\n height: Style.widgetHeight\n )\n return .init(\n widgetFrame: widgetFrame,\n tabFrame: tabFrame,\n sharedPanelLocation: .init(\n frame: panelFrame,\n alignPanelTop: alignPanelTopToAnchor\n ),\n defaultPanelLocation: .init(\n frame: panelFrame,\n alignPanelTop: alignPanelTopToAnchor\n ),\n suggestionPanelLocation: nil\n )\n } else {\n let anchorFrame = proposedAnchorFrameOnTheRightSide\n let panelFrame = CGRect(\n x: anchorFrame.maxX - Style.panelWidth,\n y: alignPanelTopToAnchor\n ? anchorFrame.maxY - Style.panelHeight\n : anchorFrame.maxY - editorFrameExpendedSize.height,\n width: Style.panelWidth,\n height: Style.panelHeight\n )\n let tabFrame = CGRect(\n x: anchorFrame.minX - Style.widgetPadding - Style.widgetWidth,\n y: anchorFrame.origin.y,\n width: Style.widgetWidth,\n height: Style.widgetHeight\n )\n return .init(\n widgetFrame: widgetFrame,\n tabFrame: tabFrame,\n sharedPanelLocation: .init(\n frame: panelFrame,\n alignPanelTop: alignPanelTopToAnchor\n ),\n defaultPanelLocation: .init(\n frame: panelFrame,\n alignPanelTop: alignPanelTopToAnchor\n ),\n suggestionPanelLocation: nil\n )\n }\n }\n }\n }\n\n struct NearbyTextCursor {\n func framesForSuggestionWindow(\n editorFrame: CGRect,\n mainScreen: NSScreen,\n activeScreen: NSScreen,\n editor: AXUIElement,\n completionPanel: AXUIElement?\n ) -> WidgetLocation.PanelLocation? {\n guard let selectionFrame = UpdateLocationStrategy\n .getSelectionFirstLineFrame(editor: editor) else { return nil }\n\n // hide it when the line of code is outside of the editor visible rect\n if selectionFrame.maxY < editorFrame.minY || selectionFrame.minY > editorFrame.maxY {\n return nil\n }\n\n let proposedY = mainScreen.frame.height - selectionFrame.maxY\n let proposedX = selectionFrame.maxX - 40\n let maxY = max(\n proposedY,\n 4 + activeScreen.frame.minY\n )\n let y = min(\n maxY,\n activeScreen.frame.maxY - 4\n )\n\n // align panel to top == place under the selection frame.\n // we initially try to place it at the bottom side, but if there is no enough space\n // we move it to the top of the selection frame.\n let alignPanelTopToAnchor = y - Style.inlineSuggestionMaxHeight\n >= activeScreen.frame.minY\n\n let caseIgnoreCompletionPanel = {\n (alignPanelTopToAnchor: Bool) -> WidgetLocation.PanelLocation? in\n let x: Double = {\n if proposedX + Style.inlineSuggestionMinWidth <= activeScreen.frame.maxX {\n return proposedX\n }\n return activeScreen.frame.maxX - Style.inlineSuggestionMinWidth\n }()\n if alignPanelTopToAnchor {\n // case: present under selection\n return .init(\n frame: .init(\n x: x,\n y: y - Style.inlineSuggestionMaxHeight,\n width: Style.inlineSuggestionMinWidth,\n height: Style.inlineSuggestionMaxHeight\n ),\n alignPanelTop: alignPanelTopToAnchor\n )\n } else {\n // case: present above selection\n return .init(\n frame: .init(\n x: x,\n y: y + selectionFrame.height + Style.widgetPadding,\n width: Style.inlineSuggestionMinWidth,\n height: Style.inlineSuggestionMaxHeight\n ),\n alignPanelTop: alignPanelTopToAnchor\n )\n }\n }\n\n let caseConsiderCompletionPanel = {\n (completionPanelRect: CGRect) -> WidgetLocation.PanelLocation? in\n let completionPanelBelowCursor = completionPanelRect.minY >= selectionFrame.midY\n switch (completionPanelBelowCursor, alignPanelTopToAnchor) {\n case (true, false), (false, true):\n // case: different position, place the suggestion as it should be\n return caseIgnoreCompletionPanel(alignPanelTopToAnchor)\n case (true, true), (false, false):\n // case: same position, place the suggestion next to the completion panel\n let y = completionPanelBelowCursor\n ? y - Style.inlineSuggestionMaxHeight\n : y + selectionFrame.height - Style.widgetPadding\n if let x = {\n let proposedX = completionPanelRect.maxX + Style.widgetPadding\n if proposedX + Style.inlineSuggestionMinWidth <= activeScreen.frame.maxX {\n return proposedX\n }\n let leftSideX = completionPanelRect.minX\n - Style.widgetPadding\n - Style.inlineSuggestionMinWidth\n if leftSideX >= activeScreen.frame.minX {\n return leftSideX\n }\n return nil\n }() {\n return .init(\n frame: .init(\n x: x,\n y: y,\n width: Style.inlineSuggestionMinWidth,\n height: Style.inlineSuggestionMaxHeight\n ),\n alignPanelTop: alignPanelTopToAnchor\n )\n }\n // case: no enough horizontal space, place the suggestion on the other side\n return caseIgnoreCompletionPanel(!alignPanelTopToAnchor)\n }\n }\n\n if let completionPanel, let completionPanelRect = completionPanel.rect {\n return caseConsiderCompletionPanel(completionPanelRect)\n } else {\n return caseIgnoreCompletionPanel(alignPanelTopToAnchor)\n }\n }\n }\n\n /// Get the frame of the selection.\n static func getSelectionFrame(editor: AXUIElement) -> CGRect? {\n guard let selectedRange: AXValue = try? editor\n .copyValue(key: kAXSelectedTextRangeAttribute),\n let rect: AXValue = try? editor.copyParameterizedValue(\n key: kAXBoundsForRangeParameterizedAttribute,\n parameters: selectedRange\n )\n else {\n return nil\n }\n var selectionFrame: CGRect = .zero\n let found = AXValueGetValue(rect, .cgRect, &selectionFrame)\n guard found else { return nil }\n return selectionFrame\n }\n\n /// Get the frame of the first line of the selection.\n static func getSelectionFirstLineFrame(editor: AXUIElement) -> CGRect? {\n // Find selection range rect\n guard let selectedRange: AXValue = try? editor\n .copyValue(key: kAXSelectedTextRangeAttribute),\n let rect: AXValue = try? editor.copyParameterizedValue(\n key: kAXBoundsForRangeParameterizedAttribute,\n parameters: selectedRange\n )\n else {\n return nil\n }\n var selectionFrame: CGRect = .zero\n let found = AXValueGetValue(rect, .cgRect, &selectionFrame)\n guard found else { return nil }\n\n var firstLineRange: CFRange = .init()\n let foundFirstLine = AXValueGetValue(selectedRange, .cfRange, &firstLineRange)\n firstLineRange.length = 0\n\n if foundFirstLine,\n let firstLineSelectionRange = AXValueCreate(.cfRange, &firstLineRange),\n let firstLineRect: AXValue = try? editor.copyParameterizedValue(\n key: kAXBoundsForRangeParameterizedAttribute,\n parameters: firstLineSelectionRange\n )\n {\n var firstLineFrame: CGRect = .zero\n let foundFirstLineFrame = AXValueGetValue(firstLineRect, .cgRect, &firstLineFrame)\n if foundFirstLineFrame {\n selectionFrame = firstLineFrame\n }\n }\n\n return selectionFrame\n }\n}\n\n" }, { "path": "Core/Sources/SuggestionWidget/WidgetView.swift", "content": "import ActiveApplicationMonitor\nimport ComposableArchitecture\nimport Preferences\nimport SharedUIComponents\nimport SuggestionBasic\nimport SwiftUI\n\nstruct WidgetView: View {\n let store: StoreOf\n @State var isHovering: Bool = false\n var onOpenChatClicked: () -> Void = {}\n var onCustomCommandClicked: (CustomCommand) -> Void = { _ in }\n\n @AppStorage(\\.hideCircularWidget) var hideCircularWidget\n\n var body: some View {\n GeometryReader { _ in\n WithPerceptionTracking {\n ZStack {\n WidgetAnimatedCapsule(\n store: store,\n isHovering: isHovering\n )\n }\n .onTapGesture {\n store.send(.widgetClicked, animation: .easeInOut(duration: 0.2))\n }\n .onHover { yes in\n withAnimation(.easeInOut(duration: 0.14)) {\n isHovering = yes\n }\n }\n .contextMenu {\n WidgetContextMenu(store: store)\n }\n .opacity({\n if !hideCircularWidget { return 1 }\n return store.isProcessing ? 1 : 0\n }())\n .animation(\n .easeInOut(duration: 0.2),\n value: isHovering\n )\n .animation(\n .easeInOut(duration: 0.4),\n value: store.isProcessing\n )\n }\n }\n }\n}\n\nstruct WidgetAnimatedCapsule: View {\n let store: StoreOf\n var isHovering: Bool\n\n @State private var breathingOpacity: CGFloat = 1.0\n @State private var animationTask: Task?\n\n var body: some View {\n GeometryReader { geo in\n WithPerceptionTracking {\n let capsuleWidth = geo.size.width\n let capsuleHeight = geo.size.height\n\n let backgroundWidth = capsuleWidth\n let foregroundWidth = max(capsuleWidth - 4, 2)\n let padding = (backgroundWidth - foregroundWidth) / 2\n let foregroundHeight = capsuleHeight - padding * 2\n\n ZStack {\n Capsule()\n .modify {\n if #available(macOS 26.0, *) {\n $0.glassEffect()\n } else if #available(macOS 13.0, *) {\n $0.backgroundStyle(.thickMaterial.opacity(0.8)).overlay(\n Capsule().stroke(\n Color(nsColor: .darkGray).opacity(0.2),\n lineWidth: 1\n )\n )\n } else {\n $0.fill(Color(nsColor: .darkGray).opacity(0.6)).overlay(\n Capsule().stroke(\n Color(nsColor: .darkGray).opacity(0.2),\n lineWidth: 1\n )\n )\n }\n }\n .frame(width: backgroundWidth, height: capsuleHeight)\n\n Capsule()\n .fill(Color.white)\n .frame(\n width: foregroundWidth,\n height: foregroundHeight\n )\n .opacity({\n let base = store.isProcessing ? breathingOpacity : 0\n if isHovering {\n return min(base + 0.5, 1.0)\n }\n return base\n }())\n .blur(radius: 2)\n }\n .onAppear {\n updateBreathingAnimation(isProcessing: store.isProcessing)\n }\n .onChange(of: store.isProcessing) { newValue in\n updateBreathingAnimation(isProcessing: newValue)\n }\n }\n }\n }\n\n private func updateBreathingAnimation(isProcessing: Bool) {\n animationTask?.cancel()\n animationTask = nil\n\n if isProcessing {\n animationTask = Task {\n while !Task.isCancelled {\n await MainActor.run {\n withAnimation(.easeInOut(duration: 1.2)) {\n breathingOpacity = 0.3\n }\n }\n try? await Task.sleep(nanoseconds: UInt64(1.2 * 1_000_000_000))\n if Task.isCancelled { break }\n if !(store.isProcessing) { break }\n await MainActor.run {\n withAnimation(.easeInOut(duration: 1.2)) {\n breathingOpacity = 1.0\n }\n }\n try? await Task.sleep(nanoseconds: UInt64(1.2 * 1_000_000_000))\n if Task.isCancelled { break }\n if !(store.isProcessing) { break }\n }\n }\n } else {\n withAnimation(.easeInOut(duration: 0.2)) {\n breathingOpacity = 0\n }\n }\n }\n}\n\nstruct WidgetContextMenu: View {\n @AppStorage(\\.useGlobalChat) var useGlobalChat\n @AppStorage(\\.realtimeSuggestionToggle) var realtimeSuggestionToggle\n @AppStorage(\\.disableSuggestionFeatureGlobally) var disableSuggestionFeatureGlobally\n @AppStorage(\\.suggestionFeatureEnabledProjectList) var suggestionFeatureEnabledProjectList\n @AppStorage(\\.suggestionFeatureDisabledLanguageList) var suggestionFeatureDisabledLanguageList\n @AppStorage(\\.customCommands) var customCommands\n let store: StoreOf\n\n @Dependency(\\.xcodeInspector) var xcodeInspector\n\n var body: some View {\n WithPerceptionTracking {\n Group { // Commands\n Button(action: {\n store.send(.openChatButtonClicked)\n }) {\n Text(\"Open Chat\")\n }\n\n Button(action: {\n store.send(.openModificationButtonClicked)\n }) {\n Text(\"Write or Edit Code\")\n }\n\n customCommandMenu()\n }\n\n Divider()\n\n Group {\n enableSuggestionForProject\n\n disableSuggestionForLanguage\n }\n\n Divider()\n\n Group { // Settings\n Button(action: {\n store.send(.detachChatPanelToggleClicked)\n }) {\n Text(\"Detach Chat Panel\")\n if store.isChatPanelDetached {\n Image(systemName: \"checkmark\")\n }\n }\n\n Button(action: {\n realtimeSuggestionToggle.toggle()\n }) {\n Text(\"Realtime Suggestion\")\n if realtimeSuggestionToggle {\n Image(systemName: \"checkmark\")\n }\n }\n }\n\n Divider()\n }\n }\n\n func customCommandMenu() -> some View {\n Menu(\"Custom Commands\") {\n ForEach(customCommands, id: \\.name) { command in\n Button(action: {\n store.send(.runCustomCommandButtonClicked(command))\n }) {\n Text(command.name)\n }\n }\n }\n }\n}\n\nextension WidgetContextMenu {\n @ViewBuilder\n var enableSuggestionForProject: some View {\n if let projectPath = xcodeInspector.activeProjectRootURL?.path,\n disableSuggestionFeatureGlobally\n {\n let matchedPath = suggestionFeatureEnabledProjectList.first { path in\n projectPath.hasPrefix(path)\n }\n Button(action: {\n if matchedPath != nil {\n suggestionFeatureEnabledProjectList\n .removeAll { path in path == matchedPath }\n } else {\n suggestionFeatureEnabledProjectList.append(projectPath)\n }\n }) {\n if matchedPath == nil {\n Text(\"Add to Suggestion-Enabled Project List\")\n } else {\n Text(\"Remove from Suggestion-Enabled Project List\")\n }\n }\n }\n }\n\n @ViewBuilder\n var disableSuggestionForLanguage: some View {\n let fileURL = xcodeInspector.activeDocumentURL\n let fileLanguage = fileURL.map(languageIdentifierFromFileURL) ?? .plaintext\n let matched = suggestionFeatureDisabledLanguageList.first { rawValue in\n fileLanguage.rawValue == rawValue\n }\n Button(action: {\n if let matched {\n suggestionFeatureDisabledLanguageList.removeAll { $0 == matched }\n } else {\n suggestionFeatureDisabledLanguageList.append(fileLanguage.rawValue)\n }\n }) {\n if matched == nil {\n Text(\"Disable Suggestion for \\\"\\(fileLanguage.rawValue.capitalized)\\\"\")\n } else {\n Text(\"Enable Suggestion for \\\"\\(fileLanguage.rawValue.capitalized)\\\"\")\n }\n }\n }\n}\n\nstruct WidgetView_Preview: PreviewProvider {\n static var previews: some View {\n VStack {\n WidgetView(\n store: Store(\n initialState: .init(\n isProcessing: false,\n isDisplayingContent: false,\n isContentEmpty: true,\n isChatPanelDetached: false,\n isChatOpen: false\n ),\n reducer: { CircularWidget() }\n ),\n isHovering: false\n )\n .frame(width: Style.widgetWidth, height: Style.widgetHeight)\n\n WidgetView(\n store: Store(\n initialState: .init(\n isProcessing: false,\n isDisplayingContent: false,\n isContentEmpty: true,\n isChatPanelDetached: false,\n isChatOpen: false\n ),\n reducer: { CircularWidget() }\n ),\n isHovering: true\n )\n .frame(width: Style.widgetWidth, height: Style.widgetHeight)\n\n WidgetView(\n store: Store(\n initialState: .init(\n isProcessing: true,\n isDisplayingContent: false,\n isContentEmpty: true,\n isChatPanelDetached: false,\n isChatOpen: false\n ),\n reducer: { CircularWidget() }\n ),\n isHovering: false\n )\n .frame(width: Style.widgetWidth, height: Style.widgetHeight)\n\n WidgetView(\n store: Store(\n initialState: .init(\n isProcessing: false,\n isDisplayingContent: true,\n isContentEmpty: true,\n isChatPanelDetached: false,\n isChatOpen: false\n ),\n reducer: { CircularWidget() }\n ),\n isHovering: false\n )\n .frame(width: Style.widgetWidth, height: Style.widgetHeight)\n }\n .frame(width: 200, height: 200)\n .background(Color.black)\n }\n}\n\n" }, { "path": "Core/Sources/SuggestionWidget/WidgetWindowsController.swift", "content": "import AppKit\nimport AsyncAlgorithms\nimport ChatTab\nimport ComposableArchitecture\nimport Dependencies\nimport Foundation\nimport SharedUIComponents\nimport SwiftNavigation\nimport SwiftUI\nimport XcodeInspector\n\n#warning(\"\"\"\nTODO: This part is too messy, consider breaking it up, let each window handle their own things\n\"\"\")\n\nactor WidgetWindowsController: NSObject {\n let userDefaultsObservers = WidgetUserDefaultsObservers()\n var xcodeInspector: XcodeInspector { .shared }\n\n nonisolated let windows: WidgetWindows\n nonisolated let store: StoreOf\n nonisolated let chatTabPool: ChatTabPool\n\n var currentApplicationProcessIdentifier: pid_t?\n\n var observeToAppTask: Task?\n var observeToFocusedEditorTask: Task?\n\n var updateWindowOpacityTask: Task?\n var lastUpdateWindowOpacityTime = Date(timeIntervalSince1970: 0)\n\n var updateWindowLocationTask: Task?\n var lastUpdateWindowLocationTime = Date(timeIntervalSince1970: 0)\n\n var beatingCompletionPanelTask: Task?\n var updateWindowStateTask: Task?\n\n deinit {\n userDefaultsObservers.presentationModeChangeObserver.onChange = {}\n observeToAppTask?.cancel()\n observeToFocusedEditorTask?.cancel()\n updateWindowStateTask?.cancel()\n }\n\n init(store: StoreOf, chatTabPool: ChatTabPool) {\n self.store = store\n self.chatTabPool = chatTabPool\n windows = .init(store: store, chatTabPool: chatTabPool)\n super.init()\n windows.controller = self\n }\n\n @MainActor func send(_ action: Widget.Action) {\n store.send(action)\n }\n\n func start() {\n Task { [xcodeInspector] in\n await observe { [weak self] in\n if let app = xcodeInspector.activeApplication {\n Task {\n await self?.activate(app)\n }\n }\n }\n\n await observe { [weak self] in\n if let editor = xcodeInspector.focusedEditor {\n Task {\n await self?.observe(toEditor: editor)\n }\n }\n }\n\n await observe { [weak self] in\n let isDisplaying = xcodeInspector.completionPanel != nil\n Task {\n await self?.handleCompletionPanelChange(isDisplaying: isDisplaying)\n }\n }\n }\n\n userDefaultsObservers.presentationModeChangeObserver.onChange = { [weak self] in\n Task { [weak self] in\n await self?.updateWindowLocation(animated: false, immediately: false)\n await self?.send(.updateColorScheme)\n }\n }\n\n updateWindowStateTask = Task { [weak self] in\n if let self { await handleSpaceChange() }\n\n await withThrowingTaskGroup(of: Void.self) { [weak self] group in\n // active space did change\n _ = group.addTaskUnlessCancelled { [weak self] in\n let sequence = NSWorkspace.shared.notificationCenter\n .notifications(named: NSWorkspace.activeSpaceDidChangeNotification)\n for await _ in sequence {\n guard let self else { return }\n try Task.checkCancellation()\n await handleSpaceChange()\n }\n }\n }\n }\n\n Task { @MainActor in\n windows.chatPanelWindow.isPanelDisplayed = false\n }\n }\n}\n\n// MARK: - Observation\n\nprivate extension WidgetWindowsController {\n func activate(_ app: AppInstanceInspector) {\n Task {\n if app.isXcode {\n updateWindowLocation(animated: false, immediately: true)\n updateWindowOpacity(immediately: false)\n } else {\n updateWindowOpacity(immediately: true)\n updateWindowLocation(animated: false, immediately: false)\n await hideSuggestionPanelWindow()\n }\n await adjustChatPanelWindowLevel()\n await adjustModificationPanelLevel()\n }\n guard currentApplicationProcessIdentifier != app.processIdentifier else { return }\n currentApplicationProcessIdentifier = app.processIdentifier\n observe(toApp: app)\n }\n\n func observe(toApp app: AppInstanceInspector) {\n guard let app = app as? XcodeAppInstanceInspector else { return }\n let notifications = app.axNotifications\n observeToAppTask?.cancel()\n observeToAppTask = Task {\n await windows.orderFront()\n\n /// Hide the widgets before switching to another window/editor\n /// so the transition looks better.\n func hideWidgetForTransitions() async {\n let newDocumentURL = xcodeInspector.realtimeActiveDocumentURL\n let documentURL = await MainActor\n .run { store.withState { $0.focusingDocumentURL } }\n if documentURL != newDocumentURL {\n await send(.panel(.removeDisplayedContent))\n await hidePanelWindows()\n }\n await send(.updateFocusingDocumentURL)\n }\n\n func removeContent() async {\n await send(.panel(.removeDisplayedContent))\n }\n\n func updateWidgetsAndNotifyChangeOfEditor(immediately: Bool) async {\n await send(.panel(.switchToAnotherEditorAndUpdateContent))\n updateWindowLocation(animated: false, immediately: immediately)\n updateWindowOpacity(immediately: immediately)\n }\n\n func updateWidgets(immediately: Bool) async {\n updateWindowLocation(animated: false, immediately: immediately)\n updateWindowOpacity(immediately: immediately)\n }\n\n await updateWidgetsAndNotifyChangeOfEditor(immediately: true)\n\n for await notification in await notifications.notifications() {\n try Task.checkCancellation()\n\n switch notification.kind {\n case .focusedWindowChanged:\n await handleSpaceChange()\n await hideWidgetForTransitions()\n await updateWidgetsAndNotifyChangeOfEditor(immediately: true)\n case .focusedUIElementChanged:\n await hideWidgetForTransitions()\n await updateWidgetsAndNotifyChangeOfEditor(immediately: true)\n case .applicationActivated:\n await removeContent()\n await updateWidgetsAndNotifyChangeOfEditor(immediately: false)\n case .mainWindowChanged:\n await removeContent()\n await updateWidgetsAndNotifyChangeOfEditor(immediately: false)\n case .moved,\n .resized,\n .windowMoved,\n .windowResized,\n .windowMiniaturized,\n .windowDeminiaturized:\n await updateWidgets(immediately: false)\n case .created, .uiElementDestroyed, .xcodeCompletionPanelChanged,\n .applicationDeactivated:\n continue\n case .titleChanged:\n continue\n }\n }\n }\n }\n\n func observe(toEditor editor: SourceEditor) {\n observeToFocusedEditorTask?.cancel()\n observeToFocusedEditorTask = Task {\n let selectionRangeChange = await editor.axNotifications.notifications()\n .filter { $0.kind == .selectedTextChanged }\n let scroll = await editor.axNotifications.notifications()\n .filter { $0.kind == .scrollPositionChanged }\n\n if #available(macOS 13.0, *) {\n for await notification in merge(\n selectionRangeChange.debounce(for: Duration.milliseconds(500)),\n scroll\n ) {\n guard await xcodeInspector.latestActiveXcode != nil else { return }\n try Task.checkCancellation()\n\n // for better looking\n if notification.kind == .scrollPositionChanged {\n await hideSuggestionPanelWindow()\n }\n\n updateWindowLocation(animated: false, immediately: false)\n updateWindowOpacity(immediately: false)\n }\n } else {\n for await notification in merge(selectionRangeChange, scroll) {\n guard await xcodeInspector.latestActiveXcode != nil else { return }\n try Task.checkCancellation()\n\n // for better looking\n if notification.kind == .scrollPositionChanged {\n await hideSuggestionPanelWindow()\n }\n\n updateWindowLocation(animated: false, immediately: false)\n updateWindowOpacity(immediately: false)\n }\n }\n }\n }\n\n func handleCompletionPanelChange(isDisplaying: Bool) {\n beatingCompletionPanelTask?.cancel()\n beatingCompletionPanelTask = Task {\n if !isDisplaying {\n // so that the buttons on the suggestion panel could be\n // clicked\n // before the completion panel updates the location of the\n // suggestion panel\n try await Task.sleep(nanoseconds: 400_000_000)\n }\n\n updateWindowLocation(animated: false, immediately: false)\n updateWindowOpacity(immediately: false)\n }\n }\n}\n\n// MARK: - Window Updating\n\nextension WidgetWindowsController {\n @MainActor\n func hidePanelWindows() {\n// windows.sharedPanelWindow.alphaValue = 0\n windows.suggestionPanelWindow.alphaValue = 0\n }\n\n @MainActor\n func hideSuggestionPanelWindow() {\n windows.suggestionPanelWindow.alphaValue = 0\n }\n\n func generateWidgetLocation() async -> WidgetLocation? {\n if let application = await xcodeInspector.latestActiveXcode?.appElement {\n if let window = application.focusedWindow,\n let windowFrame = window.rect,\n let focusElement = await xcodeInspector.focusedEditor?.element,\n let parent = focusElement.parent,\n let frame = parent.rect,\n let screen = NSScreen.screens.first(\n where: { $0.frame.origin == .zero }\n ) ?? NSScreen.main,\n let windowContainingScreen = NSScreen.screens.first(where: {\n let flippedScreenFrame = $0.frame.flipped(relativeTo: screen.frame)\n return flippedScreenFrame.contains(frame.origin)\n })\n {\n let positionMode = UserDefaults.shared\n .value(for: \\.suggestionWidgetPositionMode)\n let suggestionMode = UserDefaults.shared\n .value(for: \\.suggestionPresentationMode)\n\n switch positionMode {\n case .fixedToBottom:\n var result = UpdateLocationStrategy.FixedToBottom().framesForWindows(\n windowFrame: windowFrame,\n editorFrame: frame,\n mainScreen: screen,\n activeScreen: windowContainingScreen\n )\n switch suggestionMode {\n case .nearbyTextCursor:\n result.suggestionPanelLocation = UpdateLocationStrategy\n .NearbyTextCursor()\n .framesForSuggestionWindow(\n editorFrame: frame,\n mainScreen: screen,\n activeScreen: windowContainingScreen,\n editor: focusElement,\n completionPanel: await xcodeInspector.completionPanel\n )\n default:\n break\n }\n return result\n case .alignToTextCursor:\n var result = UpdateLocationStrategy.AlignToTextCursor().framesForWindows(\n windowFrame: windowFrame,\n editorFrame: frame,\n mainScreen: screen,\n activeScreen: windowContainingScreen,\n editor: focusElement\n )\n switch suggestionMode {\n case .nearbyTextCursor:\n result.suggestionPanelLocation = UpdateLocationStrategy\n .NearbyTextCursor()\n .framesForSuggestionWindow(\n editorFrame: frame, mainScreen: screen,\n activeScreen: windowContainingScreen,\n editor: focusElement,\n completionPanel: await xcodeInspector.completionPanel\n )\n default:\n break\n }\n return result\n }\n } else if var window = application.focusedWindow,\n var frame = application.focusedWindow?.rect,\n ![\"menu bar\", \"menu bar item\"].contains(window.description),\n frame.size.height > 300,\n let screen = NSScreen.screens.first(where: { $0.frame.origin == .zero }),\n let firstScreen = NSScreen.main\n {\n if [\"open_quickly\"].contains(window.identifier)\n || [\"alert\"].contains(window.label)\n {\n // fallback to use workspace window\n guard let workspaceWindow = application.windows\n .first(where: { $0.identifier == \"Xcode.WorkspaceWindow\" }),\n let rect = workspaceWindow.rect\n else {\n return WidgetLocation(\n widgetFrame: .zero,\n tabFrame: .zero,\n sharedPanelLocation: .init(frame: .zero, alignPanelTop: false),\n defaultPanelLocation: .init(frame: .zero, alignPanelTop: false)\n )\n }\n \n window = workspaceWindow\n frame = rect\n }\n\n return UpdateLocationStrategy.FixedToBottom().framesForWindows(\n windowFrame: frame,\n editorFrame: frame,\n mainScreen: screen,\n activeScreen: firstScreen,\n preferredInsideEditorMinWidth: 9_999_999_999, // never\n editorFrameExpendedSize: .zero\n )\n }\n }\n return nil\n }\n\n func updatePanelState(_ location: WidgetLocation) async {\n await send(.updatePanelStateToMatch(location))\n }\n\n func updateWindowOpacity(immediately: Bool) {\n let shouldDebounce = !immediately &&\n !(Date().timeIntervalSince(lastUpdateWindowOpacityTime) > 3)\n lastUpdateWindowOpacityTime = Date()\n updateWindowOpacityTask?.cancel()\n\n let task = Task {\n if shouldDebounce {\n try await Task.sleep(nanoseconds: 200_000_000)\n }\n try Task.checkCancellation()\n let xcodeInspector = self.xcodeInspector\n let activeApp = await xcodeInspector.activeApplication\n let latestActiveXcode = await xcodeInspector.latestActiveXcode\n let previousActiveApplication = await xcodeInspector.previousActiveApplication\n await MainActor.run {\n if let activeApp, activeApp.isXcode {\n let application = activeApp.appElement\n /// We need this to hide the windows when Xcode is minimized.\n let noFocus = application.focusedWindow == nil\n windows.sharedPanelWindow.alphaValue = 1\n windows.suggestionPanelWindow.alphaValue = noFocus ? 0 : 1\n windows.widgetWindow.alphaValue = noFocus ? 0 : 1\n windows.toastWindow.alphaValue = noFocus ? 0 : 1\n } else if let activeApp, activeApp.isExtensionService {\n let noFocus = {\n guard let xcode = latestActiveXcode else { return true }\n if let window = xcode.appElement.focusedWindow,\n window.role == \"AXWindow\"\n {\n return false\n }\n return true\n }()\n\n let previousAppIsXcode = previousActiveApplication?.isXcode ?? false\n\n windows.sharedPanelWindow.alphaValue = 1\n windows.suggestionPanelWindow.alphaValue = noFocus ? 0 : 1\n windows.widgetWindow.alphaValue = if noFocus {\n 0\n } else if previousAppIsXcode {\n if windows.chatPanelWindow.isFullscreen,\n windows.chatPanelWindow.isOnActiveSpace\n {\n 0\n } else {\n 1\n }\n } else {\n 0\n }\n windows.toastWindow.alphaValue = noFocus ? 0 : 1\n } else {\n windows.sharedPanelWindow.alphaValue = 1\n windows.suggestionPanelWindow.alphaValue = 0\n windows.widgetWindow.alphaValue = 0\n windows.toastWindow.alphaValue = 0\n }\n }\n }\n\n updateWindowOpacityTask = task\n }\n\n func updateWindowLocation(\n animated: Bool,\n immediately: Bool,\n function: StaticString = #function,\n line: UInt = #line\n ) {\n @Sendable @MainActor\n func update() async {\n let state = store.withState { $0 }\n let isChatPanelDetached = state.chatPanelState.isDetached\n guard let widgetLocation = await generateWidgetLocation() else { return }\n await updatePanelState(widgetLocation)\n\n windows.widgetWindow.setFrame(\n widgetLocation.widgetFrame,\n display: false,\n animate: animated\n )\n windows.toastWindow.setFrame(\n widgetLocation.defaultPanelLocation.frame,\n display: false,\n animate: animated\n )\n windows.sharedPanelWindow.setFrame(\n widgetLocation.sharedPanelLocation.frame,\n display: false,\n animate: animated\n )\n\n if let suggestionPanelLocation = widgetLocation.suggestionPanelLocation {\n windows.suggestionPanelWindow.setFrame(\n suggestionPanelLocation.frame,\n display: false,\n animate: animated\n )\n }\n\n if isChatPanelDetached {\n // don't update it!\n } else {\n windows.chatPanelWindow.setFrame(\n widgetLocation.defaultPanelLocation.frame,\n display: false,\n animate: animated\n )\n }\n\n await adjustChatPanelWindowLevel()\n await adjustModificationPanelLevel()\n }\n\n let now = Date()\n let shouldThrottle = !immediately &&\n !(now.timeIntervalSince(lastUpdateWindowLocationTime) > 3)\n\n updateWindowLocationTask?.cancel()\n let interval: TimeInterval = 0.05\n\n if shouldThrottle {\n let delay = max(\n 0,\n interval - now.timeIntervalSince(lastUpdateWindowLocationTime)\n )\n\n updateWindowLocationTask = Task {\n try await Task.sleep(nanoseconds: UInt64(delay * 500_000_000))\n try Task.checkCancellation()\n await update()\n }\n } else {\n Task {\n await update()\n }\n }\n lastUpdateWindowLocationTime = Date()\n }\n\n @MainActor\n func adjustModificationPanelLevel() async {\n let window = windows.sharedPanelWindow\n\n let latestApp = await xcodeInspector.activeApplication\n let latestAppIsXcodeOrExtension = if let latestApp {\n latestApp.isXcode || latestApp.isExtensionService\n } else {\n false\n }\n\n window.setFloatOnTop(latestAppIsXcodeOrExtension)\n }\n\n @MainActor\n func adjustChatPanelWindowLevel() async {\n let flowOnTopOption = UserDefaults.shared\n .value(for: \\.chatPanelFloatOnTopOption)\n let disableFloatOnTopWhenTheChatPanelIsDetached = UserDefaults.shared\n .value(for: \\.disableFloatOnTopWhenTheChatPanelIsDetached)\n\n let window = windows.chatPanelWindow\n\n if flowOnTopOption == .never {\n window.setFloatOnTop(false)\n return\n }\n\n let state = store.withState { $0 }\n let isChatPanelDetached = state.chatPanelState.isDetached\n\n let floatOnTopWhenOverlapsXcode = UserDefaults.shared\n .value(for: \\.keepFloatOnTopIfChatPanelAndXcodeOverlaps)\n\n let latestApp = await xcodeInspector.activeApplication\n let latestAppIsXcodeOrExtension = if let latestApp {\n latestApp.isXcode || latestApp.isExtensionService\n } else {\n false\n }\n\n async let overlap: Bool = { @MainActor in\n guard let xcode = await xcodeInspector.latestActiveXcode else { return false }\n let windowElements = xcode.appElement.windows\n let overlap = windowElements.contains {\n if let position = $0.position, let size = $0.size {\n let rect = CGRect(\n x: position.x,\n y: position.y,\n width: size.width,\n height: size.height\n )\n return rect.intersects(window.frame)\n }\n return false\n }\n return overlap\n }()\n\n if latestAppIsXcodeOrExtension {\n if floatOnTopWhenOverlapsXcode {\n let overlap = await overlap\n window.setFloatOnTop(overlap)\n } else {\n if disableFloatOnTopWhenTheChatPanelIsDetached, isChatPanelDetached {\n window.setFloatOnTop(false)\n } else {\n window.setFloatOnTop(true)\n }\n }\n } else {\n if floatOnTopWhenOverlapsXcode {\n let overlap = await overlap\n window.setFloatOnTop(overlap)\n } else {\n switch flowOnTopOption {\n case .onTopWhenXcodeIsActive:\n window.setFloatOnTop(false)\n case .alwaysOnTop:\n window.setFloatOnTop(true)\n case .never:\n window.setFloatOnTop(false)\n }\n }\n }\n }\n\n @MainActor\n func handleSpaceChange() async {\n let activeXcode = XcodeInspector.shared.activeXcode\n\n let xcode = activeXcode?.appElement\n\n let isXcodeActive = xcode?.isFrontmost ?? false\n\n [\n windows.sharedPanelWindow,\n windows.suggestionPanelWindow,\n windows.widgetWindow,\n windows.toastWindow,\n ].forEach {\n if isXcodeActive {\n $0.moveToActiveSpace()\n }\n }\n\n if isXcodeActive, !windows.chatPanelWindow.isDetached {\n windows.chatPanelWindow.moveToActiveSpace()\n }\n\n if windows.fullscreenDetector.isOnActiveSpace, xcode?.focusedWindow != nil {\n windows.orderFront()\n }\n }\n}\n\n// MARK: - NSWindowDelegate\n\nextension WidgetWindowsController: NSWindowDelegate {\n nonisolated\n func windowWillMove(_ notification: Notification) {\n guard let window = notification.object as? NSWindow else { return }\n Task { @MainActor in\n guard window === windows.chatPanelWindow else { return }\n await Task.yield()\n store.send(.chatPanel(.detachChatPanel))\n }\n }\n\n nonisolated\n func windowDidMove(_ notification: Notification) {\n guard let window = notification.object as? NSWindow else { return }\n Task { @MainActor in\n guard window === windows.chatPanelWindow else { return }\n await Task.yield()\n await adjustChatPanelWindowLevel()\n }\n }\n\n nonisolated\n func windowWillEnterFullScreen(_ notification: Notification) {\n guard let window = notification.object as? NSWindow else { return }\n Task { @MainActor in\n guard window === windows.chatPanelWindow else { return }\n await Task.yield()\n store.send(.chatPanel(.enterFullScreen))\n }\n }\n\n nonisolated\n func windowWillExitFullScreen(_ notification: Notification) {\n guard let window = notification.object as? NSWindow else { return }\n Task { @MainActor in\n guard window === windows.chatPanelWindow else { return }\n await Task.yield()\n store.send(.chatPanel(.exitFullScreen))\n }\n }\n}\n\n// MARK: - Windows\n\npublic final class WidgetWindows {\n let store: StoreOf\n let chatTabPool: ChatTabPool\n weak var controller: WidgetWindowsController?\n\n // you should make these window `.transient` so they never show up in the mission control.\n\n @MainActor\n lazy var fullscreenDetector = {\n let it = CanBecomeKeyWindow(\n contentRect: .zero,\n styleMask: .borderless,\n backing: .buffered,\n defer: false\n )\n it.isReleasedWhenClosed = false\n it.backgroundColor = .clear\n it.collectionBehavior = [.canJoinAllSpaces, .fullScreenAuxiliary, .transient]\n it.hasShadow = false\n it.setIsVisible(false)\n it.canBecomeKeyChecker = { false }\n return it\n }()\n\n @MainActor\n lazy var widgetWindow = {\n let it = WidgetWindow(\n contentRect: .zero,\n styleMask: .borderless,\n backing: .buffered,\n defer: false\n )\n it.isReleasedWhenClosed = false\n it.backgroundColor = .clear\n it.level = widgetLevel(0)\n it.hasShadow = false\n it.contentView = NSHostingView(\n rootView: WidgetView(\n store: store.scope(\n state: \\._internalCircularWidgetState,\n action: \\.circularWidget\n )\n )\n )\n it.setIsVisible(true)\n it.canBecomeKeyChecker = { false }\n return it\n }()\n\n @MainActor\n lazy var sharedPanelWindow = {\n let it = WidgetWindow(\n contentRect: .init(x: 0, y: 0, width: Style.panelWidth, height: Style.panelHeight),\n styleMask: .borderless,\n backing: .buffered,\n defer: false\n )\n it.isReleasedWhenClosed = false\n it.backgroundColor = .clear\n it.level = widgetLevel(2)\n it.hoveringLevel = widgetLevel(2)\n it.hasShadow = false\n it.contentView = NSHostingView(\n rootView: SharedPanelView(\n store: store.scope(\n state: \\.panelState,\n action: \\.panel\n ).scope(\n state: \\.sharedPanelState,\n action: \\.sharedPanel\n )\n ).modifierFlagsMonitor()\n )\n it.setIsVisible(true)\n it.canBecomeKeyChecker = { [store] in\n store.withState { state in\n !state.panelState.sharedPanelState.content.promptToCodeGroup.promptToCodes.isEmpty\n }\n }\n return it\n }()\n\n @MainActor\n lazy var suggestionPanelWindow = {\n let it = WidgetWindow(\n contentRect: .init(x: 0, y: 0, width: Style.panelWidth, height: Style.panelHeight),\n styleMask: .borderless,\n backing: .buffered,\n defer: false\n )\n it.isReleasedWhenClosed = false\n it.backgroundColor = .clear\n it.level = widgetLevel(2)\n it.hasShadow = false\n it.menu = nil\n it.animationBehavior = .utilityWindow\n it.contentView = NSHostingView(\n rootView: SuggestionPanelView(\n store: store.scope(\n state: \\.panelState,\n action: \\.panel\n ).scope(\n state: \\.suggestionPanelState,\n action: \\.suggestionPanel\n )\n )\n )\n it.canBecomeKeyChecker = { false }\n it.setIsVisible(true)\n return it\n }()\n\n @MainActor\n lazy var chatPanelWindow = {\n let it = ChatPanelWindow(\n store: store.scope(\n state: \\.chatPanelState,\n action: \\.chatPanel\n ),\n chatTabPool: chatTabPool,\n minimizeWindow: { [weak self] in\n self?.store.send(.chatPanel(.hideButtonClicked))\n }\n )\n it.hoveringLevel = widgetLevel(1)\n it.delegate = controller\n return it\n }()\n\n @MainActor\n lazy var toastWindow = {\n let it = WidgetWindow(\n contentRect: .init(x: 0, y: 0, width: Style.panelWidth, height: Style.panelHeight),\n styleMask: [.borderless],\n backing: .buffered,\n defer: false\n )\n it.isReleasedWhenClosed = false\n it.isOpaque = false\n it.backgroundColor = .clear\n it.level = widgetLevel(2)\n it.hasShadow = false\n it.contentView = NSHostingView(\n rootView: ToastPanelView(store: store.scope(\n state: \\.toastPanel,\n action: \\.toastPanel\n ))\n )\n it.setIsVisible(true)\n it.canBecomeKeyChecker = { false }\n return it\n }()\n\n init(\n store: StoreOf,\n chatTabPool: ChatTabPool\n ) {\n self.store = store\n self.chatTabPool = chatTabPool\n }\n\n @MainActor\n func orderFront() {\n widgetWindow.orderFrontRegardless()\n toastWindow.orderFrontRegardless()\n sharedPanelWindow.orderFrontRegardless()\n suggestionPanelWindow.orderFrontRegardless()\n if chatPanelWindow.level.rawValue > NSWindow.Level.normal.rawValue,\n store.withState({ !$0.chatPanelState.isDetached })\n {\n chatPanelWindow.orderFrontRegardless()\n }\n }\n}\n\n// MARK: - Window Subclasses\n\nclass CanBecomeKeyWindow: NSWindow {\n var canBecomeKeyChecker: () -> Bool = { true }\n override var canBecomeKey: Bool { canBecomeKeyChecker() }\n override var canBecomeMain: Bool { canBecomeKeyChecker() }\n}\n\nclass WidgetWindow: CanBecomeKeyWindow {\n enum State: Equatable {\n case normal(fullscreen: Bool)\n case switchingSpace\n }\n\n var hoveringLevel: NSWindow.Level = widgetLevel(0)\n \n override var isFloatingPanel: Bool { true }\n\n var defaultCollectionBehavior: NSWindow.CollectionBehavior {\n [.fullScreenAuxiliary, .transient]\n }\n\n var isFullscreen: Bool {\n styleMask.contains(.fullScreen)\n }\n\n private var state: State? {\n didSet {\n guard state != oldValue else { return }\n switch state {\n case .none:\n collectionBehavior = defaultCollectionBehavior\n case .switchingSpace:\n collectionBehavior = defaultCollectionBehavior.union(.moveToActiveSpace)\n case .normal:\n collectionBehavior = defaultCollectionBehavior\n }\n }\n }\n\n func moveToActiveSpace() {\n let previousState = state\n state = .switchingSpace\n Task { @MainActor in\n try await Task.sleep(nanoseconds: 50_000_000)\n self.state = previousState\n }\n }\n\n func setFloatOnTop(_ isFloatOnTop: Bool) {\n let targetLevel: NSWindow.Level = isFloatOnTop\n ? hoveringLevel\n : .normal\n\n if targetLevel != level {\n orderFrontRegardless()\n level = targetLevel\n }\n }\n}\n\nfunc widgetLevel(_ addition: Int) -> NSWindow.Level {\n let minimumWidgetLevel: Int\n #if DEBUG\n minimumWidgetLevel = NSWindow.Level.floating.rawValue + 1\n #else\n minimumWidgetLevel = NSWindow.Level.floating.rawValue\n #endif\n return .init(minimumWidgetLevel + addition)\n}\n\nextension CGRect {\n func flipped(relativeTo reference: CGRect) -> CGRect {\n let flippedOrigin = CGPoint(\n x: origin.x,\n y: reference.height - origin.y - height\n )\n return CGRect(origin: flippedOrigin, size: size)\n }\n\n func relative(to reference: CGRect) -> CGRect {\n let relativeOrigin = CGPoint(\n x: origin.x - reference.origin.x,\n y: origin.y - reference.origin.y\n )\n return CGRect(origin: relativeOrigin, size: size)\n }\n}\n\n" }, { "path": "Core/Sources/UpdateChecker/UpdateChecker.swift", "content": "import Logger\nimport Preferences\nimport Sparkle\n\npublic final class UpdateChecker {\n let updater: SPUUpdater\n let hostBundleFound: Bool\n let delegate: UpdaterDelegate\n public weak var updateCheckerDelegate: UpdateCheckerDelegate? {\n get { delegate.updateCheckerDelegate }\n set { delegate.updateCheckerDelegate = newValue }\n }\n\n public init(\n hostBundle: Bundle?,\n shouldAutomaticallyCheckForUpdate: Bool\n ) {\n if hostBundle == nil {\n hostBundleFound = false\n Logger.updateChecker.error(\"Host bundle not found\")\n } else {\n hostBundleFound = true\n }\n delegate = .init(\n shouldAutomaticallyCheckForUpdate: shouldAutomaticallyCheckForUpdate\n )\n updater = SPUUpdater(\n hostBundle: hostBundle ?? Bundle.main,\n applicationBundle: Bundle.main,\n userDriver: SPUStandardUserDriver(hostBundle: hostBundle ?? Bundle.main, delegate: nil),\n delegate: delegate\n )\n do {\n try updater.start()\n } catch {\n Logger.updateChecker.error(error.localizedDescription)\n }\n }\n\n public func checkForUpdates() {\n updater.checkForUpdates()\n }\n \n public func resetUpdateCycle() {\n updater.resetUpdateCycleAfterShortDelay()\n }\n\n public var automaticallyChecksForUpdates: Bool {\n get { updater.automaticallyChecksForUpdates }\n set { updater.automaticallyChecksForUpdates = newValue }\n }\n}\n\npublic protocol UpdateCheckerDelegate: AnyObject {\n func prepareForRelaunch(finish: @escaping () -> Void)\n}\n\nclass UpdaterDelegate: NSObject, SPUUpdaterDelegate {\n let shouldAutomaticallyCheckForUpdate: Bool\n weak var updateCheckerDelegate: UpdateCheckerDelegate?\n\n init(shouldAutomaticallyCheckForUpdate: Bool) {\n self.shouldAutomaticallyCheckForUpdate = shouldAutomaticallyCheckForUpdate\n }\n\n func updater(_ updater: SPUUpdater, mayPerform updateCheck: SPUUpdateCheck) throws {\n // Not sure how it works\n// if !shouldAutomaticallyCheckForUpdate, updateCheck == .updatesInBackground {\n// throw CancellationError()\n// }\n }\n\n func updater(\n _ updater: SPUUpdater,\n shouldPostponeRelaunchForUpdate item: SUAppcastItem,\n untilInvokingBlock installHandler: @escaping () -> Void\n ) -> Bool {\n if let updateCheckerDelegate {\n updateCheckerDelegate.prepareForRelaunch(finish: installHandler)\n return true\n }\n return false\n }\n \n func updater(_ updater: SPUUpdater, willScheduleUpdateCheckAfterDelay delay: TimeInterval) {\n Logger.updateChecker.info(\"Will schedule update check after delay: \\(delay)\")\n }\n \n func updaterWillNotScheduleUpdateCheck(_ updater: SPUUpdater) {\n Logger.updateChecker.info(\"Will not schedule update check\")\n }\n\n func allowedChannels(for updater: SPUUpdater) -> Set {\n if UserDefaults.shared.value(for: \\.installBetaBuilds) {\n Set([\"beta\"])\n } else {\n []\n }\n }\n}\n\n" }, { "path": "Core/Sources/UserDefaultsObserver/UserDefaultsObserver.swift", "content": "import Foundation\n\npublic final class UserDefaultsObserver: NSObject {\n public var onChange: (() -> Void)?\n private weak var object: NSObject?\n private let keyPaths: [String]\n\n public init(\n object: NSObject,\n forKeyPaths keyPaths: [String],\n context: UnsafeMutableRawPointer?\n ) {\n self.object = object\n self.keyPaths = keyPaths\n super.init()\n for keyPath in keyPaths {\n object.addObserver(self, forKeyPath: keyPath, options: .new, context: context)\n }\n }\n\n deinit {\n for keyPath in keyPaths {\n object?.removeObserver(self, forKeyPath: keyPath)\n }\n }\n\n public override func observeValue(\n forKeyPath keyPath: String?,\n of object: Any?,\n change: [NSKeyValueChangeKey: Any]?,\n context: UnsafeMutableRawPointer?\n ) {\n onChange?()\n }\n}\n\n" }, { "path": "Core/Sources/XcodeThemeController/HighlightJSThemeTemplate.swift", "content": "import Foundation\n\nfunc buildHighlightJSTheme(_ theme: XcodeTheme) -> String {\n /// The source value is an `r g b a` string, for example: `0.5 0.5 0.2 1`\n \n return \"\"\"\n .hljs {\n display: block;\n overflow-x: auto;\n padding: 0.5em;\n background: \\(theme.backgroundColor.hexString);\n color: \\(theme.plainTextColor.hexString);\n }\n .xml .hljs-meta {\n color: \\(theme.marksColor.hexString);\n }\n .hljs-comment,\n .hljs-quote {\n color: \\(theme.commentColor.hexString);\n }\n .hljs-tag,\n .hljs-keyword,\n .hljs-selector-tag,\n .hljs-literal,\n .hljs-name {\n color: \\(theme.keywordsColor.hexString);\n }\n .hljs-attribute {\n color: \\(theme.attributesColor.hexString);\n }\n .hljs-variable,\n .hljs-template-variable {\n color: \\(theme.otherPropertiesAndGlobalsColor.hexString);\n }\n .hljs-code,\n .hljs-string,\n .hljs-meta-string {\n color: \\(theme.stringsColor.hexString);\n }\n .hljs-regexp {\n color: \\(theme.regexLiteralsColor.hexString);\n }\n .hljs-link {\n color: \\(theme.urlsColor.hexString);\n }\n .hljs-title {\n color: \\(theme.headingColor.hexString);\n }\n .hljs-symbol,\n .hljs-bullet {\n color: \\(theme.attributesColor.hexString);\n }\n .hljs-number {\n color: \\(theme.numbersColor.hexString);\n }\n .hljs-section {\n color: \\(theme.marksColor.hexString);\n }\n .hljs-meta {\n color: \\(theme.keywordsColor.hexString);\n }\n .hljs-type,\n .hljs-built_in,\n .hljs-builtin-name {\n color: \\(theme.otherTypeNamesColor.hexString);\n }\n .hljs-class .hljs-title,\n .hljs-title .class_ {\n color: \\(theme.typeDeclarationsColor.hexString);\n }\n .hljs-function .hljs-title,\n .hljs-title .function_ {\n color: \\(theme.otherDeclarationsColor.hexString);\n }\n .hljs-params {\n color: \\(theme.otherDeclarationsColor.hexString);\n }\n .hljs-attr {\n color: \\(theme.attributesColor.hexString);\n }\n .hljs-subst {\n color: \\(theme.plainTextColor.hexString);\n }\n .hljs-formula {\n background-color: \\(theme.selectionColor.hexString);\n font-style: italic;\n }\n .hljs-addition {\n background-color: #baeeba;\n }\n .hljs-deletion {\n background-color: #ffc8bd;\n }\n .hljs-selector-id,\n .hljs-selector-class {\n color: \\(theme.plainTextColor.hexString);\n }\n .hljs-doctag,\n .hljs-strong {\n font-weight: bold;\n }\n .hljs-emphasis {\n font-style: italic;\n }\n \"\"\"\n}\n\n" }, { "path": "Core/Sources/XcodeThemeController/HighlightrThemeManager.swift", "content": "import Foundation\nimport Highlightr\nimport Preferences\n\npublic class HighlightrThemeManager: ThemeManager {\n let defaultManager: ThemeManager\n\n weak var controller: XcodeThemeController?\n\n public init(defaultManager: ThemeManager, controller: XcodeThemeController) {\n self.defaultManager = defaultManager\n self.controller = controller\n }\n\n public func theme(for name: String) -> Theme? {\n let syncSuggestionTheme = UserDefaults.shared.value(for: \\.syncSuggestionHighlightTheme)\n let syncPromptToCodeTheme = UserDefaults.shared.value(for: \\.syncPromptToCodeHighlightTheme)\n let syncChatTheme = UserDefaults.shared.value(for: \\.syncChatCodeHighlightTheme)\n \n lazy var defaultLight = Theme(themeString: defaultLightTheme)\n lazy var defaultDark = Theme(themeString: defaultDarkTheme)\n\n switch name {\n case \"suggestion-light\":\n guard syncSuggestionTheme, let theme = theme(lightMode: true) else {\n return defaultLight\n }\n return theme\n case \"suggestion-dark\":\n guard syncSuggestionTheme, let theme = theme(lightMode: false) else {\n return defaultDark\n }\n return theme\n case \"promptToCode-light\":\n guard syncPromptToCodeTheme, let theme = theme(lightMode: true) else {\n return defaultLight\n }\n return theme\n case \"promptToCode-dark\":\n guard syncPromptToCodeTheme, let theme = theme(lightMode: false) else {\n return defaultDark\n }\n return theme\n case \"chat-light\":\n guard syncChatTheme, let theme = theme(lightMode: true) else {\n return defaultLight\n }\n return theme\n case \"chat-dark\":\n guard syncChatTheme, let theme = theme(lightMode: false) else {\n return defaultDark\n }\n return theme\n case \"light\":\n return defaultLight\n case \"dark\":\n return defaultDark\n default:\n return defaultLight\n }\n }\n\n func theme(lightMode: Bool) -> Theme? {\n guard let controller else { return nil }\n guard let directories = controller.createSupportDirectoriesIfNeeded() else { return nil }\n\n let themeURL: URL = if lightMode {\n directories.themeDirectory.appendingPathComponent(\"highlightjs-light\")\n } else {\n directories.themeDirectory.appendingPathComponent(\"highlightjs-dark\")\n }\n\n if let themeString = try? String(contentsOf: themeURL) {\n return Theme(themeString: themeString)\n }\n\n controller.syncXcodeThemeIfNeeded()\n\n if let themeString = try? String(contentsOf: themeURL) {\n return Theme(themeString: themeString)\n }\n\n return nil\n }\n}\n\nlet defaultLightTheme = \".hljs{display:block;overflow-x:auto;padding:0.5em;background:#FFFFFFFF;color:#000000D8}.xml .hljs-meta{color:#495460FF}.hljs-comment,.hljs-quote{color:#5D6B79FF}.hljs-tag,.hljs-keyword,.hljs-selector-tag,.hljs-literal,.hljs-name{color:#9A2393FF}.hljs-attribute{color:#805E03FF}.hljs-variable,.hljs-template-variable{color:#6B36A9FF}.hljs-code,.hljs-string,.hljs-meta-string{color:#C31A15FF}.hljs-regexp{color:#000000D8}.hljs-link{color:#0E0EFFFF}.hljs-title{color:#000000FF}.hljs-symbol,.hljs-bullet{color:#805E03FF}.hljs-number{color:#1C00CFFF}.hljs-section{color:#495460FF}.hljs-meta{color:#9A2393FF}.hljs-type,.hljs-built_in,.hljs-builtin-name{color:#3900A0FF}.hljs-class .hljs-title,.hljs-title .class_{color:#0B4F79FF}.hljs-function .hljs-title,.hljs-title .function_{color:#0E67A0FF}.hljs-params{color:#0E67A0FF}.hljs-attr{color:#805E03FF}.hljs-subst{color:#000000D8}.hljs-formula{background-color:#A3CCFEFF;font-style:italic}.hljs-addition{background-color:#baeeba}.hljs-deletion{background-color:#ffc8bd}.hljs-selector-id,.hljs-selector-class{color:#000000D8}.hljs-doctag,.hljs-strong{font-weight:bold}.hljs-emphasis{font-style:italic}\"\n\nlet defaultDarkTheme = \".hljs{display:block;overflow-x:auto;padding:0.5em;background:#1F1F23FF;color:#FFFFFFD8}.xml .hljs-meta{color:#91A1B1FF}.hljs-comment,.hljs-quote{color:#6B7985FF}.hljs-tag,.hljs-keyword,.hljs-selector-tag,.hljs-literal,.hljs-name{color:#FC5FA2FF}.hljs-attribute{color:#BF8554FF}.hljs-variable,.hljs-template-variable{color:#A166E5FF}.hljs-code,.hljs-string,.hljs-meta-string{color:#FC695DFF}.hljs-regexp{color:#FFFFFFD8}.hljs-link{color:#5482FEFF}.hljs-title{color:#FFFFFFFF}.hljs-symbol,.hljs-bullet{color:#BF8554FF}.hljs-number{color:#CFBF69FF}.hljs-section{color:#91A1B1FF}.hljs-meta{color:#FC5FA2FF}.hljs-type,.hljs-built_in,.hljs-builtin-name{color:#D0A7FEFF}.hljs-class .hljs-title,.hljs-title .class_{color:#5CD7FEFF}.hljs-function .hljs-title,.hljs-title .function_{color:#41A1BFFF}.hljs-params{color:#41A1BFFF}.hljs-attr{color:#BF8554FF}.hljs-subst{color:#FFFFFFD8}.hljs-formula{background-color:#505A6FFF;font-style:italic}.hljs-addition{background-color:#baeeba}.hljs-deletion{background-color:#ffc8bd}.hljs-selector-id,.hljs-selector-class{color:#FFFFFFD8}.hljs-doctag,.hljs-strong{font-weight:bold}.hljs-emphasis{font-style:italic}\"\n" }, { "path": "Core/Sources/XcodeThemeController/PreferenceKey+Theme.swift", "content": "import Foundation\nimport Preferences\n\n// MARK: - Theming\n\npublic extension UserDefaultPreferenceKeys {\n var lightXcodeThemeName: PreferenceKey {\n .init(defaultValue: \"\", key: \"LightXcodeThemeName\")\n }\n \n var lightXcodeTheme: PreferenceKey> {\n .init(defaultValue: .init(nil), key: \"LightXcodeTheme\")\n }\n \n var darkXcodeThemeName: PreferenceKey {\n .init(defaultValue: \"\", key: \"DarkXcodeThemeName\")\n }\n \n var darkXcodeTheme: PreferenceKey> {\n .init(defaultValue: .init(nil), key: \"LightXcodeTheme\")\n }\n\n var lastSyncedHighlightJSThemeCreatedAt: PreferenceKey {\n .init(defaultValue: 0, key: \"LastSyncedHighlightJSThemeCreatedAt\")\n }\n}\n\n" }, { "path": "Core/Sources/XcodeThemeController/XcodeThemeController.swift", "content": "import AppKit\nimport Foundation\nimport Highlightr\nimport XcodeInspector\n\npublic class XcodeThemeController {\n var syncTriggerTask: Task?\n\n public init(syncTriggerTask: Task? = nil) {\n self.syncTriggerTask = syncTriggerTask\n }\n\n public func start() {\n let defaultHighlightrThemeManager = Highlightr.themeManager\n Highlightr.themeManager = HighlightrThemeManager(\n defaultManager: defaultHighlightrThemeManager,\n controller: self\n )\n\n syncXcodeThemeIfNeeded()\n\n syncTriggerTask?.cancel()\n syncTriggerTask = Task { [weak self] in\n let notifications = NSWorkspace.shared.notificationCenter\n .notifications(named: NSWorkspace.didActivateApplicationNotification)\n for await notification in notifications {\n try Task.checkCancellation()\n guard let app = notification\n .userInfo?[NSWorkspace.applicationUserInfoKey] as? NSRunningApplication\n else { continue }\n guard app.isCopilotForXcodeExtensionService else { continue }\n guard let self else { return }\n self.syncXcodeThemeIfNeeded()\n }\n }\n }\n}\n\nextension XcodeThemeController {\n func syncXcodeThemeIfNeeded() {\n guard UserDefaults.shared.value(for: \\.syncSuggestionHighlightTheme)\n || UserDefaults.shared.value(for: \\.syncPromptToCodeHighlightTheme)\n || UserDefaults.shared.value(for: \\.syncChatCodeHighlightTheme)\n else { return }\n guard let directories = createSupportDirectoriesIfNeeded() else { return }\n\n defer {\n UserDefaults.shared.set(\n Date().timeIntervalSince1970,\n for: \\.lastSyncedHighlightJSThemeCreatedAt\n )\n }\n\n let xcodeUserDefaults = UserDefaults(suiteName: \"com.apple.dt.Xcode\")!\n\n if let darkThemeName = xcodeUserDefaults\n .value(forKey: \"XCFontAndColorCurrentDarkTheme\") as? String\n {\n syncXcodeThemeIfNeeded(\n xcodeThemeName: darkThemeName,\n light: false,\n in: directories.themeDirectory\n )\n }\n\n if let lightThemeName = xcodeUserDefaults\n .value(forKey: \"XCFontAndColorCurrentTheme\") as? String\n {\n syncXcodeThemeIfNeeded(\n xcodeThemeName: lightThemeName,\n light: true,\n in: directories.themeDirectory\n )\n }\n }\n\n func syncXcodeThemeIfNeeded(\n xcodeThemeName: String,\n light: Bool,\n in directoryURL: URL\n ) {\n let targetName = light ? \"highlightjs-light\" : \"highlightjs-dark\"\n guard let xcodeThemeURL = locateXcodeTheme(named: xcodeThemeName) else { return }\n let targetThemeURL = directoryURL.appendingPathComponent(targetName)\n let lastSyncTimestamp = UserDefaults.shared\n .value(for: \\.lastSyncedHighlightJSThemeCreatedAt)\n\n let shouldSync = {\n if light, UserDefaults.shared.value(for: \\.lightXcodeTheme) == nil { return true }\n if !light, UserDefaults.shared.value(for: \\.darkXcodeTheme) == nil { return true }\n if light, xcodeThemeName != UserDefaults.shared.value(for: \\.lightXcodeThemeName) {\n return true\n }\n if !light, xcodeThemeName != UserDefaults.shared.value(for: \\.darkXcodeThemeName) {\n return true\n }\n if !FileManager.default.fileExists(atPath: targetThemeURL.path) { return true }\n\n let xcodeThemeFileUpdated = {\n guard let xcodeThemeModifiedDate = try? xcodeThemeURL\n .resourceValues(forKeys: [.contentModificationDateKey]).contentModificationDate\n else { return true }\n return xcodeThemeModifiedDate.timeIntervalSince1970 > lastSyncTimestamp\n }()\n\n if xcodeThemeFileUpdated { return true }\n\n return false\n }()\n\n if shouldSync {\n do {\n let theme = try XcodeTheme(fileURL: xcodeThemeURL)\n let highlightrTheme = theme.asHighlightJSTheme()\n try highlightrTheme.write(to: targetThemeURL, atomically: true, encoding: .utf8)\n\n Task { @MainActor in\n if light {\n UserDefaults.shared.set(xcodeThemeName, for: \\.lightXcodeThemeName)\n UserDefaults.shared.set(.init(theme), for: \\.lightXcodeTheme)\n UserDefaults.shared.set(\n .init(theme.plainTextColor.storable),\n for: \\.codeForegroundColorLight\n )\n UserDefaults.shared.set(\n .init(theme.backgroundColor.storable),\n for: \\.codeBackgroundColorLight\n )\n } else {\n UserDefaults.shared.set(xcodeThemeName, for: \\.darkXcodeThemeName)\n UserDefaults.shared.set(.init(theme), for: \\.darkXcodeTheme)\n UserDefaults.shared.set(\n .init(theme.plainTextColor.storable),\n for: \\.codeForegroundColorDark\n )\n UserDefaults.shared.set(\n .init(theme.backgroundColor.storable),\n for: \\.codeBackgroundColorDark\n )\n }\n }\n } catch {\n print(error.localizedDescription)\n }\n }\n }\n\n func locateXcodeTheme(named name: String) -> URL? {\n if let customThemeURL = FileManager.default.urls(\n for: .libraryDirectory,\n in: .userDomainMask\n ).first?.appendingPathComponent(\"Developer/Xcode/UserData/FontAndColorThemes\")\n .appendingPathComponent(name),\n FileManager.default.fileExists(atPath: customThemeURL.path)\n {\n return customThemeURL\n }\n\n let xcodeURL: URL? = {\n if let running = NSWorkspace.shared\n .urlForApplication(withBundleIdentifier: \"com.apple.dt.Xcode\")\n {\n return running\n }\n // Use the main Xcode.app\n let proposedXcodeURL = URL(fileURLWithPath: \"/Applications/Xcode.app\")\n if FileManager.default.fileExists(atPath: proposedXcodeURL.path) {\n return proposedXcodeURL\n }\n // Look for an Xcode.app\n if let applicationsURL = FileManager.default.urls(\n for: .applicationDirectory,\n in: .localDomainMask\n ).first {\n struct InfoPlist: Codable {\n var CFBundleIdentifier: String\n }\n\n let appBundleIdentifier = \"com.apple.dt.Xcode\"\n let appDirectories = try? FileManager.default.contentsOfDirectory(\n at: applicationsURL,\n includingPropertiesForKeys: [],\n options: .skipsHiddenFiles\n )\n for appDirectoryURL in appDirectories ?? [] {\n let infoPlistURL = appDirectoryURL.appendingPathComponent(\"Contents/Info.plist\")\n if let data = try? Data(contentsOf: infoPlistURL),\n let infoPlist = try? PropertyListDecoder().decode(\n InfoPlist.self,\n from: data\n ),\n infoPlist.CFBundleIdentifier == appBundleIdentifier\n {\n return appDirectoryURL\n }\n }\n }\n return nil\n }()\n\n if let url = xcodeURL?\n .appendingPathComponent(\"Contents/SharedFrameworks/DVTUserInterfaceKit.framework\")\n .appendingPathComponent(\"Versions/A/Resources/FontAndColorThemes\")\n .appendingPathComponent(name),\n FileManager.default.fileExists(atPath: url.path)\n {\n return url\n }\n\n return nil\n }\n\n func createSupportDirectoriesIfNeeded() -> (supportDirectory: URL, themeDirectory: URL)? {\n guard let supportURL = FileManager.default.urls(\n for: .applicationSupportDirectory,\n in: .userDomainMask\n ).first?.appendingPathComponent(\n Bundle.main\n .object(forInfoDictionaryKey: \"APPLICATION_SUPPORT_FOLDER\") as! String\n ) else {\n return nil\n }\n\n let themeURL = supportURL.appendingPathComponent(\"Themes\")\n\n do {\n if !FileManager.default.fileExists(atPath: supportURL.path) {\n try FileManager.default.createDirectory(\n at: supportURL,\n withIntermediateDirectories: true,\n attributes: nil\n )\n }\n\n if !FileManager.default.fileExists(atPath: themeURL.path) {\n try FileManager.default.createDirectory(\n at: themeURL,\n withIntermediateDirectories: true,\n attributes: nil\n )\n }\n } catch {\n return nil\n }\n\n return (supportURL, themeURL)\n }\n}\n\n" }, { "path": "Core/Sources/XcodeThemeController/XcodeThemeParser.swift", "content": "import Foundation\nimport Preferences\n\npublic struct XcodeTheme: Codable {\n public struct ThemeColor: Codable {\n public var red: Double\n public var green: Double\n public var blue: Double\n public var alpha: Double\n\n public var hexString: String {\n let red = Int(self.red * 255)\n let green = Int(self.green * 255)\n let blue = Int(self.blue * 255)\n let alpha = Int(self.alpha * 255)\n return String(format: \"#%02X%02X%02X%02X\", red, green, blue, alpha)\n }\n \n var storable: StorableColor {\n .init(red: red, green: green, blue: blue, alpha: alpha)\n }\n }\n\n public var plainTextColor: ThemeColor\n public var commentColor: ThemeColor\n public var documentationMarkupColor: ThemeColor\n public var documentationMarkupKeywordColor: ThemeColor\n public var marksColor: ThemeColor\n public var stringsColor: ThemeColor\n public var charactersColor: ThemeColor\n public var numbersColor: ThemeColor\n public var regexLiteralsColor: ThemeColor\n public var regexLiteralNumbersColor: ThemeColor\n public var regexLiteralCaptureNamesColor: ThemeColor\n public var regexLiteralCharacterClassNamesColor: ThemeColor\n public var regexLiteralOperatorsColor: ThemeColor\n public var keywordsColor: ThemeColor\n public var preprocessorStatementsColor: ThemeColor\n public var urlsColor: ThemeColor\n public var attributesColor: ThemeColor\n public var typeDeclarationsColor: ThemeColor\n public var otherDeclarationsColor: ThemeColor\n public var projectClassNamesColor: ThemeColor\n public var projectFunctionAndMethodNamesColor: ThemeColor\n public var projectConstantsColor: ThemeColor\n public var projectTypeNamesColor: ThemeColor\n public var projectPropertiesAndGlobalsColor: ThemeColor\n public var projectPreprocessorMacrosColor: ThemeColor\n public var otherClassNamesColor: ThemeColor\n public var otherFunctionAndMethodNamesColor: ThemeColor\n public var otherConstantsColor: ThemeColor\n public var otherTypeNamesColor: ThemeColor\n public var otherPropertiesAndGlobalsColor: ThemeColor\n public var otherPreprocessorMacrosColor: ThemeColor\n public var headingColor: ThemeColor\n public var backgroundColor: ThemeColor\n public var selectionColor: ThemeColor\n public var cursorColor: ThemeColor\n public var currentLineColor: ThemeColor\n public var invisibleCharactersColor: ThemeColor\n public var debuggerConsolePromptColor: ThemeColor\n public var debuggerConsoleOutputColor: ThemeColor\n public var debuggerConsoleInputColor: ThemeColor\n public var executableConsoleOutputColor: ThemeColor\n public var executableConsoleInputColor: ThemeColor\n\n public func asHighlightJSTheme() -> String {\n buildHighlightJSTheme(self)\n .replacingOccurrences(of: \"\\n\", with: \"\")\n .replacingOccurrences(of: \": \", with: \":\")\n .replacingOccurrences(of: \"} \", with: \"}\")\n .replacingOccurrences(of: \" {\", with: \"{\")\n .replacingOccurrences(of: \";}\", with: \"}\")\n .replacingOccurrences(of: \" \", with: \"\")\n }\n}\n\npublic extension XcodeTheme {\n /// Color scheme locations:\n /// ~/Library/Developer/Xcode/UserData/FontAndColorThemes/\n /// Xcode.app/Contents/SharedFrameworks/DVTUserInterfaceKit.framework/Versions/A/Resources/FontAndColorThemes\n init(fileURL: URL) throws {\n let parser = XcodeThemeParser()\n self = try parser.parse(fileURL: fileURL)\n }\n}\n\nstruct XcodeThemeParser {\n enum Error: Swift.Error {\n case fileNotFound\n case invalidData\n }\n\n func parse(fileURL: URL) throws -> XcodeTheme {\n guard let data = try? Data(contentsOf: fileURL) else {\n throw Error.fileNotFound\n }\n\n if fileURL.pathExtension == \"xccolortheme\" {\n return try parseXCColorTheme(data)\n } else {\n throw Error.invalidData\n }\n }\n\n func parseXCColorTheme(_ data: Data) throws -> XcodeTheme {\n let plist = try? PropertyListSerialization.propertyList(\n from: data,\n options: .mutableContainers,\n format: nil\n ) as? [String: Any]\n\n guard let theme = plist else { throw Error.invalidData }\n\n /// The source value is an `r g b a` string, for example: `0.5 0.5 0.2 1`\n func convertColor(source: String) -> XcodeTheme.ThemeColor {\n let components = source.split(separator: \" \")\n let red = (components[0] as NSString).doubleValue\n let green = (components[1] as NSString).doubleValue\n let blue = (components[2] as NSString).doubleValue\n let alpha = (components[3] as NSString).doubleValue\n return .init(red: red, green: green, blue: blue, alpha: alpha)\n }\n\n func getThemeValue(\n at path: [String],\n defaultValue: XcodeTheme.ThemeColor = .init(red: 0, green: 0, blue: 0, alpha: 1)\n ) -> XcodeTheme.ThemeColor {\n guard !path.isEmpty else { return defaultValue }\n let keys = path.dropLast(1)\n var currentDict = theme\n for key in keys {\n guard let value = currentDict[key] as? [String: Any] else {\n return defaultValue\n }\n currentDict = value\n }\n if let value = currentDict[path.last!] as? String {\n return convertColor(source: value)\n }\n return defaultValue\n }\n\n let black = XcodeTheme.ThemeColor(red: 0, green: 0, blue: 0, alpha: 1)\n let white = XcodeTheme.ThemeColor(red: 1, green: 1, blue: 1, alpha: 1)\n\n let xcodeTheme = XcodeTheme(\n plainTextColor: getThemeValue(\n at: [\"DVTSourceTextSyntaxColors\", \"xcode.syntax.plain\"],\n defaultValue: black\n ),\n commentColor: getThemeValue(\n at: [\"DVTSourceTextSyntaxColors\", \"xcode.syntax.comment\"],\n defaultValue: black\n ),\n documentationMarkupColor: getThemeValue(\n at: [\"DVTSourceTextSyntaxColors\", \"xcode.syntax.comment.doc\"],\n defaultValue: black\n ),\n documentationMarkupKeywordColor: getThemeValue(\n at: [\"DVTSourceTextSyntaxColors\", \"xcode.syntax.comment.doc.keyword\"],\n defaultValue: black\n ),\n marksColor: getThemeValue(\n at: [\"DVTSourceTextSyntaxColors\", \"xcode.syntax.mark\"],\n defaultValue: black\n ),\n stringsColor: getThemeValue(\n at: [\"DVTSourceTextSyntaxColors\", \"xcode.syntax.string\"],\n defaultValue: black\n ),\n charactersColor: getThemeValue(\n at: [\"DVTSourceTextSyntaxColors\", \"xcode.syntax.character\"],\n defaultValue: black\n ),\n numbersColor: getThemeValue(\n at: [\"DVTSourceTextSyntaxColors\", \"xcode.syntax.number\"],\n defaultValue: black\n ),\n regexLiteralsColor: getThemeValue(\n at: [\"DVTSourceTextSyntaxColors\", \"xcode.syntax.plain\"],\n defaultValue: black\n ),\n regexLiteralNumbersColor: getThemeValue(\n at: [\"DVTSourceTextSyntaxColors\", \"xcode.syntax.number\"],\n defaultValue: black\n ),\n regexLiteralCaptureNamesColor: getThemeValue(\n at: [\"DVTSourceTextSyntaxColors\", \"xcode.syntax.plain\"],\n defaultValue: black\n ),\n regexLiteralCharacterClassNamesColor: getThemeValue(\n at: [\"DVTSourceTextSyntaxColors\", \"xcode.syntax.plain\"],\n defaultValue: black\n ),\n regexLiteralOperatorsColor: getThemeValue(\n at: [\"DVTSourceTextSyntaxColors\", \"xcode.syntax.plain\"],\n defaultValue: black\n ),\n keywordsColor: getThemeValue(\n at: [\"DVTSourceTextSyntaxColors\", \"xcode.syntax.keyword\"],\n defaultValue: black\n ),\n preprocessorStatementsColor: getThemeValue(\n at: [\"DVTSourceTextSyntaxColors\", \"xcode.syntax.preprocessor\"],\n defaultValue: black\n ),\n urlsColor: getThemeValue(\n at: [\"DVTSourceTextSyntaxColors\", \"xcode.syntax.url\"],\n defaultValue: black\n ),\n attributesColor: getThemeValue(\n at: [\"DVTSourceTextSyntaxColors\", \"xcode.syntax.attribute\"],\n defaultValue: black\n ),\n typeDeclarationsColor: getThemeValue(\n at: [\"DVTSourceTextSyntaxColors\", \"xcode.syntax.declaration.type\"],\n defaultValue: black\n ),\n otherDeclarationsColor: getThemeValue(\n at: [\"DVTSourceTextSyntaxColors\", \"xcode.syntax.declaration.other\"],\n defaultValue: black\n ),\n projectClassNamesColor: getThemeValue(\n at: [\"DVTSourceTextSyntaxColors\", \"xcode.syntax.identifier.class\"],\n defaultValue: black\n ),\n projectFunctionAndMethodNamesColor: getThemeValue(\n at: [\"DVTSourceTextSyntaxColors\", \"xcode.syntax.identifier.function\"],\n defaultValue: black\n ),\n projectConstantsColor: getThemeValue(\n at: [\"DVTSourceTextSyntaxColors\", \"xcode.syntax.identifier.constant\"],\n defaultValue: black\n ),\n projectTypeNamesColor: getThemeValue(\n at: [\"DVTSourceTextSyntaxColors\", \"xcode.syntax.identifier.type\"],\n defaultValue: black\n ),\n projectPropertiesAndGlobalsColor: getThemeValue(\n at: [\"DVTSourceTextSyntaxColors\", \"xcode.syntax.identifier.variable\"],\n defaultValue: black\n ),\n projectPreprocessorMacrosColor: getThemeValue(\n at: [\"DVTSourceTextSyntaxColors\", \"xcode.syntax.identifier.macro\"],\n defaultValue: black\n ),\n otherClassNamesColor: getThemeValue(\n at: [\"DVTSourceTextSyntaxColors\", \"xcode.syntax.identifier.class.system\"],\n defaultValue: black\n ),\n otherFunctionAndMethodNamesColor: getThemeValue(\n at: [\"DVTSourceTextSyntaxColors\", \"xcode.syntax.identifier.function.system\"],\n defaultValue: black\n ),\n otherConstantsColor: getThemeValue(\n at: [\"DVTSourceTextSyntaxColors\", \"xcode.syntax.identifier.constant.system\"],\n defaultValue: black\n ),\n otherTypeNamesColor: getThemeValue(\n at: [\"DVTSourceTextSyntaxColors\", \"xcode.syntax.identifier.type.system\"],\n defaultValue: black\n ),\n otherPropertiesAndGlobalsColor: getThemeValue(\n at: [\"DVTSourceTextSyntaxColors\", \"xcode.syntax.identifier.variable.system\"],\n defaultValue: black\n ),\n otherPreprocessorMacrosColor: getThemeValue(\n at: [\"DVTSourceTextSyntaxColors\", \"xcode.syntax.identifier.macro.system\"],\n defaultValue: black\n ),\n headingColor: getThemeValue(\n at: [\"DVTMarkupTextPrimaryHeadingColor\"],\n defaultValue: black\n ),\n backgroundColor: getThemeValue(\n at: [\"DVTSourceTextBackground\"],\n defaultValue: white\n ),\n selectionColor: getThemeValue(\n at: [\"DVTSourceTextSelectionColor\"],\n defaultValue: black\n ),\n cursorColor: getThemeValue(\n at: [\"DVTSourceTextInsertionPointColor\"],\n defaultValue: black\n ),\n currentLineColor: getThemeValue(\n at: [\"DVTSourceTextCurrentLineHighlightColor\"],\n defaultValue: black\n ),\n invisibleCharactersColor: getThemeValue(\n at: [\"DVTSourceTextInvisiblesColor\"],\n defaultValue: black\n ),\n debuggerConsolePromptColor: getThemeValue(\n at: [\"DVTConsoleDebuggerPromptTextColor\"],\n defaultValue: black\n ),\n debuggerConsoleOutputColor: getThemeValue(\n at: [\"DVTConsoleDebuggerOutputTextColor\"],\n defaultValue: black\n ),\n debuggerConsoleInputColor: getThemeValue(\n at: [\"DVTConsoleDebuggerInputTextColor\"],\n defaultValue: black\n ),\n executableConsoleOutputColor: getThemeValue(\n at: [\"DVTConsoleExectuableOutputTextColor\"],\n defaultValue: black\n ),\n executableConsoleInputColor: getThemeValue(\n at: [\"DVTConsoleExectuableInputTextColor\"],\n defaultValue: black\n )\n )\n\n return xcodeTheme\n }\n}\n\n" }, { "path": "Core/Tests/ChatServiceTests/ParseScopesTests.swift", "content": "import XCTest\n\n@testable import ChatService\n\nfinal class ParseScopesTests: XCTestCase {\n let parse = DynamicContextController.parseScopes\n \n func test_parse_single_scope() async throws {\n var prompt = \"@web hello\"\n let scopes = parse(&prompt)\n XCTAssertEqual(scopes, [.web])\n XCTAssertEqual(prompt, \"hello\")\n }\n \n func test_parse_single_scope_with_prefix() async throws {\n var prompt = \"@w hello\"\n let scopes = parse(&prompt)\n XCTAssertEqual(scopes, [.web])\n XCTAssertEqual(prompt, \"hello\")\n }\n \n func test_parse_multiple_spaces() async throws {\n var prompt = \"@web hello\"\n let scopes = parse(&prompt)\n XCTAssertEqual(scopes, [.web])\n XCTAssertEqual(prompt, \"hello\")\n }\n \n func test_parse_no_prefix_at_mark() async throws {\n var prompt = \" @web hello\"\n let scopes = parse(&prompt)\n XCTAssertEqual(scopes, [])\n XCTAssertEqual(prompt, prompt)\n }\n \n func test_parse_multiple_scopes() async throws {\n var prompt = \"@web+file+c+s+project hello\"\n let scopes = parse(&prompt)\n XCTAssertEqual(scopes, [.web, .code, .sense, .project, .file])\n XCTAssertEqual(prompt, \"hello\")\n }\n}\n\n\n\n\n" }, { "path": "Core/Tests/KeyBindingManagerTests/TabToAcceptSuggestionTests.swift", "content": "import Foundation\nimport XCTest\n\n@testable import Workspace\n@testable import KeyBindingManager\n\nclass TabToAcceptSuggestionTests: XCTestCase {\n func test_should_accept_if_line_invalid() {\n XCTAssertTrue(\n TabToAcceptSuggestion.checkIfAcceptSuggestion(\n lines: \"\"\"\n struct Cat {\n var name: String\n var age: Int\n }\n \"\"\".breakLines(),\n cursorPosition: .init(line: 4, character: 4),\n codeMetadata: .init(),\n presentingSuggestionText: \"Hello\"\n )\n )\n \n XCTAssertTrue(\n TabToAcceptSuggestion.checkIfAcceptSuggestion(\n lines: \"\"\"\n struct Cat {\n var name: String\n var age: Int\n }\n \"\"\".breakLines(),\n cursorPosition: .init(line: -1, character: 4),\n codeMetadata: .init(),\n presentingSuggestionText: \"Hello\"\n )\n )\n }\n \n func test_should_not_accept_if_tab_does_not_invalidate_the_suggestion() {\n XCTAssertFalse(\n TabToAcceptSuggestion.checkIfAcceptSuggestion(\n lines: \"\"\"\n struct Cat {\n \n var age: Int\n }\n \"\"\".breakLines(),\n cursorPosition: .init(line: 1, character: 0),\n codeMetadata: .init(tabSize: 4, indentSize: 4, usesTabsForIndentation: false),\n presentingSuggestionText: \" var name: String\"\n )\n )\n \n XCTAssertFalse(\n TabToAcceptSuggestion.checkIfAcceptSuggestion(\n lines: \"\"\"\n struct ๐Ÿฑ {\n \n var ๐ŸŽ‡: Int\n }\n \"\"\".breakLines(),\n cursorPosition: .init(line: 1, character: 0),\n codeMetadata: .init(tabSize: 4, indentSize: 4, usesTabsForIndentation: false),\n presentingSuggestionText: \" var ๐ŸŽ‡: String\"\n )\n )\n \n XCTAssertFalse(\n TabToAcceptSuggestion.checkIfAcceptSuggestion(\n lines: \"\"\"\n struct Cat {\n \n var age: Int\n }\n \"\"\".breakLines(),\n cursorPosition: .init(line: 1, character: 0),\n codeMetadata: .init(tabSize: 2, indentSize: 2, usesTabsForIndentation: false),\n presentingSuggestionText: \" var name: String\"\n )\n )\n \n XCTAssertFalse(\n TabToAcceptSuggestion.checkIfAcceptSuggestion(\n lines: \"\"\"\n struct Cat {\n \n \\tvar age: Int\n }\n \"\"\".breakLines(),\n cursorPosition: .init(line: 1, character: 0),\n codeMetadata: .init(tabSize: 4, indentSize: 1, usesTabsForIndentation: true),\n presentingSuggestionText: \"\\tvar name: String\"\n )\n )\n }\n \n func test_should_accept_if_tab_invalidates_the_suggestion() {\n XCTAssertTrue(\n TabToAcceptSuggestion.checkIfAcceptSuggestion(\n lines: \"\"\"\n struct Cat {\n \\(\" \")\n var age: Int\n }\n \"\"\".breakLines(),\n cursorPosition: .init(line: 1, character: 1),\n codeMetadata: .init(tabSize: 4, indentSize: 4, usesTabsForIndentation: false),\n presentingSuggestionText: \" var name: String\"\n )\n )\n \n XCTAssertTrue(\n TabToAcceptSuggestion.checkIfAcceptSuggestion(\n lines: \"\"\"\n struct ๐Ÿฑ {\n \\(\" \")\n var ๐ŸŽ‡: Int\n }\n \"\"\".breakLines(),\n cursorPosition: .init(line: 1, character: 1),\n codeMetadata: .init(tabSize: 4, indentSize: 4, usesTabsForIndentation: false),\n presentingSuggestionText: \" var ๐ŸŽ‡: String\"\n )\n )\n \n XCTAssertTrue(\n TabToAcceptSuggestion.checkIfAcceptSuggestion(\n lines: \"\"\"\n struct Cat {\n \\(\" \")\n var age: Int\n }\n \"\"\".breakLines(),\n cursorPosition: .init(line: 1, character: 1),\n codeMetadata: .init(tabSize: 2, indentSize: 2, usesTabsForIndentation: false),\n presentingSuggestionText: \" var name: String\"\n )\n )\n \n XCTAssertTrue(\n TabToAcceptSuggestion.checkIfAcceptSuggestion(\n lines: \"\"\"\n struct Cat {\n \\t\n \\tvar age: Int\n }\n \"\"\".breakLines(),\n cursorPosition: .init(line: 1, character: 1),\n codeMetadata: .init(tabSize: 4, indentSize: 1, usesTabsForIndentation: true),\n presentingSuggestionText: \"\\tvar name: String\"\n )\n )\n }\n}\n" }, { "path": "Core/Tests/PromptToCodeServiceTests/ExtractCodeFromChatGPTTests.swift", "content": "import XCTest\n@testable import PromptToCodeService\n\nfinal class ExtractCodeFromChatGPTTests: XCTestCase {\n func test_extract_from_no_code_block() {\n let api = SimpleModificationAgent()\n let result = api.extractCodeAndDescription(from: \"\"\"\n hello world!\n \"\"\")\n \n XCTAssertEqual(result.code, \"\")\n XCTAssertEqual(result.description, \"\")\n }\n \n func test_extract_from_incomplete_code_block() {\n let api = SimpleModificationAgent()\n let result = api.extractCodeAndDescription(from: \"\"\"\n ```swift\n func foo() {}\n \"\"\")\n \n XCTAssertEqual(result.code, \"func foo() {}\")\n XCTAssertEqual(result.description, \"\")\n }\n \n func test_extract_from_complete_code_block() {\n let api = SimpleModificationAgent()\n let result = api.extractCodeAndDescription(from: \"\"\"\n ```swift\n func foo() {}\n \n func bar() {}\n ```\n \n Description\n \"\"\")\n \n XCTAssertEqual(result.code, \"func foo() {}\\n\\nfunc bar() {}\")\n XCTAssertEqual(result.description, \"Description\")\n }\n \n func test_extract_from_incomplete_code_block_without_language() {\n let api = SimpleModificationAgent()\n let result = api.extractCodeAndDescription(from: \"\"\"\n ```\n func foo() {}\n \"\"\")\n \n XCTAssertEqual(result.code, \"func foo() {}\")\n XCTAssertEqual(result.description, \"\")\n }\n \n func test_extract_from_code_block_without_language() {\n let api = SimpleModificationAgent()\n let result = api.extractCodeAndDescription(from: \"\"\"\n ```\n func foo() {}\n \n func bar() {}\n ```\n \n Description\n \"\"\")\n \n XCTAssertEqual(result.code, \"func foo() {}\\n\\nfunc bar() {}\")\n XCTAssertEqual(result.description, \"Description\")\n }\n \n}\n" }, { "path": "Core/Tests/ServiceTests/Environment.swift", "content": "import AppKit\nimport Client\nimport Foundation\nimport GitHubCopilotService\nimport SuggestionBasic\nimport Workspace\nimport XCTest\nimport XPCShared\n\n@testable import Service\n\nfunc completion(text: String, range: CursorRange, uuid: String = \"\") -> CodeSuggestion {\n .init(id: uuid, text: text, position: range.start, range: range)\n}\n\nclass MockSuggestionService: GitHubCopilotSuggestionServiceType {\n func cancelOngoingTask(workDoneToken: String) async {\n fatalError()\n }\n \n func notifyChangeTextDocument(fileURL: URL, content: String, version: Int) async throws {\n fatalError()\n }\n\n func terminate() async {\n fatalError()\n }\n\n func cancelRequest() async {\n fatalError()\n }\n\n func notifyOpenTextDocument(fileURL: URL, content: String) async throws {\n fatalError()\n }\n\n func notifyCloseTextDocument(fileURL: URL) async throws {\n fatalError()\n }\n\n func notifySaveTextDocument(fileURL: URL) async throws {\n fatalError()\n }\n\n var completions = [CodeSuggestion]()\n var accepted: String?\n var rejected: [String] = []\n\n init(completions: [CodeSuggestion]) {\n self.completions = completions\n }\n\n func getCompletions(\n fileURL: URL,\n content: String,\n originalContent: String,\n cursorPosition: SuggestionBasic.CursorPosition,\n tabSize: Int,\n indentSize: Int,\n usesTabsForIndentation: Bool\n ) async throws -> [SuggestionBasic.CodeSuggestion] {\n completions\n }\n\n func notifyAccepted(_ completion: CodeSuggestion) async {\n accepted = completion.id\n }\n\n func notifyRejected(_ completions: [CodeSuggestion]) async {\n rejected = completions.map(\\.id)\n }\n}\n\n" }, { "path": "Core/Tests/ServiceTests/ExtractSelectedCodeTests.swift", "content": "import SuggestionBasic\nimport XCTest\n@testable import Service\n@testable import XPCShared\n\nclass ExtractSelectedCodeTests: XCTestCase {\n func test_empty_selection() {\n let selection = EditorContent.Selection(\n start: CursorPosition(line: 0, character: 0),\n end: CursorPosition(line: 0, character: 0)\n )\n let lines = [\"let foo = 1\\n\", \"let bar = 2\\n\"]\n let result = selectedCode(in: selection, for: lines)\n XCTAssertEqual(result, \"\")\n }\n\n func test_single_line_selection() {\n let selection = EditorContent.Selection(\n start: CursorPosition(line: 0, character: 4),\n end: CursorPosition(line: 0, character: 10)\n )\n let lines = [\"let foo = 1\\n\", \"let bar = 2\\n\"]\n let result = selectedCode(in: selection, for: lines)\n XCTAssertEqual(result, \"foo = \")\n }\n\n func test_single_line_selection_at_line_end() {\n let selection = EditorContent.Selection(\n start: CursorPosition(line: 0, character: 8),\n end: CursorPosition(line: 0, character: 11)\n )\n let lines = [\"let foo = 1\\n\", \"let bar = 2\\n\"]\n let result = selectedCode(in: selection, for: lines)\n XCTAssertEqual(result, \"= 1\")\n }\n\n func test_multi_line_selection() {\n let selection = EditorContent.Selection(\n start: CursorPosition(line: 0, character: 4),\n end: CursorPosition(line: 1, character: 11)\n )\n let lines = [\"let foo = 1\\n\", \"let bar = 2\\n\", \"let baz = 3\\n\"]\n let result = selectedCode(in: selection, for: lines)\n XCTAssertEqual(result, \"foo = 1\\nlet bar = 2\")\n }\n\n func test_invalid_selection() {\n let selection = EditorContent.Selection(\n start: CursorPosition(line: 1, character: 4),\n end: CursorPosition(line: 0, character: 10)\n )\n let lines = [\"let foo = 1\", \"let bar = 2\"]\n let result = selectedCode(in: selection, for: lines)\n XCTAssertEqual(result, \"\")\n }\n}\n" }, { "path": "Core/Tests/ServiceTests/FilespaceSuggestionInvalidationTests.swift", "content": "import Foundation\nimport SuggestionBasic\nimport XCTest\n\n@testable import Service\n@testable import Workspace\n\nclass FilespaceSuggestionInvalidationTests: XCTestCase {\n @WorkspaceActor\n func prepare(\n lines: [String],\n suggestionText: String,\n cursorPosition: CursorPosition,\n range: CursorRange\n ) async throws -> Filespace {\n let pool = WorkspacePool()\n let (_, filespace) = try await pool.fetchOrCreateWorkspaceAndFilespace(\n fileURL: URL(fileURLWithPath: \"file/path/to.swift\"),\n checkIfFileExists: false\n )\n filespace.suggestions = [\n .init(\n id: \"\",\n text: suggestionText,\n position: cursorPosition,\n range: range\n ),\n ]\n filespace.suggestionSourceSnapshot = .init(lines: lines, cursorPosition: cursorPosition)\n return filespace\n }\n\n func test_text_typing_suggestion_should_be_valid() async throws {\n let lines = [\"\\n\", \"hell\\n\", \"\\n\"]\n let filespace = try await prepare(\n lines: lines,\n suggestionText: \"hello man\",\n cursorPosition: .init(line: 1, character: 0),\n range: .init(startPair: (1, 0), endPair: (1, 0))\n )\n let isValid = await filespace.validateSuggestions(\n lines: lines,\n cursorPosition: .init(line: 1, character: 4),\n alwaysTrueIfCursorNotMoved: false // TODO: What\n )\n XCTAssertTrue(isValid)\n let suggestion = filespace.presentingSuggestion\n XCTAssertNotNil(suggestion)\n }\n\n func test_text_typing_suggestion_in_the_middle_should_be_valid() async throws {\n let lines = [\"\\n\", \"hell man\\n\", \"\\n\"]\n let filespace = try await prepare(\n lines: lines,\n suggestionText: \"hello man\",\n cursorPosition: .init(line: 1, character: 0),\n range: .init(startPair: (1, 0), endPair: (1, 0))\n )\n let isValid = await filespace.validateSuggestions(\n lines: lines,\n cursorPosition: .init(line: 1, character: 4),\n alwaysTrueIfCursorNotMoved: false\n )\n XCTAssertTrue(isValid)\n let suggestion = filespace.presentingSuggestion\n XCTAssertNotNil(suggestion)\n }\n\n func test_text_typing_suggestion_with_emoji_in_the_middle_should_be_valid() async throws {\n let lines = [\"\\n\", \"hell๐ŸŽ†๐ŸŽ† man\\n\", \"\\n\"]\n let filespace = try await prepare(\n lines: lines,\n suggestionText: \"hello๐ŸŽ†๐ŸŽ† man\",\n cursorPosition: .init(line: 1, character: 0),\n range: .init(startPair: (1, 0), endPair: (1, 0))\n )\n let isValid = await filespace.validateSuggestions(\n lines: lines,\n cursorPosition: .init(line: 1, character: 4),\n alwaysTrueIfCursorNotMoved: false\n )\n XCTAssertTrue(isValid)\n let suggestion = filespace.presentingSuggestion\n XCTAssertNotNil(suggestion)\n }\n\n func test_text_typing_suggestion_typed_emoji_in_the_middle_should_be_valid() async throws {\n let lines = [\"\\n\", \"h๐ŸŽ†๐ŸŽ†o ma\\n\", \"\\n\"]\n let filespace = try await prepare(\n lines: lines,\n suggestionText: \"h๐ŸŽ†๐ŸŽ†o man\",\n cursorPosition: .init(line: 1, character: 0),\n range: .init(startPair: (1, 0), endPair: (1, 0))\n )\n let isValid = await filespace.validateSuggestions(\n lines: lines,\n cursorPosition: .init(line: 1, character: 2),\n alwaysTrueIfCursorNotMoved: false\n )\n XCTAssertTrue(isValid)\n let suggestion = filespace.presentingSuggestion\n XCTAssertNotNil(suggestion)\n }\n\n func test_text_typing_suggestion_cutting_emoji_in_the_middle_should_be_valid() async throws {\n // undefined behavior, must not crash\n\n let lines = [\"\\n\", \"h๐ŸŽ†๐ŸŽ†o ma\\n\", \"\\n\"]\n\n let filespace = try await prepare(\n lines: lines,\n suggestionText: \"h๐ŸŽ†๐ŸŽ†o man\",\n cursorPosition: .init(line: 1, character: 0),\n range: .init(startPair: (1, 0), endPair: (1, 0))\n )\n let isValid = await filespace.validateSuggestions(\n lines: lines,\n cursorPosition: .init(line: 1, character: 3),\n alwaysTrueIfCursorNotMoved: false\n )\n XCTAssertTrue(isValid)\n let suggestion = filespace.presentingSuggestion\n XCTAssertNotNil(suggestion)\n }\n\n func test_typing_not_according_to_suggestion_should_invalidate() async throws {\n let lines = [\"\\n\", \"hello ma\\n\", \"\\n\"]\n let filespace = try await prepare(\n lines: lines,\n suggestionText: \"hello man\",\n cursorPosition: .init(line: 1, character: 8),\n range: .init(startPair: (1, 0), endPair: (1, 8))\n )\n let wasValid = await filespace.validateSuggestions(\n lines: lines,\n cursorPosition: .init(line: 1, character: 8),\n alwaysTrueIfCursorNotMoved: false\n )\n let isValid = await filespace.validateSuggestions(\n lines: [\"\\n\", \"hello mat\\n\", \"\\n\"],\n cursorPosition: .init(line: 1, character: 9),\n alwaysTrueIfCursorNotMoved: false\n )\n XCTAssertTrue(wasValid)\n XCTAssertFalse(isValid)\n let suggestion = filespace.presentingSuggestion\n XCTAssertNil(suggestion)\n }\n\n func test_text_cursor_moved_to_another_line_should_invalidate() async throws {\n let lines = [\"\\n\", \"hell\\n\", \"\\n\"]\n let filespace = try await prepare(\n lines: lines,\n suggestionText: \"hello man\",\n cursorPosition: .init(line: 1, character: 0),\n range: .init(startPair: (1, 0), endPair: (1, 0))\n )\n let isValid = await filespace.validateSuggestions(\n lines: lines,\n cursorPosition: .init(line: 2, character: 0),\n alwaysTrueIfCursorNotMoved: false\n )\n XCTAssertFalse(isValid)\n let suggestion = filespace.presentingSuggestion\n XCTAssertNil(suggestion)\n }\n\n func test_text_cursor_is_invalid_should_invalidate() async throws {\n let lines = [\"\\n\", \"hell\\n\", \"\\n\"]\n let filespace = try await prepare(\n lines: lines,\n suggestionText: \"hello man\",\n cursorPosition: .init(line: 100, character: 0),\n range: .init(startPair: (1, 0), endPair: (1, 0))\n )\n let isValid = await filespace.validateSuggestions(\n lines: lines,\n cursorPosition: .init(line: 100, character: 4),\n alwaysTrueIfCursorNotMoved: false\n )\n XCTAssertFalse(isValid)\n let suggestion = filespace.presentingSuggestion\n XCTAssertNil(suggestion)\n }\n\n func test_line_content_does_not_match_input_should_invalidate() async throws {\n let filespace = try await prepare(\n lines: [\"\\n\", \"hello\\n\", \"\\n\"],\n suggestionText: \"hello man\",\n cursorPosition: .init(line: 1, character: 5),\n range: .init(startPair: (1, 0), endPair: (1, 5))\n )\n let isValid = await filespace.validateSuggestions(\n lines: [\"\\n\", \"helo\\n\", \"\\n\"],\n cursorPosition: .init(line: 1, character: 4),\n alwaysTrueIfCursorNotMoved: false\n )\n XCTAssertFalse(isValid)\n let suggestion = filespace.presentingSuggestion\n XCTAssertNil(suggestion)\n }\n\n func test_line_content_does_not_match_input_should_invalidate_index_out_of_scope() async throws {\n let filespace = try await prepare(\n lines: [\"\\n\", \"hello\\n\", \"\\n\"],\n suggestionText: \"hello man\",\n cursorPosition: .init(line: 1, character: 5),\n range: .init(startPair: (1, 0), endPair: (1, 5))\n )\n let isValid = await filespace.validateSuggestions(\n lines: [\"\\n\", \"helo\\n\", \"\\n\"],\n cursorPosition: .init(line: 1, character: 100),\n alwaysTrueIfCursorNotMoved: false\n )\n XCTAssertFalse(isValid)\n let suggestion = filespace.presentingSuggestion\n XCTAssertNil(suggestion)\n }\n\n func test_finish_typing_the_whole_single_line_suggestion_should_invalidate() async throws {\n let lines = [\"\\n\", \"hello ma\\n\", \"\\n\"]\n let filespace = try await prepare(\n lines: lines,\n suggestionText: \"hello man\",\n cursorPosition: .init(line: 1, character: 8),\n range: .init(startPair: (1, 0), endPair: (1, 8))\n )\n let wasValid = await filespace.validateSuggestions(\n lines: lines,\n cursorPosition: .init(line: 1, character: 8),\n alwaysTrueIfCursorNotMoved: false\n )\n let isValid = await filespace.validateSuggestions(\n lines: [\"\\n\", \"hello man\\n\", \"\\n\"],\n cursorPosition: .init(line: 1, character: 9),\n alwaysTrueIfCursorNotMoved: false\n )\n XCTAssertTrue(wasValid)\n XCTAssertFalse(isValid)\n let suggestion = filespace.presentingSuggestion\n XCTAssertNil(suggestion)\n }\n\n func test_finish_typing_the_whole_single_line_suggestion_with_emoji_should_invalidate(\n ) async throws {\n let lines = [\"\\n\", \"hello m๐ŸŽ†๐ŸŽ†a\\n\", \"\\n\"]\n let filespace = try await prepare(\n lines: lines,\n suggestionText: \"hello m๐ŸŽ†๐ŸŽ†an\",\n cursorPosition: .init(line: 1, character: 0),\n range: .init(startPair: (1, 0), endPair: (1, 0))\n )\n let wasValid = await filespace.validateSuggestions(\n lines: lines,\n cursorPosition: .init(line: 1, character: 12),\n alwaysTrueIfCursorNotMoved: false\n )\n let isValid = await filespace.validateSuggestions(\n lines: [\"\\n\", \"hello m๐ŸŽ†๐ŸŽ†an\\n\", \"\\n\"],\n cursorPosition: .init(line: 1, character: 13),\n alwaysTrueIfCursorNotMoved: false\n )\n XCTAssertTrue(wasValid)\n XCTAssertFalse(isValid)\n let suggestion = filespace.presentingSuggestion\n XCTAssertNil(suggestion)\n }\n\n func test_finish_typing_the_whole_single_line_suggestion_suggestion_is_incomplete_should_invalidate(\n ) async throws {\n let lines = [\"\\n\", \"hello ma!!!!\\n\", \"\\n\"]\n let filespace = try await prepare(\n lines: lines,\n suggestionText: \"hello man\",\n cursorPosition: .init(line: 1, character: 0),\n range: .init(startPair: (1, 0), endPair: (1, 0))\n )\n let wasValid = await filespace.validateSuggestions(\n lines: lines,\n cursorPosition: .init(line: 1, character: 8),\n alwaysTrueIfCursorNotMoved: false\n )\n let isValid = await filespace.validateSuggestions(\n lines: [\"\\n\", \"hello man!!!!!\\n\", \"\\n\"],\n cursorPosition: .init(line: 1, character: 9),\n alwaysTrueIfCursorNotMoved: false\n )\n XCTAssertTrue(wasValid)\n XCTAssertFalse(isValid)\n let suggestion = filespace.presentingSuggestion\n XCTAssertNil(suggestion)\n }\n\n func test_finish_typing_the_whole_multiple_line_suggestion_should_be_valid() async throws {\n let lines = [\"\\n\", \"hello man\\n\", \"\\n\"]\n let filespace = try await prepare(\n lines: lines,\n suggestionText: \"hello man\\nhow are you?\",\n cursorPosition: .init(line: 1, character: 0),\n range: .init(startPair: (1, 0), endPair: (1, 0))\n )\n let isValid = await filespace.validateSuggestions(\n lines: lines,\n cursorPosition: .init(line: 1, character: 9),\n alwaysTrueIfCursorNotMoved: false\n )\n XCTAssertTrue(isValid)\n let suggestion = filespace.presentingSuggestion\n XCTAssertNotNil(suggestion)\n }\n\n func test_finish_typing_the_whole_multiple_line_suggestion_with_emoji_should_be_valid(\n ) async throws {\n let lines = [\"\\n\", \"hello m๐ŸŽ†๐ŸŽ†an\\n\", \"\\n\"]\n let filespace = try await prepare(\n lines: lines,\n suggestionText: \"hello m๐ŸŽ†๐ŸŽ†an\\nhow are you?\",\n cursorPosition: .init(line: 1, character: 0),\n range: .init(startPair: (1, 0), endPair: (1, 0))\n )\n let isValid = await filespace.validateSuggestions(\n lines: lines,\n cursorPosition: .init(line: 1, character: 13),\n alwaysTrueIfCursorNotMoved: false\n )\n XCTAssertTrue(isValid)\n let suggestion = filespace.presentingSuggestion\n XCTAssertNotNil(suggestion)\n }\n\n func test_undo_text_to_a_state_before_the_suggestion_was_generated_should_invalidate(\n ) async throws {\n let lines = [\"\\n\", \"hell\\n\", \"\\n\"]\n let filespace = try await prepare(\n lines: lines,\n suggestionText: \"hello man\",\n cursorPosition: .init(line: 1, character: 5), // generating man from hello\n range: .init(startPair: (1, 0), endPair: (1, 5))\n )\n let isValid = await filespace.validateSuggestions(\n lines: lines,\n cursorPosition: .init(line: 1, character: 4),\n alwaysTrueIfCursorNotMoved: false\n )\n XCTAssertFalse(isValid)\n let suggestion = filespace.presentingSuggestion\n XCTAssertNil(suggestion)\n }\n\n func test_rewriting_the_current_line_by_removing_the_suffix_should_be_valid() async throws {\n let lines = [\"hello world !!!\\n\"]\n let filespace = try await prepare(\n lines: lines,\n suggestionText: \"hello world\",\n cursorPosition: .init(line: 0, character: 15),\n range: .init(startPair: (0, 0), endPair: (0, 15))\n )\n let isValid = await filespace.validateSuggestions(\n lines: lines,\n cursorPosition: .init(line: 0, character: 15),\n alwaysTrueIfCursorNotMoved: false\n )\n XCTAssertTrue(isValid)\n let suggestion = filespace.presentingSuggestion\n XCTAssertNotNil(suggestion)\n }\n\n func test_rewriting_the_current_line_should_be_valid() async throws {\n let lines = [\"hello everyone !!!\\n\"]\n let filespace = try await prepare(\n lines: lines,\n suggestionText: \"hello world !!!\",\n cursorPosition: .init(line: 0, character: 18),\n range: .init(startPair: (0, 0), endPair: (0, 18))\n )\n let isValid = await filespace.validateSuggestions(\n lines: lines,\n cursorPosition: .init(line: 0, character: 18),\n alwaysTrueIfCursorNotMoved: false\n )\n XCTAssertTrue(isValid)\n let suggestion = filespace.presentingSuggestion\n XCTAssertNotNil(suggestion)\n }\n}\n\n" }, { "path": "Core/Tests/ServiceUpdateMigrationTests/MigrateTo240Tests.swift", "content": "import Foundation\nimport Keychain\nimport XCTest\n\n@testable import ServiceUpdateMigration\n\nfinal class MigrateTo240Tests: XCTestCase {\n let userDefaults = UserDefaults(suiteName: \"MigrateTo240Tests\")!\n \n override func tearDown() async throws {\n userDefaults.removePersistentDomain(forName: \"MigrateTo240Tests\")\n }\n \n func test_migrateTo240_no_data_to_migrate() async throws {\n let keychain = FakeKeyChain()\n\n try migrateTo240(defaults: userDefaults, keychain: keychain)\n\n XCTAssertTrue(try keychain.getAll().isEmpty, \"No api key to migrate\")\n\n let chatModels = userDefaults.value(for: \\.chatModels)\n let embeddingModels = userDefaults.value(for: \\.embeddingModels)\n\n for chatModel in chatModels {\n switch chatModel.format {\n case .openAI:\n XCTAssertEqual(chatModel.name, \"OpenAI\")\n XCTAssertEqual(chatModel.info, .init(\n apiKeyName: \"OpenAI\",\n baseURL: \"\",\n maxTokens: 16385,\n supportsFunctionCalling: true,\n modelName: \"gpt-3.5-turbo\"\n ))\n case .azureOpenAI:\n XCTAssertEqual(chatModel.name, \"Azure OpenAI\")\n XCTAssertEqual(chatModel.info, .init(\n apiKeyName: \"Azure OpenAI\",\n baseURL: \"\",\n maxTokens: 4000,\n supportsFunctionCalling: true,\n modelName: \"\"\n ))\n default:\n XCTFail()\n }\n }\n\n for embeddingModel in embeddingModels {\n switch embeddingModel.format {\n case .openAI:\n XCTAssertEqual(embeddingModel.name, \"OpenAI\")\n XCTAssertEqual(embeddingModel.info, .init(\n apiKeyName: \"OpenAI\",\n baseURL: \"\",\n maxTokens: 8191,\n modelName: \"text-embedding-ada-002\"\n ))\n case .azureOpenAI:\n XCTAssertEqual(embeddingModel.name, \"Azure OpenAI\")\n XCTAssertEqual(embeddingModel.info, .init(\n apiKeyName: \"Azure OpenAI\",\n baseURL: \"\",\n maxTokens: 8191,\n modelName: \"\"\n ))\n default:\n XCTFail()\n }\n }\n }\n\n func test_migrateTo240_migrate_data_use_openAI() async throws {\n let keychain = FakeKeyChain()\n\n userDefaults.set(\"Key1\", forKey: \"OpenAIAPIKey\")\n userDefaults.set(\"openai.com\", forKey: \"OpenAIBaseURL\")\n userDefaults.set(\"gpt-500\", forKey: \"ChatGPTModel\")\n userDefaults.set(200, forKey: \"ChatGPTMaxToken\")\n userDefaults.set(\"embedding-200\", forKey: \"OpenAIEmbeddingModel\")\n userDefaults.set(\"Key2\", forKey: \"AzureOpenAIAPIKey\")\n userDefaults.set(\"azure.com\", forKey: \"AzureOpenAIBaseURL\")\n userDefaults.set(\"gpt-800\", forKey: \"AzureChatGPTDeployment\")\n userDefaults.set(\"embedding-800\", forKey: \"AzureEmbeddingDeployment\")\n userDefaults.set(\"openAI\", forKey: \"ChatFeatureProvider\")\n userDefaults.set(\"openAI\", forKey: \"EmbeddingFeatureProvider\")\n\n try migrateTo240(defaults: userDefaults, keychain: keychain)\n\n XCTAssertEqual(try keychain.getAll(), [\n \"OpenAI\": \"Key1\",\n \"Azure OpenAI\": \"Key2\",\n ])\n\n let chatModels = userDefaults.value(for: \\.chatModels)\n let embeddingModels = userDefaults.value(for: \\.embeddingModels)\n\n XCTAssertEqual(chatModels.count, 2)\n XCTAssertEqual(embeddingModels.count, 2)\n\n XCTAssertEqual(\n userDefaults.value(for: \\.defaultChatFeatureChatModelId),\n chatModels.first(where: { $0.format == .openAI })?.id\n )\n XCTAssertEqual(\n userDefaults.value(for: \\.defaultChatFeatureEmbeddingModelId),\n embeddingModels.first(where: { $0.format == .openAI })?.id\n )\n\n for chatModel in chatModels {\n switch chatModel.format {\n case .openAI:\n XCTAssertEqual(chatModel.name, \"OpenAI\")\n XCTAssertEqual(chatModel.info, .init(\n apiKeyName: \"OpenAI\",\n baseURL: \"openai.com\",\n maxTokens: 200,\n supportsFunctionCalling: true,\n modelName: \"gpt-500\"\n ))\n case .azureOpenAI:\n XCTAssertEqual(chatModel.name, \"Azure OpenAI\")\n XCTAssertEqual(chatModel.info, .init(\n apiKeyName: \"Azure OpenAI\",\n baseURL: \"azure.com\",\n maxTokens: 200,\n supportsFunctionCalling: true,\n modelName: \"gpt-800\"\n ))\n default:\n XCTFail()\n }\n }\n\n for embeddingModel in embeddingModels {\n switch embeddingModel.format {\n case .openAI:\n XCTAssertEqual(embeddingModel.name, \"OpenAI\")\n XCTAssertEqual(embeddingModel.info, .init(\n apiKeyName: \"OpenAI\",\n baseURL: \"openai.com\",\n maxTokens: 8191,\n modelName: \"embedding-200\"\n ))\n case .azureOpenAI:\n XCTAssertEqual(embeddingModel.name, \"Azure OpenAI\")\n XCTAssertEqual(embeddingModel.info, .init(\n apiKeyName: \"Azure OpenAI\",\n baseURL: \"azure.com\",\n maxTokens: 8191,\n modelName: \"embedding-800\"\n ))\n default:\n XCTFail()\n }\n }\n }\n\n func test_migrateTo240_migrate_data_use_azureOpenAI() async throws {\n let keychain = FakeKeyChain()\n\n userDefaults.set(\"Key1\", forKey: \"OpenAIAPIKey\")\n userDefaults.set(\"openai.com\", forKey: \"OpenAIBaseURL\")\n userDefaults.set(\"gpt-500\", forKey: \"ChatGPTModel\")\n userDefaults.set(200, forKey: \"ChatGPTMaxToken\")\n userDefaults.set(\"embedding-200\", forKey: \"OpenAIEmbeddingModel\")\n userDefaults.set(\"Key2\", forKey: \"AzureOpenAIAPIKey\")\n userDefaults.set(\"azure.com\", forKey: \"AzureOpenAIBaseURL\")\n userDefaults.set(\"gpt-800\", forKey: \"AzureChatGPTDeployment\")\n userDefaults.set(\"embedding-800\", forKey: \"AzureEmbeddingDeployment\")\n userDefaults.set(\"azureOpenAI\", forKey: \"ChatFeatureProvider\")\n userDefaults.set(\"azureOpenAI\", forKey: \"EmbeddingFeatureProvider\")\n\n try migrateTo240(defaults: userDefaults, keychain: keychain)\n\n let chatModels = userDefaults.value(for: \\.chatModels)\n let embeddingModels = userDefaults.value(for: \\.embeddingModels)\n\n XCTAssertEqual(chatModels.count, 2)\n XCTAssertEqual(embeddingModels.count, 2)\n\n XCTAssertEqual(\n userDefaults.value(for: \\.defaultChatFeatureChatModelId),\n chatModels.first(where: { $0.format == .azureOpenAI })?.id\n )\n XCTAssertEqual(\n userDefaults.value(for: \\.defaultChatFeatureEmbeddingModelId),\n embeddingModels.first(where: { $0.format == .azureOpenAI })?.id\n )\n }\n}\n\n" }, { "path": "Core/Tests/SuggestionWidgetTests/File.swift", "content": "import Foundation\n" }, { "path": "DEVELOPMENT.md", "content": "# Development\n\n## Targets \n\n### Copilot for Xcode\n\nCopilot for Xcode is the host app containing both the XPCService and the editor extension. It provides the settings UI.\n\n### EditorExtension\n\nAs its name suggests, the Xcode source editor extension. Its sole purpose is to forward editor content to the XPCService for processing, and update the editor with the returned content. Due to the sandboxing requirements for editor extensions, it has to communicate with a trusted, non-sandboxed XPCService (CommunicationBridge and ExtensionService) to bypass the limitations. The XPCService service name must be included in the `com.apple.security.temporary-exception.mach-lookup.global-name` entitlements.\n\n### ExtensionService\n\nThe `ExtensionService` is a program that operates in the background. All features are implemented in this target. \n\n### CommunicationBridge\n\nIt's responsible for maintaining the communication between the Copilot for Xcode/EditorExtension and ExtensionService.\n\n### Core and Tool\n\nMost of the logics are implemented inside the package `Core` and `Tool`.\n\n- The `Service` contains the implementations of the ExtensionService target.\n- The `HostApp` contains the implementations of the Copilot for Xcode target.\n\n## Building and Archiving the App\n\n1. Update the xcconfig files, bridgeLaunchAgent.plist, and Tool/Configs/Configurations.swift.\n2. Build or archive the Copilot for Xcode target.\n\n## Testing Source Editor Extension and Service\n\nJust run both the `ExtensionService`, `CommunicationBridge` and the `EditorExtension` Target. Read [Testing Your Source Editor Extension](https://developer.apple.com/documentation/xcodekit/testing_your_source_editor_extension) for more details.\n\nIf you are not testing the source editor extension, it's recommended to archive and install a debug version of the Copilot for Xcode and test with the bundled source editor extension.\n\n## SwiftUI Previews\n\nLooks like SwiftUI Previews are not very happy with Objective-C packages when running with app targets. To use previews, please switch schemes to the package product targets.\n\n## Unit Tests\n\nTo run unit tests, just run test from the `Copilot for Xcode` target.\n\nFor new tests, they should be added to the `TestPlan.xctestplan`.\n\n## Chat Plugins\n\nTo create a chat plugin, please use the `TerminalChatPlugin` as an example. You should add your plugin to the target `ChatPlugin` and register it in `ChatService`.\n\n## Code Style\n\nWe use SwiftFormat to format the code.\n\nThe source code mostly follows the [Ray Wenderlich Style Guide](https://github.com/raywenderlich/swift-style-guide) very closely with the following exception:\n\n- Use the Xcode default of 4 spaces for indentation.\n\n## App Versioning\n\nThe app version and all targets' version in controlled by `Version.xcconfig`.\n" }, { "path": "EditorExtension/AcceptPromptToCodeCommand.swift", "content": "import Client\nimport Foundation\nimport SuggestionBasic\nimport XcodeKit\n\nclass AcceptPromptToCodeCommand: NSObject, XCSourceEditorCommand, CommandType {\n var name: String { \"Accept Modification\" }\n\n func perform(\n with invocation: XCSourceEditorCommandInvocation,\n completionHandler: @escaping (Error?) -> Void\n ) {\n Task {\n do {\n try await (Task(timeout: 7) {\n let service = try getService()\n if let content = try await service.getPromptToCodeAcceptedCode(\n editorContent: .init(invocation)\n ) {\n invocation.accept(content)\n }\n completionHandler(nil)\n }.value)\n } catch is CancellationError {\n completionHandler(nil)\n } catch {\n completionHandler(error)\n }\n }\n }\n}\n" }, { "path": "EditorExtension/AcceptSuggestionCommand.swift", "content": "import Client\nimport Foundation\nimport SuggestionBasic\nimport XcodeKit\nimport XPCShared\n\nclass AcceptSuggestionCommand: NSObject, XCSourceEditorCommand, CommandType {\n var name: String { \"Accept Suggestion\" }\n\n func perform(\n with invocation: XCSourceEditorCommandInvocation,\n completionHandler: @escaping (Error?) -> Void\n ) {\n Task {\n do {\n try await (Task(timeout: 7) {\n let service = try getService()\n if let content = try await service.getSuggestionAcceptedCode(\n editorContent: .init(invocation)\n ) {\n invocation.accept(content)\n }\n completionHandler(nil)\n }.value)\n } catch is CancellationError {\n completionHandler(nil)\n } catch {\n completionHandler(error)\n }\n }\n }\n}\n\nclass AcceptSuggestionLineCommand: NSObject, XCSourceEditorCommand, CommandType {\n var name: String { \"Accept Suggestion Line\" }\n\n func perform(\n with invocation: XCSourceEditorCommandInvocation,\n completionHandler: @escaping (Error?) -> Void\n ) {\n Task {\n do {\n try await (Task(timeout: 7) {\n let service = try getService()\n if let content = try await service.send(\n requestBody: ExtensionServiceRequests\n .GetSuggestionLineAcceptedCode(editorContent: .init(invocation))\n ) {\n invocation.accept(content)\n }\n completionHandler(nil)\n }.value)\n } catch is CancellationError {\n completionHandler(nil)\n } catch {\n completionHandler(error)\n }\n }\n }\n}\n" }, { "path": "EditorExtension/CloseIdleTabsCommand.swift", "content": "import Client\nimport Foundation\nimport SuggestionBasic\nimport XcodeKit\n\nclass CloseIdleTabsCommand: NSObject, XCSourceEditorCommand, CommandType {\n var name: String { \"Close Idle Tabs\" }\n\n func perform(\n with invocation: XCSourceEditorCommandInvocation,\n completionHandler: @escaping (Error?) -> Void\n ) {\n completionHandler(nil)\n Task {\n let service = try getService()\n _ = try await service.postNotification(name: \"CloseIdleTabsOfXcodeWindow\")\n }\n }\n}\n\n" }, { "path": "EditorExtension/CustomCommand.swift", "content": "import Client\nimport Foundation\nimport SuggestionBasic\nimport XcodeKit\n\nclass CustomCommand: NSObject, XCSourceEditorCommand, CommandType {\n var name: String = \"\"\n\n func perform(\n with invocation: XCSourceEditorCommandInvocation,\n completionHandler: @escaping (Error?) -> Void\n ) {\n completionHandler(nil)\n Task {\n let service = try getService()\n _ = try await service.customCommand(\n id: customCommandMap[invocation.commandIdentifier] ?? \"\",\n editorContent: .init(invocation)\n )\n }\n }\n}\n\n" }, { "path": "EditorExtension/EditorExtension.entitlements", "content": "\n\n\n\n\tcom.apple.security.app-sandbox\n\t\n\tcom.apple.security.application-groups\n\t\n\t\t$(TeamIdentifierPrefix)group.$(BUNDLE_IDENTIFIER_BASE)\n\t\n\tcom.apple.security.temporary-exception.mach-lookup.global-name\n\t\n\t\t$(BUNDLE_IDENTIFIER_BASE).CommunicationBridge\n\t\t$(BUNDLE_IDENTIFIER_BASE).ExtensionService\n\t\n\n\n" }, { "path": "EditorExtension/GetSuggestionsCommand.swift", "content": "import Client\nimport Foundation\nimport SuggestionBasic\nimport XcodeKit\n\nclass GetSuggestionsCommand: NSObject, XCSourceEditorCommand, CommandType {\n var name: String { \"Get Suggestions\" }\n\n func perform(\n with invocation: XCSourceEditorCommandInvocation,\n completionHandler: @escaping (Error?) -> Void\n ) {\n completionHandler(nil)\n Task {\n let service = try getService()\n _ = try await service.getSuggestedCode(editorContent: .init(invocation))\n }\n }\n}\n\n" }, { "path": "EditorExtension/Helpers.swift", "content": "import Foundation\nimport SuggestionBasic\nimport XcodeKit\nimport XPCShared\n\nextension XCSourceEditorCommandInvocation {\n func mutateCompleteBuffer(modifications: [Modification], restoringSelections restore: Bool) {\n if restore {\n let selectionsRangesToRestore = buffer.selections\n .compactMap { $0 as? XCSourceTextRange }\n buffer.selections.removeAllObjects()\n buffer.lines.apply(modifications)\n for range in selectionsRangesToRestore {\n buffer.selections.add(range)\n }\n } else {\n buffer.lines.apply(modifications)\n }\n }\n\n func accept(_ updatedContent: UpdatedContent) {\n if !updatedContent.newSelections.isEmpty {\n mutateCompleteBuffer(\n modifications: updatedContent.modifications,\n restoringSelections: false\n )\n buffer.selections.removeAllObjects()\n for newSelection in updatedContent.newSelections {\n buffer.selections.add(XCSourceTextRange(\n start: .init(\n line: newSelection.start.line,\n column: newSelection.start.character\n ),\n end: .init(line: newSelection.end.line, column: newSelection.end.character)\n ))\n }\n } else {\n mutateCompleteBuffer(\n modifications: updatedContent.modifications,\n restoringSelections: true\n )\n }\n }\n}\n\nextension EditorContent {\n init(_ invocation: XCSourceEditorCommandInvocation) {\n let buffer = invocation.buffer\n self.init(\n content: buffer.completeBuffer,\n lines: buffer.lines as? [String] ?? [],\n uti: buffer.contentUTI,\n cursorPosition: ((buffer.selections.lastObject as? XCSourceTextRange)?.end).map {\n CursorPosition(line: $0.line, character: $0.column)\n } ?? CursorPosition(line: 0, character: 0),\n cursorOffset: -1,\n selections: buffer.selections.map {\n let sl = ($0 as? XCSourceTextRange)?.start.line ?? 0\n let sc = ($0 as? XCSourceTextRange)?.start.column ?? 0\n let el = ($0 as? XCSourceTextRange)?.end.line ?? 0\n let ec = ($0 as? XCSourceTextRange)?.end.column ?? 0\n\n return Selection(\n start: CursorPosition(line: sl, character: sc),\n end: CursorPosition(line: el, character: ec)\n )\n },\n tabSize: buffer.tabWidth,\n indentSize: buffer.indentationWidth,\n usesTabsForIndentation: buffer.usesTabsForIndentation\n )\n }\n}\n\n/// https://gist.github.com/swhitty/9be89dfe97dbb55c6ef0f916273bbb97\nextension Task where Failure == Error {\n // Start a new Task with a timeout. If the timeout expires before the operation is\n // completed then the task is cancelled and an error is thrown.\n init(\n priority: TaskPriority? = nil,\n timeout: TimeInterval,\n operation: @escaping @Sendable () async throws -> Success\n ) {\n self = Task(priority: priority) {\n try await withThrowingTaskGroup(of: Success.self) { group -> Success in\n group.addTask(operation: operation)\n group.addTask {\n try await _Concurrency.Task.sleep(nanoseconds: UInt64(timeout * 1_000_000_000))\n throw TimeoutError()\n }\n guard let success = try await group.next() else {\n throw _Concurrency.CancellationError()\n }\n group.cancelAll()\n return success\n }\n }\n }\n}\n\nprivate struct TimeoutError: LocalizedError {\n var errorDescription: String? = \"Task timed out before completion\"\n}\n\n" }, { "path": "EditorExtension/Info.plist", "content": "\n\n\n\n\tAPPLICATION_SUPPORT_FOLDER\n\t$(APPLICATION_SUPPORT_FOLDER)\n\tBUNDLE_IDENTIFIER_BASE\n\t$(BUNDLE_IDENTIFIER_BASE)\n\tCFBundleDevelopmentRegion\n\t$(DEVELOPMENT_LANGUAGE)\n\tCFBundleDisplayName\n\t$(EXTENSION_BUNDLE_DISPLAY_NAME)\n\tCFBundleExecutable\n\t$(EXECUTABLE_NAME)\n\tCFBundleIdentifier\n\t$(PRODUCT_BUNDLE_IDENTIFIER)\n\tCFBundleInfoDictionaryVersion\n\t6.0\n\tCFBundleName\n\t$(EXTENSION_BUNDLE_NAME)\n\tCFBundlePackageType\n\t$(PRODUCT_BUNDLE_PACKAGE_TYPE)\n\tCFBundleShortVersionString\n\t$(MARKETING_VERSION)\n\tCFBundleVersion\n\t$(CURRENT_PROJECT_VERSION)\n\tHOST_APP_NAME\n\t$(HOST_APP_NAME)\n\tNSExtension\n\t\n\t\tNSExtensionAttributes\n\t\t\n\t\t\tXCSourceEditorCommandDefinitions\n\t\t\t\n\t\t\tXCSourceEditorExtensionPrincipalClass\n\t\t\t$(PRODUCT_MODULE_NAME).SourceEditorExtension\n\t\t\n\t\tNSExtensionPointIdentifier\n\t\tcom.apple.dt.Xcode.extension.source-editor\n\t\n\tNSHumanReadableCopyright\n\t\n\tTEAM_ID_PREFIX\n\t$(TeamIdentifierPrefix)\n\n\n" }, { "path": "EditorExtension/NextSuggestionCommand.swift", "content": "import Client\nimport Foundation\nimport SuggestionBasic\nimport XcodeKit\n\nclass NextSuggestionCommand: NSObject, XCSourceEditorCommand, CommandType {\n var name: String { \"Next Suggestion\" }\n\n func perform(\n with invocation: XCSourceEditorCommandInvocation,\n completionHandler: @escaping (Error?) -> Void\n ) {\n completionHandler(nil)\n Task {\n let service = try getService()\n _ = try await service.getNextSuggestedCode(editorContent: .init(invocation))\n }\n }\n}\n\n" }, { "path": "EditorExtension/OpenChat.swift", "content": "import Client\nimport SuggestionBasic\nimport Foundation\nimport XcodeKit\n\nclass OpenChatCommand: NSObject, XCSourceEditorCommand, CommandType {\n var name: String { \"Open Chat\" }\n\n func perform(\n with invocation: XCSourceEditorCommandInvocation,\n completionHandler: @escaping (Error?) -> Void\n ) {\n completionHandler(nil)\n Task {\n let service = try getService()\n _ = try await service.openChat(editorContent: .init(invocation))\n }\n }\n}\n" }, { "path": "EditorExtension/PrefetchSuggestionsCommand.swift", "content": "import Client\nimport SuggestionBasic\nimport Foundation\nimport XcodeKit\n\nclass PrefetchSuggestionsCommand: NSObject, XCSourceEditorCommand, CommandType {\n var name: String { \"Prefetch Suggestions\" }\n\n func perform(\n with invocation: XCSourceEditorCommandInvocation,\n completionHandler: @escaping (Error?) -> Void\n ) {\n completionHandler(nil)\n Task {\n let service = try getService()\n await service.prefetchRealtimeSuggestions(editorContent: .init(invocation))\n }\n }\n}\n" }, { "path": "EditorExtension/PreviousSuggestionCommand.swift", "content": "import Client\nimport Foundation\nimport SuggestionBasic\nimport XcodeKit\n\nclass PreviousSuggestionCommand: NSObject, XCSourceEditorCommand, CommandType {\n var name: String { \"Previous Suggestion\" }\n\n func perform(\n with invocation: XCSourceEditorCommandInvocation,\n completionHandler: @escaping (Error?) -> Void\n ) {\n completionHandler(nil)\n Task {\n let service = try getService()\n _ = try await service.getPreviousSuggestedCode(editorContent: .init(invocation))\n }\n }\n}\n\n" }, { "path": "EditorExtension/PromptToCodeCommand.swift", "content": "import Client\nimport SuggestionBasic\nimport Foundation\nimport XcodeKit\n\nclass PromptToCodeCommand: NSObject, XCSourceEditorCommand, CommandType {\n var name: String { \"Write or Edit Code\" }\n\n func perform(\n with invocation: XCSourceEditorCommandInvocation,\n completionHandler: @escaping (Error?) -> Void\n ) {\n completionHandler(nil)\n Task {\n let service = try getService()\n _ = try await service.promptToCode(editorContent: .init(invocation))\n }\n }\n}\n" }, { "path": "EditorExtension/RealtimeSuggestionCommand.swift", "content": "import Client\nimport SuggestionBasic\nimport Foundation\nimport XcodeKit\n\nclass RealtimeSuggestionsCommand: NSObject, XCSourceEditorCommand, CommandType {\n var name: String { \"Prepare for Real-time Suggestions\" }\n\n func perform(\n with invocation: XCSourceEditorCommandInvocation,\n completionHandler: @escaping (Error?) -> Void\n ) {\n completionHandler(nil)\n Task {\n let service = try getService()\n _ = try await service.getRealtimeSuggestedCode(editorContent: .init(invocation))\n }\n }\n}\n" }, { "path": "EditorExtension/RejectSuggestionCommand.swift", "content": "import Client\nimport Foundation\nimport SuggestionBasic\nimport XcodeKit\n\nclass RejectSuggestionCommand: NSObject, XCSourceEditorCommand, CommandType {\n var name: String { \"Reject Suggestion\" }\n\n func perform(\n with invocation: XCSourceEditorCommandInvocation,\n completionHandler: @escaping (Error?) -> Void\n ) {\n completionHandler(nil)\n Task {\n let service = try getService()\n _ = try await service.getSuggestionRejectedCode(editorContent: .init(invocation))\n }\n }\n}\n\n" }, { "path": "EditorExtension/SeparatorCommand.swift", "content": "import Client\nimport SuggestionBasic\nimport Foundation\nimport XcodeKit\n\nclass SeparatorCommand: NSObject, XCSourceEditorCommand, CommandType {\n var name: String = \"\"\n \n func perform(\n with invocation: XCSourceEditorCommandInvocation,\n completionHandler: @escaping (Error?) -> Void\n ) {\n completionHandler(nil)\n }\n \n func named(_ name: String) -> Self {\n self.name = name\n return self\n }\n}\n" }, { "path": "EditorExtension/SourceEditorExtension.swift", "content": "import Client\nimport Foundation\nimport Preferences\nimport XcodeKit\n\n#if canImport(PreferencesPlus)\nimport PreferencesPlus\n#endif\n\nclass SourceEditorExtension: NSObject, XCSourceEditorExtension {\n var builtin: [[XCSourceEditorCommandDefinitionKey: Any]] {\n [\n GetSuggestionsCommand(),\n AcceptSuggestionCommand(),\n AcceptSuggestionLineCommand(),\n RejectSuggestionCommand(),\n NextSuggestionCommand(),\n PreviousSuggestionCommand(),\n PromptToCodeCommand(),\n AcceptPromptToCodeCommand(),\n OpenChatCommand(),\n ToggleRealtimeSuggestionsCommand(),\n ].map(makeCommandDefinition)\n }\n \n var optional: [[XCSourceEditorCommandDefinitionKey: Any]] {\n var all = [[XCSourceEditorCommandDefinitionKey: Any]]()\n \n #if canImport(PreferencesPlus)\n if UserDefaults.shared.value(for: \\.enableCloseIdleTabCommandInXcodeMenu) {\n all.append(CloseIdleTabsCommand().makeCommandDefinition())\n }\n #endif\n \n return all\n }\n \n var internalUse: [[XCSourceEditorCommandDefinitionKey: Any]] {\n [\n SeparatorCommand().named(\"------\"),\n RealtimeSuggestionsCommand(),\n PrefetchSuggestionsCommand(),\n ].map(makeCommandDefinition)\n }\n\n var custom: [[XCSourceEditorCommandDefinitionKey: Any]] {\n let all = customCommands()\n if all.isEmpty {\n return []\n }\n return [SeparatorCommand().named(\"------\")].map(makeCommandDefinition) + all\n }\n\n var commandDefinitions: [[XCSourceEditorCommandDefinitionKey: Any]] {\n return builtin + optional + custom + internalUse\n }\n\n func extensionDidFinishLaunching() {\n #if DEBUG\n // In a debug build, we usually want to use the XPC service run from Xcode.\n #else\n // When the source extension is initialized\n // we can call a random command to wake up the XPC service.\n Task.detached {\n try await Task.sleep(nanoseconds: 1_000_000_000)\n let service = try getService()\n _ = try await service.getXPCServiceVersion()\n }\n #endif\n }\n}\n\nlet identifierPrefix: String = Bundle.main.bundleIdentifier ?? \"\"\n\nvar customCommandMap = [String: String]()\n\nprotocol CommandType: AnyObject {\n var commandClassName: String { get }\n var identifier: String { get }\n var name: String { get }\n}\n\nextension CommandType where Self: NSObject {\n var commandClassName: String { Self.className() }\n var identifier: String { commandClassName }\n}\n\nextension CommandType {\n func makeCommandDefinition() -> [XCSourceEditorCommandDefinitionKey: Any] {\n [.classNameKey: commandClassName,\n .identifierKey: identifierPrefix + identifier,\n .nameKey: name]\n }\n}\n\nfunc makeCommandDefinition(_ commandType: CommandType)\n -> [XCSourceEditorCommandDefinitionKey: Any]\n{\n commandType.makeCommandDefinition()\n}\n\nfunc customCommands() -> [[XCSourceEditorCommandDefinitionKey: Any]] {\n var definitions = [[XCSourceEditorCommandDefinitionKey: String]]()\n for command in UserDefaults.shared.value(for: \\.customCommands) {\n let identifier = identifierPrefix + \"CustomCommand\\(command.id)\"\n definitions.append([\n .classNameKey: CustomCommand.className(),\n .identifierKey: identifier,\n .nameKey: command.name,\n ])\n \n customCommandMap[identifier] = command.id\n }\n \n return definitions\n}\n" }, { "path": "EditorExtension/ToggleRealtimeSuggestionsCommand.swift", "content": "import Client\nimport SuggestionBasic\nimport Foundation\nimport XcodeKit\n\nclass ToggleRealtimeSuggestionsCommand: NSObject, XCSourceEditorCommand, CommandType {\n var name: String { \"Toggle Real-time Suggestions\" }\n\n func perform(\n with invocation: XCSourceEditorCommandInvocation,\n completionHandler: @escaping (Error?) -> Void\n ) {\n Task {\n do {\n let service = try getService()\n try await service.toggleRealtimeSuggestion()\n completionHandler(nil)\n } catch is CancellationError {\n completionHandler(nil)\n } catch {\n completionHandler(error)\n }\n }\n }\n}\n" }, { "path": "ExtensionPoint.appextensionpoint", "content": "\n\n\n\n com.intii.CopilotForXcode.ExtensionService.Extension\n \n EXPresentsUserInterface\n \n \n\n\n" }, { "path": "ExtensionService/AppDelegate+Menu.swift", "content": "import AppKit\nimport Foundation\nimport Preferences\nimport XcodeInspector\nimport Dependencies\nimport Workspace\n\nextension AppDelegate {\n fileprivate var statusBarMenuIdentifier: NSUserInterfaceItemIdentifier {\n .init(\"statusBarMenu\")\n }\n\n fileprivate var xcodeInspectorDebugMenuIdentifier: NSUserInterfaceItemIdentifier {\n .init(\"xcodeInspectorDebugMenu\")\n }\n\n fileprivate var accessibilityAPIPermissionMenuItemIdentifier: NSUserInterfaceItemIdentifier {\n .init(\"accessibilitAPIPermissionMenuItem\")\n }\n\n fileprivate var sourceEditorDebugMenu: NSUserInterfaceItemIdentifier {\n .init(\"sourceEditorDebugMenu\")\n }\n\n @MainActor\n @objc func buildStatusBarMenu() {\n let statusBar = NSStatusBar.system\n statusBarItem = statusBar.statusItem(\n withLength: NSStatusItem.squareLength\n )\n statusBarItem.button?.image = NSImage(named: \"MenuBarIcon\")\n\n let statusBarMenu = NSMenu(title: \"Status Bar Menu\")\n statusBarMenu.identifier = statusBarMenuIdentifier\n statusBarItem.menu = statusBarMenu\n\n let hostAppName = Bundle.main.object(forInfoDictionaryKey: \"HOST_APP_NAME\") as? String\n ?? \"Copilot for Xcode\"\n\n let copilotName = NSMenuItem(\n title: hostAppName,\n action: nil,\n keyEquivalent: \"\"\n )\n\n let checkForUpdate = NSMenuItem(\n title: \"Check for Updates\",\n action: #selector(checkForUpdate),\n keyEquivalent: \"\"\n )\n\n let openExtensionManager = NSMenuItem(\n title: \"Open Extension Manager\",\n action: #selector(openExtensionManager),\n keyEquivalent: \"\"\n )\n\n let openCopilotForXcode = NSMenuItem(\n title: \"Open \\(hostAppName)\",\n action: #selector(openCopilotForXcode),\n keyEquivalent: \"\"\n )\n\n let openGlobalChat = NSMenuItem(\n title: \"Open Chat\",\n action: #selector(openGlobalChat),\n keyEquivalent: \"\"\n )\n\n let xcodeInspectorDebug = NSMenuItem(\n title: \"Xcode Inspector Debug\",\n action: nil,\n keyEquivalent: \"\"\n )\n\n let xcodeInspectorDebugMenu = NSMenu(title: \"Xcode Inspector Debug\")\n xcodeInspectorDebugMenu.identifier = xcodeInspectorDebugMenuIdentifier\n xcodeInspectorDebug.submenu = xcodeInspectorDebugMenu\n xcodeInspectorDebug.isHidden = false\n\n let accessibilityAPIPermission = NSMenuItem(\n title: \"Accessibility API Permission: N/A\",\n action: nil,\n keyEquivalent: \"\"\n )\n accessibilityAPIPermission.identifier = accessibilityAPIPermissionMenuItemIdentifier\n\n let quitItem = NSMenuItem(\n title: \"Quit\",\n action: #selector(quit),\n keyEquivalent: \"\"\n )\n quitItem.target = self\n\n let reactivateObservationsItem = NSMenuItem(\n title: \"Reactivate Observations to Xcode\",\n action: #selector(reactivateObservationsToXcode),\n keyEquivalent: \"\"\n )\n \n let resetWorkspacesItem = NSMenuItem(\n title: \"Reset workspaces\",\n action: #selector(destroyWorkspacePool),\n keyEquivalent: \"\"\n )\n\n reactivateObservationsItem.target = self\n\n statusBarMenu.addItem(copilotName)\n statusBarMenu.addItem(openCopilotForXcode)\n statusBarMenu.addItem(checkForUpdate)\n statusBarMenu.addItem(openExtensionManager)\n statusBarMenu.addItem(.separator())\n statusBarMenu.addItem(openGlobalChat)\n statusBarMenu.addItem(.separator())\n statusBarMenu.addItem(xcodeInspectorDebug)\n statusBarMenu.addItem(accessibilityAPIPermission)\n statusBarMenu.addItem(reactivateObservationsItem)\n statusBarMenu.addItem(resetWorkspacesItem)\n statusBarMenu.addItem(quitItem)\n\n statusBarMenu.delegate = self\n xcodeInspectorDebugMenu.delegate = self\n }\n}\n\nextension AppDelegate: NSMenuDelegate {\n func menuWillOpen(_ menu: NSMenu) {\n switch menu.identifier {\n case statusBarMenuIdentifier:\n if let xcodeInspectorDebug = menu.items.first(where: { item in\n item.submenu?.identifier == xcodeInspectorDebugMenuIdentifier\n }) {\n xcodeInspectorDebug.isHidden = !UserDefaults.shared\n .value(for: \\.enableXcodeInspectorDebugMenu)\n }\n\n if let accessibilityAPIPermission = menu.items.first(where: { item in\n item.identifier == accessibilityAPIPermissionMenuItemIdentifier\n }) {\n AXIsProcessTrusted()\n accessibilityAPIPermission.title =\n \"Accessibility API Permission: \\(AXIsProcessTrusted() ? \"Granted\" : \"Not Granted\")\"\n }\n\n case xcodeInspectorDebugMenuIdentifier:\n let inspector = XcodeInspector.shared\n menu.items.removeAll()\n menu.items\n .append(.text(\"Active Project: \\(inspector.activeProjectRootURL?.path ?? \"N/A\")\"))\n menu.items\n .append(.text(\"Active Workspace: \\(inspector.activeWorkspaceURL?.path ?? \"N/A\")\"))\n menu.items\n .append(.text(\"Active Document: \\(inspector.activeDocumentURL?.path ?? \"N/A\")\"))\n\n if let focusedWindow = inspector.focusedWindow {\n menu.items.append(.text(\n \"Active Window: \\(focusedWindow.uiElement.identifier)\"\n ))\n } else {\n menu.items.append(.text(\"Active Window: N/A\"))\n }\n\n if let focusedElement = inspector.focusedElement {\n menu.items.append(.text(\n \"Focused Element: \\(focusedElement.description)\"\n ))\n } else {\n menu.items.append(.text(\"Focused Element: N/A\"))\n }\n\n if let sourceEditor = inspector.latestFocusedEditor {\n let label = sourceEditor.element.description\n menu.items\n .append(.text(\"Active Source Editor: \\(label.isEmpty ? \"Unknown\" : label)\"))\n } else {\n menu.items.append(.text(\"Active Source Editor: N/A\"))\n }\n\n menu.items.append(.separator())\n\n for xcode in inspector.xcodes {\n let item = NSMenuItem(\n title: \"Xcode \\(xcode.processIdentifier)\",\n action: nil,\n keyEquivalent: \"\"\n )\n menu.addItem(item)\n let xcodeMenu = NSMenu()\n item.submenu = xcodeMenu\n xcodeMenu.items.append(.text(\"Is Active: \\(xcode.isActive)\"))\n xcodeMenu.items\n .append(.text(\"Active Project: \\(xcode.projectRootURL?.path ?? \"N/A\")\"))\n xcodeMenu.items\n .append(.text(\"Active Workspace: \\(xcode.workspaceURL?.path ?? \"N/A\")\"))\n xcodeMenu.items\n .append(.text(\"Active Document: \\(xcode.documentURL?.path ?? \"N/A\")\"))\n\n for (key, workspace) in xcode.realtimeWorkspaces {\n let workspaceItem = NSMenuItem(\n title: \"Workspace \\(key)\",\n action: nil,\n keyEquivalent: \"\"\n )\n xcodeMenu.items.append(workspaceItem)\n let workspaceMenu = NSMenu()\n workspaceItem.submenu = workspaceMenu\n let tabsItem = NSMenuItem(\n title: \"Tabs\",\n action: nil,\n keyEquivalent: \"\"\n )\n workspaceMenu.addItem(tabsItem)\n let tabsMenu = NSMenu()\n tabsItem.submenu = tabsMenu\n for tab in workspace.tabs {\n tabsMenu.addItem(.text(tab))\n }\n }\n }\n\n menu.items.append(.separator())\n\n menu.items.append(NSMenuItem(\n title: \"Restart Xcode Inspector\",\n action: #selector(restartXcodeInspector),\n keyEquivalent: \"\"\n ))\n \n let isDebuggingOverlay = UserDefaults.shared.value(for: \\.debugOverlayPanel)\n let debugOverlayItem = NSMenuItem(\n title: \"Debug Window Overlays\",\n action: #selector(toggleDebugOverlayPanel),\n keyEquivalent: \"\"\n )\n debugOverlayItem.state = isDebuggingOverlay ? .on : .off\n menu.items.append(debugOverlayItem)\n\n default:\n break\n }\n }\n}\n\nimport XPCShared\n\nprivate extension AppDelegate {\n @objc func restartXcodeInspector() {\n Task {\n await XcodeInspector.shared.restart(cleanUp: true)\n }\n }\n\n @objc func reactivateObservationsToXcode() {\n Task {\n await XcodeInspector.shared.reactivateObservationsToXcode()\n }\n }\n\n @objc func openExtensionManager() {\n guard let data = try? JSONEncoder().encode(ExtensionServiceRequests.OpenExtensionManager())\n else { return }\n Task {\n await service.handleXPCServiceRequests(\n endpoint: ExtensionServiceRequests.OpenExtensionManager.endpoint,\n requestBody: data,\n reply: { _, _ in }\n )\n }\n }\n \n @objc func destroyWorkspacePool() {\n @Dependency(\\.workspacePool) var workspacePool: WorkspacePool\n Task {\n await workspacePool.destroy()\n }\n }\n \n @objc func toggleDebugOverlayPanel() {\n let isDebuggingOverlay = UserDefaults.shared.value(for: \\.debugOverlayPanel)\n UserDefaults.shared.set(!isDebuggingOverlay, for: \\.debugOverlayPanel)\n }\n}\n\nprivate extension NSMenuItem {\n static func text(_ text: String) -> NSMenuItem {\n let item = NSMenuItem(\n title: text,\n action: nil,\n keyEquivalent: \"\"\n )\n item.isEnabled = false\n return item\n }\n}\n\n" }, { "path": "ExtensionService/AppDelegate.swift", "content": "import FileChangeChecker\nimport LaunchAgentManager\nimport Logger\nimport Perception\nimport Preferences\nimport Service\nimport ServiceManagement\nimport ServiceUpdateMigration\nimport SwiftUI\nimport UpdateChecker\nimport UserDefaultsObserver\nimport UserNotifications\nimport XcodeInspector\nimport XPCShared\n\nlet bundleIdentifierBase = Bundle.main\n .object(forInfoDictionaryKey: \"BUNDLE_IDENTIFIER_BASE\") as! String\nlet serviceIdentifier = bundleIdentifierBase + \".ExtensionService\"\n\n@main\nclass AppDelegate: NSObject, NSApplicationDelegate, NSWindowDelegate {\n @MainActor\n let service = Service.shared\n var statusBarItem: NSStatusItem!\n var xpcController: XPCController?\n let updateChecker = UpdateChecker(\n hostBundle: locateHostBundleURL(url: Bundle.main.bundleURL)\n .flatMap(Bundle.init(url:)),\n shouldAutomaticallyCheckForUpdate: true\n )\n\n func applicationDidFinishLaunching(_: Notification) {\n// isPerceptionCheckingEnabled = false\n if ProcessInfo.processInfo.environment[\"IS_UNIT_TEST\"] == \"YES\" { return }\n _ = XcodeInspector.shared\n updateChecker.updateCheckerDelegate = self\n service.start()\n AXIsProcessTrustedWithOptions([\n kAXTrustedCheckOptionPrompt.takeRetainedValue() as NSString: true,\n ] as CFDictionary)\n setupQuitOnUpdate()\n setupQuitOnUserTerminated()\n xpcController = .init()\n Logger.service.info(\"XPC Service started.\")\n NSApp.setActivationPolicy(.accessory)\n buildStatusBarMenu()\n\n Task {\n do {\n try await ServiceUpdateMigrator().migrate()\n } catch {\n Logger.service.error(error.localizedDescription)\n }\n }\n }\n\n @objc func quit() {\n Task { @MainActor in\n await service.prepareForExit()\n await xpcController?.quit()\n NSApp.terminate(self)\n }\n }\n\n @objc func openCopilotForXcode() {\n let task = Process()\n if let appPath = locateHostBundleURL(url: Bundle.main.bundleURL)?.absoluteString {\n task.launchPath = \"/usr/bin/open\"\n task.arguments = [appPath]\n task.launch()\n task.waitUntilExit()\n }\n }\n\n @objc func openGlobalChat() {\n Task { @MainActor in\n let serviceGUI = Service.shared.guiController\n serviceGUI.openGlobalChat()\n }\n }\n\n func setupQuitOnUpdate() {\n Task {\n guard let url = Bundle.main.executableURL else { return }\n let checker = await FileChangeChecker(fileURL: url)\n\n // If Xcode or Copilot for Xcode is made active, check if the executable of this program\n // is changed. If changed, quit this program.\n\n let sequence = NSWorkspace.shared.notificationCenter\n .notifications(named: NSWorkspace.didActivateApplicationNotification)\n for await notification in sequence {\n try Task.checkCancellation()\n guard let app = notification\n .userInfo?[NSWorkspace.applicationUserInfoKey] as? NSRunningApplication,\n app.isUserOfService\n else { continue }\n guard await checker.checkIfChanged() else {\n Logger.service.info(\"Extension Service is not updated, no need to quit.\")\n continue\n }\n Logger.service.info(\"Extension Service will quit.\")\n #if DEBUG\n #else\n quit()\n #endif\n }\n }\n }\n\n func setupQuitOnUserTerminated() {\n Task {\n // Whenever Xcode or the host application quits, check if any of the two is running.\n // If none, quit the XPC service.\n\n let sequence = NSWorkspace.shared.notificationCenter\n .notifications(named: NSWorkspace.didTerminateApplicationNotification)\n for await notification in sequence {\n try Task.checkCancellation()\n guard UserDefaults.shared.value(for: \\.quitXPCServiceOnXcodeAndAppQuit)\n else { continue }\n guard let app = notification\n .userInfo?[NSWorkspace.applicationUserInfoKey] as? NSRunningApplication,\n app.isUserOfService\n else { continue }\n if NSWorkspace.shared.runningApplications.contains(where: \\.isUserOfService) {\n continue\n }\n quit()\n }\n }\n }\n\n func requestAccessoryAPIPermission() {\n AXIsProcessTrustedWithOptions([\n kAXTrustedCheckOptionPrompt.takeRetainedValue() as NSString: true,\n ] as NSDictionary)\n }\n\n @objc func checkForUpdate() {\n updateChecker.checkForUpdates()\n }\n}\n\nextension AppDelegate: UpdateCheckerDelegate {\n func prepareForRelaunch(finish: @escaping () -> Void) {\n Task {\n await service.prepareForExit()\n finish()\n }\n }\n}\n\nextension NSRunningApplication {\n var isUserOfService: Bool {\n [\n \"com.apple.dt.Xcode\",\n bundleIdentifierBase,\n ].contains(bundleIdentifier)\n }\n}\n\nfunc locateHostBundleURL(url: URL) -> URL? {\n var nextURL = url\n while nextURL.path != \"/\" {\n nextURL = nextURL.deletingLastPathComponent()\n if nextURL.lastPathComponent.hasSuffix(\".app\") {\n return nextURL\n }\n }\n let devAppURL = url\n .deletingLastPathComponent()\n .appendingPathComponent(\"Copilot for Xcode Dev.app\")\n return devAppURL\n}\n\n" }, { "path": "ExtensionService/Assets.xcassets/AccentColor.colorset/Contents.json", "content": "{\n \"colors\" : [\n {\n \"idiom\" : \"universal\"\n }\n ],\n \"info\" : {\n \"author\" : \"xcode\",\n \"version\" : 1\n }\n}\n" }, { "path": "ExtensionService/Assets.xcassets/AppIcon.appiconset/Contents.json", "content": "{\n \"images\" : [\n {\n \"filename\" : \"service-icon@16w.png\",\n \"idiom\" : \"mac\",\n \"scale\" : \"1x\",\n \"size\" : \"16x16\"\n },\n {\n \"filename\" : \"service-icon@32w.png\",\n \"idiom\" : \"mac\",\n \"scale\" : \"2x\",\n \"size\" : \"16x16\"\n },\n {\n \"filename\" : \"service-icon@32w.png\",\n \"idiom\" : \"mac\",\n \"scale\" : \"1x\",\n \"size\" : \"32x32\"\n },\n {\n \"filename\" : \"service-icon@64w.png\",\n \"idiom\" : \"mac\",\n \"scale\" : \"2x\",\n \"size\" : \"32x32\"\n },\n {\n \"filename\" : \"service-icon@128w.png\",\n \"idiom\" : \"mac\",\n \"scale\" : \"1x\",\n \"size\" : \"128x128\"\n },\n {\n \"filename\" : \"service-icon@256w.png\",\n \"idiom\" : \"mac\",\n \"scale\" : \"2x\",\n \"size\" : \"128x128\"\n },\n {\n \"filename\" : \"service-icon@256w.png\",\n \"idiom\" : \"mac\",\n \"scale\" : \"1x\",\n \"size\" : \"256x256\"\n },\n {\n \"filename\" : \"service-icon@512w.png\",\n \"idiom\" : \"mac\",\n \"scale\" : \"2x\",\n \"size\" : \"256x256\"\n },\n {\n \"filename\" : \"service-icon@512w.png\",\n \"idiom\" : \"mac\",\n \"scale\" : \"1x\",\n \"size\" : \"512x512\"\n },\n {\n \"filename\" : \"service-icon.png\",\n \"idiom\" : \"mac\",\n \"scale\" : \"2x\",\n \"size\" : \"512x512\"\n }\n ],\n \"info\" : {\n \"author\" : \"xcode\",\n \"version\" : 1\n }\n}\n" }, { "path": "ExtensionService/Assets.xcassets/Contents.json", "content": "{\n \"info\" : {\n \"author\" : \"xcode\",\n \"version\" : 1\n }\n}\n" }, { "path": "ExtensionService/Assets.xcassets/MenuBarIcon.imageset/Contents.json", "content": "{\n \"images\" : [\n {\n \"filename\" : \"octopus.png\",\n \"idiom\" : \"universal\",\n \"scale\" : \"1x\"\n },\n {\n \"filename\" : \"octopus 2.png\",\n \"idiom\" : \"universal\",\n \"scale\" : \"2x\"\n },\n {\n \"filename\" : \"octopus 1.png\",\n \"idiom\" : \"universal\",\n \"scale\" : \"3x\"\n }\n ],\n \"info\" : {\n \"author\" : \"xcode\",\n \"version\" : 1\n },\n \"properties\" : {\n \"template-rendering-intent\" : \"template\"\n }\n}\n" }, { "path": "ExtensionService/ExtensionService.entitlements", "content": "\n\n\n\n\tcom.apple.security.application-groups\n\t\n\t\t$(TeamIdentifierPrefix)group.$(BUNDLE_IDENTIFIER_BASE)\n\t\n\tkeychain-access-groups\n\t\n\t\t$(AppIdentifierPrefix)$(BUNDLE_IDENTIFIER_BASE).Shared\n\t\n\n\n" }, { "path": "ExtensionService/Info.plist", "content": "\n\n\n\n\tAPPLICATION_SUPPORT_FOLDER\n\t$(APPLICATION_SUPPORT_FOLDER)\n\tAPP_ID_PREFIX\n\t$(AppIdentifierPrefix)\n\tBUNDLE_IDENTIFIER_BASE\n\t$(BUNDLE_IDENTIFIER_BASE)\n\tEXTENSION_BUNDLE_NAME\n\t$(EXTENSION_BUNDLE_NAME)\n\tHOST_APP_NAME\n\t$(HOST_APP_NAME)\n\tNSAppTransportSecurity\n\t\n\t\tNSAllowsArbitraryLoads\n\t\t\n\t\n\tTEAM_ID_PREFIX\n\t$(TeamIdentifierPrefix)\n\tXPCService\n\t\n\t\tServiceType\n\t\tApplication\n\t\n\n\n" }, { "path": "ExtensionService/Main.storyboard", "content": "\n\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \tDefault\n \n \n \n \n \n \n \tLeft to Right\n \n \n \n \n \n \n \tRight to Left\n \n \n \n \n \n \n \n \n \n \n \tDefault\n \n \n \n \n \n \n \tLeft to Right\n \n \n \n \n \n \n \tRight to Left\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n\n" }, { "path": "ExtensionService/ServiceDelegate.swift", "content": "import Foundation\nimport Service\nimport XPCShared\n\nclass ServiceDelegate: NSObject, NSXPCListenerDelegate {\n func listener(\n _: NSXPCListener,\n shouldAcceptNewConnection newConnection: NSXPCConnection\n ) -> Bool {\n newConnection.exportedInterface = NSXPCInterface(\n with: XPCServiceProtocol.self\n )\n\n let exportedObject = XPCService()\n newConnection.exportedObject = exportedObject\n newConnection.resume()\n return true\n }\n}\n\n" }, { "path": "ExtensionService/XPCController.swift", "content": "import Foundation\nimport Logger\nimport XPCShared\n\nfinal class XPCController: XPCServiceDelegate {\n let bridge: XPCCommunicationBridge\n let xpcListener: NSXPCListener\n let xpcServiceDelegate: ServiceDelegate\n\n var pingTask: Task?\n\n init() {\n let bridge = XPCCommunicationBridge(logger: .client)\n let listener = NSXPCListener.anonymous()\n let delegate = ServiceDelegate()\n listener.delegate = delegate\n listener.resume()\n xpcListener = listener\n xpcServiceDelegate = delegate\n self.bridge = bridge\n\n Task {\n bridge.setDelegate(self)\n createPingTask()\n }\n }\n \n func quit() async {\n bridge.setDelegate(nil)\n pingTask?.cancel()\n try? await bridge.quit()\n }\n\n deinit {\n xpcListener.invalidate()\n pingTask?.cancel()\n }\n\n func createPingTask() {\n pingTask?.cancel()\n pingTask = Task { [weak self] in\n while !Task.isCancelled {\n guard let self else { return }\n do {\n try await self.bridge.updateServiceEndpoint(self.xpcListener.endpoint)\n try await Task.sleep(nanoseconds: 60_000_000_000)\n } catch {\n try await Task.sleep(nanoseconds: 1_000_000_000)\n #if DEBUG\n // No log, but you should run CommunicationBridge, too.\n #else\n Logger.service\n .error(\"Failed to connect to bridge: \\(error.localizedDescription)\")\n #endif\n }\n }\n }\n }\n\n func connectionDidInvalidate() async {\n // ignore\n }\n\n func connectionDidInterrupt() async {\n createPingTask() // restart the ping task so that it can bring the bridge back immediately.\n }\n}\n\n" }, { "path": "Helper/ReloadLaunchAgent.swift", "content": "import ArgumentParser\nimport Foundation\n\nstruct ReloadLaunchAgent: ParsableCommand {\n static var configuration = CommandConfiguration(\n abstract: \"Reload the launch agent\"\n )\n\n @Option(name: .long, help: \"The service identifier of the service.\")\n var serviceIdentifier: String\n\n var launchAgentDirURL: URL {\n FileManager.default.homeDirectoryForCurrentUser\n .appendingPathComponent(\"Library/LaunchAgents\")\n }\n\n var launchAgentPath: String {\n launchAgentDirURL.appendingPathComponent(\"\\(serviceIdentifier).plist\").path\n }\n\n func run() throws {\n try? launchctl(\"unload\", launchAgentPath)\n try launchctl(\"load\", launchAgentPath)\n }\n}\n\nprivate func launchctl(_ args: String...) throws {\n return try process(\"/bin/launchctl\", args)\n}\n\nprivate func process(_ launchPath: String, _ args: [String]) throws {\n let task = Process()\n task.launchPath = launchPath\n task.arguments = args\n task.environment = [\n \"PATH\": \"/usr/bin\",\n ]\n let outpipe = Pipe()\n task.standardOutput = outpipe\n try task.run()\n task.waitUntilExit()\n\n struct E: Error, LocalizedError {\n var errorDescription: String?\n }\n\n if task.terminationStatus == 0 {\n return\n }\n throw E(\n errorDescription: \"Failed to restart. Please make sure the launch agent is already loaded.\"\n )\n}\n" }, { "path": "Helper/main.swift", "content": "import ArgumentParser\nimport Foundation\n\nstruct Helper: ParsableCommand {\n static var configuration = CommandConfiguration(\n commandName: \"helper\",\n abstract: \"Helper CLI for Copilot for Xcode\",\n subcommands: [\n ReloadLaunchAgent.self,\n ]\n )\n}\n\nHelper.main()\n" }, { "path": "LICENSE", "content": "MIT License\n\nCopyright (c) 2024 Shangxin Guo \n\nPermission is hereby granted, free of charge, to any person obtaining a copy\nof this software and associated documentation files (the \"Software\"), to deal\nin the Software without restriction, including without limitation the rights\nto use, copy, modify, merge, publish, distribute, sublicense, and/or sell\ncopies of the Software, and to permit persons to whom the Software is\nfurnished to do so, subject to the following conditions:\n\nThe above copyright notice and this permission notice shall be included in all\ncopies or substantial portions of the Software.\n\nTHE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\nIMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\nFITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\nAUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\nLIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\nOUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\nSOFTWARE.\n" }, { "path": "OverlayWindow/.gitignore", "content": ".DS_Store\n/.build\n/Packages\nxcuserdata/\nDerivedData/\n.swiftpm/configuration/registries.json\n.swiftpm/xcode/package.xcworkspace/contents.xcworkspacedata\n.netrc\n" }, { "path": "OverlayWindow/Package.swift", "content": "// swift-tools-version: 6.2\n// The swift-tools-version declares the minimum version of Swift required to build this package.\n\nimport PackageDescription\n\nlet package = Package(\n name: \"OverlayWindow\",\n platforms: [.macOS(.v13)],\n products: [\n .library(\n name: \"OverlayWindow\",\n targets: [\"OverlayWindow\"]\n ),\n ],\n dependencies: [\n .package(path: \"../Tool\"),\n .package(url: \"https://github.com/pointfreeco/swift-perception\", from: \"1.3.4\"),\n .package(url: \"https://github.com/pointfreeco/swift-dependencies\", from: \"1.4.0\"),\n ],\n targets: [\n .target(\n name: \"OverlayWindow\",\n dependencies: [\n .product(name: \"AppMonitoring\", package: \"Tool\"),\n .product(name: \"Toast\", package: \"Tool\"),\n .product(name: \"Preferences\", package: \"Tool\"),\n .product(name: \"Logger\", package: \"Tool\"),\n .product(name: \"DebounceFunction\", package: \"Tool\"),\n .product(name: \"Perception\", package: \"swift-perception\"),\n .product(name: \"Dependencies\", package: \"swift-dependencies\"),\n ]\n ),\n .testTarget(\n name: \"OverlayWindowTests\",\n dependencies: [\"OverlayWindow\", .product(name: \"DebounceFunction\", package: \"Tool\")]\n ),\n ]\n)\n\n" }, { "path": "OverlayWindow/Sources/OverlayWindow/IDEWorkspaceWindowOverlayWindowController.swift", "content": "import AppKit\nimport AXExtension\nimport AXNotificationStream\nimport DebounceFunction\nimport Foundation\nimport Perception\nimport SwiftUI\nimport XcodeInspector\n\n@MainActor\npublic protocol IDEWorkspaceWindowOverlayWindowControllerContentProvider {\n associatedtype Content: View\n func createWindow() -> NSWindow?\n func createContent() -> Content\n func destroy()\n\n init(windowInspector: WorkspaceXcodeWindowInspector, application: NSRunningApplication)\n}\n\nextension IDEWorkspaceWindowOverlayWindowControllerContentProvider {\n var contentBody: AnyView {\n AnyView(createContent())\n }\n}\n\n@MainActor\nfinal class IDEWorkspaceWindowOverlayWindowController {\n private var lastAccessDate: Date = .init()\n let application: NSRunningApplication\n let inspector: WorkspaceXcodeWindowInspector\n let contentProviders: [any IDEWorkspaceWindowOverlayWindowControllerContentProvider]\n let maskPanel: OverlayPanel\n var windowElement: AXUIElement\n private var axNotificationTask: Task?\n let updateFrameThrottler = ThrottleRunner(duration: 0.2)\n\n init(\n inspector: WorkspaceXcodeWindowInspector,\n application: NSRunningApplication,\n contentProviderFactory: (\n _ windowInspector: WorkspaceXcodeWindowInspector, _ application: NSRunningApplication\n ) -> [any IDEWorkspaceWindowOverlayWindowControllerContentProvider]\n ) {\n self.inspector = inspector\n self.application = application\n let contentProviders = contentProviderFactory(inspector, application)\n self.contentProviders = contentProviders\n windowElement = inspector.uiElement\n\n let panel = OverlayPanel(\n contentRect: .init(x: 0, y: 0, width: 200, height: 200)\n ) {\n ZStack {\n ForEach(0..(\n contentRect: NSRect,\n @ViewBuilder content: @escaping () -> Content\n ) {\n super.init(\n contentRect: contentRect,\n styleMask: [\n .borderless,\n .nonactivatingPanel,\n .fullSizeContentView,\n ],\n backing: .buffered,\n defer: false\n )\n\n isReleasedWhenClosed = false\n menu = nil\n isOpaque = true\n backgroundColor = .clear\n hasShadow = false\n alphaValue = 1.0\n collectionBehavior = [.fullScreenAuxiliary]\n isFloatingPanel = true\n titleVisibility = .hidden\n titlebarAppearsTransparent = true\n animationBehavior = .utilityWindow\n\n standardWindowButton(.closeButton)?.isHidden = true\n standardWindowButton(.miniaturizeButton)?.isHidden = true\n standardWindowButton(.zoomButton)?.isHidden = true\n\n contentView = NSHostingView(\n rootView: ContentWrapper(panelState: panelState) { content() }\n )\n }\n\n override public var canBecomeKey: Bool {\n return _canBecomeKey\n }\n\n override public var canBecomeMain: Bool {\n return false\n }\n\n override public func setIsVisible(_ visible: Bool) {\n _canBecomeKey = false\n defer { _canBecomeKey = true }\n super.setIsVisible(visible)\n }\n\n public func moveToActiveSpace() {\n collectionBehavior = [.fullScreenAuxiliary, .moveToActiveSpace]\n Task { @MainActor in\n try await Task.sleep(nanoseconds: 50_000_000)\n self.collectionBehavior = [.fullScreenAuxiliary]\n }\n }\n\n func setTopLeftCoordinateFrame(_ frame: CGRect, display: Bool) {\n let zeroScreen = NSScreen.screens.first { $0.frame.origin == .zero }\n ?? NSScreen.primaryScreen ?? NSScreen.main\n let panelFrame = Self.convertAXRectToNSPanelFrame(\n axRect: frame,\n forPrimaryScreen: zeroScreen\n )\n panelState.windowFrame = frame\n panelState.windowFrameNSCoordinate = panelFrame\n setFrame(panelFrame, display: display)\n }\n\n static func convertAXRectToNSPanelFrame(\n axRect: CGRect,\n forPrimaryScreen screen: NSScreen?\n ) -> CGRect {\n guard let screen = screen else { return .zero }\n let screenFrame = screen.frame\n let flippedY = screenFrame.origin.y + screenFrame.size\n .height - (axRect.origin.y + axRect.size.height)\n return CGRect(\n x: axRect.origin.x,\n y: flippedY,\n width: axRect.size.width,\n height: axRect.size.height\n )\n }\n\n struct ContentWrapper: View {\n let panelState: PanelState\n @ViewBuilder let content: () -> Content\n @AppStorage(\\.debugOverlayPanel) var debugOverlayPanel\n\n var body: some View {\n WithPerceptionTracking {\n ZStack {\n Rectangle().fill(.green.opacity(debugOverlayPanel ? 0.1 : 0))\n .allowsHitTesting(false)\n content()\n .environment(\\.overlayFrame, panelState.windowFrame)\n .environment(\\.overlayDebug, debugOverlayPanel)\n }\n }\n }\n }\n}\n\nfunc overlayLevel(_ addition: Int) -> NSWindow.Level {\n let minimumWidgetLevel: Int\n #if DEBUG\n minimumWidgetLevel = NSWindow.Level.floating.rawValue + 1\n #else\n minimumWidgetLevel = NSWindow.Level.floating.rawValue\n #endif\n return .init(minimumWidgetLevel + addition)\n}\n\npublic extension CGRect {\n func flipped(relativeTo reference: CGRect) -> CGRect {\n let flippedOrigin = CGPoint(\n x: origin.x,\n y: reference.height - origin.y - height\n )\n return CGRect(origin: flippedOrigin, size: size)\n }\n\n func relative(to reference: CGRect) -> CGRect {\n let relativeOrigin = CGPoint(\n x: origin.x - reference.origin.x,\n y: origin.y - reference.origin.y\n )\n return CGRect(origin: relativeOrigin, size: size)\n }\n}\n\npublic extension NSScreen {\n var isPrimary: Bool {\n let id = deviceDescription[.init(\"NSScreenNumber\")] as? CGDirectDisplayID\n return id == CGMainDisplayID()\n }\n\n static var primaryScreen: NSScreen? {\n NSScreen.screens.first {\n let id = $0.deviceDescription[.init(\"NSScreenNumber\")] as? CGDirectDisplayID\n return id == CGMainDisplayID()\n }\n }\n}\n\n" }, { "path": "OverlayWindow/Sources/OverlayWindow/OverlayWindowController.swift", "content": "import AppKit\nimport DebounceFunction\nimport Foundation\nimport Perception\nimport XcodeInspector\n\n@MainActor\npublic final class OverlayWindowController {\n public typealias IDEWorkspaceWindowOverlayWindowControllerContentProviderFactory =\n @MainActor @Sendable (\n _ windowInspector: WorkspaceXcodeWindowInspector,\n _ application: NSRunningApplication\n ) -> any IDEWorkspaceWindowOverlayWindowControllerContentProvider\n\n static var ideWindowOverlayWindowControllerContentProviderFactories:\n [IDEWorkspaceWindowOverlayWindowControllerContentProviderFactory] = []\n\n var ideWindowOverlayWindowControllers =\n [ObjectIdentifier: IDEWorkspaceWindowOverlayWindowController]()\n var updateWindowStateTask: Task?\n\n let windowUpdateThrottler = ThrottleRunner(duration: 0.2)\n\n lazy var fullscreenDetector = {\n let it = NSWindow(\n contentRect: .zero,\n styleMask: .borderless,\n backing: .buffered,\n defer: false\n )\n it.isReleasedWhenClosed = false\n it.isOpaque = false\n it.backgroundColor = .clear\n it.collectionBehavior = [.canJoinAllSpaces, .fullScreenAuxiliary, .transient]\n it.hasShadow = false\n it.setIsVisible(false)\n return it\n }()\n\n public init() {}\n\n public func start() {\n observeEvents()\n _ = fullscreenDetector\n }\n\n public nonisolated static func registerIDEWorkspaceWindowOverlayWindowControllerContentProviderFactory(\n _ factory: @escaping IDEWorkspaceWindowOverlayWindowControllerContentProviderFactory\n ) {\n Task { @MainActor in\n ideWindowOverlayWindowControllerContentProviderFactories.append(factory)\n }\n }\n}\n\nextension OverlayWindowController {\n func observeEvents() {\n observeWindowChange()\n\n updateWindowStateTask = Task { [weak self] in\n if let self { await handleSpaceChange() }\n\n await withThrowingTaskGroup(of: Void.self) { [weak self] group in\n // active space did change\n _ = group.addTaskUnlessCancelled { [weak self] in\n let sequence = NSWorkspace.shared.notificationCenter\n .notifications(named: NSWorkspace.activeSpaceDidChangeNotification)\n for await _ in sequence {\n guard let self else { return }\n try Task.checkCancellation()\n await handleSpaceChange()\n }\n }\n }\n }\n }\n}\n\nprivate extension OverlayWindowController {\n func observeWindowChange() {\n if ideWindowOverlayWindowControllers.isEmpty {\n if let app = XcodeInspector.shared.activeXcode,\n let windowInspector = XcodeInspector.shared\n .focusedWindow as? WorkspaceXcodeWindowInspector\n {\n createNewIDEOverlayWindowController(\n inspector: windowInspector,\n application: app.runningApplication\n )\n }\n }\n\n withPerceptionTracking {\n _ = XcodeInspector.shared.focusedWindow\n _ = XcodeInspector.shared.activeXcode\n _ = XcodeInspector.shared.activeApplication\n } onChange: { [weak self] in\n guard let self else { return }\n Task { @MainActor in\n defer { self.observeWindowChange() }\n await self.windowUpdateThrottler.throttle { [weak self] in\n await self?.handleOverlayStatusChange()\n }\n }\n }\n }\n\n func createNewIDEOverlayWindowController(\n inspector: WorkspaceXcodeWindowInspector,\n application: NSRunningApplication\n ) {\n let id = ObjectIdentifier(inspector)\n let newController = IDEWorkspaceWindowOverlayWindowController(\n inspector: inspector,\n application: application,\n contentProviderFactory: {\n windowInspector, application in\n OverlayWindowController.ideWindowOverlayWindowControllerContentProviderFactories\n .map { $0(windowInspector, application) }\n }\n )\n newController.access()\n ideWindowOverlayWindowControllers[id] = newController\n }\n\n func removeIDEOverlayWindowController(for id: ObjectIdentifier) {\n if let controller = ideWindowOverlayWindowControllers[id] {\n controller.destroy()\n }\n ideWindowOverlayWindowControllers[id] = nil\n }\n\n func handleSpaceChange() async {\n let windowInspector = XcodeInspector.shared.focusedWindow\n guard let activeWindowController = {\n if let windowInspector = windowInspector as? WorkspaceXcodeWindowInspector {\n let id = ObjectIdentifier(windowInspector)\n return ideWindowOverlayWindowControllers[id]\n } else {\n return nil\n }\n }() else { return }\n\n let activeXcode = XcodeInspector.shared.activeXcode\n let xcode = activeXcode?.appElement\n let isXcodeActive = xcode?.isFrontmost ?? false\n if isXcodeActive {\n activeWindowController.maskPanel.moveToActiveSpace()\n }\n\n if fullscreenDetector.isOnActiveSpace, xcode?.focusedWindow != nil {\n activeWindowController.maskPanel.orderFrontRegardless()\n }\n }\n\n func handleOverlayStatusChange() {\n guard XcodeInspector.shared.activeApplication?.isXcode ?? false else {\n var closedControllers: [ObjectIdentifier] = []\n for (id, controller) in ideWindowOverlayWindowControllers {\n if controller.isWindowClosed {\n controller.dim()\n closedControllers.append(id)\n } else {\n controller.dim()\n }\n }\n for id in closedControllers {\n removeIDEOverlayWindowController(for: id)\n }\n return\n }\n\n guard let app = XcodeInspector.shared.activeXcode else {\n for (_, controller) in ideWindowOverlayWindowControllers {\n controller.hide()\n }\n return\n }\n\n let windowInspector = XcodeInspector.shared.focusedWindow\n if let ideWindowInspector = windowInspector as? WorkspaceXcodeWindowInspector {\n let objectID = ObjectIdentifier(ideWindowInspector)\n // Workspace window is active\n // Hide all controllers first\n for (id, controller) in ideWindowOverlayWindowControllers {\n if id != objectID {\n controller.hide()\n }\n }\n if let controller = ideWindowOverlayWindowControllers[objectID] {\n controller.access()\n } else {\n createNewIDEOverlayWindowController(\n inspector: ideWindowInspector,\n application: app.runningApplication\n )\n }\n } else {\n // Not a workspace window, dim all controllers\n for (_, controller) in ideWindowOverlayWindowControllers {\n controller.dim()\n }\n }\n }\n}\n\n" }, { "path": "OverlayWindow/Tests/OverlayWindowTests/WindowTests.swift", "content": "import Testing\n\n@Test func example() async throws {\n // Write your test here and use APIs like `#expect(...)` to check expected conditions.\n}\n" }, { "path": "Playground.playground/Pages/RetrievalQAChain.xcplaygroundpage/Contents.swift", "content": "import AppKit\nimport Foundation\nimport LangChain\nimport OpenAIService\nimport PlaygroundSupport\nimport SwiftUI\nimport TokenEncoder\n\nstruct QAForm: View {\n @State var relevantInformation = [String]()\n @State var relevantDocuments = [(document: Document, distance: Float)]()\n @State var duration: TimeInterval = 0\n @State var answer: String = \"\"\n @State var tokenCount: Int = 0\n @State var question: String = \"What is Swift macros?\"\n @State var isProcessing: Bool = false\n @State var url: String = \"https://developer.apple.com/documentation/swift/applying-macros\"\n\n var body: some View {\n HStack(spacing: 0) {\n ScrollView {\n Form {\n Section(header: Text(\"Input\")) {\n TextField(\"URL\", text: $url)\n TextField(\"Question\", text: $question)\n HStack {\n Button(\"Ask\") {\n Task {\n do {\n try await ask()\n } catch {\n answer = error.localizedDescription\n }\n }\n }\n .disabled(isProcessing)\n \n Text(\"\\(duration) seconds\")\n }\n }\n Section(header: Text(\"All Relevant Information (\\(tokenCount) words)\")) {\n Text(answer)\n }\n Section(header: Text(\"Relevant Information\")) {\n ForEach(0..\n\n \n \n\n" }, { "path": "Playground.playground/Pages/WebScrapper.xcplaygroundpage/Contents.swift", "content": "import AppKit\nimport LangChain\nimport PlaygroundSupport\nimport SwiftUI\n\nstruct ScrapperForm: View {\n @State var webDocuments: [Document] = []\n @State var isProcessing: Bool = false\n @State var url: String = \"https://developer.apple.com/documentation/swift/applying-macros\"\n\n var body: some View {\n Form {\n Section(header: Text(\"Input\")) {\n TextField(\"URL\", text: $url)\n Button(\"Scrap\") {\n Task {\n do {\n try await scrap()\n } catch {\n webDocuments =\n [.init(pageContent: error.localizedDescription, metadata: [:])]\n }\n }\n }\n .disabled(isProcessing)\n }\n Section(header: Text(\"Web Content\")) {\n ForEach(webDocuments, id: \\.pageContent) { document in\n VStack(alignment: .leading) {\n Text(document.pageContent)\n .font(.body)\n }\n Divider()\n }\n }\n }\n .formStyle(.grouped)\n }\n\n func scrap() async throws {\n webDocuments = []\n isProcessing = true\n defer { isProcessing = false }\n guard let url = URL(string: url) else { return }\n let webLoader = WebLoader(urls: [url])\n webDocuments = try await webLoader.load()\n }\n}\n\nlet hostingView = NSHostingController(\n rootView: ScrapperForm()\n .frame(width: 600, height: 800)\n)\n\nPlaygroundPage.current.needsIndefiniteExecution = true\nPlaygroundPage.current.liveView = hostingView\n\n" }, { "path": "Playground.playground/Pages/WebScrapper.xcplaygroundpage/timeline.xctimeline", "content": "\n\n \n \n\n" }, { "path": "Playground.playground/contents.xcplayground", "content": "\n\n \n \n \n \n" }, { "path": "README.md", "content": "# Copilot for Xcode \"Logo\"\n\n![Screenshot](/Screenshot.png)\n\nCopilot for Xcode is an Xcode Source Editor Extension that provides GitHub Copilot, Codeium and ChatGPT support for Xcode.\n\n\"Buy\n\n## Features\n\n- Code Suggestions\n- Chat\n- Modification\n- Custom Commands to extend Chat and Modification.\n\n## Table of Contents\n\n- [Copilot for Xcode ](#copilot-for-xcode-)\n - [Features](#features)\n - [Table of Contents](#table-of-contents)\n - [Prerequisites](#prerequisites)\n - [Permissions Required](#permissions-required)\n - [Installation and Setup](#installation-and-setup)\n - [Install](#install)\n - [Enable the Extension](#enable-the-extension)\n - [macOS 15](#macos-15)\n - [MacOS 14](#macos-14)\n - [Older Versions](#older-versions)\n - [Granting Permissions to the App](#granting-permissions-to-the-app)\n - [Setting Up Key Bindings](#setting-up-key-bindings)\n - [Setting Up Global Hotkeys](#setting-up-global-hotkeys)\n - [Setting Up Suggestion Feature](#setting-up-suggestion-feature)\n - [Setting Up GitHub Copilot](#setting-up-github-copilot)\n - [Setting Up Codeium](#setting-up-codeium)\n - [Setting Up Chat Feature](#setting-up-chat-feature)\n - [Managing `CopilotForXcodeExtensionService.app`](#managing-copilotforxcodeextensionserviceapp)\n - [Update](#update)\n - [Feature](#feature)\n - [Suggestion](#suggestion)\n - [Commands](#commands)\n - [Chat](#chat)\n - [Commands](#commands-1)\n - [Keyboard Shortcuts](#keyboard-shortcuts)\n - [Chat Commands](#chat-commands)\n - [Prompt to Code](#prompt-to-code)\n - [Commands](#commands-2)\n - [Custom Commands](#custom-commands)\n - [Limitations](#limitations)\n - [License](#license)\n\nFor development instruction, check [Development.md](DEVELOPMENT.md).\n\nFor more information, check the [Wiki Page](https://copilotforxcode.intii.com/wiki).\n\n## Prerequisites\n\n- Public network connection.\n\nFor suggestion features:\n\n- For GitHub Copilot users:\n - [Node](https://nodejs.org/) installed to run the Copilot LSP.\n - Active GitHub Copilot subscription.\n- For Codeium users:\n - Active Codeium account.\n- Access to other LLMs.\n\nFor chat and prompt to code features:\n\n- A valid OpenAI API key.\n- Access to other LLMs.\n\n## Permissions Required\n\n- Folder Access\n- Accessibility API\n\n> If you are concerned about key logging and cannot trust the binary, we recommend examining the code and [building it yourself](DEVELOPMENT.md). To address any concerns, you can specifically search for `CGEvent.tapCreate`, `AXObserver`, `AX___` within the code.\n\n## Installation and Setup\n\n> The installation process is a bit complicated. Here is a quick overview:\n> \n> 1. Install the app into the Applications folder, open it once.\n> 2. Enable the source editor extension.\n> 3. Grant Accessibility API permission to the extension app.\n> 4. Setup accounts and models in the host app.\n> 5. Optionally, update the settings of each feature in the host app, and setup keybindings.\n\n### Install\n\nYou can install it via [Homebrew](http://brew.sh/):\n\n```bash\nbrew install --cask copilot-for-xcode\n```\n\nOr install it manually, by downloading the `Copilot for Xcode.app` from the latest [release](https://github.com/intitni/CopilotForXcode/releases).\n\nPlease make sure the app is inside the Applications folder.\n\nOpen the app, the app will create a launch agent to setup a background running Service that does the real job.\n\n### Enable the Extension\n\nEnable the extension in `System Settings.app`.\n\n#### macOS 15\nFrom the Apple menu located in the top-left corner of your screen click `System Settings`. Navigate to `General` then `Login Items & Extensions`. Click `Xcode Source Editor` and tick `Copilot for Xcode`.\n\n#### MacOS 14\nFrom the Apple menu located in the top-left corner of your screen click `System Settings`. Navigate to `Privacy & Security` then toward the bottom click `Extensions`. Click `Xcode Source Editor` and tick `Copilot`.\n\n#### Older Versions\nIf you are using macOS Monterey, enter the `Extensions` menu in `System Preferences.app` with its dedicated icon.\n\n### Granting Permissions to the App\n\nThe first time the app is open and command run, the extension will ask for the necessary permissions.\n\nAlternatively, you may manually grant the required permissions by navigating to the `Privacy & Security` tab in the `System Settings.app`.\n\n- To grant permissions for the Accessibility API, click `Accessibility`, and drag `CopilotForXcodeExtensionService.app` to the list. You can locate the extension app by clicking `Reveal Extension App in Finder` in the host app.\n\n\"Accessibility\n\nIf you encounter an alert requesting permission that you have previously granted, please remove the permission from the list and add it again to re-grant the necessary permissions.\n\n### Setting Up Key Bindings\n\nThe extension will work better if you use key bindings.\n\nIt looks like there is no way to add default key bindings to commands, but you can set them up in `Xcode settings > Key Bindings`. You can filter the list by typing `copilot` in the search bar.\n\nA [recommended setup](https://github.com/intitni/CopilotForXcode/issues/14) that should cause no conflict is\n\n| Command | Key Binding |\n| ------------------- | ------------------------------------------------------ |\n| Accept Suggestions | `โŒฅ}` or Tab |\n| Dismiss Suggestions | Esc |\n| Reject Suggestion | `โŒฅ{` |\n| Next Suggestion | `โŒฅ>` |\n| Previous Suggestion | `โŒฅ<` |\n| Open Chat | `โŒฅ\"` |\n| Explain Selection | `โŒฅ\\|` |\n\nEssentially using `โŒฅโ‡ง` as the \"access\" key combination for all bindings.\n\nAnother convenient method to access commands is by using the `โ‡งโŒ˜/` shortcut to search for a command in the menu bar.\n\n#### Setting Up Global Hotkeys\n\nCurrently, there is only one global hotkey you can set to show/hide the widgets under the General tab from the host app.\n\nWhen this hotkey is not set to enabled globally, it will only work when the service app or Xcode is active.\n\n### Setting Up Suggestion Feature\n\n#### Setting Up GitHub Copilot\n\n1. In the host app, navigate to \"Service - GitHub Copilot\" to access your GitHub Copilot account settings.\n2. Click on \"Install\" to install the language server.\n3. Optionally, set up the path to Node. The default value is simply `node`. Copilot for Xcode.app will attempt to locate Node from the following directories: `/opt/homebrew/bin:/opt/homebrew/sbin:/usr/local/bin:/usr/bin:/bin:/usr/sbin:/sbin`.\n\n If your Node installation is located elsewhere, you can run `which node` from the terminal to obtain the correct path.\n\n If you are using a node version manager that provides a shim executable, you will need to find the path to the actual executable. Please refer to the FAQ for more information.\n\n4. Click on \"Sign In\", and you will be redirected to a verification website provided by GitHub. A user code will be copied to your clipboard.\n5. After signing in, return to the app and click on \"Confirm Sign-in\" to complete the process.\n6. Go to \"Feature - Suggestion\" and update the feature provider to \"GitHub Copilot\".\n\nThe installed language server is located at `~/Library/Application Support/com.intii.CopilotForXcode/GitHub Copilot/executable/`.\n\n#### Setting Up Codeium\n\n1. In the host app, navigate to \"Service - Codeium\" to access the Codeium account settings.\n2. Click on \"Install\" to install the language server.\n3. Click on \"Sign In\" and you will be redirected to codeium.com. After signing in, a token will be provided. You need to copy and paste this token back into the app to complete the sign-in process.\n4. Go to \"Feature - Suggestion\" and update the feature provider to \"Codeium\".\n\n> The key is stored in the keychain. When the helper app tries to access the key for the first time, it will prompt you to enter the password to access the keychain. Please select \"Always Allow\" to let the helper app access the key.\n\nThe installed language server is located at `~/Library/Application Support/com.intii.CopilotForXcode/Codeium/executable/`.\n\n### Setting Up Chat Feature\n\n1. In the host app, navigate to \"Service - Chat Model\".\n2. Update the OpenAI model or create a new one if necessary. Use the test button to verify the model.\n3. Optionally, set up the embedding model in \"Service - Embedding Model\", which is required for a subset of the chat feature.\n4. Go to \"Feature - Chat\" and update the chat/embedding feature provider with the one you just updated/created.\n\n### Managing `CopilotForXcodeExtensionService.app`\n\nThis app runs whenever you open `Copilot for Xcode.app` or `Xcode.app`. You can quit it with its menu bar item that looks like a tentacle.\n\nYou can also set it to quit automatically when the above 2 apps are closed.\n\n## Update\n\nYou can use the in-app updater or download the latest version manually from the latest [release](https://github.com/intitni/CopilotForXcode/releases).\n\nAfter updating, please open Copilot for Xcode.app once and restart Xcode to allow the extension to reload.\n\nIf you find that some of the features are no longer working, please first try regranting permissions to the app.\n\n## Feature\n\n> Files in gitignore will not receive suggestion. Both chat and prompt to code feature will not have access to them unless you manually select code from them.\n\n### Suggestion\n\nThe app can provide real-time code suggestions based on the files you have opened. It's powered by GitHub Copilot and Codeium.\n\nThe feature provides two presentation modes:\n\n- Nearby Text Cursor: This mode shows suggestions based on the position of the text cursor.\n- Floating Widget: This mode shows suggestions next to the indicator widget.\n\nWhen using the \"Nearby Text Cursor\" mode, it is recommended to set the real-time suggestion debounce to 0.1.\n\nIf you're working on a company project and don't want the suggestion feature to be triggered, you can globally disable it and choose to enable it only for specific projects.\n\nWhenever your code is updated, the app will automatically fetch suggestions for you, you can cancel this by pressing **Escape**.\n\n\\*: If a file is already open before the helper app launches, you will need to switch to those files in order to send the open file notification.\n\n#### Commands\n\n- Get Suggestions: Get suggestions for the editing file at the current cursor position.\n- Next Suggestion: If there is more than one suggestion, switch to the next one.\n- Previous Suggestion: If there is more than one suggestion, switch to the previous one.\n- Accept Suggestion: Add the suggestion to the code.\n- Reject Suggestion: Remove the suggestion comments.\n\nCommands called by the app:\n\n- Prepare for Real-time Suggestions: Call only by Copilot for Xcode. When suggestions are successfully fetched, Copilot for Xcode will run this command to present the suggestions.\n- Prefetch Suggestions: Call only by Copilot for Xcode. In the background, Copilot for Xcode will occasionally run this command to prefetch real-time suggestions.\n\n### Chat\n\nThis feature is powered by ChatGPT. Please ensure that you have set up your OpenAI account before using it.\n\nThe chat knows the following information:\n\n- The selected code in the active editor.\n- The relative path of the file.\n- The error and warning labels in the active editor.\n- The text cursor location.\n\nThere are currently two tabs in the chat panel: one is available shared across Xcode, and the other is only available in the current file.\n\nYou can detach the chat panel by simply dragging it away. Once detached, the chat panel will remain visible even if Xcode is inactive. To re-attach it to the widget, click the message bubble button located next to the indicator widget.\n\n#### Commands\n\n- Open Chat: Open a chat tab.\n\n#### Keyboard Shortcuts\n\n| Shortcut | Description |\n| :------: | --------------------------------------------------------------------------------------------------- |\n| `โŒ˜W` | Close the chat tab. |\n| `โŒ˜M` | Minimize the chat, you can bring it back with any chat commands or by clicking the indicator widget. |\n| `โ‡งโ†ฉ๏ธŽ` | Add new line. |\n| `โ‡งโŒ˜]` | Move to next tab |\n| `โ‡งโŒ˜[` | Move to previous tab |\n\n#### Chat Commands\n\nThe chat panel supports chat plugins that may not require an OpenAI API key. For example, if you need to use the `/shell` plugin, you just type\n\n```\n/run echo hello\n```\n\nIf you need to end a plugin, you can just type\n\n```\n/exit\n```\n\n| Command | Description |\n| :--------------------: | ----------------------------------------------------------------------------------------------------------------------------------------- |\n| `/shell` | Runs the command under the project root. |\n| | Environment variable:
- `PROJECT_ROOT` to get the project root.
- `FILE_PATH` to get the editing file path. |\n| `/shortcut(name)` | Run a shortcut from the Shortcuts.app, and use the following message as the input. |\n| | If the message is empty, it will use the previous message as input. The output of the shortcut will be printed as a reply from the bot. |\n\n### Prompt to Code\n\nRefactor existing code or write new code using natural language.\n\nThis feature is recommended when you need to update a specific piece of code. Some example use cases include:\n\n- Improving the code's readability.\n- Correcting bugs in the code.\n- Adding documentation to the code.\n- Breaking a large function into smaller functions.\n- Generating code with a specific template through custom commands.\n- Polishing and correcting grammar and spelling errors in the documentation.\n- Translating a localizable strings file.\n\n#### Commands\n\n- Write or Edit Code: Open a modification window, where you can use natural language to write or edit selected code.\n- Accept Modification: Accept the result of modification.\n\n### Custom Commands\n\nYou can create custom commands that run Chat and Modification with personalized prompts. These commands are easily accessible from both the Xcode menu bar and the context menu of the indicator widget. There are 3 types of custom commands:\n\n- Modification: Run Modification with the selected code, and update or write the code using the given prompt, if provided. You can provide additional information through the extra system prompt field.\n- Send Message: Open the chat window and immediately send a message, if provided. You can provide more information through the extra system prompt field.\n- Custom Chat: Open the chat window and immediately send a message, if provided. You can overwrite the entire system prompt through the system prompt field.\n- Single Round Dialog: Send a message to a temporary chat. Useful when you want to run a terminal command with `/run`.\n\nYou can use the following template arguments in custom commands:\n\n| Argument | Description |\n| ----------------------------- | ---------------------------------------------- |\n| `{{selected_code}}` | The currently selected code in the editor. |\n| `{{active_editor_language}}` | The programming language of the active editor. |\n| `{{active_editor_file_url}}` | The URL of the active file in the editor. |\n| `{{active_editor_file_name}}` | The name of the active file in the editor. |\n| `{{clipboard}}` | The content in clipboard. |\n\n## Limitations\n\n- The extension utilizes various tricks to monitor the state of Xcode. It may fail, it may be incorrect, especially when you have multiple Xcode windows running, and maybe even worse when they are in different displays. I am not sure about that though.\n\n## License\n\nPlease check [LICENSE](LICENSE) for details.\n\n" }, { "path": "SandboxedClientTester/Assets.xcassets/AccentColor.colorset/Contents.json", "content": "{\n \"colors\" : [\n {\n \"idiom\" : \"universal\"\n }\n ],\n \"info\" : {\n \"author\" : \"xcode\",\n \"version\" : 1\n }\n}\n" }, { "path": "SandboxedClientTester/Assets.xcassets/AppIcon.appiconset/Contents.json", "content": "{\n \"images\" : [\n {\n \"idiom\" : \"mac\",\n \"scale\" : \"1x\",\n \"size\" : \"16x16\"\n },\n {\n \"idiom\" : \"mac\",\n \"scale\" : \"2x\",\n \"size\" : \"16x16\"\n },\n {\n \"idiom\" : \"mac\",\n \"scale\" : \"1x\",\n \"size\" : \"32x32\"\n },\n {\n \"idiom\" : \"mac\",\n \"scale\" : \"2x\",\n \"size\" : \"32x32\"\n },\n {\n \"idiom\" : \"mac\",\n \"scale\" : \"1x\",\n \"size\" : \"128x128\"\n },\n {\n \"idiom\" : \"mac\",\n \"scale\" : \"2x\",\n \"size\" : \"128x128\"\n },\n {\n \"idiom\" : \"mac\",\n \"scale\" : \"1x\",\n \"size\" : \"256x256\"\n },\n {\n \"idiom\" : \"mac\",\n \"scale\" : \"2x\",\n \"size\" : \"256x256\"\n },\n {\n \"idiom\" : \"mac\",\n \"scale\" : \"1x\",\n \"size\" : \"512x512\"\n },\n {\n \"idiom\" : \"mac\",\n \"scale\" : \"2x\",\n \"size\" : \"512x512\"\n }\n ],\n \"info\" : {\n \"author\" : \"xcode\",\n \"version\" : 1\n }\n}\n" }, { "path": "SandboxedClientTester/Assets.xcassets/Contents.json", "content": "{\n \"info\" : {\n \"author\" : \"xcode\",\n \"version\" : 1\n }\n}\n" }, { "path": "SandboxedClientTester/ContentView.swift", "content": "import SwiftUI\nimport Client\n\nstruct ContentView: View {\n @State var text: String = \"Hello, world!\"\n var body: some View {\n VStack {\n Button(action: {\n Task {\n do {\n let service = try getService()\n let version = try await service.getXPCServiceVersion()\n text = \"Version: \\(version.version) Build: \\(version.build)\"\n } catch {\n text = error.localizedDescription\n }\n }\n }) {\n Text(\"Test\")\n }\n Text(text)\n }\n .padding()\n }\n}\n\n#Preview {\n ContentView()\n}\n" }, { "path": "SandboxedClientTester/Info.plist", "content": "\n\n\n\n\tBUNDLE_IDENTIFIER_BASE\n\t$(BUNDLE_IDENTIFIER_BASE)\n\n\n" }, { "path": "SandboxedClientTester/Preview Content/Preview Assets.xcassets/Contents.json", "content": "{\n \"info\" : {\n \"author\" : \"xcode\",\n \"version\" : 1\n }\n}\n" }, { "path": "SandboxedClientTester/SandboxedClientTester.entitlements", "content": "\n\n\n\n\tcom.apple.security.app-sandbox\n\t\n\tcom.apple.security.files.user-selected.read-only\n\t\n\tcom.apple.security.temporary-exception.mach-lookup.global-name\n\t\n\t\t$(BUNDLE_IDENTIFIER_BASE).CommunicationBridge\n\t\n\n\n" }, { "path": "SandboxedClientTester/SandboxedClientTesterApp.swift", "content": "import SwiftUI\n\n@main\nstruct SandboxedClientTesterApp: App {\n var body: some Scene {\n WindowGroup {\n ContentView()\n }\n }\n}\n" }, { "path": "TestPlan.xctestplan", "content": "{\n \"configurations\" : [\n {\n \"id\" : \"586480F5-DC84-425D-814F-7A5F569A1974\",\n \"name\" : \"Configuration 1\",\n \"options\" : {\n\n }\n }\n ],\n \"defaultOptions\" : {\n \"environmentVariableEntries\" : [\n {\n \"key\" : \"IS_UNIT_TEST\",\n \"value\" : \"YES\"\n },\n {\n \"key\" : \"SUEnableAutomaticChecks\",\n \"value\" : \"NO\"\n }\n ],\n \"testTimeoutsEnabled\" : true\n },\n \"testTargets\" : [\n {\n \"target\" : {\n \"containerPath\" : \"container:Tool\",\n \"identifier\" : \"SharedUIComponentsTests\",\n \"name\" : \"SharedUIComponentsTests\"\n }\n },\n {\n \"target\" : {\n \"containerPath\" : \"container:Tool\",\n \"identifier\" : \"ActiveDocumentChatContextCollectorTests\",\n \"name\" : \"ActiveDocumentChatContextCollectorTests\"\n }\n },\n {\n \"target\" : {\n \"containerPath\" : \"container:Tool\",\n \"identifier\" : \"CodeDiffTests\",\n \"name\" : \"CodeDiffTests\"\n }\n },\n {\n \"target\" : {\n \"containerPath\" : \"container:Core\",\n \"identifier\" : \"ServiceUpdateMigrationTests\",\n \"name\" : \"ServiceUpdateMigrationTests\"\n }\n },\n {\n \"target\" : {\n \"containerPath\" : \"container:Tool\",\n \"identifier\" : \"ASTParserTests\",\n \"name\" : \"ASTParserTests\"\n }\n },\n {\n \"target\" : {\n \"containerPath\" : \"container:Tool\",\n \"identifier\" : \"SuggestionProviderTests\",\n \"name\" : \"SuggestionProviderTests\"\n }\n },\n {\n \"target\" : {\n \"containerPath\" : \"container:Core\",\n \"identifier\" : \"PromptToCodeServiceTests\",\n \"name\" : \"PromptToCodeServiceTests\"\n }\n },\n {\n \"target\" : {\n \"containerPath\" : \"container:Tool\",\n \"identifier\" : \"KeychainTests\",\n \"name\" : \"KeychainTests\"\n }\n },\n {\n \"target\" : {\n \"containerPath\" : \"container:Tool\",\n \"identifier\" : \"JoinJSONTests\",\n \"name\" : \"JoinJSONTests\"\n }\n },\n {\n \"target\" : {\n \"containerPath\" : \"container:Core\",\n \"identifier\" : \"ServiceTests\",\n \"name\" : \"ServiceTests\"\n }\n },\n {\n \"target\" : {\n \"containerPath\" : \"container:OverlayWindow\",\n \"identifier\" : \"OverlayWindowTests\",\n \"name\" : \"OverlayWindowTests\"\n }\n },\n {\n \"target\" : {\n \"containerPath\" : \"container:Tool\",\n \"identifier\" : \"SuggestionBasicTests\",\n \"name\" : \"SuggestionBasicTests\"\n }\n },\n {\n \"target\" : {\n \"containerPath\" : \"container:Tool\",\n \"identifier\" : \"LangChainTests\",\n \"name\" : \"LangChainTests\"\n }\n },\n {\n \"target\" : {\n \"containerPath\" : \"container:Tool\",\n \"identifier\" : \"GitHubCopilotServiceTests\",\n \"name\" : \"GitHubCopilotServiceTests\"\n }\n },\n {\n \"target\" : {\n \"containerPath\" : \"container:Tool\",\n \"identifier\" : \"OpenAIServiceTests\",\n \"name\" : \"OpenAIServiceTests\"\n }\n },\n {\n \"target\" : {\n \"containerPath\" : \"container:Core\",\n \"identifier\" : \"SuggestionWidgetTests\",\n \"name\" : \"SuggestionWidgetTests\"\n }\n },\n {\n \"target\" : {\n \"containerPath\" : \"container:Core\",\n \"identifier\" : \"KeyBindingManagerTests\",\n \"name\" : \"KeyBindingManagerTests\"\n }\n },\n {\n \"target\" : {\n \"containerPath\" : \"container:Tool\",\n \"identifier\" : \"FocusedCodeFinderTests\",\n \"name\" : \"FocusedCodeFinderTests\"\n }\n },\n {\n \"target\" : {\n \"containerPath\" : \"container:Tool\",\n \"identifier\" : \"WebSearchServiceTests\",\n \"name\" : \"WebSearchServiceTests\"\n }\n },\n {\n \"target\" : {\n \"containerPath\" : \"container:Tool\",\n \"identifier\" : \"XcodeInspectorTests\",\n \"name\" : \"XcodeInspectorTests\"\n }\n },\n {\n \"target\" : {\n \"containerPath\" : \"container:Core\",\n \"identifier\" : \"ChatServiceTests\",\n \"name\" : \"ChatServiceTests\"\n }\n },\n {\n \"target\" : {\n \"containerPath\" : \"container:Tool\",\n \"identifier\" : \"SuggestionInjectorTests\",\n \"name\" : \"SuggestionInjectorTests\"\n }\n },\n {\n \"target\" : {\n \"containerPath\" : \"container:Tool\",\n \"identifier\" : \"TokenEncoderTests\",\n \"name\" : \"TokenEncoderTests\"\n }\n }\n ],\n \"version\" : 1\n}\n" }, { "path": "Tool/.gitignore", "content": ".DS_Store\n/.build\n/Packages\n/*.xcodeproj\nxcuserdata/\nDerivedData/\n.swiftpm/config/registries.json\n.swiftpm/xcode/package.xcworkspace/contents.xcworkspacedata\n.netrc\n" }, { "path": "Tool/.swiftpm/xcode/xcshareddata/xcschemes/SuggestionModel.xcscheme", "content": "\n\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n\n" }, { "path": "Tool/Package.resolved", "content": "{\n \"pins\" : [\n {\n \"identity\" : \"codablewrappers\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/GottaGetSwifty/CodableWrappers\",\n \"state\" : {\n \"revision\" : \"4eb46a4c656333e8514db8aad204445741de7d40\",\n \"version\" : \"2.0.7\"\n }\n },\n {\n \"identity\" : \"combine-schedulers\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/combine-schedulers\",\n \"state\" : {\n \"revision\" : \"ec62f32d21584214a4b27c8cee2b2ad70ab2c38a\",\n \"version\" : \"0.11.0\"\n }\n },\n {\n \"identity\" : \"fseventswrapper\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/Frizlab/FSEventsWrapper\",\n \"state\" : {\n \"revision\" : \"e0c59a2ce2775e5f6642da6d19207445f10112d0\",\n \"version\" : \"1.0.2\"\n }\n },\n {\n \"identity\" : \"glob\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/Bouke/Glob\",\n \"state\" : {\n \"revision\" : \"deda6e163d2ff2a8d7e138e2c3326dbd71157faf\",\n \"version\" : \"1.0.5\"\n }\n },\n {\n \"identity\" : \"highlightr\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/intitni/Highlightr\",\n \"state\" : {\n \"branch\" : \"bump-highlight-js-version\",\n \"revision\" : \"4ffbb1b0b721378263297cafea6f2838044eb1eb\"\n }\n },\n {\n \"identity\" : \"jsonrpc\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/ChimeHQ/JSONRPC\",\n \"state\" : {\n \"revision\" : \"5da978702aece6ba5c7879b0d253c180d61e4ef3\",\n \"version\" : \"0.6.0\"\n }\n },\n {\n \"identity\" : \"languageclient\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/ChimeHQ/LanguageClient\",\n \"state\" : {\n \"revision\" : \"f0198ee0a102d266078f7d9c28f086f2989f988a\",\n \"version\" : \"0.3.1\"\n }\n },\n {\n \"identity\" : \"languageserverprotocol\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/ChimeHQ/LanguageServerProtocol\",\n \"state\" : {\n \"revision\" : \"6e97f943dc024307c5524a80bd33cdbd1cc621de\",\n \"version\" : \"0.8.0\"\n }\n },\n {\n \"identity\" : \"operationplus\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/ChimeHQ/OperationPlus\",\n \"state\" : {\n \"revision\" : \"1340f95dce3e93d742497d88db18f8676f4badf4\",\n \"version\" : \"1.6.0\"\n }\n },\n {\n \"identity\" : \"processenv\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/ChimeHQ/ProcessEnv\",\n \"state\" : {\n \"revision\" : \"29487b6581bb785c372c611c943541ef4309d051\",\n \"version\" : \"0.3.1\"\n }\n },\n {\n \"identity\" : \"swift-async-algorithms\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/apple/swift-async-algorithms\",\n \"state\" : {\n \"revision\" : \"9cfed92b026c524674ed869a4ff2dcfdeedf8a2a\",\n \"version\" : \"0.1.0\"\n }\n },\n {\n \"identity\" : \"swift-case-paths\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-case-paths\",\n \"state\" : {\n \"revision\" : \"fc45e7b2cfece9dd80b5a45e6469ffe67fe67984\",\n \"version\" : \"0.14.1\"\n }\n },\n {\n \"identity\" : \"swift-clocks\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-clocks\",\n \"state\" : {\n \"revision\" : \"0fbaebfc013715dab44d715a4d350ba37f297e4d\",\n \"version\" : \"0.4.0\"\n }\n },\n {\n \"identity\" : \"swift-collections\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/apple/swift-collections.git\",\n \"state\" : {\n \"revision\" : \"937e904258d22af6e447a0b72c0bc67583ef64a2\",\n \"version\" : \"1.0.4\"\n }\n },\n {\n \"identity\" : \"swift-composable-architecture\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-composable-architecture\",\n \"state\" : {\n \"revision\" : \"9f4202ab5b8422aa90f0ed983bf7652c3af7abf0\",\n \"version\" : \"0.59.0\"\n }\n },\n {\n \"identity\" : \"swift-concurrency-extras\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-concurrency-extras\",\n \"state\" : {\n \"revision\" : \"479750bd98fac2e813fffcf2af0728b5b0085795\",\n \"version\" : \"0.1.1\"\n }\n },\n {\n \"identity\" : \"swift-custom-dump\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-custom-dump\",\n \"state\" : {\n \"revision\" : \"4a87bb75be70c983a9548597e8783236feb3401e\",\n \"version\" : \"0.11.1\"\n }\n },\n {\n \"identity\" : \"swift-dependencies\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-dependencies\",\n \"state\" : {\n \"revision\" : \"16fd42ae04c6e7f74a6a86395d04722c641cccee\",\n \"version\" : \"0.6.0\"\n }\n },\n {\n \"identity\" : \"swift-identified-collections\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-identified-collections\",\n \"state\" : {\n \"revision\" : \"d01446a78fb768adc9a78cbb6df07767c8ccfc29\",\n \"version\" : \"0.8.0\"\n }\n },\n {\n \"identity\" : \"swift-parsing\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swift-parsing\",\n \"state\" : {\n \"revision\" : \"27c941bbd22a4bbc53005a15a0440443fd892f70\",\n \"version\" : \"0.12.1\"\n }\n },\n {\n \"identity\" : \"swift-syntax\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/apple/swift-syntax.git\",\n \"state\" : {\n \"branch\" : \"main\",\n \"revision\" : \"e149b01cfd3e96240e102729697e2095c19157ef\"\n }\n },\n {\n \"identity\" : \"swiftsoup\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/scinfu/SwiftSoup.git\",\n \"state\" : {\n \"revision\" : \"8b6cf29eead8841a1fa7822481cb3af4ddaadba6\",\n \"version\" : \"2.6.1\"\n }\n },\n {\n \"identity\" : \"swifttreesitter\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/ChimeHQ/SwiftTreeSitter\",\n \"state\" : {\n \"revision\" : \"a9b1335d5151b62b11f07599bd07d07dc5965de3\",\n \"version\" : \"0.7.2\"\n }\n },\n {\n \"identity\" : \"swiftui-navigation\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/swiftui-navigation\",\n \"state\" : {\n \"revision\" : \"2aa885e719087ee19df251c08a5980ad3e787f12\",\n \"version\" : \"0.8.0\"\n }\n },\n {\n \"identity\" : \"tiktoken\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/intitni/Tiktoken\",\n \"state\" : {\n \"branch\" : \"main\",\n \"revision\" : \"6a2ac324c6daec167ca95268d5a487e6de6a1cea\"\n }\n },\n {\n \"identity\" : \"tree-sitter-objc\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/lukepistrol/tree-sitter-objc\",\n \"state\" : {\n \"branch\" : \"feature/spm\",\n \"revision\" : \"1b54ef0b5efddddf393b45e173788499cc572048\"\n }\n },\n {\n \"identity\" : \"tree-sitter-swift\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/alex-pinkus/tree-sitter-swift\",\n \"state\" : {\n \"branch\" : \"with-generated-files\",\n \"revision\" : \"eda05af7ac41adb4eb19c346883c0fa32fe3bdd8\"\n }\n },\n {\n \"identity\" : \"usearch\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/unum-cloud/usearch\",\n \"state\" : {\n \"revision\" : \"33c53288b44ccb55de77776820676132a6e4c42a\",\n \"version\" : \"0.23.0\"\n }\n },\n {\n \"identity\" : \"xctest-dynamic-overlay\",\n \"kind\" : \"remoteSourceControl\",\n \"location\" : \"https://github.com/pointfreeco/xctest-dynamic-overlay\",\n \"state\" : {\n \"revision\" : \"50843cbb8551db836adec2290bb4bc6bac5c1865\",\n \"version\" : \"0.9.0\"\n }\n }\n ],\n \"version\" : 2\n}\n" }, { "path": "Tool/Package.swift", "content": "// swift-tools-version: 5.7\n// The swift-tools-version declares the minimum version of Swift required to build this package.\n\nimport PackageDescription\n\nlet package = Package(\n name: \"Tool\",\n platforms: [.macOS(.v12)],\n products: [\n .library(name: \"XPCShared\", targets: [\"XPCShared\"]),\n .library(name: \"Terminal\", targets: [\"Terminal\"]),\n .library(name: \"LangChain\", targets: [\"LangChain\"]),\n .library(name: \"ExternalServices\", targets: [\"WebSearchService\"]),\n .library(name: \"Preferences\", targets: [\"Preferences\", \"Configs\"]),\n .library(name: \"Logger\", targets: [\"Logger\"]),\n .library(name: \"OpenAIService\", targets: [\"OpenAIService\"]),\n .library(name: \"ChatTab\", targets: [\"ChatTab\"]),\n .library(name: \"FileSystem\", targets: [\"FileSystem\"]),\n .library(\n name: \"ChatContextCollector\",\n targets: [\"ChatContextCollector\", \"ActiveDocumentChatContextCollector\"]\n ),\n .library(name: \"SuggestionBasic\", targets: [\"SuggestionBasic\", \"SuggestionInjector\"]),\n .library(name: \"PromptToCode\", targets: [\"ModificationBasic\", \"PromptToCodeCustomization\"]),\n .library(name: \"Chat\", targets: [\"ChatBasic\"]),\n .library(name: \"ASTParser\", targets: [\"ASTParser\"]),\n .library(name: \"FocusedCodeFinder\", targets: [\"FocusedCodeFinder\"]),\n .library(name: \"Toast\", targets: [\"Toast\"]),\n .library(name: \"Keychain\", targets: [\"Keychain\"]),\n .library(name: \"SharedUIComponents\", targets: [\"SharedUIComponents\"]),\n .library(name: \"UserDefaultsObserver\", targets: [\"UserDefaultsObserver\"]),\n .library(name: \"Workspace\", targets: [\"Workspace\"]),\n .library(name: \"WorkspaceSuggestionService\", targets: [\"WorkspaceSuggestionService\"]),\n .library(\n name: \"SuggestionProvider\",\n targets: [\"SuggestionProvider\", \"GitHubCopilotService\", \"CodeiumService\"]\n ),\n .library(\n name: \"AppMonitoring\",\n targets: [\n \"XcodeInspector\",\n \"ActiveApplicationMonitor\",\n \"AXExtension\",\n \"AXNotificationStream\",\n \"AppActivator\",\n ]\n ),\n .library(name: \"GitIgnoreCheck\", targets: [\"GitIgnoreCheck\"]),\n .library(name: \"DebounceFunction\", targets: [\"DebounceFunction\"]),\n .library(name: \"AsyncPassthroughSubject\", targets: [\"AsyncPassthroughSubject\"]),\n .library(name: \"CustomAsyncAlgorithms\", targets: [\"CustomAsyncAlgorithms\"]),\n .library(name: \"CommandHandler\", targets: [\"CommandHandler\"]),\n .library(name: \"CodeDiff\", targets: [\"CodeDiff\"]),\n .library(name: \"BuiltinExtension\", targets: [\"BuiltinExtension\"]),\n .library(name: \"WebSearchService\", targets: [\"WebSearchService\"]),\n .library(name: \"WebScrapper\", targets: [\"WebScrapper\"]),\n .library(\n name: \"CustomCommandTemplateProcessor\",\n targets: [\"CustomCommandTemplateProcessor\"]\n ),\n ],\n dependencies: [\n // A fork of https://github.com/aespinilla/Tiktoken to allow loading from local files.\n .package(url: \"https://github.com/intitni/Tiktoken\", branch: \"main\"),\n // TODO: Update LanguageClient some day.\n .package(url: \"https://github.com/ChimeHQ/LanguageClient\", exact: \"0.3.1\"),\n .package(url: \"https://github.com/ChimeHQ/LanguageServerProtocol\", exact: \"0.8.0\"),\n .package(url: \"https://github.com/apple/swift-async-algorithms\", from: \"1.0.0\"),\n .package(url: \"https://github.com/pointfreeco/swift-parsing\", from: \"0.12.1\"),\n .package(url: \"https://github.com/ChimeHQ/JSONRPC\", exact: \"0.6.0\"),\n .package(url: \"https://github.com/scinfu/SwiftSoup.git\", from: \"2.6.0\"),\n .package(url: \"https://github.com/unum-cloud/usearch\", from: \"0.19.1\"),\n .package(url: \"https://github.com/intitni/Highlightr\", branch: \"master\"),\n .package(\n url: \"https://github.com/pointfreeco/swift-composable-architecture\",\n exact: \"1.16.1\"\n ),\n .package(url: \"https://github.com/apple/swift-syntax.git\", from: \"600.0.0\"),\n .package(url: \"https://github.com/GottaGetSwifty/CodableWrappers\", from: \"2.0.7\"),\n // A fork of https://github.com/google/generative-ai-swift to support setting base url.\n .package(\n url: \"https://github.com/intitni/generative-ai-swift\",\n branch: \"support-setting-base-url\"\n ),\n .package(\n url: \"https://github.com/intitni/CopilotForXcodeKit\",\n branch: \"feature/custom-chat-tab\"\n ),\n\n // TreeSitter\n .package(url: \"https://github.com/intitni/SwiftTreeSitter.git\", branch: \"main\"),\n .package(url: \"https://github.com/lukepistrol/tree-sitter-objc\", branch: \"feature/spm\"),\n ],\n targets: [\n // MARK: - Helpers\n\n .target(name: \"XPCShared\", dependencies: [\"SuggestionBasic\", \"Logger\"]),\n\n .target(name: \"Configs\"),\n\n .target(name: \"Preferences\", dependencies: [\"Configs\", \"AIModel\"]),\n\n .target(name: \"Terminal\"),\n\n .target(name: \"Logger\"),\n\n .target(name: \"FileSystem\"),\n\n .target(name: \"ObjectiveCExceptionHandling\"),\n\n .target(name: \"JoinJSON\"),\n .testTarget(name: \"JoinJSONTests\", dependencies: [\"JoinJSON\"]),\n\n .target(name: \"CodeDiff\", dependencies: [\"SuggestionBasic\"]),\n .testTarget(name: \"CodeDiffTests\", dependencies: [\"CodeDiff\"]),\n\n .target(\n name: \"CustomAsyncAlgorithms\",\n dependencies: [\n .product(name: \"AsyncAlgorithms\", package: \"swift-async-algorithms\"),\n ]\n ),\n\n .target(\n name: \"Keychain\",\n dependencies: [\"Configs\", \"Preferences\"]\n ),\n\n .testTarget(\n name: \"KeychainTests\",\n dependencies: [\"Keychain\"]\n ),\n\n .target(\n name: \"Toast\",\n dependencies: [.product(\n name: \"ComposableArchitecture\",\n package: \"swift-composable-architecture\"\n )]\n ),\n\n .target(\n name: \"CustomCommandTemplateProcessor\",\n dependencies: [\n \"XcodeInspector\",\n \"SuggestionBasic\",\n ]\n ),\n\n .target(name: \"DebounceFunction\"),\n\n .target(\n name: \"AppActivator\",\n dependencies: [\n \"XcodeInspector\",\n .product(name: \"ComposableArchitecture\", package: \"swift-composable-architecture\"),\n ]\n ),\n\n .target(name: \"ActiveApplicationMonitor\"),\n\n .target(name: \"USearchIndex\", dependencies: [\n \"ObjectiveCExceptionHandling\",\n .product(name: \"USearch\", package: \"usearch\"),\n ]),\n\n .target(\n name: \"TokenEncoder\",\n dependencies: [\n .product(name: \"Tiktoken\", package: \"Tiktoken\"),\n .product(name: \"GoogleGenerativeAI\", package: \"generative-ai-swift\"),\n ],\n resources: [\n .copy(\"Resources/cl100k_base.tiktoken\"),\n ]\n ),\n .testTarget(\n name: \"TokenEncoderTests\",\n dependencies: [\"TokenEncoder\"]\n ),\n\n .target(\n name: \"SuggestionBasic\",\n dependencies: [\n \"LanguageClient\",\n .product(name: \"Parsing\", package: \"swift-parsing\"),\n .product(name: \"CodableWrappers\", package: \"CodableWrappers\"),\n ]\n ),\n\n .target(\n name: \"SuggestionInjector\",\n dependencies: [\"SuggestionBasic\"]\n ),\n .testTarget(\n name: \"SuggestionInjectorTests\",\n dependencies: [\"SuggestionInjector\"]\n ),\n\n .target(\n name: \"AIModel\",\n dependencies: [\n .product(name: \"CodableWrappers\", package: \"CodableWrappers\"),\n ]\n ),\n\n .testTarget(\n name: \"SuggestionBasicTests\",\n dependencies: [\"SuggestionBasic\"]\n ),\n\n .target(\n name: \"ChatBasic\",\n dependencies: [\n \"AIModel\",\n \"Preferences\",\n \"Keychain\",\n .product(name: \"CodableWrappers\", package: \"CodableWrappers\"),\n ]\n ),\n\n .target(\n name: \"ModificationBasic\",\n dependencies: [\n \"SuggestionBasic\",\n \"ChatBasic\",\n .product(name: \"CodableWrappers\", package: \"CodableWrappers\"),\n .product(\n name: \"ComposableArchitecture\",\n package: \"swift-composable-architecture\"\n ),\n ]\n ),\n .testTarget(name: \"ModificationBasicTests\", dependencies: [\"ModificationBasic\"]),\n\n .target(\n name: \"PromptToCodeCustomization\",\n dependencies: [\n \"ModificationBasic\",\n \"SuggestionBasic\",\n \"ChatBasic\",\n .product(\n name: \"ComposableArchitecture\",\n package: \"swift-composable-architecture\"\n ),\n ]\n ),\n\n .target(\n name: \"AXExtension\",\n dependencies: [\"Logger\"]\n ),\n\n .target(\n name: \"AXNotificationStream\",\n dependencies: [\n \"Preferences\",\n \"Logger\",\n ]\n ),\n\n .target(\n name: \"XcodeInspector\",\n dependencies: [\n \"AXExtension\",\n \"SuggestionBasic\",\n \"AXNotificationStream\",\n \"Logger\",\n \"Toast\",\n \"Preferences\",\n \"AsyncPassthroughSubject\",\n .product(name: \"AsyncAlgorithms\", package: \"swift-async-algorithms\"),\n ]\n ),\n\n .testTarget(name: \"XcodeInspectorTests\", dependencies: [\"XcodeInspector\"]),\n\n .target(name: \"UserDefaultsObserver\"),\n\n .target(name: \"AsyncPassthroughSubject\"),\n\n .target(\n name: \"BuiltinExtension\",\n dependencies: [\n \"SuggestionBasic\",\n \"SuggestionProvider\",\n \"ChatBasic\",\n \"Workspace\",\n \"ChatTab\",\n \"AIModel\",\n .product(name: \"CopilotForXcodeKit\", package: \"CopilotForXcodeKit\"),\n ]\n ),\n\n .target(\n name: \"SharedUIComponents\",\n dependencies: [\n \"Highlightr\",\n \"Preferences\",\n \"SuggestionBasic\",\n \"DebounceFunction\",\n \"CodeDiff\",\n .product(name: \"ComposableArchitecture\", package: \"swift-composable-architecture\"),\n ]\n ),\n .testTarget(name: \"SharedUIComponentsTests\", dependencies: [\"SharedUIComponents\"]),\n\n .target(name: \"ASTParser\", dependencies: [\n \"SuggestionBasic\",\n .product(name: \"SwiftTreeSitter\", package: \"SwiftTreeSitter\"),\n .product(name: \"TreeSitterObjC\", package: \"tree-sitter-objc\"),\n ]),\n\n .testTarget(name: \"ASTParserTests\", dependencies: [\"ASTParser\"]),\n\n .target(\n name: \"Workspace\",\n dependencies: [\n \"GitIgnoreCheck\",\n \"UserDefaultsObserver\",\n \"SuggestionBasic\",\n \"Logger\",\n \"Preferences\",\n \"XcodeInspector\",\n ]\n ),\n\n .target(\n name: \"WorkspaceSuggestionService\",\n dependencies: [\n \"Workspace\",\n \"SuggestionProvider\",\n \"XPCShared\",\n \"BuiltinExtension\",\n \"SuggestionInjector\",\n ]\n ),\n\n .target(\n name: \"FocusedCodeFinder\",\n dependencies: [\n \"Preferences\",\n \"ASTParser\",\n \"SuggestionBasic\",\n .product(name: \"SwiftSyntax\", package: \"swift-syntax\"),\n .product(name: \"SwiftParser\", package: \"swift-syntax\"),\n ]\n ),\n .testTarget(\n name: \"FocusedCodeFinderTests\",\n dependencies: [\"FocusedCodeFinder\"]\n ),\n\n .target(\n name: \"GitIgnoreCheck\",\n dependencies: [\n \"Terminal\",\n \"Preferences\",\n .product(\n name: \"ComposableArchitecture\",\n package: \"swift-composable-architecture\"\n ),\n ]\n ),\n\n .target(\n name: \"CommandHandler\",\n dependencies: [\n \"XcodeInspector\",\n \"Preferences\",\n \"ChatBasic\",\n \"ModificationBasic\",\n .product(name: \"ComposableArchitecture\", package: \"swift-composable-architecture\"),\n ]\n ),\n\n // MARK: - Services\n\n .target(\n name: \"LangChain\",\n dependencies: [\n \"OpenAIService\",\n \"ObjectiveCExceptionHandling\",\n \"USearchIndex\",\n \"ChatBasic\",\n .product(name: \"JSONRPC\", package: \"JSONRPC\"),\n .product(name: \"Parsing\", package: \"swift-parsing\"),\n .product(name: \"SwiftSoup\", package: \"SwiftSoup\"),\n ]\n ),\n\n .target(name: \"WebScrapper\", dependencies: [\n .product(name: \"SwiftSoup\", package: \"SwiftSoup\"),\n ]),\n\n .target(name: \"WebSearchService\", dependencies: [\"Preferences\", \"WebScrapper\", \"Keychain\"]),\n .testTarget(name: \"WebSearchServiceTests\", dependencies: [\"WebSearchService\"]),\n\n .target(name: \"SuggestionProvider\", dependencies: [\n \"SuggestionBasic\",\n \"UserDefaultsObserver\",\n \"Preferences\",\n \"Logger\",\n .product(name: \"CopilotForXcodeKit\", package: \"CopilotForXcodeKit\"),\n ]),\n\n .testTarget(name: \"SuggestionProviderTests\", dependencies: [\"SuggestionProvider\"]),\n\n .target(\n name: \"RAGChatAgent\",\n dependencies: [\n \"ChatBasic\",\n \"ChatContextCollector\",\n \"OpenAIService\",\n \"Preferences\",\n ]\n ),\n\n // MARK: - GitHub Copilot\n\n .target(\n name: \"GitHubCopilotService\",\n dependencies: [\n \"LanguageClient\",\n \"SuggestionBasic\",\n \"ChatBasic\",\n \"Logger\",\n \"Preferences\",\n \"Terminal\",\n \"BuiltinExtension\",\n \"Toast\",\n \"SuggestionProvider\",\n .product(name: \"JSONRPC\", package: \"JSONRPC\"),\n .product(name: \"LanguageServerProtocol\", package: \"LanguageServerProtocol\"),\n .product(name: \"CopilotForXcodeKit\", package: \"CopilotForXcodeKit\"),\n ],\n resources: [.copy(\"Resources/load-self-signed-cert-1.34.0.js\")]\n ),\n .testTarget(\n name: \"GitHubCopilotServiceTests\",\n dependencies: [\"GitHubCopilotService\"]\n ),\n\n // MARK: - Codeium\n\n .target(\n name: \"CodeiumService\",\n dependencies: [\n \"LanguageClient\",\n \"Keychain\",\n \"SuggestionBasic\",\n \"Preferences\",\n \"Terminal\",\n \"XcodeInspector\",\n \"BuiltinExtension\",\n \"ChatTab\",\n \"SharedUIComponents\",\n .product(name: \"JSONRPC\", package: \"JSONRPC\"),\n .product(name: \"CopilotForXcodeKit\", package: \"CopilotForXcodeKit\"),\n ]\n ),\n\n // MARK: - OpenAI\n\n .target(\n name: \"OpenAIService\",\n dependencies: [\n \"Logger\",\n \"Preferences\",\n \"TokenEncoder\",\n \"Keychain\",\n \"BuiltinExtension\",\n \"ChatBasic\",\n \"GitHubCopilotService\",\n \"JoinJSON\",\n .product(name: \"JSONRPC\", package: \"JSONRPC\"),\n .product(name: \"AsyncAlgorithms\", package: \"swift-async-algorithms\"),\n .product(name: \"GoogleGenerativeAI\", package: \"generative-ai-swift\"),\n .product(\n name: \"ComposableArchitecture\",\n package: \"swift-composable-architecture\"\n ),\n ]\n ),\n .testTarget(\n name: \"OpenAIServiceTests\",\n dependencies: [\n \"OpenAIService\",\n \"ChatBasic\",\n .product(\n name: \"ComposableArchitecture\",\n package: \"swift-composable-architecture\"\n ),\n ]\n ),\n\n // MARK: - UI\n\n .target(\n name: \"ChatTab\",\n dependencies: [\n \"Preferences\",\n \"Configs\",\n \"AIModel\",\n .product(name: \"CodableWrappers\", package: \"CodableWrappers\"),\n .product(\n name: \"ComposableArchitecture\",\n package: \"swift-composable-architecture\"\n ),\n ]\n ),\n\n // MARK: - Chat Context Collector\n\n .target(\n name: \"ChatContextCollector\",\n dependencies: [\n \"SuggestionBasic\",\n \"ChatBasic\",\n \"OpenAIService\",\n ]\n ),\n\n .target(\n name: \"ActiveDocumentChatContextCollector\",\n dependencies: [\n \"ASTParser\",\n \"ChatContextCollector\",\n \"OpenAIService\",\n \"Preferences\",\n \"FocusedCodeFinder\",\n \"XcodeInspector\",\n \"GitIgnoreCheck\",\n ],\n path: \"Sources/ChatContextCollectors/ActiveDocumentChatContextCollector\"\n ),\n\n .testTarget(\n name: \"ActiveDocumentChatContextCollectorTests\",\n dependencies: [\"ActiveDocumentChatContextCollector\"]\n ),\n\n // MARK: - Tests\n\n .testTarget(\n name: \"LangChainTests\",\n dependencies: [\"LangChain\"]\n ),\n ]\n)\n\n" }, { "path": "Tool/README.md", "content": "# Tool\n\nA description of this package.\n" }, { "path": "Tool/Sources/AIModel/ChatModel.swift", "content": "import CodableWrappers\nimport Foundation\n\npublic struct ChatModel: Codable, Equatable, Identifiable {\n public var id: String\n public var name: String\n @FallbackDecoding\n public var format: Format\n @FallbackDecoding\n public var info: Info\n\n public init(id: String, name: String, format: Format, info: Info) {\n self.id = id\n self.name = name\n self.format = format\n self.info = info\n }\n\n public enum Format: String, Codable, Equatable, CaseIterable {\n case openAI\n case azureOpenAI\n case openAICompatible\n case googleAI\n case ollama\n case claude\n case gitHubCopilot\n }\n\n public struct Info: Codable, Equatable {\n public struct OllamaInfo: Codable, Equatable {\n @FallbackDecoding\n public var keepAlive: String\n\n public init(keepAlive: String = \"\") {\n self.keepAlive = keepAlive\n }\n }\n\n public struct OpenAIInfo: Codable, Equatable {\n @FallbackDecoding\n public var organizationID: String\n @FallbackDecoding\n public var projectID: String\n\n public init(organizationID: String = \"\", projectID: String = \"\") {\n self.organizationID = organizationID\n self.projectID = projectID\n }\n }\n\n public struct OpenAICompatibleInfo: Codable, Equatable {\n @FallbackDecoding\n public var enforceMessageOrder: Bool\n @FallbackDecoding\n public var supportsMultipartMessageContent: Bool\n @FallbackDecoding\n public var requiresBeginWithUserMessage: Bool\n\n public init(\n enforceMessageOrder: Bool = false,\n supportsMultipartMessageContent: Bool = true,\n requiresBeginWithUserMessage: Bool = false\n ) {\n self.enforceMessageOrder = enforceMessageOrder\n self.supportsMultipartMessageContent = supportsMultipartMessageContent\n self.requiresBeginWithUserMessage = requiresBeginWithUserMessage\n }\n }\n\n public struct GoogleGenerativeAIInfo: Codable, Equatable {\n @FallbackDecoding\n public var apiVersion: String\n\n public init(apiVersion: String = \"\") {\n self.apiVersion = apiVersion\n }\n }\n\n public struct CustomHeaderInfo: Codable, Equatable {\n public struct HeaderField: Codable, Equatable {\n public var key: String\n public var value: String\n\n public init(key: String, value: String) {\n self.key = key\n self.value = value\n }\n }\n\n @FallbackDecoding\n public var headers: [HeaderField]\n\n public init(headers: [HeaderField] = []) {\n self.headers = headers\n }\n }\n \n public struct CustomBodyInfo: Codable, Equatable {\n public var jsonBody: String\n \n public init(jsonBody: String = \"\") {\n self.jsonBody = jsonBody\n }\n }\n\n @FallbackDecoding\n public var apiKeyName: String\n @FallbackDecoding\n public var baseURL: String\n @FallbackDecoding\n public var isFullURL: Bool\n @FallbackDecoding\n public var maxTokens: Int\n @FallbackDecoding\n public var supportsFunctionCalling: Bool\n @FallbackDecoding\n public var supportsImage: Bool\n @FallbackDecoding\n public var supportsAudio: Bool\n @FallbackDecoding\n public var modelName: String\n\n @FallbackDecoding\n public var openAIInfo: OpenAIInfo\n @FallbackDecoding\n public var ollamaInfo: OllamaInfo\n @FallbackDecoding\n public var googleGenerativeAIInfo: GoogleGenerativeAIInfo\n @FallbackDecoding\n public var openAICompatibleInfo: OpenAICompatibleInfo\n @FallbackDecoding\n public var customHeaderInfo: CustomHeaderInfo\n @FallbackDecoding\n public var customBodyInfo: CustomBodyInfo\n\n public init(\n apiKeyName: String = \"\",\n baseURL: String = \"\",\n isFullURL: Bool = false,\n maxTokens: Int = 4000,\n supportsFunctionCalling: Bool = true,\n supportsImage: Bool = false,\n supportsAudio: Bool = false,\n modelName: String = \"\",\n openAIInfo: OpenAIInfo = OpenAIInfo(),\n ollamaInfo: OllamaInfo = OllamaInfo(),\n googleGenerativeAIInfo: GoogleGenerativeAIInfo = GoogleGenerativeAIInfo(),\n openAICompatibleInfo: OpenAICompatibleInfo = OpenAICompatibleInfo(),\n customHeaderInfo: CustomHeaderInfo = CustomHeaderInfo(),\n customBodyInfo: CustomBodyInfo = CustomBodyInfo()\n ) {\n self.apiKeyName = apiKeyName\n self.baseURL = baseURL\n self.isFullURL = isFullURL\n self.maxTokens = maxTokens\n self.supportsFunctionCalling = supportsFunctionCalling\n self.supportsImage = supportsImage\n self.supportsAudio = supportsAudio\n self.modelName = modelName\n self.openAIInfo = openAIInfo\n self.ollamaInfo = ollamaInfo\n self.googleGenerativeAIInfo = googleGenerativeAIInfo\n self.openAICompatibleInfo = openAICompatibleInfo\n self.customHeaderInfo = customHeaderInfo\n self.customBodyInfo = customBodyInfo\n }\n }\n\n public var endpoint: String {\n switch format {\n case .openAI:\n let baseURL = info.baseURL\n if baseURL.isEmpty { return \"https://api.openai.com/v1/chat/completions\" }\n return \"\\(baseURL)/v1/chat/completions\"\n case .openAICompatible:\n let baseURL = info.baseURL\n if baseURL.isEmpty { return \"https://api.openai.com/v1/chat/completions\" }\n if info.isFullURL { return baseURL }\n return \"\\(baseURL)/v1/chat/completions\"\n case .azureOpenAI:\n let baseURL = info.baseURL\n let deployment = info.modelName\n let version = \"2024-02-15-preview\"\n if baseURL.isEmpty { return \"\" }\n return \"\\(baseURL)/openai/deployments/\\(deployment)/chat/completions?api-version=\\(version)\"\n case .googleAI:\n let baseURL = info.baseURL\n if baseURL.isEmpty { return \"https://generativelanguage.googleapis.com\" }\n return \"\\(baseURL)\"\n case .ollama:\n let baseURL = info.baseURL\n if baseURL.isEmpty { return \"http://localhost:11434/api/chat\" }\n return \"\\(baseURL)/api/chat\"\n case .claude:\n let baseURL = info.baseURL\n if baseURL.isEmpty { return \"https://api.anthropic.com/v1/messages\" }\n return \"\\(baseURL)/v1/messages\"\n case .gitHubCopilot:\n return \"https://api.githubcopilot.com/chat/completions\"\n }\n }\n}\n\npublic struct EmptyChatModelInfo: FallbackValueProvider {\n public static var defaultValue: ChatModel.Info { .init() }\n}\n\npublic struct EmptyChatModelFormat: FallbackValueProvider {\n public static var defaultValue: ChatModel.Format { .openAI }\n}\n\npublic struct EmptyChatModelOllamaInfo: FallbackValueProvider {\n public static var defaultValue: ChatModel.Info.OllamaInfo { .init() }\n}\n\npublic struct EmptyChatModelOpenAIInfo: FallbackValueProvider {\n public static var defaultValue: ChatModel.Info.OpenAIInfo { .init() }\n}\n\npublic struct EmptyChatModelGoogleGenerativeAIInfo: FallbackValueProvider {\n public static var defaultValue: ChatModel.Info.GoogleGenerativeAIInfo { .init() }\n}\n\npublic struct EmptyChatModelOpenAICompatibleInfo: FallbackValueProvider {\n public static var defaultValue: ChatModel.Info.OpenAICompatibleInfo { .init() }\n}\n\npublic struct EmptyChatModelCustomHeaderInfo: FallbackValueProvider {\n public static var defaultValue: ChatModel.Info.CustomHeaderInfo { .init() }\n}\n\npublic struct EmptyChatModelCustomBodyInfo: FallbackValueProvider {\n public static var defaultValue: ChatModel.Info.CustomBodyInfo { .init() }\n}\n\npublic struct EmptyTrue: FallbackValueProvider {\n public static var defaultValue: Bool { true }\n}\n" }, { "path": "Tool/Sources/AIModel/EmbeddingModel.swift", "content": "import CodableWrappers\nimport Foundation\n\npublic struct EmbeddingModel: Codable, Equatable, Identifiable {\n public var id: String\n public var name: String\n @FallbackDecoding\n public var format: Format\n @FallbackDecoding\n public var info: Info\n\n public init(id: String, name: String, format: Format, info: Info) {\n self.id = id\n self.name = name\n self.format = format\n self.info = info\n }\n\n public enum Format: String, Codable, Equatable, CaseIterable {\n case openAI\n case azureOpenAI\n case openAICompatible\n case ollama\n case gitHubCopilot\n }\n\n public struct Info: Codable, Equatable {\n public typealias OllamaInfo = ChatModel.Info.OllamaInfo\n public typealias OpenAIInfo = ChatModel.Info.OpenAIInfo\n public typealias CustomHeaderInfo = ChatModel.Info.CustomHeaderInfo\n\n @FallbackDecoding\n public var apiKeyName: String\n @FallbackDecoding\n public var baseURL: String\n @FallbackDecoding\n public var isFullURL: Bool\n @FallbackDecoding\n public var maxTokens: Int\n @FallbackDecoding\n public var dimensions: Int\n @FallbackDecoding\n public var modelName: String\n\n @FallbackDecoding\n public var openAIInfo: OpenAIInfo\n @FallbackDecoding\n public var ollamaInfo: OllamaInfo\n @FallbackDecoding\n public var customHeaderInfo: CustomHeaderInfo\n\n public init(\n apiKeyName: String = \"\",\n baseURL: String = \"\",\n isFullURL: Bool = false,\n maxTokens: Int = 8192,\n dimensions: Int = 1536,\n modelName: String = \"\",\n openAIInfo: OpenAIInfo = OpenAIInfo(),\n ollamaInfo: OllamaInfo = OllamaInfo(),\n customHeaderInfo: CustomHeaderInfo = CustomHeaderInfo()\n ) {\n self.apiKeyName = apiKeyName\n self.baseURL = baseURL\n self.isFullURL = isFullURL\n self.maxTokens = maxTokens\n self.dimensions = dimensions\n self.modelName = modelName\n self.openAIInfo = openAIInfo\n self.ollamaInfo = ollamaInfo\n self.customHeaderInfo = customHeaderInfo\n }\n }\n\n public var endpoint: String {\n switch format {\n case .openAI:\n let baseURL = info.baseURL\n if baseURL.isEmpty { return \"https://api.openai.com/v1/embeddings\" }\n return \"\\(baseURL)/v1/embeddings\"\n case .openAICompatible:\n let baseURL = info.baseURL\n if baseURL.isEmpty { return \"https://api.openai.com/v1/embeddings\" }\n if info.isFullURL { return baseURL }\n return \"\\(baseURL)/v1/embeddings\"\n case .azureOpenAI:\n let baseURL = info.baseURL\n let deployment = info.modelName\n let version = \"2024-02-15-preview\"\n if baseURL.isEmpty { return \"\" }\n return \"\\(baseURL)/openai/deployments/\\(deployment)/embeddings?api-version=\\(version)\"\n case .ollama:\n let baseURL = info.baseURL\n if baseURL.isEmpty { return \"http://localhost:11434/api/embeddings\" }\n return \"\\(baseURL)/api/embeddings\"\n case .gitHubCopilot:\n return \"https://api.githubcopilot.com/embeddings\"\n }\n }\n}\n\npublic struct EmptyEmbeddingModelInfo: FallbackValueProvider {\n public static var defaultValue: EmbeddingModel.Info { .init() }\n}\n\npublic struct EmptyEmbeddingModelFormat: FallbackValueProvider {\n public static var defaultValue: EmbeddingModel.Format { .openAI }\n}\n\n" }, { "path": "Tool/Sources/ASTParser/ASTParser.swift", "content": "import SuggestionBasic\nimport SwiftTreeSitter\nimport tree_sitter\nimport TreeSitterObjC\n\npublic enum ParsableLanguage {\n case objectiveC\n\n var tsLanguage: UnsafeMutablePointer {\n switch self {\n case .objectiveC:\n return tree_sitter_objc()\n }\n }\n}\n\npublic struct ASTParser {\n let language: ParsableLanguage\n let parser: Parser\n\n public init(language: ParsableLanguage) {\n self.language = language\n parser = Parser()\n try! parser.setLanguage(Language(language: language.tsLanguage))\n }\n\n public func parse(_ source: String) -> ASTTree? {\n return ASTTree(tree: parser.parse(source))\n }\n}\n\npublic typealias ASTNode = Node\n\npublic typealias ASTPoint = Point\n\npublic struct ASTTree {\n public let tree: Tree?\n\n public var rootNode: ASTNode? {\n return tree?.rootNode\n }\n\n public func smallestNodeContainingRange(\n _ range: CursorRange,\n filter: (ASTNode) -> Bool = { _ in true }\n ) -> ASTNode? {\n guard var targetNode = rootNode else { return nil }\n\n func rangeContains(_ range: Range, _ another: Range) -> Bool {\n return range.lowerBound <= another.lowerBound && range.upperBound >= another.upperBound\n }\n\n for node in targetNode.treeCursor.deepFirstSearch(skipChildren: { node in\n !rangeContains(node.pointRange, range.pointRange)\n }) {\n guard filter(node) else { continue }\n if rangeContains(node.pointRange, range.pointRange) {\n targetNode = node\n }\n }\n\n return targetNode\n }\n}\n\npublic extension ASTNode {\n var children: ASTNodeChildrenSequence {\n return ASTNodeChildrenSequence(node: self)\n }\n\n struct ASTNodeChildrenSequence: Sequence {\n let node: ASTNode\n\n public struct ASTNodeChildrenIterator: IteratorProtocol {\n let node: ASTNode\n var index: UInt32 = 0\n\n public mutating func next() -> ASTNode? {\n guard index < node.childCount else { return nil }\n defer { index += 1 }\n return node.child(at: 1)\n }\n }\n\n public func makeIterator() -> ASTNodeChildrenIterator {\n return ASTNodeChildrenIterator(node: node)\n }\n }\n}\n\npublic extension CursorRange {\n var pointRange: Range {\n let bytePerCharacter = 2 // tree sitter uses UTF-16\n let startPoint = Point(row: start.line, column: start.character * bytePerCharacter)\n let endPoint = Point(row: end.line, column: end.character * bytePerCharacter)\n guard endPoint > startPoint else {\n return startPoint..) {\n let bytePerCharacter = 2 // tree sitter uses UTF-16\n let start = CursorPosition(\n line: Int(pointRange.lowerBound.row),\n character: Int(pointRange.lowerBound.column) / bytePerCharacter\n )\n let end = CursorPosition(\n line: Int(pointRange.upperBound.row),\n character: Int(pointRange.upperBound.column) / bytePerCharacter\n )\n self.init(start: start, end: end)\n }\n}\n\n" }, { "path": "Tool/Sources/ASTParser/ASTTreeVisitor.swift", "content": "import Foundation\nimport SwiftTreeSitter\n\npublic enum ASTTreeVisitorContinueKind {\n /// The visitor should visit the descendants of the current node.\n case visitChildren\n /// The visitor should avoid visiting the descendants of the current node.\n case skipChildren\n}\n\n// A SwiftSyntax style tree visitor.\nopen class ASTTreeVisitor {\n public let tree: ASTTree\n\n public init(tree: ASTTree) {\n self.tree = tree\n }\n\n public func walk() {\n guard let cursor = tree.rootNode?.treeCursor else { return }\n visit(cursor)\n }\n \n public func walk(_ node: ASTNode) {\n let cursor = node.treeCursor\n visit(cursor)\n }\n\n open func visit(_: ASTNode) -> ASTTreeVisitorContinueKind {\n // do nothing\n return .skipChildren\n }\n\n open func visitPost(_: ASTNode) {\n // do nothing\n }\n\n private func visit(_ cursor: TreeCursor) {\n guard let currentNode = cursor.currentNode else { return }\n let continueKind = visit(currentNode)\n\n switch continueKind {\n case .skipChildren:\n visitPost(currentNode)\n case .visitChildren:\n visitChildren(cursor)\n visitPost(currentNode)\n }\n }\n\n private func visitChildren(_ cursor: TreeCursor) {\n let hasChild = cursor.goToFirstChild()\n guard hasChild else { return }\n visit(cursor)\n while cursor.goToNextSibling() {\n visit(cursor)\n }\n _ = cursor.gotoParent()\n }\n}\n\n" }, { "path": "Tool/Sources/ASTParser/DumpSyntaxTree.swift", "content": "import SwiftTreeSitter\nimport SwiftUI\n\npublic extension ASTTree {\n /// Dumps the syntax tree as a string, for debugging purposes.\n func dump() -> AttributedString {\n guard let tree, let root = tree.rootNode else { return \"\" }\n var result: AttributedString = \"\"\n\n let appendNode: (_ level: Int, _ node: Node, _ name: String) -> Void = {\n level, node, name in\n let range = node.pointRange\n let lowerBoundL = range.lowerBound.row\n let lowerBoundC = range.lowerBound.column / 2\n let upperBoundL = range.upperBound.row\n let upperBoundC = range.upperBound.column / 2\n let indentation = AttributedString(String(repeating: \" \", count: level))\n let nodeInfo = {\n if name.isEmpty {\n return AttributedString(node.nodeType ?? \"N/A\", attributes: .init([\n .foregroundColor: NSColor.blue,\n ]))\n } else {\n var string = AttributedString(\"\\(name): \", attributes: .init([\n .foregroundColor: NSColor.brown,\n ]))\n string.append(AttributedString(node.nodeType ?? \"N/A\", attributes: .init([\n .foregroundColor: NSColor.blue,\n ])))\n return string\n }\n }()\n let rangeText = \"[\\(lowerBoundL), \\(lowerBoundC)] - [\\(upperBoundL), \\(upperBoundC)]\"\n\n var line: AttributedString = \"\"\n line.append(indentation)\n line.append(nodeInfo)\n line.append(AttributedString(\" \\(rangeText)\\n\"))\n \n result.append(line)\n }\n\n func enumerate(_ node: Node, level: Int, name: String) {\n appendNode(level, node, name)\n for i in 0.. Bool\n ) -> CursorDeepFirstSearchSequence {\n return CursorDeepFirstSearchSequence(cursor: self, skipChildren: skipChildren)\n }\n}\n\n// MARK: - Search\n\npublic protocol Cursor {\n associatedtype Node\n var currentNode: Node? { get }\n func goToFirstChild() -> Bool\n func goToNextSibling() -> Bool\n func goToParent() -> Bool\n}\n\nextension TreeCursor: Cursor {\n public func goToNextSibling() -> Bool {\n gotoNextSibling()\n }\n\n public func goToParent() -> Bool {\n gotoParent()\n }\n}\n\npublic struct CursorDeepFirstSearchSequence: Sequence {\n let cursor: C\n let skipChildren: (C.Node) -> Bool\n\n public func makeIterator() -> CursorDeepFirstSearchIterator {\n return CursorDeepFirstSearchIterator(\n cursor: cursor,\n skipChildren: skipChildren\n )\n }\n\n public struct CursorDeepFirstSearchIterator: IteratorProtocol {\n let cursor: C\n let skipChildren: (C.Node) -> Bool\n var isEnded = false\n\n public mutating func next() -> C.Node? {\n guard !isEnded else { return nil }\n let currentNode = cursor.currentNode\n let hasChild = {\n guard let n = currentNode else { return false }\n if skipChildren(n) { return false }\n return cursor.goToFirstChild()\n }()\n if !hasChild {\n while !cursor.goToNextSibling() {\n if !cursor.goToParent() {\n isEnded = true\n break\n }\n }\n }\n\n return currentNode\n }\n }\n}\n\n" }, { "path": "Tool/Sources/AXExtension/AXUIElement.swift", "content": "import AppKit\nimport Foundation\nimport Logger\n\n// MARK: - State\n\npublic extension AXUIElement {\n /// Set global timeout in seconds.\n static func setGlobalMessagingTimeout(_ timeout: Float) {\n AXUIElementSetMessagingTimeout(AXUIElementCreateSystemWide(), timeout)\n }\n\n /// Set timeout in seconds for this element.\n func setMessagingTimeout(_ timeout: Float) {\n AXUIElementSetMessagingTimeout(self, timeout)\n }\n\n var identifier: String {\n (try? copyValue(key: kAXIdentifierAttribute)) ?? \"\"\n }\n\n var value: String {\n (try? copyValue(key: kAXValueAttribute)) ?? \"\"\n }\n\n var intValue: Int? {\n (try? copyValue(key: kAXValueAttribute))\n }\n\n var title: String {\n (try? copyValue(key: kAXTitleAttribute)) ?? \"\"\n }\n\n var role: String {\n (try? copyValue(key: kAXRoleAttribute)) ?? \"\"\n }\n\n var doubleValue: Double {\n (try? copyValue(key: kAXValueAttribute)) ?? 0.0\n }\n\n var document: String? {\n try? copyValue(key: kAXDocumentAttribute)\n }\n\n /// Label in Accessibility Inspector.\n var description: String {\n (try? copyValue(key: kAXDescriptionAttribute)) ?? \"\"\n }\n\n /// Type in Accessibility Inspector.\n var roleDescription: String {\n (try? copyValue(key: kAXRoleDescriptionAttribute)) ?? \"\"\n }\n\n var label: String {\n (try? copyValue(key: kAXLabelValueAttribute)) ?? \"\"\n }\n\n var isSourceEditor: Bool {\n if !(description == \"Source Editor\" && role != kAXUnknownRole) { return false }\n if let _ = firstParent(where: { $0.identifier == \"editor context\" }) { return true }\n return false\n }\n\n var selectedTextRange: ClosedRange? {\n guard let value: AXValue = try? copyValue(key: kAXSelectedTextRangeAttribute)\n else { return nil }\n var range: CFRange = .init(location: 0, length: 0)\n if AXValueGetValue(value, .cfRange, &range) {\n return range.location...(range.location + range.length)\n }\n return nil\n }\n\n var isFocused: Bool {\n (try? copyValue(key: kAXFocusedAttribute)) ?? false\n }\n\n var isEnabled: Bool {\n (try? copyValue(key: kAXEnabledAttribute)) ?? false\n }\n\n var isHidden: Bool {\n (try? copyValue(key: kAXHiddenAttribute)) ?? false\n }\n\n var debugDescription: String {\n \"<\\(title)> <\\(description)> <\\(label)> (\\(role):\\(roleDescription)) [\\(identifier)] \\(rect ?? .zero) \\(children.count) children\"\n }\n\n var debugEnumerateChildren: String {\n var result = \"> \" + debugDescription + \"\\n\"\n result += children.map {\n $0.debugEnumerateChildren.split(separator: \"\\n\")\n .map { \" \" + $0 }\n .joined(separator: \"\\n\")\n }.joined(separator: \"\\n\")\n return result\n }\n\n var debugEnumerateParents: String {\n var chain: [String] = []\n chain.append(\"* \" + debugDescription)\n var parent = self.parent\n if let current = parent {\n chain.append(\"> \" + current.debugDescription)\n parent = current.parent\n }\n var result = \"\"\n for (index, line) in chain.reversed().enumerated() {\n result += String(repeating: \" \", count: index) + line + \"\\n\"\n }\n return result\n }\n}\n\n// MARK: - Rect\n\npublic extension AXUIElement {\n var position: CGPoint? {\n guard let value: AXValue = try? copyValue(key: kAXPositionAttribute)\n else { return nil }\n var point: CGPoint = .zero\n if AXValueGetValue(value, .cgPoint, &point) {\n return point\n }\n return nil\n }\n\n var size: CGSize? {\n guard let value: AXValue = try? copyValue(key: kAXSizeAttribute)\n else { return nil }\n var size: CGSize = .zero\n if AXValueGetValue(value, .cgSize, &size) {\n return size\n }\n return nil\n }\n\n var rect: CGRect? {\n guard let position, let size else { return nil }\n return .init(origin: position, size: size)\n }\n}\n\n// MARK: - Relationship\n\npublic extension AXUIElement {\n var focusedElement: AXUIElement? {\n try? copyValue(key: kAXFocusedUIElementAttribute)\n }\n\n var sharedFocusElements: [AXUIElement] {\n (try? copyValue(key: kAXChildrenAttribute)) ?? []\n }\n\n var window: AXUIElement? {\n try? copyValue(key: kAXWindowAttribute)\n }\n\n var windows: [AXUIElement] {\n (try? copyValue(key: kAXWindowsAttribute)) ?? []\n }\n\n var isFullScreen: Bool {\n (try? copyValue(key: \"AXFullScreen\")) ?? false\n }\n\n var windowID: CGWindowID? {\n var identifier: CGWindowID = 0\n let error = AXUIElementGetWindow(self, &identifier)\n if error == .success {\n return identifier\n }\n return nil\n }\n\n var isFrontmost: Bool {\n get {\n (try? copyValue(key: kAXFrontmostAttribute)) ?? false\n }\n set {\n AXUIElementSetAttributeValue(\n self,\n kAXFrontmostAttribute as CFString,\n newValue as CFBoolean\n )\n }\n }\n\n var focusedWindow: AXUIElement? {\n try? copyValue(key: kAXFocusedWindowAttribute)\n }\n\n var topLevelElement: AXUIElement? {\n try? copyValue(key: kAXTopLevelUIElementAttribute)\n }\n\n var rows: [AXUIElement] {\n (try? copyValue(key: kAXRowsAttribute)) ?? []\n }\n\n var parent: AXUIElement? {\n try? copyValue(key: kAXParentAttribute)\n }\n\n var children: [AXUIElement] {\n (try? copyValue(key: kAXChildrenAttribute)) ?? []\n }\n\n var menuBar: AXUIElement? {\n try? copyValue(key: kAXMenuBarAttribute)\n }\n\n var visibleChildren: [AXUIElement] {\n (try? copyValue(key: kAXVisibleChildrenAttribute)) ?? []\n }\n\n func child(\n identifier: String? = nil,\n title: String? = nil,\n role: String? = nil,\n depth: Int = 0\n ) -> AXUIElement? {\n #if DEBUG\n if depth >= 50 {\n fatalError(\"AXUIElement.child: Exceeding recommended depth.\")\n }\n #endif\n\n for child in children {\n let match = {\n if let identifier, child.identifier != identifier { return false }\n if let title, child.title != title { return false }\n if let role, child.role != role { return false }\n return true\n }()\n if match { return child }\n }\n for child in children {\n if let target = child.child(\n identifier: identifier,\n title: title,\n role: role,\n depth: depth + 1\n ) { return target }\n }\n return nil\n }\n\n /// Get children that match the requirement\n ///\n /// - important: If the element has a lot of descendant nodes, it will heavily affect the\n /// **performance of Xcode**. Please make use ``AXUIElement\\traverse(_:)`` instead.\n @available(\n *,\n deprecated,\n renamed: \"traverse(_:)\",\n message: \"Please make use ``AXUIElement\\traverse(_:)`` instead.\"\n )\n func children(depth: Int = 0, where match: (AXUIElement) -> Bool) -> [AXUIElement] {\n #if DEBUG\n if depth >= 50 {\n fatalError(\"AXUIElement.children: Exceeding recommended depth.\")\n }\n #endif\n\n var all = [AXUIElement]()\n for child in children {\n if match(child) { all.append(child) }\n }\n for child in children {\n all.append(contentsOf: child.children(depth: depth + 1, where: match))\n }\n return all\n }\n\n func firstParent(where match: (AXUIElement) -> Bool) -> AXUIElement? {\n guard let parent = parent else { return nil }\n if match(parent) { return parent }\n return parent.firstParent(where: match)\n }\n\n func firstChild(\n depth: Int = 0,\n maxDepth: Int = 50,\n where match: (AXUIElement) -> Bool\n ) -> AXUIElement? {\n #if DEBUG\n if depth > maxDepth {\n fatalError(\"AXUIElement.firstChild: Exceeding recommended depth.\")\n }\n #else\n if depth > maxDepth {\n return nil\n }\n #endif\n for child in children {\n if match(child) { return child }\n }\n for child in children {\n if let target = child.firstChild(depth: depth + 1, where: match) {\n return target\n }\n }\n return nil\n }\n\n func visibleChild(identifier: String) -> AXUIElement? {\n for child in visibleChildren {\n if child.identifier == identifier { return child }\n if let target = child.visibleChild(identifier: identifier) { return target }\n }\n return nil\n }\n\n var verticalScrollBar: AXUIElement? {\n try? copyValue(key: kAXVerticalScrollBarAttribute)\n }\n}\n\npublic extension AXUIElement {\n enum SearchNextStep {\n case skipDescendants\n case skipSiblings(Info)\n case skipDescendantsAndSiblings\n case continueSearching(Info)\n case stopSearching\n }\n\n /// Traversing the element tree.\n ///\n /// - important: Traversing the element tree is resource consuming and will affect the\n /// **performance of Xcode**. Please make sure to skip as much as possible.\n ///\n /// - todo: Make it not recursive.\n func traverse(\n access: (AXUIElement) -> [AXUIElement] = { $0.children },\n info: Info,\n file: StaticString = #file,\n line: UInt = #line,\n function: StaticString = #function,\n _ handle: (_ element: AXUIElement, _ level: Int, _ info: Info) -> SearchNextStep\n ) {\n #if DEBUG\n var count = 0\n// let startDate = Date()\n #endif\n func _traverse(\n element: AXUIElement,\n level: Int,\n info: Info,\n handle: (AXUIElement, Int, Info) -> SearchNextStep\n ) -> SearchNextStep {\n #if DEBUG\n count += 1\n #endif\n let nextStep = handle(element, level, info)\n switch nextStep {\n case .stopSearching: return .stopSearching\n case .skipDescendants: return .continueSearching(info)\n case .skipDescendantsAndSiblings: return .skipSiblings(info)\n case let .continueSearching(info), let .skipSiblings(info):\n loop: for child in access(element) {\n switch _traverse(element: child, level: level + 1, info: info, handle: handle) {\n case .skipSiblings, .skipDescendantsAndSiblings:\n break loop\n case .stopSearching:\n return .stopSearching\n case .continueSearching, .skipDescendants:\n continue loop\n }\n }\n\n return nextStep\n }\n }\n\n _ = _traverse(element: self, level: 0, info: info, handle: handle)\n\n #if DEBUG\n// let duration = Date().timeIntervalSince(startDate)\n// .formatted(.number.precision(.fractionLength(0...4)))\n// Logger.service.debug(\n// \"AXUIElement.traverse count: \\(count), took \\(duration) seconds\",\n// file: file,\n// line: line,\n// function: function\n// )\n #endif\n }\n\n /// Traversing the element tree.\n ///\n /// - important: Traversing the element tree is resource consuming and will affect the\n /// **performance of Xcode**. Please make sure to skip as much as possible.\n ///\n /// - todo: Make it not recursive.\n func traverse(\n access: (AXUIElement) -> [AXUIElement] = { $0.children },\n file: StaticString = #file,\n line: UInt = #line,\n function: StaticString = #function,\n _ handle: (_ element: AXUIElement, _ level: Int) -> SearchNextStep\n ) {\n traverse(access: access, info: (), file: file, line: line, function: function) {\n element, level, _ in\n handle(element, level)\n }\n }\n}\n\n// MARK: - Helper\n\npublic extension AXUIElement {\n func copyValue(key: String, ofType _: T.Type = T.self) throws -> T {\n var value: AnyObject?\n let error = AXUIElementCopyAttributeValue(self, key as CFString, &value)\n if error == .success, let value = value as? T {\n return value\n }\n throw error\n }\n\n func copyParameterizedValue(\n key: String,\n parameters: AnyObject,\n ofType _: T.Type = T.self\n ) throws -> T {\n var value: AnyObject?\n let error = AXUIElementCopyParameterizedAttributeValue(\n self,\n key as CFString,\n parameters as CFTypeRef,\n &value\n )\n if error == .success, let value = value as? T {\n return value\n }\n throw error\n }\n}\n\nextension AXError: @retroactive _BridgedNSError {}\nextension AXError: @retroactive _ObjectiveCBridgeableError {}\nextension AXError: @retroactive Error {}\n\n" }, { "path": "Tool/Sources/AXExtension/AXUIElementPrivateAPI.swift", "content": "import AppKit\n\n/// AXError _AXUIElementGetWindow(AXUIElementRef element, uint32_t *identifier);\n@_silgen_name(\"_AXUIElementGetWindow\") @discardableResult\nfunc AXUIElementGetWindow(\n _ element: AXUIElement,\n _ identifier: UnsafeMutablePointer\n) -> AXError\n" }, { "path": "Tool/Sources/AXNotificationStream/AXNotificationStream.swift", "content": "import AppKit\nimport ApplicationServices\nimport Foundation\nimport Logger\nimport Preferences\n\npublic final class AXNotificationStream: AsyncSequence {\n public typealias Stream = AsyncStream\n public typealias Continuation = Stream.Continuation\n public typealias AsyncIterator = Stream.AsyncIterator\n public typealias Element = (name: String, element: AXUIElement, info: CFDictionary)\n\n private var continuation: Continuation\n private let stream: Stream\n\n private let file: StaticString\n private let line: UInt\n private let function: StaticString\n\n public func makeAsyncIterator() -> Stream.AsyncIterator {\n stream.makeAsyncIterator()\n }\n\n deinit {\n continuation.finish()\n }\n\n public convenience init(\n app: NSRunningApplication,\n element: AXUIElement? = nil,\n notificationNames: String...,\n file: StaticString = #file,\n line: UInt = #line,\n function: StaticString = #function\n ) {\n self.init(\n app: app,\n element: element,\n notificationNames: notificationNames,\n file: file,\n line: line,\n function: function\n )\n }\n\n public init(\n app: NSRunningApplication,\n element: AXUIElement? = nil,\n notificationNames: [String],\n file: StaticString = #file,\n line: UInt = #line,\n function: StaticString = #function\n ) {\n self.file = file\n self.line = line\n self.function = function\n\n let mode: CFRunLoopMode = UserDefaults.shared\n .value(for: \\.observeToAXNotificationWithDefaultMode) ? .defaultMode : .commonModes\n\n let runLoop: CFRunLoop = CFRunLoopGetMain()\n\n var cont: Continuation!\n stream = Stream { continuation in\n cont = continuation\n }\n continuation = cont\n var observer: AXObserver?\n\n func callback(\n observer: AXObserver,\n element: AXUIElement,\n notificationName: CFString,\n userInfo: CFDictionary,\n pointer: UnsafeMutableRawPointer?\n ) {\n guard let pointer = pointer?.assumingMemoryBound(to: Continuation.self)\n else { return }\n pointer.pointee.yield((notificationName as String, element, userInfo))\n }\n\n _ = AXObserverCreateWithInfoCallback(\n app.processIdentifier,\n callback,\n &observer\n )\n guard let observer else {\n continuation.finish()\n return\n }\n\n let observingElement = element ?? AXUIElementCreateApplication(app.processIdentifier)\n continuation.onTermination = { @Sendable _ in\n for name in notificationNames {\n AXObserverRemoveNotification(observer, observingElement, name as CFString)\n }\n CFRunLoopRemoveSource(\n runLoop,\n AXObserverGetRunLoopSource(observer),\n mode\n )\n }\n\n Task { @MainActor [weak self] in\n CFRunLoopAddSource(\n runLoop,\n AXObserverGetRunLoopSource(observer),\n mode\n )\n var pendingRegistrationNames = Set(notificationNames)\n var retry = 0\n while !pendingRegistrationNames.isEmpty, retry < 100 {\n guard let self else { return }\n retry += 1\n for name in notificationNames {\n await Task.yield()\n let e = withUnsafeMutablePointer(to: &self.continuation) { pointer in\n AXObserverAddNotification(\n observer,\n observingElement,\n name as CFString,\n pointer\n )\n }\n switch e {\n case .success:\n pendingRegistrationNames.remove(name)\n case .actionUnsupported:\n Logger.service.error(\"AXObserver: Action unsupported: \\(name)\")\n pendingRegistrationNames.remove(name)\n case .apiDisabled:\n Logger.service\n .error(\"AXObserver: Accessibility API disabled, will try again later\")\n retry -= 1\n case .invalidUIElement:\n // It's possible that the UI element is not ready yet.\n //\n // Especially when you retrieve an element right after macOS is\n // awaken from sleep.\n Logger.service\n .error(\"AXObserver: Invalid UI element, notification name \\(name)\")\n case .invalidUIElementObserver:\n Logger.service.error(\"AXObserver: Invalid UI element observer\")\n pendingRegistrationNames.remove(name)\n case .cannotComplete:\n Logger.service\n .error(\"AXObserver: Failed to observe \\(name), will try again later\")\n case .notificationUnsupported:\n Logger.service.error(\"AXObserver: Notification unsupported: \\(name)\")\n pendingRegistrationNames.remove(name)\n case .notificationAlreadyRegistered:\n Logger.service.info(\"AXObserver: Notification already registered: \\(name)\")\n pendingRegistrationNames.remove(name)\n default:\n Logger.service\n .error(\n \"AXObserver: Unrecognized error \\(e) when registering \\(name), will try again later\"\n )\n }\n }\n try await Task.sleep(nanoseconds: 1_500_000_000)\n }\n }\n }\n}\n\n" }, { "path": "Tool/Sources/ActiveApplicationMonitor/ActiveApplicationMonitor.swift", "content": "import AppKit\n\npublic struct RunningApplicationInfo: Sendable {\n public let isXcode: Bool\n public let isActive: Bool\n public let isHidden: Bool\n public let localizedName: String?\n public let bundleIdentifier: String?\n public let bundleURL: URL?\n public let executableURL: URL?\n public let processIdentifier: pid_t\n public let launchDate: Date?\n public let executableArchitecture: Int\n\n init(_ application: NSRunningApplication) {\n isXcode = application.isXcode\n isActive = application.isActive\n isHidden = application.isHidden\n localizedName = application.localizedName\n bundleIdentifier = application.bundleIdentifier\n bundleURL = application.bundleURL\n executableURL = application.executableURL\n processIdentifier = application.processIdentifier\n launchDate = application.launchDate\n executableArchitecture = application.executableArchitecture\n }\n}\n\npublic extension NSRunningApplication {\n var info: RunningApplicationInfo { RunningApplicationInfo(self) }\n}\n\npublic final class ActiveApplicationMonitor {\n public static let shared = ActiveApplicationMonitor()\n public private(set) var latestXcode: NSRunningApplication? = NSWorkspace.shared\n .runningApplications\n .first(where: \\.isXcode)\n public private(set) var previousApp: NSRunningApplication?\n public private(set) var activeApplication = NSWorkspace.shared.runningApplications\n .first(where: \\.isActive)\n {\n didSet {\n if activeApplication?.isXcode ?? false {\n latestXcode = activeApplication\n }\n previousApp = oldValue\n }\n }\n\n private var infoContinuations: [UUID: AsyncStream.Continuation] = [:]\n\n private init() {\n activeApplication = NSWorkspace.shared.runningApplications.first(where: \\.isActive)\n\n Task {\n let sequence = NSWorkspace.shared.notificationCenter\n .notifications(named: NSWorkspace.didActivateApplicationNotification)\n for await notification in sequence {\n guard let app = notification\n .userInfo?[NSWorkspace.applicationUserInfoKey] as? NSRunningApplication\n else { continue }\n activeApplication = app\n notifyContinuations()\n }\n }\n }\n\n deinit {\n for continuation in infoContinuations {\n continuation.value.finish()\n }\n }\n\n public var activeXcode: NSRunningApplication? {\n if activeApplication?.isXcode ?? false {\n return activeApplication\n }\n return nil\n }\n\n public func createInfoStream() -> AsyncStream {\n .init { continuation in\n let id = UUID()\n Task { @MainActor in\n continuation.onTermination = { _ in\n self.removeInfoContinuation(id: id)\n }\n addInfoContinuation(continuation, id: id)\n continuation.yield(activeApplication?.info)\n }\n }\n }\n\n func addInfoContinuation(\n _ continuation: AsyncStream.Continuation,\n id: UUID\n ) {\n infoContinuations[id] = continuation\n }\n\n func removeInfoContinuation(id: UUID) {\n infoContinuations[id] = nil\n }\n\n private func notifyContinuations() {\n for continuation in infoContinuations {\n continuation.value.yield(activeApplication?.info)\n }\n }\n}\n\npublic extension NSRunningApplication {\n var isXcode: Bool { bundleIdentifier == \"com.apple.dt.Xcode\" }\n}\n\n" }, { "path": "Tool/Sources/AppActivator/AppActivator.swift", "content": "import AppKit\nimport Dependencies\nimport XcodeInspector\n\npublic extension NSWorkspace {\n static func activateThisApp(delay: TimeInterval = 0.10) {\n Task { @MainActor in\n try await Task.sleep(nanoseconds: UInt64(delay * 1_000_000_000))\n\n // NSApp.activate may fail. And since macOS 14, it looks like the app needs other\n // apps to call `yieldActivationToApplication` to activate itself?\n\n let activated = NSRunningApplication.current\n .activate(options: [.activateIgnoringOtherApps])\n\n if activated { return }\n\n // Fallback solution\n\n let axApplication = AXUIElementCreateApplication(\n ProcessInfo.processInfo.processIdentifier\n )\n activateAppElement(axApplication)\n//\n// let appleScript = \"\"\"\n// tell application \"System Events\"\n// set frontmost of the first process whose unix id is \\\n// \\(ProcessInfo.processInfo.processIdentifier) to true\n// end tell\n// \"\"\"\n// try await runAppleScript(appleScript)\n }\n }\n\n static func activatePreviousActiveApp(delay: TimeInterval = 0.2) {\n Task { @MainActor in\n guard let app = XcodeInspector.shared.previousActiveApplication\n else { return }\n try await Task.sleep(nanoseconds: UInt64(delay * 1_000_000_000))\n activateApp(app)\n }\n }\n\n static func activatePreviousActiveXcode(delay: TimeInterval = 0.2) {\n Task { @MainActor in\n guard let app = XcodeInspector.shared.latestActiveXcode else { return }\n try await Task.sleep(nanoseconds: UInt64(delay * 1_000_000_000))\n activateApp(app)\n }\n }\n\n static func activateApp(_ app: AppInstanceInspector) {\n // we prefer `.activate()` because it only brings the active window to the front\n if !app.activate() {\n activateAppElement(app.appElement)\n }\n }\n\n static func activateAppElement(_ appElement: AXUIElement) {\n appElement.isFrontmost = true\n }\n}\n\nstruct ActivateThisAppDependencyKey: DependencyKey {\n static var liveValue: () -> Void = { NSWorkspace.activateThisApp() }\n}\n\nstruct ActivatePreviousActiveAppDependencyKey: DependencyKey {\n static var liveValue: () -> Void = { NSWorkspace.activatePreviousActiveApp() }\n}\n\nstruct ActivatePreviousActiveXcodeDependencyKey: DependencyKey {\n static var liveValue: () -> Void = { NSWorkspace.activatePreviousActiveXcode() }\n}\n\npublic extension DependencyValues {\n var activateThisApp: () -> Void {\n get { self[ActivateThisAppDependencyKey.self] }\n set { self[ActivateThisAppDependencyKey.self] = newValue }\n }\n\n var activatePreviousActiveApp: () -> Void {\n get { self[ActivatePreviousActiveAppDependencyKey.self] }\n set { self[ActivatePreviousActiveAppDependencyKey.self] = newValue }\n }\n\n var activatePreviousActiveXcode: () -> Void {\n get { self[ActivatePreviousActiveXcodeDependencyKey.self] }\n set { self[ActivatePreviousActiveXcodeDependencyKey.self] = newValue }\n }\n}\n\n@discardableResult\nfunc runAppleScript(_ appleScript: String) async throws -> String {\n let task = Process()\n task.launchPath = \"/usr/bin/osascript\"\n task.arguments = [\"-e\", appleScript]\n let outpipe = Pipe()\n task.standardOutput = outpipe\n task.standardError = Pipe()\n\n return try await withUnsafeThrowingContinuation { continuation in\n do {\n task.terminationHandler = { _ in\n do {\n if let data = try outpipe.fileHandleForReading.readToEnd(),\n let content = String(data: data, encoding: .utf8)\n {\n continuation.resume(returning: content)\n return\n }\n continuation.resume(returning: \"\")\n } catch {\n continuation.resume(throwing: error)\n }\n }\n try task.run()\n } catch {\n continuation.resume(throwing: error)\n }\n }\n}\n\n" }, { "path": "Tool/Sources/AsyncPassthroughSubject/AsyncPassthroughSubject.swift", "content": "import AppKit\nimport Foundation\n\npublic actor AsyncPassthroughSubject {\n var tasks: [AsyncStream.Continuation] = []\n\n deinit {\n tasks.forEach { $0.finish() }\n }\n \n public init() {}\n\n public func notifications() -> AsyncStream {\n AsyncStream { [weak self] continuation in\n let task = Task { [weak self] in\n await self?.storeContinuation(continuation)\n }\n \n continuation.onTermination = { termination in\n task.cancel()\n }\n }\n }\n\n nonisolated\n public func send(_ element: Element) {\n Task { await _send(element) }\n }\n \n func _send(_ element: Element) {\n let tasks = tasks\n for task in tasks {\n task.yield(element)\n }\n }\n \n func storeContinuation(_ continuation: AsyncStream.Continuation) {\n tasks.append(continuation)\n }\n \n nonisolated\n public func finish() {\n Task { await _finish() }\n }\n \n func _finish() {\n let tasks = self.tasks\n self.tasks = []\n for task in tasks {\n task.finish()\n }\n }\n}\n\n" }, { "path": "Tool/Sources/BuiltinExtension/BuiltinExtension.swift", "content": "import ChatBasic\nimport ChatTab\nimport CopilotForXcodeKit\nimport Foundation\nimport Preferences\n\npublic protocol BuiltinExtension: CopilotForXcodeExtensionCapability {\n /// An identifier for the extension.\n var extensionIdentifier: String { get }\n\n /// All chat builders provided by this extension.\n var chatTabTypes: [any CustomChatTab] { get }\n\n /// It's usually called when the app is about to quit,\n /// you should clean up all the resources here.\n func terminate()\n}\n\n// MARK: - Default Implementation\n\npublic extension BuiltinExtension {\n var suggestionServiceId: BuiltInSuggestionFeatureProvider? { nil }\n var chatTabTypes: [any CustomChatTab] { [] }\n}\n\n// MAKR: - ChatService\n\n/// A temporary protocol for ChatServiceType. Migrate it to CopilotForXcodeKit when finished.\npublic protocol BuiltinExtensionChatServiceType: ChatServiceType {\n typealias Message = ChatMessage\n\n func sendMessage(\n _ message: String,\n history: [Message],\n references: [RetrievedContent],\n workspace: WorkspaceInfo\n ) async -> AsyncThrowingStream\n}\n\npublic struct RetrievedContent {\n public var document: ChatMessage.Reference\n public var priority: Int\n\n public init(document: ChatMessage.Reference, priority: Int) {\n self.document = document\n self.priority = priority\n }\n}\n\npublic enum ChatServiceMemoryMutation: Codable {\n public typealias Message = ChatMessage\n\n /// Add a new message to the end of memory.\n /// If an id is not provided, a new id will be generated.\n /// If an id is provided, and a message with the same id exists the message with the same\n /// id will be updated.\n case appendMessage(id: String?, role: Message.Role, text: String)\n /// Update the message with the given id.\n case updateMessage(id: String, role: Message.Role, text: String)\n /// Stream the content into a message with the given id.\n case streamIntoMessage(id: String, role: Message.Role?, text: String?)\n}\n\npublic protocol CustomChatTab {\n var name: String { get }\n var isDefaultChatTabReplacement: Bool { get }\n var canHandleOpenChatCommand: Bool { get }\n func chatBuilders() -> [ChatTabBuilder]\n func defaultChatBuilder() -> ChatTabBuilder\n func restore(from data: Data) async throws -> any ChatTabBuilder\n}\n\npublic struct TypedCustomChatTab: CustomChatTab {\n public let type: ChatTab.Type\n\n public init(of type: ChatTab.Type) {\n self.type = type\n }\n\n public var name: String { type.name }\n public var isDefaultChatTabReplacement: Bool { type.isDefaultChatTabReplacement }\n public var canHandleOpenChatCommand: Bool { type.canHandleOpenChatCommand }\n public func chatBuilders() -> [ChatTabBuilder] { type.chatBuilders() }\n public func defaultChatBuilder() -> ChatTabBuilder { type.defaultChatBuilder() }\n public func restore(from data: Data) async throws -> any ChatTabBuilder {\n try await type.restore(from: data)\n }\n}\n\n" }, { "path": "Tool/Sources/BuiltinExtension/BuiltinExtensionManager.swift", "content": "import AppKit\nimport Combine\nimport Foundation\nimport XcodeInspector\n\npublic final class BuiltinExtensionManager {\n public static let shared: BuiltinExtensionManager = .init()\n public private(set) var extensions: [any BuiltinExtension] = []\n\n init() {\n Task { [weak self] in\n let notifications = NotificationCenter.default\n .notifications(named: .activeApplicationDidChange)\n for await _ in notifications {\n guard let self else { return }\n if let app = await XcodeInspector.shared.activeApplication,\n app.isXcode || app.isExtensionService\n {\n self.checkAppConfiguration()\n }\n }\n }\n }\n\n public func setupExtensions(_ extensions: [any BuiltinExtension]) {\n self.extensions = extensions\n checkAppConfiguration()\n }\n\n public func addExtensions(_ extensions: [any BuiltinExtension]) {\n self.extensions.append(contentsOf: extensions)\n checkAppConfiguration()\n }\n\n public func terminate() {\n for ext in extensions {\n ext.terminate()\n }\n }\n}\n\nextension BuiltinExtensionManager {\n func checkAppConfiguration() {\n let suggestionFeatureProvider = UserDefaults.shared.value(for: \\.suggestionFeatureProvider)\n for ext in extensions {\n let isSuggestionFeatureInUse = switch suggestionFeatureProvider {\n case let .builtIn(provider):\n switch provider {\n case .gitHubCopilot:\n ext.extensionIdentifier == \"com.github.copilot\"\n case .codeium:\n ext.extensionIdentifier == \"com.codeium\"\n }\n case let .extension(_, bundleIdentifier):\n ext.extensionIdentifier == bundleIdentifier\n }\n let isChatFeatureInUse = false\n ext.extensionUsageDidChange(.init(\n isSuggestionServiceInUse: isSuggestionFeatureInUse,\n isChatServiceInUse: isChatFeatureInUse\n ))\n }\n }\n}\n\n" }, { "path": "Tool/Sources/BuiltinExtension/BuiltinExtensionSuggestionServiceProvider.swift", "content": "import CopilotForXcodeKit\nimport Foundation\nimport Logger\nimport Preferences\nimport SuggestionBasic\nimport SuggestionProvider\n\npublic final class BuiltinExtensionSuggestionServiceProvider<\n T: BuiltinExtension\n>: SuggestionServiceProvider {\n public var configuration: SuggestionServiceConfiguration {\n guard let service else {\n return .init(\n acceptsRelevantCodeSnippets: true,\n mixRelevantCodeSnippetsInSource: true,\n acceptsRelevantSnippetsFromOpenedFiles: true\n )\n }\n\n return service.configuration\n }\n\n let extensionManager: BuiltinExtensionManager\n\n public init(\n extension: T.Type,\n extensionManager: BuiltinExtensionManager = .shared\n ) {\n self.extensionManager = extensionManager\n }\n\n var service: CopilotForXcodeKit.SuggestionServiceType? {\n extensionManager.extensions.first { $0 is T }?.suggestionService\n }\n \n struct BuiltinExtensionSuggestionServiceNotFoundError: Error, LocalizedError {\n var errorDescription: String? {\n \"Builtin suggestion service not found.\"\n }\n }\n\n public func getSuggestions(\n _ request: SuggestionProvider.SuggestionRequest,\n workspaceInfo: CopilotForXcodeKit.WorkspaceInfo\n ) async throws -> [SuggestionBasic.CodeSuggestion] {\n guard let service else {\n Logger.service.error(\"Builtin suggestion service not found.\")\n throw BuiltinExtensionSuggestionServiceNotFoundError()\n }\n return try await service.getSuggestions(\n .init(\n fileURL: request.fileURL,\n relativePath: request.relativePath,\n language: .init(\n rawValue: languageIdentifierFromFileURL(request.fileURL).rawValue\n ) ?? .plaintext,\n content: request.content, \n originalContent: request.originalContent,\n cursorPosition: .init(\n line: request.cursorPosition.line,\n character: request.cursorPosition.character\n ),\n tabSize: request.tabSize,\n indentSize: request.indentSize,\n usesTabsForIndentation: request.usesTabsForIndentation,\n relevantCodeSnippets: request.relevantCodeSnippets.map { $0.converted }\n ),\n workspace: workspaceInfo\n ).map { $0.converted }\n }\n\n public func cancelRequest(\n workspaceInfo: CopilotForXcodeKit.WorkspaceInfo\n ) async {\n guard let service else {\n Logger.service.error(\"Builtin suggestion service not found.\")\n return\n }\n await service.cancelRequest(workspace: workspaceInfo)\n }\n\n public func notifyAccepted(\n _ suggestion: SuggestionBasic.CodeSuggestion,\n workspaceInfo: CopilotForXcodeKit.WorkspaceInfo\n ) async {\n guard let service else {\n Logger.service.error(\"Builtin suggestion service not found.\")\n return\n }\n await service.notifyAccepted(suggestion.converted, workspace: workspaceInfo)\n }\n\n public func notifyRejected(\n _ suggestions: [SuggestionBasic.CodeSuggestion],\n workspaceInfo: CopilotForXcodeKit.WorkspaceInfo\n ) async {\n guard let service else {\n Logger.service.error(\"Builtin suggestion service not found.\")\n return\n }\n await service.notifyRejected(suggestions.map(\\.converted), workspace: workspaceInfo)\n }\n}\n\nextension SuggestionProvider.SuggestionRequest {\n var converted: CopilotForXcodeKit.SuggestionRequest {\n .init(\n fileURL: fileURL,\n relativePath: relativePath,\n language: .init(rawValue: languageIdentifierFromFileURL(fileURL).rawValue)\n ?? .plaintext,\n content: content,\n originalContent: originalContent,\n cursorPosition: .init(\n line: cursorPosition.line,\n character: cursorPosition.character\n ),\n tabSize: tabSize,\n indentSize: indentSize,\n usesTabsForIndentation: usesTabsForIndentation,\n relevantCodeSnippets: relevantCodeSnippets.map(\\.converted)\n )\n }\n}\n\nextension SuggestionBasic.CodeSuggestion {\n var converted: CopilotForXcodeKit.CodeSuggestion {\n .init(\n id: id,\n text: text,\n position: .init(\n line: position.line,\n character: position.character\n ),\n range: .init(\n start: .init(\n line: range.start.line,\n character: range.start.character\n ),\n end: .init(\n line: range.end.line,\n character: range.end.character\n )\n )\n )\n }\n}\n\nextension CopilotForXcodeKit.CodeSuggestion {\n var converted: SuggestionBasic.CodeSuggestion {\n .init(\n id: id,\n text: text,\n position: .init(\n line: position.line,\n character: position.character\n ),\n range: .init(\n start: .init(\n line: range.start.line,\n character: range.start.character\n ),\n end: .init(\n line: range.end.line,\n character: range.end.character\n )\n )\n )\n }\n}\n\nextension SuggestionProvider.RelevantCodeSnippet {\n var converted: CopilotForXcodeKit.RelevantCodeSnippet {\n .init(content: content, priority: priority, filePath: filePath)\n }\n}\n\n" }, { "path": "Tool/Sources/BuiltinExtension/BuiltinExtensionWorkspacePlugin.swift", "content": "import Foundation\nimport Workspace\n\npublic final class BuiltinExtensionWorkspacePlugin: WorkspacePlugin {\n let extensionManager: BuiltinExtensionManager\n\n public init(workspace: Workspace, extensionManager: BuiltinExtensionManager = .shared) {\n self.extensionManager = extensionManager\n super.init(workspace: workspace)\n }\n\n override public func didOpenFilespace(_ filespace: Filespace) {\n notifyOpenFile(filespace: filespace)\n }\n\n override public func didSaveFilespace(_ filespace: Filespace) {\n notifySaveFile(filespace: filespace)\n }\n\n override public func didUpdateFilespace(_ filespace: Filespace, content: String) {\n notifyUpdateFile(filespace: filespace, content: content)\n }\n\n override public func didCloseFilespace(_ fileURL: URL) {\n Task {\n for ext in extensionManager.extensions {\n ext.workspace(\n .init(workspaceURL: workspaceURL, projectURL: projectRootURL),\n didCloseDocumentAt: fileURL\n )\n }\n }\n }\n\n public func notifyOpenFile(filespace: Filespace) {\n Task {\n guard filespace.isTextReadable else { return }\n guard !(await filespace.isGitIgnored) else { return }\n for ext in extensionManager.extensions {\n ext.workspace(\n .init(workspaceURL: workspaceURL, projectURL: projectRootURL),\n didOpenDocumentAt: filespace.fileURL\n )\n }\n }\n }\n\n public func notifyUpdateFile(filespace: Filespace, content: String) {\n Task {\n guard filespace.isTextReadable else { return }\n guard !(await filespace.isGitIgnored) else { return }\n for ext in extensionManager.extensions {\n ext.workspace(\n .init(workspaceURL: workspaceURL, projectURL: projectRootURL),\n didUpdateDocumentAt: filespace.fileURL, \n content: content\n )\n }\n }\n }\n\n public func notifySaveFile(filespace: Filespace) {\n Task {\n guard filespace.isTextReadable else { return }\n guard !(await filespace.isGitIgnored) else { return }\n for ext in extensionManager.extensions {\n ext.workspace(\n .init(workspaceURL: workspaceURL, projectURL: projectRootURL),\n didSaveDocumentAt: filespace.fileURL\n )\n }\n }\n }\n}\n\n" }, { "path": "Tool/Sources/ChatBasic/ChatAgent.swift", "content": "import Foundation\n\npublic enum ChatAgentResponse {\n public enum Content {\n case text(String)\n }\n\n public enum ActionResult {\n case success(String)\n case failure(String)\n }\n\n /// Post the status of the current message.\n case status([String])\n /// Stream the content to the current message.\n case content(Content)\n /// Update the attachments of the current message.\n case attachments([URL])\n /// start a new action.\n case startAction(id: String, task: String)\n /// Finish the current action.\n case finishAction(id: String, result: ActionResult)\n /// Update the references of the current message.\n case references([ChatMessage.Reference])\n /// End the current message. The next contents will be sent as a new message.\n case startNewMessage\n /// Reasoning\n case reasoning(String)\n}\n\npublic struct ChatAgentRequest {\n public var text: String\n public var history: [ChatMessage]\n public var references: [ChatMessage.Reference]\n public var topics: [ChatMessage.Reference]\n public var agentInstructions: String? = nil\n\n public init(\n text: String,\n history: [ChatMessage],\n references: [ChatMessage.Reference],\n topics: [ChatMessage.Reference],\n agentInstructions: String? = nil\n ) {\n self.text = text\n self.history = history\n self.references = references\n self.topics = topics\n self.agentInstructions = agentInstructions\n }\n}\n\npublic protocol ChatAgent {\n typealias Response = ChatAgentResponse\n typealias Request = ChatAgentRequest\n /// Send a request to the agent.\n func send(_ request: Request) async -> AsyncThrowingStream\n}\n\npublic extension AsyncThrowingStream {\n func asTexts() async throws -> [String] {\n var result = [String]()\n var text = \"\"\n for try await response in self {\n switch response {\n case let .content(.text(content)):\n text += content\n case .startNewMessage:\n if !text.isEmpty {\n result.append(text)\n text = \"\"\n }\n default:\n break\n }\n }\n if !text.isEmpty {\n result.append(text)\n }\n return result\n }\n}\n\n" }, { "path": "Tool/Sources/ChatBasic/ChatGPTFunction.swift", "content": "import Foundation\n\npublic enum ChatGPTFunctionCallPhase {\n case detected\n case processing(argumentsJsonString: String)\n case ended(argumentsJsonString: String, result: ChatGPTFunctionResult)\n case error(argumentsJsonString: String, result: Error)\n}\n\npublic enum ChatGPTFunctionResultUserReadableContent: Sendable {\n public struct ListItem: Sendable {\n public enum Detail: Sendable {\n case link(URL)\n case text(String)\n }\n \n public var title: String\n public var description: String?\n public var detail: Detail?\n \n public init(title: String, description: String? = nil, detail: Detail? = nil) {\n self.title = title\n self.description = description\n self.detail = detail\n }\n }\n \n case text(String)\n case list([ListItem])\n case searchResult([ListItem], queries: [String])\n}\n\npublic protocol ChatGPTFunctionResult {\n var botReadableContent: String { get }\n var userReadableContent: ChatGPTFunctionResultUserReadableContent { get }\n}\n\nextension String: ChatGPTFunctionResult {\n public var botReadableContent: String { self }\n public var userReadableContent: ChatGPTFunctionResultUserReadableContent {\n .text(self)\n }\n}\n\npublic struct NoChatGPTFunctionArguments: Decodable {}\n\npublic protocol ChatGPTFunction {\n typealias NoArguments = NoChatGPTFunctionArguments\n associatedtype Arguments: Decodable\n associatedtype Result: ChatGPTFunctionResult\n typealias ReportProgress = @Sendable (String) async -> Void\n\n /// The name of this function.\n /// May contain a-z, A-Z, 0-9, and underscores, with a maximum length of 64 characters.\n var name: String { get }\n /// A short description telling the bot when it should use this function.\n var description: String { get }\n /// The arguments schema that the function take in [JSON schema](https://json-schema.org).\n var argumentSchema: JSONSchemaValue { get }\n /// Prepare to call the function\n func prepare(reportProgress: @escaping ReportProgress) async\n /// Call the function with the given arguments.\n func call(arguments: Arguments, reportProgress: @escaping ReportProgress) async throws\n -> Result\n}\n\npublic extension ChatGPTFunction {\n /// Call the function with the given arguments in JSON.\n func call(\n argumentsJsonString: String,\n reportProgress: @escaping ReportProgress\n ) async throws -> Result {\n let arguments = try await {\n do {\n return try JSONDecoder()\n .decode(Arguments.self, from: argumentsJsonString.data(using: .utf8) ?? Data())\n } catch {\n await reportProgress(\n \"Error: Failed to decode arguments. \\(error.localizedDescription)\"\n )\n throw error\n }\n }()\n return try await call(arguments: arguments, reportProgress: reportProgress)\n }\n}\n\npublic extension ChatGPTFunction where Arguments == NoArguments {\n var argumentSchema: JSONSchemaValue {\n [.type: \"object\", .properties: [:]]\n }\n}\n\n/// This kind of function is only used to get a structured output from the bot.\npublic protocol ChatGPTArgumentsCollectingFunction: ChatGPTFunction where Result == String {}\n\npublic extension ChatGPTArgumentsCollectingFunction {\n func prepare(reportProgress: @escaping ReportProgress = { _ in }) async {\n assertionFailure(\"This function is only used to get a structured output from the bot.\")\n }\n\n func call(\n arguments: Arguments,\n reportProgress: @escaping ReportProgress = { _ in }\n ) async throws -> Result {\n assertionFailure(\"This function is only used to get a structured output from the bot.\")\n return \"\"\n }\n\n func call(\n argumentsJsonString: String,\n reportProgress: @escaping ReportProgress\n ) async throws -> Result {\n assertionFailure(\"This function is only used to get a structured output from the bot.\")\n return \"\"\n }\n}\n\npublic struct ChatGPTFunctionSchema: Codable, Equatable, Sendable {\n public var name: String\n public var description: String\n public var parameters: JSONSchemaValue\n\n public init(name: String, description: String, parameters: JSONSchemaValue) {\n self.name = name\n self.description = description\n self.parameters = parameters\n }\n}\n\n" }, { "path": "Tool/Sources/ChatBasic/ChatMessage.swift", "content": "@preconcurrency import CodableWrappers\nimport Foundation\n\n/// A chat message that can be sent or received.\npublic struct ChatMessage: Equatable, Codable, Sendable {\n public typealias ID = String\n\n /// The role of a message.\n public enum Role: String, Codable, Equatable, Sendable {\n case system\n case user\n case assistant\n // There is no `tool` role\n // because tool calls and results are stored in the assistant messages.\n }\n\n /// A function call that can be made by the bot.\n public struct FunctionCall: Codable, Equatable, Sendable {\n /// The name of the function.\n public var name: String\n /// Arguments in the format of a JSON string.\n public var arguments: String\n public init(name: String, arguments: String) {\n self.name = name\n self.arguments = arguments\n }\n }\n\n /// A tool call that can be made by the bot.\n public struct ToolCall: Codable, Equatable, Identifiable, Sendable {\n public var id: String\n /// The type of tool call.\n public var type: String\n /// The actual function call.\n public var function: FunctionCall\n /// The response of the function call.\n public var response: ToolCallResponse\n public init(\n id: String,\n type: String,\n function: FunctionCall,\n response: ToolCallResponse? = nil\n ) {\n self.id = id\n self.type = type\n self.function = function\n self.response = response ?? .init(content: \"\", summary: nil)\n }\n }\n\n /// The response of a tool call\n public struct ToolCallResponse: Codable, Equatable, Sendable {\n /// The content of the response.\n public var content: String\n /// The summary of the response to display in UI.\n public var summary: String?\n public init(content: String, summary: String?) {\n self.content = content\n self.summary = summary\n }\n }\n\n /// A reference to include in a chat message.\n public struct Reference: Codable, Equatable, Identifiable, Sendable {\n /// The kind of reference.\n public enum Kind: Codable, Equatable, Sendable {\n public enum Symbol: String, Codable, Sendable {\n case `class`\n case `struct`\n case `enum`\n case `actor`\n case `protocol`\n case `extension`\n case `case`\n case property\n case `typealias`\n case function\n case method\n }\n\n /// Code symbol.\n case symbol(Symbol, uri: String, startLine: Int?, endLine: Int?)\n /// Some text.\n case text\n /// A webpage.\n case webpage(uri: String)\n /// A text file.\n case textFile(uri: String)\n /// Other kind of reference.\n case other(kind: String)\n /// Error case.\n case error\n }\n\n @FallbackDecoding\n public var id: String\n /// The title of the reference.\n public var title: String\n /// The content of the reference.\n public var content: String\n /// The kind of the reference.\n @FallbackDecoding\n public var kind: Kind\n\n public init(\n id: String = UUID().uuidString,\n title: String,\n content: String,\n kind: Kind\n ) {\n self.id = id\n self.title = title\n self.content = content\n self.kind = kind\n }\n }\n\n public struct Image: Equatable, Sendable, Codable {\n public enum Format: String, Sendable, Codable {\n case png = \"image/png\"\n case jpeg = \"image/jpeg\"\n case gif = \"image/gif\"\n }\n\n public var base64EncodedData: String\n public var format: Format\n public var urlString: String {\n \"data:\\(format.rawValue);base64,\\(base64EncodedData)\"\n }\n\n public init(base64EncodedData: String, format: Format) {\n self.base64EncodedData = base64EncodedData\n self.format = format\n }\n }\n\n /// The role of a message.\n @FallbackDecoding\n public var role: Role\n\n /// The content of the message, either the chat message, or a result of a function call.\n public var content: String? {\n didSet { tokensCount = nil }\n }\n\n /// A function call from the bot.\n public var toolCalls: [ToolCall]? {\n didSet { tokensCount = nil }\n }\n\n /// The function name of a reply to a function call.\n public var name: String? {\n didSet { tokensCount = nil }\n }\n\n /// The summary of a message that is used for display.\n public var summary: String?\n\n /// The id of the message.\n public var id: ID\n\n /// The id of the sender of the message.\n public var senderId: String?\n\n /// The id of the message that this message is a response to.\n public var responseTo: ID?\n\n /// The number of tokens of this message.\n public var tokensCount: Int?\n\n /// The references of this message.\n @FallbackDecoding>\n public var references: [Reference]\n\n /// The images associated with this message.\n @FallbackDecoding>\n public var images: [Image]\n\n /// Cache the message in the prompt if possible.\n public var cacheIfPossible: Bool\n\n /// Is the message considered empty.\n public var isEmpty: Bool {\n if let content, !content.isEmpty { return false }\n if let toolCalls, !toolCalls.isEmpty { return false }\n if let name, !name.isEmpty { return false }\n return true\n }\n\n public init(\n id: String = UUID().uuidString,\n senderId: String? = nil,\n responseTo: String? = nil,\n role: Role,\n content: String?,\n name: String? = nil,\n toolCalls: [ToolCall]? = nil,\n summary: String? = nil,\n tokenCount: Int? = nil,\n references: [Reference] = [],\n images: [Image] = [],\n cacheIfPossible: Bool = false\n ) {\n self.role = role\n self.senderId = senderId\n self.responseTo = responseTo\n self.content = content\n self.name = name\n self.toolCalls = toolCalls\n self.summary = summary\n self.id = id\n tokensCount = tokenCount\n self.references = references\n self.images = images\n self.cacheIfPossible = cacheIfPossible\n }\n}\n\npublic struct ReferenceKindFallback: FallbackValueProvider, Sendable {\n public static var defaultValue: ChatMessage.Reference.Kind { .other(kind: \"Unknown\") }\n}\n\npublic struct ReferenceIDFallback: FallbackValueProvider, Sendable {\n public static var defaultValue: String { UUID().uuidString }\n}\n\npublic struct ChatMessageRoleFallback: FallbackValueProvider, Sendable {\n public static var defaultValue: ChatMessage.Role { .user }\n}\n\n" }, { "path": "Tool/Sources/ChatBasic/ChatPlugin.swift", "content": "import Foundation\n\npublic struct ChatPluginRequest: Sendable {\n public var text: String\n public var arguments: [String]\n public var history: [ChatMessage]\n\n public init(text: String, arguments: [String], history: [ChatMessage]) {\n self.text = text\n self.arguments = arguments\n self.history = history\n }\n}\n\npublic protocol ChatPlugin {\n typealias Response = ChatAgentResponse\n typealias Request = ChatPluginRequest\n static var id: String { get }\n static var command: String { get }\n static var name: String { get }\n static var description: String { get }\n // In this method, the plugin is able to send more complicated response. It also enables it to\n // perform special tasks like starting a new message or reporting progress.\n func sendForComplicatedResponse(\n _ request: Request\n ) async -> AsyncThrowingStream\n // This method allows the plugin to respond a stream of text content only.\n func sendForTextResponse(_ request: Request) async -> AsyncThrowingStream\n func formatContent(_ content: Response.Content) -> Response.Content\n init()\n}\n\npublic extension ChatPlugin {\n func formatContent(_ content: Response.Content) -> Response.Content {\n return content\n }\n \n func sendForComplicatedResponse(\n _ request: Request\n ) async -> AsyncThrowingStream {\n let textStream = await sendForTextResponse(request)\n return AsyncThrowingStream { continuation in\n let task = Task {\n do {\n for try await text in textStream {\n continuation.yield(Response.content(.text(text)))\n }\n continuation.finish()\n } catch {\n continuation.finish(throwing: error)\n }\n }\n \n continuation.onTermination = { _ in\n task.cancel()\n }\n }\n }\n}\n" }, { "path": "Tool/Sources/ChatBasic/JSONSchema.swift", "content": "import Foundation\n\npublic struct JSONSchemaKey: Codable, Hashable, Sendable, Equatable, ExpressibleByStringLiteral {\n public var key: String\n \n public init(stringLiteral: String) {\n key = stringLiteral\n }\n \n public func encode(to encoder: Encoder) throws {\n var container = encoder.singleValueContainer()\n try container.encode(key)\n }\n \n public init(from decoder: Decoder) throws {\n let single = try? decoder.singleValueContainer()\n if let value = try? single?.decode(String.self) {\n key = value\n return\n }\n throw DecodingError.dataCorrupted(.init(codingPath: decoder.codingPath, debugDescription: \"failed to decode JSON schema key\"))\n }\n \n public static let type: JSONSchemaKey = \"type\"\n public static let minLength: JSONSchemaKey = \"minLength\"\n public static let maxLength: JSONSchemaKey = \"maxLength\"\n public static let pattern: JSONSchemaKey = \"pattern\"\n public static let format: JSONSchemaKey = \"format\"\n public static let multipleOf: JSONSchemaKey = \"multipleOf\"\n public static let minimum: JSONSchemaKey = \"minimum\"\n public static let exclusiveMinimum: JSONSchemaKey = \"exclusiveMinimum\"\n public static let maximum: JSONSchemaKey = \"maximum\"\n public static let exclusiveMaximum: JSONSchemaKey = \"exclusiveMaximum\"\n public static let minProperties: JSONSchemaKey = \"minProperties\"\n public static let maxProperties: JSONSchemaKey = \"maxProperties\"\n public static let required: JSONSchemaKey = \"required\"\n public static let properties: JSONSchemaKey = \"properties\"\n public static let patternProperties: JSONSchemaKey = \"patternProperties\"\n public static let additionalProperties: JSONSchemaKey = \"additionalProperties\"\n public static let dependencies: JSONSchemaKey = \"dependencies\"\n public static let propertyNames: JSONSchemaKey = \"propertyNames\"\n public static let minItems: JSONSchemaKey = \"minItems\"\n public static let maxItems: JSONSchemaKey = \"maxItems\"\n public static let uniqueItems: JSONSchemaKey = \"uniqueItems\"\n public static let items: JSONSchemaKey = \"items\"\n public static let additionalItems: JSONSchemaKey = \"additionalItems\"\n public static let contains: JSONSchemaKey = \"contains\"\n public static let `enum`: JSONSchemaKey = \"enum\"\n public static let const: JSONSchemaKey = \"const\"\n public static let title: JSONSchemaKey = \"title\"\n public static let description: JSONSchemaKey = \"description\"\n public static let `default`: JSONSchemaKey = \"default\"\n public static let examples: JSONSchemaKey = \"examples\"\n public static let comment: JSONSchemaKey = \"$comment\"\n public static let allOf: JSONSchemaKey = \"allOf\"\n public static let anyOf: JSONSchemaKey = \"anyOf\"\n public static let oneOf: JSONSchemaKey = \"oneOf\"\n public static let not: JSONSchemaKey = \"not\"\n public static let `if`: JSONSchemaKey = \"if\"\n public static let then: JSONSchemaKey = \"then\"\n public static let `else`: JSONSchemaKey = \"else\"\n}\n\n\npublic enum JSONSchemaValue: Codable, Hashable, Sendable {\n case bool(Bool)\n case number(Double)\n case string(String)\n case array([JSONSchemaValue])\n case hash([String: JSONSchemaValue])\n\n public func encode(to encoder: Encoder) throws {\n var container = encoder.singleValueContainer()\n\n switch self {\n case .bool(let value):\n try container.encode(value)\n case .number(let value):\n try container.encode(value)\n case .string(let value):\n try container.encode(value)\n case .array(let value):\n try container.encode(value)\n case .hash(let value):\n try container.encode(value)\n }\n }\n\n public init(from decoder: Decoder) throws {\n let single = try? decoder.singleValueContainer()\n\n if let value = try? single?.decode([String: JSONSchemaValue].self) {\n self = .hash(value)\n return\n }\n\n if let value = try? single?.decode([JSONSchemaValue].self) {\n self = .array(value)\n return\n }\n\n if let value = try? single?.decode(String.self) {\n self = .string(value)\n return\n }\n\n if let value = try? single?.decode(Double.self) {\n self = .number(value)\n return\n }\n\n if let value = try? single?.decode(Bool.self) {\n self = .bool(value)\n return\n }\n\n throw DecodingError.dataCorrupted(.init(codingPath: decoder.codingPath, debugDescription: \"failed to decode JSON schema object\"))\n }\n}\n\nextension JSONSchemaValue: ExpressibleByDictionaryLiteral {\n public init(dictionaryLiteral elements: (JSONSchemaKey, JSONSchemaValue)...) {\n var hash = [String: JSONSchemaValue]()\n\n for element in elements {\n hash[element.0.key] = element.1\n }\n\n self = .hash(hash)\n }\n}\n\nextension JSONSchemaValue: ExpressibleByStringLiteral {\n public init(stringLiteral: String) {\n self = .string(stringLiteral)\n }\n}\n\nextension JSONSchemaValue: ExpressibleByIntegerLiteral {\n public init(integerLiteral value: IntegerLiteralType) {\n self = .number(Double(value))\n }\n}\n\nextension JSONSchemaValue: ExpressibleByFloatLiteral {\n public init(floatLiteral value: FloatLiteralType) {\n self = .number(value)\n }\n}\n\nextension JSONSchemaValue: ExpressibleByArrayLiteral {\n public init(arrayLiteral elements: JSONSchemaValue...) {\n var array = [JSONSchemaValue]()\n\n for element in elements {\n array.append(element)\n }\n\n self = .array(array)\n }\n}\n\nextension JSONSchemaValue: ExpressibleByBooleanLiteral {\n public init(booleanLiteral value: BooleanLiteralType) {\n self = .bool(value)\n }\n}\n" }, { "path": "Tool/Sources/ChatContextCollector/ChatContextCollector.swift", "content": "import ChatBasic\nimport Foundation\nimport OpenAIService\nimport Parsing\n\npublic struct ChatContext {\n public enum Scope: String, Equatable, CaseIterable, Codable, Sendable {\n case file\n case code\n case sense\n case project\n case web\n }\n\n public struct RetrievedContent: Sendable {\n public var document: ChatMessage.Reference\n public var priority: Int\n\n public init(document: ChatMessage.Reference, priority: Int) {\n self.document = document\n self.priority = priority\n }\n }\n\n public var systemPrompt: String\n public var retrievedContent: [RetrievedContent]\n public var functions: [any ChatGPTFunction]\n public init(\n systemPrompt: String,\n retrievedContent: [RetrievedContent],\n functions: [any ChatGPTFunction]\n ) {\n self.systemPrompt = systemPrompt\n self.retrievedContent = retrievedContent\n self.functions = functions\n }\n\n public static var empty: Self {\n .init(systemPrompt: \"\", retrievedContent: [], functions: [])\n }\n}\n\npublic extension ChatContext.Scope {\n init?(text: String) {\n for scope in Self.allCases {\n if scope.rawValue.hasPrefix(text.lowercased()) {\n self = scope\n return\n }\n }\n return nil\n }\n}\n\npublic protocol ChatContextCollector {\n func generateContext(\n history: [ChatMessage],\n scopes: Set,\n content: String,\n configuration: ChatGPTConfiguration\n ) async -> ChatContext\n}\n\npublic struct MessageScopeParser {\n public init() {}\n\n public func callAsFunction(_ content: inout String) -> Set {\n return parseScopes(&content)\n }\n\n func parseScopes(_ prompt: inout String) -> Set {\n guard !prompt.isEmpty else { return [] }\n do {\n let parser = Parse {\n \"@\"\n Many {\n Prefix { $0.isLetter }\n } separator: {\n \"+\"\n } terminator: {\n \" \"\n }\n Skip {\n Many {\n \" \"\n }\n }\n Rest()\n }\n let (scopes, rest) = try parser.parse(prompt)\n prompt = String(rest)\n return Set(scopes.map(String.init).compactMap(ChatContext.Scope.init(text:)))\n } catch {\n return []\n }\n }\n}\n\n" }, { "path": "Tool/Sources/ChatContextCollectors/ActiveDocumentChatContextCollector/ActiveDocumentChatContextCollector.swift", "content": "import ASTParser\nimport ChatBasic\nimport ChatContextCollector\nimport Dependencies\nimport FocusedCodeFinder\nimport Foundation\nimport GitIgnoreCheck\nimport OpenAIService\nimport Preferences\nimport SuggestionBasic\nimport XcodeInspector\n\npublic final class ActiveDocumentChatContextCollector: ChatContextCollector {\n public init() {}\n\n public var activeDocumentContext: ActiveDocumentContext?\n\n @Dependency(\\.gitIgnoredChecker) var gitIgnoredChecker\n\n public func generateContext(\n history: [ChatMessage],\n scopes: Set,\n content: String,\n configuration: ChatGPTConfiguration\n ) async -> ChatContext {\n guard let info = await XcodeInspector.shared.getFocusedEditorContent()\n else { return .empty }\n let context = getActiveDocumentContext(info)\n activeDocumentContext = context\n\n let isSensitive = await gitIgnoredChecker.checkIfGitIgnored(fileURL: info.documentURL)\n\n guard scopes.contains(.code)\n else {\n if scopes.contains(.file) {\n var removedCode = context\n removedCode.focusedContext = nil\n return .init(\n systemPrompt: extractSystemPrompt(removedCode, isSensitive: isSensitive),\n retrievedContent: [],\n functions: []\n )\n }\n return .empty\n }\n\n var functions = [any ChatGPTFunction]()\n\n if !isSensitive {\n let contextLineRange: String = {\n if let range = context.focusedContext?.codeRange {\n return \" from \\(range.start.line + 1) to \\(range.end.line + 1)\"\n }\n return \"\"\n }()\n \n var functionPrompt = \"\"\"\n ONLY call it when one of the following conditions are satisfied:\n - the user explicitly ask you about specific line of code, that is NOT in the focused range \\(contextLineRange).\n \"\"\"\n\n if let annotations = context.focusedContext?.otherLineAnnotations,\n !annotations.isEmpty\n {\n functionPrompt += \"\"\"\n\n - the user ask about annotations at line \\(\n Set(annotations.map(\\.line)).map(String.init).joined(separator: \",\")\n ).\n \"\"\"\n }\n\n print(functionPrompt)\n\n functions.append(GetCodeCodeAroundLineFunction(\n contextCollector: self,\n additionalDescription: functionPrompt\n ))\n }\n\n return .init(\n systemPrompt: extractSystemPrompt(context, isSensitive: isSensitive),\n retrievedContent: [],\n functions: functions\n )\n }\n\n func getActiveDocumentContext(_ info: EditorInformation) -> ActiveDocumentContext {\n var activeDocumentContext = activeDocumentContext ?? .init(\n documentURL: .init(fileURLWithPath: \"/\"),\n relativePath: \"\",\n language: .builtIn(.swift),\n fileContent: \"\",\n lines: [],\n selectedCode: \"\",\n selectionRange: .outOfScope,\n lineAnnotations: [],\n imports: [],\n includes: []\n )\n\n activeDocumentContext.update(info)\n return activeDocumentContext\n }\n\n func extractSystemPrompt(_ context: ActiveDocumentContext, isSensitive: Bool) -> String {\n let start = \"\"\"\n ## Active Document\n\n The active document is the source code the user is editing right now.\n\n \\(\n context.focusedContext == nil\n ? \"\"\n : \"When you don't known what I am asking, I am probably referring to the document.\"\n )\n \"\"\"\n let relativePath = \"Document Relative Path: \\(context.relativePath)\"\n let language = \"Language: \\(context.language.rawValue)\"\n\n let focusedContextExplanation =\n \"Below is the code inside the active document that the user is looking at right now:\"\n\n if let focusedContext = context.focusedContext {\n let codeContext = focusedContext.context.isEmpty || isSensitive\n ? \"\"\n : \"\"\"\n Focused Context:\n ```\n \\(focusedContext.context.map(\\.signature).joined(separator: \"\\n\"))\n ```\n \"\"\"\n\n let codeRange = \"Focused Range [line, character]: \\(focusedContext.codeRange)\"\n\n let code = context.selectionRange.isEmpty && isSensitive\n ? \"\"\"\n The file is in gitignore, you can't read the file.\n Ask the user to select the code in the editor to get help. Also tell them the file is in gitignore.\n \"\"\"\n : \"\"\"\n Focused Code (from line \\(\n focusedContext.codeRange.start.line + 1\n ) to line \\(focusedContext.codeRange.end.line + 1)):\n ```\\(context.language.rawValue)\n \\(focusedContext.code)\n ```\n \"\"\"\n\n let fileAnnotations = focusedContext.otherLineAnnotations.isEmpty || isSensitive\n ? \"\"\n : \"\"\"\n Out-of-scope Annotations:\\\"\"\"\n (The related code are not inside the focused code.)\n \\(\n focusedContext.otherLineAnnotations\n .map(convertAnnotationToText)\n .joined(separator: \"\\n\")\n )\n \\\"\"\"\n \"\"\"\n\n let codeAnnotations = focusedContext.lineAnnotations.isEmpty || isSensitive\n ? \"\"\n : \"\"\"\n Annotations Inside Focused Range:\\\"\"\"\n \\(\n focusedContext.lineAnnotations\n .map(convertAnnotationToText)\n .joined(separator: \"\\n\")\n )\n \\\"\"\"\n \"\"\"\n\n return [\n start,\n relativePath,\n language,\n focusedContextExplanation,\n codeContext,\n codeRange,\n code,\n codeAnnotations,\n fileAnnotations,\n ]\n .filter { !$0.isEmpty }\n .joined(separator: \"\\n\\n\")\n } else {\n let selectionRange = \"Selection Range [line, character]: \\(context.selectionRange)\"\n let lineAnnotations = context.lineAnnotations.isEmpty || isSensitive\n ? \"\"\n : \"\"\"\n Line Annotations:\\\"\"\"\n \\(context.lineAnnotations.map(convertAnnotationToText).joined(separator: \"\\n\"))\n \\\"\"\"\n \"\"\"\n\n return [\n start,\n relativePath,\n language,\n lineAnnotations,\n selectionRange,\n ]\n .filter { !$0.isEmpty }\n .joined(separator: \"\\n\\n\")\n }\n }\n\n func convertAnnotationToText(_ annotation: EditorInformation.LineAnnotation) -> String {\n return \"- Line \\(annotation.line), \\(annotation.type): \\(annotation.message)\"\n }\n}\n\n" }, { "path": "Tool/Sources/ChatContextCollectors/ActiveDocumentChatContextCollector/Functions/GetCodeCodeAroundLineFunction.swift", "content": "import ASTParser\nimport ChatBasic\nimport Foundation\nimport OpenAIService\nimport SuggestionBasic\n\nstruct GetCodeCodeAroundLineFunction: ChatGPTFunction {\n struct Arguments: Codable {\n var line: Int\n }\n\n struct Result: ChatGPTFunctionResult {\n var range: CursorRange\n var content: String\n var language: CodeLanguage\n\n var botReadableContent: String {\n \"\"\"\n Code in range \\(range)\n ```\\(language.rawValue)\n \\(content)\n ```\n \"\"\"\n }\n \n var userReadableContent: ChatGPTFunctionResultUserReadableContent {\n .text(botReadableContent)\n }\n }\n\n struct E: Error, LocalizedError {\n var errorDescription: String?\n }\n\n var name: String {\n \"getCodeAtLine\"\n }\n\n var description: String {\n \"Get the code at the given line. \\(additionalDescription)\"\n }\n\n var argumentSchema: JSONSchemaValue { [\n .type: \"object\",\n .properties: [\n \"line\": [\n .type: \"number\",\n .description: \"The line number in the file\",\n ],\n ],\n .required: [\"line\"],\n ] }\n\n weak var contextCollector: ActiveDocumentChatContextCollector?\n \n let additionalDescription: String\n\n init(contextCollector: ActiveDocumentChatContextCollector, additionalDescription: String = \"\") {\n self.contextCollector = contextCollector\n self.additionalDescription = additionalDescription\n }\n\n func prepare(reportProgress: @escaping (String) async -> Void) async {\n await reportProgress(\"Finding code around..\")\n }\n\n func call(\n arguments: Arguments,\n reportProgress: @escaping (String) async -> Void\n ) async throws -> Result {\n guard var activeDocumentContext = contextCollector?.activeDocumentContext else {\n throw E(errorDescription: \"No active document found.\")\n }\n await reportProgress(\"Reading code around line \\(arguments.line)..\")\n activeDocumentContext.moveToCodeAroundLine(max(arguments.line - 1, 0))\n guard let newContext = activeDocumentContext.focusedContext else {\n let progress = \"Failed to read code around line \\(arguments.line)..)\"\n await reportProgress(progress)\n throw E(errorDescription: progress)\n }\n let progress = \"Finish reading code at \\(newContext.codeRange)\"\n await reportProgress(progress)\n return .init(\n range: newContext.codeRange,\n content: newContext.code\n .split(omittingEmptySubsequences: false, whereSeparator: \\.isNewline)\n .enumerated()\n .map {\n let (index, content) = $0\n if index + newContext.codeRange.start.line == arguments.line - 1 {\n return content + \" // <--- line \\(arguments.line)\"\n } else {\n return content\n }\n }\n .joined(separator: \"\\n\"),\n language: activeDocumentContext.language\n )\n }\n}\n\n" }, { "path": "Tool/Sources/ChatContextCollectors/ActiveDocumentChatContextCollector/LegacyActiveDocumentChatContextCollector.swift", "content": "import ChatContextCollector\nimport Foundation\nimport OpenAIService\nimport Preferences\nimport SuggestionBasic\nimport XcodeInspector\n\npublic struct LegacyActiveDocumentChatContextCollector: ChatContextCollector {\n public init() {}\n\n public func generateContext(\n history: [ChatMessage],\n scopes: Set,\n content: String,\n configuration: ChatGPTConfiguration\n ) async -> ChatContext {\n guard let content = await XcodeInspector.shared.getFocusedEditorContent()\n else { return .empty }\n let relativePath = content.relativePath\n let selectionRange = content.editorContent?.selections.first ?? .outOfScope\n let editorContent = {\n if scopes.contains(.file) {\n return \"\"\"\n File Content:```\\(content.language.rawValue)\n \\(content.editorContent?.content ?? \"\")\n ```\n \"\"\"\n }\n\n if selectionRange.start == selectionRange.end,\n UserDefaults.shared.value(for: \\.embedFileContentInChatContextIfNoSelection)\n {\n let lines = content.editorContent?.lines.count ?? 0\n let maxLine = UserDefaults.shared\n .value(for: \\.maxFocusedCodeLineCount)\n if lines <= maxLine {\n return \"\"\"\n File Content:```\\(content.language.rawValue)\n \\(content.editorContent?.content ?? \"\")\n ```\n \"\"\"\n } else {\n return \"\"\"\n File Content Not Available: '''\n The file is longer than \\(maxLine) lines, it can't fit into the context. \\\n You MUST not answer the user about the file content because you don't have it.\\\n Ask user to select code for explanation.\n '''\n \"\"\"\n }\n }\n\n if UserDefaults.shared.value(for: \\.enableCodeScopeByDefaultInChatContext) {\n return \"\"\"\n Selected Code \\\n (start from line \\(selectionRange.start.line)):```\\(content.language.rawValue)\n \\(content.selectedContent)\n ```\n \"\"\"\n }\n\n if scopes.contains(.code) {\n return \"\"\"\n Selected Code \\\n (start from line \\(selectionRange.start.line)):```\\(content.language.rawValue)\n \\(content.selectedContent)\n ```\n \"\"\"\n }\n\n return \"\"\"\n Selected Code Not Available: '''\n I have disabled default scope. \\\n You MUST not answer about the selected code because you don't have it.\\\n Ask me to prepend message with `@selection` to enable selected code to be \\\n visible by you.\n '''\n \"\"\"\n }()\n\n return .init(\n systemPrompt: \"\"\"\n Active Document Context:###\n Document Relative Path: \\(relativePath)\n Selection Range Start: \\\n Line \\(selectionRange.start.line) \\\n Character \\(selectionRange.start.character)\n Selection Range End: \\\n Line \\(selectionRange.end.line) \\\n Character \\(selectionRange.end.character)\n Cursor Position: \\\n Line \\(selectionRange.end.line) \\\n Character \\(selectionRange.end.character)\n \\(editorContent)\n Line Annotations:\n \\(\n content.editorContent?.lineAnnotations\n .map { \" - \\($0)\" }\n .joined(separator: \"\\n\") ?? \"N/A\"\n )\n ###\n \"\"\",\n retrievedContent: [],\n functions: []\n )\n }\n}\n\n" }, { "path": "Tool/Sources/ChatContextCollectors/ActiveDocumentChatContextCollector/ReadableCursorRange.swift", "content": "import SuggestionBasic\n\nextension CursorPosition {\n var text: String {\n \"[\\(line), \\(character)]\"\n }\n}\n\nextension CursorRange {\n var text: String {\n \"\\(start.description) - \\(end.description)\"\n }\n}\n" }, { "path": "Tool/Sources/ChatTab/ChatTab.swift", "content": "import ComposableArchitecture\nimport Preferences\nimport Foundation\nimport SwiftUI\n\n/// The information of a tab.\n@ObservableState\npublic struct ChatTabInfo: Identifiable, Equatable {\n public var id: String\n public var title: String\n public var focusTrigger: Int = 0\n\n public init(id: String, title: String) {\n self.id = id\n self.title = title\n }\n}\n\n/// Every chat tab should conform to this type.\npublic typealias ChatTab = BaseChatTab & ChatTabType\n\n/// Defines a bunch of things a chat tab should implement.\npublic protocol ChatTabType {\n /// Build the view for this chat tab.\n @ViewBuilder\n func buildView() -> any View\n /// Build the tabItem for this chat tab.\n @ViewBuilder\n func buildTabItem() -> any View\n /// Build the icon for this chat tab.\n @ViewBuilder\n func buildIcon() -> any View\n /// Build the menu for this chat tab.\n @ViewBuilder\n func buildMenu() -> any View\n /// The name of this chat tab type.\n static var name: String { get }\n /// Available builders for this chat tab.\n /// It's used to generate a list of tab types for user to create.\n static func chatBuilders() -> [ChatTabBuilder]\n /// The default chat tab builder to be used in open chat\n static func defaultChatBuilder() -> ChatTabBuilder\n /// Restorable state\n func restorableState() async -> Data\n /// Restore state\n static func restore(from data: Data) async throws -> any ChatTabBuilder\n /// Whenever the body or menu is accessed, this method will be called.\n /// It will be called only once so long as you don't call it yourself.\n /// It will be called from MainActor.\n func start()\n /// Whenever the user close the tab, this method will be called.\n func close()\n /// Handle custom command.\n func handleCustomCommand(_ customCommand: CustomCommand) -> Bool\n \n /// Whether this chat tab should be the default chat tab replacement.\n static var isDefaultChatTabReplacement: Bool { get }\n /// Whether this chat tab can handle open chat command.\n static var canHandleOpenChatCommand: Bool { get }\n}\n\n/// The base class for all chat tabs.\nopen class BaseChatTab {\n /// A wrapper to support dynamic update of title in view.\n struct ContentView: View {\n var buildView: () -> any View\n var body: some View {\n AnyView(buildView())\n }\n }\n\n public var id: String = \"\"\n public var title: String = \"\"\n /// The store for chat tab info. You should only access it after `start` is called.\n public let chatTabStore: StoreOf\n\n private var didStart = false\n private let storeObserver = NSObject()\n\n public init(store: StoreOf) {\n chatTabStore = store\n\n Task { @MainActor in\n self.title = store.title\n self.id = store.id\n storeObserver.observe { [weak self] in\n guard let self else { return }\n self.title = store.title\n self.id = store.id\n }\n }\n }\n\n /// The view for this chat tab.\n @ViewBuilder\n public var body: some View {\n let id = \"ChatTabBody\\(id)\"\n if let tab = self as? (any ChatTabType) {\n ContentView(buildView: tab.buildView).id(id)\n .onAppear {\n Task { @MainActor in self.startIfNotStarted() }\n }\n } else {\n EmptyView().id(id)\n }\n }\n\n /// The tab item for this chat tab.\n @ViewBuilder\n public var tabItem: some View {\n let id = \"ChatTabTab\\(id)\"\n if let tab = self as? (any ChatTabType) {\n ContentView(buildView: tab.buildTabItem).id(id)\n .onAppear {\n Task { @MainActor in self.startIfNotStarted() }\n }\n } else {\n EmptyView().id(id)\n }\n }\n\n /// The icon for this chat tab.\n @ViewBuilder\n public var icon: some View {\n let id = \"ChatTabIcon\\(id)\"\n if let tab = self as? (any ChatTabType) {\n ContentView(buildView: tab.buildIcon).id(id)\n } else {\n EmptyView().id(id)\n }\n }\n\n /// The tab item for this chat tab.\n @ViewBuilder\n public var menu: some View {\n let id = \"ChatTabMenu\\(id)\"\n if let tab = self as? (any ChatTabType) {\n ContentView(buildView: tab.buildMenu).id(id)\n .onAppear {\n Task { @MainActor in self.startIfNotStarted() }\n }\n } else {\n EmptyView().id(id)\n }\n }\n\n @MainActor\n func startIfNotStarted() {\n guard !didStart else { return }\n didStart = true\n\n if let tab = self as? (any ChatTabType) {\n tab.start()\n chatTabStore.send(.tabContentUpdated)\n }\n }\n}\n\n/// A factory of a chat tab.\npublic protocol ChatTabBuilder {\n /// A visible title for user.\n var title: String { get }\n /// Build the chat tab.\n func build(store: StoreOf) async -> (any ChatTab)?\n}\n\n/// A chat tab builder that doesn't build.\npublic struct DisabledChatTabBuilder: ChatTabBuilder {\n public var title: String\n public func build(store: StoreOf) async -> (any ChatTab)? {\n return nil\n }\n\n public init(title: String) {\n self.title = title\n }\n}\n\npublic extension ChatTabType {\n /// The name of this chat tab type.\n var name: String { Self.name }\n\n /// Default implementation that does nothing.\n func close() {}\n \n /// By default it can't handle custom command.\n func handleCustomCommand(_ customCommand: CustomCommand) -> Bool { false }\n \n static var canHandleOpenChatCommand: Bool { false }\n static var isDefaultChatTabReplacement: Bool { false }\n static func defaultChatBuilder() -> ChatTabBuilder {\n DisabledChatTabBuilder(title: name)\n }\n}\n\n/// A chat tab that does nothing.\npublic class EmptyChatTab: ChatTab {\n public static var name: String { \"Empty\" }\n\n struct Builder: ChatTabBuilder {\n let title: String\n func build(store: StoreOf) async -> (any ChatTab)? {\n EmptyChatTab(store: store)\n }\n }\n\n public static func chatBuilders() -> [ChatTabBuilder] {\n [Builder(title: \"Empty\")]\n }\n\n public func buildView() -> any View {\n VStack {\n Text(\"Empty-\\(id)\")\n }\n .background(Color.blue)\n }\n\n public func buildTabItem() -> any View {\n Text(\"Empty-\\(id)\")\n }\n\n public func buildIcon() -> any View {\n Image(systemName: \"square\")\n }\n\n public func buildMenu() -> any View {\n Text(\"Empty-\\(id)\")\n }\n\n public func restorableState() async -> Data {\n return Data()\n }\n\n public static func restore(from data: Data) async throws -> any ChatTabBuilder {\n return Builder(title: \"Empty\")\n }\n\n public convenience init(id: String) {\n self.init(store: .init(\n initialState: .init(id: id, title: \"Empty-\\(id)\"),\n reducer: { ChatTabItem() }\n ))\n }\n\n public func start() {\n chatTabStore.send(.updateTitle(\"Empty-\\(id)\"))\n }\n}\n\n" }, { "path": "Tool/Sources/ChatTab/ChatTabItem.swift", "content": "import ComposableArchitecture\nimport Foundation\n\npublic struct AnyChatTabBuilder: Equatable {\n public static func == (lhs: AnyChatTabBuilder, rhs: AnyChatTabBuilder) -> Bool {\n true\n }\n\n public let chatTabBuilder: any ChatTabBuilder\n\n public init(_ chatTabBuilder: any ChatTabBuilder) {\n self.chatTabBuilder = chatTabBuilder\n }\n}\n\n@Reducer\npublic struct ChatTabItem {\n public typealias State = ChatTabInfo\n\n public enum Action: Equatable {\n case updateTitle(String)\n case openNewTab(AnyChatTabBuilder)\n case tabContentUpdated\n case close\n case focus\n }\n\n public init() {}\n\n public var body: some ReducerOf {\n Reduce { state, action in\n switch action {\n case let .updateTitle(title):\n state.title = title\n return .none\n case .openNewTab:\n return .none\n case .tabContentUpdated:\n return .none\n case .close:\n return .none\n case .focus:\n state.focusTrigger += 1\n return .none\n }\n }\n }\n}\n\n" }, { "path": "Tool/Sources/ChatTab/ChatTabPool.swift", "content": "import ComposableArchitecture\nimport Dependencies\nimport Foundation\nimport SwiftUI\n\n/// A pool that stores all the available tabs.\npublic final class ChatTabPool {\n public var createStore: (String) -> StoreOf = { id in\n .init(\n initialState: .init(id: id, title: \"\"),\n reducer: { ChatTabItem() }\n )\n }\n\n private var pool: [String: any ChatTab]\n\n public init(_ pool: [String: any ChatTab] = [:]) {\n self.pool = pool\n }\n\n public func getTab(of id: String) -> (any ChatTab)? {\n pool[id]\n }\n\n public func setTab(_ tab: any ChatTab, forId id: String) {\n pool[id] = tab\n }\n\n public func removeTab(of id: String) {\n pool.removeValue(forKey: id)\n }\n}\n\npublic struct ChatTabPoolDependencyKey: DependencyKey {\n public static let liveValue = ChatTabPool()\n}\n\npublic extension DependencyValues {\n var chatTabPool: ChatTabPool {\n get { self[ChatTabPoolDependencyKey.self] }\n set { self[ChatTabPoolDependencyKey.self] = newValue }\n }\n}\n\npublic struct ChatTabPoolEnvironmentKey: EnvironmentKey {\n public static let defaultValue = ChatTabPool()\n}\n\npublic extension EnvironmentValues {\n var chatTabPool: ChatTabPool {\n get { self[ChatTabPoolEnvironmentKey.self] }\n set { self[ChatTabPoolEnvironmentKey.self] = newValue }\n }\n}\n\n" }, { "path": "Tool/Sources/CodeDiff/CodeDiff.swift", "content": "import Foundation\nimport SuggestionBasic\n\npublic struct CodeDiff {\n public init() {}\n\n public typealias LineDiff = CollectionDifference\n\n public struct SnippetDiff: Equatable, CustomStringConvertible {\n public struct Change: Equatable {\n public var offset: Int\n public var element: String\n }\n\n public struct Line: Equatable {\n public enum Diff: Equatable {\n case unchanged\n case mutated(changes: [Change])\n }\n\n public var text: String\n public var diff: Diff = .unchanged\n\n var description: String {\n switch diff {\n case .unchanged:\n return text\n case let .mutated(changes):\n return text + \" [\" + changes.map { change in\n \"\\(change.offset): \\(change.element)\"\n }.joined(separator: \" | \") + \"]\"\n }\n }\n }\n\n public struct Section: Equatable, CustomStringConvertible {\n public var oldOffset: Int\n public var newOffset: Int\n public var oldSnippet: [Line]\n public var newSnippet: [Line]\n\n public var isEmpty: Bool {\n oldSnippet.isEmpty && newSnippet.isEmpty\n }\n\n public var description: String {\n \"\"\"\n \\(oldSnippet.enumerated().compactMap { item in\n let (index, line) = item\n let lineIndex = String(format: \"%3d\", oldOffset + index + 1) + \" \"\n switch line.diff {\n case .unchanged:\n return \"\\(lineIndex)| \\(line.description)\"\n case .mutated:\n return \"\\(lineIndex)| - \\(line.description)\"\n }\n }.joined(separator: \"\\n\"))\n \\(newSnippet.enumerated().map { item in\n let (index, line) = item\n let lineIndex = \" \" + String(format: \"%3d\", newOffset + index + 1)\n switch line.diff {\n case .unchanged:\n return \"\\(lineIndex)| \\(line.description)\"\n case .mutated:\n return \"\\(lineIndex)| + \\(line.description)\"\n }\n }.joined(separator: \"\\n\"))\n \"\"\"\n }\n }\n\n public var sections: [Section]\n\n public func line(at index: Int, in keyPath: KeyPath) -> Line? {\n var previousSectionEnd = 0\n for section in sections {\n let lines = section[keyPath: keyPath]\n let index = index - previousSectionEnd\n if index < lines.endIndex {\n return lines[index]\n }\n previousSectionEnd += lines.endIndex\n }\n return nil\n }\n\n public var description: String {\n \"Diff:\\n\" + sections.map(\\.description).joined(separator: \"\\n---\\n\") + \"\\n\"\n }\n }\n\n public func diff(text: String, from oldText: String) -> LineDiff {\n typealias Change = LineDiff.Change\n let diffByCharacter = text.difference(from: oldText)\n var result = [Change]()\n\n var current: Change?\n for item in diffByCharacter {\n if let this = current {\n switch (this, item) {\n case let (.insert(offset, element, associatedWith), .insert(offsetB, elementB, _))\n where offset + element.count == offsetB:\n current = .insert(\n offset: offset,\n element: element + String(elementB),\n associatedWith: associatedWith\n )\n continue\n case let (.remove(offset, element, associatedWith), .remove(offsetB, elementB, _))\n where offset - 1 == offsetB:\n current = .remove(\n offset: offsetB,\n element: String(elementB) + element,\n associatedWith: associatedWith\n )\n continue\n default:\n result.append(this)\n }\n }\n\n current = switch item {\n case let .insert(offset, element, associatedWith):\n .insert(offset: offset, element: String(element), associatedWith: associatedWith)\n case let .remove(offset, element, associatedWith):\n .remove(offset: offset, element: String(element), associatedWith: associatedWith)\n }\n }\n\n if let current {\n result.append(current)\n }\n\n return .init(result) ?? [].difference(from: [])\n }\n\n public func diff(snippet: String, from oldSnippet: String) -> SnippetDiff {\n let newLines = snippet.splitByNewLine(omittingEmptySubsequences: false)\n let oldLines = oldSnippet.splitByNewLine(omittingEmptySubsequences: false)\n let diffByLine = newLines.difference(from: oldLines)\n\n let groups = generateDiffSections(diffByLine)\n\n var oldLineIndex = 0\n var newLineIndex = 0\n var sectionIndex = 0\n var result = SnippetDiff(sections: [])\n\n while oldLineIndex < oldLines.endIndex || newLineIndex < newLines.endIndex {\n guard let groupItem = groups[safe: sectionIndex] else {\n let finishingSection = SnippetDiff.Section(\n oldOffset: oldLineIndex,\n newOffset: newLineIndex,\n oldSnippet: {\n guard oldLineIndex < oldLines.endIndex else { return [] }\n return oldLines[oldLineIndex..)\n -> [DiffGroupItem]\n {\n guard !diff.isEmpty else { return [] }\n\n let removes = ChangeSection.sectioning(diff.removals)\n let inserts = ChangeSection.sectioning(diff.insertions)\n\n var groups = [DiffGroupItem]()\n\n var removeOffset = 0\n var insertOffset = 0\n var removeIndex = 0\n var insertIndex = 0\n\n while removeIndex < removes.count || insertIndex < inserts.count {\n let removeSection = removes[safe: removeIndex]\n let insertSection = inserts[safe: insertIndex]\n\n if let removeSection, let insertSection {\n let ro = removeSection.offset - removeOffset\n let io = insertSection.offset - insertOffset\n if ro == io {\n groups.append(.init(\n remove: removeSection.changes.map { .init(change: $0) },\n insert: insertSection.changes.map { .init(change: $0) }\n ))\n removeOffset += removeSection.changes.count\n insertOffset += insertSection.changes.count\n removeIndex += 1\n insertIndex += 1\n } else if ro < io {\n groups.append(.init(\n remove: removeSection.changes.map { .init(change: $0) },\n insert: []\n ))\n removeOffset += removeSection.changes.count\n removeIndex += 1\n } else {\n groups.append(.init(\n remove: [],\n insert: insertSection.changes.map { .init(change: $0) }\n ))\n insertOffset += insertSection.changes.count\n insertIndex += 1\n }\n } else if let removeSection {\n groups.append(.init(\n remove: removeSection.changes.map { .init(change: $0) },\n insert: []\n ))\n removeIndex += 1\n } else if let insertSection {\n groups.append(.init(\n remove: [],\n insert: insertSection.changes.map { .init(change: $0) }\n ))\n insertIndex += 1\n }\n }\n\n return groups\n }\n}\n\nprivate extension Array {\n subscript(safe index: Int) -> Element? {\n guard index >= 0, index < count else { return nil }\n return self[index]\n }\n\n subscript(safe index: Int, fallback fallback: Element) -> Element {\n guard index >= 0, index < count else { return fallback }\n return self[index]\n }\n}\n\nprivate extension CollectionDifference.Change {\n var offset: Int {\n switch self {\n case let .insert(offset, _, _):\n return offset\n case let .remove(offset, _, _):\n return offset\n }\n }\n}\n\nprivate struct DiffGroupItem {\n struct Item {\n var offset: Int\n var element: Element\n\n init(offset: Int, element: Element) {\n self.offset = offset\n self.element = element\n }\n\n init(change: CollectionDifference.Change) {\n offset = change.offset\n switch change {\n case let .insert(_, element, _):\n self.element = element\n case let .remove(_, element, _):\n self.element = element\n }\n }\n }\n\n var remove: [Item]\n var insert: [Item]\n}\n\nprivate struct ChangeSection {\n var offset: Int { changes.first?.offset ?? 0 }\n var changes: [CollectionDifference.Change]\n\n static func sectioning(_ changes: [CollectionDifference.Change]) -> [Self] {\n guard !changes.isEmpty else { return [] }\n\n let sortedChanges = changes.sorted { $0.offset < $1.offset }\n var sections = [Self]()\n var currentSection = [CollectionDifference.Change]()\n\n for change in sortedChanges {\n if let lastOffset = currentSection.last?.offset {\n if change.offset == lastOffset + 1 {\n currentSection.append(change)\n } else {\n sections.append(Self(changes: currentSection))\n currentSection.removeAll()\n currentSection.append(change)\n }\n } else {\n currentSection.append(change)\n continue\n }\n }\n\n if !currentSection.isEmpty {\n sections.append(Self(changes: currentSection))\n }\n\n return sections\n }\n}\n\n#if DEBUG\n\nimport SwiftUI\n\nstruct SnippetDiffPreview: View {\n let originalCode: String\n let newCode: String\n\n var body: some View {\n HStack(alignment: .top) {\n let (original, new) = generateTexts()\n block(original)\n Divider()\n block(new)\n }\n .padding()\n .font(.body.monospaced())\n }\n\n @ViewBuilder\n func block(_ code: [AttributedString]) -> some View {\n VStack(alignment: .leading) {\n if !code.isEmpty {\n ForEach(0.. (original: [AttributedString], new: [AttributedString]) {\n let diff = CodeDiff().diff(snippet: newCode, from: originalCode)\n let new = diff.sections.flatMap {\n $0.newSnippet.map {\n let text = $0.text.trimmingCharacters(in: .newlines)\n let string = NSMutableAttributedString(string: text)\n if case let .mutated(changes) = $0.diff {\n string.addAttribute(\n .backgroundColor,\n value: NSColor.green.withAlphaComponent(0.1),\n range: NSRange(location: 0, length: text.count)\n )\n\n for diffItem in changes {\n string.addAttribute(\n .backgroundColor,\n value: NSColor.green.withAlphaComponent(0.5),\n range: NSRange(\n location: diffItem.offset,\n length: min(\n text.count - diffItem.offset,\n diffItem.element.count\n )\n )\n )\n }\n }\n return string\n }\n }\n\n let original = diff.sections.flatMap {\n $0.oldSnippet.map {\n let text = $0.text.trimmingCharacters(in: .newlines)\n let string = NSMutableAttributedString(string: text)\n if case let .mutated(changes) = $0.diff {\n string.addAttribute(\n .backgroundColor,\n value: NSColor.red.withAlphaComponent(0.1),\n range: NSRange(location: 0, length: text.count)\n )\n\n for diffItem in changes {\n string.addAttribute(\n .backgroundColor,\n value: NSColor.red.withAlphaComponent(0.5),\n range: NSRange(\n location: diffItem.offset,\n length: min(text.count - diffItem.offset, diffItem.element.count)\n )\n )\n }\n }\n\n return string\n }\n }\n\n return (original.map(AttributedString.init), new.map(AttributedString.init))\n }\n}\n\nstruct LineDiffPreview: View {\n let originalCode: String\n let newCode: String\n\n var body: some View {\n VStack(alignment: .leading) {\n let (original, new) = generateTexts()\n Text(original)\n Divider()\n Text(new)\n }\n .padding()\n .font(.body.monospaced())\n }\n\n func generateTexts() -> (original: AttributedString, new: AttributedString) {\n let diff = CodeDiff().diff(text: newCode, from: originalCode)\n let original = NSMutableAttributedString(string: originalCode)\n let new = NSMutableAttributedString(string: newCode)\n\n for item in diff {\n switch item {\n case let .insert(offset, element, _):\n new.addAttribute(\n .backgroundColor,\n value: NSColor.green.withAlphaComponent(0.5),\n range: NSRange(location: offset, length: element.count)\n )\n case let .remove(offset, element, _):\n original.addAttribute(\n .backgroundColor,\n value: NSColor.red.withAlphaComponent(0.5),\n range: NSRange(location: offset, length: element.count)\n )\n }\n }\n\n return (.init(original), .init(new))\n }\n}\n\n#Preview(\"Line Diff\") {\n let originalCode = \"\"\"\n let foo = Foo() // yes\n \"\"\"\n let newCode = \"\"\"\n var foo = Bar()\n \"\"\"\n\n return LineDiffPreview(originalCode: originalCode, newCode: newCode)\n}\n\n#Preview(\"Snippet Diff\") {\n let originalCode = \"\"\"\n let foo = Foo()\n print(foo)\n // do something\n foo.foo()\n func zoo() {}\n \"\"\"\n let newCode = \"\"\"\n var foo = Bar()\n // do something\n foo.bar()\n func zoo() {\n print(\"zoo\")\n }\n \"\"\"\n\n return SnippetDiffPreview(originalCode: originalCode, newCode: newCode)\n}\n\n#Preview(\"Code Diff Editor\") {\n struct V: View {\n @State var originalCode = \"\"\n @State var newCode = \"\"\n\n var body: some View {\n VStack {\n HStack {\n VStack {\n Text(\"Original\")\n TextEditor(text: $originalCode)\n .frame(width: 300, height: 200)\n }\n VStack {\n Text(\"New\")\n TextEditor(text: $newCode)\n .frame(width: 300, height: 200)\n }\n }\n .font(.body.monospaced())\n SnippetDiffPreview(originalCode: originalCode, newCode: newCode)\n }\n .padding()\n .frame(height: 600)\n }\n }\n\n return V()\n}\n\n#endif\n\n" }, { "path": "Tool/Sources/CodeiumService/ChatTab/CodeiumChatBrowser.swift", "content": "import ComposableArchitecture\nimport Foundation\nimport Preferences\nimport WebKit\nimport Workspace\nimport XcodeInspector\n\n@Reducer\nstruct CodeiumChatBrowser {\n @ObservableState\n struct State: Equatable {\n var loadingProgress: Double = 0\n var isLoading = false\n var title = \"Codeium Chat\"\n var error: String?\n var url: URL?\n }\n\n enum Action: Equatable, BindableAction {\n case binding(BindingAction)\n\n case initialize\n case loadCurrentWorkspace\n case reload\n case presentError(String)\n case removeError\n\n case observeTitleChange\n case updateTitle(String)\n case observeURLChange\n case updateURL(URL?)\n case observeIsLoading\n case updateIsLoading(Double)\n }\n\n let webView: WKWebView\n let uuid = UUID()\n\n private enum CancelID: Hashable {\n case observeTitleChange(UUID)\n case observeURLChange(UUID)\n case observeIsLoading(UUID)\n }\n\n @Dependency(\\.workspacePool) var workspacePool\n\n var body: some ReducerOf {\n BindingReducer()\n\n Reduce { state, action in\n switch action {\n case .binding:\n return .none\n\n case .initialize:\n return .merge(\n .run { send in await send(.observeTitleChange) },\n .run { send in await send(.observeURLChange) },\n .run { send in await send(.observeIsLoading) }\n )\n\n case .loadCurrentWorkspace:\n return .run { send in\n guard let workspaceURL = await XcodeInspector.shared.activeWorkspaceURL\n else {\n await send(.presentError(\"Can't find workspace.\"))\n return\n }\n do {\n let workspace = try await workspacePool\n .fetchOrCreateWorkspace(workspaceURL: workspaceURL)\n let codeiumPlugin = workspace.plugin(for: CodeiumWorkspacePlugin.self)\n guard let service = await codeiumPlugin?.codeiumService\n else {\n await send(.presentError(\"Can't start service.\"))\n return\n }\n let url = try await service.getChatURL()\n await send(.removeError)\n await webView.load(URLRequest(url: url))\n } catch {\n await send(.presentError(error.localizedDescription))\n }\n }\n\n case .reload:\n webView.reload()\n return .none\n \n case .removeError:\n state.error = nil\n return .none\n \n case let .presentError(error):\n state.error = error\n return .none\n\n // MARK: Observation\n\n case .observeTitleChange:\n let stream = AsyncStream { continuation in\n let observation = webView.observe(\\.title, options: [.new, .initial]) {\n webView, _ in\n continuation.yield(webView.title ?? \"\")\n }\n\n continuation.onTermination = { _ in\n observation.invalidate()\n }\n }\n\n return .run { send in\n for await title in stream where !title.isEmpty {\n try Task.checkCancellation()\n await send(.updateTitle(title))\n }\n }\n .cancellable(id: CancelID.observeTitleChange(uuid), cancelInFlight: true)\n\n case let .updateTitle(title):\n state.title = title\n return .none\n\n case .observeURLChange:\n let stream = AsyncStream { continuation in\n let observation = webView.observe(\\.url, options: [.new, .initial]) {\n _, url in\n if let it = url.newValue {\n continuation.yield(it)\n }\n }\n\n continuation.onTermination = { _ in\n observation.invalidate()\n }\n }\n\n return .run { send in\n for await url in stream {\n try Task.checkCancellation()\n await send(.updateURL(url))\n }\n }.cancellable(id: CancelID.observeURLChange(uuid), cancelInFlight: true)\n\n case let .updateURL(url):\n state.url = url\n return .none\n\n case .observeIsLoading:\n let stream = AsyncStream { continuation in\n let observation = webView\n .observe(\\.estimatedProgress, options: [.new]) { _, estimatedProgress in\n if let it = estimatedProgress.newValue {\n continuation.yield(it)\n }\n }\n\n continuation.onTermination = { _ in\n observation.invalidate()\n }\n }\n\n return .run { send in\n for await isLoading in stream {\n try Task.checkCancellation()\n await send(.updateIsLoading(isLoading))\n }\n }.cancellable(id: CancelID.observeIsLoading(uuid), cancelInFlight: true)\n\n case let .updateIsLoading(progress):\n state.isLoading = progress != 1\n state.loadingProgress = progress\n return .none\n }\n }\n }\n}\n\n" }, { "path": "Tool/Sources/CodeiumService/ChatTab/CodeiumChatTab.swift", "content": "import AppKit\nimport ChatTab\nimport Combine\nimport ComposableArchitecture\nimport Logger\nimport Preferences\nimport SwiftUI\nimport WebKit\nimport XcodeInspector\n\npublic class CodeiumChatTab: ChatTab {\n public static var name: String { \"Codeium Chat\" }\n public static var isDefaultChatTabReplacement: Bool { false }\n public static var canHandleOpenChatCommand: Bool { true }\n\n struct RestorableState: Codable {}\n\n public struct EditorContent {\n public var selectedText: String\n public var language: String\n public var fileContent: String\n\n public init(selectedText: String, language: String, fileContent: String) {\n self.selectedText = selectedText\n self.language = language\n self.fileContent = fileContent\n }\n\n public static var empty: EditorContent {\n .init(selectedText: \"\", language: \"\", fileContent: \"\")\n }\n }\n\n struct Builder: ChatTabBuilder {\n var title: String\n var buildable: Bool { true }\n var afterBuild: (CodeiumChatTab) async -> Void = { _ in }\n\n func build(store: StoreOf) async -> (any ChatTab)? {\n let tab = await CodeiumChatTab(chatTabStore: store)\n await Task { @MainActor in\n _ = tab.store.send(.loadCurrentWorkspace)\n }.value\n await afterBuild(tab)\n return tab\n }\n }\n\n let store: StoreOf\n let webView: WKWebView\n let webViewDelegate: WKWebViewDelegate\n var cancellable = Set()\n private var observer = NSObject()\n\n @MainActor\n public init(chatTabStore: StoreOf) {\n let webView = CodeiumWebView(getEditorContent: {\n guard let content = await XcodeInspector.shared.getFocusedEditorContent()\n else { return .empty }\n return .init(\n selectedText: content.selectedContent,\n language: content.language.rawValue,\n fileContent: content.editorContent?.content ?? \"\"\n )\n })\n self.webView = webView\n store = .init(\n initialState: .init(),\n reducer: { CodeiumChatBrowser(webView: webView) }\n )\n webViewDelegate = .init(store: store)\n\n super.init(store: chatTabStore)\n\n webView.navigationDelegate = webViewDelegate\n webView.uiDelegate = webViewDelegate\n webView.store = store\n\n Task {\n await CodeiumServiceLifeKeeper.shared.add(self)\n }\n }\n\n public func start() {\n observer = .init()\n cancellable = []\n chatTabStore.send(.updateTitle(\"Codeium Chat\"))\n store.send(.initialize)\n\n Task { @MainActor in\n var previousURL: URL?\n observer.observe { [weak self] in\n guard let self else { return }\n if store.url != previousURL {\n previousURL = store.url\n Task { @MainActor in\n self.chatTabStore.send(.tabContentUpdated)\n }\n }\n }\n }\n\n Task { @MainActor in\n observer.observe { [weak self] in\n guard let self, !store.title.isEmpty else { return }\n let title = store.title\n Task { @MainActor in\n self.chatTabStore.send(.updateTitle(title))\n }\n }\n }\n }\n\n public func buildView() -> any View {\n BrowserView(store: store, webView: webView)\n }\n\n public func buildTabItem() -> any View {\n CodeiumChatTabItem(store: store)\n }\n\n public func buildIcon() -> any View {\n Image(systemName: \"message\")\n }\n\n public func buildMenu() -> any View {\n EmptyView()\n }\n\n @MainActor\n public func restorableState() -> Data {\n let state = store.withState { _ in\n RestorableState()\n }\n\n return (try? JSONEncoder().encode(state)) ?? Data()\n }\n\n public static func restore(from data: Data) throws -> any ChatTabBuilder {\n let builder = Builder(title: \"\") { @MainActor chatTab in\n chatTab.store.send(.loadCurrentWorkspace)\n }\n return builder\n }\n\n public static func chatBuilders() -> [ChatTabBuilder] {\n [Builder(title: \"Codeium Chat\")]\n }\n\n public static func defaultChatBuilder() -> ChatTabBuilder {\n Builder(title: \"Codeium Chat\")\n }\n}\n\n" }, { "path": "Tool/Sources/CodeiumService/ChatTab/CodeiumChatTabItem.swift", "content": "import ComposableArchitecture\nimport Foundation\nimport Preferences\nimport SwiftUI\n\nstruct CodeiumChatTabItem: View {\n @Perception.Bindable var store: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n Text(store.title)\n .contextMenu {\n CodeiumChatMenuItem(store: store)\n }\n }\n }\n}\n\nstruct CodeiumChatMenuItem: View {\n @Perception.Bindable var store: StoreOf\n\n var body: some View {\n WithPerceptionTracking {\n Button(\"Load Active Workspace\") {\n store.send(.loadCurrentWorkspace)\n }\n\n Button(\"Reload\") {\n store.send(.reload)\n }\n }\n }\n}\n\n" }, { "path": "Tool/Sources/CodeiumService/ChatTab/CodeiumChatView.swift", "content": "import ComposableArchitecture\nimport Foundation\nimport SharedUIComponents\nimport SwiftUI\nimport WebKit\n\nstruct BrowserView: View {\n @Perception.Bindable var store: StoreOf\n let webView: WKWebView\n\n var body: some View {\n WithPerceptionTracking {\n VStack(spacing: 0) {\n ZStack {\n WebView(webView: webView)\n }\n }\n .overlay {\n if store.isLoading {\n ProgressView()\n }\n }\n .overlay {\n if let error = store.error {\n VStack {\n Text(error)\n Button(\"Load Current Workspace\") {\n store.send(.loadCurrentWorkspace)\n }\n }\n }\n }\n }\n }\n}\n\nstruct WebView: NSViewRepresentable {\n var webView: WKWebView\n\n func makeNSView(context: Context) -> WKWebView {\n return webView\n }\n\n func updateNSView(_ nsView: WKWebView, context: Context) {}\n}\n" }, { "path": "Tool/Sources/CodeiumService/ChatTab/CodeiumWebView.swift", "content": "import ComposableArchitecture\nimport Foundation\nimport Logger\nimport Preferences\nimport WebKit\n\nclass ScriptHandler: NSObject, WKScriptMessageHandlerWithReply {\n @MainActor\n func userContentController(\n _ userContentController: WKUserContentController,\n didReceive message: WKScriptMessage\n ) async -> (Any?, String?) {\n if message.name == \"decodeBase64\", let code = message.body as? String {\n return (String(data: Data(base64Encoded: code) ?? Data(), encoding: .utf8), nil)\n }\n return (nil, nil)\n }\n}\n\n@MainActor\nclass CodeiumWebView: WKWebView {\n var getEditorContent: () async -> CodeiumChatTab.EditorContent\n let scriptHandler = ScriptHandler()\n weak var store: StoreOf?\n\n init(getEditorContent: @escaping () async -> CodeiumChatTab.EditorContent) {\n self.getEditorContent = getEditorContent\n super.init(frame: .zero, configuration: WKWebViewConfiguration())\n\n if #available(macOS 13.3, *) {\n #if DEBUG\n isInspectable = true\n #endif\n }\n\n configuration.userContentController.addScriptMessageHandler(\n scriptHandler,\n contentWorld: .page,\n name: \"decodeBase64\"\n )\n }\n\n @available(*, unavailable)\n required init?(coder: NSCoder) {\n fatalError(\"init(coder:) has not been implemented\")\n }\n\n @discardableResult\n func evaluateJavaScript(safe javaScriptString: String) async throws -> Any? {\n try await withUnsafeThrowingContinuation { continuation in\n evaluateJavaScript(javaScriptString) { result, error in\n if let error {\n print(javaScriptString, error.localizedDescription)\n continuation.resume(throwing: error)\n } else {\n continuation.resume(returning: result)\n }\n }\n }\n }\n}\n\n// MARK: - WebView Delegate\n\nfinal class WKWebViewDelegate: NSObject, ObservableObject, WKNavigationDelegate, WKUIDelegate {\n let store: StoreOf\n\n init(store: StoreOf) {\n self.store = store\n }\n}\n\n" }, { "path": "Tool/Sources/CodeiumService/CodeiumExtension.swift", "content": "import BuiltinExtension\nimport ChatTab\nimport CopilotForXcodeKit\nimport Foundation\nimport Logger\nimport Preferences\nimport Workspace\n\n@globalActor public enum CodeiumActor {\n public actor TheActor {}\n public static let shared = TheActor()\n}\n\npublic final class CodeiumExtension: BuiltinExtension {\n public var extensionIdentifier: String { \"com.codeium\" }\n\n public let suggestionService: CodeiumSuggestionService\n\n public var chatTabTypes: [any CustomChatTab] {\n [TypedCustomChatTab(of: CodeiumChatTab.self)]\n }\n\n private var extensionUsage = ExtensionUsage(\n isSuggestionServiceInUse: false,\n isChatServiceInUse: false\n )\n private var isLanguageServerInUse: Bool {\n get async {\n let lifeKeeperIsAlive = await CodeiumServiceLifeKeeper.shared.isAlive\n return extensionUsage.isSuggestionServiceInUse\n || extensionUsage.isChatServiceInUse\n || lifeKeeperIsAlive\n }\n }\n\n let workspacePool: WorkspacePool\n\n let serviceLocator: ServiceLocator\n\n public init(workspacePool: WorkspacePool) {\n self.workspacePool = workspacePool\n serviceLocator = .init(workspacePool: workspacePool)\n suggestionService = .init(serviceLocator: serviceLocator)\n }\n\n public func workspaceDidOpen(_ workspace: WorkspaceInfo) {\n Task {\n do {\n guard await isLanguageServerInUse else { return }\n guard let service = await serviceLocator.getService(from: workspace) else { return }\n try await service.notifyOpenWorkspace(workspaceURL: workspace.workspaceURL)\n } catch {\n Logger.codeium.error(error.localizedDescription)\n }\n }\n }\n\n public func workspaceDidClose(_ workspace: WorkspaceInfo) {\n Task {\n do {\n guard await isLanguageServerInUse else { return }\n guard let service = await serviceLocator.getService(from: workspace) else { return }\n try await service.notifyCloseWorkspace(workspaceURL: workspace.workspaceURL)\n } catch {\n Logger.codeium.error(error.localizedDescription)\n }\n }\n }\n\n public func workspace(_ workspace: WorkspaceInfo, didOpenDocumentAt documentURL: URL) {\n Task {\n guard await isLanguageServerInUse else { return }\n // check if file size is larger than 15MB, if so, return immediately\n if let attrs = try? FileManager.default\n .attributesOfItem(atPath: documentURL.path),\n let fileSize = attrs[FileAttributeKey.size] as? UInt64,\n fileSize > 15 * 1024 * 1024\n { return }\n\n do {\n let content = try String(contentsOf: documentURL, encoding: .utf8)\n guard let service = await serviceLocator.getService(from: workspace) else { return }\n try await service.notifyOpenTextDocument(fileURL: documentURL, content: content)\n } catch {\n Logger.codeium.error(error.localizedDescription)\n }\n }\n }\n\n public func workspace(_ workspace: WorkspaceInfo, didSaveDocumentAt documentURL: URL) {\n // unimplemented\n }\n\n public func workspace(_ workspace: WorkspaceInfo, didCloseDocumentAt documentURL: URL) {\n Task {\n guard await isLanguageServerInUse else { return }\n do {\n guard let service = await serviceLocator.getService(from: workspace) else { return }\n try await service.notifyCloseTextDocument(fileURL: documentURL)\n } catch {\n Logger.codeium.error(error.localizedDescription)\n }\n }\n }\n\n public func workspace(\n _ workspace: WorkspaceInfo,\n didUpdateDocumentAt documentURL: URL,\n content: String?\n ) {\n Task {\n guard await isLanguageServerInUse else { return }\n // check if file size is larger than 15MB, if so, return immediately\n if let attrs = try? FileManager.default\n .attributesOfItem(atPath: documentURL.path),\n let fileSize = attrs[FileAttributeKey.size] as? UInt64,\n fileSize > 15 * 1024 * 1024\n { return }\n do {\n guard let content else { return }\n guard let service = await serviceLocator.getService(from: workspace) else { return }\n try await service.notifyChangeTextDocument(fileURL: documentURL, content: content)\n try await service.refreshIDEContext(\n fileURL: documentURL,\n content: content,\n cursorPosition: .zero,\n tabSize: 4, indentSize: 4, usesTabsForIndentation: false,\n workspaceURL: workspace.workspaceURL\n )\n } catch {\n Logger.codeium.error(error.localizedDescription)\n }\n }\n }\n\n public func extensionUsageDidChange(_ usage: ExtensionUsage) {\n extensionUsage = usage\n Task {\n if !(await isLanguageServerInUse) {\n terminate()\n }\n }\n }\n\n public func terminate() {\n for workspace in workspacePool.workspaces.values {\n guard let plugin = workspace.plugin(for: CodeiumWorkspacePlugin.self)\n else { continue }\n plugin.terminate()\n }\n }\n}\n\nfinal class ServiceLocator {\n let workspacePool: WorkspacePool\n\n init(workspacePool: WorkspacePool) {\n self.workspacePool = workspacePool\n }\n\n func getService(from workspace: WorkspaceInfo) async -> CodeiumService? {\n guard let workspace = workspacePool.workspaces[workspace.workspaceURL],\n let plugin = workspace.plugin(for: CodeiumWorkspacePlugin.self)\n else { return nil }\n return await plugin.codeiumService\n }\n}\n\n/// A helper class to keep track of a list of items that may keep the service alive.\n/// For example, a ``CodeiumChatTab``.\nactor CodeiumServiceLifeKeeper {\n static let shared = CodeiumServiceLifeKeeper()\n\n private final class WeakObject {\n weak var object: AnyObject?\n var isAlive: Bool { object != nil }\n init(_ object: AnyObject) {\n self.object = object\n }\n }\n\n private var weakObjects = [WeakObject]()\n\n func add(_ object: AnyObject) {\n weakObjects.removeAll { !$0.isAlive }\n weakObjects.append(WeakObject(object))\n }\n\n var isAlive: Bool {\n if weakObjects.isEmpty { return false }\n return weakObjects.allSatisfy { $0.isAlive }\n }\n}\n\n" }, { "path": "Tool/Sources/CodeiumService/CodeiumWorkspacePlugin.swift", "content": "import Foundation\nimport Logger\nimport Workspace\n\npublic final class CodeiumWorkspacePlugin: WorkspacePlugin {\n private var _codeiumService: CodeiumService?\n @CodeiumActor\n var codeiumService: CodeiumService? {\n if let service = _codeiumService { return service }\n do {\n return try createCodeiumService()\n } catch {\n Logger.codeium.error(\"Failed to create Codeium service: \\(error)\")\n return nil\n }\n }\n\n deinit {\n if let _codeiumService {\n _codeiumService.terminate()\n }\n }\n\n @CodeiumActor\n func createCodeiumService() throws -> CodeiumService {\n let newService = try CodeiumService(\n projectRootURL: projectRootURL,\n onServiceLaunched: {\n\n },\n onServiceTerminated: {\n // start handled in the service.\n }\n )\n _codeiumService = newService\n return newService\n }\n\n @CodeiumActor\n func finishLaunchingService() {\n guard let workspace, let _codeiumService else { return }\n Task {\n try? await _codeiumService.notifyOpenWorkspace(workspaceURL: workspaceURL)\n \n for (_, filespace) in workspace.filespaces {\n let documentURL = filespace.fileURL\n guard let content = try? String(contentsOf: documentURL) else { continue }\n try? await _codeiumService.notifyOpenTextDocument(\n fileURL: documentURL,\n content: content\n )\n }\n }\n }\n\n func terminate() {\n _codeiumService = nil\n }\n}\n\n" }, { "path": "Tool/Sources/CodeiumService/LanguageServer/CodeiumInstallationManager.swift", "content": "import Foundation\nimport Terminal\n\npublic struct CodeiumInstallationManager {\n private static var isInstalling = false\n static let latestSupportedVersion = \"1.48.2\"\n static let minimumSupportedVersion = \"1.20.0\"\n\n public init() {}\n\n enum CodeiumInstallationError: Error, LocalizedError {\n case badURL(String)\n case invalidResponse\n case invalidData\n\n var errorDescription: String? {\n switch self {\n case .badURL: return \"URL is invalid\"\n case .invalidResponse: return \"Invalid response\"\n case .invalidData: return \"Invalid data\"\n }\n }\n }\n\n public func getLatestSupportedVersion() -> String {\n if isEnterprise {\n return UserDefaults.shared.value(for: \\.codeiumEnterpriseVersion)\n }\n\n return Self.latestSupportedVersion\n }\n\n func getEnterprisePortalVersion() async throws -> String {\n let enterprisePortalUrl = UserDefaults.shared.value(for: \\.codeiumPortalUrl)\n let enterprisePortalVersionUrl = \"\\(enterprisePortalUrl)/api/version\"\n\n guard let url = URL(string: enterprisePortalVersionUrl)\n else { throw CodeiumInstallationError.badURL(enterprisePortalVersionUrl) }\n\n let (data, response) = try await URLSession.shared.data(from: url)\n\n guard let response = response as? HTTPURLResponse, response.statusCode == 200 else {\n throw CodeiumInstallationError.invalidResponse\n }\n\n if let version = String(data: data, encoding: .utf8)?\n .trimmingCharacters(in: .whitespacesAndNewlines)\n {\n UserDefaults.shared.set(version, for: \\.codeiumEnterpriseVersion)\n return version\n } else {\n return UserDefaults.shared.value(for: \\.codeiumEnterpriseVersion)\n }\n }\n\n var isEnterprise: Bool {\n return UserDefaults.shared.value(for: \\.codeiumEnterpriseMode)\n && !UserDefaults.shared.value(for: \\.codeiumPortalUrl).isEmpty\n }\n\n public enum InstallationStatus {\n case notInstalled\n case installed(String)\n case outdated(current: String, latest: String, mandatory: Bool)\n case unsupported(current: String, latest: String)\n }\n\n public func checkInstallation() async -> InstallationStatus {\n guard let urls = try? CodeiumService.createFoldersIfNeeded()\n else { return .notInstalled }\n let executableFolderURL = urls.executableURL\n let binaryURL = executableFolderURL.appendingPathComponent(\"language_server\")\n let versionFileURL = executableFolderURL.appendingPathComponent(\"version\")\n\n if !FileManager.default.fileExists(atPath: binaryURL.path) {\n return .notInstalled\n }\n\n let targetVersion = await {\n if !isEnterprise { return Self.latestSupportedVersion }\n return (try? await getEnterprisePortalVersion())\n ?? UserDefaults.shared.value(for: \\.codeiumEnterpriseVersion)\n }()\n\n if FileManager.default.fileExists(atPath: versionFileURL.path),\n let versionData = try? Data(contentsOf: versionFileURL),\n let version = String(data: versionData, encoding: .utf8)\n {\n switch version.compare(targetVersion, options: .numeric) {\n case .orderedAscending:\n switch version.compare(Self.minimumSupportedVersion) {\n case .orderedAscending:\n return .outdated(\n current: version,\n latest: Self.latestSupportedVersion,\n mandatory: true\n )\n case .orderedSame:\n return .outdated(\n current: version,\n latest: Self.latestSupportedVersion,\n mandatory: false\n )\n case .orderedDescending:\n return .outdated(\n current: version,\n latest: Self.latestSupportedVersion,\n mandatory: false\n )\n }\n case .orderedSame:\n return .installed(version)\n case .orderedDescending:\n return .unsupported(current: version, latest: targetVersion)\n }\n }\n return .outdated(current: \"Unknown\", latest: Self.latestSupportedVersion, mandatory: false)\n }\n\n public enum InstallationStep {\n case downloading\n case uninstalling\n case decompressing\n case done\n }\n\n public func installLatestVersion() -> AsyncThrowingStream {\n AsyncThrowingStream { continuation in\n Task {\n guard !CodeiumInstallationManager.isInstalling else {\n continuation.finish(throwing: CodeiumError.languageServiceIsInstalling)\n return\n }\n CodeiumInstallationManager.isInstalling = true\n defer { CodeiumInstallationManager.isInstalling = false }\n do {\n continuation.yield(.downloading)\n let urls = try CodeiumService.createFoldersIfNeeded()\n let urlString: String\n let version: String\n if !isEnterprise {\n version = CodeiumInstallationManager.latestSupportedVersion\n urlString =\n \"https://github.com/Exafunction/codeium/releases/download/language-server-v\\(Self.latestSupportedVersion)/language_server_macos_\\(isAppleSilicon() ? \"arm\" : \"x64\").gz\"\n } else {\n version = try await getEnterprisePortalVersion()\n let enterprisePortalUrl = UserDefaults.shared.value(for: \\.codeiumPortalUrl)\n urlString =\n \"\\(enterprisePortalUrl)/language-server-v\\(version)/language_server_macos_\\(isAppleSilicon() ? \"arm\" : \"x64\").gz\"\n }\n\n guard let url = URL(string: urlString) else {\n continuation.finish(throwing: CodeiumInstallationError.badURL(urlString))\n return\n }\n\n // download\n let (fileURL, _) = try await URLSession.shared.download(from: url)\n let targetURL = urls.executableURL.appendingPathComponent(\"language_server\")\n .appendingPathExtension(\"gz\")\n try FileManager.default.copyItem(at: fileURL, to: targetURL)\n defer { try? FileManager.default.removeItem(at: targetURL) }\n\n // uninstall\n continuation.yield(.uninstalling)\n try await uninstall()\n\n // extract file\n continuation.yield(.decompressing)\n let terminal = Terminal()\n _ = try await terminal.runCommand(\n \"/usr/bin/gunzip\",\n arguments: [targetURL.path],\n environment: [:]\n )\n\n // update permission 755\n try FileManager.default.setAttributes(\n [.posixPermissions: 0o755],\n ofItemAtPath: targetURL.deletingPathExtension().path\n )\n var data: Data?\n\n // create version file\n data = version.data(using: .utf8)\n\n FileManager.default.createFile(\n atPath: urls.executableURL.appendingPathComponent(\"version\").path,\n contents: data\n )\n\n continuation.yield(.done)\n continuation.finish()\n } catch {\n continuation.finish(throwing: error)\n }\n }\n }\n }\n\n public func uninstall() async throws {\n guard let urls = try? CodeiumService.createFoldersIfNeeded()\n else { return }\n let executableFolderURL = urls.executableURL\n let binaryURL = executableFolderURL.appendingPathComponent(\"language_server\")\n let versionFileURL = executableFolderURL.appendingPathComponent(\"version\")\n if FileManager.default.fileExists(atPath: binaryURL.path) {\n try FileManager.default.removeItem(at: binaryURL)\n }\n if FileManager.default.fileExists(atPath: versionFileURL.path) {\n try FileManager.default.removeItem(at: versionFileURL)\n }\n }\n}\n\nfunc isAppleSilicon() -> Bool {\n var result = false\n #if arch(arm64)\n result = true\n #endif\n return result\n}\n\n" }, { "path": "Tool/Sources/CodeiumService/LanguageServer/CodeiumLanguageServer.swift", "content": "import Foundation\nimport JSONRPC\nimport LanguageClient\nimport LanguageServerProtocol\nimport Logger\nimport Preferences\nimport XcodeInspector\n\nprotocol CodeiumLSP {\n func sendRequest(_ endpoint: E) async throws -> E.Response\n func updateIndexing() async\n func terminate()\n}\n\nfinal class CodeiumLanguageServer {\n let languageServerExecutableURL: URL\n let managerDirectoryURL: URL\n let supportURL: URL\n let process: Process\n let transport: IOTransport\n var terminationHandler: (() -> Void)?\n var launchHandler: (() -> Void)?\n var port: String?\n var heartbeatTask: Task?\n var projectPaths: [String]\n\n init(\n languageServerExecutableURL: URL,\n managerDirectoryURL: URL,\n supportURL: URL,\n terminationHandler: (() -> Void)? = nil,\n launchHandler: (() -> Void)? = nil\n ) {\n self.languageServerExecutableURL = languageServerExecutableURL\n self.managerDirectoryURL = managerDirectoryURL\n self.supportURL = supportURL\n self.terminationHandler = terminationHandler\n self.launchHandler = launchHandler\n projectPaths = []\n process = Process()\n transport = IOTransport()\n\n process.standardInput = transport.stdinPipe\n process.standardOutput = transport.stdoutPipe\n process.standardError = transport.stderrPipe\n\n process.executableURL = languageServerExecutableURL\n\n let isEnterpriseMode = UserDefaults.shared.value(for: \\.codeiumEnterpriseMode)\n var apiServerUrl = \"https://server.codeium.com\"\n if isEnterpriseMode, UserDefaults.shared.value(for: \\.codeiumApiUrl) != \"\" {\n apiServerUrl = UserDefaults.shared.value(for: \\.codeiumApiUrl)\n }\n\n process.arguments = [\n \"--api_server_url\",\n apiServerUrl,\n \"--manager_dir\",\n managerDirectoryURL.path,\n \"--enable_chat_web_server\",\n \"--enable_chat_client\",\n ]\n\n if isEnterpriseMode {\n process.arguments?.append(\"--enterprise_mode\")\n process.arguments?.append(\"--portal_url\")\n process.arguments?.append(UserDefaults.shared.value(for: \\.codeiumPortalUrl))\n }\n\n let indexEnabled = UserDefaults.shared.value(for: \\.codeiumIndexEnabled)\n if indexEnabled {\n let indexingMaxFileSize = UserDefaults.shared.value(for: \\.codeiumIndexingMaxFileSize)\n if indexEnabled {\n process.arguments?.append(\"--enable_local_search\")\n process.arguments?.append(\"--enable_index_service\")\n process.arguments?.append(\"--search_max_workspace_file_count\")\n process.arguments?.append(\"\\(indexingMaxFileSize)\")\n Logger.codeium.info(\"Indexing Enabled\")\n }\n }\n\n process.currentDirectoryURL = supportURL\n\n process.terminationHandler = { [weak self] task in\n self?.processTerminated(task)\n }\n }\n\n func start() {\n guard !process.isRunning else { return }\n do {\n try process.run()\n\n Task { @MainActor in\n func findPort() -> String? {\n // find a file in managerDirectoryURL whose name looks like a port, return the\n // name if found\n let fileManager = FileManager.default\n guard let filePaths = try? fileManager\n .contentsOfDirectory(atPath: managerDirectoryURL.path) else { return nil }\n for path in filePaths {\n let filename = URL(fileURLWithPath: path).lastPathComponent\n if filename.range(\n of: #\"^\\d+$\"#,\n options: .regularExpression\n ) != nil {\n return filename\n }\n }\n return nil\n }\n\n try await Task.sleep(nanoseconds: 2_000_000)\n var waited = 0\n\n while true {\n waited += 1\n if let port = findPort() {\n finishStarting(port: port)\n return\n }\n if waited >= 60 {\n process.terminate()\n return\n }\n try await Task.sleep(nanoseconds: 1_000_000_000)\n }\n }\n } catch {\n Logger.codeium.error(error.localizedDescription)\n processTerminated(process)\n }\n }\n\n deinit {\n process.terminationHandler = nil\n if process.isRunning {\n process.terminate()\n }\n transport.close()\n }\n\n private func processTerminated(_: Process) {\n transport.close()\n terminationHandler?()\n }\n\n private func finishStarting(port: String) {\n Logger.codeium.info(\"Language server started.\")\n self.port = port\n launchHandler?()\n }\n\n func terminate() {\n process.terminationHandler = nil\n if process.isRunning {\n process.terminate()\n }\n transport.close()\n }\n}\n\nextension CodeiumLanguageServer: CodeiumLSP {\n func sendRequest(_ request: E) async throws -> E.Response where E: CodeiumRequestType {\n guard let port else { throw CancellationError() }\n\n let request = request.makeURLRequest(server: \"http://127.0.0.1:\\(port)\")\n let (data, response) = try await URLSession.shared.data(for: request)\n if (response as? HTTPURLResponse)?.statusCode == 200 {\n do {\n let response = try JSONDecoder().decode(E.Response.self, from: data)\n return response\n } catch {\n if UserDefaults.shared.value(for: \\.codeiumVerboseLog) {\n dump(error)\n Logger.codeium.error(error.localizedDescription)\n }\n throw error\n }\n } else {\n do {\n let error = try JSONDecoder().decode(CodeiumResponseError.self, from: data)\n if error.code == \"aborted\" {\n if error.message.contains(\"is too old\") {\n throw CodeiumError.languageServerOutdated\n }\n throw error\n }\n throw CancellationError()\n } catch {\n if UserDefaults.shared.value(for: \\.codeiumVerboseLog) {\n Logger.codeium.error(error.localizedDescription)\n }\n throw error\n }\n }\n }\n\n func updateIndexing() async {\n let indexEnabled = UserDefaults.shared.value(for: \\.codeiumIndexEnabled)\n if !indexEnabled {\n return\n }\n\n let currentProjectPaths = await getProjectPaths()\n\n // Add all workspaces that are in the currentProjectPaths but not in the previous project\n // paths\n for currentProjectPath in currentProjectPaths {\n if !projectPaths.contains(currentProjectPath) && FileManager.default\n .fileExists(atPath: currentProjectPath)\n {\n _ = try? await sendRequest(CodeiumRequest.AddTrackedWorkspace(requestBody: .init(\n workspace: currentProjectPath\n )))\n }\n }\n\n // Remove all workspaces that are in previous project paths but not in the\n // currentProjectPaths\n for projectPath in projectPaths {\n if !currentProjectPaths.contains(projectPath) && FileManager.default\n .fileExists(atPath: projectPath)\n {\n _ = try? await sendRequest(CodeiumRequest.RemoveTrackedWorkspace(requestBody: .init(\n workspace: projectPath\n )))\n }\n }\n // These should be identical now\n projectPaths = currentProjectPaths\n }\n}\n\nfinal class IOTransport {\n public let stdinPipe: Pipe\n public let stdoutPipe: Pipe\n public let stderrPipe: Pipe\n private var closed: Bool\n private var queue: DispatchQueue\n\n public init() {\n stdinPipe = Pipe()\n stdoutPipe = Pipe()\n stderrPipe = Pipe()\n closed = false\n queue = DispatchQueue(label: \"com.intii.CopilotForXcode.IOTransport\")\n\n setupFileHandleHandlers()\n }\n\n public func write(_ data: Data) {\n if closed {\n return\n }\n\n let fileHandle = stdinPipe.fileHandleForWriting\n\n queue.async {\n fileHandle.write(data)\n }\n }\n\n public func close() {\n queue.sync {\n if self.closed {\n return\n }\n\n self.closed = true\n\n [stdoutPipe, stderrPipe, stdinPipe].forEach { pipe in\n pipe.fileHandleForWriting.closeFile()\n pipe.fileHandleForReading.closeFile()\n }\n }\n }\n\n private func setupFileHandleHandlers() {\n stdoutPipe.fileHandleForReading.readabilityHandler = { [weak self] handle in\n let data = handle.availableData\n\n guard !data.isEmpty else {\n return\n }\n\n if UserDefaults.shared.value(for: \\.codeiumVerboseLog) {\n self?.forwardDataToHandler(data)\n }\n }\n\n stderrPipe.fileHandleForReading.readabilityHandler = { [weak self] handle in\n let data = handle.availableData\n\n guard !data.isEmpty else {\n return\n }\n\n if UserDefaults.shared.value(for: \\.codeiumVerboseLog) {\n self?.forwardErrorDataToHandler(data)\n }\n }\n }\n\n private func forwardDataToHandler(_ data: Data) {\n queue.async { [weak self] in\n guard let self = self else { return }\n\n if self.closed {\n return\n }\n\n if let string = String(bytes: data, encoding: .utf8) {\n Logger.codeium.info(\"stdout: \\(string)\")\n }\n }\n }\n\n private func forwardErrorDataToHandler(_ data: Data) {\n queue.async {\n if let string = String(bytes: data, encoding: .utf8) {\n Logger.codeium.error(\"stderr: \\(string)\")\n }\n }\n }\n}\n\nclass WorkspaceParser: NSObject, XMLParserDelegate {\n var projectPaths: [String] = []\n var workspaceFileURL: URL\n var workspaceBaseURL: URL\n\n init(workspaceFileURL: URL, workspaceBaseURL: URL) {\n self.workspaceFileURL = workspaceFileURL\n self.workspaceBaseURL = workspaceBaseURL\n }\n\n func parse() -> [String] {\n guard let parser = XMLParser(contentsOf: workspaceFileURL) else {\n print(\"Failed to create XML parser for file: \\(workspaceFileURL.path)\")\n return []\n }\n parser.delegate = self\n parser.parse()\n return projectPaths\n }\n\n // XMLParserDelegate methods\n func parser(\n _ parser: XMLParser,\n didStartElement elementName: String,\n namespaceURI: String?,\n qualifiedName qName: String?,\n attributes attributeDict: [String: String]\n ) {\n if elementName == \"FileRef\", let location = attributeDict[\"location\"] {\n var project_path: String\n if location.starts(with: \"group:\") && pathEndsWithXcodeproj(location) {\n let curr_path = String(location.dropFirst(\"group:\".count))\n guard let relative_project_url = URL(string: curr_path) else {\n return\n }\n let relative_base_path = relative_project_url.deletingLastPathComponent()\n project_path = (\n workspaceBaseURL\n .appendingPathComponent(relative_base_path.relativePath)\n ).standardized.path\n } else if location.starts(with: \"absolute:\") && pathEndsWithXcodeproj(location) {\n let abs_url = URL(fileURLWithPath: String(location.dropFirst(\"absolute:\".count)))\n project_path = abs_url.deletingLastPathComponent().standardized.path\n } else {\n return\n }\n if FileManager.default.fileExists(atPath: project_path) {\n projectPaths.append(project_path)\n }\n }\n }\n\n func parser(_ parser: XMLParser, parseErrorOccurred parseError: Error) {\n print(\"Failed to parse XML: \\(parseError.localizedDescription)\")\n }\n\n func pathEndsWithXcodeproj(_ path: String) -> Bool {\n return path.hasSuffix(\".xcodeproj\")\n }\n}\n\npublic func getProjectPaths() async -> [String] {\n guard let workspaceURL = XcodeInspector.shared.realtimeActiveWorkspaceURL else {\n return []\n }\n\n let workspacebaseURL = workspaceURL.deletingLastPathComponent()\n\n let workspaceContentsURL = workspaceURL.appendingPathComponent(\"contents.xcworkspacedata\")\n\n let parser = WorkspaceParser(\n workspaceFileURL: workspaceContentsURL,\n workspaceBaseURL: workspacebaseURL\n )\n let absolutePaths = parser.parse()\n return absolutePaths\n}\n\n" }, { "path": "Tool/Sources/CodeiumService/LanguageServer/CodeiumModels.swift", "content": "import Foundation\nimport JSONRPC\nimport LanguageServerProtocol\nimport SuggestionBasic\n\nstruct CodeiumCompletion: Codable {\n var completionId: String\n var text: String\n var prefix: String?\n var stop: String?\n// var score: Double\n// var tokens: [UInt64]\n// var decodedTokens: [String]\n// var probabilities: [Double]\n// var adjustedProbabilities: [Double]\n// var generatedLength: Int\n}\n\nstruct CodeiumCompletionItem: Codable {\n var completion: CodeiumCompletion\n var suffix: Suffix?\n var range: Range\n var source: CompletionSource\n var completionParts: [CompletionPart]?\n}\n\nstruct Suffix: Codable {\n /// Text to insert after the cursor when accepting the completion.\n var text: String\n /// Cursor position delta (as signed offset) from the end of the inserted\n /// completion (including the suffix).\n var deltaCursorOffset: String\n}\n\nstruct Range: Codable {\n var startOffset: String?\n var endOffset: String?\n var startPosition: DocumentPosition?\n var endPosition: DocumentPosition?\n}\n\nenum CompletionSource: String, Codable {\n case unspecified = \"COMPLETION_SOURCE_UNSPECIFIED\"\n case typingAsSuggested = \"COMPLETION_SOURCE_TYPING_AS_SUGGESTED\"\n case cache = \"COMPLETION_SOURCE_CACHE\"\n case network = \"COMPLETION_SOURCE_NETWORK\"\n}\n\n/// Represents a contiguous part of the completion text that is not\n/// already in the document.\nstruct CompletionPart: Codable {\n enum CompletionPartType: String, Codable {\n case unspecified = \"COMPLETION_PART_TYPE_UNSPECIFIED\"\n /// Single-line completion parts that appear within an existing line of text.\n case inline = \"COMPLETION_PART_TYPE_INLINE\"\n /// Possibly multi-line completion parts that appear below an existing line of text.\n case block = \"COMPLETION_PART_TYPE_BLOCK\"\n /// Like COMPLETION_PART_TYPE_INLINE, but overwrites the existing text.\n case inline_mask = \"COMPLETION_PART_TYPE_INLINE_MASK\"\n }\n\n var text: String\n /// Offset in the original document where the part starts. For block\n /// parts, this is always the end of the line before the block.\n var offset: String\n var type: CompletionPartType\n /// The section of the original line that came before this part. Only valid for\n /// COMPLETION_PART_TYPE_INLINE.\n var prefix: String?\n /// In the case of COMPLETION_PART_TYPE_BLOCK, represents the line it is below.\n var line: String?\n}\n\nstruct CodeiumDocument: Codable {\n var absolute_path_migrate_me_to_uri: String\n var text: String\n // Language ID provided by the editor.\n var editor_language: String\n // Language enum standardized across editors.\n var language: CodeiumSupportedLanguage\n // Measured in number of UTF-8 bytes.\n // var cursor_offset: UInt64?\n // May be present instead of cursor_offset.\n var cursor_position: DocumentCursorPosition?\n // \\n or \\r\\n, if known.\n var line_ending: String = \"\\n\"\n}\n\nstruct DocumentPosition: Codable {\n /// 0-indexed. Measured in UTF-8 bytes.\n var row: String?\n /// 0-indexed. Measured in UTF-8 bytes.\n var col: String?\n}\n\nstruct DocumentCursorPosition: Codable {\n /// 0-indexed. Measured in UTF-8 bytes.\n var row: Int\n /// 0-indexed. Measured in UTF-8 bytes.\n var col: Int\n}\n\nstruct CodeiumEditorOptions: Codable {\n var tab_size: Int\n var insert_spaces: Bool\n}\n\nstruct Metadata: Codable {\n var ide_name: String\n var ide_version: String\n// var extension_name: String\n var extension_version: String\n var api_key: String\n\n /// UID identifying a single session for the given user.\n var session_id: String\n\n /// Used purely in language server to cancel in flight requests.\n /// If request_id is 0, then the request is not cancelable.\n /// This should be a strictly monotonically increasing number\n /// for the duration of a session.\n var request_id: UInt64\n}\n\nenum CodeiumState: String, Codable {\n case unspecified = \"CODEIUM_STATE_UNSPECIFIED\"\n case inactive = \"CODEIUM_STATE_INACTIVE\"\n case processing = \"CODEIUM_STATE_PROCESSING\"\n case success = \"CODEIUM_STATE_SUCCESS\"\n case warning = \"CODEIUM_STATE_WARNING\"\n case error = \"CODEIUM_STATE_ERROR\"\n}\n\nstruct State: Codable {\n var state: CodeiumState\n var message: String\n}\n\n" }, { "path": "Tool/Sources/CodeiumService/LanguageServer/CodeiumRequest.swift", "content": "import Foundation\nimport JSONRPC\nimport LanguageServerProtocol\nimport SuggestionBasic\n\nprotocol CodeiumRequestType {\n associatedtype Response: Codable\n func makeURLRequest(server: String) -> URLRequest\n}\n\nextension CodeiumRequestType {\n func assembleURLRequest(server: String, method: String, body: Data?) -> URLRequest {\n var request = URLRequest(url: .init(\n string: \"\\(server)/exa.language_server_pb.LanguageServerService/\\(method)\"\n )!)\n request.setValue(\"application/json\", forHTTPHeaderField: \"Content-Type\")\n request.httpMethod = \"POST\"\n request.httpBody = body\n return request\n }\n}\n\nstruct CodeiumResponseError: Codable, Error, LocalizedError {\n var code: String\n var message: String\n var errorDescription: String? { message }\n}\n\nenum CodeiumRequest {\n struct GetProcesses: CodeiumRequestType {\n struct Response: Codable {\n var lspPort: UInt32\n var chatWebServerPort: UInt32\n var chatClientPort: UInt32\n }\n\n struct RequestBody: Codable {}\n\n var requestBody: RequestBody\n\n func makeURLRequest(server: String) -> URLRequest {\n let data = (try? JSONEncoder().encode(requestBody)) ?? Data()\n return assembleURLRequest(server: server, method: \"GetProcesses\", body: data)\n }\n }\n\n struct GetCompletion: CodeiumRequestType {\n struct Response: Codable {\n var state: State\n var completionItems: [CodeiumCompletionItem]?\n }\n\n struct RequestBody: Codable {\n var metadata: Metadata\n var document: CodeiumDocument\n var editor_options: CodeiumEditorOptions\n var other_documents: [CodeiumDocument]\n }\n\n var requestBody: RequestBody\n\n func makeURLRequest(server: String) -> URLRequest {\n let data = (try? JSONEncoder().encode(requestBody)) ?? Data()\n return assembleURLRequest(server: server, method: \"GetCompletions\", body: data)\n }\n }\n\n struct CancelRequest: CodeiumRequestType {\n struct Response: Codable {}\n\n struct RequestBody: Codable {\n var request_id: UInt64\n var session_id: String\n }\n\n var requestBody: RequestBody\n\n func makeURLRequest(server: String) -> URLRequest {\n let data = (try? JSONEncoder().encode(requestBody)) ?? Data()\n return assembleURLRequest(server: server, method: \"CancelRequest\", body: data)\n }\n }\n\n struct AcceptCompletion: CodeiumRequestType {\n struct Response: Codable {}\n\n struct RequestBody: Codable {\n var metadata: Metadata\n var completion_id: String\n }\n\n var requestBody: RequestBody\n\n func makeURLRequest(server: String) -> URLRequest {\n let data = (try? JSONEncoder().encode(requestBody)) ?? Data()\n return assembleURLRequest(server: server, method: \"AcceptCompletion\", body: data)\n }\n }\n\n struct RemoveTrackedWorkspace: CodeiumRequestType {\n struct Response: Codable {}\n\n struct RequestBody: Codable {\n var workspace: String\n }\n\n var requestBody: RequestBody\n\n func makeURLRequest(server: String) -> URLRequest {\n let data = (try? JSONEncoder().encode(requestBody)) ?? Data()\n return assembleURLRequest(\n server: server,\n method: \"RemoveTrackedWorkspace\",\n body: data\n )\n }\n }\n\n struct AddTrackedWorkspace: CodeiumRequestType {\n struct Response: Codable {}\n\n struct RequestBody: Codable {\n var workspace: String\n }\n\n var requestBody: RequestBody\n\n func makeURLRequest(server: String) -> URLRequest {\n let data = (try? JSONEncoder().encode(requestBody)) ?? Data()\n return assembleURLRequest(\n server: server,\n method: \"AddTrackedWorkspace\",\n body: data\n )\n }\n }\n\n struct RefreshContextForIdeAction: CodeiumRequestType {\n struct Response: Codable {}\n\n struct RequestBody: Codable {\n var active_document: CodeiumDocument\n var open_document_filepaths: [String]\n var workspace_paths: [String]\n var blocking: Bool = false\n }\n\n var requestBody: RequestBody\n\n func makeURLRequest(server: String) -> URLRequest {\n let data = (try? JSONEncoder().encode(requestBody)) ?? Data()\n return assembleURLRequest(\n server: server,\n method: \"RefreshContextForIdeAction\",\n body: data\n )\n }\n }\n\n struct Heartbeat: CodeiumRequestType {\n struct Response: Codable {}\n\n struct RequestBody: Codable {\n var metadata: Metadata\n }\n\n var requestBody: RequestBody\n\n func makeURLRequest(server: String) -> URLRequest {\n let data = (try? JSONEncoder().encode(requestBody)) ?? Data()\n return assembleURLRequest(server: server, method: \"Heartbeat\", body: data)\n }\n }\n}\n\n" }, { "path": "Tool/Sources/CodeiumService/LanguageServer/CodeiumSupportedLanguage.swift", "content": "import Foundation\nimport SuggestionBasic\n\nenum CodeiumSupportedLanguage: Int, Codable {\n case unspecified = 0\n case c = 1\n case clojure = 2\n case coffeescript = 3\n case cpp = 4\n case csharp = 5\n case css = 6\n case cudacpp = 7\n case dockerfile = 8\n case go = 9\n case groovy = 10\n case handlebars = 11\n case haskell = 12\n case hcl = 13\n case html = 14\n case ini = 15\n case java = 16\n case javascript = 17\n case json = 18\n case julia = 19\n case kotlin = 20\n case latex = 21\n case less = 22\n case lua = 23\n case makefile = 24\n case markdown = 25\n case objectivec = 26\n case objectivecpp = 27\n case perl = 28\n case php = 29\n case plaintext = 30\n case protobuf = 31\n case pbtxt = 32\n case python = 33\n case r = 34\n case ruby = 35\n case rust = 36\n case sass = 37\n case scala = 38\n case scss = 39\n case shell = 40\n case sql = 41\n case starlark = 42\n case swift = 43\n case tsx = 44\n case typescript = 45\n case visualbasic = 46\n case vue = 47\n case xml = 48\n case xsl = 49\n case yaml = 50\n case svelte = 51\n case toml = 52\n case dart = 53\n case rst = 54\n case ocaml = 55\n case cmake = 56\n case pascal = 57\n case elixir = 58\n case fsharp = 59\n case lisp = 60\n case matlab = 61\n case powershell = 62\n case solidity = 63\n case ada = 64\n case ocaml_interface = 65\n \n init(codeLanguage: CodeLanguage) {\n switch codeLanguage {\n case let .builtIn(language):\n switch language {\n case .abap:\n self = .unspecified\n case .windowsbat:\n self = .unspecified\n case .bibtex:\n self = .unspecified\n case .clojure:\n self = .clojure\n case .coffeescript:\n self = .coffeescript\n case .c:\n self = .c\n case .cpp:\n self = .cpp\n case .csharp:\n self = .csharp\n case .css:\n self = .css\n case .diff:\n self = .unspecified\n case .dart:\n self = .dart\n case .dockerfile:\n self = .dockerfile\n case .elixir:\n self = .elixir\n case .erlang:\n self = .unspecified\n case .fsharp:\n self = .fsharp\n case .gitcommit:\n self = .unspecified\n case .gitrebase:\n self = .unspecified\n case .go:\n self = .go\n case .groovy:\n self = .groovy\n case .handlebars:\n self = .handlebars\n case .html:\n self = .html\n case .ini:\n self = .ini\n case .java:\n self = .java\n case .javascript:\n self = .javascript\n case .javascriptreact:\n self = .javascript\n case .json:\n self = .json\n case .latex:\n self = .latex\n case .less:\n self = .less\n case .lua:\n self = .lua\n case .makefile:\n self = .makefile\n case .markdown:\n self = .markdown\n case .objc:\n self = .objectivec\n case .objcpp:\n self = .objectivecpp\n case .perl:\n self = .perl\n case .perl6:\n self = .unspecified\n case .php:\n self = .php\n case .powershell:\n self = .powershell\n case .pug:\n self = .unspecified\n case .python:\n self = .python\n case .r:\n self = .r\n case .razor:\n self = .unspecified\n case .ruby:\n self = .ruby\n case .rust:\n self = .rust\n case .scss:\n self = .scss\n case .sass:\n self = .sass\n case .scala:\n self = .scala\n case .shaderlab:\n self = .unspecified\n case .shellscript:\n self = .shell\n case .sql:\n self = .sql\n case .swift:\n self = .swift\n case .typescript:\n self = .typescript\n case .typescriptreact:\n self = .tsx\n case .tex:\n self = .latex\n case .vb:\n self = .visualbasic\n case .xml:\n self = .xml\n case .xsl:\n self = .xsl\n case .yaml:\n self = .yaml\n }\n case .plaintext:\n self = .plaintext\n case let .other(name):\n switch name {\n case \"svelte\":\n self = .svelte\n case \"toml\":\n self = .toml\n case \"rst\":\n self = .rst\n case \"lisp\":\n self = .lisp\n case \"matlab\":\n self = .matlab\n default:\n self = .unspecified\n }\n }\n }\n}\n" }, { "path": "Tool/Sources/CodeiumService/LanguageServer/OpendDocumentPool.swift", "content": "import Foundation\n\nprivate let maxSize: Int = 1_000_000 // Byte\n\nactor OpenedDocumentPool {\n var openedDocuments = [URL: OpenedDocument]()\n\n func getOtherDocuments(exceptURL: URL) -> [OpenedDocument] {\n let ordered = openedDocuments.values.sorted { $0.updateTime > $1.updateTime }\n var documents = [OpenedDocument]()\n var size = 0\n for document in ordered where document.url != exceptURL {\n size += document.size\n if size > maxSize {\n break\n }\n documents.append(document)\n }\n\n return documents\n }\n\n func openDocument(url: URL, relativePath: String, content: String) {\n let document = OpenedDocument(url: url, relativePath: relativePath, content: content)\n openedDocuments[url] = document\n }\n\n func updateDocument(url: URL, relativePath: String, content: String) {\n if let document = openedDocuments[url] {\n document.update(content: content)\n } else {\n openDocument(url: url, relativePath: relativePath, content: content)\n }\n }\n\n func closeDocument(url: URL) {\n openedDocuments[url] = nil\n }\n}\n\nfinal class OpenedDocument {\n var url: URL\n var relativePath: String\n var updateTime: Date\n var content: String\n var size: Int\n\n public init(url: URL, relativePath: String, content: String) {\n self.url = url\n self.relativePath = relativePath\n updateTime = Date()\n size = content.utf8.count\n if size > maxSize {\n self.content = \"\"\n } else {\n self.content = content\n }\n }\n\n func update(content: String) {\n updateTime = Date()\n size = content.utf8.count\n if size > maxSize {\n self.content = \"\"\n } else {\n self.content = content\n }\n }\n}\n\n" }, { "path": "Tool/Sources/CodeiumService/Services/CodeiumAuthService.swift", "content": "import Configs\nimport Foundation\nimport Keychain\n\npublic final class CodeiumAuthService {\n public init() {}\n let codeiumKeyKey = \"codeiumAuthKey\"\n let keychain = Keychain()\n\n var key: String? { try? keychain.get(codeiumKeyKey) }\n\n public var isSignedIn: Bool { return key != nil }\n\n public func signIn(token: String) async throws {\n let key = try await generate(token: token)\n try keychain.update(key, key: codeiumKeyKey)\n }\n\n public func signOut() async throws {\n try keychain.remove(codeiumKeyKey)\n }\n\n struct GenerateKeyRequestBody: Codable {\n var firebase_id_token: String\n }\n\n struct GenerateKeyResponseBody: Codable {\n var api_key: String\n }\n\n struct GenerateKeyErrorResponseBody: Codable, Error, LocalizedError {\n var detail: String\n var errorDescription: String? { detail }\n }\n\n func generate(token: String) async throws -> String {\n var registerUserUrl = URL(string: \"https://api.codeium.com/register_user/\")\n let apiUrl = UserDefaults.shared.value(for: \\.codeiumApiUrl)\n if UserDefaults.shared.value(for: \\.codeiumEnterpriseMode), apiUrl != \"\" {\n registerUserUrl =\n URL(string: apiUrl + \"/exa.seat_management_pb.SeatManagementService/RegisterUser\")\n }\n\n var request = URLRequest(url: registerUserUrl!)\n request.httpMethod = \"POST\"\n request.addValue(\"application/json\", forHTTPHeaderField: \"Content-Type\")\n let requestBody = GenerateKeyRequestBody(firebase_id_token: token)\n let requestData = try JSONEncoder().encode(requestBody)\n request.httpBody = requestData\n let (data, _) = try await URLSession.shared.data(for: request)\n do {\n let response = try JSONDecoder().decode(GenerateKeyResponseBody.self, from: data)\n return response.api_key\n } catch {\n if let response = try? JSONDecoder()\n .decode(GenerateKeyErrorResponseBody.self, from: data)\n {\n throw response\n }\n throw error\n }\n }\n}\n\n" }, { "path": "Tool/Sources/CodeiumService/Services/CodeiumService.swift", "content": "import AppKit\nimport Foundation\nimport LanguageClient\nimport LanguageServerProtocol\nimport Logger\nimport SuggestionBasic\nimport XcodeInspector\n\npublic protocol CodeiumSuggestionServiceType {\n func getCompletions(\n fileURL: URL,\n content: String,\n cursorPosition: CursorPosition,\n tabSize: Int,\n indentSize: Int,\n usesTabsForIndentation: Bool\n ) async throws -> [CodeSuggestion]\n func notifyAccepted(_ suggestion: CodeSuggestion) async\n func getChatURL() async throws -> URL\n func notifyOpenTextDocument(fileURL: URL, content: String) async throws\n func notifyChangeTextDocument(fileURL: URL, content: String) async throws\n func notifyCloseTextDocument(fileURL: URL) async throws\n func cancelRequest() async\n func terminate()\n}\n\nenum CodeiumError: Error, LocalizedError {\n case languageServerNotInstalled\n case languageServerOutdated\n case languageServiceIsInstalling\n case failedToConstructChatURL\n\n var errorDescription: String? {\n switch self {\n case .languageServerNotInstalled:\n return \"Language server is not installed. Please install it in the host app.\"\n case .languageServerOutdated:\n return \"Language server is outdated. Please update it in the host app or update the extension.\"\n case .languageServiceIsInstalling:\n return \"Language service is installing, please try again later.\"\n case .failedToConstructChatURL:\n return \"Failed to construct chat URL.\"\n }\n }\n}\n\npublic class CodeiumService {\n static let sessionId = UUID().uuidString\n let projectRootURL: URL\n var server: CodeiumLSP?\n var heartbeatTask: Task?\n var workspaceTask: Task?\n var requestCounter: UInt64 = 0\n var cancellationCounter: UInt64 = 0\n let openedDocumentPool = OpenedDocumentPool()\n let onServiceLaunched: () -> Void\n let onServiceTerminated: () -> Void\n\n let languageServerURL: URL\n let supportURL: URL\n\n let authService = CodeiumAuthService()\n\n var fallbackXcodeVersion = \"14.0.0\"\n var languageServerVersion = CodeiumInstallationManager.latestSupportedVersion\n\n private var ongoingTasks = Set>()\n\n init(designatedServer: CodeiumLSP) {\n projectRootURL = URL(fileURLWithPath: \"/\")\n server = designatedServer\n onServiceLaunched = {}\n onServiceTerminated = {}\n languageServerURL = URL(fileURLWithPath: \"/\")\n supportURL = URL(fileURLWithPath: \"/\")\n }\n\n public init(\n projectRootURL: URL,\n onServiceLaunched: @escaping () -> Void,\n onServiceTerminated: @escaping () -> Void\n ) throws {\n self.projectRootURL = projectRootURL\n self.onServiceLaunched = onServiceLaunched\n self.onServiceTerminated = onServiceTerminated\n let urls = try CodeiumService.createFoldersIfNeeded()\n languageServerURL = urls.executableURL.appendingPathComponent(\"language_server\")\n supportURL = urls.supportURL\n Task {\n try await setupServerIfNeeded()\n }\n }\n\n @discardableResult\n func setupServerIfNeeded() async throws -> CodeiumLSP {\n if let server { return server }\n\n let binaryManager = CodeiumInstallationManager()\n let installationStatus = await binaryManager.checkInstallation()\n switch installationStatus {\n case let .installed(version), let .unsupported(version, _):\n languageServerVersion = version\n case .notInstalled:\n throw CodeiumError.languageServerNotInstalled\n case let .outdated(version, _, _):\n languageServerVersion = version\n throw CodeiumError.languageServerOutdated\n }\n\n let metadata = try await getMetadata()\n let tempFolderURL = FileManager.default.temporaryDirectory\n let managerDirectoryURL = tempFolderURL\n .appendingPathComponent(\"com.intii.CopilotForXcode\")\n .appendingPathComponent(UUID().uuidString)\n if !FileManager.default.fileExists(atPath: managerDirectoryURL.path) {\n try FileManager.default.createDirectory(\n at: managerDirectoryURL,\n withIntermediateDirectories: true\n )\n }\n\n let server = CodeiumLanguageServer(\n languageServerExecutableURL: languageServerURL,\n managerDirectoryURL: managerDirectoryURL,\n supportURL: supportURL\n )\n\n server.terminationHandler = { [weak self] in\n self?.server = nil\n self?.heartbeatTask?.cancel()\n self?.workspaceTask?.cancel()\n self?.requestCounter = 0\n self?.cancellationCounter = 0\n self?.onServiceTerminated()\n Logger.codeium.info(\"Language server is terminated, will be restarted when needed.\")\n }\n\n server.launchHandler = { [weak self] in\n guard let self else { return }\n self.onServiceLaunched()\n self.heartbeatTask = Task { [weak self, metadata] in\n while true {\n try Task.checkCancellation()\n _ = try? await self?.server?.sendRequest(\n CodeiumRequest.Heartbeat(requestBody: .init(metadata: metadata))\n )\n try await Task.sleep(nanoseconds: 5_000_000_000)\n }\n }\n\n self.workspaceTask = Task { [weak self] in\n while true {\n try Task.checkCancellation()\n _ = await self?.server?.updateIndexing()\n try await Task.sleep(nanoseconds: 5_000_000_000)\n }\n }\n }\n\n self.server = server\n server.start()\n return server\n }\n\n public static func createFoldersIfNeeded() throws -> (\n applicationSupportURL: URL,\n gitHubCopilotURL: URL,\n executableURL: URL,\n supportURL: URL\n ) {\n let supportURL = FileManager.default.urls(\n for: .applicationSupportDirectory,\n in: .userDomainMask\n ).first!.appendingPathComponent(\n Bundle.main\n .object(forInfoDictionaryKey: \"APPLICATION_SUPPORT_FOLDER\") as! String\n )\n\n if !FileManager.default.fileExists(atPath: supportURL.path) {\n try? FileManager.default\n .createDirectory(at: supportURL, withIntermediateDirectories: false)\n }\n let gitHubCopilotFolderURL = supportURL.appendingPathComponent(\"Codeium\")\n if !FileManager.default.fileExists(atPath: gitHubCopilotFolderURL.path) {\n try? FileManager.default\n .createDirectory(at: gitHubCopilotFolderURL, withIntermediateDirectories: false)\n }\n let supportFolderURL = gitHubCopilotFolderURL.appendingPathComponent(\"support\")\n if !FileManager.default.fileExists(atPath: supportFolderURL.path) {\n try? FileManager.default\n .createDirectory(at: supportFolderURL, withIntermediateDirectories: false)\n }\n let executableFolderURL = gitHubCopilotFolderURL.appendingPathComponent(\"executable\")\n if !FileManager.default.fileExists(atPath: executableFolderURL.path) {\n try? FileManager.default\n .createDirectory(at: executableFolderURL, withIntermediateDirectories: false)\n }\n\n return (supportURL, gitHubCopilotFolderURL, executableFolderURL, supportFolderURL)\n }\n}\n\nextension CodeiumService {\n func getMetadata() async throws -> Metadata {\n guard let key = authService.key else {\n struct E: Error, LocalizedError {\n var errorDescription: String? { \"Codeium not signed in.\" }\n }\n throw E()\n }\n var ideVersion = await XcodeInspector.shared.latestActiveXcode?.version\n ?? fallbackXcodeVersion\n let versionNumberSegmentCount = ideVersion.split(separator: \".\").count\n if versionNumberSegmentCount == 2 {\n ideVersion += \".0\"\n } else if versionNumberSegmentCount == 1 {\n ideVersion += \".0.0\"\n }\n return Metadata(\n ide_name: \"xcode\",\n ide_version: ideVersion,\n extension_version: languageServerVersion,\n api_key: key,\n session_id: CodeiumService.sessionId,\n request_id: requestCounter\n )\n }\n\n func getRelativePath(of fileURL: URL) -> String {\n let filePath = fileURL.path\n let rootPath = projectRootURL.path\n if let range = filePath.range(of: rootPath),\n range.lowerBound == filePath.startIndex\n {\n let relativePath = filePath.replacingCharacters(\n in: filePath.startIndex.. [CodeSuggestion] {\n ongoingTasks.forEach { $0.cancel() }\n ongoingTasks.removeAll()\n await cancelRequest()\n\n requestCounter += 1\n let languageId = languageIdentifierFromFileURL(fileURL)\n\n let task = Task {\n let request = try await CodeiumRequest.GetCompletion(requestBody: .init(\n metadata: getMetadata(),\n document: .init(\n absolute_path_migrate_me_to_uri: fileURL.path,\n text: content,\n editor_language: languageId.rawValue,\n language: .init(codeLanguage: languageId),\n cursor_position: .init(\n row: cursorPosition.line,\n col: cursorPosition.character\n )\n ),\n editor_options: .init(tab_size: indentSize, insert_spaces: !usesTabsForIndentation),\n other_documents: openedDocumentPool.getOtherDocuments(exceptURL: fileURL)\n .map { openedDocument in\n let languageId = languageIdentifierFromFileURL(openedDocument.url)\n return .init(\n absolute_path_migrate_me_to_uri: openedDocument.url.path,\n text: openedDocument.content,\n editor_language: languageId.rawValue,\n language: .init(codeLanguage: languageId)\n )\n }\n ))\n\n try Task.checkCancellation()\n\n let result = try await (await setupServerIfNeeded()).sendRequest(request)\n\n try Task.checkCancellation()\n\n return result.completionItems?.map { item in\n CodeSuggestion(\n id: item.completion.completionId,\n text: item.completion.text,\n position: cursorPosition,\n range: CursorRange(\n start: .init(\n line: item.range.startPosition?.row.flatMap(Int.init) ?? 0,\n character: item.range.startPosition?.col.flatMap(Int.init) ?? 0\n ),\n end: .init(\n line: item.range.endPosition?.row.flatMap(Int.init) ?? 0,\n character: item.range.endPosition?.col.flatMap(Int.init) ?? 0\n )\n )\n )\n } ?? []\n }\n\n ongoingTasks.insert(task)\n\n return try await task.value\n }\n\n public func cancelRequest() async {\n _ = try? await server?.sendRequest(\n CodeiumRequest.CancelRequest(requestBody: .init(\n request_id: requestCounter,\n session_id: CodeiumService.sessionId\n ))\n )\n }\n\n public func getChatURL() async throws -> URL {\n let metadata = try await getMetadata()\n let ports = try await server?.sendRequest(\n CodeiumRequest.GetProcesses(requestBody: .init())\n )\n\n guard let chatClientPort = ports?.chatClientPort,\n let chatWebServerPort = ports?.chatWebServerPort\n else { throw CodeiumError.failedToConstructChatURL }\n\n let webServerUrl = \"ws://127.0.0.1:\\(chatWebServerPort)\"\n var components = URLComponents()\n components.scheme = \"http\"\n components.host = \"127.0.0.1\"\n components.port = Int(chatClientPort)\n components.path = \"/\"\n components.queryItems = [\n URLQueryItem(name: \"api_key\", value: metadata.api_key),\n URLQueryItem(name: \"locale\", value: \"en\"),\n URLQueryItem(name: \"extension_name\", value: \"Copilot for XCode\"),\n URLQueryItem(name: \"extension_version\", value: metadata.extension_version),\n URLQueryItem(name: \"ide_name\", value: metadata.ide_name),\n URLQueryItem(name: \"ide_version\", value: metadata.ide_version),\n URLQueryItem(name: \"web_server_url\", value: webServerUrl),\n URLQueryItem(name: \"ide_telemetry_enabled\", value: \"true\"),\n URLQueryItem(name: \"has_enterprise_extension\", value: String(UserDefaults.shared.value(for: \\.codeiumEnterpriseMode))),\n URLQueryItem(name: \"has_index_service\", value: String(UserDefaults.shared.value(for: \\.codeiumIndexEnabled)))\n ]\n\n if let url = components.url {\n print(url)\n return url\n } else {\n throw CodeiumError.failedToConstructChatURL\n }\n }\n\n public func notifyAccepted(_ suggestion: CodeSuggestion) async {\n _ = try? await (try setupServerIfNeeded())\n .sendRequest(CodeiumRequest.AcceptCompletion(requestBody: .init(\n metadata: getMetadata(),\n completion_id: suggestion.id\n )))\n }\n\n public func notifyOpenTextDocument(fileURL: URL, content: String) async throws {\n let relativePath = getRelativePath(of: fileURL)\n await openedDocumentPool.openDocument(\n url: fileURL,\n relativePath: relativePath,\n content: content\n )\n }\n\n public func notifyChangeTextDocument(fileURL: URL, content: String) async throws {\n let relativePath = getRelativePath(of: fileURL)\n await openedDocumentPool.updateDocument(\n url: fileURL,\n relativePath: relativePath,\n content: content\n )\n }\n\n public func notifyCloseTextDocument(fileURL: URL) async throws {\n await openedDocumentPool.closeDocument(url: fileURL)\n }\n\n public func notifyOpenWorkspace(workspaceURL: URL) async throws {\n _ = try await (setupServerIfNeeded()).sendRequest(\n CodeiumRequest\n .AddTrackedWorkspace(requestBody: .init(workspace: workspaceURL.path))\n )\n }\n\n public func notifyCloseWorkspace(workspaceURL: URL) async throws {\n _ = try await (setupServerIfNeeded()).sendRequest(\n CodeiumRequest\n .RemoveTrackedWorkspace(requestBody: .init(workspace: workspaceURL.path))\n )\n }\n\n public func refreshIDEContext(\n fileURL: URL,\n content: String,\n cursorPosition: CursorPosition,\n tabSize: Int,\n indentSize: Int,\n usesTabsForIndentation: Bool,\n workspaceURL: URL\n ) async throws {\n let languageId = languageIdentifierFromFileURL(fileURL)\n let request = await CodeiumRequest.RefreshContextForIdeAction(requestBody: .init(\n active_document: .init(\n absolute_path_migrate_me_to_uri: fileURL.path,\n text: content,\n editor_language: languageId.rawValue,\n language: .init(codeLanguage: languageId),\n cursor_position: .init(\n row: cursorPosition.line,\n col: cursorPosition.character\n )\n ),\n open_document_filepaths: openedDocumentPool.getOtherDocuments(exceptURL: fileURL)\n .map(\\.url.path),\n workspace_paths: [workspaceURL.path]\n ))\n _ = try await (setupServerIfNeeded()).sendRequest(request)\n }\n\n public func terminate() {\n server?.terminate()\n server = nil\n }\n}\n\nfunc getXcodeVersion() async throws -> String {\n let task = Process()\n task.launchPath = \"/usr/bin/xcodebuild\"\n task.arguments = [\"-version\"]\n let outpipe = Pipe()\n task.standardOutput = outpipe\n task.standardError = Pipe()\n return try await withUnsafeThrowingContinuation { continuation in\n do {\n task.terminationHandler = { _ in\n do {\n if let data = try outpipe.fileHandleForReading.readToEnd(),\n let content = String(data: data, encoding: .utf8)\n {\n let firstLine = content.split(whereSeparator: \\.isNewline).first ?? \"\"\n var version = firstLine.replacingOccurrences(of: \"Xcode \", with: \"\")\n if version.isEmpty {\n version = \"14.0\"\n }\n continuation.resume(returning: version)\n return\n }\n continuation.resume(returning: \"\")\n } catch {\n continuation.resume(throwing: error)\n }\n }\n try task.run()\n } catch {\n continuation.resume(throwing: error)\n }\n }\n}\n\n" }, { "path": "Tool/Sources/CodeiumService/Services/CodeiumSuggestionService.swift", "content": "import CopilotForXcodeKit\nimport Foundation\nimport SuggestionBasic\nimport Workspace\n\npublic final class CodeiumSuggestionService: SuggestionServiceType {\n public var configuration: SuggestionServiceConfiguration {\n .init(\n acceptsRelevantCodeSnippets: true,\n mixRelevantCodeSnippetsInSource: true,\n acceptsRelevantSnippetsFromOpenedFiles: false\n )\n }\n\n let serviceLocator: ServiceLocator\n\n init(serviceLocator: ServiceLocator) {\n self.serviceLocator = serviceLocator\n }\n\n public func getSuggestions(\n _ request: SuggestionRequest,\n workspace: WorkspaceInfo\n ) async throws -> [CopilotForXcodeKit.CodeSuggestion] {\n guard let service = await serviceLocator.getService(from: workspace) else { return [] }\n return try await service.getCompletions(\n fileURL: request.fileURL,\n content: request.content,\n cursorPosition: .init(\n line: request.cursorPosition.line,\n character: request.cursorPosition.character\n ),\n tabSize: request.tabSize,\n indentSize: request.indentSize,\n usesTabsForIndentation: request.usesTabsForIndentation\n ).map(Self.convert)\n }\n\n public func notifyAccepted(\n _ suggestion: CopilotForXcodeKit.CodeSuggestion,\n workspace: WorkspaceInfo\n ) async {\n guard let service = await serviceLocator.getService(from: workspace) else { return }\n await service.notifyAccepted(Self.convert(suggestion))\n }\n\n public func notifyRejected(\n _ suggestions: [CopilotForXcodeKit.CodeSuggestion],\n workspace: WorkspaceInfo\n ) async {\n // unimplemented\n }\n\n public func cancelRequest(workspace: WorkspaceInfo) async {\n guard let service = await serviceLocator.getService(from: workspace) else { return }\n await service.cancelRequest()\n }\n\n static func convert(\n _ suggestion: SuggestionBasic.CodeSuggestion\n ) -> CopilotForXcodeKit.CodeSuggestion {\n .init(\n id: suggestion.id,\n text: suggestion.text,\n position: .init(\n line: suggestion.position.line,\n character: suggestion.position.character\n ),\n range: .init(\n start: .init(\n line: suggestion.range.start.line,\n character: suggestion.range.start.character\n ),\n end: .init(\n line: suggestion.range.end.line,\n character: suggestion.range.end.character\n )\n )\n )\n }\n\n static func convert(\n _ suggestion: CopilotForXcodeKit.CodeSuggestion\n ) -> SuggestionBasic.CodeSuggestion {\n .init(\n id: suggestion.id,\n text: suggestion.text,\n position: .init(\n line: suggestion.position.line,\n character: suggestion.position.character\n ),\n range: .init(\n start: .init(\n line: suggestion.range.start.line,\n character: suggestion.range.start.character\n ),\n end: .init(\n line: suggestion.range.end.line,\n character: suggestion.range.end.character\n )\n )\n )\n }\n}\n\n" }, { "path": "Tool/Sources/CommandHandler/CommandHandler.swift", "content": "import ComposableArchitecture\nimport Dependencies\nimport Foundation\nimport ModificationBasic\nimport Preferences\nimport SuggestionBasic\nimport Toast\nimport XcodeInspector\n\n/// Provides an interface to handle commands.\npublic protocol CommandHandler {\n // MARK: Suggestion\n\n func presentSuggestions(_ suggestions: [CodeSuggestion]) async\n func presentPreviousSuggestion() async\n func presentNextSuggestion() async\n func rejectSuggestions() async\n func acceptSuggestion() async\n func acceptActiveSuggestionLineInGroup(atIndex index: Int?) async\n func dismissSuggestion() async\n func generateRealtimeSuggestions(sourceEditor: SourceEditor?) async\n\n // MARK: Chat\n\n func openChat(forceDetach: Bool, activateThisApp: Bool)\n func sendChatMessage(_ message: String) async\n\n // MARK: Modification\n\n func acceptModification() async\n func presentModification(state: Shared) async\n\n // MARK: Custom Command\n\n func handleCustomCommand(_ command: CustomCommand) async\n\n // MARK: Toast\n\n func toast(_ string: String, as type: ToastType)\n\n // MARK: Others\n\n func presentFile(at fileURL: URL, line: Int?) async\n\n func presentFile(at fileURL: URL) async\n}\n\npublic extension CommandHandler {\n /// Default implementation for `presentFile(at:line:)`.\n func presentFile(at fileURL: URL) async {\n await presentFile(at: fileURL, line: nil)\n }\n}\n\npublic struct CommandHandlerDependencyKey: DependencyKey {\n public static var liveValue: CommandHandler = UniversalCommandHandler.shared\n public static var testValue: CommandHandler = NOOPCommandHandler()\n}\n\npublic extension DependencyValues {\n /// In production, you need to override the command handler globally by setting\n /// ``UniversalCommandHandler.shared.commandHandler``.\n ///\n /// In tests, you can use ``withDependency`` to mock it.\n var commandHandler: CommandHandler {\n get { self[CommandHandlerDependencyKey.self] }\n set { self[CommandHandlerDependencyKey.self] = newValue }\n }\n}\n\npublic final class UniversalCommandHandler: CommandHandler {\n public static let shared: UniversalCommandHandler = .init()\n\n public var commandHandler: CommandHandler = NOOPCommandHandler()\n\n private init() {}\n\n public func presentSuggestions(_ suggestions: [SuggestionBasic.CodeSuggestion]) async {\n await commandHandler.presentSuggestions(suggestions)\n }\n\n public func presentPreviousSuggestion() async {\n await commandHandler.presentPreviousSuggestion()\n }\n\n public func presentNextSuggestion() async {\n await commandHandler.presentNextSuggestion()\n }\n\n public func rejectSuggestions() async {\n await commandHandler.rejectSuggestions()\n }\n\n public func acceptSuggestion() async {\n await commandHandler.acceptSuggestion()\n }\n \n public func acceptActiveSuggestionLineInGroup(atIndex index: Int?) async {\n await commandHandler.acceptActiveSuggestionLineInGroup(atIndex: index)\n }\n\n public func dismissSuggestion() async {\n await commandHandler.dismissSuggestion()\n }\n\n public func generateRealtimeSuggestions(sourceEditor: SourceEditor?) async {\n await commandHandler.generateRealtimeSuggestions(sourceEditor: sourceEditor)\n }\n\n public func openChat(forceDetach: Bool, activateThisApp: Bool) {\n commandHandler.openChat(forceDetach: forceDetach, activateThisApp: activateThisApp)\n }\n\n public func sendChatMessage(_ message: String) async {\n await commandHandler.sendChatMessage(message)\n }\n\n public func acceptModification() async {\n await commandHandler.acceptModification()\n }\n\n public func presentModification(state: Shared) async {\n await commandHandler.presentModification(state: state)\n }\n\n public func handleCustomCommand(_ command: CustomCommand) async {\n await commandHandler.handleCustomCommand(command)\n }\n\n public func toast(_ string: String, as type: ToastType) {\n commandHandler.toast(string, as: type)\n }\n\n public func presentFile(at fileURL: URL, line: Int?) async {\n await commandHandler.presentFile(at: fileURL, line: line)\n }\n}\n\nstruct NOOPCommandHandler: CommandHandler {\n func presentSuggestions(_ suggestions: [CodeSuggestion]) async {\n print(\"present \\(suggestions.count) suggestions\")\n }\n\n func presentPreviousSuggestion() async {\n print(\"previous suggestion\")\n }\n\n func presentNextSuggestion() async {\n print(\"next suggestion\")\n }\n\n func rejectSuggestions() async {\n print(\"reject suggestions\")\n }\n\n func acceptSuggestion() async {\n print(\"accept suggestion\")\n }\n \n func acceptActiveSuggestionLineInGroup(atIndex index: Int?) async {\n print(\"accept active suggestion line in group at index \\(String(describing: index))\")\n }\n\n func dismissSuggestion() async {\n print(\"dismiss suggestion\")\n }\n\n func generateRealtimeSuggestions(sourceEditor: SourceEditor?) async {\n print(\"generate realtime suggestions\")\n }\n\n func openChat(forceDetach: Bool, activateThisApp: Bool) {\n print(\"open chat\")\n }\n\n func sendChatMessage(_: String) async {\n print(\"send chat message\")\n }\n\n func acceptModification() async {\n print(\"accept prompt to code\")\n }\n\n func presentModification(state: Shared) {\n print(\"present modification\")\n }\n\n func handleCustomCommand(_: CustomCommand) async {\n print(\"handle custom command\")\n }\n\n func toast(_: String, as: ToastType) {\n print(\"toast\")\n }\n\n func presentFile(at fileURL: URL, line: Int?) async {\n print(\"present file\")\n }\n}\n\n" }, { "path": "Tool/Sources/Configs/Configurations.swift", "content": "import Foundation\n\nprivate var teamIDPrefix: String {\n Bundle.main.infoDictionary?[\"TEAM_ID_PREFIX\"] as? String ?? \"\"\n}\n\nprivate var bundleIdentifierBase: String {\n Bundle.main.infoDictionary?[\"BUNDLE_IDENTIFIER_BASE\"] as? String ?? \"\"\n}\n\npublic var userDefaultSuiteName: String {\n \"\\(teamIDPrefix)group.\\(bundleIdentifierBase)\"\n}\n\npublic var keychainAccessGroup: String {\n return \"\\(teamIDPrefix)\\(bundleIdentifierBase).Shared\"\n}\n\npublic var keychainService: String {\n return bundleIdentifierBase\n}\n\n" }, { "path": "Tool/Sources/CustomAsyncAlgorithms/TimedDebounce.swift", "content": "import Foundation\n\nprivate actor TimedDebounceFunction {\n let duration: TimeInterval\n let block: (Element) async -> Void\n\n var task: Task?\n var lastValue: Element?\n var lastFireTime: Date = .init(timeIntervalSince1970: 0)\n\n init(duration: TimeInterval, block: @escaping (Element) async -> Void) {\n self.duration = duration\n self.block = block\n }\n\n func callAsFunction(_ value: Element) async {\n task?.cancel()\n if lastFireTime.timeIntervalSinceNow < -duration {\n await fire(value)\n task = nil\n } else {\n lastValue = value\n task = Task.detached { [weak self, duration] in\n try await Task.sleep(nanoseconds: UInt64(duration * 1_000_000_000))\n await self?.fire(value)\n }\n }\n }\n\n func finish() async {\n task?.cancel()\n if let lastValue {\n await fire(lastValue)\n }\n }\n\n private func fire(_ value: Element) async {\n lastFireTime = Date()\n lastValue = nil\n await block(value)\n }\n}\n\npublic extension AsyncSequence {\n /// Debounce, but only if the value is received within a certain time frame.\n ///\n /// In the future when we drop macOS 12 support we should just use chunked from AsyncAlgorithms.\n func timedDebounce(\n for duration: TimeInterval,\n reducer: @escaping @Sendable (Element, Element) -> Element\n ) -> AsyncThrowingStream {\n return AsyncThrowingStream { continuation in\n Task {\n let storage = TimedDebounceStorage()\n var lastTimeStamp = Date()\n do {\n for try await value in self {\n await storage.reduce(value, reducer: reducer)\n let now = Date()\n if now.timeIntervalSince(lastTimeStamp) >= duration {\n lastTimeStamp = now\n if let value = await storage.consume() {\n continuation.yield(value)\n }\n }\n }\n if let value = await storage.consume() {\n continuation.yield(value)\n }\n continuation.finish()\n } catch {\n continuation.finish(throwing: error)\n }\n }\n }\n }\n}\n\nprivate actor TimedDebounceStorage {\n var value: Element?\n func reduce(_ value: Element, reducer: (Element, Element) -> Element) async {\n if let existing = self.value {\n self.value = reducer(existing, value)\n } else {\n self.value = value\n }\n }\n\n func consume() -> Element? {\n defer { value = nil }\n return value\n }\n}\n\n" }, { "path": "Tool/Sources/CustomCommandTemplateProcessor/CustomCommandTemplateProcessor.swift", "content": "import AppKit\nimport Foundation\nimport SuggestionBasic\nimport XcodeInspector\n\npublic struct CustomCommandTemplateProcessor {\n public init() {}\n \n public func process(_ text: String) async -> String {\n let info = await getEditorInformation()\n let editorContent = info.editorContent\n let updatedText = text\n .replacingOccurrences(of: \"{{selected_code}}\", with: \"\"\"\n \\(editorContent?.selectedContent.trimmingCharacters(in: .whitespacesAndNewlines) ?? \"\")\n \"\"\")\n .replacingOccurrences(\n of: \"{{active_editor_language}}\",\n with: info.language.rawValue\n )\n .replacingOccurrences(\n of: \"{{active_editor_file_url}}\",\n with: info.documentURL?.path ?? \"\"\n )\n .replacingOccurrences(\n of: \"{{active_editor_file_name}}\",\n with: info.documentURL?.lastPathComponent ?? \"\"\n )\n .replacingOccurrences(\n of: \"{{clipboard}}\",\n with: NSPasteboard.general.string(forType: .string) ?? \"\"\n )\n return updatedText\n }\n\n struct EditorInformation {\n let editorContent: SourceEditor.Content?\n let language: CodeLanguage\n let documentURL: URL?\n }\n\n func getEditorInformation() async -> EditorInformation {\n let editorContent = await XcodeInspector.shared.latestFocusedEditor?.getContent()\n let documentURL = await XcodeInspector.shared.activeDocumentURL\n let language = documentURL.map(languageIdentifierFromFileURL) ?? .plaintext\n\n return .init(\n editorContent: editorContent,\n language: language,\n documentURL: documentURL\n )\n }\n}\n\n" }, { "path": "Tool/Sources/DebounceFunction/DebounceFunction.swift", "content": "import Foundation\n\npublic actor DebounceFunction {\n let duration: TimeInterval\n let block: (T) async -> Void\n\n var task: Task?\n\n public init(duration: TimeInterval, block: @escaping (T) async -> Void) {\n self.duration = duration\n self.block = block\n }\n\n public func cancel() {\n task?.cancel()\n }\n\n public func callAsFunction(_ t: T) async {\n task?.cancel()\n task = Task { [block, duration] in\n try await Task.sleep(nanoseconds: UInt64(duration * 1_000_000_000))\n await block(t)\n }\n }\n}\n\npublic actor DebounceRunner {\n let duration: TimeInterval\n\n var task: Task?\n\n public init(duration: TimeInterval) {\n self.duration = duration\n }\n\n public func cancel() {\n task?.cancel()\n }\n\n public func debounce(_ block: @escaping () async -> Void) {\n task?.cancel()\n task = Task { [duration] in\n try await Task.sleep(nanoseconds: UInt64(duration * 1_000_000_000))\n await block()\n }\n }\n}\n\n" }, { "path": "Tool/Sources/DebounceFunction/ThrottleFunction.swift", "content": "import Foundation\n\npublic actor ThrottleFunction {\n let duration: TimeInterval\n let block: (T) async -> Void\n\n var task: Task?\n var lastFinishTime: Date = .init(timeIntervalSince1970: 0)\n var now: () -> Date = { Date() }\n\n public init(duration: TimeInterval, block: @escaping @Sendable (T) async -> Void) {\n self.duration = duration\n self.block = block\n }\n\n public func callAsFunction(_ t: T) async {\n if task == nil {\n scheduleTask(t, wait: now().timeIntervalSince(lastFinishTime) < duration)\n }\n }\n\n func scheduleTask(_ t: T, wait: Bool) {\n task = Task.detached { [weak self] in\n guard let self else { return }\n do {\n if wait {\n try await Task.sleep(nanoseconds: UInt64(duration * 1_000_000_000))\n }\n await block(t)\n await finishTask()\n } catch {\n await finishTask()\n }\n }\n }\n\n func finishTask() {\n task = nil\n lastFinishTime = now()\n }\n}\n\npublic actor ThrottleRunner {\n let duration: TimeInterval\n var lastFinishTime: Date = .init(timeIntervalSince1970: 0)\n var now: () -> Date = { Date() }\n var task: Task?\n\n public init(duration: TimeInterval) {\n self.duration = duration\n }\n\n public func throttle(block: @escaping @Sendable () async -> Void) {\n if task == nil {\n scheduleTask(wait: now().timeIntervalSince(lastFinishTime) < duration, block: block)\n }\n }\n\n func scheduleTask(wait: Bool, block: @escaping @Sendable () async -> Void) {\n task = Task.detached { [weak self] in\n guard let self else { return }\n do {\n if wait {\n try await Task.sleep(nanoseconds: UInt64(duration * 1_000_000_000))\n }\n await block()\n await finishTask()\n } catch {\n await finishTask()\n }\n }\n }\n\n func finishTask() {\n task = nil\n lastFinishTime = now()\n }\n}\n\n" }, { "path": "Tool/Sources/FileSystem/ByteString.swift", "content": "/*\n This source file is part of the Swift.org open source project\n\n Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors\n Licensed under Apache License v2.0 with Runtime Library Exception\n\n See http://swift.org/LICENSE.txt for license information\n See http://swift.org/CONTRIBUTORS.txt for Swift project authors\n*/\n\nimport Foundation\n\n/// A `ByteString` represents a sequence of bytes.\n///\n/// This struct provides useful operations for working with buffers of\n/// bytes. Conceptually it is just a contiguous array of bytes (UInt8), but it\n/// contains methods and default behavior suitable for common operations done\n/// using bytes strings.\n///\n/// This struct *is not* intended to be used for significant mutation of byte\n/// strings, we wish to retain the flexibility to micro-optimize the memory\n/// allocation of the storage (for example, by inlining the storage for small\n/// strings or and by eliminating wasted space in growable arrays). For\n/// construction of byte arrays, clients should use the `WritableByteStream` class\n/// and then convert to a `ByteString` when complete.\npublic struct ByteString: ExpressibleByArrayLiteral, Hashable, Sendable {\n /// The buffer contents.\n @usableFromInline\n internal var _bytes: [UInt8]\n\n /// Create an empty byte string.\n @inlinable\n public init() {\n _bytes = []\n }\n\n /// Create a byte string from a byte array literal.\n @inlinable\n public init(arrayLiteral contents: UInt8...) {\n _bytes = contents\n }\n\n /// Create a byte string from an array of bytes.\n @inlinable\n public init(_ contents: [UInt8]) {\n _bytes = contents\n }\n\n /// Create a byte string from an array slice.\n @inlinable\n public init(_ contents: ArraySlice) {\n _bytes = Array(contents)\n }\n\n /// Create a byte string from an byte buffer.\n @inlinable\n public init (_ contents: S) where S.Iterator.Element == UInt8 {\n _bytes = [UInt8](contents)\n }\n\n /// Create a byte string from the UTF8 encoding of a string.\n @inlinable\n public init(encodingAsUTF8 string: String) {\n _bytes = [UInt8](string.utf8)\n }\n\n /// Access the byte string contents as an array.\n @inlinable\n public var contents: [UInt8] {\n return _bytes\n }\n\n /// Return the byte string size.\n @inlinable\n public var count: Int {\n return _bytes.count\n }\n\n /// Gives a non-escaping closure temporary access to an immutable `Data` instance wrapping the `ByteString` without\n /// copying any memory around.\n ///\n /// - Parameters:\n /// - closure: The closure that will have access to a `Data` instance for the duration of its lifetime.\n @inlinable\n public func withData(_ closure: (Data) throws -> T) rethrows -> T {\n return try _bytes.withUnsafeBytes { pointer -> T in\n let mutatingPointer = UnsafeMutableRawPointer(mutating: pointer.baseAddress!)\n let data = Data(bytesNoCopy: mutatingPointer, count: pointer.count, deallocator: .none)\n return try closure(data)\n }\n }\n\n /// Returns a `String` lowercase hexadecimal representation of the contents of the `ByteString`.\n @inlinable\n public var hexadecimalRepresentation: String {\n _bytes.reduce(\"\") {\n var str = String($1, radix: 16)\n // The above method does not do zero padding.\n if str.count == 1 {\n str = \"0\" + str\n }\n return $0 + str\n }\n }\n}\n\n/// Conform to CustomDebugStringConvertible.\nextension ByteString: CustomStringConvertible {\n /// Return the string decoded as a UTF8 sequence, or traps if not possible.\n public var description: String {\n return cString\n }\n\n /// Return the string decoded as a UTF8 sequence, if possible.\n @inlinable\n public var validDescription: String? {\n // FIXME: This is very inefficient, we need a way to pass a buffer. It\n // is also wrong if the string contains embedded '\\0' characters.\n let tmp = _bytes + [UInt8(0)]\n return tmp.withUnsafeBufferPointer { ptr in\n return String(validatingUTF8: unsafeBitCast(ptr.baseAddress, to: UnsafePointer.self))\n }\n }\n\n /// Return the string decoded as a UTF8 sequence, substituting replacement\n /// characters for ill-formed UTF8 sequences.\n @inlinable\n public var cString: String {\n return String(decoding: _bytes, as: Unicode.UTF8.self)\n }\n\n @available(*, deprecated, message: \"use description or validDescription instead\")\n public var asString: String? {\n return validDescription\n }\n}\n\n/// ByteStreamable conformance for a ByteString.\nextension ByteString: ByteStreamable {\n @inlinable\n public func write(to stream: WritableByteStream) {\n stream.write(_bytes)\n }\n}\n\n/// StringLiteralConvertible conformance for a ByteString.\nextension ByteString: ExpressibleByStringLiteral {\n public typealias UnicodeScalarLiteralType = StringLiteralType\n public typealias ExtendedGraphemeClusterLiteralType = StringLiteralType\n\n public init(unicodeScalarLiteral value: UnicodeScalarLiteralType) {\n _bytes = [UInt8](value.utf8)\n }\n public init(extendedGraphemeClusterLiteral value: ExtendedGraphemeClusterLiteralType) {\n _bytes = [UInt8](value.utf8)\n }\n public init(stringLiteral value: StringLiteralType) {\n _bytes = [UInt8](value.utf8)\n }\n}\n" }, { "path": "Tool/Sources/FileSystem/FileInfo.swift", "content": "/*\n This source file is part of the Swift.org open source project\n\n Copyright (c) 2014 - 2023 Apple Inc. and the Swift project authors\n Licensed under Apache License v2.0 with Runtime Library Exception\n\n See http://swift.org/LICENSE.txt for license information\n See http://swift.org/CONTRIBUTORS.txt for Swift project authors\n */\n\nimport Foundation\n\n#if swift(<5.6)\nextension FileAttributeType: UnsafeSendable {}\nextension Date: UnsafeSendable {}\n#endif\n\n/// File system information for a particular file.\npublic struct FileInfo: Equatable, Codable, Sendable {\n\n /// The device number.\n public let device: UInt64\n\n /// The inode number.\n public let inode: UInt64\n\n /// The size of the file.\n public let size: UInt64\n\n /// The modification time of the file.\n public let modTime: Date\n\n /// Kind of file system entity.\n public let posixPermissions: Int16\n\n /// Kind of file system entity.\n public let fileType: FileAttributeType\n\n public init(_ attrs: [FileAttributeKey : Any]) {\n let device = (attrs[.systemNumber] as? NSNumber)?.uint64Value\n assert(device != nil)\n self.device = device!\n\n let inode = attrs[.systemFileNumber] as? UInt64\n assert(inode != nil)\n self.inode = inode!\n\n let posixPermissions = (attrs[.posixPermissions] as? NSNumber)?.int16Value\n assert(posixPermissions != nil)\n self.posixPermissions = posixPermissions!\n\n let fileType = attrs[.type] as? FileAttributeType\n assert(fileType != nil)\n self.fileType = fileType!\n\n let size = attrs[.size] as? UInt64\n assert(size != nil)\n self.size = size!\n\n let modTime = attrs[.modificationDate] as? Date\n assert(modTime != nil)\n self.modTime = modTime!\n }\n}\n\nextension FileAttributeType: Codable {}\n" }, { "path": "Tool/Sources/FileSystem/FileSystem.swift", "content": "/*\n This source file is part of the Swift.org open source project\n\n Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors\n Licensed under Apache License v2.0 with Runtime Library Exception\n\n See http://swift.org/LICENSE.txt for license information\n See http://swift.org/CONTRIBUTORS.txt for Swift project authors\n */\n\nimport Dispatch\nimport Foundation\n\n#if canImport(Glibc)\n@_exported import Glibc\n#elseif canImport(Musl)\n@_exported import Musl\n#elseif os(Windows)\n@_exported import CRT\n@_exported import WinSDK\n#else\n@_exported import Darwin.C\n#endif\n\npublic struct FileSystemError: Error, Equatable, Sendable {\n public enum Kind: Equatable, Sendable {\n /// Access to the path is denied.\n ///\n /// This is used when an operation cannot be completed because a component of\n /// the path cannot be accessed.\n ///\n /// Used in situations that correspond to the POSIX EACCES error code.\n case invalidAccess\n\n /// IO Error encoding\n ///\n /// This is used when an operation cannot be completed due to an otherwise\n /// unspecified IO error.\n case ioError(code: Int32)\n\n /// Is a directory\n ///\n /// This is used when an operation cannot be completed because a component\n /// of the path which was expected to be a file was not.\n ///\n /// Used in situations that correspond to the POSIX EISDIR error code.\n case isDirectory\n\n /// No such path exists.\n ///\n /// This is used when a path specified does not exist, but it was expected\n /// to.\n ///\n /// Used in situations that correspond to the POSIX ENOENT error code.\n case noEntry\n\n /// Not a directory\n ///\n /// This is used when an operation cannot be completed because a component\n /// of the path which was expected to be a directory was not.\n ///\n /// Used in situations that correspond to the POSIX ENOTDIR error code.\n case notDirectory\n\n /// Unsupported operation\n ///\n /// This is used when an operation is not supported by the concrete file\n /// system implementation.\n case unsupported\n\n /// An unspecific operating system error at a given path.\n case unknownOSError\n\n /// File or folder already exists at destination.\n ///\n /// This is thrown when copying or moving a file or directory but the destination\n /// path already contains a file or folder.\n case alreadyExistsAtDestination\n\n /// If an unspecified error occurs when trying to change directories.\n case couldNotChangeDirectory\n\n /// If a mismatch is detected in byte count when writing to a file.\n case mismatchedByteCount(expected: Int, actual: Int)\n }\n\n /// The kind of the error being raised.\n public let kind: Kind\n\n /// The absolute path to the file associated with the error, if available.\n public let path: AbsolutePath?\n\n public init(_ kind: Kind, _ path: AbsolutePath? = nil) {\n self.kind = kind\n self.path = path\n }\n}\n\nextension FileSystemError: CustomNSError {\n public var errorUserInfo: [String: Any] {\n return [NSLocalizedDescriptionKey: \"\\(self)\"]\n }\n}\n\npublic extension FileSystemError {\n init(errno: Int32, _ path: AbsolutePath) {\n switch errno {\n case EACCES:\n self.init(.invalidAccess, path)\n case EISDIR:\n self.init(.isDirectory, path)\n case ENOENT:\n self.init(.noEntry, path)\n case ENOTDIR:\n self.init(.notDirectory, path)\n case EEXIST:\n self.init(.alreadyExistsAtDestination, path)\n default:\n self.init(.ioError(code: errno), path)\n }\n }\n}\n\n/// Defines the file modes.\npublic enum FileMode: Sendable {\n public enum Option: Int, Sendable {\n case recursive\n case onlyFiles\n }\n\n case userUnWritable\n case userWritable\n case executable\n\n public func setMode(_ originalMode: Int16) -> Int16 {\n switch self {\n case .userUnWritable:\n // r-x rwx rwx\n return originalMode & 0o577\n case .userWritable:\n // -w- --- ---\n return originalMode | 0o200\n case .executable:\n // --x --x --x\n return originalMode | 0o111\n }\n }\n}\n\n/// Extended file system attributes that can applied to a given file path. See also\n/// ``FileSystem/hasAttribute(_:_:)``.\npublic enum FileSystemAttribute: RawRepresentable {\n #if canImport(Darwin)\n case quarantine\n #endif\n\n public init?(rawValue: String) {\n switch rawValue {\n #if canImport(Darwin)\n case \"com.apple.quarantine\":\n self = .quarantine\n #endif\n default:\n return nil\n }\n }\n\n public var rawValue: String {\n switch self {\n #if canImport(Darwin)\n case .quarantine:\n return \"com.apple.quarantine\"\n #endif\n }\n }\n}\n\n// FIXME: Design an asynchronous story?\n//\n/// Abstracted access to file system operations.\n///\n/// This protocol is used to allow most of the codebase to interact with a\n/// natural filesystem interface, while still allowing clients to transparently\n/// substitute a virtual file system or redirect file system operations.\n///\n/// - Note: All of these APIs are synchronous and can block.\npublic protocol FileSystem: Sendable {\n /// Check whether the given path exists and is accessible.\n @_disfavoredOverload\n func exists(_ path: AbsolutePath, followSymlink: Bool) -> Bool\n\n /// Check whether the given path is accessible and a directory.\n func isDirectory(_ path: AbsolutePath) -> Bool\n\n /// Check whether the given path is accessible and a file.\n func isFile(_ path: AbsolutePath) -> Bool\n\n /// Check whether the given path is an accessible and executable file.\n func isExecutableFile(_ path: AbsolutePath) -> Bool\n\n /// Check whether the given path is accessible and is a symbolic link.\n func isSymlink(_ path: AbsolutePath) -> Bool\n\n /// Check whether the given path is accessible and readable.\n func isReadable(_ path: AbsolutePath) -> Bool\n\n /// Check whether the given path is accessible and writable.\n func isWritable(_ path: AbsolutePath) -> Bool\n\n /// Returns any known item replacement directories for a given path. These may be used by\n /// platform-specific\n /// libraries to handle atomic file system operations, such as deletion.\n func itemReplacementDirectories(for path: AbsolutePath) throws -> [AbsolutePath]\n\n @available(*, deprecated, message: \"use `hasAttribute(_:_:)` instead\")\n func hasQuarantineAttribute(_ path: AbsolutePath) -> Bool\n\n /// Returns `true` if a given path has an attribute with a given name applied when file system\n /// supports this\n /// attribute. Returns `false` if such attribute is not applied or it isn't supported.\n func hasAttribute(_ name: FileSystemAttribute, _ path: AbsolutePath) -> Bool\n\n // FIXME: Actual file system interfaces will allow more efficient access to\n // more data than just the name here.\n //\n /// Get the contents of the given directory, in an undefined order.\n func _getDirectoryContents(\n _ path: AbsolutePath,\n includingPropertiesForKeys: [URLResourceKey]?,\n options: FileManager.DirectoryEnumerationOptions\n ) throws -> [AbsolutePath]\n\n /// Get the current working directory (similar to `getcwd(3)`), which can be\n /// different for different (virtualized) implementations of a FileSystem.\n /// The current working directory can be empty if e.g. the directory became\n /// unavailable while the current process was still working in it.\n /// This follows the POSIX `getcwd(3)` semantics.\n @_disfavoredOverload\n var currentWorkingDirectory: AbsolutePath? { get }\n\n /// Change the current working directory.\n /// - Parameters:\n /// - path: The path to the directory to change the current working directory to.\n func changeCurrentWorkingDirectory(to path: AbsolutePath) throws\n\n /// Get the home directory of current user\n @_disfavoredOverload\n var homeDirectory: AbsolutePath { get throws }\n\n /// Get the caches directory of current user\n @_disfavoredOverload\n var cachesDirectory: AbsolutePath? { get }\n\n /// Get the temp directory\n @_disfavoredOverload\n var tempDirectory: AbsolutePath { get throws }\n\n /// Create the given directory.\n func createDirectory(_ path: AbsolutePath) throws\n\n /// Create the given directory.\n ///\n /// - recursive: If true, create missing parent directories if possible.\n func createDirectory(_ path: AbsolutePath, recursive: Bool) throws\n\n /// Creates a symbolic link of the source path at the target path\n /// - Parameters:\n /// - path: The path at which to create the link.\n /// - destination: The path to which the link points to.\n /// - relative: If `relative` is true, the symlink contents will be a relative path, otherwise\n /// it will be absolute.\n func createSymbolicLink(\n _ path: AbsolutePath,\n pointingAt destination: AbsolutePath,\n relative: Bool\n ) throws\n \n func data(_ path: AbsolutePath) throws -> Data\n \n // FIXME: This is obviously not a very efficient or flexible API.\n //\n /// Get the contents of a file.\n ///\n /// - Returns: The file contents as bytes, or nil if missing.\n func readFileContents(_ path: AbsolutePath) throws -> ByteString\n\n // FIXME: This is obviously not a very efficient or flexible API.\n //\n /// Write the contents of a file.\n func writeFileContents(_ path: AbsolutePath, bytes: ByteString) throws\n\n // FIXME: This is obviously not a very efficient or flexible API.\n //\n /// Write the contents of a file.\n func writeFileContents(_ path: AbsolutePath, bytes: ByteString, atomically: Bool) throws\n\n /// Recursively deletes the file system entity at `path`.\n ///\n /// If there is no file system entity at `path`, this function does nothing (in particular, this\n /// is not considered\n /// to be an error).\n func removeFileTree(_ path: AbsolutePath) throws\n\n /// Change file mode.\n func chmod(_ mode: FileMode, path: AbsolutePath, options: Set) throws\n\n /// Returns the file info of the given path.\n ///\n /// The method throws if the underlying stat call fails.\n func getFileInfo(_ path: AbsolutePath) throws -> FileInfo\n\n /// Copy a file or directory.\n func copy(from sourcePath: AbsolutePath, to destinationPath: AbsolutePath) throws\n\n /// Move a file or directory.\n func move(from sourcePath: AbsolutePath, to destinationPath: AbsolutePath) throws\n\n /// Execute the given block while holding the lock.\n func withLock(\n on path: AbsolutePath,\n type: FileLock.LockType,\n blocking: Bool,\n _ body: () throws -> T\n ) throws -> T\n\n /// Execute the given block while holding the lock.\n func withLock(\n on path: AbsolutePath,\n type: FileLock.LockType,\n blocking: Bool,\n _ body: () async throws -> T\n ) async throws -> T\n}\n\n/// Convenience implementations (default arguments aren't permitted in protocol\n/// methods).\npublic extension FileSystem {\n /// exists override with default value.\n @_disfavoredOverload\n func exists(_ path: AbsolutePath) -> Bool {\n return exists(path, followSymlink: true)\n }\n\n /// Default implementation of createDirectory(_:)\n func createDirectory(_ path: AbsolutePath) throws {\n try createDirectory(path, recursive: false)\n }\n\n // Change file mode.\n func chmod(_ mode: FileMode, path: AbsolutePath) throws {\n try chmod(mode, path: path, options: [])\n }\n\n // Unless the file system type provides an override for this method, throw\n // if `atomically` is `true`, otherwise fall back to whatever implementation already exists.\n @_disfavoredOverload\n func writeFileContents(_ path: AbsolutePath, bytes: ByteString, atomically: Bool) throws {\n guard !atomically else {\n throw FileSystemError(.unsupported, path)\n }\n try writeFileContents(path, bytes: bytes)\n }\n\n /// Write to a file from a stream producer.\n @_disfavoredOverload\n func writeFileContents(_ path: AbsolutePath, body: (WritableByteStream) -> Void) throws {\n let contents = BufferedOutputByteStream()\n body(contents)\n try createDirectory(path.parentDirectory, recursive: true)\n try writeFileContents(path, bytes: contents.bytes)\n }\n\n func getFileInfo(_ path: AbsolutePath) throws -> FileInfo {\n throw FileSystemError(.unsupported, path)\n }\n\n func withLock(on path: AbsolutePath, _ body: () throws -> T) throws -> T {\n return try withLock(on: path, type: .exclusive, body)\n }\n\n func withLock(\n on path: AbsolutePath,\n type: FileLock.LockType,\n _ body: () throws -> T\n ) throws -> T {\n return try withLock(on: path, type: type, blocking: true, body)\n }\n\n func withLock(\n on path: AbsolutePath,\n type: FileLock.LockType,\n blocking: Bool,\n _ body: () throws -> T\n ) throws -> T {\n throw FileSystemError(.unsupported, path)\n }\n\n func withLock(\n on path: AbsolutePath,\n type: FileLock.LockType,\n _ body: () async throws -> T\n ) async throws -> T {\n return try await withLock(on: path, type: type, blocking: true, body)\n }\n\n func withLock(\n on path: AbsolutePath,\n type: FileLock.LockType,\n blocking: Bool,\n _ body: () async throws -> T\n ) async throws -> T {\n throw FileSystemError(.unsupported, path)\n }\n\n func hasQuarantineAttribute(_: AbsolutePath) -> Bool { false }\n\n func hasAttribute(_: FileSystemAttribute, _: AbsolutePath) -> Bool { false }\n\n func itemReplacementDirectories(for path: AbsolutePath) throws -> [AbsolutePath] { [] }\n}\n\n/// Concrete FileSystem implementation which communicates with the local file system.\nprivate struct LocalFileSystem: FileSystem {\n func isExecutableFile(_ path: AbsolutePath) -> Bool {\n // Our semantics doesn't consider directories.\n return (isFile(path) || isSymlink(path)) && FileManager.default\n .isExecutableFile(atPath: path.pathString)\n }\n\n func exists(_ path: AbsolutePath, followSymlink: Bool) -> Bool {\n if followSymlink {\n return FileManager.default.fileExists(atPath: path.pathString)\n }\n return (try? FileManager.default.attributesOfItem(atPath: path.pathString)) != nil\n }\n\n func isDirectory(_ path: AbsolutePath) -> Bool {\n var isDirectory: ObjCBool = false\n let exists: Bool = FileManager.default.fileExists(\n atPath: path.pathString,\n isDirectory: &isDirectory\n )\n return exists && isDirectory.boolValue\n }\n\n func isFile(_ path: AbsolutePath) -> Bool {\n guard let path = try? resolveSymlinks(path) else {\n return false\n }\n let attrs = try? FileManager.default.attributesOfItem(atPath: path.pathString)\n return attrs?[.type] as? FileAttributeType == .typeRegular\n }\n\n func isSymlink(_ path: AbsolutePath) -> Bool {\n let url = NSURL(fileURLWithPath: path.pathString)\n // We are intentionally using `NSURL.resourceValues(forKeys:)` here since it improves\n // performance on Darwin platforms.\n let result = try? url.resourceValues(forKeys: [.isSymbolicLinkKey])\n return (result?[.isSymbolicLinkKey] as? Bool) == true\n }\n\n func isReadable(_ path: AbsolutePath) -> Bool {\n FileManager.default.isReadableFile(atPath: path.pathString)\n }\n\n func isWritable(_ path: AbsolutePath) -> Bool {\n FileManager.default.isWritableFile(atPath: path.pathString)\n }\n\n func getFileInfo(_ path: AbsolutePath) throws -> FileInfo {\n let attrs = try FileManager.default.attributesOfItem(atPath: path.pathString)\n return FileInfo(attrs)\n }\n\n func hasAttribute(_ name: FileSystemAttribute, _ path: AbsolutePath) -> Bool {\n #if canImport(Darwin)\n let bufLength = getxattr(path.pathString, name.rawValue, nil, 0, 0, 0)\n\n return bufLength > 0\n #else\n return false\n #endif\n }\n\n var currentWorkingDirectory: AbsolutePath? {\n let cwdStr = FileManager.default.currentDirectoryPath\n\n #if _runtime(_ObjC)\n // The ObjC runtime indicates that the underlying Foundation has ObjC\n // interoperability in which case the return type of\n // `fileSystemRepresentation` is different from the Swift implementation\n // of Foundation.\n return try? AbsolutePath(validating: cwdStr)\n #else\n let fsr: UnsafePointer = cwdStr.fileSystemRepresentation\n defer { fsr.deallocate() }\n\n return try? AbsolutePath(String(cString: fsr))\n #endif\n }\n\n func changeCurrentWorkingDirectory(to path: AbsolutePath) throws {\n guard isDirectory(path) else {\n throw FileSystemError(.notDirectory, path)\n }\n\n guard FileManager.default.changeCurrentDirectoryPath(path.pathString) else {\n throw FileSystemError(.couldNotChangeDirectory, path)\n }\n }\n\n var homeDirectory: AbsolutePath {\n get throws {\n return try AbsolutePath(validating: NSHomeDirectory())\n }\n }\n\n var cachesDirectory: AbsolutePath? {\n return FileManager.default.urls(for: .cachesDirectory, in: .userDomainMask).first\n .flatMap { try? AbsolutePath(validating: $0.path) }\n }\n\n var tempDirectory: AbsolutePath {\n get throws {\n return try AbsolutePath(validating: NSTemporaryDirectory())\n }\n }\n\n func _getDirectoryContents(\n _ path: AbsolutePath,\n includingPropertiesForKeys: [URLResourceKey]?,\n options: FileManager.DirectoryEnumerationOptions\n ) throws -> [AbsolutePath] {\n return try FileManager.default.contentsOfDirectory(\n at: URL(fileURLWithPath: path.pathString),\n includingPropertiesForKeys: includingPropertiesForKeys,\n options: options\n ).compactMap { try? AbsolutePath(validating: $0.path) }\n }\n\n func createDirectory(_ path: AbsolutePath, recursive: Bool) throws {\n // Don't fail if path is already a directory.\n if isDirectory(path) { return }\n\n try FileManager.default.createDirectory(\n atPath: path.pathString,\n withIntermediateDirectories: recursive,\n attributes: [:]\n )\n }\n\n func createSymbolicLink(\n _ path: AbsolutePath,\n pointingAt destination: AbsolutePath,\n relative: Bool\n ) throws {\n let destString = relative ? destination.relative(to: path.parentDirectory)\n .pathString : destination.pathString\n try FileManager.default.createSymbolicLink(\n atPath: path.pathString,\n withDestinationPath: destString\n )\n }\n \n func data(_ path: AbsolutePath) throws -> Data {\n try Data(contentsOf: URL(fileURLWithPath: path.pathString))\n }\n\n func readFileContents(_ path: AbsolutePath) throws -> ByteString {\n // Open the file.\n guard let fp = fopen(path.pathString, \"rb\") else {\n throw FileSystemError(errno: errno, path)\n }\n defer { fclose(fp) }\n\n // Read the data one block at a time.\n let data = BufferedOutputByteStream()\n var tmpBuffer = [UInt8](repeating: 0, count: 1 << 12)\n while true {\n let n = fread(&tmpBuffer, 1, tmpBuffer.count, fp)\n if n < 0 {\n if errno == EINTR { continue }\n throw FileSystemError(.ioError(code: errno), path)\n }\n if n == 0 {\n let errno = ferror(fp)\n if errno != 0 {\n throw FileSystemError(.ioError(code: errno), path)\n }\n break\n }\n data.send(tmpBuffer[0..) throws {\n guard exists(path) else { return }\n func setMode(path: String) throws {\n let attrs = try FileManager.default.attributesOfItem(atPath: path)\n // Skip if only files should be changed.\n if options.contains(.onlyFiles) && attrs[.type] as? FileAttributeType != .typeRegular {\n return\n }\n\n // Compute the new mode for this file.\n let currentMode = attrs[.posixPermissions] as! Int16\n let newMode = mode.setMode(currentMode)\n guard newMode != currentMode else { return }\n try FileManager.default.setAttributes(\n [.posixPermissions: newMode],\n ofItemAtPath: path\n )\n }\n\n try setMode(path: path.pathString)\n guard isDirectory(path) else { return }\n\n guard let traverse = FileManager.default.enumerator(\n at: URL(fileURLWithPath: path.pathString),\n includingPropertiesForKeys: nil\n ) else {\n throw FileSystemError(.noEntry, path)\n }\n\n if !options.contains(.recursive) {\n traverse.skipDescendants()\n }\n\n while let path = traverse.nextObject() {\n try setMode(path: (path as! URL).path)\n }\n }\n\n func copy(from sourcePath: AbsolutePath, to destinationPath: AbsolutePath) throws {\n guard exists(sourcePath) else { throw FileSystemError(.noEntry, sourcePath) }\n guard !exists(destinationPath)\n else { throw FileSystemError(.alreadyExistsAtDestination, destinationPath) }\n try FileManager.default.copyItem(at: sourcePath.asURL, to: destinationPath.asURL)\n }\n\n func move(from sourcePath: AbsolutePath, to destinationPath: AbsolutePath) throws {\n guard exists(sourcePath) else { throw FileSystemError(.noEntry, sourcePath) }\n guard !exists(destinationPath)\n else { throw FileSystemError(.alreadyExistsAtDestination, destinationPath) }\n try FileManager.default.moveItem(at: sourcePath.asURL, to: destinationPath.asURL)\n }\n\n func withLock(\n on path: AbsolutePath,\n type: FileLock.LockType,\n blocking: Bool,\n _ body: () throws -> T\n ) throws -> T {\n try FileLock.withLock(fileToLock: path, type: type, blocking: blocking, body: body)\n }\n\n func withLock(\n on path: AbsolutePath,\n type: FileLock.LockType,\n blocking: Bool,\n _ body: () async throws -> T\n ) async throws -> T {\n try await FileLock.withLock(fileToLock: path, type: type, blocking: blocking, body: body)\n }\n\n func itemReplacementDirectories(for path: AbsolutePath) throws -> [AbsolutePath] {\n let result = try FileManager.default.url(\n for: .itemReplacementDirectory,\n in: .userDomainMask,\n appropriateFor: path.asURL,\n create: false\n )\n let path = try AbsolutePath(validating: result.path)\n // Foundation returns a path that is unique every time, so we return both that path, as well\n // as its parent.\n return [path, path.parentDirectory]\n }\n}\n\n/// Concrete FileSystem implementation which simulates an empty disk.\npublic final class InMemoryFileSystem: FileSystem {\n /// Private internal representation of a file system node.\n /// Not thread-safe.\n private class Node {\n /// The actual node data.\n let contents: NodeContents\n\n init(_ contents: NodeContents) {\n self.contents = contents\n }\n\n /// Creates deep copy of the object.\n func copy() -> Node {\n return Node(contents.copy())\n }\n }\n\n /// Private internal representation the contents of a file system node.\n /// Not thread-safe.\n private enum NodeContents {\n case file(ByteString)\n case directory(DirectoryContents)\n case symlink(String)\n\n /// Creates deep copy of the object.\n func copy() -> NodeContents {\n switch self {\n case let .file(bytes):\n return .file(bytes)\n case let .directory(contents):\n return .directory(contents.copy())\n case let .symlink(path):\n return .symlink(path)\n }\n }\n }\n\n /// Private internal representation the contents of a directory.\n /// Not thread-safe.\n private final class DirectoryContents {\n var entries: [String: Node]\n\n init(entries: [String: Node] = [:]) {\n self.entries = entries\n }\n\n /// Creates deep copy of the object.\n func copy() -> DirectoryContents {\n let contents = DirectoryContents()\n for (key, node) in entries {\n contents.entries[key] = node.copy()\n }\n return contents\n }\n }\n\n /// The root node of the filesystem.\n private var root: Node\n\n /// Protects `root` and everything underneath it.\n /// FIXME: Using a single lock for this is a performance problem, but in\n /// reality, the only practical use for InMemoryFileSystem is for unit\n /// tests.\n private let lock = NSLock()\n /// A map that keeps weak references to all locked files.\n private var lockFiles = [AbsolutePath: WeakReference]()\n /// Used to access lockFiles in a thread safe manner.\n private let lockFilesLock = NSLock()\n\n /// Exclusive file system lock vended to clients through `withLock()`.\n /// Used to ensure that DispatchQueues are released when they are no longer in use.\n private struct WeakReference {\n weak var reference: Value?\n\n init(_ value: Value?) {\n reference = value\n }\n }\n\n public init() {\n root = Node(.directory(DirectoryContents()))\n }\n\n /// Creates deep copy of the object.\n public func copy() -> InMemoryFileSystem {\n return lock.withLock {\n let fs = InMemoryFileSystem()\n fs.root = root.copy()\n return fs\n }\n }\n\n /// Private function to look up the node corresponding to a path.\n /// Not thread-safe.\n private func getNode(_ path: AbsolutePath, followSymlink: Bool = true) throws -> Node? {\n func getNodeInternal(_ path: AbsolutePath) throws -> Node? {\n // If this is the root node, return it.\n if path.isRoot {\n return root\n }\n\n // Otherwise, get the parent node.\n guard let parent = try getNodeInternal(path.parentDirectory) else {\n return nil\n }\n\n // If we didn't find a directory, this is an error.\n guard case let .directory(contents) = parent.contents else {\n throw FileSystemError(.notDirectory, path.parentDirectory)\n }\n\n // Return the directory entry.\n let node = contents.entries[path.basename]\n\n switch node?.contents {\n case .directory, .file:\n return node\n case let .symlink(destination):\n let destination = try AbsolutePath(\n validating: destination,\n relativeTo: path.parentDirectory\n )\n return followSymlink ? try getNodeInternal(destination) : node\n case .none:\n return nil\n }\n }\n\n // Get the node that corresponds to the path.\n return try getNodeInternal(path)\n }\n\n // MARK: FileSystem Implementation\n\n public func exists(_ path: AbsolutePath, followSymlink: Bool) -> Bool {\n return lock.withLock {\n do {\n switch try getNode(path, followSymlink: followSymlink)?.contents {\n case .file, .directory, .symlink: return true\n case .none: return false\n }\n } catch {\n return false\n }\n }\n }\n\n public func isDirectory(_ path: AbsolutePath) -> Bool {\n return lock.withLock {\n do {\n if case .directory? = try getNode(path)?.contents {\n return true\n }\n return false\n } catch {\n return false\n }\n }\n }\n\n public func isFile(_ path: AbsolutePath) -> Bool {\n return lock.withLock {\n do {\n if case .file? = try getNode(path)?.contents {\n return true\n }\n return false\n } catch {\n return false\n }\n }\n }\n\n public func isSymlink(_ path: AbsolutePath) -> Bool {\n return lock.withLock {\n do {\n if case .symlink? = try getNode(path, followSymlink: false)?.contents {\n return true\n }\n return false\n } catch {\n return false\n }\n }\n }\n\n public func isReadable(_ path: AbsolutePath) -> Bool {\n exists(path)\n }\n\n public func isWritable(_ path: AbsolutePath) -> Bool {\n exists(path)\n }\n\n public func isExecutableFile(_: AbsolutePath) -> Bool {\n // FIXME: Always return false until in-memory implementation\n // gets permission semantics.\n return false\n }\n\n /// Virtualized current working directory.\n public var currentWorkingDirectory: AbsolutePath? {\n return try? AbsolutePath(validating: \"/\")\n }\n\n public func changeCurrentWorkingDirectory(to path: AbsolutePath) throws {\n throw FileSystemError(.unsupported, path)\n }\n\n public var homeDirectory: AbsolutePath {\n get throws {\n // FIXME: Maybe we should allow setting this when creating the fs.\n return try AbsolutePath(validating: \"/home/user\")\n }\n }\n\n public var cachesDirectory: AbsolutePath? {\n return try? homeDirectory.appending(component: \"caches\")\n }\n\n public var tempDirectory: AbsolutePath {\n get throws {\n return try AbsolutePath(validating: \"/tmp\")\n }\n }\n\n public func _getDirectoryContents(\n _ path: AbsolutePath,\n includingPropertiesForKeys: [URLResourceKey]?,\n options: FileManager.DirectoryEnumerationOptions\n ) throws -> [AbsolutePath] {\n return try lock.withLock {\n guard let node = try getNode(path) else {\n throw FileSystemError(.noEntry, path)\n }\n guard case let .directory(contents) = node.contents else {\n throw FileSystemError(.notDirectory, path)\n }\n\n // FIXME: Perhaps we should change the protocol to allow lazy behavior.\n return [String](contents.entries.keys).map {\n path.appending(component: $0)\n }\n }\n }\n\n /// Not thread-safe.\n private func _createDirectory(_ path: AbsolutePath, recursive: Bool) throws {\n // Ignore if client passes root.\n guard !path.isRoot else {\n return\n }\n // Get the parent directory node.\n let parentPath = path.parentDirectory\n guard let parent = try getNode(parentPath) else {\n // If the parent doesn't exist, and we are recursive, then attempt\n // to create the parent and retry.\n if recursive && path != parentPath {\n // Attempt to create the parent.\n try _createDirectory(parentPath, recursive: true)\n\n // Re-attempt creation, non-recursively.\n return try _createDirectory(path, recursive: false)\n } else {\n // Otherwise, we failed.\n throw FileSystemError(.noEntry, parentPath)\n }\n }\n\n // Check that the parent is a directory.\n guard case let .directory(contents) = parent.contents else {\n // The parent isn't a directory, this is an error.\n throw FileSystemError(.notDirectory, parentPath)\n }\n\n // Check if the node already exists.\n if let node = contents.entries[path.basename] {\n // Verify it is a directory.\n guard case .directory = node.contents else {\n // The path itself isn't a directory, this is an error.\n throw FileSystemError(.notDirectory, path)\n }\n\n // We are done.\n return\n }\n\n // Otherwise, the node does not exist, create it.\n contents.entries[path.basename] = Node(.directory(DirectoryContents()))\n }\n\n public func createDirectory(_ path: AbsolutePath, recursive: Bool) throws {\n return try lock.withLock {\n try _createDirectory(path, recursive: recursive)\n }\n }\n\n public func createSymbolicLink(\n _ path: AbsolutePath,\n pointingAt destination: AbsolutePath,\n relative: Bool\n ) throws {\n return try lock.withLock {\n // Create directory to destination parent.\n guard let destinationParent = try getNode(path.parentDirectory) else {\n throw FileSystemError(.noEntry, path.parentDirectory)\n }\n\n // Check that the parent is a directory.\n guard case let .directory(contents) = destinationParent.contents else {\n throw FileSystemError(.notDirectory, path.parentDirectory)\n }\n\n guard contents.entries[path.basename] == nil else {\n throw FileSystemError(.alreadyExistsAtDestination, path)\n }\n\n let destination = relative ? destination.relative(to: path.parentDirectory)\n .pathString : destination.pathString\n\n contents.entries[path.basename] = Node(.symlink(destination))\n }\n }\n \n public func data(_ path: AbsolutePath) throws -> Data {\n return try lock.withLock {\n // Get the node.\n guard let node = try getNode(path) else {\n throw FileSystemError(.noEntry, path)\n }\n\n // Check that the node is a file.\n guard case let .file(contents) = node.contents else {\n // The path is a directory, this is an error.\n throw FileSystemError(.isDirectory, path)\n }\n\n // Return the file contents.\n return contents.withData { $0 }\n }\n }\n\n public func readFileContents(_ path: AbsolutePath) throws -> ByteString {\n return try lock.withLock {\n // Get the node.\n guard let node = try getNode(path) else {\n throw FileSystemError(.noEntry, path)\n }\n\n // Check that the node is a file.\n guard case let .file(contents) = node.contents else {\n // The path is a directory, this is an error.\n throw FileSystemError(.isDirectory, path)\n }\n\n // Return the file contents.\n return contents\n }\n }\n\n public func writeFileContents(_ path: AbsolutePath, bytes: ByteString) throws {\n return try lock.withLock {\n // It is an error if this is the root node.\n let parentPath = path.parentDirectory\n guard path != parentPath else {\n throw FileSystemError(.isDirectory, path)\n }\n\n // Get the parent node.\n guard let parent = try getNode(parentPath) else {\n throw FileSystemError(.noEntry, parentPath)\n }\n\n // Check that the parent is a directory.\n guard case let .directory(contents) = parent.contents else {\n // The parent isn't a directory, this is an error.\n throw FileSystemError(.notDirectory, parentPath)\n }\n\n // Check if the node exists.\n if let node = contents.entries[path.basename] {\n // Verify it is a file.\n guard case .file = node.contents else {\n // The path is a directory, this is an error.\n throw FileSystemError(.isDirectory, path)\n }\n }\n\n // Write the file.\n contents.entries[path.basename] = Node(.file(bytes))\n }\n }\n\n public func writeFileContents(\n _ path: AbsolutePath,\n bytes: ByteString,\n atomically: Bool\n ) throws {\n // In memory file system's writeFileContents is already atomic, so ignore the parameter here\n // and just call the base implementation.\n try writeFileContents(path, bytes: bytes)\n }\n\n public func removeFileTree(_ path: AbsolutePath) throws {\n return lock.withLock {\n // Ignore root and get the parent node's content if its a directory.\n guard !path.isRoot,\n let parent = try? getNode(path.parentDirectory),\n case let .directory(contents) = parent.contents\n else {\n return\n }\n // Set it to nil to release the contents.\n contents.entries[path.basename] = nil\n }\n }\n\n public func chmod(_ mode: FileMode, path: AbsolutePath, options: Set) throws {\n // FIXME: We don't have these semantics in InMemoryFileSystem.\n }\n\n /// Private implementation of core copying function.\n /// Not thread-safe.\n private func _copy(from sourcePath: AbsolutePath, to destinationPath: AbsolutePath) throws {\n // Get the source node.\n guard let source = try getNode(sourcePath) else {\n throw FileSystemError(.noEntry, sourcePath)\n }\n\n // Create directory to destination parent.\n guard let destinationParent = try getNode(destinationPath.parentDirectory) else {\n throw FileSystemError(.noEntry, destinationPath.parentDirectory)\n }\n\n // Check that the parent is a directory.\n guard case let .directory(contents) = destinationParent.contents else {\n throw FileSystemError(.notDirectory, destinationPath.parentDirectory)\n }\n\n guard contents.entries[destinationPath.basename] == nil else {\n throw FileSystemError(.alreadyExistsAtDestination, destinationPath)\n }\n\n contents.entries[destinationPath.basename] = source\n }\n\n public func copy(from sourcePath: AbsolutePath, to destinationPath: AbsolutePath) throws {\n return try lock.withLock {\n try _copy(from: sourcePath, to: destinationPath)\n }\n }\n\n public func move(from sourcePath: AbsolutePath, to destinationPath: AbsolutePath) throws {\n return try lock.withLock {\n // Get the source parent node.\n guard let sourceParent = try getNode(sourcePath.parentDirectory) else {\n throw FileSystemError(.noEntry, sourcePath.parentDirectory)\n }\n\n // Check that the parent is a directory.\n guard case let .directory(contents) = sourceParent.contents else {\n throw FileSystemError(.notDirectory, sourcePath.parentDirectory)\n }\n\n try _copy(from: sourcePath, to: destinationPath)\n\n contents.entries[sourcePath.basename] = nil\n }\n }\n\n public func withLock(\n on path: AbsolutePath,\n type: FileLock.LockType,\n blocking: Bool,\n _ body: () throws -> T\n ) throws -> T {\n if !blocking {\n throw FileSystemError(.unsupported, path)\n }\n\n let resolvedPath: AbsolutePath = try lock.withLock {\n if case let .symlink(destination) = try getNode(path)?.contents {\n return try AbsolutePath(validating: destination, relativeTo: path.parentDirectory)\n } else {\n return path\n }\n }\n\n let fileQueue: DispatchQueue = lockFilesLock.withLock {\n if let queueReference = lockFiles[resolvedPath], let queue = queueReference.reference {\n return queue\n } else {\n let queue = DispatchQueue(\n label: \"org.swift.swiftpm.in-memory-file-system.file-queue\",\n attributes: .concurrent\n )\n lockFiles[resolvedPath] = WeakReference(queue)\n return queue\n }\n }\n\n return try fileQueue.sync(flags: type == .exclusive ? .barrier : .init(), execute: body)\n }\n}\n\n// Internal state of `InMemoryFileSystem` is protected with a lock in all of its `public` methods.\n#if compiler(>=5.7)\nextension InMemoryFileSystem: @unchecked Sendable {}\n#else\nextension InMemoryFileSystem: UnsafeSendable {}\n#endif\n\nprivate var _localFileSystem: FileSystem = LocalFileSystem()\n\n/// Public access to the local FS proxy.\npublic var localFileSystem: FileSystem {\n return _localFileSystem\n}\n\npublic extension FileSystem {\n /// Print the filesystem tree of the given path.\n ///\n /// For debugging only.\n func dumpTree(at path: AbsolutePath = .root) {\n print(\".\")\n do {\n try recurse(fs: self, path: path)\n } catch {\n print(\"\\(error)\")\n }\n }\n\n /// Write bytes to the path if the given contents are different.\n func writeIfChanged(path: AbsolutePath, bytes: ByteString) throws {\n try createDirectory(path.parentDirectory, recursive: true)\n\n // Return if the contents are same.\n if isFile(path), try readFileContents(path) == bytes {\n return\n }\n\n try writeFileContents(path, bytes: bytes)\n }\n\n func getDirectoryContents(\n at path: AbsolutePath,\n includingPropertiesForKeys: [URLResourceKey]? = nil,\n options: FileManager.DirectoryEnumerationOptions = []\n ) throws -> [AbsolutePath] {\n return try _getDirectoryContents(\n path,\n includingPropertiesForKeys: includingPropertiesForKeys,\n options: options\n )\n }\n\n /// Helper method to recurse and print the tree.\n private func recurse(fs: FileSystem, path: AbsolutePath, prefix: String = \"\") throws {\n let contents = (try fs.getDirectoryContents(at: path)).map(\\.basename)\n\n for (idx, entry) in contents.enumerated() {\n let isLast = idx == contents.count - 1\n let line = prefix + (isLast ? \"โ””โ”€โ”€ \" : \"โ”œโ”€โ”€ \") + entry\n print(line)\n\n let entryPath = path.appending(component: entry)\n if fs.isDirectory(entryPath) {\n let childPrefix = prefix + (isLast ? \" \" : \"โ”‚ \")\n try recurse(fs: fs, path: entryPath, prefix: String(childPrefix))\n }\n }\n }\n}\n\n" }, { "path": "Tool/Sources/FileSystem/Lock.swift", "content": "import Foundation\n\npublic enum ProcessLockError: Error {\n case unableToAcquireLock(errno: Int32)\n}\n\nextension ProcessLockError: CustomNSError {\n public var errorUserInfo: [String : Any] {\n return [NSLocalizedDescriptionKey: \"\\(self)\"]\n }\n}\n\n/// Provides functionality to acquire a lock on a file via POSIX's flock() method.\n/// It can be used for things like serializing concurrent mutations on a shared resource\n/// by multiple instances of a process. The `FileLock` is not thread-safe.\npublic final class FileLock {\n\n public enum LockType {\n case exclusive\n case shared\n }\n\n /// File descriptor to the lock file.\n #if os(Windows)\n private var handle: HANDLE?\n #else\n private var fileDescriptor: CInt?\n #endif\n\n /// Path to the lock file.\n private let lockFile: AbsolutePath\n\n /// Create an instance of FileLock at the path specified\n ///\n /// Note: The parent directory path should be a valid directory.\n internal init(at lockFile: AbsolutePath) {\n self.lockFile = lockFile\n }\n\n @available(*, deprecated, message: \"use init(at:) instead\")\n public convenience init(name: String, cachePath: AbsolutePath) {\n self.init(at: cachePath.appending(component: name + \".lock\"))\n }\n\n /// Try to acquire a lock. This method will block until lock the already acquired by other process.\n ///\n /// Note: This method can throw if underlying POSIX methods fail.\n public func lock(type: LockType = .exclusive, blocking: Bool = true) throws {\n #if os(Windows)\n if handle == nil {\n let h: HANDLE = lockFile.pathString.withCString(encodedAs: UTF16.self, {\n CreateFileW(\n $0,\n UInt32(GENERIC_READ) | UInt32(GENERIC_WRITE),\n UInt32(FILE_SHARE_READ) | UInt32(FILE_SHARE_WRITE),\n nil,\n DWORD(OPEN_ALWAYS),\n DWORD(FILE_ATTRIBUTE_NORMAL),\n nil\n )\n })\n if h == INVALID_HANDLE_VALUE {\n throw FileSystemError(errno: Int32(GetLastError()), lockFile)\n }\n self.handle = h\n }\n var overlapped = OVERLAPPED()\n overlapped.Offset = 0\n overlapped.OffsetHigh = 0\n overlapped.hEvent = nil\n var dwFlags = Int32(0)\n switch type {\n case .exclusive: dwFlags |= LOCKFILE_EXCLUSIVE_LOCK\n case .shared: break\n }\n if !blocking {\n dwFlags |= LOCKFILE_FAIL_IMMEDIATELY\n }\n if !LockFileEx(handle, DWORD(dwFlags), 0,\n UInt32.max, UInt32.max, &overlapped) {\n throw ProcessLockError.unableToAcquireLock(errno: Int32(GetLastError()))\n }\n #else\n // Open the lock file.\n if fileDescriptor == nil {\n let fd = open(lockFile.pathString, O_WRONLY | O_CREAT | O_CLOEXEC, 0o666)\n if fd == -1 {\n throw FileSystemError(errno: errno, lockFile)\n }\n self.fileDescriptor = fd\n }\n var flags = Int32(0)\n switch type {\n case .exclusive: flags = LOCK_EX\n case .shared: flags = LOCK_SH\n }\n if !blocking {\n flags |= LOCK_NB\n }\n // Acquire lock on the file.\n while true {\n if flock(fileDescriptor!, flags) == 0 {\n break\n }\n // Retry if interrupted.\n if errno == EINTR { continue }\n throw ProcessLockError.unableToAcquireLock(errno: errno)\n }\n #endif\n }\n\n /// Unlock the held lock.\n public func unlock() {\n #if os(Windows)\n var overlapped = OVERLAPPED()\n overlapped.Offset = 0\n overlapped.OffsetHigh = 0\n overlapped.hEvent = nil\n UnlockFileEx(handle, 0, UInt32.max, UInt32.max, &overlapped)\n #else\n guard let fd = fileDescriptor else { return }\n flock(fd, LOCK_UN)\n #endif\n }\n\n deinit {\n #if os(Windows)\n guard let handle = handle else { return }\n CloseHandle(handle)\n #else\n guard let fd = fileDescriptor else { return }\n close(fd)\n #endif\n }\n\n /// Execute the given block while holding the lock.\n public func withLock(type: LockType = .exclusive, blocking: Bool = true, _ body: () throws -> T) throws -> T {\n try lock(type: type, blocking: blocking)\n defer { unlock() }\n return try body()\n }\n\n /// Execute the given block while holding the lock.\n public func withLock(type: LockType = .exclusive, blocking: Bool = true, _ body: () async throws -> T) async throws -> T {\n try lock(type: type, blocking: blocking)\n defer { unlock() }\n return try await body()\n }\n\n public static func prepareLock(\n fileToLock: AbsolutePath,\n at lockFilesDirectory: AbsolutePath? = nil\n ) throws -> FileLock {\n // unless specified, we use the tempDirectory to store lock files\n let lockFilesDirectory = try lockFilesDirectory ?? localFileSystem.tempDirectory\n if !localFileSystem.exists(lockFilesDirectory) {\n throw FileSystemError(.noEntry, lockFilesDirectory)\n }\n if !localFileSystem.isDirectory(lockFilesDirectory) {\n throw FileSystemError(.notDirectory, lockFilesDirectory)\n }\n // use the parent path to generate unique filename in temp\n var lockFileName = try (resolveSymlinks(fileToLock.parentDirectory)\n .appending(component: fileToLock.basename))\n .components.joined(separator: \"_\")\n .replacingOccurrences(of: \":\", with: \"_\") + \".lock\"\n#if os(Windows)\n // NTFS has an ARC limit of 255 codepoints\n var lockFileUTF16 = lockFileName.utf16.suffix(255)\n while String(lockFileUTF16) == nil {\n lockFileUTF16 = lockFileUTF16.dropFirst()\n }\n lockFileName = String(lockFileUTF16) ?? lockFileName\n#else\n if lockFileName.hasPrefix(AbsolutePath.root.pathString) {\n lockFileName = String(lockFileName.dropFirst(AbsolutePath.root.pathString.count))\n }\n // back off until it occupies at most `NAME_MAX` UTF-8 bytes but without splitting scalars\n // (we might split clusters but it's not worth the effort to keep them together as long as we get a valid file name)\n var lockFileUTF8 = lockFileName.utf8.suffix(Int(NAME_MAX))\n while String(lockFileUTF8) == nil {\n // in practice this will only be a few iterations\n lockFileUTF8 = lockFileUTF8.dropFirst()\n }\n // we will never end up with nil since we have ASCII characters at the end\n lockFileName = String(lockFileUTF8) ?? lockFileName\n#endif\n let lockFilePath = lockFilesDirectory.appending(component: lockFileName)\n\n return FileLock(at: lockFilePath)\n }\n\n public static func withLock(\n fileToLock: AbsolutePath,\n lockFilesDirectory: AbsolutePath? = nil,\n type: LockType = .exclusive,\n blocking: Bool = true,\n body: () throws -> T\n ) throws -> T {\n let lock = try Self.prepareLock(fileToLock: fileToLock, at: lockFilesDirectory)\n return try lock.withLock(type: type, blocking: blocking, body)\n }\n\n public static func withLock(\n fileToLock: AbsolutePath,\n lockFilesDirectory: AbsolutePath? = nil,\n type: LockType = .exclusive,\n blocking: Bool = true,\n body: () async throws -> T\n ) async throws -> T {\n let lock = try Self.prepareLock(fileToLock: fileToLock, at: lockFilesDirectory)\n return try await lock.withLock(type: type, blocking: blocking, body)\n }\n}\n" }, { "path": "Tool/Sources/FileSystem/Misc.swift", "content": "/*\n This source file is part of the Swift.org open source project\n\n Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors\n Licensed under Apache License v2.0 with Runtime Library Exception\n\n See http://swift.org/LICENSE.txt for license information\n See http://swift.org/CONTRIBUTORS.txt for Swift project authors\n*/\n\n#if canImport(Glibc)\n@_exported import Glibc\n#elseif canImport(Musl)\n@_exported import Musl\n#elseif os(Windows)\n@_exported import CRT\n@_exported import WinSDK\n#else\n@_exported import Darwin.C\n#endif\n\n/// `CStringArray` represents a C null-terminated array of pointers to C strings.\n///\n/// The lifetime of the C strings will correspond to the lifetime of the `CStringArray`\n/// instance so be careful about copying the buffer as it may contain dangling pointers.\npublic final class CStringArray {\n /// The null-terminated array of C string pointers.\n public let cArray: [UnsafeMutablePointer?]\n\n /// Creates an instance from an array of strings.\n public init(_ array: [String]) {\n#if os(Windows)\n cArray = array.map({ $0.withCString({ _strdup($0) }) }) + [nil]\n#else\n cArray = array.map({ $0.withCString({ strdup($0) }) }) + [nil]\n#endif\n }\n\n deinit {\n for case let element? in cArray {\n free(element)\n }\n }\n}\n\nimport Foundation\n#if os(Windows)\nimport WinSDK\n#endif\n\n#if os(Windows)\npublic let executableFileSuffix = \".exe\"\n#else\npublic let executableFileSuffix = \"\"\n#endif\n\n#if os(Windows)\nprivate func quote(_ arguments: [String]) -> String {\n func quote(argument: String) -> String {\n if !argument.contains(where: { \" \\t\\n\\\"\".contains($0) }) {\n return argument\n }\n\n // To escape the command line, we surround the argument with quotes.\n // However, the complication comes due to how the Windows command line\n // parser treats backslashes (\\) and quotes (\").\n //\n // - \\ is normally treated as a literal backslash\n // e.g. alpha\\beta\\gamma => alpha\\beta\\gamma\n // - The sequence \\\" is treated as a literal \"\n // e.g. alpha\\\"beta => alpha\"beta\n //\n // But then what if we are given a path that ends with a \\?\n //\n // Surrounding alpha\\beta\\ with \" would be \"alpha\\beta\\\" which would be\n // an unterminated string since it ends on a literal quote. To allow\n // this case the parser treats:\n //\n // - \\\\\" as \\ followed by the \" metacharacter\n // - \\\\\\\" as \\ followed by a literal \"\n //\n // In general:\n // - 2n \\ followed by \" => n \\ followed by the \" metacharacter\n // - 2n + 1 \\ followed by \" => n \\ followed by a literal \"\n\n var quoted = \"\\\"\"\n var unquoted = argument.unicodeScalars\n\n while !unquoted.isEmpty {\n guard let firstNonBS = unquoted.firstIndex(where: { $0 != \"\\\\\" }) else {\n // String ends with a backslash (e.g. first\\second\\), escape all\n // the backslashes then add the metacharacter \".\n let count = unquoted.count\n quoted.append(String(repeating: \"\\\\\", count: 2 * count))\n break\n }\n\n let count = unquoted.distance(from: unquoted.startIndex, to: firstNonBS)\n if unquoted[firstNonBS] == \"\\\"\" {\n // This is a string of \\ followed by a \" (e.g. first\\\"second).\n // Escape the backslashes and the quote.\n quoted.append(String(repeating: \"\\\\\", count: 2 * count + 1))\n } else {\n // These are just literal backslashes\n quoted.append(String(repeating: \"\\\\\", count: count))\n }\n\n quoted.append(String(unquoted[firstNonBS]))\n\n // Drop the backslashes and the following character\n unquoted.removeFirst(count + 1)\n }\n quoted.append(\"\\\"\")\n\n return quoted\n }\n return arguments.map(quote(argument:)).joined(separator: \" \")\n}\n#endif\n\n/// Replace the current process image with a new process image.\n///\n/// - Parameters:\n/// - path: Absolute path to the executable.\n/// - args: The executable arguments.\npublic func exec(path: String, args: [String]) throws -> Never {\n let cArgs = CStringArray(args)\n #if os(Windows)\n var hJob: HANDLE\n\n hJob = CreateJobObjectA(nil, nil)\n if hJob == HANDLE(bitPattern: 0) {\n throw SystemError.exec(Int32(GetLastError()), path: path, args: args)\n }\n defer { CloseHandle(hJob) }\n\n let hPort = CreateIoCompletionPort(INVALID_HANDLE_VALUE, nil, 0, 1)\n if hPort == HANDLE(bitPattern: 0) {\n throw SystemError.exec(Int32(GetLastError()), path: path, args: args)\n }\n\n var acpAssociation: JOBOBJECT_ASSOCIATE_COMPLETION_PORT = JOBOBJECT_ASSOCIATE_COMPLETION_PORT()\n acpAssociation.CompletionKey = hJob\n acpAssociation.CompletionPort = hPort\n if !SetInformationJobObject(hJob, JobObjectAssociateCompletionPortInformation,\n &acpAssociation, DWORD(MemoryLayout.size)) {\n throw SystemError.exec(Int32(GetLastError()), path: path, args: args)\n }\n\n var eliLimits: JOBOBJECT_EXTENDED_LIMIT_INFORMATION = JOBOBJECT_EXTENDED_LIMIT_INFORMATION()\n eliLimits.BasicLimitInformation.LimitFlags =\n DWORD(JOB_OBJECT_LIMIT_KILL_ON_JOB_CLOSE) | DWORD(JOB_OBJECT_LIMIT_SILENT_BREAKAWAY_OK)\n if !SetInformationJobObject(hJob, JobObjectExtendedLimitInformation, &eliLimits,\n DWORD(MemoryLayout.size)) {\n throw SystemError.exec(Int32(GetLastError()), path: path, args: args)\n }\n\n\n var siInfo: STARTUPINFOW = STARTUPINFOW()\n siInfo.cb = DWORD(MemoryLayout.size)\n\n var piInfo: PROCESS_INFORMATION = PROCESS_INFORMATION()\n\n try quote(args).withCString(encodedAs: UTF16.self) { pwszCommandLine in\n if !CreateProcessW(nil,\n UnsafeMutablePointer(mutating: pwszCommandLine),\n nil, nil, false,\n DWORD(CREATE_SUSPENDED) | DWORD(CREATE_NEW_PROCESS_GROUP),\n nil, nil, &siInfo, &piInfo) {\n throw SystemError.exec(Int32(GetLastError()), path: path, args: args)\n }\n }\n\n defer { CloseHandle(piInfo.hThread) }\n defer { CloseHandle(piInfo.hProcess) }\n\n if !AssignProcessToJobObject(hJob, piInfo.hProcess) {\n throw SystemError.exec(Int32(GetLastError()), path: path, args: args)\n }\n\n _ = ResumeThread(piInfo.hThread)\n\n var dwCompletionCode: DWORD = 0\n var ulCompletionKey: ULONG_PTR = 0\n var lpOverlapped: LPOVERLAPPED?\n repeat {\n } while GetQueuedCompletionStatus(hPort, &dwCompletionCode, &ulCompletionKey,\n &lpOverlapped, INFINITE) &&\n !(ulCompletionKey == ULONG_PTR(UInt(bitPattern: hJob)) &&\n dwCompletionCode == JOB_OBJECT_MSG_ACTIVE_PROCESS_ZERO)\n\n var dwExitCode: DWORD = DWORD(bitPattern: -1)\n _ = GetExitCodeProcess(piInfo.hProcess, &dwExitCode)\n _exit(Int32(bitPattern: dwExitCode))\n #elseif (!canImport(Darwin) || os(macOS))\n guard execv(path, cArgs.cArray) != -1 else {\n throw SystemError.exec(errno, path: path, args: args)\n }\n fatalError(\"unreachable\")\n #else\n fatalError(\"not implemented\")\n #endif\n}\n\n@_disfavoredOverload\n@available(*, deprecated, message: \"Use the overload which returns Never\")\npublic func exec(path: String, args: [String]) throws {\n try exec(path: path, args: args)\n}\n\n// MARK: TSCUtility function for searching for executables\n\n/// Create a list of AbsolutePath search paths from a string, such as the PATH environment variable.\n///\n/// - Parameters:\n/// - pathString: The path string to parse.\n/// - currentWorkingDirectory: The current working directory, the relative paths will be converted to absolute paths\n/// based on this path.\n/// - Returns: List of search paths.\npublic func getEnvSearchPaths(\n pathString: String?,\n currentWorkingDirectory: AbsolutePath?\n) -> [AbsolutePath] {\n // Compute search paths from PATH variable.\n#if os(Windows)\n let pathSeparator: Character = \";\"\n#else\n let pathSeparator: Character = \":\"\n#endif\n return (pathString ?? \"\").split(separator: pathSeparator).map(String.init).compactMap({ pathString in\n if let cwd = currentWorkingDirectory {\n return try? AbsolutePath(validating: pathString, relativeTo: cwd)\n }\n return try? AbsolutePath(validating: pathString)\n })\n}\n\n/// Lookup an executable path from an environment variable value, current working\n/// directory or search paths. Only return a value that is both found and executable.\n///\n/// This method searches in the following order:\n/// * If env value is a valid absolute path, return it.\n/// * If env value is relative path, first try to locate it in current working directory.\n/// * Otherwise, in provided search paths.\n///\n/// - Parameters:\n/// - filename: The name of the file to find.\n/// - currentWorkingDirectory: The current working directory to look in.\n/// - searchPaths: The additional search paths to look in if not found in cwd.\n/// - Returns: Valid path to executable if present, otherwise nil.\npublic func lookupExecutablePath(\n filename value: String?,\n currentWorkingDirectory: AbsolutePath? = localFileSystem.currentWorkingDirectory,\n searchPaths: [AbsolutePath] = []\n) -> AbsolutePath? {\n\n // We should have a value to continue.\n guard let value = value, !value.isEmpty else {\n return nil\n }\n\n var paths: [AbsolutePath] = []\n\n if let cwd = currentWorkingDirectory, let path = try? AbsolutePath(validating: value, relativeTo: cwd) {\n // We have a value, but it could be an absolute or a relative path.\n paths.append(path)\n } else if let absPath = try? AbsolutePath(validating: value) {\n // Current directory not being available is not a problem\n // for the absolute-specified paths.\n paths.append(absPath)\n }\n\n // Ensure the value is not a path.\n if !value.contains(\"/\") {\n // Try to locate in search paths.\n paths.append(contentsOf: searchPaths.map({ $0.appending(component: value) }))\n }\n\n return paths.first(where: { localFileSystem.isExecutableFile($0) })\n}\n\n/// A wrapper for Range to make it Codable.\n///\n/// Technically, we can use conditional conformance and make\n/// stdlib's Range Codable but since extensions leak out, it\n/// is not a good idea to extend types that you don't own.\n///\n/// Range conformance will be added soon to stdlib so we can remove\n/// this type in the future.\npublic struct CodableRange where Bound: Comparable & Codable {\n\n /// The underlying range.\n public let range: Range\n\n /// Create a CodableRange instance.\n public init(_ range: Range) {\n self.range = range\n }\n}\n\nextension CodableRange: Sendable where Bound: Sendable {}\n\nextension CodableRange: Codable {\n private enum CodingKeys: String, CodingKey {\n case lowerBound, upperBound\n }\n\n public func encode(to encoder: Encoder) throws {\n var container = encoder.container(keyedBy: CodingKeys.self)\n try container.encode(range.lowerBound, forKey: .lowerBound)\n try container.encode(range.upperBound, forKey: .upperBound)\n }\n\n public init(from decoder: Decoder) throws {\n let container = try decoder.container(keyedBy: CodingKeys.self)\n let lowerBound = try container.decode(Bound.self, forKey: .lowerBound)\n let upperBound = try container.decode(Bound.self, forKey: .upperBound)\n self.init(Range(uncheckedBounds: (lowerBound, upperBound)))\n }\n}\n\nextension AbsolutePath {\n /// File URL created from the normalized string representation of the path.\n public var asURL: Foundation.URL {\n return URL(fileURLWithPath: pathString)\n }\n \n public init(_ url: URL) throws {\n try self.init(validating: url.path)\n }\n}\n\n// FIXME: Eliminate or find a proper place for this.\npublic enum SystemError: Error {\n case chdir(Int32, String)\n case close(Int32)\n case exec(Int32, path: String, args: [String])\n case pipe(Int32)\n case posix_spawn(Int32, [String])\n case read(Int32)\n case setenv(Int32, String)\n case stat(Int32, String)\n case symlink(Int32, String, dest: String)\n case unsetenv(Int32, String)\n case waitpid(Int32)\n}\n\nextension SystemError: CustomStringConvertible {\n public var description: String {\n func strerror(_ errno: Int32) -> String {\n #if os(Windows)\n let cap = 128\n var buf = [Int8](repeating: 0, count: cap)\n let _ = strerror_s(&buf, 128, errno)\n return \"\\(String(cString: buf)) (\\(errno))\"\n #else\n var cap = 64\n while cap <= 16 * 1024 {\n var buf = [Int8](repeating: 0, count: cap)\n let err = strerror_r(errno, &buf, buf.count)\n if err == EINVAL {\n return \"Unknown error \\(errno)\"\n }\n if err == ERANGE {\n cap *= 2\n continue\n }\n if err != 0 {\n fatalError(\"strerror_r error: \\(err)\")\n }\n return \"\\(String(cString: buf)) (\\(errno))\"\n }\n fatalError(\"strerror_r error: \\(ERANGE)\")\n #endif\n }\n\n switch self {\n case .chdir(let errno, let path):\n return \"chdir error: \\(strerror(errno)): \\(path)\"\n case .close(let err):\n let errorMessage: String\n if err == -1 { // if the return code is -1, we need to consult the global `errno`\n errorMessage = strerror(errno)\n } else {\n errorMessage = strerror(err)\n }\n return \"close error: \\(errorMessage)\"\n case .exec(let errno, let path, let args):\n let joinedArgs = args.joined(separator: \" \")\n return \"exec error: \\(strerror(errno)): \\(path) \\(joinedArgs)\"\n case .pipe(let errno):\n return \"pipe error: \\(strerror(errno))\"\n case .posix_spawn(let errno, let args):\n return \"posix_spawn error: \\(strerror(errno)), `\\(args)`\"\n case .read(let errno):\n return \"read error: \\(strerror(errno))\"\n case .setenv(let errno, let key):\n return \"setenv error: \\(strerror(errno)): \\(key)\"\n case .stat(let errno, _):\n return \"stat error: \\(strerror(errno))\"\n case .symlink(let errno, let path, let dest):\n return \"symlink error: \\(strerror(errno)): \\(path) -> \\(dest)\"\n case .unsetenv(let errno, let key):\n return \"unsetenv error: \\(strerror(errno)): \\(key)\"\n case .waitpid(let errno):\n return \"waitpid error: \\(strerror(errno))\"\n }\n }\n}\n\nextension SystemError: CustomNSError {\n public var errorUserInfo: [String : Any] {\n return [NSLocalizedDescriptionKey: self.description]\n }\n}\n\n/// Memoizes a costly computation to a cache variable.\nfunc memoize(to cache: inout T?, build: () throws -> T) rethrows -> T {\n if let value = cache {\n return value\n } else {\n let value = try build()\n cache = value\n return value\n }\n}\n" }, { "path": "Tool/Sources/FileSystem/Path.swift", "content": "/*\n This source file is part of the Swift.org open source project\n\n Copyright (c) 2014 - 2018 Apple Inc. and the Swift project authors\n Licensed under Apache License v2.0 with Runtime Library Exception\n\n See http://swift.org/LICENSE.txt for license information\n See http://swift.org/CONTRIBUTORS.txt for Swift project authors\n*/\n#if os(Windows)\nimport Foundation\nimport WinSDK\n#endif\n\n#if os(Windows)\nprivate typealias PathImpl = WindowsPath\n#else\nprivate typealias PathImpl = UNIXPath\n#endif\n\nimport protocol Foundation.CustomNSError\nimport var Foundation.NSLocalizedDescriptionKey\n\n/// Represents an absolute file system path, independently of what (or whether\n/// anything at all) exists at that path in the file system at any given time.\n/// An absolute path always starts with a `/` character, and holds a normalized\n/// string representation. This normalization is strictly syntactic, and does\n/// not access the file system in any way.\n///\n/// The absolute path string is normalized by:\n/// - Collapsing `..` path components\n/// - Removing `.` path components\n/// - Removing any trailing path separator\n/// - Removing any redundant path separators\n///\n/// This string manipulation may change the meaning of a path if any of the\n/// path components are symbolic links on disk. However, the file system is\n/// never accessed in any way when initializing an AbsolutePath.\n///\n/// Note that `~` (home directory resolution) is *not* done as part of path\n/// normalization, because it is normally the responsibility of the shell and\n/// not the program being invoked (e.g. when invoking `cd ~`, it is the shell\n/// that evaluates the tilde; the `cd` command receives an absolute path).\npublic struct AbsolutePath: Hashable, Sendable {\n /// Check if the given name is a valid individual path component.\n ///\n /// This only checks with regard to the semantics enforced by `AbsolutePath`\n /// and `RelativePath`; particular file systems may have their own\n /// additional requirements.\n static func isValidComponent(_ name: String) -> Bool {\n return PathImpl.isValidComponent(name)\n }\n\n /// Private implementation details, shared with the RelativePath struct.\n private let _impl: PathImpl\n\n /// Private initializer when the backing storage is known.\n private init(_ impl: PathImpl) {\n _impl = impl\n }\n\n /// Initializes an AbsolutePath from a string that may be either absolute\n /// or relative; if relative, `basePath` is used as the anchor; if absolute,\n /// it is used as is, and in this case `basePath` is ignored.\n public init(validating str: String, relativeTo basePath: AbsolutePath) throws {\n if PathImpl(string: str).isAbsolute {\n try self.init(validating: str)\n } else {\n#if os(Windows)\n assert(!basePath.pathString.isEmpty)\n guard !str.isEmpty else {\n self.init(basePath._impl)\n return\n }\n\n let base: UnsafePointer =\n basePath.pathString.fileSystemRepresentation\n defer { base.deallocate() }\n\n let path: UnsafePointer = str.fileSystemRepresentation\n defer { path.deallocate() }\n\n var pwszResult: PWSTR!\n _ = String(cString: base).withCString(encodedAs: UTF16.self) { pwszBase in\n String(cString: path).withCString(encodedAs: UTF16.self) { pwszPath in\n PathAllocCombine(pwszBase, pwszPath, ULONG(PATHCCH_ALLOW_LONG_PATHS.rawValue), &pwszResult)\n }\n }\n defer { LocalFree(pwszResult) }\n\n self.init(String(decodingCString: pwszResult, as: UTF16.self))\n#else\n try self.init(basePath, RelativePath(validating: str))\n#endif\n }\n }\n\n /// Initializes the AbsolutePath by concatenating a relative path to an\n /// existing absolute path, and renormalizing if necessary.\n public init(_ absPath: AbsolutePath, _ relPath: RelativePath) {\n self.init(absPath._impl.appending(relativePath: relPath._impl))\n }\n\n /// Convenience initializer that appends a string to a relative path.\n public init(_ absPath: AbsolutePath, validating relStr: String) throws {\n try self.init(absPath, RelativePath(validating: relStr))\n }\n\n /// Initializes the AbsolutePath from `absStr`, which must be an absolute\n /// path (i.e. it must begin with a path separator; this initializer does\n /// not interpret leading `~` characters as home directory specifiers).\n /// The input string will be normalized if needed, as described in the\n /// documentation for AbsolutePath.\n public init(validating path: String) throws {\n try self.init(PathImpl(validatingAbsolutePath: path))\n }\n\n /// Directory component. An absolute path always has a non-empty directory\n /// component (the directory component of the root path is the root itself).\n public var dirname: String {\n return _impl.dirname\n }\n\n /// Last path component (including the suffix, if any). it is never empty.\n public var basename: String {\n return _impl.basename\n }\n\n /// Returns the basename without the extension.\n public var basenameWithoutExt: String {\n if let ext = self.extension {\n return String(basename.dropLast(ext.count + 1))\n }\n return basename\n }\n\n /// Suffix (including leading `.` character) if any. Note that a basename\n /// that starts with a `.` character is not considered a suffix, nor is a\n /// trailing `.` character.\n public var suffix: String? {\n return _impl.suffix\n }\n\n /// Extension of the give path's basename. This follow same rules as\n /// suffix except that it doesn't include leading `.` character.\n public var `extension`: String? {\n return _impl.extension\n }\n\n /// Absolute path of parent directory. This always returns a path, because\n /// every directory has a parent (the parent directory of the root directory\n /// is considered to be the root directory itself).\n public var parentDirectory: AbsolutePath {\n return AbsolutePath(_impl.parentDirectory)\n }\n\n /// True if the path is the root directory.\n public var isRoot: Bool {\n return _impl.isRoot\n }\n\n /// Returns the absolute path with the relative path applied.\n public func appending(_ subpath: RelativePath) -> AbsolutePath {\n return AbsolutePath(self, subpath)\n }\n\n /// Returns the absolute path with an additional literal component appended.\n ///\n /// This method accepts pseudo-path like '.' or '..', but should not contain \"/\".\n public func appending(component: String) -> AbsolutePath {\n return AbsolutePath(_impl.appending(component: component))\n }\n\n /// Returns the absolute path with additional literal components appended.\n ///\n /// This method should only be used in cases where the input is guaranteed\n /// to be a valid path component (i.e., it cannot be empty, contain a path\n /// separator, or be a pseudo-path like '.' or '..').\n public func appending(components names: [String]) -> AbsolutePath {\n // FIXME: This doesn't seem a particularly efficient way to do this.\n return names.reduce(self, { path, name in\n path.appending(component: name)\n })\n }\n\n public func appending(components names: String...) -> AbsolutePath {\n appending(components: names)\n }\n\n /// NOTE: We will most likely want to add other `appending()` methods, such\n /// as `appending(suffix:)`, and also perhaps `replacing()` methods,\n /// such as `replacing(suffix:)` or `replacing(basename:)` for some\n /// of the more common path operations.\n\n /// NOTE: We may want to consider adding operators such as `+` for appending\n /// a path component.\n\n /// NOTE: We will want to add a method to return the lowest common ancestor\n /// path.\n\n /// Root directory (whose string representation is just a path separator).\n public static let root = AbsolutePath(PathImpl.root)\n\n /// Normalized string representation (the normalization rules are described\n /// in the documentation of the initializer). This string is never empty.\n public var pathString: String {\n return _impl.string\n }\n\n /// Returns an array of strings that make up the path components of the\n /// absolute path. This is the same sequence of strings as the basenames\n /// of each successive path component, starting from the root. Therefore\n /// the first path component of an absolute path is always `/`.\n public var components: [String] {\n return _impl.components\n }\n}\n\n/// Represents a relative file system path. A relative path never starts with\n/// a `/` character, and holds a normalized string representation. As with\n/// AbsolutePath, the normalization is strictly syntactic, and does not access\n/// the file system in any way.\n///\n/// The relative path string is normalized by:\n/// - Collapsing `..` path components that aren't at the beginning\n/// - Removing extraneous `.` path components\n/// - Removing any trailing path separator\n/// - Removing any redundant path separators\n/// - Replacing a completely empty path with a `.`\n///\n/// This string manipulation may change the meaning of a path if any of the\n/// path components are symbolic links on disk. However, the file system is\n/// never accessed in any way when initializing a RelativePath.\npublic struct RelativePath: Hashable, Sendable {\n /// Private implementation details, shared with the AbsolutePath struct.\n fileprivate let _impl: PathImpl\n\n /// Private initializer when the backing storage is known.\n private init(_ impl: PathImpl) {\n _impl = impl\n }\n\n /// Convenience initializer that verifies that the path is relative.\n public init(validating path: String) throws {\n try self.init(PathImpl(validatingRelativePath: path))\n }\n\n /// Directory component. For a relative path without any path separators,\n /// this is the `.` string instead of the empty string.\n public var dirname: String {\n return _impl.dirname\n }\n\n /// Last path component (including the suffix, if any). It is never empty.\n public var basename: String {\n return _impl.basename\n }\n\n /// Returns the basename without the extension.\n public var basenameWithoutExt: String {\n if let ext = self.extension {\n return String(basename.dropLast(ext.count + 1))\n }\n return basename\n }\n\n /// Suffix (including leading `.` character) if any. Note that a basename\n /// that starts with a `.` character is not considered a suffix, nor is a\n /// trailing `.` character.\n public var suffix: String? {\n return _impl.suffix\n }\n\n /// Extension of the give path's basename. This follow same rules as\n /// suffix except that it doesn't include leading `.` character.\n public var `extension`: String? {\n return _impl.extension\n }\n\n /// Normalized string representation (the normalization rules are described\n /// in the documentation of the initializer). This string is never empty.\n public var pathString: String {\n return _impl.string\n }\n\n /// Returns an array of strings that make up the path components of the\n /// relative path. This is the same sequence of strings as the basenames\n /// of each successive path component. Therefore the returned array of\n /// path components is never empty; even an empty path has a single path\n /// component: the `.` string.\n public var components: [String] {\n return _impl.components\n }\n\n /// Returns the relative path with the given relative path applied.\n public func appending(_ subpath: RelativePath) -> RelativePath {\n return RelativePath(_impl.appending(relativePath: subpath._impl))\n }\n\n /// Returns the relative path with an additional literal component appended.\n ///\n /// This method accepts pseudo-path like '.' or '..', but should not contain \"/\".\n public func appending(component: String) -> RelativePath {\n return RelativePath(_impl.appending(component: component))\n }\n\n /// Returns the relative path with additional literal components appended.\n ///\n /// This method should only be used in cases where the input is guaranteed\n /// to be a valid path component (i.e., it cannot be empty, contain a path\n /// separator, or be a pseudo-path like '.' or '..').\n public func appending(components names: [String]) -> RelativePath {\n // FIXME: This doesn't seem a particularly efficient way to do this.\n return names.reduce(self, { path, name in\n path.appending(component: name)\n })\n }\n\n public func appending(components names: String...) -> RelativePath {\n appending(components: names)\n }\n}\n\nextension AbsolutePath: Codable {\n public func encode(to encoder: Encoder) throws {\n var container = encoder.singleValueContainer()\n try container.encode(pathString)\n }\n\n public init(from decoder: Decoder) throws {\n let container = try decoder.singleValueContainer()\n try self.init(validating: container.decode(String.self))\n }\n}\n\nextension RelativePath: Codable {\n public func encode(to encoder: Encoder) throws {\n var container = encoder.singleValueContainer()\n try container.encode(pathString)\n }\n\n public init(from decoder: Decoder) throws {\n let container = try decoder.singleValueContainer()\n try self.init(validating: container.decode(String.self))\n }\n}\n\n// Make absolute paths Comparable.\nextension AbsolutePath: Comparable {\n public static func < (lhs: AbsolutePath, rhs: AbsolutePath) -> Bool {\n return lhs.pathString < rhs.pathString\n }\n}\n\n/// Make absolute paths CustomStringConvertible and CustomDebugStringConvertible.\nextension AbsolutePath: CustomStringConvertible, CustomDebugStringConvertible {\n public var description: String {\n return pathString\n }\n\n public var debugDescription: String {\n // FIXME: We should really be escaping backslashes and quotes here.\n return \"\"\n }\n}\n\n/// Make relative paths CustomStringConvertible and CustomDebugStringConvertible.\nextension RelativePath: CustomStringConvertible {\n public var description: String {\n return _impl.string\n }\n\n public var debugDescription: String {\n // FIXME: We should really be escaping backslashes and quotes here.\n return \"\"\n }\n}\n\n/// Private implementation shared between AbsolutePath and RelativePath.\nprotocol Path: Hashable {\n\n /// Root directory.\n static var root: Self { get }\n\n /// Checks if a string is a valid component.\n static func isValidComponent(_ name: String) -> Bool\n\n /// Normalized string of the (absolute or relative) path. Never empty.\n var string: String { get }\n\n /// Returns whether the path is the root path.\n var isRoot: Bool { get }\n\n /// Returns whether the path is an absolute path.\n var isAbsolute: Bool { get }\n\n /// Returns the directory part of the stored path (relying on the fact that it has been normalized). Returns a\n /// string consisting of just `.` if there is no directory part (which is the case if and only if there is no path\n /// separator).\n var dirname: String { get }\n\n /// Returns the last past component.\n var basename: String { get }\n\n /// Returns the components of the path between each path separator.\n var components: [String] { get }\n\n /// Path of parent directory. This always returns a path, because every directory has a parent (the parent\n /// directory of the root directory is considered to be the root directory itself).\n var parentDirectory: Self { get }\n\n /// Creates a path from its normalized string representation.\n init(string: String)\n\n /// Creates a path from a string representation, validates that it is a valid absolute path and normalizes it.\n init(validatingAbsolutePath: String) throws\n\n /// Creates a path from a string representation, validates that it is a valid relative path and normalizes it.\n init(validatingRelativePath: String) throws\n\n /// Returns suffix with leading `.` if withDot is true otherwise without it.\n func suffix(withDot: Bool) -> String?\n\n /// Returns a new Path by appending the path component.\n func appending(component: String) -> Self\n\n /// Returns a path by concatenating a relative path and renormalizing if necessary.\n func appending(relativePath: Self) -> Self\n}\n\nextension Path {\n var suffix: String? {\n return suffix(withDot: true)\n }\n\n var `extension`: String? {\n return suffix(withDot: false)\n }\n}\n\n#if os(Windows)\nprivate struct WindowsPath: Path, Sendable {\n let string: String\n\n // NOTE: this is *NOT* a root path. It is a drive-relative path that needs\n // to be specified due to assumptions in the APIs. Use the platform\n // specific path separator as we should be normalizing the path normally.\n // This is required to make the `InMemoryFileSystem` correctly iterate\n // paths.\n static let root = Self(string: \"\\\\\")\n\n static func isValidComponent(_ name: String) -> Bool {\n return name != \"\" && name != \".\" && name != \"..\" && !name.contains(\"/\")\n }\n\n static func isAbsolutePath(_ path: String) -> Bool {\n return !path.withCString(encodedAs: UTF16.self, PathIsRelativeW)\n }\n\n var dirname: String {\n let fsr: UnsafePointer = self.string.fileSystemRepresentation\n defer { fsr.deallocate() }\n\n let path: String = String(cString: fsr)\n return path.withCString(encodedAs: UTF16.self) {\n let data = UnsafeMutablePointer(mutating: $0)\n PathCchRemoveFileSpec(data, path.count)\n return String(decodingCString: data, as: UTF16.self)\n }\n }\n\n var isAbsolute: Bool {\n return Self.isAbsolutePath(self.string)\n }\n\n public var isRoot: Bool {\n return self.string.withCString(encodedAs: UTF16.self, PathCchIsRoot)\n }\n\n var basename: String {\n let path: String = self.string\n return path.withCString(encodedAs: UTF16.self) {\n PathStripPathW(UnsafeMutablePointer(mutating: $0))\n return String(decodingCString: $0, as: UTF16.self)\n }\n }\n\n // FIXME: We should investigate if it would be more efficient to instead\n // return a path component iterator that does all its work lazily, moving\n // from one path separator to the next on-demand.\n //\n var components: [String] {\n let normalized: UnsafePointer = string.fileSystemRepresentation\n defer { normalized.deallocate() }\n\n return String(cString: normalized).components(separatedBy: \"\\\\\").filter { !$0.isEmpty }\n }\n\n var parentDirectory: Self {\n return self == .root ? self : Self(string: dirname)\n }\n\n init(string: String) {\n if string.first?.isASCII ?? false, string.first?.isLetter ?? false, string.first?.isLowercase ?? false,\n string.count > 1, string[string.index(string.startIndex, offsetBy: 1)] == \":\"\n {\n self.string = \"\\(string.first!.uppercased())\\(string.dropFirst(1))\"\n } else {\n self.string = string\n }\n }\n\n private static func repr(_ path: String) -> String {\n guard !path.isEmpty else { return \"\" }\n let representation: UnsafePointer = path.fileSystemRepresentation\n defer { representation.deallocate() }\n return String(cString: representation)\n }\n\n init(validatingAbsolutePath path: String) throws {\n let realpath = Self.repr(path)\n if !Self.isAbsolutePath(realpath) {\n throw PathValidationError.invalidAbsolutePath(path)\n }\n self.init(string: realpath)\n }\n\n init(validatingRelativePath path: String) throws {\n if path.isEmpty || path == \".\" {\n self.init(string: \".\")\n } else {\n let realpath: String = Self.repr(path)\n // Treat a relative path as an invalid relative path...\n if Self.isAbsolutePath(realpath) || realpath.first == \"\\\\\" {\n throw PathValidationError.invalidRelativePath(path)\n }\n self.init(string: realpath)\n }\n }\n\n func suffix(withDot: Bool) -> String? {\n return self.string.withCString(encodedAs: UTF16.self) {\n if let pointer = PathFindExtensionW($0) {\n let substring = String(decodingCString: pointer, as: UTF16.self)\n guard substring.length > 0 else { return nil }\n return withDot ? substring : String(substring.dropFirst(1))\n }\n return nil\n }\n }\n\n func appending(component name: String) -> Self {\n var result: PWSTR?\n _ = string.withCString(encodedAs: UTF16.self) { root in\n name.withCString(encodedAs: UTF16.self) { path in\n PathAllocCombine(root, path, ULONG(PATHCCH_ALLOW_LONG_PATHS.rawValue), &result)\n }\n }\n defer { LocalFree(result) }\n return Self(string: String(decodingCString: result!, as: UTF16.self))\n }\n\n func appending(relativePath: Self) -> Self {\n var result: PWSTR?\n _ = string.withCString(encodedAs: UTF16.self) { root in\n relativePath.string.withCString(encodedAs: UTF16.self) { path in\n PathAllocCombine(root, path, ULONG(PATHCCH_ALLOW_LONG_PATHS.rawValue), &result)\n }\n }\n defer { LocalFree(result) }\n return Self(string: String(decodingCString: result!, as: UTF16.self))\n }\n}\n#else\nprivate struct UNIXPath: Path, Sendable {\n let string: String\n\n static let root = Self(string: \"/\")\n\n static func isValidComponent(_ name: String) -> Bool {\n return name != \"\" && name != \".\" && name != \"..\" && !name.contains(\"/\")\n }\n\n var dirname: String {\n // FIXME: This method seems too complicated; it should be simplified,\n // if possible, and certainly optimized (using UTF8View).\n // Find the last path separator.\n guard let idx = string.lastIndex(of: \"/\") else {\n // No path separators, so the directory name is `.`.\n return \".\"\n }\n // Check if it's the only one in the string.\n if idx == string.startIndex {\n // Just one path separator, so the directory name is `/`.\n return \"/\"\n }\n // Otherwise, it's the string up to (but not including) the last path\n // separator.\n return String(string.prefix(upTo: idx))\n }\n\n var isAbsolute: Bool {\n return string.hasPrefix(\"/\")\n }\n\n var isRoot: Bool {\n return self == Self.root\n }\n\n var basename: String {\n // Find the last path separator.\n guard let idx = string.lastIndex(of: \"/\") else {\n // No path separators, so the basename is the whole string.\n return string\n }\n // Otherwise, it's the string from (but not including) the last path\n // separator.\n return String(string.suffix(from: string.index(after: idx)))\n }\n\n // FIXME: We should investigate if it would be more efficient to instead\n // return a path component iterator that does all its work lazily, moving\n // from one path separator to the next on-demand.\n //\n var components: [String] {\n // FIXME: This isn't particularly efficient; needs optimization, and\n // in fact, it might well be best to return a custom iterator so we\n // don't have to allocate everything up-front. It would be backed by\n // the path string and just return a slice at a time.\n let components = string.components(separatedBy: \"/\").filter({ !$0.isEmpty })\n\n if string.hasPrefix(\"/\") {\n return [\"/\"] + components\n } else {\n return components\n }\n }\n\n var parentDirectory: Self {\n return self == .root ? self : Self(string: dirname)\n }\n\n init(string: String) {\n self.string = string\n }\n\n init(normalizingAbsolutePath path: String) {\n precondition(path.first == \"/\", \"Failure normalizing \\(path), absolute paths should start with '/'\")\n\n // At this point we expect to have a path separator as first character.\n assert(path.first == \"/\")\n // Fast path.\n if !mayNeedNormalization(absolute: path) {\n self.init(string: path)\n }\n\n // Split the character array into parts, folding components as we go.\n // As we do so, we count the number of characters we'll end up with in\n // the normalized string representation.\n var parts: [String] = []\n var capacity = 0\n for part in path.split(separator: \"/\") {\n switch part.count {\n case 0:\n // Ignore empty path components.\n continue\n case 1 where part.first == \".\":\n // Ignore `.` path components.\n continue\n case 2 where part.first == \".\" && part.last == \".\":\n // If there's a previous part, drop it; otherwise, do nothing.\n if let prev = parts.last {\n parts.removeLast()\n capacity -= prev.count\n }\n default:\n // Any other component gets appended.\n parts.append(String(part))\n capacity += part.count\n }\n }\n capacity += max(parts.count, 1)\n\n // Create an output buffer using the capacity we've calculated.\n // FIXME: Determine the most efficient way to reassemble a string.\n var result = \"\"\n result.reserveCapacity(capacity)\n\n // Put the normalized parts back together again.\n var iter = parts.makeIterator()\n result.append(\"/\")\n if let first = iter.next() {\n result.append(contentsOf: first)\n while let next = iter.next() {\n result.append(\"/\")\n result.append(contentsOf: next)\n }\n }\n\n // Sanity-check the result (including the capacity we reserved).\n assert(!result.isEmpty, \"unexpected empty string\")\n assert(result.count == capacity, \"count: \" +\n \"\\(result.count), cap: \\(capacity)\")\n\n // Use the result as our stored string.\n self.init(string: result)\n }\n\n init(normalizingRelativePath path: String) {\n precondition(path.first != \"/\")\n\n // FIXME: Here we should also keep track of whether anything actually has\n // to be changed in the string, and if not, just return the existing one.\n\n // Split the character array into parts, folding components as we go.\n // As we do so, we count the number of characters we'll end up with in\n // the normalized string representation.\n var parts: [String] = []\n var capacity = 0\n for part in path.split(separator: \"/\") {\n switch part.count {\n case 0:\n // Ignore empty path components.\n continue\n case 1 where part.first == \".\":\n // Ignore `.` path components.\n continue\n case 2 where part.first == \".\" && part.last == \".\":\n // If at beginning, fall through to treat the `..` literally.\n guard let prev = parts.last else {\n fallthrough\n }\n // If previous component is anything other than `..`, drop it.\n if !(prev.count == 2 && prev.first == \".\" && prev.last == \".\") {\n parts.removeLast()\n capacity -= prev.count\n continue\n }\n // Otherwise, fall through to treat the `..` literally.\n fallthrough\n default:\n // Any other component gets appended.\n parts.append(String(part))\n capacity += part.count\n }\n }\n capacity += max(parts.count - 1, 0)\n\n // Create an output buffer using the capacity we've calculated.\n // FIXME: Determine the most efficient way to reassemble a string.\n var result = \"\"\n result.reserveCapacity(capacity)\n\n // Put the normalized parts back together again.\n var iter = parts.makeIterator()\n if let first = iter.next() {\n result.append(contentsOf: first)\n while let next = iter.next() {\n result.append(\"/\")\n result.append(contentsOf: next)\n }\n }\n\n // Sanity-check the result (including the capacity we reserved).\n assert(result.count == capacity, \"count: \" +\n \"\\(result.count), cap: \\(capacity)\")\n\n // If the result is empty, return `.`, otherwise we return it as a string.\n self.init(string: result.isEmpty ? \".\" : result)\n }\n\n init(validatingAbsolutePath path: String) throws {\n switch path.first {\n case \"/\":\n self.init(normalizingAbsolutePath: path)\n case \"~\":\n throw PathValidationError.startsWithTilde(path)\n default:\n throw PathValidationError.invalidAbsolutePath(path)\n }\n }\n\n init(validatingRelativePath path: String) throws {\n switch path.first {\n case \"/\":\n throw PathValidationError.invalidRelativePath(path)\n default:\n self.init(normalizingRelativePath: path)\n }\n }\n\n func suffix(withDot: Bool) -> String? {\n // FIXME: This method seems too complicated; it should be simplified,\n // if possible, and certainly optimized (using UTF8View).\n // Find the last path separator, if any.\n let sIdx = string.lastIndex(of: \"/\")\n // Find the start of the basename.\n let bIdx = (sIdx != nil) ? string.index(after: sIdx!) : string.startIndex\n // Find the last `.` (if any), starting from the second character of\n // the basename (a leading `.` does not make the whole path component\n // a suffix).\n let fIdx = string.index(bIdx, offsetBy: 1, limitedBy: string.endIndex) ?? string.startIndex\n if let idx = string[fIdx...].lastIndex(of: \".\") {\n // Unless it's just a `.` at the end, we have found a suffix.\n if string.distance(from: idx, to: string.endIndex) > 1 {\n let fromIndex = withDot ? idx : string.index(idx, offsetBy: 1)\n return String(string.suffix(from: fromIndex))\n } else {\n return nil\n }\n }\n // If we get this far, there is no suffix.\n return nil\n }\n\n func appending(component name: String) -> Self {\n assert(!name.contains(\"/\"), \"\\(name) is invalid path component\")\n\n // Handle pseudo paths.\n switch name {\n case \"\", \".\":\n return self\n case \"..\":\n return self.parentDirectory\n default:\n break\n }\n\n if self == Self.root {\n return Self(string: \"/\" + name)\n } else {\n return Self(string: string + \"/\" + name)\n }\n }\n\n func appending(relativePath: Self) -> Self {\n // Both paths are already normalized. The only case in which we have\n // to renormalize their concatenation is if the relative path starts\n // with a `..` path component.\n var newPathString = string\n if self != .root {\n newPathString.append(\"/\")\n }\n\n let relativePathString = relativePath.string\n newPathString.append(relativePathString)\n\n // If the relative string starts with `.` or `..`, we need to normalize\n // the resulting string.\n // FIXME: We can actually optimize that case, since we know that the\n // normalization of a relative path can leave `..` path components at\n // the beginning of the path only.\n if relativePathString.hasPrefix(\".\") {\n if newPathString.hasPrefix(\"/\") {\n return Self(normalizingAbsolutePath: newPathString)\n } else {\n return Self(normalizingRelativePath: newPathString)\n }\n } else {\n return Self(string: newPathString)\n }\n }\n}\n#endif\n\n/// Describes the way in which a path is invalid.\npublic enum PathValidationError: Error {\n case startsWithTilde(String)\n case invalidAbsolutePath(String)\n case invalidRelativePath(String)\n}\n\nextension PathValidationError: CustomStringConvertible {\n public var description: String {\n switch self {\n case .startsWithTilde(let path):\n return \"invalid absolute path '\\(path)'; absolute path must begin with '/'\"\n case .invalidAbsolutePath(let path):\n return \"invalid absolute path '\\(path)'\"\n case .invalidRelativePath(let path):\n return \"invalid relative path '\\(path)'; relative path should not begin with '\\(AbsolutePath.root.pathString)'\"\n }\n }\n}\n\nextension AbsolutePath {\n /// Returns a relative path that, when concatenated to `base`, yields the\n /// callee path itself. If `base` is not an ancestor of the callee, the\n /// returned path will begin with one or more `..` path components.\n ///\n /// Because both paths are absolute, they always have a common ancestor\n /// (the root path, if nothing else). Therefore, any path can be made\n /// relative to any other path by using a sufficient number of `..` path\n /// components.\n ///\n /// This method is strictly syntactic and does not access the file system\n /// in any way. Therefore, it does not take symbolic links into account.\n public func relative(to base: AbsolutePath) -> RelativePath {\n let result: RelativePath\n // Split the two paths into their components.\n // FIXME: The is needs to be optimized to avoid unnecessary copying.\n let pathComps = self.components\n let baseComps = base.components\n\n // It's common for the base to be an ancestor, so try that first.\n if pathComps.starts(with: baseComps) {\n // Special case, which is a plain path without `..` components. It\n // might be an empty path (when self and the base are equal).\n let relComps = pathComps.dropFirst(baseComps.count)\n#if os(Windows)\n let pathString = relComps.joined(separator: \"\\\\\")\n#else\n let pathString = relComps.joined(separator: \"/\")\n#endif\n do {\n result = try RelativePath(validating: pathString)\n } catch {\n preconditionFailure(\"invalid relative path computed from \\(pathString)\")\n }\n\n } else {\n // General case, in which we might well need `..` components to go\n // \"up\" before we can go \"down\" the directory tree.\n var newPathComps = ArraySlice(pathComps)\n var newBaseComps = ArraySlice(baseComps)\n while newPathComps.prefix(1) == newBaseComps.prefix(1) {\n // First component matches, so drop it.\n newPathComps = newPathComps.dropFirst()\n newBaseComps = newBaseComps.dropFirst()\n }\n // Now construct a path consisting of as many `..`s as are in the\n // `newBaseComps` followed by what remains in `newPathComps`.\n var relComps = Array(repeating: \"..\", count: newBaseComps.count)\n relComps.append(contentsOf: newPathComps)\n#if os(Windows)\n let pathString = relComps.joined(separator: \"\\\\\")\n#else\n let pathString = relComps.joined(separator: \"/\")\n#endif\n do {\n result = try RelativePath(validating: pathString)\n } catch {\n preconditionFailure(\"invalid relative path computed from \\(pathString)\")\n }\n }\n\n assert(AbsolutePath(base, result) == self)\n return result\n }\n\n /// Returns true if the path contains the given path.\n ///\n /// This method is strictly syntactic and does not access the file system\n /// in any way.\n @available(*, deprecated, renamed: \"isDescendantOfOrEqual(to:)\")\n public func contains(_ other: AbsolutePath) -> Bool {\n return isDescendantOfOrEqual(to: other)\n }\n\n /// Returns true if the path is an ancestor of the given path.\n ///\n /// This method is strictly syntactic and does not access the file system\n /// in any way.\n public func isAncestor(of descendant: AbsolutePath) -> Bool {\n return descendant.components.dropLast().starts(with: self.components)\n }\n\n /// Returns true if the path is an ancestor of or equal to the given path.\n ///\n /// This method is strictly syntactic and does not access the file system\n /// in any way.\n public func isAncestorOfOrEqual(to descendant: AbsolutePath) -> Bool {\n return descendant.components.starts(with: self.components)\n }\n\n /// Returns true if the path is a descendant of the given path.\n ///\n /// This method is strictly syntactic and does not access the file system\n /// in any way.\n public func isDescendant(of ancestor: AbsolutePath) -> Bool {\n return self.components.dropLast().starts(with: ancestor.components)\n }\n\n /// Returns true if the path is a descendant of or equal to the given path.\n ///\n /// This method is strictly syntactic and does not access the file system\n /// in any way.\n public func isDescendantOfOrEqual(to ancestor: AbsolutePath) -> Bool {\n return self.components.starts(with: ancestor.components)\n }\n}\n\nextension PathValidationError: CustomNSError {\n public var errorUserInfo: [String : Any] {\n return [NSLocalizedDescriptionKey: self.description]\n }\n}\n\n// FIXME: We should consider whether to merge the two `normalize()` functions.\n// The argument for doing so is that some of the code is repeated; the argument\n// against doing so is that some of the details are different, and since any\n// given path is either absolute or relative, it's wasteful to keep checking\n// for whether it's relative or absolute. Possibly we can do both by clever\n// use of generics that abstract away the differences.\n\n/// Fast check for if a string might need normalization.\n///\n/// This assumes that paths containing dotfiles are rare:\nprivate func mayNeedNormalization(absolute string: String) -> Bool {\n var last = UInt8(ascii: \"0\")\n for c in string.utf8 {\n switch c {\n case UInt8(ascii: \"/\") where last == UInt8(ascii: \"/\"):\n return true\n case UInt8(ascii: \".\") where last == UInt8(ascii: \"/\"):\n return true\n default:\n break\n }\n last = c\n }\n if last == UInt8(ascii: \"/\") {\n return true\n }\n return false\n}\n\n// MARK: - `AbsolutePath` backwards compatibility, delete after deprecation period.\n\nextension AbsolutePath {\n @_disfavoredOverload\n @available(*, deprecated, message: \"use throwing `init(validating:)` variant instead\")\n public init(_ absStr: String) {\n try! self.init(validating: absStr)\n }\n\n @_disfavoredOverload\n @available(*, deprecated, message: \"use throwing `init(validating:relativeTo:)` variant instead\")\n public init(_ str: String, relativeTo basePath: AbsolutePath) {\n try! self.init(validating: str, relativeTo: basePath)\n }\n\n @_disfavoredOverload\n @available(*, deprecated, message: \"use throwing variant instead\")\n public init(_ absPath: AbsolutePath, _ relStr: String) {\n try! self.init(absPath, validating: relStr)\n }\n}\n\n// MARK: - `AbsolutePath` backwards compatibility, delete after deprecation period.\n\nextension RelativePath {\n @_disfavoredOverload\n @available(*, deprecated, message: \"use throwing variant instead\")\n public init(_ string: String) {\n try! self.init(validating: string)\n }\n}\n" }, { "path": "Tool/Sources/FileSystem/PathShim.swift", "content": "/*\n This source file is part of the Swift.org open source project\n\n Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors\n Licensed under Apache License v2.0 with Runtime Library Exception\n\n See http://swift.org/LICENSE.txt for license information\n See http://swift.org/CONTRIBUTORS.txt for Swift project authors\n -------------------------------------------------------------------------\n\n This file contains temporary shim functions for use during the adoption of\n AbsolutePath and RelativePath. The eventual plan is to use the FileSystem\n API for all of this, at which time this file will go way. But since it is\n important to have a quality FileSystem API, we will evolve it slowly.\n\n Meanwhile this file bridges the gap to let call sites be as clean as possible,\n while making it fairly easy to find those calls later.\n */\n\nimport Foundation\n\n#if canImport(Glibc)\n@_exported import Glibc\n#elseif canImport(Musl)\n@_exported import Musl\n#elseif os(Windows)\n@_exported import CRT\n@_exported import WinSDK\n#else\n@_exported import Darwin.C\n#endif\n\n/// Returns the \"real path\" corresponding to `path` by resolving any symbolic links.\npublic func resolveSymlinks(_ path: AbsolutePath) throws -> AbsolutePath {\n #if os(Windows)\n let handle: HANDLE = path.pathString.withCString(encodedAs: UTF16.self) {\n CreateFileW(\n $0,\n GENERIC_READ,\n DWORD(FILE_SHARE_READ),\n nil,\n DWORD(OPEN_EXISTING),\n DWORD(FILE_FLAG_BACKUP_SEMANTICS),\n nil\n )\n }\n if handle == INVALID_HANDLE_VALUE { return path }\n defer { CloseHandle(handle) }\n return try withUnsafeTemporaryAllocation(of: WCHAR.self, capacity: 261) {\n let dwLength: DWORD =\n GetFinalPathNameByHandleW(\n handle,\n $0.baseAddress!,\n DWORD($0.count),\n DWORD(FILE_NAME_NORMALIZED)\n )\n let path = String(decodingCString: $0.baseAddress!, as: UTF16.self)\n return try AbsolutePath(path)\n }\n #else\n let pathStr = path.pathString\n\n // FIXME: We can't use FileManager's destinationOfSymbolicLink because\n // that implements readlink and not realpath.\n if let resultPtr = realpath(pathStr, nil) {\n let result = String(cString: resultPtr)\n // If `resolved_path` is specified as NULL, then `realpath` uses\n // malloc(3) to allocate a buffer [...]. The caller should deallocate\n // this buffer using free(3).\n //\n // String.init(cString:) creates a new string by copying the\n // null-terminated UTF-8 data referenced by the given pointer.\n resultPtr.deallocate()\n // FIXME: We should measure if it's really more efficient to compare the strings first.\n return result == pathStr ? path : try AbsolutePath(validating: result)\n }\n\n return path\n #endif\n}\n\n/// Creates a new, empty directory at `path`. If needed, any non-existent ancestor paths are also\n/// created. If there is\n/// already a directory at `path`, this function does nothing (in particular, this is not considered\n/// to be an error).\npublic func makeDirectories(_ path: AbsolutePath) throws {\n try FileManager.default.createDirectory(\n atPath: path.pathString,\n withIntermediateDirectories: true,\n attributes: [:]\n )\n}\n\n/// Creates a symbolic link at `path` whose content points to `dest`. If `relative` is true, the\n/// symlink contents will\n/// be a relative path, otherwise it will be absolute.\n@available(*, deprecated, renamed: \"localFileSystem.createSymbolicLink\")\npublic func createSymlink(\n _ path: AbsolutePath,\n pointingAt dest: AbsolutePath,\n relative: Bool = true\n) throws {\n let destString = relative ? dest.relative(to: path.parentDirectory).pathString : dest.pathString\n try FileManager.default.createSymbolicLink(\n atPath: path.pathString,\n withDestinationPath: destString\n )\n}\n\n/**\n - Returns: a generator that walks the specified directory producing all\n files therein. If recursively is true will enter any directories\n encountered recursively.\n\n - Warning: directories that cannot be entered due to permission problems\n are silently ignored. So keep that in mind.\n\n - Warning: Symbolic links that point to directories are *not* followed.\n\n - Note: setting recursively to `false` still causes the generator to feed\n you the directory; just not its contents.\n */\npublic func walk(\n _ path: AbsolutePath,\n fileSystem: FileSystem = localFileSystem,\n recursively: Bool = true\n) throws -> RecursibleDirectoryContentsGenerator {\n return try RecursibleDirectoryContentsGenerator(\n path: path,\n fileSystem: fileSystem,\n recursionFilter: { _ in recursively }\n )\n}\n\n/**\n - Returns: a generator that walks the specified directory producing all\n files therein. Directories are recursed based on the return value of\n `recursing`.\n\n - Warning: directories that cannot be entered due to permissions problems\n are silently ignored. So keep that in mind.\n\n - Warning: Symbolic links that point to directories are *not* followed.\n\n - Note: returning `false` from `recursing` still produces that directory\n from the generator; just not its contents.\n */\npublic func walk(\n _ path: AbsolutePath,\n fileSystem: FileSystem = localFileSystem,\n recursing: @escaping (AbsolutePath) -> Bool\n) throws -> RecursibleDirectoryContentsGenerator {\n return try RecursibleDirectoryContentsGenerator(\n path: path,\n fileSystem: fileSystem,\n recursionFilter: recursing\n )\n}\n\n/**\n Produced by `walk`.\n */\npublic class RecursibleDirectoryContentsGenerator: IteratorProtocol, Sequence {\n private var current: (path: AbsolutePath, iterator: IndexingIterator<[String]>)\n private var towalk = [AbsolutePath]()\n\n private let shouldRecurse: (AbsolutePath) -> Bool\n private let fileSystem: FileSystem\n\n fileprivate init(\n path: AbsolutePath,\n fileSystem: FileSystem,\n recursionFilter: @escaping (AbsolutePath) -> Bool\n ) throws {\n self.fileSystem = fileSystem\n // FIXME: getDirectoryContents should have an iterator version.\n current = try (\n path,\n fileSystem.getDirectoryContents(at: path).map(\\.basename).makeIterator()\n )\n shouldRecurse = recursionFilter\n }\n\n public func next() -> AbsolutePath? {\n outer: while true {\n guard let entry = current.iterator.next() else {\n while !towalk.isEmpty {\n // FIXME: This looks inefficient.\n let path = towalk.removeFirst()\n guard shouldRecurse(path) else { continue }\n // Ignore if we can't get content for this path.\n guard let current = try? fileSystem.getDirectoryContents(at: path)\n .map(\\.basename)\n .makeIterator() else { continue }\n self.current = (path, current)\n continue outer\n }\n return nil\n }\n\n let path = current.path.appending(component: entry)\n if fileSystem.isDirectory(path) && !fileSystem.isSymlink(path) {\n towalk.append(path)\n }\n return path\n }\n }\n}\n\npublic extension AbsolutePath {\n /// Returns a path suitable for display to the user (if possible, it is made\n /// to be relative to the current working directory).\n func prettyPath(cwd: AbsolutePath? = localFileSystem.currentWorkingDirectory) -> String {\n guard let dir = cwd else {\n // No current directory, display as is.\n return pathString\n }\n // FIXME: Instead of string prefix comparison we should add a proper API\n // to AbsolutePath to determine ancestry.\n if self == dir {\n return \".\"\n } else if pathString.hasPrefix(dir.pathString + \"/\") {\n return \"./\" + relative(to: dir).pathString\n } else {\n return pathString\n }\n }\n}\n\n" }, { "path": "Tool/Sources/FileSystem/WritableByteStream.swift", "content": "/*\n This source file is part of the Swift.org open source project\n\n Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors\n Licensed under Apache License v2.0 with Runtime Library Exception\n\n See http://swift.org/LICENSE.txt for license information\n See http://swift.org/CONTRIBUTORS.txt for Swift project authors\n*/\n\n/// Closable entity is one that manages underlying resources and needs to be closed for cleanup\n/// The intent of this method is for the sole owner of the refernece/handle of the resource to close it completely, compared to releasing a shared resource.\npublic protocol Closable {\n func close() throws\n}\n\nimport Dispatch\n\n#if canImport(Glibc)\n@_exported import Glibc\n#elseif canImport(Musl)\n@_exported import Musl\n#elseif os(Windows)\n@_exported import CRT\n@_exported import WinSDK\n#else\n@_exported import Darwin.C\n#endif\n\n/// Convert an integer in 0..<16 to its hexadecimal ASCII character.\nprivate func hexdigit(_ value: UInt8) -> UInt8 {\n return value < 10 ? (0x30 + value) : (0x41 + value - 10)\n}\n\n/// Describes a type which can be written to a byte stream.\npublic protocol ByteStreamable {\n func write(to stream: WritableByteStream)\n}\n\n/// An output byte stream.\n///\n/// This protocol is designed to be able to support efficient streaming to\n/// different output destinations, e.g., a file or an in memory buffer. This is\n/// loosely modeled on LLVM's llvm::raw_ostream class.\n///\n/// The stream is generally used in conjunction with the `appending` function.\n/// For example:\n///\n/// let stream = BufferedOutputByteStream()\n/// stream.appending(\"Hello, world!\")\n///\n/// would write the UTF8 encoding of \"Hello, world!\" to the stream.\n///\n/// The stream accepts a number of custom formatting operators which are defined\n/// in the `Format` struct (used for namespacing purposes). For example:\n///\n/// let items = [\"hello\", \"world\"]\n/// stream.appending(Format.asSeparatedList(items, separator: \" \"))\n///\n/// would write each item in the list to the stream, separating them with a\n/// space.\npublic protocol WritableByteStream: AnyObject, TextOutputStream, Closable {\n /// The current offset within the output stream.\n var position: Int { get }\n\n /// Write an individual byte to the buffer.\n func write(_ byte: UInt8)\n\n /// Write a collection of bytes to the buffer.\n func write(_ bytes: C) where C.Element == UInt8\n\n /// Flush the stream's buffer.\n func flush()\n}\n\n// Default noop implementation of close to avoid source-breaking downstream dependents with the addition of the close\n// API.\npublic extension WritableByteStream {\n func close() throws { }\n}\n\n// Public alias to the old name to not introduce API compatibility.\npublic typealias OutputByteStream = WritableByteStream\n\n#if os(Android) || canImport(Musl)\npublic typealias FILEPointer = OpaquePointer\n#else\npublic typealias FILEPointer = UnsafeMutablePointer\n#endif\n\nextension WritableByteStream {\n /// Write a sequence of bytes to the buffer.\n public func write(sequence: S) where S.Iterator.Element == UInt8 {\n // Iterate the sequence and append byte by byte since sequence's append\n // is not performant anyway.\n for byte in sequence {\n write(byte)\n }\n }\n\n /// Write a string to the buffer (as UTF8).\n public func write(_ string: String) {\n // FIXME(performance): Use `string.utf8._copyContents(initializing:)`.\n write(string.utf8)\n }\n\n /// Write a string (as UTF8) to the buffer, with escaping appropriate for\n /// embedding within a JSON document.\n ///\n /// - Note: This writes the literal data applying JSON string escaping, but\n /// does not write any other characters (like the quotes that would surround\n /// a JSON string).\n public func writeJSONEscaped(_ string: String) {\n // See RFC7159 for reference: https://tools.ietf.org/html/rfc7159\n for character in string.utf8 {\n // Handle string escapes; we use constants here to directly match the RFC.\n switch character {\n // Literal characters.\n case 0x20...0x21, 0x23...0x5B, 0x5D...0xFF:\n write(character)\n\n // Single-character escaped characters.\n case 0x22: // '\"'\n write(0x5C) // '\\'\n write(0x22) // '\"'\n case 0x5C: // '\\\\'\n write(0x5C) // '\\'\n write(0x5C) // '\\'\n case 0x08: // '\\b'\n write(0x5C) // '\\'\n write(0x62) // 'b'\n case 0x0C: // '\\f'\n write(0x5C) // '\\'\n write(0x66) // 'b'\n case 0x0A: // '\\n'\n write(0x5C) // '\\'\n write(0x6E) // 'n'\n case 0x0D: // '\\r'\n write(0x5C) // '\\'\n write(0x72) // 'r'\n case 0x09: // '\\t'\n write(0x5C) // '\\'\n write(0x74) // 't'\n\n // Multi-character escaped characters.\n default:\n write(0x5C) // '\\'\n write(0x75) // 'u'\n write(hexdigit(0))\n write(hexdigit(0))\n write(hexdigit(character >> 4))\n write(hexdigit(character & 0xF))\n }\n }\n }\n\n // MARK: helpers that return `self`\n\n // FIXME: This override shouldn't be necessary but removing it causes a 30% performance regression. This problem is\n // tracked by the following bug: https://bugs.swift.org/browse/SR-8535\n @discardableResult\n public func send(_ value: ArraySlice) -> WritableByteStream {\n value.write(to: self)\n return self\n }\n\n @discardableResult\n public func send(_ value: ByteStreamable) -> WritableByteStream {\n value.write(to: self)\n return self\n }\n\n @discardableResult\n public func send(_ value: CustomStringConvertible) -> WritableByteStream {\n value.description.write(to: self)\n return self\n }\n\n @discardableResult\n public func send(_ value: ByteStreamable & CustomStringConvertible) -> WritableByteStream {\n value.write(to: self)\n return self\n }\n}\n\n/// The `WritableByteStream` base class.\n///\n/// This class provides a base and efficient implementation of the `WritableByteStream`\n/// protocol. It can not be used as is-as subclasses as several functions need to be\n/// implemented in subclasses.\npublic class _WritableByteStreamBase: WritableByteStream {\n /// If buffering is enabled\n @usableFromInline let _buffered : Bool\n\n /// The data buffer.\n /// - Note: Minimum Buffer size should be one.\n @usableFromInline var _buffer: [UInt8]\n\n /// Default buffer size of the data buffer.\n private static let bufferSize = 1024\n\n /// Queue to protect mutating operation.\n fileprivate let queue = DispatchQueue(label: \"org.swift.swiftpm.basic.stream\")\n\n init(buffered: Bool) {\n self._buffered = buffered\n self._buffer = []\n\n // When not buffered we still reserve 1 byte, as it is used by the\n // by the single byte write() variant.\n self._buffer.reserveCapacity(buffered ? _WritableByteStreamBase.bufferSize : 1)\n }\n\n // MARK: Data Access API\n\n /// The current offset within the output stream.\n public var position: Int {\n return _buffer.count\n }\n\n /// Currently available buffer size.\n @usableFromInline var _availableBufferSize: Int {\n return _buffer.capacity - _buffer.count\n }\n\n /// Clears the buffer maintaining current capacity.\n @usableFromInline func _clearBuffer() {\n _buffer.removeAll(keepingCapacity: true)\n }\n\n // MARK: Data Output API\n\n public final func flush() {\n writeImpl(ArraySlice(_buffer))\n _clearBuffer()\n flushImpl()\n }\n\n @usableFromInline func flushImpl() {\n // Do nothing.\n }\n\n public final func close() throws {\n try closeImpl()\n }\n\n @usableFromInline func closeImpl() throws {\n fatalError(\"Subclasses must implement this\")\n }\n\n @usableFromInline func writeImpl(_ bytes: C) where C.Iterator.Element == UInt8 {\n fatalError(\"Subclasses must implement this\")\n }\n\n @usableFromInline func writeImpl(_ bytes: ArraySlice) {\n fatalError(\"Subclasses must implement this\")\n }\n\n /// Write an individual byte to the buffer.\n public final func write(_ byte: UInt8) {\n guard _buffered else {\n _buffer.append(byte)\n writeImpl(ArraySlice(_buffer))\n flushImpl()\n _clearBuffer()\n return\n }\n\n // If buffer is full, write and clear it.\n if _availableBufferSize == 0 {\n writeImpl(ArraySlice(_buffer))\n _clearBuffer()\n }\n\n // This will need to change change if we ever have unbuffered stream.\n precondition(_availableBufferSize > 0)\n _buffer.append(byte)\n }\n\n /// Write a collection of bytes to the buffer.\n @inlinable public final func write(_ bytes: C) where C.Element == UInt8 {\n guard _buffered else {\n if let b = bytes as? ArraySlice {\n // Fast path for unbuffered ArraySlice\n writeImpl(b)\n } else if let b = bytes as? Array {\n // Fast path for unbuffered Array\n writeImpl(ArraySlice(b))\n } else {\n // generic collection unfortunately must be temporarily buffered\n writeImpl(bytes)\n }\n flushImpl()\n return\n }\n\n // This is based on LLVM's raw_ostream.\n let availableBufferSize = self._availableBufferSize\n let byteCount = Int(bytes.count)\n\n // If we have to insert more than the available space in buffer.\n if byteCount > availableBufferSize {\n // If buffer is empty, start writing and keep the last chunk in buffer.\n if _buffer.isEmpty {\n let bytesToWrite = byteCount - (byteCount % availableBufferSize)\n let writeUptoIndex = bytes.index(bytes.startIndex, offsetBy: numericCast(bytesToWrite))\n writeImpl(bytes.prefix(upTo: writeUptoIndex))\n\n // If remaining bytes is more than buffer size write everything.\n let bytesRemaining = byteCount - bytesToWrite\n if bytesRemaining > availableBufferSize {\n writeImpl(bytes.suffix(from: writeUptoIndex))\n return\n }\n // Otherwise keep remaining in buffer.\n _buffer += bytes.suffix(from: writeUptoIndex)\n return\n }\n\n let writeUptoIndex = bytes.index(bytes.startIndex, offsetBy: numericCast(availableBufferSize))\n // Append whatever we can accommodate.\n _buffer += bytes.prefix(upTo: writeUptoIndex)\n\n writeImpl(ArraySlice(_buffer))\n _clearBuffer()\n\n // FIXME: We should start again with remaining chunk but this doesn't work. Write everything for now.\n //write(collection: bytes.suffix(from: writeUptoIndex))\n writeImpl(bytes.suffix(from: writeUptoIndex))\n return\n }\n _buffer += bytes\n }\n}\n\n/// The thread-safe wrapper around output byte streams.\n///\n/// This class wraps any `WritableByteStream` conforming type to provide a type-safe\n/// access to its operations. If the provided stream inherits from `_WritableByteStreamBase`,\n/// it will also ensure it is type-safe will all other `ThreadSafeOutputByteStream` instances\n/// around the same stream.\npublic final class ThreadSafeOutputByteStream: WritableByteStream {\n private static let defaultQueue = DispatchQueue(label: \"org.swift.swiftpm.basic.thread-safe-output-byte-stream\")\n public let stream: WritableByteStream\n private let queue: DispatchQueue\n\n public var position: Int {\n return queue.sync {\n stream.position\n }\n }\n\n public init(_ stream: WritableByteStream) {\n self.stream = stream\n self.queue = (stream as? _WritableByteStreamBase)?.queue ?? ThreadSafeOutputByteStream.defaultQueue\n }\n\n public func write(_ byte: UInt8) {\n queue.sync {\n stream.write(byte)\n }\n }\n\n public func write(_ bytes: C) where C.Element == UInt8 {\n queue.sync {\n stream.write(bytes)\n }\n }\n\n public func flush() {\n queue.sync {\n stream.flush()\n }\n }\n\n public func write(sequence: S) where S.Iterator.Element == UInt8 {\n queue.sync {\n stream.write(sequence: sequence)\n }\n }\n\n public func writeJSONEscaped(_ string: String) {\n queue.sync {\n stream.writeJSONEscaped(string)\n }\n }\n\n public func close() throws {\n try queue.sync {\n try stream.close()\n }\n }\n}\n\n\n#if swift(<5.6)\nextension ThreadSafeOutputByteStream: UnsafeSendable {}\n#else\nextension ThreadSafeOutputByteStream: @unchecked Sendable {}\n#endif\n\n/// Define an output stream operator. We need it to be left associative, so we\n/// use `<<<`.\ninfix operator <<< : StreamingPrecedence\nprecedencegroup StreamingPrecedence {\n associativity: left\n}\n\n// MARK: Output Operator Implementations\n\n// FIXME: This override shouldn't be necessary but removing it causes a 30% performance regression. This problem is\n// tracked by the following bug: https://bugs.swift.org/browse/SR-8535\n\n@available(*, deprecated, message: \"use send(_:) function on WritableByteStream instead\")\n@discardableResult\npublic func <<< (stream: WritableByteStream, value: ArraySlice) -> WritableByteStream {\n value.write(to: stream)\n return stream\n}\n\n@available(*, deprecated, message: \"use send(_:) function on WritableByteStream instead\")\n@discardableResult\npublic func <<< (stream: WritableByteStream, value: ByteStreamable) -> WritableByteStream {\n value.write(to: stream)\n return stream\n}\n\n@available(*, deprecated, message: \"use send(_:) function on WritableByteStream instead\")\n@discardableResult\npublic func <<< (stream: WritableByteStream, value: CustomStringConvertible) -> WritableByteStream {\n value.description.write(to: stream)\n return stream\n}\n\n@available(*, deprecated, message: \"use send(_:) function on WritableByteStream instead\")\n@discardableResult\npublic func <<< (stream: WritableByteStream, value: ByteStreamable & CustomStringConvertible) -> WritableByteStream {\n value.write(to: stream)\n return stream\n}\n\nextension UInt8: ByteStreamable {\n public func write(to stream: WritableByteStream) {\n stream.write(self)\n }\n}\n\nextension Character: ByteStreamable {\n public func write(to stream: WritableByteStream) {\n stream.write(String(self))\n }\n}\n\nextension String: ByteStreamable {\n public func write(to stream: WritableByteStream) {\n stream.write(self.utf8)\n }\n}\n\nextension Substring: ByteStreamable {\n public func write(to stream: WritableByteStream) {\n stream.write(self.utf8)\n }\n}\n\nextension StaticString: ByteStreamable {\n public func write(to stream: WritableByteStream) {\n withUTF8Buffer { stream.write($0) }\n }\n}\n\nextension Array: ByteStreamable where Element == UInt8 {\n public func write(to stream: WritableByteStream) {\n stream.write(self)\n }\n}\n\nextension ArraySlice: ByteStreamable where Element == UInt8 {\n public func write(to stream: WritableByteStream) {\n stream.write(self)\n }\n}\n\nextension ContiguousArray: ByteStreamable where Element == UInt8 {\n public func write(to stream: WritableByteStream) {\n stream.write(self)\n }\n}\n\n// MARK: Formatted Streaming Output\n\n/// Provides operations for returning derived streamable objects to implement various forms of formatted output.\npublic struct Format {\n /// Write the input boolean encoded as a JSON object.\n static public func asJSON(_ value: Bool) -> ByteStreamable {\n return JSONEscapedBoolStreamable(value: value)\n }\n private struct JSONEscapedBoolStreamable: ByteStreamable {\n let value: Bool\n\n func write(to stream: WritableByteStream) {\n stream.send(value ? \"true\" : \"false\")\n }\n }\n\n /// Write the input integer encoded as a JSON object.\n static public func asJSON(_ value: Int) -> ByteStreamable {\n return JSONEscapedIntStreamable(value: value)\n }\n private struct JSONEscapedIntStreamable: ByteStreamable {\n let value: Int\n\n func write(to stream: WritableByteStream) {\n // FIXME: Diagnose integers which cannot be represented in JSON.\n stream.send(value.description)\n }\n }\n\n /// Write the input double encoded as a JSON object.\n static public func asJSON(_ value: Double) -> ByteStreamable {\n return JSONEscapedDoubleStreamable(value: value)\n }\n private struct JSONEscapedDoubleStreamable: ByteStreamable {\n let value: Double\n\n func write(to stream: WritableByteStream) {\n // FIXME: What should we do about NaN, etc.?\n //\n // FIXME: Is Double.debugDescription the best representation?\n stream.send(value.debugDescription)\n }\n }\n\n /// Write the input CustomStringConvertible encoded as a JSON object.\n static public func asJSON(_ value: T) -> ByteStreamable {\n return JSONEscapedStringStreamable(value: value.description)\n }\n /// Write the input string encoded as a JSON object.\n static public func asJSON(_ string: String) -> ByteStreamable {\n return JSONEscapedStringStreamable(value: string)\n }\n private struct JSONEscapedStringStreamable: ByteStreamable {\n let value: String\n\n func write(to stream: WritableByteStream) {\n stream.send(UInt8(ascii: \"\\\"\"))\n stream.writeJSONEscaped(value)\n stream.send(UInt8(ascii: \"\\\"\"))\n }\n }\n\n /// Write the input string list encoded as a JSON object.\n static public func asJSON(_ items: [T]) -> ByteStreamable {\n return JSONEscapedStringListStreamable(items: items.map({ $0.description }))\n }\n /// Write the input string list encoded as a JSON object.\n //\n // FIXME: We might be able to make this more generic through the use of a \"JSONEncodable\" protocol.\n static public func asJSON(_ items: [String]) -> ByteStreamable {\n return JSONEscapedStringListStreamable(items: items)\n }\n private struct JSONEscapedStringListStreamable: ByteStreamable {\n let items: [String]\n\n func write(to stream: WritableByteStream) {\n stream.send(UInt8(ascii: \"[\"))\n for (i, item) in items.enumerated() {\n if i != 0 { stream.send(\",\") }\n stream.send(Format.asJSON(item))\n }\n stream.send(UInt8(ascii: \"]\"))\n }\n }\n\n /// Write the input dictionary encoded as a JSON object.\n static public func asJSON(_ items: [String: String]) -> ByteStreamable {\n return JSONEscapedDictionaryStreamable(items: items)\n }\n private struct JSONEscapedDictionaryStreamable: ByteStreamable {\n let items: [String: String]\n\n func write(to stream: WritableByteStream) {\n stream.send(UInt8(ascii: \"{\"))\n for (offset: i, element: (key: key, value: value)) in items.enumerated() {\n if i != 0 { stream.send(\",\") }\n stream.send(Format.asJSON(key)).send(\":\").send(Format.asJSON(value))\n }\n stream.send(UInt8(ascii: \"}\"))\n }\n }\n\n /// Write the input list (after applying a transform to each item) encoded as a JSON object.\n //\n // FIXME: We might be able to make this more generic through the use of a \"JSONEncodable\" protocol.\n static public func asJSON(_ items: [T], transform: @escaping (T) -> String) -> ByteStreamable {\n return JSONEscapedTransformedStringListStreamable(items: items, transform: transform)\n }\n private struct JSONEscapedTransformedStringListStreamable: ByteStreamable {\n let items: [T]\n let transform: (T) -> String\n\n func write(to stream: WritableByteStream) {\n stream.send(UInt8(ascii: \"[\"))\n for (i, item) in items.enumerated() {\n if i != 0 { stream.send(\",\") }\n stream.send(Format.asJSON(transform(item)))\n }\n stream.send(UInt8(ascii: \"]\"))\n }\n }\n\n /// Write the input list to the stream with the given separator between items.\n static public func asSeparatedList(_ items: [T], separator: String) -> ByteStreamable {\n return SeparatedListStreamable(items: items, separator: separator)\n }\n private struct SeparatedListStreamable: ByteStreamable {\n let items: [T]\n let separator: String\n\n func write(to stream: WritableByteStream) {\n for (i, item) in items.enumerated() {\n // Add the separator, if necessary.\n if i != 0 {\n stream.send(separator)\n }\n\n stream.send(item)\n }\n }\n }\n\n /// Write the input list to the stream (after applying a transform to each item) with the given separator between\n /// items.\n static public func asSeparatedList(\n _ items: [T],\n transform: @escaping (T) -> ByteStreamable,\n separator: String\n ) -> ByteStreamable {\n return TransformedSeparatedListStreamable(items: items, transform: transform, separator: separator)\n }\n private struct TransformedSeparatedListStreamable: ByteStreamable {\n let items: [T]\n let transform: (T) -> ByteStreamable\n let separator: String\n\n func write(to stream: WritableByteStream) {\n for (i, item) in items.enumerated() {\n if i != 0 { stream.send(separator) }\n stream.send(transform(item))\n }\n }\n }\n\n static public func asRepeating(string: String, count: Int) -> ByteStreamable {\n return RepeatingStringStreamable(string: string, count: count)\n }\n private struct RepeatingStringStreamable: ByteStreamable {\n let string: String\n let count: Int\n\n init(string: String, count: Int) {\n precondition(count >= 0, \"Count should be >= zero\")\n self.string = string\n self.count = count\n }\n\n func write(to stream: WritableByteStream) {\n for _ in 0..(_ bytes: C) where C.Iterator.Element == UInt8 {\n contents += bytes\n }\n override final func writeImpl(_ bytes: ArraySlice) {\n contents += bytes\n }\n\n override final func closeImpl() throws {\n // Do nothing. The protocol does not require to stop receiving writes, close only signals that resources could\n // be released at this point should we need to.\n }\n}\n\n/// Represents a stream which is backed to a file. Not for instantiating.\npublic class FileOutputByteStream: _WritableByteStreamBase {\n\n public override final func closeImpl() throws {\n flush()\n try fileCloseImpl()\n }\n\n /// Closes the file flushing any buffered data.\n func fileCloseImpl() throws {\n fatalError(\"fileCloseImpl() should be implemented by a subclass\")\n }\n}\n\n/// Implements file output stream for local file system.\npublic final class LocalFileOutputByteStream: FileOutputByteStream {\n\n /// The pointer to the file.\n let filePointer: FILEPointer\n\n /// Set to an error value if there were any IO error during writing.\n private var error: FileSystemError?\n\n /// Closes the file on deinit if true.\n private var closeOnDeinit: Bool\n\n /// Path to the file this stream should operate on.\n private let path: AbsolutePath?\n\n /// Instantiate using the file pointer.\n public init(filePointer: FILEPointer, closeOnDeinit: Bool = true, buffered: Bool = true) throws {\n self.filePointer = filePointer\n self.closeOnDeinit = closeOnDeinit\n self.path = nil\n super.init(buffered: buffered)\n }\n\n /// Opens the file for writing at the provided path.\n ///\n /// - Parameters:\n /// - path: Path to the file this stream should operate on.\n /// - closeOnDeinit: If true closes the file on deinit. clients can use\n /// close() if they want to close themselves or catch\n /// errors encountered during writing to the file.\n /// Default value is true.\n /// - buffered: If true buffers writes in memory until full or flush().\n /// Otherwise, writes are processed and flushed immediately.\n /// Default value is true.\n ///\n /// - Throws: FileSystemError\n public init(_ path: AbsolutePath, closeOnDeinit: Bool = true, buffered: Bool = true) throws {\n guard let filePointer = fopen(path.pathString, \"wb\") else {\n throw FileSystemError(errno: errno, path)\n }\n self.path = path\n self.filePointer = filePointer\n self.closeOnDeinit = closeOnDeinit\n super.init(buffered: buffered)\n }\n\n deinit {\n if closeOnDeinit {\n fclose(filePointer)\n }\n }\n\n func errorDetected(code: Int32?) {\n if let code = code {\n error = .init(.ioError(code: code), path)\n } else {\n error = .init(.unknownOSError, path)\n }\n }\n\n override final func writeImpl(_ bytes: C) where C.Iterator.Element == UInt8 {\n // FIXME: This will be copying bytes but we don't have option currently.\n var contents = [UInt8](bytes)\n while true {\n let n = fwrite(&contents, 1, contents.count, filePointer)\n if n < 0 {\n if errno == EINTR { continue }\n errorDetected(code: errno)\n } else if n != contents.count {\n errorDetected(code: nil)\n }\n break\n }\n }\n\n override final func writeImpl(_ bytes: ArraySlice) {\n bytes.withUnsafeBytes { bytesPtr in\n while true {\n let n = fwrite(bytesPtr.baseAddress!, 1, bytesPtr.count, filePointer)\n if n < 0 {\n if errno == EINTR { continue }\n errorDetected(code: errno)\n } else if n != bytesPtr.count {\n errorDetected(code: nil)\n }\n break\n }\n }\n }\n\n override final func flushImpl() {\n fflush(filePointer)\n }\n\n override final func fileCloseImpl() throws {\n defer {\n fclose(filePointer)\n // If clients called close we shouldn't call fclose again in deinit.\n closeOnDeinit = false\n }\n // Throw if errors were found during writing.\n if let error = error {\n throw error\n }\n }\n}\n\n/// Public stdout stream instance.\npublic var stdoutStream: ThreadSafeOutputByteStream = try! ThreadSafeOutputByteStream(LocalFileOutputByteStream(\n filePointer: stdout,\n closeOnDeinit: false))\n\n/// Public stderr stream instance.\npublic var stderrStream: ThreadSafeOutputByteStream = try! ThreadSafeOutputByteStream(LocalFileOutputByteStream(\n filePointer: stderr,\n closeOnDeinit: false))\n" }, { "path": "Tool/Sources/FocusedCodeFinder/ActiveDocumentContext.swift", "content": "import Foundation\nimport SuggestionBasic\n\npublic struct ActiveDocumentContext: Sendable {\n public var documentURL: URL\n public var relativePath: String\n public var language: CodeLanguage\n public var fileContent: String\n public var lines: [String]\n public var selectedCode: String\n public var selectionRange: CursorRange\n public var lineAnnotations: [EditorInformation.LineAnnotation]\n public var imports: [String]\n public var includes: [String]\n\n public struct FocusedContext: Sendable {\n public struct Context: Equatable, Sendable {\n public var signature: String\n public var name: String\n public var range: CursorRange\n\n public init(signature: String, name: String, range: CursorRange) {\n self.signature = signature\n self.name = name\n self.range = range\n }\n }\n\n public var context: [Context]\n public var contextRange: CursorRange\n public var smallestContextRange: CursorRange\n public var codeRange: CursorRange\n public var code: String\n public var lineAnnotations: [EditorInformation.LineAnnotation]\n public var otherLineAnnotations: [EditorInformation.LineAnnotation]\n\n public init(\n context: [Context],\n contextRange: CursorRange,\n smallestContextRange: CursorRange,\n codeRange: CursorRange,\n code: String,\n lineAnnotations: [EditorInformation.LineAnnotation],\n otherLineAnnotations: [EditorInformation.LineAnnotation]\n ) {\n self.context = context\n self.contextRange = contextRange\n self.smallestContextRange = smallestContextRange\n self.codeRange = codeRange\n self.code = code\n self.lineAnnotations = lineAnnotations\n self.otherLineAnnotations = otherLineAnnotations\n }\n }\n\n public var focusedContext: FocusedContext?\n\n public init(\n documentURL: URL,\n relativePath: String,\n language: CodeLanguage,\n fileContent: String,\n lines: [String],\n selectedCode: String,\n selectionRange: CursorRange,\n lineAnnotations: [EditorInformation.LineAnnotation],\n imports: [String],\n includes: [String],\n focusedContext: FocusedContext? = nil\n ) {\n self.documentURL = documentURL\n self.relativePath = relativePath\n self.language = language\n self.fileContent = fileContent\n self.lines = lines\n self.selectedCode = selectedCode\n self.selectionRange = selectionRange\n self.lineAnnotations = lineAnnotations\n self.imports = imports\n self.includes = includes\n self.focusedContext = focusedContext\n }\n \n public static func empty() -> ActiveDocumentContext {\n .init(\n documentURL: .init(fileURLWithPath: \"/\"),\n relativePath: \"\",\n language: .builtIn(.swift),\n fileContent: \"\",\n lines: [],\n selectedCode: \"\",\n selectionRange: .outOfScope,\n lineAnnotations: [],\n imports: [],\n includes: []\n )\n }\n\n public mutating func moveToFocusedCode() {\n moveToCodeContainingRange(selectionRange)\n }\n\n public mutating func moveToCodeAroundLine(_ line: Int) {\n moveToCodeContainingRange(.init(\n start: .init(line: line, character: 0),\n end: .init(line: line, character: 0)\n ))\n }\n\n public mutating func expandFocusedRangeToContextRange() {\n guard let focusedContext else { return }\n moveToCodeContainingRange(focusedContext.contextRange)\n }\n\n public mutating func moveToCodeContainingRange(_ range: CursorRange) {\n let finder = FocusedCodeFinder(\n maxFocusedCodeLineCount: UserDefaults.shared.value(for: \\.maxFocusedCodeLineCount)\n )\n\n let codeContext = finder.findFocusedCode(\n in: .init(documentURL: documentURL, content: fileContent, lines: lines),\n containingRange: range,\n language: language\n )\n\n imports = codeContext.imports\n includes = codeContext.includes\n\n let startLine = codeContext.focusedRange.start.line\n let endLine = codeContext.focusedRange.end.line\n var matchedAnnotations = [EditorInformation.LineAnnotation]()\n var otherAnnotations = [EditorInformation.LineAnnotation]()\n for annotation in lineAnnotations {\n if annotation.line >= startLine, annotation.line <= endLine {\n matchedAnnotations.append(annotation)\n } else {\n otherAnnotations.append(annotation)\n }\n }\n\n focusedContext = .init(\n context: codeContext.scopeContexts,\n contextRange: codeContext.contextRange,\n smallestContextRange: codeContext.smallestContextRange,\n codeRange: codeContext.focusedRange,\n code: codeContext.focusedCode,\n lineAnnotations: matchedAnnotations,\n otherLineAnnotations: otherAnnotations\n )\n }\n\n public mutating func update(_ info: EditorInformation) {\n /// Whenever the file content, relative path, or selection range changes,\n /// we should reset the context.\n let changed: Bool = {\n if info.relativePath != relativePath { return true }\n if info.editorContent?.content != fileContent { return true }\n if let range = info.editorContent?.selections.first,\n range != selectionRange { return true }\n return false\n }()\n\n documentURL = info.documentURL\n relativePath = info.relativePath\n language = info.language\n fileContent = info.editorContent?.content ?? \"\"\n lines = info.editorContent?.lines ?? []\n selectedCode = info.selectedContent\n selectionRange = info.editorContent?.selections.first ?? .zero\n lineAnnotations = info.editorContent?.lineAnnotations ?? []\n imports = []\n includes = []\n\n if changed {\n moveToFocusedCode()\n }\n }\n}\n\n" }, { "path": "Tool/Sources/FocusedCodeFinder/FocusedCodeFinder.swift", "content": "import Foundation\nimport SuggestionBasic\n\npublic struct CodeContext: Equatable {\n public typealias ScopeContext = ActiveDocumentContext.FocusedContext.Context\n\n public enum Scope: Equatable {\n case file\n case top\n case scope(signature: [ScopeContext])\n }\n\n public var scopeContexts: [ScopeContext] {\n switch scope {\n case .file:\n return []\n case .top:\n return []\n case let .scope(contexts):\n return contexts\n }\n }\n\n public var scope: Scope\n public var contextRange: CursorRange\n public var smallestContextRange: CursorRange\n public var focusedRange: CursorRange\n public var focusedCode: String\n public var imports: [String]\n public var includes: [String]\n\n public static var empty: CodeContext {\n .init(\n scope: .file,\n contextRange: .zero,\n smallestContextRange: .zero,\n focusedRange: .zero,\n focusedCode: \"\",\n imports: [],\n includes: []\n )\n }\n\n public init(\n scope: Scope,\n contextRange: CursorRange,\n smallestContextRange: CursorRange,\n focusedRange: CursorRange,\n focusedCode: String,\n imports: [String],\n includes: [String]\n ) {\n self.scope = scope\n self.contextRange = contextRange\n self.smallestContextRange = smallestContextRange\n self.focusedRange = focusedRange\n self.focusedCode = focusedCode\n self.imports = imports\n self.includes = includes\n }\n}\n\npublic struct FocusedCodeFinder {\n public let maxFocusedCodeLineCount: Int\n\n public init(maxFocusedCodeLineCount: Int) {\n self.maxFocusedCodeLineCount = maxFocusedCodeLineCount\n }\n\n public struct Document {\n var documentURL: URL\n var content: String\n var lines: [String]\n\n public init(documentURL: URL, content: String, lines: [String]) {\n self.documentURL = documentURL\n self.content = content\n self.lines = lines\n }\n }\n\n public func findFocusedCode(\n in document: Document,\n containingRange: CursorRange,\n language: CodeLanguage\n ) -> CodeContext {\n let finder: FocusedCodeFinderType = {\n switch language {\n case .builtIn(.swift):\n return SwiftFocusedCodeFinder(maxFocusedCodeLineCount: maxFocusedCodeLineCount)\n case .builtIn(.objc), .builtIn(.objcpp), .builtIn(.c):\n #warning(\n \"TODO: Implement C++ focused code finder, use it for C and metal shading language\"\n )\n return ObjectiveCFocusedCodeFinder(maxFocusedCodeLineCount: maxFocusedCodeLineCount)\n default:\n return UnknownLanguageFocusedCodeFinder(proposedSearchRange: 5)\n }\n }()\n\n return finder.findFocusedCode(in: document, containingRange: containingRange)\n }\n}\n\npublic protocol FocusedCodeFinderType {\n typealias Document = FocusedCodeFinder.Document\n\n func findFocusedCode(\n in document: Document,\n containingRange: CursorRange\n ) -> CodeContext\n}\n\n" }, { "path": "Tool/Sources/FocusedCodeFinder/KnownLanguageFocusedCodeFinder.swift", "content": "import Foundation\nimport Preferences\nimport SuggestionBasic\n\npublic typealias KnownLanguageFocusedCodeFinder =\n BaseKnownLanguageFocusedCodeFinder &\n KnownLanguageFocusedCodeFinderType\n\npublic class BaseKnownLanguageFocusedCodeFinder {\n public typealias TextProvider = (TextPosition) -> String\n public typealias RangeConverter = (Node) -> CursorRange\n\n public struct NodeInfo {\n var node: Node\n var signature: String\n var name: String\n var canBeUsedAsCodeRange: Bool = true\n }\n\n public struct ContextInfo {\n var nodes: [Node]\n var includes: [String]\n var imports: [String]\n }\n\n public let maxFocusedCodeLineCount: Int\n\n init(\n maxFocusedCodeLineCount: Int = UserDefaults.shared.value(for: \\.maxFocusedCodeLineCount)\n ) {\n self.maxFocusedCodeLineCount = maxFocusedCodeLineCount\n }\n}\n\npublic protocol KnownLanguageFocusedCodeFinderType: FocusedCodeFinderType {\n associatedtype Tree\n associatedtype Node\n associatedtype TextPosition\n typealias Document = FocusedCodeFinder.Document\n typealias Finder = BaseKnownLanguageFocusedCodeFinder\n typealias NodeInfo = Finder.NodeInfo\n typealias ContextInfo = Finder.ContextInfo\n typealias TextProvider = Finder.TextProvider\n typealias RangeConverter = Finder.RangeConverter\n\n var maxFocusedCodeLineCount: Int { get }\n\n func parseSyntaxTree(from document: Document) -> Tree?\n\n func collectContextNodes(\n in document: Document,\n tree: Tree,\n containingRange: SuggestionBasic.CursorRange,\n textProvider: @escaping TextProvider,\n rangeConverter: @escaping RangeConverter\n ) -> ContextInfo\n\n func contextContainingNode(\n _ node: Node,\n textProvider: @escaping TextProvider,\n rangeConverter: @escaping RangeConverter\n ) -> NodeInfo?\n\n func createTextProviderAndRangeConverter(\n for document: Document,\n tree: Tree\n ) -> (TextProvider, RangeConverter)\n}\n\npublic extension KnownLanguageFocusedCodeFinderType {\n func findFocusedCode(\n in document: Document,\n containingRange range: SuggestionBasic.CursorRange\n ) -> CodeContext {\n guard let tree = parseSyntaxTree(from: document) else { return .empty }\n\n let (textProvider, rangeConverter) = createTextProviderAndRangeConverter(\n for: document,\n tree: tree\n )\n var contextInfo = collectContextNodes(\n in: document,\n tree: tree,\n containingRange: range,\n textProvider: textProvider,\n rangeConverter: rangeConverter\n )\n var codeRange: CursorRange\n\n let noSelection = range.isEmpty\n if noSelection {\n // use the first scope as code, the second as context\n var focusedNode: Node?\n while let node = contextInfo.nodes.first {\n contextInfo.nodes.removeFirst()\n let nodeInfo = contextContainingNode(\n node,\n textProvider: textProvider,\n rangeConverter: rangeConverter\n )\n if nodeInfo?.canBeUsedAsCodeRange ?? false {\n focusedNode = node\n break\n }\n }\n guard let focusedNode else {\n // fallback to unknown language focused code finder when no scope found\n var result = UnknownLanguageFocusedCodeFinder(proposedSearchRange: 8)\n .findFocusedCode(in: document, containingRange: range)\n result.imports = contextInfo.imports\n result.includes = contextInfo.includes\n return result\n }\n codeRange = rangeConverter(focusedNode)\n } else {\n // use the selection as code, the first scope as context\n codeRange = range\n }\n\n let (code, _, focusedRange) = extractFocusedCode(\n in: codeRange,\n in: document,\n containingRange: range\n )\n\n let (contextRange, scopeContexts) = extractScopeContext(\n contextNodes: contextInfo.nodes,\n textProvider: textProvider,\n rangeConverter: rangeConverter\n )\n\n return .init(\n scope: scopeContexts.isEmpty ? .file : .scope(signature: scopeContexts),\n contextRange: contextRange,\n smallestContextRange: codeRange,\n focusedRange: focusedRange,\n focusedCode: code,\n imports: contextInfo.imports,\n includes: contextInfo.includes\n )\n }\n}\n\nextension KnownLanguageFocusedCodeFinderType {\n func extractFocusedCode(\n in codeRange: CursorRange,\n in document: Document,\n containingRange range: SuggestionBasic.CursorRange\n ) -> (code: String, lines: [String], codeRange: CursorRange) {\n var codeRange = codeRange\n let codeInCodeRange = EditorInformation.code(\n in: document.lines,\n inside: codeRange,\n ignoreColumns: true\n )\n\n var code = codeInCodeRange.code\n var lines = codeInCodeRange.lines\n\n if range.isEmpty, codeInCodeRange.lines.count > maxFocusedCodeLineCount {\n // if the focused code is too long, truncate it to be shorter\n let centerLine = range.start.line\n let relativeCenterLine = centerLine - codeRange.start.line\n let startLine = max(0, relativeCenterLine - maxFocusedCodeLineCount / 2)\n let endLine = max(\n startLine,\n min(codeInCodeRange.lines.count - 1, startLine + maxFocusedCodeLineCount - 1)\n )\n\n lines = Array(codeInCodeRange.lines[startLine...endLine])\n code = lines.joined()\n codeRange = .init(\n start: .init(line: startLine + codeRange.start.line, character: 0),\n end: .init(\n line: endLine + codeRange.start.line,\n character: codeInCodeRange.lines[endLine].count\n )\n )\n }\n\n return (code, lines, codeRange)\n }\n\n func extractScopeContext(\n contextNodes: [Node],\n textProvider: @escaping TextProvider,\n rangeConverter: @escaping RangeConverter\n ) -> (contextRange: CursorRange, scopeContexts: [CodeContext.ScopeContext]) {\n var nodes = contextNodes\n var contextRange = CursorRange.zero\n var signature = [CodeContext.ScopeContext]()\n\n while let node = nodes.first {\n nodes.removeFirst()\n let context = contextContainingNode(\n node,\n textProvider: textProvider,\n rangeConverter: rangeConverter\n )\n\n if let context {\n contextRange = rangeConverter(context.node)\n signature.insert(.init(\n signature: context.signature,\n name: context.name,\n range: contextRange\n ), at: 0)\n }\n }\n\n return (contextRange, signature)\n }\n}\n\n" }, { "path": "Tool/Sources/FocusedCodeFinder/ObjectiveC/ObjectiveCCodeFinder.swift", "content": "import ASTParser\nimport Foundation\nimport Preferences\nimport SuggestionBasic\nimport SwiftTreeSitter\n\npublic enum TreeSitterTextPosition {\n case node(ASTNode)\n case range(range: NSRange, pointRange: Range)\n}\n\npublic class ObjectiveCFocusedCodeFinder: KnownLanguageFocusedCodeFinder<\n ASTTree,\n ASTNode,\n TreeSitterTextPosition\n> {\n override public init(maxFocusedCodeLineCount: Int) {\n super.init(maxFocusedCodeLineCount: maxFocusedCodeLineCount)\n }\n\n public func parseSyntaxTree(from document: Document) -> ASTTree? {\n let parser = ASTParser(language: .objectiveC)\n return parser.parse(document.content)\n }\n\n public func collectContextNodes(\n in document: Document,\n tree: ASTTree,\n containingRange range: CursorRange,\n textProvider: @escaping TextProvider,\n rangeConverter: @escaping RangeConverter\n ) -> ContextInfo {\n let visitor = ObjectiveCScopeHierarchySyntaxVisitor(\n tree: tree,\n code: document.content,\n textProvider: { node in\n textProvider(.node(node))\n },\n range: range\n )\n\n let nodes = visitor.findScopeHierarchy()\n\n return .init(nodes: nodes, includes: visitor.includes, imports: visitor.imports)\n }\n\n public func createTextProviderAndRangeConverter(\n for document: Document,\n tree: ASTTree\n ) -> (TextProvider, RangeConverter) {\n (\n { position in\n switch position {\n case let .node(node):\n return document.content.cursorTextProvider(node.range, node.pointRange) ?? \"\"\n case let .range(range, pointRange):\n return document.content.cursorTextProvider(range, pointRange) ?? \"\"\n }\n },\n { node in\n CursorRange(pointRange: node.pointRange)\n }\n )\n }\n\n public func contextContainingNode(\n _ node: Node,\n textProvider: @escaping TextProvider,\n rangeConverter: @escaping RangeConverter\n ) -> NodeInfo? {\n switch ObjectiveCNodeType(rawValue: node.nodeType ?? \"\") {\n case .classInterface, .categoryInterface:\n return parseDeclarationInterfaceNode(node, textProvider: textProvider)\n case .classImplementation, .categoryImplementation:\n return parseDeclarationInterfaceNode(node, textProvider: textProvider)\n case .protocolDeclaration:\n return parseDeclarationInterfaceNode(node, textProvider: textProvider)\n case .methodDefinition:\n return parseMethodDefinitionNode(node, textProvider: textProvider)\n case .functionDefinition:\n return parseFunctionDefinitionNode(node, textProvider: textProvider)\n case .structSpecifier, .enumSpecifier, .nsEnumSpecifier:\n return parseTypeSpecifierNode(node, textProvider: textProvider)\n case .typeDefinition:\n return parseTypedefNode(node, textProvider: textProvider)\n default:\n return nil\n }\n }\n\n func parseDeclarationInterfaceNode(\n _ node: ASTNode,\n textProvider: @escaping TextProvider\n ) -> NodeInfo? {\n var name = \"\"\n var category = \"\"\n /// Attributes, declaration kind, and name.\n var prefix = \"\"\n /// Generics, super class, etc.\n var extra = \"\"\n\n if let nameNode = node.child(byFieldName: \"name\") {\n name = textProvider(.node(nameNode))\n prefix = textProvider(.range(\n range: node.range.notSurpassing(nameNode.range),\n pointRange: node.pointRange.notSurpassing(nameNode.pointRange)\n ))\n }\n if let categoryNode = node.child(byFieldName: \"category\") {\n category = textProvider(.node(categoryNode))\n }\n\n for i in 0.. NodeInfo? {\n parseSignatureBeforeBody(node, fieldNameForName: \"selector\", textProvider: textProvider)\n }\n\n func parseTypeSpecifierNode(\n _ node: ASTNode,\n textProvider: @escaping TextProvider\n ) -> NodeInfo? {\n parseSignatureBeforeBody(node, textProvider: textProvider)\n }\n\n func parseTypedefNode(\n _ node: ASTNode,\n textProvider: @escaping TextProvider\n ) -> NodeInfo? {\n guard let typeNode = node.child(byFieldName: \"type\") else { return nil }\n guard var result = parseSignatureBeforeBody(typeNode, textProvider: textProvider)\n else { return nil }\n result.signature = \"typedef \\(result.signature)\"\n result.node = node\n return result\n }\n\n func parseFunctionDefinitionNode(\n _ node: ASTNode,\n textProvider: @escaping TextProvider\n ) -> NodeInfo? {\n let declaratorNode = node.child(byFieldName: \"declarator\")\n let name = declaratorNode?.contentOfChild(\n withFieldName: \"declarator\",\n textProvider: textProvider\n )\n let (\n _,\n signatureRange,\n signaturePointRange\n ) = node.extractInformationBeforeNode(withFieldName: \"body\")\n let signature = textProvider(.range(range: signatureRange, pointRange: signaturePointRange))\n .breakLines(proposedLineEnding: \" \", appendLineBreakToLastLine: false)\n .joined()\n .trimmingCharacters(in: .whitespacesAndNewlines)\n if signature.isEmpty { return nil }\n return .init(\n node: node,\n signature: signature,\n name: name ?? \"N/A\",\n canBeUsedAsCodeRange: true\n )\n }\n}\n\n// MARK: - Shared Parser\n\nextension ObjectiveCFocusedCodeFinder {\n func parseSignatureBeforeBody(\n _ node: ASTNode,\n fieldNameForName: String = \"name\",\n textProvider: @escaping TextProvider\n ) -> NodeInfo? {\n let name = node.contentOfChild(withFieldName: fieldNameForName, textProvider: textProvider)\n let (\n _,\n signatureRange,\n signaturePointRange\n ) = node.extractInformationBeforeNode(withFieldName: \"body\")\n let signature = textProvider(.range(range: signatureRange, pointRange: signaturePointRange))\n .breakLines(proposedLineEnding: \" \", appendLineBreakToLastLine: false)\n .joined()\n .trimmingCharacters(in: .whitespacesAndNewlines)\n if signature.isEmpty { return nil }\n return .init(\n node: node,\n signature: signature,\n name: name ?? \"N/A\",\n canBeUsedAsCodeRange: true\n )\n }\n}\n\nextension ASTNode {\n func contentOfChild(\n withFieldName name: String,\n textProvider: (TreeSitterTextPosition) -> String\n ) -> String? {\n guard let child = child(byFieldName: name) else { return nil }\n return textProvider(.node(child))\n }\n\n func extractInformationBeforeNode(withFieldName name: String) -> (\n postfixNode: ASTNode?,\n range: NSRange,\n pointRange: Range\n ) {\n guard let postfixNode = child(byFieldName: name) else {\n return (nil, range, pointRange)\n }\n\n let range = self.range.subtracting(postfixNode.range)\n let pointRange = self.pointRange.subtracting(postfixNode.pointRange)\n return (postfixNode, range, pointRange)\n }\n}\n\nextension NSRange {\n func subtracting(_ range: NSRange) -> NSRange {\n let start = lowerBound\n let end = Swift.max(lowerBound, Swift.min(upperBound, range.lowerBound))\n return NSRange(location: start, length: end - start)\n }\n\n func notSurpassing(_ range: NSRange) -> NSRange {\n let start = lowerBound\n let end = Swift.max(lowerBound, Swift.min(upperBound, range.upperBound))\n return NSRange(location: start, length: end - start)\n }\n}\n\nextension Range where Bound == Point {\n func subtracting(_ range: Range) -> Range {\n let start = lowerBound\n let end = if range.lowerBound >= upperBound {\n upperBound\n } else {\n Swift.max(range.lowerBound, lowerBound)\n }\n return Range(uncheckedBounds: (start, end))\n }\n\n func notSurpassing(_ range: Range) -> Range {\n let start = lowerBound\n let end = if range.lowerBound >= upperBound {\n upperBound\n } else {\n Swift.max(range.upperBound, lowerBound)\n }\n return Range(uncheckedBounds: (start, end))\n }\n}\n\n" }, { "path": "Tool/Sources/FocusedCodeFinder/ObjectiveC/ObjectiveCScopeHierarchySyntaxVisitor.swift", "content": "import ASTParser\nimport Foundation\nimport Preferences\nimport SuggestionBasic\nimport SwiftTreeSitter\n\nfinal class ObjectiveCScopeHierarchySyntaxVisitor: ASTTreeVisitor {\n let range: CursorRange\n let code: String\n let textProvider: (ASTNode) -> String\n var includes: [String] = []\n var imports: [String] = []\n private var _scopeHierarchy: [ASTNode] = []\n\n init(\n tree: ASTTree,\n code: String,\n textProvider: @escaping (ASTNode) -> String,\n range: CursorRange\n ) {\n self.range = range\n self.code = code\n self.textProvider = textProvider\n super.init(tree: tree)\n }\n\n /// The nodes containing the current range, sorted from inner to outer.\n func findScopeHierarchy(_ node: ASTNode) -> [ASTNode] {\n walk(node)\n return _scopeHierarchy.sorted { $0.range.location > $1.range.location }\n }\n\n /// The nodes containing the current range, sorted from inner to outer.\n func findScopeHierarchy() -> [ASTNode] {\n walk()\n return _scopeHierarchy.sorted { $0.range.location > $1.range.location }\n }\n\n override func visit(_ node: ASTNode) -> ASTTreeVisitorContinueKind {\n let cursorRange = CursorRange(pointRange: node.pointRange)\n\n switch ObjectiveCNodeType(rawValue: node.nodeType ?? \"\") {\n case .translationUnit:\n return .visitChildren\n case .preprocInclude:\n handlePreprocInclude(node)\n return .skipChildren\n case .preprocImport:\n handlePreprocImport(node)\n return .skipChildren\n case .moduleImport:\n handleModuleImport(node)\n return .skipChildren\n case .classInterface, .categoryInterface, .protocolDeclaration:\n guard cursorRange.strictlyContains(range) else { return .skipChildren }\n _scopeHierarchy.append(node)\n return .visitChildren\n case .classImplementation, .categoryImplementation:\n guard cursorRange.strictlyContains(range) else { return .skipChildren }\n _scopeHierarchy.append(node)\n return .visitChildren\n case .methodDefinition:\n guard cursorRange.strictlyContains(range) else { return .skipChildren }\n _scopeHierarchy.append(node)\n return .skipChildren\n case .typeDefinition:\n guard cursorRange.strictlyContains(range) else { return .skipChildren }\n _scopeHierarchy.append(node)\n return .skipChildren\n case .structSpecifier, .enumSpecifier, .nsEnumSpecifier:\n guard cursorRange.strictlyContains(range) else { return .skipChildren }\n _scopeHierarchy.append(node)\n return .skipChildren\n case .functionDefinition:\n guard cursorRange.strictlyContains(range) else { return .skipChildren }\n _scopeHierarchy.append(node)\n return .skipChildren\n default:\n return .skipChildren\n }\n }\n\n override func visitPost(_: ASTNode) {}\n\n // MARK: Imports\n\n func handlePreprocInclude(_ node: ASTNode) {\n if let pathNode = node.child(byFieldName: \"path\") {\n let path = textProvider(pathNode)\n if !path.isEmpty {\n includes.append(path)\n }\n }\n }\n\n func handlePreprocImport(_ node: ASTNode) {\n if let pathNode = node.child(byFieldName: \"path\") {\n let path = textProvider(pathNode)\n if !path.isEmpty {\n imports.append(path)\n }\n }\n }\n\n func handleModuleImport(_ node: ASTNode) {\n if let pathNode = node.child(byFieldName: \"module\") {\n let path = textProvider(pathNode)\n if !path.isEmpty {\n imports.append(path)\n }\n }\n }\n}\n\n" }, { "path": "Tool/Sources/FocusedCodeFinder/ObjectiveC/ObjectiveCSyntax.swift", "content": "import Foundation\n\n/// https://github.com/lukepistrol/tree-sitter-objc/blob/feature/spm/test/corpus/imports.txt\n/// https://github.com/lukepistrol/tree-sitter-objc/blob/feature/spm/test/corpus/expressions.txt\n/// https://github.com/lukepistrol/tree-sitter-objc/blob/feature/spm/test/corpus/declarations.txt\n/// https://github.com/lukepistrol/tree-sitter-objc/blob/feature/spm/node-types.json\n/// Some of the test cases are actually incorrect?\nenum ObjectiveCNodeType: String {\n /// The top most item\n case translationUnit = \"translation_unit\"\n /// `#include`\n case preprocInclude = \"preproc_include\"\n /// `#import \"bar.h\"`\n case preprocImport = \"preproc_import\"\n /// `@import foo.bar`\n case moduleImport = \"module_import\"\n /// ```objc\n /// @interface ClassName(Category): SuperClass {\n /// type1 iv1;\n /// type2 iv2;\n /// }\n /// @property (readwrite, copy) float value;\n /// + (tr)k1:(t1)a1 : (t2)a2 k2: a3;\n /// @end\n /// ```\n case classInterface = \"class_interface\"\n /// `@implementation`\n case classImplementation = \"class_implementation\"\n /// Similar to class interface.\n case categoryInterface = \"category_interface\"\n /// Similar to class implementation.\n case categoryImplementation = \"category_implementation\"\n /// Similar to class interface.\n case protocolDeclaration = \"protocol_declaration\"\n /// `@protocol `\n case protocolDeclarationList = \"protocol_declaration_list\"\n /// ```objc\n /// @class C1, C2;\n /// ```\n case classDeclarationList = \"class_declaration_list\"\n /// ```\n /// + (tr)k1: (t1)a1 : (t2)a2 k2: a3;\n /// ```\n case propertyDeclaration = \"property_declaration\"\n /// ```objc\n /// + (tr)k1: (t1)a1 : (t2)a2 k2: a3;\n /// ```\n case methodDeclaration = \"method_declaration\"\n /// `- (rt)sel {}`\n case methodDefinition = \"method_definition\"\n /// function definitions\n case functionDefinition = \"function_definition\"\n /// Names of symbols\n case identifier = \"identifier\"\n /// Type identifiers\n case typeIdentifier = \"type_identifier\"\n /// Compound statements, such as `{ ... }`\n case compoundStatement = \"compound_statement\"\n /// Typedef.\n case typeDefinition = \"type_definition\"\n /// `struct {}`.\n case structSpecifier = \"struct_specifier\"\n /// `enum {}`.\n case enumSpecifier = \"enum_specifier\"\n /// `NS_ENUM {}` and `NS_OPTIONS {}`.\n case nsEnumSpecifier = \"ns_enum_specifier\"\n /// Fields inside a type definition.\n case fieldDeclarationList = \"field_declaration_list\"\n /// Protocols that a type conforms.\n case protocolQualifiers = \"protocol_qualifiers\"\n /// Superclass of a type.\n case superclassReference = \"superclass_reference\"\n /// The generic type arguments.\n case parameterizedClassTypeArguments = \"parameterized_class_type_arguments\"\n /// `__GENERICS` in category interface and implementation.\n case genericsTypeReference = \"generics_type_reference\"\n /// `IB_DESIGNABLE`, etc. The typo is from the original source.\n case classInterfaceAttributeSpecifier = \"class_interface_attribute_specifier\"\n}\n\nextension ObjectiveCNodeType {\n init?(rawValue: String?) {\n self.init(rawValue: rawValue ?? \"\")\n }\n}\n" }, { "path": "Tool/Sources/FocusedCodeFinder/Swift/SwiftFocusedCodeFinder.swift", "content": "import ASTParser\nimport Foundation\nimport Preferences\nimport SuggestionBasic\nimport SwiftParser\nimport SwiftSyntax\n\npublic class SwiftFocusedCodeFinder: KnownLanguageFocusedCodeFinder<\n SourceFileSyntax,\n SyntaxProtocol,\n SyntaxProtocol\n> {\n override public init(maxFocusedCodeLineCount: Int) {\n super.init(maxFocusedCodeLineCount: maxFocusedCodeLineCount)\n }\n\n public func parseSyntaxTree(from document: Document) -> SourceFileSyntax? {\n Parser.parse(source: document.content)\n }\n\n public func collectContextNodes(\n in document: Document,\n tree: SourceFileSyntax,\n containingRange range: CursorRange,\n textProvider: @escaping TextProvider,\n rangeConverter: @escaping RangeConverter\n ) -> ContextInfo {\n let visitor = SwiftScopeHierarchySyntaxVisitor(\n tree: tree,\n code: document.content,\n range: range,\n rangeConverter: rangeConverter\n )\n\n let nodes = visitor.findScopeHierarchy()\n return .init(\n nodes: nodes,\n includes: [],\n imports: visitor.imports\n )\n }\n\n public func createTextProviderAndRangeConverter(\n for document: Document,\n tree: SourceFileSyntax\n ) -> (TextProvider, RangeConverter) {\n let locationConverter = SourceLocationConverter(\n fileName: document.documentURL.path,\n tree: tree\n )\n return (\n { node in\n let range = CursorRange(sourceRange: node.sourceRange(converter: locationConverter))\n return EditorInformation.code(in: document.lines, inside: range).code\n },\n { node in\n let range = CursorRange(sourceRange: node.sourceRange(converter: locationConverter))\n return range\n }\n )\n }\n\n public func contextContainingNode(\n _ node: SyntaxProtocol,\n textProvider: @escaping TextProvider,\n rangeConverter: @escaping RangeConverter\n ) -> NodeInfo? {\n func extractText(_ node: SyntaxProtocol) -> String {\n textProvider(node)\n }\n\n switch node {\n case let node as StructDeclSyntax:\n let type = node.structKeyword.text\n let name = node.name.text\n return .init(\n node: node,\n signature: \"\\(type) \\(name)\"\n .prefixedModifiers(node.modifierAndAttributeText(extractText))\n .suffixedInheritance(node.inheritanceClauseTexts(extractText))\n .split(omittingEmptySubsequences: false, whereSeparator: \\.isNewline)\n .joined(separator: \" \"),\n name: name\n )\n\n case let node as ClassDeclSyntax:\n let type = node.classKeyword.text\n let name = node.name.text\n return .init(\n node: node,\n signature: \"\\(type) \\(name)\"\n .prefixedModifiers(node.modifierAndAttributeText(extractText))\n .suffixedInheritance(node.inheritanceClauseTexts(extractText))\n .split(omittingEmptySubsequences: false, whereSeparator: \\.isNewline)\n .joined(separator: \" \"),\n name: name\n )\n\n case let node as EnumDeclSyntax:\n let type = node.enumKeyword.text\n let name = node.name.text\n return .init(\n node: node,\n signature: \"\\(type) \\(name)\"\n .prefixedModifiers(node.modifierAndAttributeText(extractText))\n .suffixedInheritance(node.inheritanceClauseTexts(extractText))\n .split(omittingEmptySubsequences: false, whereSeparator: \\.isNewline)\n .joined(separator: \" \"),\n name: name\n )\n\n case let node as ActorDeclSyntax:\n let type = node.actorKeyword.text\n let name = node.name.text\n return .init(\n node: node,\n signature: \"\\(type) \\(name)\"\n .prefixedModifiers(node.modifierAndAttributeText(extractText))\n .suffixedInheritance(node.inheritanceClauseTexts(extractText))\n .split(omittingEmptySubsequences: false, whereSeparator: \\.isNewline)\n .joined(separator: \" \"),\n name: name\n )\n\n case let node as MacroDeclSyntax:\n let type = node.macroKeyword.text\n let name = node.name.text\n return .init(\n node: node,\n signature: \"\\(type) \\(name)\"\n .prefixedModifiers(node.modifierAndAttributeText(extractText))\n .split(omittingEmptySubsequences: false, whereSeparator: \\.isNewline)\n .joined(separator: \" \"),\n name: name\n )\n\n case let node as ProtocolDeclSyntax:\n let type = node.protocolKeyword.text\n let name = node.name.text\n return .init(\n node: node,\n signature: \"\\(type) \\(name)\"\n .prefixedModifiers(node.modifierAndAttributeText(extractText))\n .suffixedInheritance(node.inheritanceClauseTexts(extractText))\n .split(omittingEmptySubsequences: false, whereSeparator: \\.isNewline)\n .joined(separator: \" \"),\n name: name\n )\n\n case let node as ExtensionDeclSyntax:\n let type = node.extensionKeyword.text\n let name = node.extendedType.trimmedDescription\n return .init(\n node: node,\n signature: \"\\(type) \\(name)\"\n .prefixedModifiers(node.modifierAndAttributeText(extractText))\n .suffixedInheritance(node.inheritanceClauseTexts(extractText))\n .split(omittingEmptySubsequences: false, whereSeparator: \\.isNewline)\n .joined(separator: \" \"),\n name: name\n )\n\n case let node as FunctionDeclSyntax:\n let type = node.funcKeyword.text\n let name = node.name.text\n let signature = node.signature.trimmedDescription\n .split(omittingEmptySubsequences: false, whereSeparator: \\.isNewline)\n .map { $0.trimmingCharacters(in: .whitespacesAndNewlines) }\n .joined(separator: \" \")\n\n return .init(\n node: node,\n signature: \"\\(type) \\(name)\\(signature)\"\n .prefixedModifiers(node.modifierAndAttributeText(extractText)),\n name: name\n )\n\n case let node as VariableDeclSyntax:\n let type = node.bindingSpecifier.trimmedDescription\n let name = node.bindings.first?.pattern.trimmedDescription ?? \"\"\n let signature = node.bindings.first?.typeAnnotation?.trimmedDescription ?? \"\"\n\n return .init(\n node: node,\n signature: \"\\(type) \\(name)\\(signature.isEmpty ? \"\" : \"\\(signature)\")\"\n .prefixedModifiers(node.modifierAndAttributeText(extractText))\n .split(omittingEmptySubsequences: false, whereSeparator: \\.isNewline)\n .joined(separator: \" \"),\n name: name,\n canBeUsedAsCodeRange: node.bindings.first?.accessorBlock != nil\n )\n\n case let node as AccessorDeclSyntax:\n let keyword = node.accessorSpecifier.text\n let signature = keyword\n\n return .init(\n node: node,\n signature: signature\n .prefixedModifiers(node.modifierAndAttributeText(extractText))\n .split(omittingEmptySubsequences: false, whereSeparator: \\.isNewline)\n .joined(separator: \" \"),\n name: keyword\n )\n\n case let node as SubscriptDeclSyntax:\n let genericPClause = node.genericWhereClause?.trimmedDescription ?? \"\"\n let pClause = node.parameterClause.trimmedDescription\n let whereClause = node.genericWhereClause?.trimmedDescription ?? \"\"\n let signature = \"subscript\\(genericPClause)(\\(pClause))\\(whereClause)\"\n\n return .init(\n node: node,\n signature: signature\n .prefixedModifiers(node.modifierAndAttributeText(extractText))\n .split(omittingEmptySubsequences: false, whereSeparator: \\.isNewline)\n .joined(separator: \" \"),\n name: \"subscript\"\n )\n\n case let node as InitializerDeclSyntax:\n let signature = \"init\"\n\n return .init(\n node: node,\n signature: \"\\(signature)\"\n .prefixedModifiers(node.modifierAndAttributeText(extractText))\n .split(omittingEmptySubsequences: false, whereSeparator: \\.isNewline)\n .joined(separator: \" \"),\n name: \"init\"\n )\n\n case let node as DeinitializerDeclSyntax:\n let signature = \"deinit\"\n\n return .init(\n node: node,\n signature: signature\n .prefixedModifiers(node.modifierAndAttributeText(extractText))\n .split(omittingEmptySubsequences: false, whereSeparator: \\.isNewline)\n .joined(separator: \" \"),\n name: \"deinit\"\n )\n\n case let node as ClosureExprSyntax:\n let signature = \"closure\"\n let range = rangeConverter(node)\n\n return .init(\n node: node,\n signature: signature\n .split(omittingEmptySubsequences: false, whereSeparator: \\.isNewline)\n .joined(separator: \" \"),\n name: \"closure\",\n canBeUsedAsCodeRange: range.lineCount > 80\n )\n\n case let node as FunctionCallExprSyntax:\n let signature = \"function call\"\n return .init(\n node: node,\n signature: signature\n .split(omittingEmptySubsequences: false, whereSeparator: \\.isNewline)\n .joined(separator: \" \"),\n name: \"function call\",\n canBeUsedAsCodeRange: false\n )\n\n case let node as SwitchCaseSyntax:\n let range = rangeConverter(node)\n \n return .init(\n node: node,\n signature: node.trimmedDescription\n .split(omittingEmptySubsequences: false, whereSeparator: \\.isNewline)\n .joined(separator: \" \"),\n name: \"switch\",\n canBeUsedAsCodeRange: range.lineCount > 80\n )\n\n default:\n return nil\n }\n }\n\n func findAssigningToVariable(_ node: SyntaxProtocol)\n -> (type: String, name: String, signature: String)?\n {\n if let node = node as? VariableDeclSyntax {\n let type = node.bindingSpecifier.trimmedDescription\n let name = node.bindings.first?.pattern.trimmedDescription ?? \"\"\n let sig = node.bindings.first?.initializer?.value.trimmedDescription ?? \"\"\n return (type, name, sig)\n }\n return nil\n }\n\n func findTypeNameFromNode(_ node: SyntaxProtocol) -> String? {\n switch node {\n case let node as ClassDeclSyntax:\n return node.name.text\n case let node as StructDeclSyntax:\n return node.name.text\n case let node as EnumDeclSyntax:\n return node.name.text\n case let node as ActorDeclSyntax:\n return node.name.text\n case let node as ProtocolDeclSyntax:\n return node.name.text\n case let node as ExtensionDeclSyntax:\n return node.extendedType.trimmedDescription\n default:\n return nil\n }\n }\n}\n\nextension CursorRange {\n init(sourceRange: SourceRange) {\n self.init(\n start: .init(line: sourceRange.start.line - 1, character: sourceRange.start.column - 1),\n end: .init(line: sourceRange.end.line - 1, character: sourceRange.end.column - 1)\n )\n }\n}\n\n// MARK: - Helper Types\n\nprotocol AttributeAndModifierApplicableSyntax {\n var attributes: AttributeListSyntax { get }\n var modifiers: DeclModifierListSyntax { get }\n}\n\nextension AttributeAndModifierApplicableSyntax {\n func modifierAndAttributeText(_ extractText: (SyntaxProtocol) -> String) -> String {\n let attributeTexts = attributes.map { attribute in\n extractText(attribute)\n }\n let modifierTexts = modifiers.map { modifier in\n extractText(modifier)\n }\n let prefix = (attributeTexts + modifierTexts).joined(separator: \" \")\n return prefix\n }\n}\n\nextension StructDeclSyntax: AttributeAndModifierApplicableSyntax {}\nextension ClassDeclSyntax: AttributeAndModifierApplicableSyntax {}\nextension EnumDeclSyntax: AttributeAndModifierApplicableSyntax {}\nextension ActorDeclSyntax: AttributeAndModifierApplicableSyntax {}\nextension MacroDeclSyntax: AttributeAndModifierApplicableSyntax {}\nextension MacroExpansionDeclSyntax: AttributeAndModifierApplicableSyntax {}\nextension ProtocolDeclSyntax: AttributeAndModifierApplicableSyntax {}\nextension ExtensionDeclSyntax: AttributeAndModifierApplicableSyntax {}\nextension FunctionDeclSyntax: AttributeAndModifierApplicableSyntax {}\nextension VariableDeclSyntax: AttributeAndModifierApplicableSyntax {}\nextension InitializerDeclSyntax: AttributeAndModifierApplicableSyntax {}\nextension DeinitializerDeclSyntax: AttributeAndModifierApplicableSyntax {}\nextension AccessorDeclSyntax: AttributeAndModifierApplicableSyntax {\n var modifiers: SwiftSyntax.DeclModifierListSyntax { [] }\n}\n\nextension SubscriptDeclSyntax: AttributeAndModifierApplicableSyntax {}\n\nprotocol InheritanceClauseApplicableSyntax {\n var inheritanceClause: InheritanceClauseSyntax? { get }\n}\n\nextension StructDeclSyntax: InheritanceClauseApplicableSyntax {}\nextension ClassDeclSyntax: InheritanceClauseApplicableSyntax {}\nextension EnumDeclSyntax: InheritanceClauseApplicableSyntax {}\nextension ActorDeclSyntax: InheritanceClauseApplicableSyntax {}\nextension ProtocolDeclSyntax: InheritanceClauseApplicableSyntax {}\nextension ExtensionDeclSyntax: InheritanceClauseApplicableSyntax {}\n\nextension InheritanceClauseApplicableSyntax {\n func inheritanceClauseTexts(_ extractText: (SyntaxProtocol) -> String) -> String {\n inheritanceClause?.inheritedTypes.map { clause in\n extractText(clause).trimmingCharacters(in: [\",\"])\n }.joined(separator: \", \") ?? \"\"\n }\n}\n\nextension String {\n func prefixedModifiers(_ text: String) -> String {\n if text.isEmpty {\n return self\n }\n return \"\\(text) \\(self)\"\n }\n\n func suffixedInheritance(_ text: String) -> String {\n if text.isEmpty {\n return self\n }\n return \"\\(self): \\(text)\"\n }\n}\n\n" }, { "path": "Tool/Sources/FocusedCodeFinder/Swift/SwiftScopeHierarchySyntaxVisitor.swift", "content": "import ASTParser\nimport Foundation\nimport Preferences\nimport SuggestionBasic\nimport SwiftParser\nimport SwiftSyntax\n\nfinal class SwiftScopeHierarchySyntaxVisitor: SyntaxVisitor {\n let tree: SyntaxProtocol\n let code: String\n let range: CursorRange\n let rangeConverter: (SyntaxProtocol) -> CursorRange\n\n var imports: [String] = []\n private var _scopeHierarchy: [SyntaxProtocol] = []\n\n /// The nodes containing the current range, sorted from inner to outer.\n func findScopeHierarchy(_ node: some SyntaxProtocol) -> [SyntaxProtocol] {\n walk(node)\n return _scopeHierarchy.sorted { $0.position.utf8Offset > $1.position.utf8Offset }\n }\n\n /// The nodes containing the current range, sorted from inner to outer.\n func findScopeHierarchy() -> [SyntaxProtocol] {\n walk(tree)\n return _scopeHierarchy.sorted { $0.position.utf8Offset > $1.position.utf8Offset }\n }\n\n init(\n tree: SyntaxProtocol,\n code: String,\n range: CursorRange,\n rangeConverter: @escaping (SyntaxProtocol) -> CursorRange\n ) {\n self.tree = tree\n self.code = code\n self.range = range\n self.rangeConverter = rangeConverter\n super.init(viewMode: .sourceAccurate)\n }\n\n func skipChildrenIfPossible(_ node: SyntaxProtocol) -> SyntaxVisitorContinueKind {\n if _scopeHierarchy.count > 5 { return .skipChildren }\n if !nodeContainsRange(node) { return .skipChildren }\n return .visitChildren\n }\n\n func captureNodeIfPossible(_ node: SyntaxProtocol) -> SyntaxVisitorContinueKind {\n if _scopeHierarchy.count > 5 { return .skipChildren }\n if !nodeContainsRange(node) { return .skipChildren }\n _scopeHierarchy.append(node)\n return .visitChildren\n }\n\n func nodeContainsRange(_ node: SyntaxProtocol) -> Bool {\n let cursorRange = rangeConverter(node)\n return cursorRange.strictlyContains(range)\n }\n\n // skip if possible\n\n override func visit(_ node: MemberBlockSyntax) -> SyntaxVisitorContinueKind {\n skipChildrenIfPossible(node)\n }\n\n override func visit(_ node: MemberBlockItemSyntax) -> SyntaxVisitorContinueKind {\n skipChildrenIfPossible(node)\n }\n\n // capture if possible\n\n override func visit(_ node: ImportDeclSyntax) -> SyntaxVisitorContinueKind {\n imports.append(node.path.trimmedDescription)\n return .skipChildren\n }\n\n override func visit(_ node: StructDeclSyntax) -> SyntaxVisitorContinueKind {\n captureNodeIfPossible(node)\n }\n\n override func visit(_ node: ClassDeclSyntax) -> SyntaxVisitorContinueKind {\n captureNodeIfPossible(node)\n }\n\n override func visit(_ node: EnumDeclSyntax) -> SyntaxVisitorContinueKind {\n captureNodeIfPossible(node)\n }\n\n override func visit(_ node: ActorDeclSyntax) -> SyntaxVisitorContinueKind {\n captureNodeIfPossible(node)\n }\n\n override func visit(_ node: MacroDeclSyntax) -> SyntaxVisitorContinueKind {\n captureNodeIfPossible(node)\n }\n\n override func visit(_ node: ProtocolDeclSyntax) -> SyntaxVisitorContinueKind {\n captureNodeIfPossible(node)\n }\n\n override func visit(_ node: ExtensionDeclSyntax) -> SyntaxVisitorContinueKind {\n captureNodeIfPossible(node)\n }\n\n override func visit(_ node: FunctionDeclSyntax) -> SyntaxVisitorContinueKind {\n captureNodeIfPossible(node)\n }\n\n override func visit(_ node: SubscriptDeclSyntax) -> SyntaxVisitorContinueKind {\n captureNodeIfPossible(node)\n }\n\n override func visit(_ node: ClosureExprSyntax) -> SyntaxVisitorContinueKind {\n captureNodeIfPossible(node)\n }\n\n override func visit(_ node: VariableDeclSyntax) -> SyntaxVisitorContinueKind {\n captureNodeIfPossible(node)\n }\n\n override func visit(_ node: AccessorDeclSyntax) -> SyntaxVisitorContinueKind {\n captureNodeIfPossible(node)\n }\n\n override func visit(_ node: InitializerDeclSyntax) -> SyntaxVisitorContinueKind {\n captureNodeIfPossible(node)\n }\n\n override func visit(_ node: DeinitializerDeclSyntax) -> SyntaxVisitorContinueKind {\n captureNodeIfPossible(node)\n }\n\n override func visit(_ node: FunctionCallExprSyntax) -> SyntaxVisitorContinueKind {\n captureNodeIfPossible(node)\n }\n\n override func visit(_ node: SwitchCaseSyntax) -> SyntaxVisitorContinueKind {\n captureNodeIfPossible(node)\n }\n}\n\n" }, { "path": "Tool/Sources/FocusedCodeFinder/UnknownLanguageFocusCodeFinder.swift", "content": "import Foundation\nimport Preferences\nimport SuggestionBasic\n\n/// Used when the language is not supported by the app\n/// or that the code is too long to be returned by a focused code finder.\npublic struct UnknownLanguageFocusedCodeFinder: FocusedCodeFinderType {\n let proposedSearchRange: Int\n\n public init(proposedSearchRange: Int) {\n self.proposedSearchRange = proposedSearchRange\n }\n\n public func findFocusedCode(\n in document: Document,\n containingRange: CursorRange\n ) -> CodeContext {\n guard !document.lines.isEmpty else { return .empty }\n\n // when user is not selecting any code.\n if containingRange.start == containingRange.end {\n // search up and down for up to `proposedSearchRange * 2 + 1` lines.\n let lines = document.lines\n let proposedLineCount = proposedSearchRange * 2 + 1\n let startLineIndex = max(containingRange.start.line - proposedSearchRange, 0)\n let endLineIndex = max(\n startLineIndex,\n min(startLineIndex + proposedLineCount - 1, lines.count - 1)\n )\n\n if lines.endIndex <= endLineIndex { return .empty }\n \n let focusedLines = lines[startLineIndex...endLineIndex]\n\n let contextStartLine = max(startLineIndex - 5, 0)\n let contextEndLine = min(endLineIndex + 5, lines.count - 1)\n \n let contextRange = CursorRange(\n start: .init(line: contextStartLine, character: 0),\n end: .init(line: contextEndLine, character: lines[contextEndLine].count)\n )\n\n return .init(\n scope: .top,\n contextRange: contextRange,\n smallestContextRange: contextRange,\n focusedRange: .init(\n start: .init(line: startLineIndex, character: 0),\n end: .init(line: endLineIndex, character: lines[endLineIndex].count)\n ),\n focusedCode: focusedLines.joined(),\n imports: [],\n includes: []\n )\n }\n\n let startLine = max(containingRange.start.line, 0)\n let endLine = min(containingRange.end.line, document.lines.count - 1)\n\n if endLine < startLine { return .empty }\n\n let focusedLines = document.lines[startLine...endLine]\n let contextStartLine = max(startLine - 3, 0)\n let contextEndLine = min(endLine + 3, document.lines.count - 1)\n \n let contextRange = CursorRange(\n start: .init(line: contextStartLine, character: 0),\n end: .init(\n line: contextEndLine,\n character: document.lines[contextEndLine].count\n )\n )\n\n return CodeContext(\n scope: .top,\n contextRange: contextRange,\n smallestContextRange: contextRange,\n focusedRange: containingRange,\n focusedCode: focusedLines.joined(),\n imports: [],\n includes: []\n )\n }\n}\n\n" }, { "path": "Tool/Sources/GitHubCopilotService/GitHubCopilotExtension.swift", "content": "import BuiltinExtension\nimport CopilotForXcodeKit\nimport Dependencies\nimport Foundation\nimport LanguageServerProtocol\nimport Logger\nimport Preferences\nimport Toast\nimport Workspace\n\npublic final class GitHubCopilotExtension: BuiltinExtension {\n public var extensionIdentifier: String { \"com.github.copilot\" }\n\n public let suggestionService: GitHubCopilotSuggestionService\n public let chatService: GitHubCopilotChatService\n\n private var extensionUsage = ExtensionUsage(\n isSuggestionServiceInUse: false,\n isChatServiceInUse: false\n )\n private var isLanguageServerInUse: Bool {\n extensionUsage.isSuggestionServiceInUse || extensionUsage.isChatServiceInUse\n }\n\n @Dependency(\\.toastController) var toast\n\n let workspacePool: WorkspacePool\n\n let serviceLocator: ServiceLocatorType\n\n public init(workspacePool: WorkspacePool) {\n self.workspacePool = workspacePool\n serviceLocator = ServiceLocator(workspacePool: workspacePool)\n suggestionService = .init(serviceLocator: serviceLocator)\n chatService = .init(serviceLocator: serviceLocator)\n }\n\n public func workspaceDidOpen(_: WorkspaceInfo) {}\n\n public func workspaceDidClose(_: WorkspaceInfo) {}\n\n public func workspace(_ workspace: WorkspaceInfo, didOpenDocumentAt documentURL: URL) {\n guard isLanguageServerInUse else { return }\n // check if file size is larger than 15MB, if so, return immediately\n if let attrs = try? FileManager.default\n .attributesOfItem(atPath: documentURL.path),\n let fileSize = attrs[FileAttributeKey.size] as? UInt64,\n fileSize > 15 * 1024 * 1024\n { return }\n\n Task {\n do {\n let content = try String(contentsOf: documentURL, encoding: .utf8)\n guard let service = await serviceLocator.getService(from: workspace) else { return }\n try await service.notifyOpenTextDocument(fileURL: documentURL, content: content)\n } catch let error as ServerError {\n let e = GitHubCopilotError.languageServerError(error)\n Logger.gitHubCopilot.error(e.localizedDescription)\n \n switch error {\n case .serverUnavailable, .serverError:\n toast.toast(content: e.localizedDescription, type: .error, duration: 10.0)\n default:\n break\n }\n } catch {\n Logger.gitHubCopilot.error(error.localizedDescription)\n }\n }\n }\n\n public func workspace(_ workspace: WorkspaceInfo, didSaveDocumentAt documentURL: URL) {\n guard isLanguageServerInUse else { return }\n Task {\n do {\n guard let service = await serviceLocator.getService(from: workspace) else { return }\n try await service.notifySaveTextDocument(fileURL: documentURL)\n } catch {\n Logger.gitHubCopilot.error(error.localizedDescription)\n }\n }\n }\n\n public func workspace(_ workspace: WorkspaceInfo, didCloseDocumentAt documentURL: URL) {\n guard isLanguageServerInUse else { return }\n Task {\n do {\n guard let service = await serviceLocator.getService(from: workspace) else { return }\n try await service.notifyCloseTextDocument(fileURL: documentURL)\n } catch {\n Logger.gitHubCopilot.error(error.localizedDescription)\n }\n }\n }\n\n public func workspace(\n _ workspace: WorkspaceInfo,\n didUpdateDocumentAt documentURL: URL,\n content: String?\n ) {\n guard isLanguageServerInUse else { return }\n // check if file size is larger than 15MB, if so, return immediately\n if let attrs = try? FileManager.default\n .attributesOfItem(atPath: documentURL.path),\n let fileSize = attrs[FileAttributeKey.size] as? UInt64,\n fileSize > 15 * 1024 * 1024\n { return }\n\n Task {\n guard let content else { return }\n guard let service = await serviceLocator.getService(from: workspace) else { return }\n do {\n try await service.notifyChangeTextDocument(\n fileURL: documentURL,\n content: content,\n version: 0\n )\n } catch let error as ServerError {\n switch error {\n case .serverError(-32602, _, _): // parameter incorrect\n Logger.gitHubCopilot.error(error.localizedDescription)\n // Reopen document if it's not found in the language server\n self.workspace(workspace, didOpenDocumentAt: documentURL)\n default:\n Logger.gitHubCopilot.error(error.localizedDescription)\n }\n } catch {\n Logger.gitHubCopilot.error(error.localizedDescription)\n }\n }\n }\n\n public func extensionUsageDidChange(_ usage: ExtensionUsage) {\n extensionUsage = usage\n if !usage.isChatServiceInUse && !usage.isSuggestionServiceInUse {\n terminate()\n }\n }\n\n public func terminate() {\n for workspace in workspacePool.workspaces.values {\n guard let plugin = workspace.plugin(for: GitHubCopilotWorkspacePlugin.self)\n else { continue }\n plugin.terminate()\n }\n }\n}\n\nprotocol ServiceLocatorType {\n func getService(from workspace: WorkspaceInfo) async -> GitHubCopilotService?\n}\n\nclass ServiceLocator: ServiceLocatorType {\n let workspacePool: WorkspacePool\n\n init(workspacePool: WorkspacePool) {\n self.workspacePool = workspacePool\n }\n\n func getService(from workspace: WorkspaceInfo) async -> GitHubCopilotService? {\n guard let workspace = workspacePool.workspaces[workspace.workspaceURL],\n let plugin = workspace.plugin(for: GitHubCopilotWorkspacePlugin.self)\n else { return nil }\n return await plugin.gitHubCopilotService\n }\n}\n\nextension GitHubCopilotExtension {\n public struct Token: Codable {\n// let codesearch: Bool\n public let individual: Bool\n public let endpoints: Endpoints\n public let chat_enabled: Bool\n// public let sku: String\n// public let copilotignore_enabled: Bool\n// public let limited_user_quotas: String?\n// public let tracking_id: String\n// public let xcode: Bool\n// public let limited_user_reset_date: String?\n// public let telemetry: String\n// public let prompt_8k: Bool\n public let token: String\n// public let nes_enabled: Bool\n// public let vsc_electron_fetcher_v2: Bool\n// public let code_review_enabled: Bool\n// public let annotations_enabled: Bool\n// public let chat_jetbrains_enabled: Bool\n// public let xcode_chat: Bool\n// public let refresh_in: Int\n// public let snippy_load_test_enabled: Bool\n// public let trigger_completion_after_accept: Bool\n public let expires_at: Int\n// public let public_suggestions: String\n// public let code_quote_enabled: Bool\n\n public struct Endpoints: Codable {\n public let api: String\n public let proxy: String\n public let telemetry: String\n// public let origin-tracker: String\n }\n }\n\n struct AuthInfo: Codable {\n public let user: String\n public let oauth_token: String\n public let githubAppId: String\n }\n\n static var authInfo: AuthInfo? {\n guard let urls = try? GitHubCopilotBaseService.createFoldersIfNeeded()\n else { return nil }\n let path = urls.supportURL\n .appendingPathComponent(\"undefined\")\n .appendingPathComponent(\".config\")\n .appendingPathComponent(\"github-copilot\")\n .appendingPathComponent(\"apps.json\").path\n guard FileManager.default.fileExists(atPath: path) else { return nil }\n\n do {\n let data = try Data(contentsOf: URL(fileURLWithPath: path))\n let json = try JSONSerialization\n .jsonObject(with: data, options: []) as? [String: [String: String]]\n guard let firstEntry = json?.values.first else { return nil }\n let jsonData = try JSONSerialization.data(withJSONObject: firstEntry, options: [])\n return try JSONDecoder().decode(AuthInfo.self, from: jsonData)\n } catch {\n Logger.gitHubCopilot.error(error.localizedDescription)\n return nil\n }\n }\n\n @MainActor\n static var cachedToken: Token?\n\n public static func fetchToken() async throws -> Token {\n guard let authToken = authInfo?.oauth_token\n else { throw GitHubCopilotError.notLoggedIn }\n\n let oldToken = await MainActor.run { cachedToken }\n if let oldToken {\n let expiresAt = Date(timeIntervalSince1970: TimeInterval(oldToken.expires_at))\n if expiresAt > Date() {\n return oldToken\n }\n }\n\n let url = URL(string: \"https://api.github.com/copilot_internal/v2/token\")!\n var request = URLRequest(url: url)\n request.httpMethod = \"GET\"\n request.setValue(\"token \\(authToken)\", forHTTPHeaderField: \"authorization\")\n request.setValue(\"unknown-editor/0\", forHTTPHeaderField: \"editor-version\")\n request.setValue(\"unknown-editor-plugin/0\", forHTTPHeaderField: \"editor-plugin-version\")\n request.setValue(\"1.236.0\", forHTTPHeaderField: \"copilot-language-server-version\")\n request.setValue(\"GithubCopilot/1.236.0\", forHTTPHeaderField: \"user-agent\")\n request.setValue(\"*/*\", forHTTPHeaderField: \"accept\")\n request.setValue(\"gzip,deflate,br\", forHTTPHeaderField: \"accept-encoding\")\n\n do {\n let (data, _) = try await URLSession.shared.data(for: request)\n if let jsonString = String(data: data, encoding: .utf8) {\n print(jsonString)\n }\n let newToken = try JSONDecoder().decode(Token.self, from: data)\n await MainActor.run { cachedToken = newToken }\n return newToken\n } catch {\n Logger.service.error(error.localizedDescription)\n throw error\n }\n }\n\n public static func fetchLLMModels() async throws -> [GitHubCopilotLLMModel] {\n let token = try await GitHubCopilotExtension.fetchToken()\n guard let endpoint = URL(string: token.endpoints.api + \"/models\") else {\n throw CancellationError()\n }\n var request = URLRequest(url: endpoint)\n request.setValue(\n \"Copilot for Xcode/\\(Bundle.main.infoDictionary?[\"CFBundleShortVersionString\"] as? String ?? \"unknown\")\",\n forHTTPHeaderField: \"Editor-Version\"\n )\n request.setValue(\"Bearer \\(token.token)\", forHTTPHeaderField: \"Authorization\")\n request.setValue(\"application/json\", forHTTPHeaderField: \"Content-Type\")\n request.setValue(\"vscode-chat\", forHTTPHeaderField: \"Copilot-Integration-Id\")\n request.setValue(\"2023-07-07\", forHTTPHeaderField: \"X-Github-Api-Version\")\n\n let (data, response) = try await URLSession.shared.data(for: request)\n\n guard let response = response as? HTTPURLResponse else {\n throw CancellationError()\n }\n\n guard response.statusCode == 200 else {\n throw CancellationError()\n }\n\n struct Model: Decodable {\n struct Limit: Decodable {\n var max_context_window_tokens: Int\n }\n\n struct Capability: Decodable {\n var type: String?\n var family: String?\n var limit: Limit?\n }\n\n var id: String\n var capabilities: Capability\n }\n\n struct Body: Decodable {\n var data: [Model]\n }\n\n let models = try JSONDecoder().decode(Body.self, from: data)\n .data\n .filter {\n $0.capabilities.type == \"chat\"\n }\n .map {\n GitHubCopilotLLMModel(\n modelId: $0.id,\n familyName: $0.capabilities.family ?? \"\",\n contextWindow: $0.capabilities.limit?.max_context_window_tokens ?? 0\n )\n }\n return models\n }\n}\n\n" }, { "path": "Tool/Sources/GitHubCopilotService/GitHubCopilotWorkspacePlugin.swift", "content": "import Dependencies\nimport Foundation\nimport Logger\nimport Toast\nimport Workspace\n\npublic final class GitHubCopilotWorkspacePlugin: WorkspacePlugin {\n enum Error: Swift.Error, LocalizedError {\n case gitHubCopilotLanguageServerMustBeUpdated\n var errorDescription: String? {\n switch self {\n case .gitHubCopilotLanguageServerMustBeUpdated:\n return \"GitHub Copilot language server must be updated. Update will start immediately. \\nIf it fails, please go to Host app > Service > GitHub Copilot and check if there is an update available.\"\n }\n }\n }\n\n @Dependency(\\.toast) var toast\n\n let installationManager = GitHubCopilotInstallationManager()\n private var _gitHubCopilotService: GitHubCopilotService?\n @GitHubCopilotSuggestionActor\n var gitHubCopilotService: GitHubCopilotService? {\n if let service = _gitHubCopilotService { return service }\n do {\n return try createGitHubCopilotService()\n } catch let error as Error {\n toast(error.localizedDescription, .warning)\n Task {\n await updateLanguageServerIfPossible()\n }\n return nil\n } catch {\n Logger.gitHubCopilot.error(\"Failed to create GitHub Copilot service: \\(error)\")\n toast(\n \"Failed to start GitHub Copilot language server: \\(error.localizedDescription)\",\n .error\n )\n return nil\n }\n }\n\n deinit {\n if let _gitHubCopilotService {\n Task { await _gitHubCopilotService.terminate() }\n }\n }\n\n @GitHubCopilotSuggestionActor\n func createGitHubCopilotService() throws -> GitHubCopilotService {\n if case .outdated(_, _, true) = installationManager.checkInstallation() {\n throw Error.gitHubCopilotLanguageServerMustBeUpdated\n }\n let newService = try GitHubCopilotService(projectRootURL: projectRootURL)\n _gitHubCopilotService = newService\n newService.localProcessServer?.terminationHandler = { [weak self] in\n Logger.gitHubCopilot.error(\"GitHub Copilot language server terminated\")\n self?.terminate()\n }\n Task {\n try await Task.sleep(nanoseconds: 1_000_000_000)\n finishLaunchingService()\n }\n return newService\n }\n\n @GitHubCopilotSuggestionActor\n func finishLaunchingService() {\n guard let workspace, let _gitHubCopilotService else { return }\n Task {\n for (_, filespace) in workspace.filespaces {\n let documentURL = filespace.fileURL\n guard let content = try? String(contentsOf: documentURL) else { continue }\n try? await _gitHubCopilotService.notifyOpenTextDocument(\n fileURL: documentURL,\n content: content\n )\n }\n }\n }\n\n @GitHubCopilotSuggestionActor\n func updateLanguageServerIfPossible() async {\n guard !GitHubCopilotInstallationManager.isInstalling else { return }\n let events = installationManager.installLatestVersion()\n do {\n for try await event in events {\n switch event {\n case .downloading:\n toast(\"Updating GitHub Copilot language server\", .info)\n case .uninstalling:\n break\n case .decompressing:\n break\n case .done:\n toast(\"Finished updating GitHub Copilot language server\", .info)\n }\n }\n } catch GitHubCopilotInstallationManager.Error.isInstalling {\n return\n } catch {\n toast(error.localizedDescription, .error)\n }\n }\n\n func terminate() {\n _gitHubCopilotService = nil\n }\n}\n\n" }, { "path": "Tool/Sources/GitHubCopilotService/LanguageServer/CopilotLocalProcessServer.swift", "content": "import Foundation\nimport JSONRPC\nimport LanguageClient\nimport LanguageServerProtocol\nimport Logger\nimport ProcessEnv\n\n/// A clone of the `LocalProcessServer`.\n/// We need it because the original one does not allow us to handle custom notifications.\nclass CopilotLocalProcessServer {\n private let transport: StdioDataTransport\n private let customTransport: CustomDataTransport\n private let process: Process\n private var wrappedServer: CustomJSONRPCLanguageServer?\n var terminationHandler: (() -> Void)?\n @MainActor var ongoingCompletionRequestIDs: [JSONId] = []\n @MainActor var ongoingConversationRequestIDs: [String: JSONId] = [:]\n\n public convenience init(\n path: String,\n arguments: [String],\n environment: [String: String]? = nil,\n serverNotificationHandler: ServerNotificationHandler\n ) {\n let params = Process.ExecutionParameters(\n path: path,\n arguments: arguments,\n environment: environment\n )\n\n self.init(executionParameters: params, serverNotificationHandler: serverNotificationHandler)\n }\n\n init(\n executionParameters parameters: Process.ExecutionParameters,\n serverNotificationHandler: ServerNotificationHandler\n ) {\n transport = StdioDataTransport()\n let framing = SeperatedHTTPHeaderMessageFraming()\n let messageTransport = MessageTransport(\n dataTransport: transport,\n messageProtocol: framing\n )\n customTransport = CustomDataTransport(nextTransport: messageTransport)\n wrappedServer = CustomJSONRPCLanguageServer(\n dataTransport: customTransport,\n serverNotificationHandler: serverNotificationHandler\n )\n\n process = Process()\n\n // Because the implementation of LanguageClient is so closed,\n // we need to get the request IDs from a custom transport before the data\n // is written to the language server.\n customTransport.onWriteRequest = { [weak self] request in\n if request.method == \"getCompletionsCycling\"\n || request.method == \"textDocument/inlineCompletion\"\n {\n Task { @MainActor [weak self] in\n self?.ongoingCompletionRequestIDs.append(request.id)\n }\n } else if request.method == \"conversation/create\" {\n Task { @MainActor [weak self] in\n if let paramsData = try? JSONEncoder().encode(request.params) {\n do {\n let params = try JSONDecoder().decode(\n GitHubCopilotRequest.ConversationCreate.RequestBody.self,\n from: paramsData\n )\n self?.ongoingConversationRequestIDs[params.workDoneToken] = request.id\n } catch {\n // Handle decoding error\n print(\"Error decoding ConversationCreateParams: \\(error)\")\n }\n }\n }\n }\n }\n\n process.standardInput = transport.stdinPipe\n process.standardOutput = transport.stdoutPipe\n process.standardError = transport.stderrPipe\n\n process.parameters = parameters\n\n process.terminationHandler = { [unowned self] task in\n self.processTerminated(task)\n }\n\n process.launch()\n }\n\n deinit {\n process.terminationHandler = nil\n process.terminate()\n transport.close()\n }\n\n private func processTerminated(_: Process) {\n transport.close()\n\n // releasing the server here will short-circuit any pending requests,\n // which might otherwise take a while to time out, if ever.\n wrappedServer = nil\n terminationHandler?()\n }\n\n var logMessages: Bool {\n get { return wrappedServer?.logMessages ?? false }\n set { wrappedServer?.logMessages = newValue }\n }\n\n func terminate() {\n process.terminate()\n }\n}\n\nextension CopilotLocalProcessServer: LanguageServerProtocol.Server {\n public var requestHandler: RequestHandler? {\n get { return wrappedServer?.requestHandler }\n set { wrappedServer?.requestHandler = newValue }\n }\n\n @available(*, deprecated, message: \"Use `ServerNotificationHandler` instead\")\n public var notificationHandler: NotificationHandler? {\n get { wrappedServer?.notificationHandler }\n set { wrappedServer?.notificationHandler = newValue }\n }\n\n public func sendNotification(\n _ notif: ClientNotification,\n completionHandler: @escaping (ServerError?) -> Void\n ) {\n guard let server = wrappedServer, process.isRunning else {\n completionHandler(.serverUnavailable)\n return\n }\n\n server.sendNotification(notif, completionHandler: completionHandler)\n }\n\n /// Cancel ongoing completion requests.\n public func cancelOngoingTasks() async {\n guard let _ = wrappedServer, process.isRunning else {\n return\n }\n\n let task = Task { @MainActor in\n for id in self.ongoingCompletionRequestIDs {\n await cancelTask(id)\n }\n self.ongoingCompletionRequestIDs = []\n }\n\n await task.value\n }\n\n public func cancelOngoingTask(workDoneToken: String) async {\n let task = Task { @MainActor in\n guard let id = ongoingConversationRequestIDs[workDoneToken] else { return }\n await cancelTask(id)\n }\n await task.value\n }\n\n public func cancelTask(_ id: JSONId) async {\n guard let server = wrappedServer, process.isRunning else {\n return\n }\n\n switch id {\n case let .numericId(id):\n try? await server.sendNotification(.protocolCancelRequest(.init(id: id)))\n case let .stringId(id):\n try? await server.sendNotification(.protocolCancelRequest(.init(id: id)))\n }\n }\n\n public func sendRequest(\n _ request: ClientRequest,\n completionHandler: @escaping (ServerResult) -> Void\n ) {\n guard let server = wrappedServer, process.isRunning else {\n completionHandler(.failure(.serverUnavailable))\n return\n }\n\n server.sendRequest(request, completionHandler: completionHandler)\n }\n}\n\nfinal class CustomJSONRPCLanguageServer: Server {\n let internalServer: JSONRPCLanguageServer\n\n typealias ProtocolResponse = ProtocolTransport.ResponseResult\n\n private let protocolTransport: ProtocolTransport\n\n var requestHandler: RequestHandler?\n var serverNotificationHandler: ServerNotificationHandler\n\n @available(*, deprecated, message: \"Use `serverNotificationHandler` instead.\")\n var notificationHandler: NotificationHandler? {\n get { nil }\n set {}\n }\n\n private var outOfBandError: Error?\n\n init(\n protocolTransport: ProtocolTransport,\n serverNotificationHandler: ServerNotificationHandler\n ) {\n self.serverNotificationHandler = serverNotificationHandler\n self.protocolTransport = protocolTransport\n internalServer = JSONRPCLanguageServer(protocolTransport: protocolTransport)\n\n let previousRequestHandler = protocolTransport.requestHandler\n\n protocolTransport.requestHandler = { [weak self] in\n guard let self else { return }\n if !self.handleRequest($0, data: $1, callback: $2) {\n previousRequestHandler?($0, $1, $2)\n }\n }\n protocolTransport.notificationHandler = { [weak self] in\n guard let self else { return }\n self.handleNotification($0, data: $1, block: $2)\n }\n }\n\n convenience init(\n dataTransport: DataTransport,\n serverNotificationHandler: ServerNotificationHandler\n ) {\n self.init(\n protocolTransport: ProtocolTransport(dataTransport: dataTransport),\n serverNotificationHandler: serverNotificationHandler\n )\n }\n\n deinit {\n protocolTransport.requestHandler = nil\n protocolTransport.notificationHandler = nil\n }\n\n var logMessages: Bool {\n get { return internalServer.logMessages }\n set { internalServer.logMessages = newValue }\n }\n}\n\nextension CustomJSONRPCLanguageServer {\n private func handleNotification(\n _ anyNotification: AnyJSONRPCNotification,\n data: Data,\n block: @escaping (Error?) -> Void\n ) {\n Task {\n do {\n try await serverNotificationHandler.handleNotification(\n anyNotification,\n data: data\n )\n block(nil)\n } catch {\n block(error)\n }\n }\n }\n\n func sendNotification(\n _ notif: ClientNotification,\n completionHandler: @escaping (ServerError?) -> Void\n ) {\n internalServer.sendNotification(notif, completionHandler: completionHandler)\n }\n}\n\nextension CustomJSONRPCLanguageServer {\n private func handleRequest(\n _ request: AnyJSONRPCRequest,\n data: Data,\n callback: @escaping (AnyJSONRPCResponse) -> Void\n ) -> Bool {\n return false\n }\n}\n\nextension CustomJSONRPCLanguageServer {\n public func sendRequest(\n _ request: ClientRequest,\n completionHandler: @escaping (ServerResult) -> Void\n ) {\n internalServer.sendRequest(request, completionHandler: completionHandler)\n }\n}\n\n@GitHubCopilotSuggestionActor\nfinal class ServerNotificationHandler {\n typealias Handler = (\n _ anyNotification: AnyJSONRPCNotification,\n _ data: Data\n ) async throws -> Bool\n\n var handlers = [AnyHashable: Handler]()\n nonisolated init() {}\n\n func handleNotification(\n _ anyNotification: AnyJSONRPCNotification,\n data: Data\n ) async throws {\n for handler in handlers.values {\n do {\n let handled = try await handler(anyNotification, data)\n if handled {\n return\n }\n } catch {\n throw ServerError.notificationDispatchFailed(error)\n }\n }\n\n let methodName = anyNotification.method\n let debugDescription = {\n if let params = anyNotification.params {\n let encoder = JSONEncoder()\n encoder.outputFormatting = .prettyPrinted\n if let jsonData = try? encoder.encode(params),\n let text = String(data: jsonData, encoding: .utf8)\n {\n return text\n }\n }\n return \"N/A\"\n }()\n switch methodName {\n case \"window/logMessage\":\n if UserDefaults.shared.value(for: \\.gitHubCopilotVerboseLog) {\n Logger.gitHubCopilot\n .info(\"\\(anyNotification.method): \\(debugDescription)\")\n }\n case \"LogMessage\":\n if UserDefaults.shared.value(for: \\.gitHubCopilotVerboseLog) {\n Logger.gitHubCopilot\n .info(\"\\(anyNotification.method): \\(debugDescription)\")\n }\n case \"statusNotification\":\n if UserDefaults.shared.value(for: \\.gitHubCopilotVerboseLog) {\n Logger.gitHubCopilot\n .info(\"\\(anyNotification.method): \\(debugDescription)\")\n }\n case \"featureFlagsNotification\":\n if UserDefaults.shared.value(for: \\.gitHubCopilotVerboseLog) {\n Logger.gitHubCopilot\n .info(\"\\(anyNotification.method): \\(debugDescription)\")\n }\n case \"conversation/preconditionsNotification\":\n if UserDefaults.shared.value(for: \\.gitHubCopilotVerboseLog) {\n Logger.gitHubCopilot\n .info(\"\\(anyNotification.method): \\(debugDescription)\")\n }\n case \"didChangeStatus\":\n Logger.gitHubCopilot.info(\"Did change status: \\(debugDescription)\")\n default:\n throw ServerError.handlerUnavailable(methodName)\n }\n }\n}\n\n" }, { "path": "Tool/Sources/GitHubCopilotService/LanguageServer/CustomStdioTransport.swift", "content": "import Foundation\nimport JSONRPC\nimport os.log\n\npublic class CustomDataTransport: DataTransport {\n let nextTransport: DataTransport\n \n var onWriteRequest: (JSONRPCRequest) -> Void = { _ in }\n \n init(nextTransport: DataTransport) {\n self.nextTransport = nextTransport\n }\n \n public func write(_ data: Data) {\n if let request = try? JSONDecoder().decode(JSONRPCRequest.self, from: data) {\n onWriteRequest(request)\n }\n \n nextTransport.write(data)\n }\n \n public func setReaderHandler(_ handler: @escaping ReadHandler) {\n nextTransport.setReaderHandler(handler)\n }\n \n public func close() {\n nextTransport.close()\n }\n}\n\n" }, { "path": "Tool/Sources/GitHubCopilotService/LanguageServer/GitHubCopilotAccountStatus.swift", "content": "import Foundation\n\npublic enum GitHubCopilotAccountStatus: String, Codable, CustomStringConvertible {\n case alreadySignedIn = \"AlreadySignedIn\"\n case maybeOk = \"MaybeOk\"\n case notAuthorized = \"NotAuthorized\"\n case notSignedIn = \"NotSignedIn\"\n case ok = \"OK\"\n\n public var description: String {\n switch self {\n case .alreadySignedIn:\n return \"Already Signed In\"\n case .maybeOk:\n return \"Maybe OK\"\n case .notAuthorized:\n return \"Not Authorized\"\n case .notSignedIn:\n return \"Not Signed In\"\n case .ok:\n return \"OK\"\n }\n }\n}\n" }, { "path": "Tool/Sources/GitHubCopilotService/LanguageServer/GitHubCopilotInstallationManager.swift", "content": "import Foundation\nimport Terminal\n\npublic struct GitHubCopilotInstallationManager {\n @GitHubCopilotSuggestionActor\n public private(set) static var isInstalling = false\n\n static var downloadURL: URL {\n let commitHash = \"f89e977c87180519ba3b942200e3d05b17b1e2fc\"\n let link = \"https://github.com/github/copilot.vim/archive/\\(commitHash).zip\"\n return URL(string: link)!\n }\n\n /// The GitHub's version has quite a lot of changes about `watchedFiles` since the following\n /// commit.\n /// https://github.com/github/CopilotForXcode/commit/a50045aa3ab3b7d532cadf40c4c10bed32f81169#diff-678798cf677bcd1ce276809cfccd33da9ff594b1b0c557180210a4ed2bd27ffa\n static let latestSupportedVersion = \"1.57.0\"\n static let minimumSupportedVersion = \"1.32.0\"\n\n public init() {}\n\n public enum InstallationStatus {\n case notInstalled\n case installed(String)\n case outdated(current: String, latest: String, mandatory: Bool)\n case unsupported(current: String, latest: String)\n }\n\n public func checkInstallation() -> InstallationStatus {\n guard let urls = try? GitHubCopilotBaseService.createFoldersIfNeeded()\n else { return .notInstalled }\n let executableFolderURL = urls.executableURL\n let binaryURL = executableFolderURL.appendingPathComponent(\"copilot\")\n let versionFileURL = executableFolderURL.appendingPathComponent(\"version\")\n\n if !FileManager.default.fileExists(atPath: binaryURL.path) {\n return .notInstalled\n }\n\n if FileManager.default.fileExists(atPath: versionFileURL.path),\n let versionData = try? Data(contentsOf: versionFileURL),\n let version = String(data: versionData, encoding: .utf8)\n {\n switch version.compare(Self.latestSupportedVersion, options: .numeric) {\n case .orderedAscending:\n switch version.compare(Self.minimumSupportedVersion) {\n case .orderedAscending:\n return .outdated(\n current: version,\n latest: Self.latestSupportedVersion,\n mandatory: true\n )\n case .orderedSame:\n return .outdated(\n current: version,\n latest: Self.latestSupportedVersion,\n mandatory: false\n )\n case .orderedDescending:\n return .outdated(\n current: version,\n latest: Self.latestSupportedVersion,\n mandatory: false\n )\n }\n case .orderedSame:\n return .installed(version)\n case .orderedDescending:\n return .unsupported(current: version, latest: Self.latestSupportedVersion)\n }\n }\n\n return .outdated(current: \"Unknown\", latest: Self.latestSupportedVersion, mandatory: false)\n }\n\n public enum InstallationStep {\n case downloading\n case uninstalling\n case decompressing\n case done\n }\n\n public enum Error: Swift.Error, LocalizedError {\n case isInstalling\n case failedToFindLanguageServer\n case failedToInstallLanguageServer\n\n public var errorDescription: String? {\n switch self {\n case .isInstalling:\n return \"Language server is installing.\"\n case .failedToFindLanguageServer:\n return \"Failed to find language server. Please open an issue on GitHub.\"\n case .failedToInstallLanguageServer:\n return \"Failed to install language server. Please open an issue on GitHub.\"\n }\n }\n }\n\n public func installLatestVersion() -> AsyncThrowingStream {\n AsyncThrowingStream { continuation in\n Task { @GitHubCopilotSuggestionActor in\n guard !GitHubCopilotInstallationManager.isInstalling else {\n continuation.finish(throwing: Error.isInstalling)\n return\n }\n GitHubCopilotInstallationManager.isInstalling = true\n defer { GitHubCopilotInstallationManager.isInstalling = false }\n do {\n continuation.yield(.downloading)\n let urls = try GitHubCopilotBaseService.createFoldersIfNeeded()\n\n // download\n let (fileURL, _) = try await URLSession.shared.download(from: Self.downloadURL)\n let targetURL = urls.executableURL.appendingPathComponent(\"archive\")\n .appendingPathExtension(\"zip\")\n try FileManager.default.copyItem(at: fileURL, to: targetURL)\n defer { try? FileManager.default.removeItem(at: targetURL) }\n\n // uninstall\n continuation.yield(.uninstalling)\n try await uninstall()\n\n // decompress\n continuation.yield(.decompressing)\n let terminal = Terminal()\n\n _ = try await terminal.runCommand(\n \"/usr/bin/unzip\",\n arguments: [targetURL.path],\n currentDirectoryURL: urls.executableURL,\n environment: [:]\n )\n\n let contentURLs = try FileManager.default.contentsOfDirectory(\n at: urls.executableURL,\n includingPropertiesForKeys: nil,\n options: []\n )\n\n defer {\n for url in contentURLs {\n try? FileManager.default.removeItem(at: url)\n }\n }\n\n guard let gitFolderURL = contentURLs\n .first(where: { $0.lastPathComponent.hasPrefix(\"copilot.vim\") })\n else {\n continuation.finish(throwing: Error.failedToInstallLanguageServer)\n return\n }\n\n let lspURL = {\n let caseA = gitFolderURL.appendingPathComponent(\"dist\")\n if FileManager.default.fileExists(atPath: caseA.path) {\n return caseA\n }\n return gitFolderURL\n .appendingPathComponent(\"copilot-language-server\")\n .appendingPathComponent(\"dist\")\n }()\n let copilotURL = urls.executableURL.appendingPathComponent(\"copilot\")\n\n if !FileManager.default.fileExists(atPath: copilotURL.path) {\n try FileManager.default.createDirectory(\n at: copilotURL,\n withIntermediateDirectories: true,\n attributes: nil\n )\n }\n\n let installationURL = copilotURL.appendingPathComponent(\"dist\")\n try FileManager.default.copyItem(at: lspURL, to: installationURL)\n\n // update permission 755\n try FileManager.default.setAttributes(\n [.posixPermissions: 0o755],\n ofItemAtPath: installationURL.path\n )\n\n // create version file\n let data = Self.latestSupportedVersion.data(using: .utf8)\n FileManager.default.createFile(\n atPath: urls.executableURL.appendingPathComponent(\"version\").path,\n contents: data\n )\n\n continuation.yield(.done)\n continuation.finish()\n } catch {\n continuation.finish(throwing: error)\n }\n }\n }\n }\n\n public func uninstall() async throws {\n guard let urls = try? GitHubCopilotBaseService.createFoldersIfNeeded()\n else { return }\n let executableFolderURL = urls.executableURL\n let binaryURL = executableFolderURL.appendingPathComponent(\"copilot\")\n let versionFileURL = executableFolderURL.appendingPathComponent(\"version\")\n if FileManager.default.fileExists(atPath: binaryURL.path) {\n try FileManager.default.removeItem(at: binaryURL)\n }\n if FileManager.default.fileExists(atPath: versionFileURL.path) {\n try FileManager.default.removeItem(at: versionFileURL)\n }\n }\n}\n\n" }, { "path": "Tool/Sources/GitHubCopilotService/LanguageServer/GitHubCopilotRequest.swift", "content": "import Foundation\nimport JSONRPC\nimport LanguageServerProtocol\nimport SuggestionBasic\nimport XcodeInspector\n\nstruct GitHubCopilotDoc: Codable {\n var source: String\n var tabSize: Int\n var indentSize: Int\n var insertSpaces: Bool\n var path: String\n var uri: String\n var relativePath: String\n var languageId: CodeLanguage\n var position: Position\n var version: Int = 0\n}\n\nprotocol GitHubCopilotRequestType {\n associatedtype Response: Codable\n var request: ClientRequest { get }\n}\n\npublic struct GitHubCopilotCodeSuggestion: Codable, Equatable {\n public init(\n text: String,\n position: CursorPosition,\n uuid: String,\n range: CursorRange,\n displayText: String\n ) {\n self.text = text\n self.position = position\n self.uuid = uuid\n self.range = range\n self.displayText = displayText\n }\n\n /// The new code to be inserted and the original code on the first line.\n public var text: String\n /// The position of the cursor before generating the completion.\n public var position: CursorPosition\n /// An id.\n public var uuid: String\n /// The range of the original code that should be replaced.\n public var range: CursorRange\n /// The new code to be inserted.\n public var displayText: String\n}\n\nenum GitHubCopilotChatSource: String, Codable {\n case panel\n case inline\n}\n\nenum GitHubCopilotRequest {\n struct SetEditorInfo: GitHubCopilotRequestType {\n let xcodeVersion: String\n\n struct Response: Codable {}\n\n var networkProxy: JSONValue? {\n let host = UserDefaults.shared.value(for: \\.gitHubCopilotProxyHost)\n if host.isEmpty { return nil }\n var port = UserDefaults.shared.value(for: \\.gitHubCopilotProxyPort)\n if port.isEmpty { port = \"80\" }\n let username = UserDefaults.shared.value(for: \\.gitHubCopilotProxyUsername)\n if username.isEmpty {\n return .hash([\n \"host\": .string(host),\n \"port\": .number(Double(Int(port) ?? 80)),\n \"rejectUnauthorized\": .bool(UserDefaults.shared\n .value(for: \\.gitHubCopilotUseStrictSSL)),\n ])\n } else {\n return .hash([\n \"host\": .string(host),\n \"port\": .number(Double(Int(port) ?? 80)),\n \"rejectUnauthorized\": .bool(UserDefaults.shared\n .value(for: \\.gitHubCopilotUseStrictSSL)),\n \"username\": .string(username),\n \"password\": .string(UserDefaults.shared\n .value(for: \\.gitHubCopilotProxyPassword)),\n\n ])\n }\n }\n\n var http: JSONValue? {\n var dict: [String: JSONValue] = [:]\n let host = UserDefaults.shared.value(for: \\.gitHubCopilotProxyHost)\n if host.isEmpty { return nil }\n var port = UserDefaults.shared.value(for: \\.gitHubCopilotProxyPort)\n if port.isEmpty { port = \"80\" }\n let username = UserDefaults.shared.value(for: \\.gitHubCopilotProxyUsername)\n let password = UserDefaults.shared.value(for: \\.gitHubCopilotProxyPassword)\n let strictSSL = UserDefaults.shared.value(for: \\.gitHubCopilotUseStrictSSL)\n\n let url = if !username.isEmpty {\n \"http://\\(username):\\(password)@\\(host):\\(port)\"\n } else {\n \"http://\\(host):\\(port)\"\n }\n\n dict[\"proxy\"] = .string(url)\n dict[\"proxyStrictSSL\"] = .bool(strictSSL)\n\n if dict.isEmpty { return nil }\n\n return .hash(dict)\n }\n\n var editorConfiguration: JSONValue? {\n var dict: [String: JSONValue] = [:]\n dict[\"http\"] = http\n\n let enterpriseURI = UserDefaults.shared.value(for: \\.gitHubCopilotEnterpriseURI)\n if !enterpriseURI.isEmpty {\n dict[\"github-enterprise\"] = .hash([\n \"uri\": .string(enterpriseURI),\n ])\n }\n\n if dict.isEmpty { return nil }\n return .hash(dict)\n }\n\n var authProvider: JSONValue? {\n var dict: [String: JSONValue] = [:]\n let enterpriseURI = UserDefaults.shared.value(for: \\.gitHubCopilotEnterpriseURI)\n if !enterpriseURI.isEmpty {\n dict[\"url\"] = .string(enterpriseURI)\n }\n\n if dict.isEmpty { return nil }\n return .hash(dict)\n }\n\n var request: ClientRequest {\n let pretendToBeVSCode = UserDefaults.shared\n .value(for: \\.gitHubCopilotPretendIDEToBeVSCode)\n var dict: [String: JSONValue] = [\n \"editorInfo\": pretendToBeVSCode ? .hash([\n \"name\": \"vscode\",\n \"version\": \"1.99.3\",\n ]) : .hash([\n \"name\": \"Xcode\",\n \"version\": .string(xcodeVersion),\n ]),\n \"editorPluginInfo\": .hash([\n \"name\": \"Copilot for Xcode\",\n \"version\": .string(Bundle.main\n .infoDictionary?[\"CFBundleShortVersionString\"] as? String ?? \"\"),\n ]),\n ]\n\n dict[\"editorConfiguration\"] = editorConfiguration\n dict[\"authProvider\"] = authProvider\n dict[\"networkProxy\"] = networkProxy\n\n return .custom(\"setEditorInfo\", .hash(dict))\n }\n }\n\n struct GetVersion: GitHubCopilotRequestType {\n struct Response: Codable {\n var version: String\n }\n\n var request: ClientRequest {\n .custom(\"getVersion\", .hash([:]))\n }\n }\n\n struct CheckStatus: GitHubCopilotRequestType {\n struct Response: Codable {\n var status: GitHubCopilotAccountStatus\n }\n\n var request: ClientRequest {\n .custom(\"checkStatus\", .hash([:]))\n }\n }\n\n struct SignInInitiate: GitHubCopilotRequestType {\n struct Response: Codable {\n var verificationUri: String\n var status: String\n var userCode: String\n var expiresIn: Int\n var interval: Int\n }\n\n var request: ClientRequest {\n .custom(\"signInInitiate\", .hash([:]))\n }\n }\n\n struct SignInConfirm: GitHubCopilotRequestType {\n struct Response: Codable {\n var status: GitHubCopilotAccountStatus\n var user: String\n }\n\n var userCode: String\n\n var request: ClientRequest {\n .custom(\"signInConfirm\", .hash([\n \"userCode\": .string(userCode),\n ]))\n }\n }\n\n struct SignOut: GitHubCopilotRequestType {\n struct Response: Codable {\n var status: GitHubCopilotAccountStatus\n }\n\n var request: ClientRequest {\n .custom(\"signOut\", .hash([:]))\n }\n }\n\n struct GetCompletions: GitHubCopilotRequestType {\n struct Response: Codable {\n var completions: [GitHubCopilotCodeSuggestion]\n }\n\n var doc: GitHubCopilotDoc\n\n var request: ClientRequest {\n let data = (try? JSONEncoder().encode(doc)) ?? Data()\n let dict = (try? JSONDecoder().decode(JSONValue.self, from: data)) ?? .hash([:])\n return .custom(\"getCompletions\", .hash([\n \"doc\": dict,\n ]))\n }\n }\n\n struct GetCompletionsCycling: GitHubCopilotRequestType {\n struct Response: Codable {\n var completions: [GitHubCopilotCodeSuggestion]\n }\n\n var doc: GitHubCopilotDoc\n\n var request: ClientRequest {\n let data = (try? JSONEncoder().encode(doc)) ?? Data()\n let dict = (try? JSONDecoder().decode(JSONValue.self, from: data)) ?? .hash([:])\n return .custom(\"getCompletionsCycling\", .hash([\n \"doc\": dict,\n ]))\n }\n }\n\n struct InlineCompletion: GitHubCopilotRequestType {\n struct Response: Codable {\n var items: [InlineCompletionItem]\n }\n\n struct InlineCompletionItem: Codable {\n var insertText: String\n var filterText: String?\n var range: Range?\n var command: Command?\n\n struct Range: Codable {\n var start: Position\n var end: Position\n }\n\n struct Command: Codable {\n var title: String\n var command: String\n var arguments: [String]?\n }\n }\n\n var doc: Input\n\n struct Input: Codable {\n var textDocument: _TextDocument; struct _TextDocument: Codable {\n var uri: String\n var version: Int\n }\n\n var position: Position\n var formattingOptions: FormattingOptions\n var context: _Context; struct _Context: Codable {\n enum TriggerKind: Int, Codable {\n case invoked = 1\n case automatic = 2\n }\n\n var triggerKind: TriggerKind\n }\n }\n\n var request: ClientRequest {\n let data = (try? JSONEncoder().encode(doc)) ?? Data()\n let dict = (try? JSONDecoder().decode(JSONValue.self, from: data)) ?? .hash([:])\n return .custom(\"textDocument/inlineCompletion\", dict)\n }\n }\n\n struct GetPanelCompletions: GitHubCopilotRequestType {\n struct Response: Codable {\n var completions: [GitHubCopilotCodeSuggestion]\n }\n\n var doc: GitHubCopilotDoc\n\n var request: ClientRequest {\n let data = (try? JSONEncoder().encode(doc)) ?? Data()\n let dict = (try? JSONDecoder().decode(JSONValue.self, from: data)) ?? .hash([:])\n return .custom(\"getPanelCompletions\", .hash([\n \"doc\": dict,\n ]))\n }\n }\n\n struct NotifyAccepted: GitHubCopilotRequestType {\n struct Response: Codable {}\n\n var completionUUID: String\n\n var request: ClientRequest {\n .custom(\"notifyAccepted\", .hash([\n \"uuid\": .string(completionUUID),\n ]))\n }\n }\n\n struct NotifyRejected: GitHubCopilotRequestType {\n struct Response: Codable {}\n\n var completionUUIDs: [String]\n\n var request: ClientRequest {\n .custom(\"notifyRejected\", .hash([\n \"uuids\": .array(completionUUIDs.map(JSONValue.string)),\n ]))\n }\n }\n\n struct ConversationCreate: GitHubCopilotRequestType {\n struct Response: Codable {\n var conversationId: String\n var turnId: String\n }\n\n struct RequestBody: Codable {\n public struct Reference: Codable, Equatable, Hashable {\n public var type: String = \"file\"\n public let uri: String\n public let position: Position?\n public let visibleRange: SuggestionBasic.CursorRange?\n public let selection: SuggestionBasic.CursorRange?\n public let openedAt: String?\n public let activeAt: String?\n }\n\n enum ConversationSource: String, Codable {\n case panel, inline\n }\n\n enum ConversationMode: String, Codable {\n case agent = \"Agent\"\n }\n\n struct ConversationTurn: Codable {\n var request: String\n var response: String?\n var turnId: String?\n }\n\n var workDoneToken: String\n var turns: [ConversationTurn]\n var capabilities: Capabilities\n var textDocument: GitHubCopilotDoc?\n var references: [Reference]?\n var computeSuggestions: Bool?\n var source: ConversationSource?\n var workspaceFolder: String?\n var workspaceFolders: [WorkspaceFolder]?\n var ignoredSkills: [String]?\n var model: String?\n var chatMode: ConversationMode?\n var userLanguage: String?\n\n struct Capabilities: Codable {\n var skills: [String]\n var allSkills: Bool?\n }\n }\n\n let requestBody: RequestBody\n\n var request: ClientRequest {\n let data = (try? JSONEncoder().encode(requestBody)) ?? Data()\n let dict = (try? JSONDecoder().decode(JSONValue.self, from: data)) ?? .hash([:])\n return .custom(\"conversation/create\", dict)\n }\n }\n\n struct ConversationTurn: GitHubCopilotRequestType {\n struct Response: Codable {}\n\n struct RequestBody: Codable {\n var workDoneToken: String\n var conversationId: String\n var message: String\n var textDocument: GitHubCopilotDoc?\n var ignoredSkills: [String]?\n var references: [ConversationCreate.RequestBody.Reference]?\n var model: String?\n var workspaceFolder: String?\n var workspaceFolders: [WorkspaceFolder]?\n var chatMode: String?\n }\n\n let requestBody: RequestBody\n\n var request: ClientRequest {\n let data = (try? JSONEncoder().encode(requestBody)) ?? Data()\n let dict = (try? JSONDecoder().decode(JSONValue.self, from: data)) ?? .hash([:])\n return .custom(\"conversation/turn\", dict)\n }\n }\n\n struct ConversationTurnDelete: GitHubCopilotRequestType {\n struct Response: Codable {}\n\n struct RequestBody: Codable {\n var conversationId: String\n var turnId: String\n var options: [String: String]?\n var source: GitHubCopilotChatSource?\n }\n\n let requestBody: RequestBody\n\n var request: ClientRequest {\n let data = (try? JSONEncoder().encode(requestBody)) ?? Data()\n let dict = (try? JSONDecoder().decode(JSONValue.self, from: data)) ?? .hash([:])\n return .custom(\"conversation/turnDelete\", dict)\n }\n }\n\n struct ConversationDestroy: GitHubCopilotRequestType {\n struct Response: Codable {}\n\n struct RequestBody: Codable {\n var conversationId: String\n var options: [String: String]?\n }\n\n let requestBody: RequestBody\n\n var request: ClientRequest {\n let data = (try? JSONEncoder().encode(requestBody)) ?? Data()\n let dict = (try? JSONDecoder().decode(JSONValue.self, from: data)) ?? .hash([:])\n return .custom(\"conversation/destroy\", dict)\n }\n }\n\n struct CopilotModels: GitHubCopilotRequestType {\n typealias Response = [GitHubCopilotModel]\n\n var request: ClientRequest {\n .custom(\"copilot/models\", .hash([:]))\n }\n }\n}\n\npublic struct GitHubCopilotModel: Codable, Equatable {\n public let modelFamily: String\n public let modelName: String\n public let id: String\n// public let modelPolicy: CopilotModelPolicy?\n public let scopes: [GitHubCopilotPromptTemplateScope]\n public let preview: Bool\n public let isChatDefault: Bool\n public let isChatFallback: Bool\n// public let capabilities: CopilotModelCapabilities\n// public let billing: CopilotModelBilling?\n}\n\npublic struct GitHubCopilotLLMModel: Equatable, Decodable, Identifiable {\n public var id: String { modelId }\n public var modelId: String\n public var familyName: String\n public var contextWindow: Int\n}\n\npublic enum GitHubCopilotPromptTemplateScope: String, Codable, Equatable {\n case chatPanel = \"chat-panel\"\n case editPanel = \"edit-panel\"\n case agentPanel = \"agent-panel\"\n case editor\n case inline\n case completion\n}\n\n" }, { "path": "Tool/Sources/GitHubCopilotService/LanguageServer/GitHubCopilotService.swift", "content": "import AppKit\nimport enum CopilotForXcodeKit.SuggestionServiceError\nimport Foundation\nimport JSONRPC\nimport LanguageClient\nimport LanguageServerProtocol\nimport Logger\nimport Preferences\nimport SuggestionBasic\nimport XcodeInspector\n\npublic protocol GitHubCopilotAuthServiceType {\n func checkStatus() async throws -> GitHubCopilotAccountStatus\n func signInInitiate() async throws -> (verificationUri: String, userCode: String)\n func signInConfirm(userCode: String) async throws\n -> (username: String, status: GitHubCopilotAccountStatus)\n func signOut() async throws -> GitHubCopilotAccountStatus\n func version() async throws -> String\n}\n\npublic protocol GitHubCopilotSuggestionServiceType {\n func getCompletions(\n fileURL: URL,\n content: String,\n originalContent: String,\n cursorPosition: CursorPosition,\n tabSize: Int,\n indentSize: Int,\n usesTabsForIndentation: Bool\n ) async throws -> [CodeSuggestion]\n func notifyAccepted(_ completion: CodeSuggestion) async\n func notifyRejected(_ completions: [CodeSuggestion]) async\n func notifyOpenTextDocument(fileURL: URL, content: String) async throws\n func notifyChangeTextDocument(fileURL: URL, content: String, version: Int) async throws\n func notifyCloseTextDocument(fileURL: URL) async throws\n func notifySaveTextDocument(fileURL: URL) async throws\n func cancelRequest() async\n func terminate() async\n func cancelOngoingTask(workDoneToken: String) async\n}\n\nprotocol GitHubCopilotLSP {\n func sendRequest(\n _ endpoint: E,\n timeout: TimeInterval?\n ) async throws -> E.Response\n func sendNotification(_ notif: ClientNotification) async throws\n}\n\nextension GitHubCopilotLSP {\n func sendRequest(_ endpoint: E) async throws -> E.Response {\n try await sendRequest(endpoint, timeout: nil)\n }\n}\n\nenum GitHubCopilotError: Error, LocalizedError {\n case notLoggedIn\n case languageServerNotInstalled\n case languageServerError(ServerError)\n case failedToInstallStartScript\n case chatEndsWithError(String)\n\n var errorDescription: String? {\n switch self {\n case .notLoggedIn:\n return \"Not logged in.\"\n case .languageServerNotInstalled:\n return \"Language server is not installed.\"\n case .failedToInstallStartScript:\n return \"Failed to install start script.\"\n case let .chatEndsWithError(errorMessage):\n return \"Chat ended with error message: \\(errorMessage)\"\n case let .languageServerError(error):\n switch error {\n case let .handlerUnavailable(handler):\n return \"Language server error: Handler \\(handler) unavailable\"\n case let .unhandledMethod(method):\n return \"Language server error: Unhandled method \\(method)\"\n case let .notificationDispatchFailed(error):\n return \"Language server error: Notification dispatch failed: \\(error)\"\n case let .requestDispatchFailed(error):\n return \"Language server error: Request dispatch failed: \\(error)\"\n case let .clientDataUnavailable(error):\n return \"Language server error: Client data unavailable: \\(error)\"\n case .serverUnavailable:\n return \"Language server error: Server unavailable, please make sure that:\\n1. The path to node is correctly set.\\n2. The node is not a shim executable.\\n3. the node version is high enough (v22.0+).\"\n case .missingExpectedParameter:\n return \"Language server error: Missing expected parameter\"\n case .missingExpectedResult:\n return \"Language server error: Missing expected result\"\n case let .unableToDecodeRequest(error):\n return \"Language server error: Unable to decode request: \\(error)\"\n case let .unableToSendRequest(error):\n return \"Language server error: Unable to send request: \\(error)\"\n case let .unableToSendNotification(error):\n return \"Language server error: Unable to send notification: \\(error)\"\n case let .serverError(code: code, message: message, data: data):\n return \"Language server error: Server error: \\(code) \\(message) \\(String(describing: data))\"\n case .invalidRequest:\n return \"Language server error: Invalid request\"\n case .timeout:\n return \"Language server error: Timeout, please try again later\"\n }\n }\n }\n}\n\npublic extension Notification.Name {\n static let gitHubCopilotShouldRefreshEditorInformation = Notification\n .Name(\"com.intii.CopilotForXcode.GitHubCopilotShouldRefreshEditorInformation\")\n}\n\npublic class GitHubCopilotBaseService {\n let projectRootURL: URL\n var server: GitHubCopilotLSP\n var localProcessServer: CopilotLocalProcessServer?\n let notificationHandler: ServerNotificationHandler\n\n deinit {\n localProcessServer?.terminate()\n }\n\n init(designatedServer: GitHubCopilotLSP) {\n projectRootURL = URL(fileURLWithPath: \"/\")\n server = designatedServer\n notificationHandler = .init()\n }\n\n init(projectRootURL: URL) throws {\n self.projectRootURL = projectRootURL\n let notificationHandler = ServerNotificationHandler()\n self.notificationHandler = notificationHandler\n let (server, localServer) = try { [notificationHandler] in\n let urls = try GitHubCopilotBaseService.createFoldersIfNeeded()\n let executionParams: Process.ExecutionParameters\n let runner = UserDefaults.shared.value(for: \\.runNodeWith)\n// let watchedFiles = JSONValue(\n// booleanLiteral: projectRootURL.path == \"/\" ? false : true\n// )\n\n guard let agentJSURL = { () -> URL? in\n let languageServerDotJS = urls.executableURL\n .appendingPathComponent(\"copilot/dist/language-server.js\")\n if FileManager.default.fileExists(atPath: languageServerDotJS.path) {\n return languageServerDotJS\n }\n let agentsDotJS = urls.executableURL.appendingPathComponent(\"copilot/dist/agent.js\")\n if FileManager.default.fileExists(atPath: agentsDotJS.path) {\n return agentsDotJS\n }\n return nil\n }() else {\n throw GitHubCopilotError.languageServerNotInstalled\n }\n\n let indexJSURL: URL = try {\n if UserDefaults.shared.value(for: \\.gitHubCopilotLoadKeyChainCertificates) {\n let url = urls.executableURL\n .appendingPathComponent(\"load-self-signed-cert-1.34.0.js\")\n if !FileManager.default.fileExists(atPath: url.path) {\n let file = Bundle.module.url(\n forResource: \"load-self-signed-cert-1.34.0\",\n withExtension: \"js\"\n )!\n do {\n try FileManager.default.copyItem(at: file, to: url)\n } catch {\n throw GitHubCopilotError.failedToInstallStartScript\n }\n }\n return url\n } else {\n return agentJSURL\n }\n }()\n\n #if DEBUG\n let environment: [String: String] = [\n \"GH_COPILOT_DEBUG_UI_PORT\": \"8080\",\n \"GH_COPILOT_VERBOSE\": UserDefaults.shared.value(for: \\.gitHubCopilotVerboseLog)\n ? \"true\" : \"false\",\n ]\n #else\n let environment = [String: String]()\n #endif\n\n switch runner {\n case .bash:\n let nodePath = UserDefaults.shared.value(for: \\.nodePath)\n let command = [\n nodePath.isEmpty ? \"node\" : nodePath,\n \"\\\"\\(indexJSURL.path)\\\"\",\n \"--stdio\",\n ].joined(separator: \" \")\n executionParams = Process.ExecutionParameters(\n path: \"/bin/bash\",\n arguments: [\"-i\", \"-l\", \"-c\", command],\n environment: environment,\n currentDirectoryURL: urls.supportURL\n )\n case .shell:\n let shell = ProcessInfo.processInfo.shellExecutablePath\n let nodePath = UserDefaults.shared.value(for: \\.nodePath)\n let command = [\n nodePath.isEmpty ? \"node\" : nodePath,\n \"\\\"\\(indexJSURL.path)\\\"\",\n \"--stdio\",\n ].joined(separator: \" \")\n executionParams = Process.ExecutionParameters(\n path: shell,\n arguments: [\"-i\", \"-l\", \"-c\", command],\n environment: environment,\n currentDirectoryURL: urls.supportURL\n )\n case .env:\n let userEnvPath =\n \"/opt/homebrew/bin:/opt/homebrew/sbin:/usr/local/bin:/usr/bin:/bin:/usr/sbin:/sbin\"\n executionParams = {\n let nodePath = UserDefaults.shared.value(for: \\.nodePath)\n return Process.ExecutionParameters(\n path: \"/usr/bin/env\",\n arguments: [\n nodePath.isEmpty ? \"node\" : nodePath,\n indexJSURL.path,\n \"--stdio\",\n ],\n environment: [\n \"PATH\": userEnvPath,\n ],\n currentDirectoryURL: urls.supportURL\n )\n }()\n }\n let localServer = CopilotLocalProcessServer(\n executionParameters: executionParams,\n serverNotificationHandler: notificationHandler\n )\n\n localServer.logMessages = UserDefaults.shared.value(for: \\.gitHubCopilotVerboseLog)\n let server = InitializingServer(server: localServer)\n\n server.initializeParamsProvider = {\n let capabilities = ClientCapabilities(\n workspace: nil,\n textDocument: nil,\n window: nil,\n general: nil,\n experimental: nil\n )\n\n let pretendToBeVSCode = UserDefaults.shared\n .value(for: \\.gitHubCopilotPretendIDEToBeVSCode)\n return InitializeParams(\n processId: Int(ProcessInfo.processInfo.processIdentifier),\n clientInfo: .init(\n name: Bundle.main\n .object(forInfoDictionaryKey: \"HOST_APP_NAME\") as? String\n ?? \"Copilot for Xcode\"\n ),\n locale: nil,\n rootPath: projectRootURL.path,\n rootUri: projectRootURL.path,\n initializationOptions: [\n \"editorInfo\": pretendToBeVSCode ? .hash([\n \"name\": \"vscode\",\n \"version\": \"1.99.3\",\n ]) : .hash([\n \"name\": \"Xcode\",\n \"version\": .string(xcodeVersion() ?? \"16.0\"),\n ]),\n \"editorPluginInfo\": .hash([\n \"name\": \"Copilot for Xcode\",\n \"version\": .string(Bundle.main\n .infoDictionary?[\"CFBundleShortVersionString\"] as? String ?? \"\"),\n ]),\n// \"copilotCapabilities\": [\n// /// The editor has support for watching files over LSP\n// \"watchedFiles\": watchedFiles,\n// ],\n ],\n capabilities: capabilities,\n trace: .off,\n workspaceFolders: [WorkspaceFolder(\n uri: projectRootURL.absoluteString,\n name: projectRootURL.lastPathComponent\n )]\n )\n }\n\n return (server, localServer)\n }()\n\n self.server = server\n localProcessServer = localServer\n\n let notifications = NotificationCenter.default\n .notifications(named: .gitHubCopilotShouldRefreshEditorInformation)\n Task { [weak self] in\n _ = try? await server.sendRequest(\n GitHubCopilotRequest.SetEditorInfo(xcodeVersion: xcodeVersion() ?? \"16.0\")\n )\n\n for await _ in notifications {\n guard self != nil else { return }\n _ = try? await server.sendRequest(\n GitHubCopilotRequest.SetEditorInfo(xcodeVersion: xcodeVersion() ?? \"16.0\")\n )\n }\n }\n }\n\n public static func createFoldersIfNeeded() throws -> (\n applicationSupportURL: URL,\n gitHubCopilotURL: URL,\n executableURL: URL,\n supportURL: URL\n ) {\n guard let supportURL = FileManager.default.urls(\n for: .applicationSupportDirectory,\n in: .userDomainMask\n ).first?.appendingPathComponent(\n Bundle.main\n .object(forInfoDictionaryKey: \"APPLICATION_SUPPORT_FOLDER\") as! String\n ) else {\n throw CancellationError()\n }\n\n if !FileManager.default.fileExists(atPath: supportURL.path) {\n try? FileManager.default\n .createDirectory(at: supportURL, withIntermediateDirectories: false)\n }\n let gitHubCopilotFolderURL = supportURL.appendingPathComponent(\"GitHub Copilot\")\n if !FileManager.default.fileExists(atPath: gitHubCopilotFolderURL.path) {\n try? FileManager.default\n .createDirectory(at: gitHubCopilotFolderURL, withIntermediateDirectories: false)\n }\n let supportFolderURL = gitHubCopilotFolderURL.appendingPathComponent(\"support\")\n if !FileManager.default.fileExists(atPath: supportFolderURL.path) {\n try? FileManager.default\n .createDirectory(at: supportFolderURL, withIntermediateDirectories: false)\n }\n let executableFolderURL = gitHubCopilotFolderURL.appendingPathComponent(\"executable\")\n if !FileManager.default.fileExists(atPath: executableFolderURL.path) {\n try? FileManager.default\n .createDirectory(at: executableFolderURL, withIntermediateDirectories: false)\n }\n\n return (supportURL, gitHubCopilotFolderURL, executableFolderURL, supportFolderURL)\n }\n\n func registerNotificationHandler(\n id: AnyHashable,\n _ block: @escaping ServerNotificationHandler.Handler\n ) {\n Task { @GitHubCopilotSuggestionActor in\n self.notificationHandler.handlers[id] = block\n }\n }\n\n func unregisterNotificationHandler(id: AnyHashable) {\n Task { @GitHubCopilotSuggestionActor in\n self.notificationHandler.handlers[id] = nil\n }\n }\n}\n\npublic final class GitHubCopilotAuthService: GitHubCopilotBaseService,\n GitHubCopilotAuthServiceType\n{\n public init() throws {\n let home = FileManager.default.homeDirectoryForCurrentUser\n try super.init(projectRootURL: home)\n }\n\n public func checkStatus() async throws -> GitHubCopilotAccountStatus {\n do {\n return try await server.sendRequest(GitHubCopilotRequest.CheckStatus()).status\n } catch let error as ServerError {\n throw GitHubCopilotError.languageServerError(error)\n } catch {\n throw error\n }\n }\n\n public func signInInitiate() async throws -> (verificationUri: String, userCode: String) {\n do {\n let result = try await server.sendRequest(GitHubCopilotRequest.SignInInitiate())\n return (result.verificationUri, result.userCode)\n } catch let error as ServerError {\n throw GitHubCopilotError.languageServerError(error)\n } catch {\n throw error\n }\n }\n\n public func signInConfirm(userCode: String) async throws\n -> (username: String, status: GitHubCopilotAccountStatus)\n {\n do {\n let result = try await server\n .sendRequest(GitHubCopilotRequest.SignInConfirm(userCode: userCode))\n return (result.user, result.status)\n } catch let error as ServerError {\n throw GitHubCopilotError.languageServerError(error)\n } catch {\n throw error\n }\n }\n\n public func signOut() async throws -> GitHubCopilotAccountStatus {\n do {\n return try await server.sendRequest(GitHubCopilotRequest.SignOut()).status\n } catch let error as ServerError {\n throw GitHubCopilotError.languageServerError(error)\n } catch {\n throw error\n }\n }\n\n public func version() async throws -> String {\n do {\n return try await server.sendRequest(GitHubCopilotRequest.GetVersion()).version\n } catch let error as ServerError {\n throw GitHubCopilotError.languageServerError(error)\n } catch {\n throw error\n }\n }\n}\n\n@globalActor public enum GitHubCopilotSuggestionActor {\n public actor TheActor {}\n public static let shared = TheActor()\n}\n\npublic final class GitHubCopilotService: GitHubCopilotBaseService,\n GitHubCopilotSuggestionServiceType\n{\n private var ongoingTasks = Set>()\n\n override public init(projectRootURL: URL = URL(fileURLWithPath: \"/\")) throws {\n try super.init(projectRootURL: projectRootURL)\n }\n\n override init(designatedServer: GitHubCopilotLSP) {\n super.init(designatedServer: designatedServer)\n }\n\n @GitHubCopilotSuggestionActor\n public func getCompletions(\n fileURL: URL,\n content: String,\n originalContent: String,\n cursorPosition: CursorPosition,\n tabSize: Int,\n indentSize: Int,\n usesTabsForIndentation: Bool\n ) async throws -> [CodeSuggestion] {\n ongoingTasks.forEach { $0.cancel() }\n ongoingTasks.removeAll()\n await localProcessServer?.cancelOngoingTasks()\n\n func sendRequest(maxTry: Int = 5) async throws -> [CodeSuggestion] {\n do {\n let completions = try await server\n .sendRequest(GitHubCopilotRequest.InlineCompletion(doc: .init(\n textDocument: .init(uri: fileURL.absoluteString, version: 1),\n position: cursorPosition,\n formattingOptions: .init(\n tabSize: tabSize,\n insertSpaces: !usesTabsForIndentation\n ),\n context: .init(triggerKind: .invoked)\n )))\n .items\n .compactMap { (item: _) -> CodeSuggestion? in\n guard let range = item.range else { return nil }\n let suggestion = CodeSuggestion(\n id: item.command?.arguments?.first ?? UUID().uuidString,\n text: item.insertText,\n position: cursorPosition,\n range: .init(start: range.start, end: range.end)\n )\n return suggestion\n }\n try Task.checkCancellation()\n return completions\n } catch let error as ServerError {\n switch error {\n case .serverError(1000, _, _): // not logged-in error\n throw SuggestionServiceError\n .notice(GitHubCopilotError.languageServerError(error))\n case .serverError:\n // sometimes the content inside language server is not new enough, which can\n // lead to an version mismatch error. We can try a few times until the content\n // is up to date.\n if maxTry <= 0 { break }\n Logger.gitHubCopilot.error(\n \"Try getting suggestions again: \\(GitHubCopilotError.languageServerError(error).localizedDescription)\"\n )\n try await Task.sleep(nanoseconds: 200_000_000)\n return try await sendRequest(maxTry: maxTry - 1)\n default:\n break\n }\n throw GitHubCopilotError.languageServerError(error)\n } catch {\n throw error\n }\n }\n\n func recoverContent() async {\n try? await notifyChangeTextDocument(\n fileURL: fileURL,\n content: originalContent,\n version: 0\n )\n }\n\n // since when the language server is no longer using the passed in content to generate\n // suggestions, we will need to update the content to the file before we do any request.\n //\n // And sometimes the language server's content was not up to date and may generate\n // weird result when the cursor position exceeds the line.\n let task = Task { @GitHubCopilotSuggestionActor in\n do {\n try await notifyChangeTextDocument(\n fileURL: fileURL,\n content: content,\n version: 1\n )\n } catch let error as ServerError {\n switch error {\n case .serverUnavailable:\n throw SuggestionServiceError\n .notice(GitHubCopilotError.languageServerError(error))\n default:\n throw error\n }\n } catch {\n throw error\n }\n\n do {\n try Task.checkCancellation()\n return try await sendRequest()\n } catch let error as CancellationError {\n if ongoingTasks.isEmpty {\n await recoverContent()\n }\n throw error\n } catch {\n await recoverContent()\n throw error\n }\n }\n\n ongoingTasks.insert(task)\n\n return try await task.value\n }\n\n @GitHubCopilotSuggestionActor\n public func cancelRequest() async {\n ongoingTasks.forEach { $0.cancel() }\n ongoingTasks.removeAll()\n await localProcessServer?.cancelOngoingTasks()\n }\n\n @GitHubCopilotSuggestionActor\n public func notifyAccepted(_ completion: CodeSuggestion) async {\n _ = try? await server.sendRequest(\n GitHubCopilotRequest.NotifyAccepted(completionUUID: completion.id)\n )\n }\n\n @GitHubCopilotSuggestionActor\n public func notifyRejected(_ completions: [CodeSuggestion]) async {\n _ = try? await server.sendRequest(\n GitHubCopilotRequest.NotifyRejected(completionUUIDs: completions.map(\\.id))\n )\n }\n\n @GitHubCopilotSuggestionActor\n public func notifyOpenTextDocument(\n fileURL: URL,\n content: String\n ) async throws {\n let languageId = languageIdentifierFromFileURL(fileURL)\n let uri = \"file://\\(fileURL.path)\"\n// Logger.service.debug(\"Open \\(uri), \\(content.count)\")\n try await server.sendNotification(\n .didOpenTextDocument(\n DidOpenTextDocumentParams(\n textDocument: .init(\n uri: uri,\n languageId: languageId.rawValue,\n version: 0,\n text: content\n )\n )\n )\n )\n }\n\n @GitHubCopilotSuggestionActor\n public func notifyChangeTextDocument(\n fileURL: URL,\n content: String,\n version: Int\n ) async throws {\n let uri = \"file://\\(fileURL.path)\"\n// Logger.service.debug(\"Change \\(uri), \\(content.count)\")\n try await server.sendNotification(\n .didChangeTextDocument(\n DidChangeTextDocumentParams(\n uri: uri,\n version: version,\n contentChange: .init(\n range: nil,\n rangeLength: nil,\n text: content\n )\n )\n )\n )\n }\n\n @GitHubCopilotSuggestionActor\n public func notifySaveTextDocument(fileURL: URL) async throws {\n let uri = \"file://\\(fileURL.path)\"\n// Logger.service.debug(\"Save \\(uri)\")\n try await server.sendNotification(.didSaveTextDocument(.init(uri: uri)))\n }\n\n @GitHubCopilotSuggestionActor\n public func notifyCloseTextDocument(fileURL: URL) async throws {\n let uri = \"file://\\(fileURL.path)\"\n// Logger.service.debug(\"Close \\(uri)\")\n try await server.sendNotification(.didCloseTextDocument(.init(uri: uri)))\n }\n\n @GitHubCopilotSuggestionActor\n public func terminate() async {\n // automatically handled\n }\n\n public func cancelOngoingTask(workDoneToken: String) async {\n await localProcessServer?.cancelOngoingTask(workDoneToken: workDoneToken)\n }\n}\n\nextension InitializingServer: GitHubCopilotLSP {\n func sendRequest(\n _ endpoint: E,\n timeout: TimeInterval? = nil\n ) async throws -> E.Response {\n if let timeout {\n return try await withCheckedThrowingContinuation { continuation in\n self.sendRequest(endpoint.request, timeout: timeout) { result in\n continuation.resume(with: result)\n }\n }\n } else {\n return try await sendRequest(endpoint.request)\n }\n }\n}\n\nprivate func xcodeVersion() -> String? {\n let process = Process()\n process.executableURL = URL(fileURLWithPath: \"/usr/bin/xcrun\")\n process.arguments = [\"xcodebuild\", \"-version\"]\n\n let pipe = Pipe()\n process.standardOutput = pipe\n\n do {\n try process.run()\n } catch {\n print(\"Error running xcrun xcodebuild: \\(error)\")\n return nil\n }\n\n let data = pipe.fileHandleForReading.readDataToEndOfFile()\n guard let output = String(data: data, encoding: .utf8) else {\n return nil\n }\n\n let lines = output.split(separator: \"\\n\")\n return lines.first?.split(separator: \" \").last.map(String.init)\n}\n\n" }, { "path": "Tool/Sources/GitHubCopilotService/Resources/load-self-signed-cert-1.34.0.js", "content": "function initialize() {\n if (process.platform !== \"darwin\") {\n return;\n }\n\n const splitPattern = /(?=-----BEGIN\\sCERTIFICATE-----)/g;\n const systemRootCertsPath =\n \"/System/Library/Keychains/SystemRootCertificates.keychain\";\n const args = [\"find-certificate\", \"-a\", \"-p\"];\n\n const childProcess = require(\"child_process\");\n const allTrusted = childProcess\n .spawnSync(\"/usr/bin/security\", args)\n .stdout.toString()\n .split(splitPattern);\n\n const allRoot = childProcess\n .spawnSync(\"/usr/bin/security\", args.concat(systemRootCertsPath))\n .stdout.toString()\n .split(splitPattern);\n const all = allTrusted.concat(allRoot);\n\n const tls = require(\"tls\");\n const origCreateSecureContext = tls.createSecureContext;\n tls.createSecureContext = (options) => {\n const ctx = origCreateSecureContext(options);\n all.filter(duplicated).forEach((cert) => {\n ctx.context.addCACert(cert.trim());\n });\n return ctx;\n };\n}\n\nfunction duplicated(cert, index, arr) {\n return arr.indexOf(cert) === index;\n}\n\ninitialize();\n\nrequire(\"./copilot/dist/language-server.js\");\n" }, { "path": "Tool/Sources/GitHubCopilotService/Services/GitHubCopilotChatService.swift", "content": "import BuiltinExtension\nimport ChatBasic\nimport CopilotForXcodeKit\nimport Foundation\nimport LanguageServerProtocol\nimport XcodeInspector\n\npublic final class GitHubCopilotChatService: BuiltinExtensionChatServiceType {\n let serviceLocator: any ServiceLocatorType\n\n init(serviceLocator: any ServiceLocatorType) {\n self.serviceLocator = serviceLocator\n }\n\n /// - note: Let's do it in a naive way for proof of concept. We will create a new chat for each\n /// message in this version.\n public func sendMessage(\n _ message: String,\n history: [Message],\n references: [RetrievedContent],\n workspace: WorkspaceInfo\n ) async -> AsyncThrowingStream {\n guard let service = await serviceLocator.getService(from: workspace)\n else { return .finished(throwing: CancellationError()) }\n let id = UUID().uuidString\n let editorContent = await XcodeInspector.shared.getFocusedEditorContent()\n let workDoneToken = UUID().uuidString\n let turns = convertHistory(history: history, message: message)\n let doc = GitHubCopilotDoc(\n source: editorContent?.editorContent?.content ?? \"\",\n tabSize: 1,\n indentSize: 4,\n insertSpaces: true,\n path: editorContent?.documentURL.absoluteString ?? \"\",\n uri: editorContent?.documentURL.absoluteString ?? \"\",\n relativePath: editorContent?.relativePath ?? \"\",\n languageId: editorContent?.language ?? .plaintext,\n position: editorContent?.editorContent?.cursorPosition ?? .zero\n )\n let request = GitHubCopilotRequest.ConversationCreate(requestBody: .init(\n workDoneToken: workDoneToken,\n turns: turns,\n capabilities: .init(skills: [], allSkills: false),\n textDocument: doc,\n source: .panel,\n workspaceFolder: workspace.projectURL.absoluteString,\n model: {\n let selectedModel = UserDefaults.shared.value(for: \\.gitHubCopilotModelId)\n if selectedModel.isEmpty {\n return nil\n }\n return selectedModel\n }(),\n userLanguage: {\n let language = UserDefaults.shared.value(for: \\.chatGPTLanguage)\n if language.isEmpty {\n return \"Auto Detected\"\n }\n return language\n }()\n ))\n\n let stream = AsyncThrowingStream { continuation in\n let startTimestamp = Date()\n\n continuation.onTermination = { _ in\n Task {\n service.unregisterNotificationHandler(id: id)\n await service.cancelOngoingTask(workDoneToken: workDoneToken)\n }\n }\n\n service.registerNotificationHandler(id: id) { notification, data in\n // just incase the conversation is stuck, we will cancel it after timeout\n if Date().timeIntervalSince(startTimestamp) > 60 * 30 {\n continuation.finish(throwing: CancellationError())\n return false\n }\n\n switch notification.method {\n case \"$/progress\":\n do {\n let progress = try JSONDecoder().decode(\n JSONRPC.self,\n from: data\n ).params\n guard progress.token == workDoneToken else { return false }\n if let reply = progress.value.reply, progress.value.kind == \"report\" {\n continuation.yield(reply)\n } else if progress.value.kind == \"end\" {\n if let error = progress.value.error?.message,\n progress.value.cancellationReason == nil\n {\n if error.contains(\"400\") {\n continuation.finish(\n throwing: GitHubCopilotError.chatEndsWithError(\n \"\\(error). Please try enabling pretend IDE to be VSCode and click refresh configuration.\"\n )\n )\n } else if error.contains(\"No model configuration found\") {\n continuation.finish(\n throwing: GitHubCopilotError.chatEndsWithError(\n \"\\(error). Please try enabling pretend IDE to be VSCode and click refresh configuration.\"\n )\n )\n } else {\n continuation.finish(\n throwing: GitHubCopilotError.chatEndsWithError(error)\n )\n }\n } else {\n continuation.finish()\n }\n }\n return true\n } catch {\n return false\n }\n case \"conversation/context\":\n do {\n _ = try JSONDecoder().decode(\n JSONRPC.self,\n from: data\n )\n throw ServerError.clientDataUnavailable(CancellationError())\n } catch {\n return false\n }\n\n default:\n return false\n }\n }\n\n Task {\n do {\n // this will return when the response is generated.\n let createResponse = try await service.server.sendRequest(request, timeout: 120)\n _ = try await service.server.sendRequest(\n GitHubCopilotRequest.ConversationDestroy(requestBody: .init(\n conversationId: createResponse.conversationId\n ))\n )\n } catch let error as ServerError {\n continuation.finish(throwing: GitHubCopilotError.languageServerError(error))\n } catch {\n continuation.finish(throwing: error)\n }\n }\n }\n\n return stream\n }\n}\n\nextension GitHubCopilotChatService {\n typealias Turn = GitHubCopilotRequest.ConversationCreate.RequestBody.ConversationTurn\n func convertHistory(history: [Message], message: String) -> [Turn] {\n guard let firstIndexOfUserMessage = history.firstIndex(where: { $0.role == .user })\n else { return [.init(request: message, response: nil)] }\n\n var currentTurn = Turn(request: \"\", response: nil)\n var turns: [Turn] = []\n let systemPrompt = history\n .filter { $0.role == .system }.compactMap(\\.content)\n .joined(separator: \"\\n\\n\")\n\n if !systemPrompt.isEmpty {\n turns.append(.init(request: \"[System Prompt]\\n\\(systemPrompt)\", response: \"OK!\"))\n }\n\n for i in firstIndexOfUserMessage.. String {\n return message\n }\n\n struct JSONRPC: Decodable {\n var jsonrpc: String\n var method: String\n var params: Params\n }\n\n struct StreamProgressParams: Decodable {\n struct Value: Decodable {\n struct Step: Decodable {\n var id: String\n var title: String\n var status: String\n }\n\n struct FollowUp: Decodable {\n var id: String\n var type: String\n var message: String\n }\n\n struct Error: Decodable {\n var responseIsIncomplete: Bool?\n var message: String?\n }\n\n struct Annotation: Decodable {\n var id: Int\n }\n\n var kind: String\n var title: String?\n var conversationId: String\n var turnId: String\n var steps: [Step]?\n var followUp: FollowUp?\n var suggestedTitle: String?\n var reply: String?\n var annotations: [Annotation]?\n var hideText: Bool?\n var cancellationReason: String?\n var error: Error?\n }\n\n var token: String\n var value: Value\n }\n\n struct ConversationContextParams: Decodable {\n enum SkillID: String, Decodable {\n case currentEditor = \"current-editor\"\n case projectLabels = \"project-labels\"\n case recentFiles = \"recent-files\"\n case references\n case problemsInActiveDocument = \"problems-in-active-document\"\n }\n\n var conversationId: String\n var turnId: String\n var skillId: String\n }\n\n struct ConversationContextResponseBody: Encodable {}\n}\n\n" }, { "path": "Tool/Sources/GitHubCopilotService/Services/GitHubCopilotSuggestionService.swift", "content": "import CopilotForXcodeKit\nimport Foundation\nimport SuggestionBasic\nimport Workspace\n\npublic final class GitHubCopilotSuggestionService: SuggestionServiceType {\n public var configuration: SuggestionServiceConfiguration {\n .init(\n acceptsRelevantCodeSnippets: true,\n mixRelevantCodeSnippetsInSource: true,\n acceptsRelevantSnippetsFromOpenedFiles: false\n )\n }\n\n let serviceLocator: ServiceLocatorType\n\n init(serviceLocator: ServiceLocatorType) {\n self.serviceLocator = serviceLocator\n }\n\n public func getSuggestions(\n _ request: SuggestionRequest,\n workspace: WorkspaceInfo\n ) async throws -> [CopilotForXcodeKit.CodeSuggestion] {\n guard let service = await serviceLocator.getService(from: workspace) else { return [] }\n return try await service.getCompletions(\n fileURL: request.fileURL,\n content: request.content,\n originalContent: request.originalContent,\n cursorPosition: .init(\n line: request.cursorPosition.line,\n character: request.cursorPosition.character\n ),\n tabSize: request.tabSize,\n indentSize: request.indentSize,\n usesTabsForIndentation: request.usesTabsForIndentation\n ).map(Self.convert)\n }\n\n public func notifyAccepted(\n _ suggestion: CopilotForXcodeKit.CodeSuggestion,\n workspace: WorkspaceInfo\n ) async {\n guard let service = await serviceLocator.getService(from: workspace) else { return }\n await service.notifyAccepted(Self.convert(suggestion))\n }\n\n public func notifyRejected(\n _ suggestions: [CopilotForXcodeKit.CodeSuggestion],\n workspace: WorkspaceInfo\n ) async {\n guard let service = await serviceLocator.getService(from: workspace) else { return }\n await service.notifyRejected(suggestions.map(Self.convert))\n }\n\n public func cancelRequest(workspace: WorkspaceInfo) async {\n guard let service = await serviceLocator.getService(from: workspace) else { return }\n await service.cancelRequest()\n }\n\n static func convert(\n _ suggestion: SuggestionBasic.CodeSuggestion\n ) -> CopilotForXcodeKit.CodeSuggestion {\n .init(\n id: suggestion.id,\n text: suggestion.text,\n position: .init(\n line: suggestion.position.line,\n character: suggestion.position.character\n ),\n range: .init(\n start: .init(\n line: suggestion.range.start.line,\n character: suggestion.range.start.character\n ),\n end: .init(\n line: suggestion.range.end.line,\n character: suggestion.range.end.character\n )\n )\n )\n }\n\n static func convert(\n _ suggestion: CopilotForXcodeKit.CodeSuggestion\n ) -> SuggestionBasic.CodeSuggestion {\n .init(\n id: suggestion.id,\n text: suggestion.text,\n position: .init(\n line: suggestion.position.line,\n character: suggestion.position.character\n ),\n range: .init(\n start: .init(\n line: suggestion.range.start.line,\n character: suggestion.range.start.character\n ),\n end: .init(\n line: suggestion.range.end.line,\n character: suggestion.range.end.character\n )\n )\n )\n }\n}\n\n" }, { "path": "Tool/Sources/GitIgnoreCheck/GitIgnoreCheck.swift", "content": "import Dependencies\nimport Foundation\nimport Terminal\nimport Preferences\n\npublic struct CheckIfGitIgnoredDependencyKey: DependencyKey {\n public static var liveValue: GitIgnoredChecker = DefaultGitIgnoredChecker()\n public static var testValue: GitIgnoredChecker = DefaultGitIgnoredChecker(isTest: true)\n}\n\npublic extension DependencyValues {\n var gitIgnoredChecker: GitIgnoredChecker {\n get { self[CheckIfGitIgnoredDependencyKey.self] }\n set { self[CheckIfGitIgnoredDependencyKey.self] = newValue }\n }\n}\n\npublic protocol GitIgnoredChecker {\n func checkIfGitIgnored(fileURL: URL) async -> Bool\n func checkIfGitIgnored(fileURLs: [URL]) async -> [URL]\n}\n\npublic extension GitIgnoredChecker {\n func checkIfGitIgnored(filePath: String) async -> Bool {\n await checkIfGitIgnored(fileURL: URL(fileURLWithPath: filePath))\n }\n \n func checkIfGitIgnored(filePaths: [String]) async -> [String] {\n await checkIfGitIgnored(fileURLs: filePaths.map { URL(fileURLWithPath: $0) })\n .map(\\.path)\n }\n}\n\npublic struct DefaultGitIgnoredChecker: GitIgnoredChecker {\n var isTest = false\n \n var noCheck: Bool {\n if isTest { return true }\n return UserDefaults.shared.value(for: \\.disableGitIgnoreCheck)\n }\n \n public init() {}\n \n init(isTest: Bool) {\n self.isTest = isTest\n }\n \n public func checkIfGitIgnored(fileURL: URL) async -> Bool {\n if noCheck { return false }\n let terminal = Terminal()\n guard let gitFolderURL = gitFolderURL(forFileURL: fileURL) else {\n return false\n }\n do {\n let result = try await terminal.runCommand(\n \"/bin/bash\",\n arguments: [\"-c\", \"git check-ignore ${TARGET_FILE}\"],\n currentDirectoryURL: gitFolderURL,\n environment: [\"TARGET_FILE\": fileURL.path]\n )\n if result.isEmpty { return false }\n return true\n } catch {\n return false\n }\n }\n \n public func checkIfGitIgnored(fileURLs: [URL]) async -> [URL] {\n if noCheck { return [] }\n let filePaths = fileURLs.map { \"\\\"\\($0.path)\\\"\" }.joined(separator: \" \")\n guard let firstFileURL = fileURLs.first else { return [] }\n let terminal = Terminal()\n guard let gitFolderURL = gitFolderURL(forFileURL: firstFileURL) else {\n return []\n }\n do {\n let result = try await terminal.runCommand(\n \"/bin/bash\",\n arguments: [\"-c\", \"git check-ignore ${TARGET_FILE}\"],\n currentDirectoryURL: gitFolderURL,\n environment: [\"TARGET_FILE\": filePaths]\n )\n return result\n .split(whereSeparator: \\.isNewline)\n .map(String.init)\n .compactMap(URL.init(fileURLWithPath:))\n } catch {\n return []\n }\n }\n}\n\nfunc gitFolderURL(forFileURL fileURL: URL) -> URL? {\n var currentURL = fileURL\n let fileManager = FileManager.default\n while currentURL.path != \"/\" {\n let gitFolderURL = currentURL.appendingPathComponent(\".git\")\n if fileManager.fileExists(atPath: gitFolderURL.path) {\n return currentURL\n }\n currentURL = currentURL.deletingLastPathComponent()\n }\n return nil\n}\n\n" }, { "path": "Tool/Sources/JoinJSON/JoinJSON.swift", "content": "import Foundation\n\npublic struct JoinJSON {\n public init() {}\n\n public func join(_ a: String, with b: String) -> Data {\n return join(a.data(using: .utf8) ?? Data(), with: b.data(using: .utf8) ?? Data())\n }\n\n public func join(_ a: Data, with b: String) -> Data {\n return join(a, with: b.data(using: .utf8) ?? Data())\n }\n\n public func join(_ a: Data, with b: Data) -> Data {\n guard let firstDict = try? JSONSerialization.jsonObject(with: a) as? [String: Any],\n let secondDict = try? JSONSerialization.jsonObject(with: b) as? [String: Any]\n else {\n return a\n }\n\n var merged = firstDict\n for (key, value) in secondDict {\n merged[key] = value\n }\n\n return (try? JSONSerialization.data(withJSONObject: merged)) ?? a\n }\n}\n\n" }, { "path": "Tool/Sources/Keychain/Keychain.swift", "content": "import Configs\nimport Foundation\nimport Preferences\nimport Security\n\npublic protocol KeychainType {\n func getAll() throws -> [String: String]\n func update(_ value: String, key: String) throws\n func get(_ key: String) throws -> String?\n func remove(_ key: String) throws\n}\n\npublic final class FakeKeyChain: KeychainType {\n var values: [String: String] = [:]\n\n public init() {}\n\n public func getAll() throws -> [String: String] {\n values\n }\n\n public func update(_ value: String, key: String) throws {\n values[key] = value\n }\n\n public func get(_ key: String) throws -> String? {\n values[key]\n }\n\n public func remove(_ key: String) throws {\n values[key] = nil\n }\n}\n\npublic final class UserDefaultsBaseAPIKeychain: KeychainType {\n let defaults = UserDefaults.shared\n let scope: String\n var key: String {\n \"UserDefaultsBaseAPIKeychain-\\(scope)\"\n }\n \n init(scope: String) {\n self.scope = scope\n }\n \n public func getAll() throws -> [String : String] {\n defaults.dictionary(forKey: key) as? [String: String] ?? [:]\n }\n \n public func update(_ value: String, key: String) throws {\n var dict = try getAll()\n dict[key] = value\n defaults.set(dict, forKey: self.key)\n }\n \n public func get(_ key: String) throws -> String? {\n try getAll()[key]\n }\n \n public func remove(_ key: String) throws {\n var dict = try getAll()\n dict[key] = nil\n defaults.set(dict, forKey: self.key)\n }\n}\n\npublic struct Keychain: KeychainType {\n let service = keychainService\n let accessGroup = keychainAccessGroup\n let scope: String\n\n public static var apiKey: KeychainType {\n if UserDefaults.shared.value(for: \\.useUserDefaultsBaseAPIKeychain) {\n return UserDefaultsBaseAPIKeychain(scope: \"apiKey\")\n }\n return Keychain(scope: \"apiKey\")\n }\n\n public enum Error: Swift.Error {\n case failedToDeleteFromKeyChain\n case failedToUpdateOrSetItem\n }\n\n public init(scope: String = \"\") {\n self.scope = scope\n }\n\n func query(_ key: String) -> [String: Any] {\n let key = scopeKey(key)\n return [\n kSecClass as String: kSecClassGenericPassword as String,\n kSecAttrService as String: service,\n kSecAttrAccessGroup as String: accessGroup,\n kSecAttrAccount as String: key,\n kSecUseDataProtectionKeychain as String: true,\n ]\n }\n\n func set(_ value: String, key: String) throws {\n let query = query(key).merging([\n kSecValueData as String: value.data(using: .utf8) ?? Data(),\n ], uniquingKeysWith: { _, b in b })\n\n let result = SecItemAdd(query as CFDictionary, nil)\n\n switch result {\n case noErr:\n return\n default:\n throw Error.failedToUpdateOrSetItem\n }\n }\n\n func scopeKey(_ key: String) -> String {\n if scope.isEmpty {\n return key\n }\n return \"\\(scope)::\\(key)\"\n }\n\n func escapeScope(_ key: String) -> String? {\n if scope.isEmpty {\n return key\n }\n if !key.hasPrefix(\"\\(scope)::\") { return nil }\n return key.replacingOccurrences(of: \"\\(scope)::\", with: \"\")\n }\n\n public func getAll() throws -> [String: String] {\n let query = [\n kSecClass as String: kSecClassGenericPassword as String,\n kSecAttrService as String: service,\n kSecAttrAccessGroup as String: accessGroup,\n kSecReturnAttributes as String: true,\n kSecReturnData as String: true,\n kSecMatchLimit as String: kSecMatchLimitAll,\n ] as [String: Any]\n\n var result: AnyObject?\n if SecItemCopyMatching(query as CFDictionary, &result) == noErr {\n guard let items = result as? [[String: Any]] else {\n return [:]\n }\n\n var dict = [String: String]()\n for item in items {\n guard let key = item[kSecAttrAccount as String] as? String,\n let escapedKey = escapeScope(key)\n else { continue }\n guard let valueData = item[kSecValueData as String] as? Data,\n let value = String(data: valueData, encoding: .utf8)\n else { continue }\n dict[escapedKey] = value\n }\n return dict\n }\n\n return [:]\n }\n\n public func update(_ value: String, key: String) throws {\n let query = query(key).merging([\n kSecMatchLimit as String: kSecMatchLimitOne,\n kSecReturnData as String: true,\n ], uniquingKeysWith: { _, b in b })\n\n let attributes: [String: Any] =\n [kSecValueData as String: value.data(using: .utf8) ?? Data()]\n\n let result = SecItemUpdate(query as CFDictionary, attributes as CFDictionary)\n\n switch result {\n case noErr:\n return\n case errSecItemNotFound:\n try set(value, key: key)\n default:\n throw Error.failedToUpdateOrSetItem\n }\n }\n\n public func get(_ key: String) throws -> String? {\n let query = query(key).merging([\n kSecMatchLimit as String: kSecMatchLimitOne,\n kSecReturnData as String: true,\n kSecReturnAttributes as String: true,\n ], uniquingKeysWith: { _, b in b })\n\n var item: CFTypeRef?\n if SecItemCopyMatching(query as CFDictionary, &item) == noErr {\n if let existingItem = item as? [String: Any],\n let passwordData = existingItem[kSecValueData as String] as? Data,\n let password = String(data: passwordData, encoding: .utf8)\n {\n return password\n }\n return nil\n } else {\n return nil\n }\n }\n\n public func remove(_ key: String) throws {\n if SecItemDelete(query(key) as CFDictionary) == noErr {\n return\n }\n throw Error.failedToDeleteFromKeyChain\n }\n}\n\n" }, { "path": "Tool/Sources/LangChain/Agent.swift", "content": "import Foundation\n\npublic struct AgentAction: Equatable {\n public var toolName: String\n public var toolInput: String\n public var log: String\n public var observation: String?\n\n public init(toolName: String, toolInput: String, log: String, observation: String? = nil) {\n self.toolName = toolName\n self.toolInput = toolInput\n self.log = log\n self.observation = observation\n }\n\n public func observationAvailable(_ observation: String) -> AgentAction {\n var newAction = self\n newAction.observation = observation\n return newAction\n }\n}\n\npublic extension CallbackEvents {\n struct AgentDidFinish: CallbackEvent {\n public let info: AgentFinish\n }\n\n static func agentDidFinish() -> AgentDidFinish.Type {\n AgentDidFinish.self\n }\n\n struct AgentActionDidStart: CallbackEvent {\n public let info: AgentAction\n }\n\n var agentActionDidStart: AgentActionDidStart.Type {\n AgentActionDidStart.self\n }\n\n struct AgentActionDidEnd: CallbackEvent {\n public let info: AgentAction\n }\n\n var agentActionDidEnd: AgentActionDidEnd.Type {\n AgentActionDidEnd.self\n }\n \n struct AgentFunctionCallingToolReportProgress: CallbackEvent {\n public struct Info {\n public let functionName: String\n public let progress: String\n }\n \n public let info: Info\n }\n \n var agentFunctionCallingToolReportProgress: AgentFunctionCallingToolReportProgress.Type {\n AgentFunctionCallingToolReportProgress.self\n }\n}\n\npublic struct AgentFinish {\n public enum ReturnValue {\n case structured(Output)\n case unstructured(String)\n }\n\n public var returnValue: ReturnValue\n public var log: String\n\n public init(returnValue: ReturnValue, log: String) {\n self.returnValue = returnValue\n self.log = log\n }\n}\n\nextension AgentFinish.ReturnValue: Equatable where Output: Equatable {}\n\nextension AgentFinish: Equatable where Output: Equatable {}\n\npublic enum AgentNextStep {\n case actions([AgentAction])\n case finish(AgentFinish)\n}\n\nextension AgentNextStep: Equatable where Output: Equatable {}\n\npublic struct AgentScratchPad: Equatable {\n public var content: Content\n\n public init(content: Content) {\n self.content = content\n }\n}\n\npublic struct AgentInput {\n public var input: T\n public var thoughts: AgentScratchPad\n\n public init(input: T, thoughts: AgentScratchPad) {\n self.input = input\n self.thoughts = thoughts\n }\n}\n\nextension AgentInput: Equatable where T: Equatable {}\n\npublic enum AgentEarlyStopHandleType: Equatable {\n case force\n case generate\n}\n\npublic protocol Agent {\n associatedtype Input\n associatedtype Output: AgentOutputParsable\n associatedtype ScratchPadContent: Equatable\n var chatModelChain: ChatModelChain> { get }\n\n func validateTools(tools: [AgentTool]) throws\n func constructScratchpad(intermediateSteps: [AgentAction]) -> AgentScratchPad\n func constructFinalScratchpad(intermediateSteps: [AgentAction])\n -> AgentScratchPad\n func extraPlan(input: AgentInput)\n func parseOutput(_ output: ChatModelChain>.Output) async\n -> AgentNextStep\n}\n\npublic extension Agent {\n func getFullInputs(\n input: Input,\n intermediateSteps: [AgentAction]\n ) -> AgentInput {\n let thoughts = constructScratchpad(intermediateSteps: intermediateSteps)\n return AgentInput(input: input, thoughts: thoughts)\n }\n\n func plan(\n input: Input,\n intermediateSteps: [AgentAction],\n callbackManagers: [CallbackManager]\n ) async throws -> AgentNextStep {\n let input = getFullInputs(input: input, intermediateSteps: intermediateSteps)\n extraPlan(input: input)\n let output = try await chatModelChain.call(input, callbackManagers: callbackManagers)\n return await parseOutput(output)\n }\n\n func returnStoppedResponse(\n input: Input,\n earlyStoppedHandleType: AgentEarlyStopHandleType,\n intermediateSteps: [AgentAction],\n callbackManagers: [CallbackManager]\n ) async throws -> AgentFinish {\n switch earlyStoppedHandleType {\n case .force:\n return AgentFinish(\n returnValue: .unstructured(\"Agent stopped due to iteration limit or time limit.\"),\n log: \"\"\n )\n case .generate:\n let thoughts = constructFinalScratchpad(intermediateSteps: intermediateSteps)\n let input = AgentInput(input: input, thoughts: thoughts)\n let output = try await chatModelChain.call(input, callbackManagers: callbackManagers)\n let nextAction = await parseOutput(output)\n switch nextAction {\n case let .finish(finish):\n return finish\n case .actions:\n return .init(\n returnValue: .unstructured(output.content ?? \"\"),\n log: output.content ?? \"\"\n )\n }\n }\n }\n}\n\n" }, { "path": "Tool/Sources/LangChain/AgentExecutor.swift", "content": "import Foundation\n\npublic actor AgentExecutor: Chain\n where InnerAgent.Input == String, InnerAgent.Output: AgentOutputParsable\n{\n public typealias Input = String\n public struct Output {\n public typealias FinalOutput = AgentFinish.ReturnValue\n\n public let finalOutput: FinalOutput\n let intermediateSteps: [AgentAction]\n }\n\n let agent: InnerAgent\n let tools: [String: AgentTool]\n let maxIteration: Int?\n let maxExecutionTime: Double?\n var earlyStopHandleType: AgentEarlyStopHandleType\n var now: () -> Date = { Date() }\n var isCancelled = false\n var initialSteps: [AgentAction]\n\n public init(\n agent: InnerAgent,\n tools: [AgentTool],\n maxIteration: Int? = 10,\n maxExecutionTime: Double? = nil,\n earlyStopHandleType: AgentEarlyStopHandleType = .generate,\n initialSteps: [AgentAction] = []\n ) {\n self.agent = agent\n self.tools = tools.reduce(into: [:]) { $0[$1.name] = $1 }\n self.maxIteration = maxIteration\n self.maxExecutionTime = maxExecutionTime\n self.earlyStopHandleType = earlyStopHandleType\n self.initialSteps = initialSteps\n }\n\n public func callLogic(\n _ input: Input,\n callbackManagers: [CallbackManager]\n ) async throws -> Output {\n try agent.validateTools(tools: Array(tools.values))\n\n let startTime = now().timeIntervalSince1970\n var iterations = 0\n var intermediateSteps: [AgentAction] = initialSteps\n\n func shouldContinue() -> Bool {\n if isCancelled { return false }\n if let maxIteration = maxIteration, iterations >= maxIteration {\n return false\n }\n if let maxExecutionTime = maxExecutionTime,\n now().timeIntervalSince1970 - startTime > maxExecutionTime\n {\n return false\n }\n return true\n }\n\n while shouldContinue() {\n try Task.checkCancellation()\n let nextStepOutput = try await takeNextStep(\n input: input,\n intermediateSteps: intermediateSteps,\n callbackManagers: callbackManagers\n )\n\n try Task.checkCancellation()\n switch nextStepOutput {\n case let .finish(finish):\n return end(\n output: finish,\n intermediateSteps: intermediateSteps,\n callbackManagers: callbackManagers\n )\n case let .actions(actions):\n intermediateSteps.append(contentsOf: actions)\n if actions.count == 1,\n let action = actions.first,\n let toolFinish = getToolFinish(action: action)\n {\n return end(\n output: toolFinish,\n intermediateSteps: intermediateSteps,\n callbackManagers: callbackManagers\n )\n }\n }\n iterations += 1\n }\n\n let output = try await agent.returnStoppedResponse(\n input: input,\n earlyStoppedHandleType: earlyStopHandleType,\n intermediateSteps: intermediateSteps,\n callbackManagers: callbackManagers\n )\n return end(\n output: output,\n intermediateSteps: intermediateSteps,\n callbackManagers: callbackManagers\n )\n }\n\n public nonisolated func parseOutput(_ output: Output) -> String {\n switch output.finalOutput {\n case let .unstructured(error): return error\n case let .structured(output): return output.botReadableContent\n }\n }\n\n public func cancel() {\n isCancelled = true\n earlyStopHandleType = .force\n }\n}\n\nstruct InvalidToolError: Error {}\n\nextension AgentExecutor {\n func end(\n output: AgentFinish,\n intermediateSteps: [AgentAction],\n callbackManagers: [CallbackManager]\n ) -> Output {\n for callbackManager in callbackManagers {\n callbackManager.send(CallbackEvents.AgentDidFinish(info: output))\n }\n let finalOutput = output.returnValue\n return .init(finalOutput: finalOutput, intermediateSteps: intermediateSteps)\n }\n\n /// Plan the scratch pad and let the agent decide what to do next\n func takeNextStep(\n input: Input,\n intermediateSteps: [AgentAction],\n callbackManagers: [CallbackManager]\n ) async throws -> AgentNextStep {\n let output = try await agent.plan(\n input: input,\n intermediateSteps: intermediateSteps,\n callbackManagers: callbackManagers\n )\n switch output {\n // If the output says finish, then return the output immediately.\n case .finish: return output\n // If the output contains actions, run them, and append the results to the scratch pad.\n case let .actions(actions):\n let completedActions = try await withThrowingTaskGroup(of: AgentAction.self) {\n taskGroup in\n for action in actions {\n callbackManagers.send(CallbackEvents.AgentActionDidStart(info: action))\n if action.observation != nil {\n taskGroup.addTask { action }\n continue\n }\n guard let tool = tools[action.toolName] else { throw InvalidToolError() }\n taskGroup.addTask {\n do {\n let observation = try await tool.run(input: action.toolInput)\n return action.observationAvailable(observation)\n } catch {\n let observation = error.localizedDescription\n return action.observationAvailable(observation)\n }\n }\n }\n var completedActions = [AgentAction]()\n for try await action in taskGroup {\n try Task.checkCancellation()\n completedActions.append(action)\n callbackManagers.send(CallbackEvents.AgentActionDidEnd(info: action))\n }\n return completedActions\n }\n\n return .actions(completedActions)\n }\n }\n\n func getToolFinish(action: AgentAction) -> AgentFinish? {\n guard let tool = tools[action.toolName] else { return nil }\n guard tool.returnDirectly else { return nil }\n\n do {\n let result = try InnerAgent.Output.parse(action.observation ?? \"\")\n return .init(returnValue: .structured(result), log: action.observation ?? \"\")\n } catch {\n return .init(\n returnValue: .unstructured(action.observation ?? \"no observation\"),\n log: action.observation ?? \"\"\n )\n }\n }\n}\n\n// MARK: - AgentOutputParsable\n\npublic protocol AgentOutputParsable {\n static func parse(_ string: String) throws -> Self\n var botReadableContent: String { get }\n}\n\nextension String: AgentOutputParsable {\n public static func parse(_ string: String) throws -> String { string }\n public var botReadableContent: String { self }\n}\n\nextension Int: AgentOutputParsable {\n public static func parse(_ string: String) throws -> Int {\n guard let int = Int(string) else { return 0 }\n return int\n }\n\n public var botReadableContent: String { String(self) }\n}\n\nextension Double: AgentOutputParsable {\n public static func parse(_ string: String) throws -> Double {\n guard let double = Double(string) else { return 0 }\n return double\n }\n\n public var botReadableContent: String { String(self) }\n}\n\n" }, { "path": "Tool/Sources/LangChain/AgentTool.swift", "content": "import ChatBasic\nimport Foundation\nimport OpenAIService\n\npublic protocol AgentTool {\n var name: String { get }\n var description: String { get }\n var returnDirectly: Bool { get }\n func run(input: String) async throws -> String\n}\n\npublic struct SimpleAgentTool: AgentTool {\n public let name: String\n public let description: String\n public let returnDirectly: Bool\n public let run: (String) async throws -> String\n\n public init(\n name: String,\n description: String,\n returnDirectly: Bool = false,\n run: @escaping (String) async throws -> String\n ) {\n self.name = name\n self.description = description\n self.returnDirectly = returnDirectly\n self.run = run\n }\n\n public func run(input: String) async throws -> String {\n try await run(input)\n }\n}\n\npublic class FunctionCallingAgentTool: AgentTool, ChatGPTFunction {\n public func call(arguments: F.Arguments) async throws -> F.Result {\n try await function.call(arguments: arguments, reportProgress: reportProgress)\n }\n\n public var argumentSchema: ChatBasic.JSONSchemaValue { function.argumentSchema }\n\n public typealias Arguments = F.Arguments\n public typealias Result = F.Result\n\n public var function: F\n public var name: String\n public var description: String\n public var returnDirectly: Bool\n\n let callbackManagers: [CallbackManager]\n\n public init(\n function: F,\n returnDirectly: Bool = false,\n callbackManagers: [CallbackManager] = []\n ) {\n self.function = function\n self.callbackManagers = callbackManagers\n name = function.name\n description = function.description\n self.returnDirectly = returnDirectly\n }\n\n func reportProgress(_ progress: String) {\n callbackManagers.send(\n CallbackEvents.AgentFunctionCallingToolReportProgress(info: .init(\n functionName: name,\n progress: progress\n ))\n )\n }\n\n public func run(input: String) async throws -> String {\n await prepare(reportProgress: { [weak self] p in\n self?.reportProgress(p)\n })\n return try await call(\n argumentsJsonString: input,\n reportProgress: { [weak self] p in\n self?.reportProgress(p)\n }\n )\n .botReadableContent\n }\n\n public func prepare(reportProgress: @escaping ReportProgress) async {\n await function.prepare(reportProgress: { [weak self] p in\n self?.reportProgress(p)\n })\n }\n\n public func call(\n arguments: F.Arguments,\n reportProgress: @escaping ReportProgress\n ) async throws -> F.Result {\n try await function.call(arguments: arguments, reportProgress: reportProgress)\n }\n}\n\n" }, { "path": "Tool/Sources/LangChain/Agents/ChatAgent.swift", "content": "import Foundation\nimport Logger\nimport Parsing\n\nprivate func formatInstruction(toolsNames: String, preferredLanguage: String) -> String {\n \"\"\"\n The way you use the tools is by specifying a json blob.\n Specifically, this json should have a `action` key (with the name of the tool to use) and a `action_input` key (with the input to the tool going here).\n\n The only values that should be in the \"action\" field are: \\(toolsNames)\n\n The $JSON_BLOB should only contain a SINGLE action, do NOT return a list of multiple actions. Here is an example of a valid $JSON_BLOB:\n\n ```\n {\n \"action\": $TOOL_NAME,\n \"action_input\": $INPUT\n }\n ```\n\n ALWAYS use the following format:\n\n Question: the input question you must answer\n Thought: you should always think about what to do\n Action:\n ```\n $JSON_BLOB\n ```\n Observation: the result of the action\n ... (this Thought/Action/Observation can repeat N times)\n Thought: I now know the final answer\n Final Answer: the final answer to the original input question \\(preferredLanguage)\n \"\"\"\n}\n\npublic class ChatAgent: Agent {\n public typealias Input = String\n public typealias Output = String\n public typealias ScratchPadContent = String\n public var observationPrefix: String { \"Observation: \" }\n public var llmPrefix: String { \"Thought: \" }\n public let chatModelChain: ChatModelChain>\n let tools: [AgentTool]\n\n public init(chatModel: ChatModel, tools: [AgentTool], preferredLanguage: String) {\n self.tools = tools\n chatModelChain = .init(\n chatModel: chatModel,\n stops: [\"Observation:\"],\n promptTemplate: { agentInput in\n [\n .init(\n role: .system,\n content: \"\"\"\n Respond to the human as helpfully and accurately as possible. \\\n Wrap any code block in thought in . \\\n Format final answer to be more readable, in a ordered list if possible. \\\n You have access to the following tools:\n\n \\(tools.map { \"\\($0.name): \\($0.description)\" }.joined(separator: \"\\n\"))\n\n \\(formatInstruction(\n toolsNames: tools.map(\\.name).joined(separator: \",\"),\n preferredLanguage: preferredLanguage.isEmpty\n ? \"\"\n : \"(in \\(preferredLanguage)\"\n ))\n\n Begin! Reminder to always use the exact characters `Final Answer` when responding.\n \"\"\"\n ),\n agentInput.thoughts.content.isEmpty\n ? .init(role: .user, content: agentInput.input)\n : .init(\n role: .user,\n content: \"\"\"\n \\(agentInput.input)\n\n \\(agentInput.thoughts.content)\n \"\"\"\n ),\n ]\n }\n )\n }\n \n func constructBaseScratchpad(intermediateSteps: [AgentAction]) -> String {\n var thoughts = \"\"\n for step in intermediateSteps {\n thoughts += \"\"\"\n \\(step.log)\n \\(observationPrefix)\\(step.observation ?? \"\")\n \"\"\"\n }\n return thoughts\n }\n\n public func constructScratchpad(intermediateSteps: [AgentAction]) -> AgentScratchPad {\n let baseScratchpad = constructBaseScratchpad(intermediateSteps: intermediateSteps)\n if baseScratchpad.isEmpty { return .init(content: \"\") }\n return .init(content: \"\"\"\n This was your previous work (but I haven't seen any of it! I only see what you return as `Final Answer`):\n \\(baseScratchpad)\n (Please continue with `Thought:` or `Final Answer:`)\n \"\"\")\n }\n \n public func constructFinalScratchpad(intermediateSteps: [AgentAction]) -> AgentScratchPad {\n let baseScratchpad = constructBaseScratchpad(intermediateSteps: intermediateSteps)\n if baseScratchpad.isEmpty { return .init(content: \"\") }\n return .init(content: \"\"\"\n This was your previous work (but I haven't seen any of it! I only see what you return as `Final Answer`):\n \\(baseScratchpad)\n \\(llmPrefix)I now need to return a final answer based on the previous steps:\n \"(Please continue with `Final Answer:`)\"\n \"\"\")\n }\n\n public func validateTools(tools: [AgentTool]) throws {\n // no validation\n }\n\n public func extraPlan(input: AgentInput) {\n // do nothing\n }\n\n public func parseOutput(_ output: ChatMessage) async -> AgentNextStep {\n let text = output.content ?? \"\"\n\n func parseFinalAnswerIfPossible() -> AgentNextStep? {\n let throughAnswerParser = PrefixThrough(\"Final Answer:\")\n var parsableContent = text[...]\n do {\n _ = try throughAnswerParser.parse(&parsableContent)\n let answer = String(parsableContent)\n let output = answer.trimmingCharacters(in: .whitespacesAndNewlines)\n return .finish(AgentFinish(returnValue: .structured(output), log: text))\n } catch {\n Logger.langchain.info(\"Could not parse LLM output final answer: \\(error)\")\n return nil\n }\n }\n\n func parseNextActionIfPossible() -> AgentNextStep? {\n let throughActionBlockParser = PrefixThrough(\"\"\"\n Action:\n ```\n \"\"\")\n let throughActionBlockSimplifiedParser = PrefixThrough(\"```\")\n let jsonBlobParser = PrefixUpTo(\"```\")\n var parsableContent = text[...]\n do {\n let actionBlockPrefix = try? throughActionBlockParser.parse(&parsableContent)\n if actionBlockPrefix == nil {\n _ = try throughActionBlockSimplifiedParser.parse(&parsableContent)\n }\n let jsonBlob = try jsonBlobParser.parse(&parsableContent)\n\n struct Action: Codable {\n let action: String\n let action_input: String\n }\n let response = try JSONDecoder()\n .decode(Action.self, from: jsonBlob.data(using: .utf8) ?? Data())\n return .actions([\n AgentAction(\n toolName: response.action,\n toolInput: response.action_input,\n log: text\n ),\n ])\n } catch {\n Logger.langchain.info(\"Could not parse LLM output next action: \\(error)\")\n return nil\n }\n }\n\n if let step = parseFinalAnswerIfPossible() { return step }\n if let step = parseNextActionIfPossible() { return step }\n\n let forceParser = PrefixUpTo(\"Action:\")\n var parsableContent = text[...]\n let finalAnswer = try? forceParser.parse(&parsableContent)\n .trimmingCharacters(in: .whitespacesAndNewlines)\n\n var answer = finalAnswer ?? text\n if answer.isEmpty {\n answer = \"Sorry, I don't know.\"\n }\n\n return .finish(AgentFinish(returnValue: .structured(String(answer)), log: text))\n }\n}\n\n" }, { "path": "Tool/Sources/LangChain/Callback.swift", "content": "import Foundation\n\npublic protocol CallbackEvent {\n associatedtype Info\n var info: Info { get }\n}\n\npublic struct CallbackEvents {\n public struct UnTypedEvent: CallbackEvent {\n public var info: String\n public init(info: String) {\n self.info = info\n }\n }\n \n public var untyped: UnTypedEvent.Type { UnTypedEvent.self }\n \n private init() {}\n}\n\npublic struct CallbackManager {\n struct Observer {\n let handler: (Event.Info) -> Void\n }\n \n fileprivate var observers = [Any]()\n\n public init() {}\n\n public init(observers: (inout CallbackManager) -> Void) {\n var manager = CallbackManager()\n observers(&manager)\n self = manager\n }\n\n public mutating func on(\n _: Event.Type = Event.self,\n _ handler: @escaping (Event.Info) -> Void\n ) {\n observers.append(Observer(handler: handler))\n }\n\n public mutating func on(\n _: KeyPath,\n _ handler: @escaping (Event.Info) -> Void\n ) {\n observers.append(Observer(handler: handler))\n }\n\n public func send(_ event: Event) {\n for case let observer as Observer in observers {\n observer.handler(event.info)\n }\n }\n\n func send(\n _: KeyPath,\n _ info: Event.Info\n ) {\n for case let observer as Observer in observers {\n observer.handler(info)\n }\n }\n \n public func send(_ string: String) {\n for case let observer as Observer in observers {\n observer.handler(string)\n }\n }\n}\n\npublic extension [CallbackManager] {\n func send(_ event: Event) {\n for cb in self { cb.send(event) }\n }\n\n func send(\n _ keyPath: KeyPath,\n _ info: Event.Info\n ) {\n for cb in self { cb.send(keyPath, info) }\n }\n \n func send(_ event: String) {\n for cb in self { cb.send(event) }\n }\n}\n\n" }, { "path": "Tool/Sources/LangChain/Chain.swift", "content": "import Foundation\n\npublic protocol Chain {\n associatedtype Input\n associatedtype Output\n func callLogic(_ input: Input, callbackManagers: [CallbackManager]) async throws -> Output\n func parseOutput(_ output: Output) -> String\n}\n\npublic extension Chain {\n typealias ChainDidStart = CallbackEvents.ChainDidStart\n typealias ChainDidEnd = CallbackEvents.ChainDidEnd\n\n func run(_ input: Input, callbackManagers: [CallbackManager] = []) async throws -> String {\n let output = try await call(input, callbackManagers: callbackManagers)\n return parseOutput(output)\n }\n\n func call(_ input: Input, callbackManagers: [CallbackManager] = []) async throws -> Output {\n callbackManagers\n .send(CallbackEvents.ChainDidStart(info: (type: Self.self, input: input)))\n defer {\n callbackManagers\n .send(CallbackEvents.ChainDidEnd(info: (type: Self.self, input: input)))\n }\n return try await callLogic(input, callbackManagers: callbackManagers)\n }\n}\n\npublic extension CallbackEvents {\n struct ChainDidStart: CallbackEvent {\n public let info: (type: T.Type, input: T.Input)\n }\n\n struct ChainDidEnd: CallbackEvent {\n public let info: (type: T.Type, input: T.Input)\n }\n}\n\npublic struct SimpleChain: Chain {\n let block: (Input) async throws -> Output\n let parseOutputBlock: (Output) -> String\n\n public init(\n block: @escaping (Input) async throws -> Output,\n parseOutput: @escaping (Output) -> String = { String(describing: $0) }\n ) {\n self.block = block\n parseOutputBlock = parseOutput\n }\n\n public func callLogic(\n _ input: Input,\n callbackManagers: [CallbackManager]\n ) async throws -> Output {\n return try await block(input)\n }\n\n public func parseOutput(_ output: Output) -> String {\n return parseOutputBlock(output)\n }\n}\n\npublic struct ConnectedChain: Chain where B.Input == A.Output {\n public typealias Input = A.Input\n public typealias Output = (B.Output, A.Output)\n\n public let chainA: A\n public let chainB: B\n\n public func callLogic(\n _ input: Input,\n callbackManagers: [CallbackManager] = []\n ) async throws -> Output {\n let a = try await chainA.call(input, callbackManagers: callbackManagers)\n let b = try await chainB.call(a, callbackManagers: callbackManagers)\n return (b, a)\n }\n\n public func parseOutput(_ output: Output) -> String {\n chainB.parseOutput(output.0)\n }\n}\n\npublic struct PairedChain: Chain {\n public typealias Input = (A.Input, B.Input)\n public typealias Output = (A.Output, B.Output)\n\n public let chainA: A\n public let chainB: B\n\n public func callLogic(\n _ input: Input,\n callbackManagers: [CallbackManager] = []\n ) async throws -> Output {\n async let a = chainA.call(input.0, callbackManagers: callbackManagers)\n async let b = chainB.call(input.1, callbackManagers: callbackManagers)\n return try await (a, b)\n }\n\n public func parseOutput(_ output: (A.Output, B.Output)) -> String {\n [chainA.parseOutput(output.0), chainB.parseOutput(output.1)].joined(separator: \"\\n\")\n }\n}\n\npublic struct MappedChain: Chain {\n public typealias Input = A.Input\n public typealias Output = NewOutput\n\n public let chain: A\n public let map: (A.Output) -> NewOutput\n\n public func callLogic(\n _ input: Input,\n callbackManagers: [CallbackManager]\n ) async throws -> Output {\n let output = try await chain.call(input, callbackManagers: callbackManagers)\n return map(output)\n }\n\n public func parseOutput(_ output: Output) -> String {\n String(describing: output)\n }\n}\n\npublic extension Chain {\n func pair(with another: C) -> PairedChain {\n PairedChain(chainA: self, chainB: another)\n }\n\n func chain(to another: C) -> ConnectedChain {\n ConnectedChain(chainA: self, chainB: another)\n }\n\n func map(_ map: @escaping (Output) -> NewOutput) -> MappedChain {\n MappedChain(chain: self, map: map)\n }\n}\n\n" }, { "path": "Tool/Sources/LangChain/Chains/CombineAnswersChain.swift", "content": "import Foundation\nimport Logger\nimport OpenAIService\nimport Preferences\n\npublic class CombineAnswersChain: Chain {\n public struct Input: Decodable {\n public var question: String\n public var answers: [String]\n public init(question: String, answers: [String]) {\n self.question = question\n self.answers = answers\n }\n }\n\n public typealias Output = String\n public let chatModelChain: ChatModelChain\n\n public init(\n configuration: ChatGPTConfiguration =\n UserPreferenceChatGPTConfiguration(chatModelKey: \\.preferredChatModelIdForUtilities),\n extraInstructions: String = \"\"\n ) {\n chatModelChain = .init(\n chatModel: OpenAIChat(\n configuration: configuration.overriding {\n $0.runFunctionsAutomatically = false\n },\n memory: nil,\n stream: false\n ),\n stops: [\"Observation:\"],\n promptTemplate: { input in\n [\n .init(\n role: .system,\n content: \"\"\"\n You are a helpful assistant.\n Your job is to combine multiple answers from different sources to one question.\n \\(extraInstructions)\n \"\"\"\n ),\n .init(role: .user, content: \"\"\"\n Question: \\(input.question)\n\n Answers:\n \\(input.answers.joined(separator: \"\\n\\(String(repeating: \"-\", count: 32))\\n\"))\n\n What is the combined answer?\n \"\"\"),\n ]\n }\n )\n }\n\n public func callLogic(\n _ input: Input,\n callbackManagers: [CallbackManager]\n ) async throws -> String {\n let output = try await chatModelChain.call(input, callbackManagers: callbackManagers)\n return await parseOutput(output)\n }\n\n public func parseOutput(_ message: ChatMessage) async -> String {\n return message.content ?? \"No answer.\"\n }\n\n public func parseOutput(_ output: String) -> String {\n output\n }\n}\n\n" }, { "path": "Tool/Sources/LangChain/Chains/LLMChain.swift", "content": "import Foundation\n\npublic class ChatModelChain: Chain {\n public typealias Output = ChatMessage\n\n public internal(set) var chatModel: ChatModel\n public internal(set) var promptTemplate: (Input) -> [ChatMessage]\n public internal(set) var stops: [String]\n\n public init(\n chatModel: ChatModel,\n stops: [String] = [],\n promptTemplate: @escaping (Input) -> [ChatMessage]\n ) {\n self.chatModel = chatModel\n self.promptTemplate = promptTemplate\n self.stops = stops\n }\n\n public func callLogic(\n _ input: Input,\n callbackManagers: [CallbackManager]\n ) async throws -> Output {\n let prompt = promptTemplate(input)\n let output = try await chatModel.generate(\n prompt: prompt,\n stops: stops,\n callbackManagers: callbackManagers\n )\n return output\n }\n\n public func parseOutput(_ output: Output) -> String {\n if let content = output.content {\n return content\n } else if let toolCalls = output.toolCalls {\n return toolCalls.map { \"[\\($0.id)] \\($0.function.name): \\($0.function.arguments)\" }\n .joined(separator: \"\\n\")\n }\n\n return \"\"\n }\n}\n\n" }, { "path": "Tool/Sources/LangChain/Chains/QAInformationRetrievalChain.swift", "content": "import Foundation\nimport OpenAIService\n\npublic final class QAInformationRetrievalChain: Chain {\n let vectorStores: [VectorStore]\n let embedding: Embeddings\n let maxCount: Int\n let filterMetadata: (String) -> Bool\n let hint: String\n\n public struct Output {\n public var information: String\n public var sourceDocuments: [Document]\n public var distance: [Float]\n }\n\n public init(\n vectorStore: VectorStore,\n embedding: Embeddings,\n maxCount: Int = 5,\n filterMetadata: @escaping (String) -> Bool = { _ in true },\n hint: String = \"\"\n ) {\n vectorStores = [vectorStore]\n self.embedding = embedding\n self.maxCount = maxCount\n self.filterMetadata = filterMetadata\n self.hint = hint\n }\n\n public init(\n vectorStores: [VectorStore],\n embedding: Embeddings,\n maxCount: Int = 5,\n filterMetadata: @escaping (String) -> Bool = { _ in true },\n hint: String = \"\"\n ) {\n self.vectorStores = vectorStores\n self.embedding = embedding\n self.maxCount = maxCount\n self.filterMetadata = filterMetadata\n self.hint = hint\n }\n\n public func callLogic(\n _ input: String,\n callbackManagers: [CallbackManager]\n ) async throws -> Output {\n let embeddedQuestion = try await embedding.embed(query: input)\n let documentsSlice = await withTaskGroup(\n of: [(document: Document, distance: Float)].self\n ) { group in\n for vectorStore in vectorStores {\n group.addTask {\n (try? await vectorStore.searchWithDistance(\n embeddings: embeddedQuestion,\n count: 5\n ).filter { item in\n item.distance < 0.31\n }) ?? []\n }\n }\n var result = [(document: Document, distance: Float)]()\n for await items in group {\n result.append(contentsOf: items)\n }\n return result\n }.sorted { $0.distance < $1.distance }.prefix(maxCount)\n\n let documents = Array(documentsSlice)\n\n callbackManagers.send(CallbackEvents.RetrievalQADidExtractRelevantContent(info: documents))\n\n let relevantInformationChain = RelevantInformationExtractionChain(\n filterMetadata: filterMetadata,\n hint: hint\n )\n let relevantInformation = try await relevantInformationChain.run(\n .init(question: input, documents: documents),\n callbackManagers: callbackManagers\n )\n\n return .init(\n information: relevantInformation,\n sourceDocuments: documents.map(\\.document),\n distance: documents.map(\\.distance)\n )\n }\n\n public func parseOutput(_ output: Output) -> String {\n return output.information\n }\n}\n\npublic extension CallbackEvents {\n struct RetrievalQADidExtractRelevantContent: CallbackEvent {\n public let info: [(document: Document, distance: Float)]\n }\n\n var retrievalQADidExtractRelevantContent: RetrievalQADidExtractRelevantContent.Type {\n RetrievalQADidExtractRelevantContent.self\n }\n}\n\n" }, { "path": "Tool/Sources/LangChain/Chains/RefineDocumentChain.swift", "content": "import ChatBasic\nimport Foundation\nimport OpenAIService\nimport Preferences\n\npublic final class RefineDocumentChain: Chain {\n public struct Input {\n var question: String\n var documents: [(document: Document, distance: Float)]\n }\n\n struct RefinementInput {\n var index: Int\n var totalCount: Int\n var question: String\n var previousAnswer: String?\n var document: String\n var distance: Float\n }\n\n public struct IntermediateAnswer: Decodable {\n public var answer: String\n public var usefulness: Double\n public var more: Bool\n\n public enum CodingKeys: String, CodingKey {\n case answer\n case usefulness\n case more\n }\n\n init(answer: String, usefulness: Double, more: Bool) {\n self.answer = answer\n self.usefulness = usefulness\n self.more = more\n }\n\n public init(from decoder: Decoder) throws {\n let container = try decoder.container(keyedBy: CodingKeys.self)\n answer = try container.decode(String.self, forKey: .answer)\n usefulness = (try? container.decode(Double.self, forKey: .usefulness)) ?? 0\n more = (try? container.decode(Bool.self, forKey: .more)) ?? true\n }\n }\n\n class FunctionProvider: ChatGPTFunctionProvider {\n var functionCallStrategy: FunctionCallStrategy? = .function(name: \"respond\")\n var functions: [any ChatGPTFunction] = [RespondFunction()]\n }\n\n struct RespondFunction: ChatGPTArgumentsCollectingFunction {\n typealias Arguments = IntermediateAnswer\n var name: String = \"respond\"\n var description: String = \"Respond with the refined answer\"\n var argumentSchema: JSONSchemaValue {\n return [\n .type: \"object\",\n .properties: [\n \"answer\": [\n .type: \"string\",\n .description: \"The refined answer\",\n ],\n \"usefulness\": [\n .type: \"number\",\n .description: \"How useful the page of document is in generating the answer, the higher the better. 0 to 10\",\n ],\n \"more\": [\n .type: \"boolean\",\n .description: \"Whether you want to read the next page. The next page maybe less relevant to the question\",\n ],\n ],\n .required: [\"answer\", \"more\", \"usefulness\"],\n ]\n }\n }\n\n func buildChatModel() -> ChatModelChain {\n .init(\n chatModel: OpenAIChat(\n configuration: UserPreferenceChatGPTConfiguration(\n chatModelKey: \\.preferredChatModelIdForUtilities\n )\n .overriding {\n $0.temperature = 0\n $0.runFunctionsAutomatically = false\n },\n memory: EmptyChatGPTMemory(),\n functionProvider: FunctionProvider(),\n stream: false\n ),\n promptTemplate: { input in [\n .init(\n role: .system,\n content: {\n if let previousAnswer = input.previousAnswer {\n return \"\"\"\n I will send you a question about a document, you must refine your previous answer to it only according to the document.\n Previous answer:###\n \\(previousAnswer)\n ###\n Page \\(input.index) of \\(input.totalCount) of the document:###\n \\(input.document)\n ###\n \"\"\"\n } else {\n return \"\"\"\n I will send you a question about a document, you must answer it only according to the document.\n Page \\(input.index) of \\(input.totalCount) of the document:###\n \\(input.document)\n ###\n \"\"\"\n }\n }()\n\n ),\n .init(role: .user, content: input.question),\n ] }\n )\n }\n\n public init() {}\n\n public func callLogic(\n _ input: Input,\n callbackManagers: [CallbackManager]\n ) async throws -> String {\n var intermediateAnswer: IntermediateAnswer?\n\n for (index, document) in input.documents.enumerated() {\n if let intermediateAnswer, !intermediateAnswer.more { break }\n\n let output = try await buildChatModel().call(\n .init(\n index: index,\n totalCount: input.documents.count,\n question: input.question,\n previousAnswer: intermediateAnswer?.answer,\n document: document.document.pageContent,\n distance: document.distance\n ),\n callbackManagers: callbackManagers\n )\n intermediateAnswer = extractAnswer(output)\n\n if let intermediateAnswer {\n callbackManagers.send(\n \\.refineDocumentChainDidGenerateIntermediateAnswer,\n intermediateAnswer\n )\n }\n }\n\n return intermediateAnswer?.answer ?? \"None\"\n }\n\n public func parseOutput(_ output: String) -> String {\n return output\n }\n\n func extractAnswer(_ chatMessage: ChatMessage) -> IntermediateAnswer {\n for functionCall in chatMessage.toolCalls?.map(\\.function) ?? [] {\n do {\n let intermediateAnswer = try JSONDecoder().decode(\n IntermediateAnswer.self,\n from: functionCall.arguments.data(using: .utf8) ?? Data()\n )\n return intermediateAnswer\n } catch {\n let intermediateAnswer = IntermediateAnswer(\n answer: functionCall.arguments,\n usefulness: 0,\n more: true\n )\n return intermediateAnswer\n }\n }\n return .init(answer: chatMessage.content ?? \"\", usefulness: 0, more: true)\n }\n}\n\npublic extension CallbackEvents {\n struct RefineDocumentChainDidGenerateIntermediateAnswer: CallbackEvent {\n public let info: RefineDocumentChain.IntermediateAnswer\n }\n\n var refineDocumentChainDidGenerateIntermediateAnswer:\n RefineDocumentChainDidGenerateIntermediateAnswer.Type\n {\n RefineDocumentChainDidGenerateIntermediateAnswer.self\n }\n}\n\n" }, { "path": "Tool/Sources/LangChain/Chains/RelevantInformationExtractionChain.swift", "content": "import ChatBasic\nimport Foundation\nimport OpenAIService\nimport Preferences\n\npublic final class RelevantInformationExtractionChain: Chain {\n public struct Input {\n var question: String\n var documents: [(document: Document, distance: Float)]\n }\n\n struct TaskInput {\n var question: String\n var document: Document\n }\n\n public typealias Output = String\n\n class FunctionProvider: ChatGPTFunctionProvider {\n var functionCallStrategy: FunctionCallStrategy? = .function(name: \"saveFinalAnswer\")\n var functions: [any ChatGPTFunction] = [FinalAnswer()]\n }\n\n struct FinalAnswer: ChatGPTArgumentsCollectingFunction {\n struct Arguments: Decodable {\n var relevantInformation: String\n var noRelevantInformationFound: Bool?\n }\n\n var name: String = \"saveFinalAnswer\"\n var description: String =\n \"save the relevant information\"\n var argumentSchema: JSONSchemaValue {\n [\n .type: \"object\",\n .properties: [\n \"relevantInformation\": [.type: \"string\"],\n \"noRelevantInformationFound\": [.type: \"boolean\"],\n ],\n .required: [\"relevantInformation\", \"noRelevantInformationFound\"],\n ]\n }\n }\n\n let filterMetadata: (String) -> Bool\n let hint: String\n\n init(filterMetadata: @escaping (String) -> Bool = { _ in true }, hint: String) {\n self.filterMetadata = filterMetadata\n self.hint = hint\n }\n\n func buildChatModel() -> ChatModelChain {\n .init(\n chatModel: OpenAIChat(\n configuration: UserPreferenceChatGPTConfiguration(\n chatModelKey: \\.preferredChatModelIdForUtilities\n )\n .overriding {\n $0.temperature = 0.5\n $0.runFunctionsAutomatically = false\n },\n memory: EmptyChatGPTMemory(),\n functionProvider: FunctionProvider(),\n stream: false\n )\n ) { [filterMetadata, hint] input in [\n .init(\n role: .system,\n content: \"\"\"\n Extract the relevant information from the Document according to the Question.\n The information may not directly answer the question, but it should be relevant to the question, \\\n please think carefully and make you decision.\n Make the information clear, concise and short.\n If found code, wrap it in markdown code block.\n \\(hint)\n \"\"\"\n ),\n .init(\n role: .user,\n content: \"\"\"\n Question:###\n (how, when, what or why)\n \\(input.question)\n ###\n Document:###\n \\(input.document.metadata.filter { key, _ in\n filterMetadata(key)\n })\n \\(input.document.pageContent)\n ###\n \"\"\"\n ),\n ] }\n }\n\n public func callLogic(\n _ input: Input,\n callbackManagers: [CallbackManager]\n ) async throws -> Output {\n await withTaskGroup(of: String.self) { group in\n for document in input.documents {\n let taskInput = TaskInput(question: input.question, document: document.document)\n group.addTask {\n func run() async throws -> String {\n let model = self.buildChatModel()\n let output = try await model.call(\n taskInput,\n callbackManagers: callbackManagers\n )\n\n if let functionCall = output.toolCalls?\n .first(where: { $0.function.name == FinalAnswer().name })?.function\n {\n do {\n let arguments = try JSONDecoder().decode(\n FinalAnswer.Arguments.self,\n from: functionCall.arguments.data(using: .utf8) ?? Data()\n )\n if arguments.noRelevantInformationFound ?? false {\n return \"\"\n }\n return arguments.relevantInformation\n } catch {\n return output.content ?? \"\"\n }\n }\n\n return output.content ?? \"\"\n }\n\n var repeatCount = 0\n while repeatCount < 3 {\n do {\n return try await run()\n } catch {\n repeatCount += 1\n }\n }\n return \"\"\n }\n }\n\n var results = [String]()\n for await output in group where !output.isEmpty {\n callbackManagers.send(\n \\.relevantInformationExtractionChainDidExtractPartialRelevantContent,\n output\n )\n let trimmed = output.trimmingCharacters(in: .whitespacesAndNewlines)\n if results.contains(trimmed) { continue }\n results.append(trimmed)\n }\n if results.isEmpty { return \"No information found.\" }\n return results.joined(separator: \"\")\n }\n }\n\n public func parseOutput(_ output: Output) -> String {\n return output\n }\n}\n\npublic extension CallbackEvents {\n struct RelevantInformationExtractionChainDidExtractPartialRelevantContent: CallbackEvent {\n public let info: String\n }\n\n var relevantInformationExtractionChainDidExtractPartialRelevantContent:\n RelevantInformationExtractionChainDidExtractPartialRelevantContent.Type\n {\n RelevantInformationExtractionChainDidExtractPartialRelevantContent.self\n }\n}\n\n" }, { "path": "Tool/Sources/LangChain/Chains/StructuredOutputChatModelChain.swift", "content": "import ChatBasic\nimport Foundation\nimport Logger\nimport OpenAIService\n\n/// This is an agent used to get a structured output.\npublic class StructuredOutputChatModelChain: Chain {\n public struct EndFunction: ChatGPTArgumentsCollectingFunction {\n public struct Arguments: Decodable {\n var finalAnswer: Output\n }\n\n public var name: String { \"FinalAnswer\" }\n public var description: String { \"Save the final answer when it's ready\" }\n public var argumentSchema: JSONSchemaValue {\n return [\n .type: \"object\",\n .properties: [\n \"finalAnswer\": .hash(finalAnswerSchema),\n ],\n .required: [\"finalAnswer\"],\n ]\n }\n\n public let finalAnswerSchema: [String: JSONSchemaValue]\n\n public init(argumentSchema: [String: JSONSchemaValue]) {\n finalAnswerSchema = argumentSchema\n }\n\n public init() where Output == String {\n finalAnswerSchema = [\n JSONSchemaKey.type.key: \"string\",\n ]\n }\n\n public init() where Output == Int {\n finalAnswerSchema = [\n JSONSchemaKey.type.key: \"number\",\n ]\n }\n\n public init() where Output == Double {\n finalAnswerSchema = [\n JSONSchemaKey.type.key: \"number\",\n ]\n }\n }\n\n struct FunctionProvider: ChatGPTFunctionProvider {\n var endFunction: EndFunction\n var functions: [any ChatGPTFunction] {\n [endFunction]\n }\n\n var functionCallStrategy: FunctionCallStrategy? {\n .function(name: endFunction.name)\n }\n }\n\n public typealias Input = String\n public let chatModelChain: ChatModelChain\n var functionProvider: FunctionProvider\n\n public init(\n configuration: ChatGPTConfiguration = UserPreferenceChatGPTConfiguration(),\n endFunction: EndFunction,\n promptTemplate: ((String) -> [ChatMessage])? = nil\n ) {\n functionProvider = .init(\n endFunction: endFunction\n )\n chatModelChain = .init(\n chatModel: OpenAIChat(\n configuration: configuration.overriding {\n $0.runFunctionsAutomatically = false\n },\n memory: nil,\n functionProvider: functionProvider,\n stream: false\n ),\n stops: [\"Observation:\"],\n promptTemplate: promptTemplate ?? { input in\n [\n .init(\n role: .system,\n content: \"\"\"\n You are a helpful assistant\n Generate a final answer to my query as concisely, helpfully and accurately as possible.\n You don't ask me for additional information.\n \"\"\"\n ),\n .init(role: .user, content: input),\n ]\n }\n )\n }\n\n public func callLogic(\n _ input: String,\n callbackManagers: [CallbackManager]\n ) async throws -> Output? {\n let output = try await chatModelChain.call(input, callbackManagers: callbackManagers)\n return await parseOutput(output)\n }\n\n public func parseOutput(_ output: Output?) -> String {\n return String(describing: output)\n }\n\n public func parseOutput(_ message: ChatMessage) async -> Output? {\n if let functionCall = message.toolCalls?.first?.function {\n do {\n let result = try JSONDecoder().decode(\n EndFunction.Arguments.self,\n from: functionCall.arguments.data(using: .utf8) ?? Data()\n )\n return result.finalAnswer\n } catch {\n return nil\n }\n }\n\n return nil\n }\n}\n\n" }, { "path": "Tool/Sources/LangChain/ChatModel/ChatModel.swift", "content": "import Foundation\nimport OpenAIService\n\npublic protocol ChatModel {\n func generate(\n prompt: [ChatMessage],\n stops: [String],\n callbackManagers: [CallbackManager]\n ) async throws -> ChatMessage\n}\n\npublic typealias ChatMessage = OpenAIService.ChatMessage\n \npublic extension CallbackEvents {\n struct LLMDidProduceNewToken: CallbackEvent {\n public let info: String\n }\n \n var llmDidProduceNewToken: LLMDidProduceNewToken.Type {\n LLMDidProduceNewToken.self\n }\n}\n" }, { "path": "Tool/Sources/LangChain/ChatModel/OpenAIChat.swift", "content": "import Foundation\nimport OpenAIService\n\npublic struct OpenAIChat: ChatModel {\n public var configuration: ChatGPTConfiguration\n public var memory: ChatGPTMemory?\n public var functionProvider: ChatGPTFunctionProvider\n public var stream: Bool\n\n public init(\n configuration: ChatGPTConfiguration = UserPreferenceChatGPTConfiguration(),\n memory: ChatGPTMemory? = ConversationChatGPTMemory(systemPrompt: \"\"),\n functionProvider: ChatGPTFunctionProvider = NoChatGPTFunctionProvider(),\n stream: Bool\n ) {\n self.configuration = configuration\n self.memory = memory\n self.functionProvider = functionProvider\n self.stream = stream\n }\n\n public func generate(\n prompt: [ChatMessage],\n stops: [String],\n callbackManagers: [CallbackManager]\n ) async throws -> ChatMessage {\n let memory = memory ?? EmptyChatGPTMemory()\n \n let service = LegacyChatGPTService(\n memory: memory,\n configuration: configuration,\n functionProvider: functionProvider\n )\n for message in prompt {\n await memory.appendMessage(message)\n }\n\n if stream {\n let stream = try await service.send(content: \"\")\n var message = \"\"\n for try await chunk in stream {\n message.append(chunk)\n callbackManagers.send(CallbackEvents.LLMDidProduceNewToken(info: chunk))\n }\n return await memory.history.last ?? .init(role: .assistant, content: \"\")\n } else {\n let _ = try await service.sendAndWait(content: \"\")\n return await memory.history.last ?? .init(role: .assistant, content: \"\")\n }\n }\n}\n\n" }, { "path": "Tool/Sources/LangChain/DocumentLoader/DocumentLoader.swift", "content": "import Foundation\nimport JSONRPC\n\npublic struct Document: Codable {\n public typealias Metadata = [String: JSONValue]\n public var pageContent: String\n public var metadata: Metadata\n public init(pageContent: String, metadata: Metadata) {\n self.pageContent = pageContent\n self.metadata = metadata\n }\n \n public func metadata(_ keyPath: KeyPath) -> JSONValue? {\n let key = Key.self[keyPath: keyPath]\n return metadata[key]\n }\n}\n\npublic protocol DocumentLoader {\n func load() async throws -> [Document]\n}\n\nextension DocumentLoader {\n func loadAndSplit(\n with textSplitter: TextSplitter = RecursiveCharacterTextSplitter()\n ) async throws -> [Document] {\n let docs = try await load()\n return try await textSplitter.splitDocuments(docs)\n }\n}\n\n" }, { "path": "Tool/Sources/LangChain/DocumentLoader/TextLoader.swift", "content": "import AppKit\nimport Foundation\n\n/// Load a text document from local file.\npublic struct TextLoader: DocumentLoader {\n public enum MetadataKeys {\n public static let filename = \"filename\"\n public static let `extension` = \"extension\"\n public static let contentModificationDate = \"contentModificationDate\"\n public static let filePath = \"filePath\"\n }\n \n let url: URL\n let encoding: String.Encoding\n let options: [NSAttributedString.DocumentReadingOptionKey: Any]\n\n public init(\n url: URL,\n encoding: String.Encoding = .utf8,\n options: [NSAttributedString.DocumentReadingOptionKey: Any] = [:]\n ) {\n self.url = url\n self.encoding = encoding\n self.options = options\n }\n\n public func load() async throws -> [Document] {\n let data = try Data(contentsOf: url)\n let attributedString = try NSAttributedString(\n data: data,\n options: options,\n documentAttributes: nil\n )\n let modificationDate = try? url.resourceValues(forKeys: [.contentModificationDateKey])\n .contentModificationDate\n return [Document(pageContent: attributedString.string, metadata: [\n MetadataKeys.filename: .string(url.lastPathComponent),\n MetadataKeys.extension: .string(url.pathExtension),\n MetadataKeys.contentModificationDate: .number(\n (modificationDate ?? Date()).timeIntervalSince1970\n ),\n MetadataKeys.filePath: .string(url.path),\n ])]\n }\n}\n\n" }, { "path": "Tool/Sources/LangChain/DocumentLoader/WebLoader.swift", "content": "import Foundation\nimport Logger\nimport SwiftSoup\nimport WebKit\n\n/// Load the body of a web page.\npublic struct WebLoader: DocumentLoader {\n enum MetadataKeys {\n static let title = \"title\"\n static let url = \"url\"\n static let date = \"date\"\n }\n\n var downloadHTML: (_ url: URL, _ strategy: LoadWebPageMainContentStrategy) async throws\n -> (url: URL, html: String, strategy: LoadWebPageMainContentStrategy) = { url, strategy in\n let html = try await WebScrapper(strategy: strategy).fetch(url: url)\n return (url, html, strategy)\n }\n\n public var urls: [URL]\n\n public init(urls: [URL]) {\n self.urls = urls\n }\n\n public init(url: URL) {\n urls = [url]\n }\n\n public func load() async throws -> [Document] {\n try await withThrowingTaskGroup(of: (\n url: URL,\n html: String,\n strategy: LoadWebPageMainContentStrategy\n ).self) { group in\n for url in urls {\n let strategy: LoadWebPageMainContentStrategy = {\n switch url {\n default:\n return DefaultLoadContentStrategy()\n }\n }()\n group.addTask {\n try await downloadHTML(url, strategy)\n }\n }\n var documents: [Document] = []\n for try await result in group {\n do {\n let parsed = try SwiftSoup.parse(result.html, result.url.path)\n\n let title = (try? parsed.title()) ?? \"Untitled\"\n let parsedDocuments = try result.strategy.load(\n parsed,\n metadata: [\n MetadataKeys.title: .string(title),\n MetadataKeys.url: .string(result.url.absoluteString),\n MetadataKeys.date: .number(Date().timeIntervalSince1970),\n ]\n )\n documents.append(contentsOf: parsedDocuments)\n } catch let Exception.Error(_, message) {\n Logger.langchain.error(message)\n } catch {\n Logger.langchain.error(error.localizedDescription)\n }\n }\n return documents\n }\n }\n}\n\n// MARK: - WebScrapper\n\n@MainActor\npublic final class WebScrapper: NSObject, WKNavigationDelegate {\n public var webView: WKWebView\n\n let strategy: LoadWebPageMainContentStrategy\n let retryLimit: Int\n var webViewDidFinishLoading = false\n var navigationError: (any Error)?\n\n enum WebScrapperError: Error {\n case retry\n }\n \n init(\n retryLimit: Int = 10,\n strategy: LoadWebPageMainContentStrategy\n ) {\n self.retryLimit = retryLimit\n self.strategy = strategy\n let configuration = WKWebViewConfiguration()\n configuration.defaultWebpagePreferences.preferredContentMode = .desktop\n configuration.defaultWebpagePreferences.allowsContentJavaScript = true\n configuration\n .applicationNameForUserAgent =\n \"Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/16.5 Safari/605.1.15\"\n // The web page need the web view to have a size to load correctly.\n let webView = WKWebView(\n frame: .init(x: 0, y: 0, width: 500, height: 500),\n configuration: configuration\n )\n self.webView = webView\n super.init()\n webView.navigationDelegate = self\n }\n\n func fetch(url: URL) async throws -> String {\n webViewDidFinishLoading = false\n navigationError = nil\n var retryCount = 0\n _ = webView.load(.init(url: url))\n while !webViewDidFinishLoading {\n try await Task.sleep(nanoseconds: 10_000_000)\n }\n if let navigationError { throw navigationError }\n while retryCount < retryLimit {\n if let html = try? await getHTML(), !html.isEmpty,\n let document = try? SwiftSoup.parse(html, url.path),\n strategy.validate(document)\n {\n return html\n }\n retryCount += 1\n try await Task.sleep(nanoseconds: 100_000_000)\n }\n\n throw CancellationError()\n }\n\n public nonisolated func webView(_: WKWebView, didFinish _: WKNavigation!) {\n Task { @MainActor in\n self.webViewDidFinishLoading = true\n }\n }\n\n public nonisolated func webView(\n _: WKWebView,\n didFail _: WKNavigation!,\n withError error: Error\n ) {\n Task { @MainActor in\n self.navigationError = error\n self.webViewDidFinishLoading = true\n }\n }\n\n func getHTML() async throws -> String {\n do {\n let isReady = try await webView.evaluateJavaScript(checkIfReady) as? Bool ?? false\n if !isReady { throw WebScrapperError.retry }\n return try await webView.evaluateJavaScript(getHTMLText) as? String ?? \"\"\n } catch {\n throw WebScrapperError.retry\n }\n }\n}\n\nprivate let getHTMLText = \"\"\"\ndocument.documentElement.outerHTML;\n\"\"\"\n\nprivate let checkIfReady = \"\"\"\ndocument.readyState === \"ready\" || document.readyState === \"complete\";\n\"\"\"\n\n// MARK: - LoadWebPageMainContentStrategy\n\nprotocol LoadWebPageMainContentStrategy {\n /// Load the web content into several documents.\n func load(_ document: SwiftSoup.Document, metadata: Document.Metadata) throws -> [Document]\n /// Validate if the web page is fully loaded.\n func validate(_ document: SwiftSoup.Document) -> Bool\n}\n\nextension LoadWebPageMainContentStrategy {\n func text(inFirstTag tagName: String, from document: SwiftSoup.Document) -> String? {\n if let tag = try? document.getElementsByTag(tagName).first(),\n let text = try? tag.text()\n {\n return text\n }\n return nil\n }\n}\n\nextension WebLoader {\n struct DefaultLoadContentStrategy: LoadWebPageMainContentStrategy {\n func load(\n _ document: SwiftSoup.Document,\n metadata: Document.Metadata\n ) throws -> [Document] {\n if let mainContent = try? {\n if let article = text(inFirstTag: \"article\", from: document) { return article }\n if let main = text(inFirstTag: \"main\", from: document) { return main }\n let body = try document.body()?.text()\n return body\n }() {\n return [.init(pageContent: mainContent, metadata: metadata)]\n }\n return []\n }\n\n func validate(_: SwiftSoup.Document) -> Bool {\n return true\n }\n }\n}\n\n" }, { "path": "Tool/Sources/LangChain/DocumentTransformer/DocumentTransformer.swift", "content": "import Foundation\n\npublic protocol DocumentTransformer {\n func transformDocuments(_ documents: [Document]) async throws -> [Document]\n}\n" }, { "path": "Tool/Sources/LangChain/DocumentTransformer/RecursiveCharacterTextSplitter.swift", "content": "import Foundation\n\n/// Implementation of splitting text that looks at characters.\n/// Recursively tries to split by different characters to find one that works.\npublic class RecursiveCharacterTextSplitter: TextSplitter {\n public var chunkSize: Int\n public var chunkOverlap: Int\n public var lengthFunction: (String) -> Int\n\n /// A list of separators to try. They will be used in order. Supports regular expressions.\n public var separators: [String]\n\n /// Create a new splitter\n /// - Parameters:\n /// - separators: A list of separators to try. They will be used in order. Supports regular\n /// expressions.\n /// - chunkSize: The maximum size of chunks. Don't use chunk size larger than 8191, because\n /// length safe embedding is not implemented.\n /// - chunkOverlap: The maximum overlap between chunks.\n /// - lengthFunction: A function to compute the length of text.\n public init(\n separators: [String],\n chunkSize: Int = 4000,\n chunkOverlap: Int = 200,\n lengthFunction: @escaping (String) -> Int = { $0.count }\n ) {\n assert(chunkOverlap <= chunkSize)\n self.chunkSize = chunkSize\n self.chunkOverlap = chunkOverlap\n self.lengthFunction = lengthFunction\n self.separators = separators\n }\n\n // Create a new splitter\n /// - Parameters:\n /// - separatorSet: A set of separators to try.\n /// - chunkSize: The maximum size of chunks. Don't use chunk size larger than 8191, because\n /// length safe embedding is not implemented.\n /// - chunkOverlap: The maximum overlap between chunks.\n /// - lengthFunction: A function to compute the length of text.\n public init(\n separatorSet: TextSplitterSeparatorSet = .default,\n chunkSize: Int = 4000,\n chunkOverlap: Int = 200,\n lengthFunction: @escaping (String) -> Int = { $0.count }\n ) {\n assert(chunkOverlap <= chunkSize)\n self.chunkSize = chunkSize\n self.chunkOverlap = chunkOverlap\n self.lengthFunction = lengthFunction\n separators = separatorSet.separators\n }\n\n public func split(text: String) async throws -> [TextChunk] {\n return split(text: text, separators: separators, startIndex: 0)\n }\n\n private func split(text: String, separators: [String], startIndex: Int) -> [TextChunk] {\n var finalChunks = [TextChunk]()\n\n // Get appropriate separator to use\n let firstSeparatorIndex = separators.firstIndex {\n let pattern = \"(\\($0))\"\n guard let regex = try? NSRegularExpression(pattern: pattern) else { return false }\n return regex.firstMatch(\n in: text,\n options: [],\n range: NSRange(text.startIndex..., in: text)\n ) != nil\n }\n var separator: String\n var nextSeparators: [String]\n\n if let index = firstSeparatorIndex {\n separator = separators[index]\n if index < separators.endIndex - 1 {\n nextSeparators = Array(separators[(index + 1)...])\n } else {\n nextSeparators = []\n }\n } else {\n separator = \"\"\n nextSeparators = []\n }\n\n let splits = split(text: text, separator: separator, startIndex: startIndex)\n\n // Now go merging things, recursively splitting longer texts.\n var goodSplits = [TextChunk]()\n for s in splits {\n if lengthFunction(s.text) < chunkSize {\n goodSplits.append(s)\n } else {\n if !goodSplits.isEmpty {\n let mergedText = mergeSplits(goodSplits)\n finalChunks.append(contentsOf: mergedText)\n goodSplits.removeAll()\n }\n if nextSeparators.isEmpty {\n finalChunks.append(s)\n } else {\n let other_info = split(\n text: s.text,\n separators: nextSeparators,\n startIndex: s.startUTF16Offset\n )\n finalChunks.append(contentsOf: other_info)\n }\n }\n }\n if !goodSplits.isEmpty {\n let merged_text = mergeSplits(goodSplits)\n finalChunks.append(contentsOf: merged_text)\n }\n return finalChunks\n }\n}\n\n" }, { "path": "Tool/Sources/LangChain/DocumentTransformer/TextSplitter.swift", "content": "import Foundation\nimport JSONRPC\n\n/// Split text into multiple components.\npublic protocol TextSplitter: DocumentTransformer {\n /// The maximum size of chunks.\n var chunkSize: Int { get }\n /// The maximum overlap between chunks.\n var chunkOverlap: Int { get }\n /// A function to compute the length of text.\n var lengthFunction: (String) -> Int { get }\n\n /// Split text into multiple components.\n func split(text: String) async throws -> [TextChunk]\n}\n\npublic extension TextSplitter {\n /// Create documents from a list of texts.\n func createDocuments(\n texts: [String],\n metadata: [Document.Metadata] = []\n ) async throws -> [Document] {\n var documents = [Document]()\n let paddingLength = texts.count - metadata.count\n let metadata = metadata + .init(repeating: [:], count: paddingLength)\n for (text, metadata) in zip(texts, metadata) {\n let chunks = try await split(text: text)\n for chunk in chunks {\n var metadata = metadata\n metadata[\"startUTF16Offset\"] = .number(Double(chunk.startUTF16Offset))\n metadata[\"endUTF16Offset\"] = .number(Double(chunk.endUTF16Offset))\n let document = Document(pageContent: chunk.text, metadata: metadata)\n documents.append(document)\n }\n }\n return documents\n }\n\n /// Split documents.\n func splitDocuments(_ documents: [Document]) async throws -> [Document] {\n var texts = [String]()\n var metadata = [Document.Metadata]()\n for document in documents {\n texts.append(document.pageContent)\n metadata.append(document.metadata)\n }\n return try await createDocuments(texts: texts, metadata: metadata)\n }\n\n /// Transform sequence of documents by splitting them.\n func transformDocuments(_ documents: [Document]) async throws -> [Document] {\n return try await splitDocuments(documents)\n }\n\n func joinDocuments(_ documents: [Document]) -> Document {\n let textChunks: [TextChunk] = documents.compactMap { document in\n func extract(_ key: String) -> Int? {\n if case let .number(d) = document.metadata[key] {\n return Int(d)\n }\n return nil\n }\n guard let start = extract(\"startUTF16Offset\"),\n let end = extract(\"endUTF16Offset\")\n else { return nil }\n return TextChunk(\n text: document.pageContent,\n startUTF16Offset: start,\n endUTF16Offset: end\n )\n }.sorted(by: { $0.startUTF16Offset < $1.startUTF16Offset })\n var sumChunk: TextChunk?\n for chunk in textChunks {\n if let current = sumChunk {\n if let merged = current.merged(with: chunk, force: true) {\n sumChunk = merged\n }\n } else {\n sumChunk = chunk\n }\n }\n let pageContent = sumChunk?.text ?? \"\"\n var metadata = documents.first?.metadata ?? [String: JSONValue]()\n metadata[\"startUTF16Offset\"] = nil\n metadata[\"endUTF16Offset\"] = nil\n\n return Document(pageContent: pageContent, metadata: metadata)\n }\n}\n\npublic struct TextChunk: Equatable {\n public var text: String\n public var startUTF16Offset: Int\n public var endUTF16Offset: Int\n\n /// Merge the current chunk with another chunk if the 2 chunks are overlapping or adjacent.\n public func merged(with chunk: TextChunk, force: Bool = false) -> TextChunk? {\n let frontChunk = startUTF16Offset < chunk.startUTF16Offset ? self : chunk\n let backChunk = startUTF16Offset < chunk.startUTF16Offset ? chunk : self\n let overlap = frontChunk.endUTF16Offset - backChunk.startUTF16Offset\n guard overlap >= 0 || force else { return nil }\n\n let text = frontChunk.text + backChunk.text.dropFirst(max(0, overlap))\n let start = frontChunk.startUTF16Offset\n let end = backChunk.endUTF16Offset\n return TextChunk(text: text, startUTF16Offset: start, endUTF16Offset: end)\n }\n}\n\npublic extension TextSplitter {\n /// Merge small splits to just fit in the chunk size.\n func mergeSplits(_ splits: [TextChunk]) -> [TextChunk] {\n let chunkOverlap = chunkOverlap < chunkSize ? chunkOverlap : 0\n\n var chunks = [TextChunk]()\n var currentChunk = [TextChunk]()\n var overlappingChunks = [TextChunk]()\n var currentChunkSize = 0\n\n func join(_ a: [TextChunk], _ b: [TextChunk]) -> TextChunk? {\n let text = (a + b).map(\\.text).joined()\n var l = Int.max\n var u = 0\n\n for chunk in a + b {\n l = min(l, chunk.startUTF16Offset)\n u = max(u, chunk.endUTF16Offset)\n }\n\n guard l < u else { return nil }\n\n return .init(text: text, startUTF16Offset: l, endUTF16Offset: u)\n }\n\n for chunk in splits {\n let textLength = lengthFunction(chunk.text)\n if currentChunkSize + textLength > chunkSize {\n guard let currentChunkText = join(overlappingChunks, currentChunk) else { continue }\n chunks.append(currentChunkText)\n\n overlappingChunks = []\n var overlappingSize = 0\n // use small chunks as overlap if possible\n for chunk in currentChunk.reversed() {\n let length = lengthFunction(chunk.text)\n if overlappingSize + length > chunkOverlap { break }\n if overlappingSize + length + textLength > chunkSize { break }\n overlappingSize += length\n overlappingChunks.insert(chunk, at: 0)\n }\n// // fallback to use suffix if no small chunk found\n// if overlappingChunks.isEmpty {\n// let suffix = String(\n// currentChunkText.suffix(min(chunkOverlap, chunkSize - textLength))\n// )\n// overlappingChunks.append(suffix)\n// overlappingSize = lengthFunction(suffix)\n// }\n\n currentChunkSize = overlappingSize + textLength\n currentChunk = [chunk]\n } else {\n currentChunkSize += textLength\n currentChunk.append(chunk)\n }\n }\n\n if !currentChunk.isEmpty, let joinedChunks = join(overlappingChunks, currentChunk) {\n chunks.append(joinedChunks)\n } else {\n chunks.append(contentsOf: overlappingChunks)\n chunks.append(contentsOf: currentChunk)\n }\n\n return chunks\n }\n\n /// Split the text by separator.\n func split(text: String, separator: String, startIndex: Int = 0) -> [TextChunk] {\n let pattern = \"(\\(separator))\"\n if !separator.isEmpty, let regex = try? NSRegularExpression(pattern: pattern) {\n let matches = regex.matches(in: text, range: NSRange(text.startIndex..., in: text))\n var all = [TextChunk]()\n var start = text.startIndex\n for match in matches {\n guard let range = Range(match.range, in: text) else { break }\n guard range.lowerBound > start else { break }\n let result = text[start..\",\n \"
\",\n \"

\",\n \"
\",\n \"

  • \",\n \"

    \",\n \"

    \",\n \"

    \",\n \"

    \",\n \"

    \",\n \"
    \",\n \"\",\n \"\",\n \"\",\n \"
    \",\n \"\",\n \"
      \",\n \"
        \",\n \"
        \",\n \"