Add update migration

Author: intitni

Core/Sources/ServiceUpdateMigration/ServiceUpdateMigrator.swift

@@ -1,30 +1,75 @@
 import Foundation
+import GitHubCopilotService
 import Preferences
 
 extension UserDefaultPreferenceKeys {
     struct OldMigrationVersion: UserDefaultPreferenceKey {
-        typealias PreferenceValueType = String
-        static let key = "OldMigrationVersion"
+        var defaultValue: String = "0"
+        let key = "OldMigrationVersion"
     }
-    
+
     var oldMigrationVersion: OldMigrationVersion { .init() }
 }
 
-struct ServiceUpdateMigrator {
-    func migrate() {
-        migrate(
-            from: UserDefaults.shared.value(for: \.oldMigrationVersion),
-            to: Bundle.main.object(forInfoDictionaryKey: "CFBundleVersion")
-        )
-    }
+public struct ServiceUpdateMigrator {
+    public init() {}
 
-    func migrate(from oldVersion: String, to currentVersion: String) {
-        guard let old = Int(oldVersion), let new = Int(currentVersion) else { return }
-        guard old != new else { return }
+    public func migrate() async throws {
+        let version = Bundle.main.object(forInfoDictionaryKey: "CFBundleVersion") as? String ?? "0"
+
+        try await migrate(from: UserDefaults.shared.value(for: \.oldMigrationVersion), to: version)
+        UserDefaults.shared.set(version, for: \.oldMigrationVersion)
+    }
+
+    func migrate(from oldVersion: String, to currentVersion: String) async throws {
+        guard let old = Int(oldVersion) else { return }
         if old <= 135 {
-            migrateFromLowerThanOrEqualToVersion135()
+           try migrateFromLowerThanOrEqualToVersion135()
         }
     }
 }
 
-func migrateFromLowerThanOrEqualToVersion135() {}
+func migrateFromLowerThanOrEqualToVersion135() throws {
+    // 0. Create the application support folder if it doesn't exist
+
+    let urls = try GitHubCopilotBaseService.createFoldersIfNeeded()
+
+    // 1. Move the undefined folder in application support into a sub folder called `GitHub
+    // Copilot/support`
+
+    let undefinedFolderURL = urls.applicationSupportURL.appendingPathComponent("undefined")
+    var isUndefinedADirectory: ObjCBool = false
+    let isUndefinedExisted = FileManager.default.fileExists(
+        atPath: undefinedFolderURL.path,
+        isDirectory: &isUndefinedADirectory
+    )
+    if isUndefinedExisted, isUndefinedADirectory.boolValue {
+        try FileManager.default.moveItem(
+            at: undefinedFolderURL,
+            to: urls.supportURL.appendingPathComponent("undefined")
+        )
+    }
+
+    // 2. Copy the GitHub copilot language service to `GitHub Copilot/executable`
+
+    let copilotFolderURL = urls.executableURL.appendingPathComponent("copilot")
+    var copilotIsFolder: ObjCBool = false
+    let executable = Bundle.main.resourceURL?.appendingPathComponent("copilot")
+    if let executable,
+       FileManager.default.fileExists(atPath: executable.path, isDirectory: &copilotIsFolder),
+       !FileManager.default.fileExists(atPath: copilotFolderURL.path)
+    {
+        try FileManager.default.copyItem(
+            at: executable,
+            to: urls.executableURL.appendingPathComponent("copilot")
+        )
+    }
+
+    // 3. Use chmod to change the permission of the executable to 755
+
+    try FileManager.default.setAttributes(
+        [.posixPermissions: 0o755],
+        ofItemAtPath: copilotFolderURL.path
+    )
+}
+

ExtensionService/AppDelegate.swift

@@ -1,14 +1,15 @@
 import AppKit
+import Environment
 import FileChangeChecker
 import LaunchAgentManager
 import Logger
 import Preferences
 import Service
 import ServiceManagement
+import ServiceUpdateMigration
 import SwiftUI
 import UpdateChecker
 import UserNotifications
-import Environment
 
 let bundleIdentifierBase = Bundle.main
     .object(forInfoDictionaryKey: "BUNDLE_IDENTIFIER_BASE") as! String
@@ -36,6 +37,13 @@ class AppDelegate: NSObject, NSApplicationDelegate, NSWindowDelegate {
         Logger.service.info("XPC Service started.")
         NSApp.setActivationPolicy(.accessory)
         buildStatusBarMenu()
+        Task {
+            do {
+                try await ServiceUpdateMigrator().migrate()
+            } catch {
+                Logger.service.error(error.localizedDescription)
+            }
+        }
     }
 
     @objc private func buildStatusBarMenu() {
@@ -64,7 +72,7 @@ class AppDelegate: NSObject, NSApplicationDelegate, NSWindowDelegate {
             keyEquivalent: ""
         )
         #endif
-        
+
         let checkForUpdate = NSMenuItem(
             title: "Check for Updates",
             action: #selector(checkForUpdate),
@@ -219,7 +227,7 @@ private class UserDefaultsObserver: NSObject {
     ) {
         onChange?(keyPath)
     }
-    
+
     deinit {
         removeObserver(self, forKeyPath: UserDefaultPreferenceKeys().realtimeSuggestionToggle.key)
     }
@@ -256,3 +264,4 @@ func locateHostBundleURL(url: URL) -> URL? {
         .appendingPathComponent("Copilot for Xcode Dev.app")
     return devAppURL
 }
+