Support plus feature flag

Author: intitni

Core/Package.swift

@@ -263,6 +263,12 @@ let package = Package(
             ]
         ),
         .target(name: "UserDefaultsObserver"),
+        .target(
+            name: "PlusFeatureFlag",
+            dependencies: [
+                .product(name: "LicenseManagement", package: "Tool"),
+            ]
+        ),
 
         // MARK: - GitHub Copilot
 

Core/Sources/ChatGPTChatTab/ChatGPTChatTab.swift

@@ -15,6 +15,7 @@ public class ChatGPTChatTab: ChatTab {
 
     struct Builder: ChatTabBuilder {
         var title: String
+        var buildable: Bool { true }
         var customCommand: CustomCommand?
 
         func build() -> any ChatTab {

Core/Sources/HostApp/HostApp.swift

@@ -1,28 +1,49 @@
-import Foundation
+import Client
 import ComposableArchitecture
+import Foundation
+
+#if canImport(LicenseManagement)
+import LicenseManagement
+#endif
 
 struct HostApp: ReducerProtocol {
     struct State: Equatable {
         var general = General.State()
     }
-    
+
     enum Action: Equatable {
         case appear
+        case informExtensionServiceAboutLicenseKeyChange
         case general(General.Action)
     }
-    
+
+    @Dependency(\.toast) var toast
+
     var body: some ReducerProtocol<State, Action> {
         Scope(state: \.general, action: /Action.general) {
             General()
         }
-        
+
         Reduce { _, action in
             switch action {
             case .appear:
                 return .none
+            case .informExtensionServiceAboutLicenseKeyChange:
+                return .run { _ in
+                    #if canImport(LicenseManagement)
+                    let service = try getService()
+                    do {
+                        try await service
+                            .postNotification(name: Notification.Name.licenseKeyChanged.rawValue)
+                    } catch {
+                        toast(error.localizedDescription, .error)
+                    }
+                    #endif
+                }
             case .general:
                 return .none
             }
         }
     }
 }
+

Core/Sources/HostApp/TabContainer.swift

@@ -59,7 +59,9 @@ public struct TabContainer: View {
                     image: "command.square"
                 )
                 #if canImport(ProHostApp)
-                PlusView().tabBarItem(
+                PlusView(onLicenseKeyChanged: {
+                    store.send(.informExtensionServiceAboutLicenseKeyChange)
+                }).tabBarItem(
                     tag: 5,
                     title: "Plus",
                     image: "plus.diamond"

Core/Sources/PlusFeatureFlag/PlusFeatureFlag.swift

@@ -0,0 +1,27 @@
+import Dependencies
+import Foundation
+import SwiftUI
+
+#if canImport(LicenseManagement)
+
+import LicenseManagement
+
+public func withFeatureEnabled(
+    _ flag: KeyPath<PlusFeatureFlags, PlusFeatureFlag>,
+    then: () throws -> Void
+) rethrows {
+    try LicenseManagement.withFeatureEnabled(flag, then: then)
+}
+
+public func withFeatureEnabled(
+    _ flag: KeyPath<PlusFeatureFlags, PlusFeatureFlag>,
+    then: () async throws -> Void
+) async rethrows {
+    try await LicenseManagement.withFeatureEnabled(flag, then: then)
+}
+
+public func isFeatureAvailable(_ flag: KeyPath<PlusFeatureFlags, PlusFeatureFlag>) -> Bool {
+    LicenseManagement.isFeatureAvailable(flag)
+}
+
+#endif

Core/Sources/Service/GUI/GraphicalUserInterfaceController.swift

@@ -41,7 +41,8 @@ struct GUI: ReducerProtocol {
             Reduce { _, action in
                 switch action {
                 case let .createNewTapButtonClicked(kind):
-                    let chatTap = kind?.builder.build() ?? ChatGPTChatTab()
+                    guard let builder = kind?.builder, builder.buildable else { return .none }
+                    let chatTap = builder.build()
                     return .run { send in
                         await send(.appendAndSelectTab(chatTap))
                     }

Core/Sources/SuggestionWidget/ChatWindowView.swift

@@ -224,7 +224,7 @@ struct ChatTabBar: View {
                             store.send(.createNewTapButtonClicked(kind: kind))
                         }) {
                             Text(kind.title)
-                        }
+                        }.disabled(!kind.builder.buildable)
                     case let .folder(title, list):
                         Menu {
                             ForEach(0..<list.endIndex, id: \.self) { index in
@@ -360,6 +360,7 @@ class FakeChatTab: ChatTab {
 
     struct Builder: ChatTabBuilder {
         var title: String = "Title"
+        var buildable: Bool { true }
 
         func build() -> any ChatTab {
             return FakeChatTab(id: "id", title: "Title")

Pro

@@ -94,10 +94,24 @@ open class BaseChatTab: Equatable {
 public protocol ChatTabBuilder {
     /// A visible title for user.
     var title: String { get }
+    /// whether the chat tab is buildable.
+    var buildable: Bool { get }
     /// Build the chat tab.
     func build() -> any ChatTab
 }
 
+public struct DisabledChatTabBuilder: ChatTabBuilder {
+    public var title: String
+    public var buildable: Bool { false }
+    public func build() -> any ChatTab {
+        EmptyChatTab(id: UUID().uuidString)
+    }
+
+    public init(title: String) {
+        self.title = title
+    }
+}
+
 public protocol ChatTabType {
     /// The type of the external dependency required by this chat tab.
     associatedtype ExternalDependency
@@ -127,6 +141,7 @@ public class EmptyChatTab: ChatTab {
 
     struct Builder: ChatTabBuilder {
         let title: String
+        var buildable: Bool { true }
         func build() -> any ChatTab {
             EmptyChatTab()
         }