AI Chat fullpage UI notices and polish

components/ai_chat/core/browser/ai_chat_service.cc

@@ -117,6 +117,10 @@ AIChatService::AIChatService(
       prefs::kStorageEnabled,
       base::BindRepeating(&AIChatService::MaybeInitStorage,
                           weak_ptr_factory_.GetWeakPtr()));
+  pref_change_registrar_.Add(
+      prefs::kUserDismissedPremiumPrompt,
+      base::BindRepeating(&AIChatService::OnStateChanged,
+                          weak_ptr_factory_.GetWeakPtr()));
 
   MaybeInitStorage();
 }

ios/brave-ios/Sources/AIChat/ModelView/AIChatViewModel.swift

@@ -29,7 +29,7 @@ public class AIChatViewModel: NSObject, ObservableObject {
 
   @Published var siteInfo: AiChat.SiteInfo?
   @Published var _shouldSendPageContents: Bool = true
-  @Published var canShowPremiumPrompt: Bool = false
+  @Published var _canShowPremiumPrompt: Bool = false
   @Published var premiumStatus: AiChat.PremiumStatus = .inactive
   @Published var suggestedQuestions: [String] = []
   @Published var suggestionsStatus: AiChat.SuggestionGenerationStatus = .none
@@ -68,11 +68,11 @@ public class AIChatViewModel: NSObject, ObservableObject {
 
   public var shouldShowPremiumPrompt: Bool {
     get {
-      return canShowPremiumPrompt
+      return _canShowPremiumPrompt
     }
 
     set {  // swiftlint:disable:this unused_setter_value
-      self.canShowPremiumPrompt = newValue
+      _canShowPremiumPrompt = newValue
       if !newValue {
         api.dismissPremiumPrompt()
       }
@@ -333,4 +333,8 @@ extension AIChatViewModel: AIChatDelegate {
     self.siteInfo = siteInfo
     self._shouldSendPageContents = shouldSendPageContents
   }
+
+  public func onServiceStateChanged(_ state: AiChat.ServiceState) {
+    self._canShowPremiumPrompt = state.canShowPremiumPrompt
+  }
 }

ios/browser/api/ai_chat/ai_chat.mm

@@ -61,6 +61,9 @@ - (instancetype)initWithProfileIOS:(ProfileIOS*)profile
     current_content_ = std::make_unique<ai_chat::AssociatedContentDriverIOS>(
         profile_->GetSharedURLLoaderFactory(), delegate);
 
+    conversation_client_ = std::make_unique<ai_chat::ConversationClient>(
+        service_.get(), delegate_);
+
     [self createNewConversation];
   }
   return self;
@@ -78,8 +81,7 @@ - (void)dealloc {
 - (void)createNewConversation {
   current_conversation_ = service_->CreateConversationHandlerForContent(
       current_content_->GetContentId(), current_content_->GetWeakPtr());
-  conversation_client_ = std::make_unique<ai_chat::ConversationClient>(
-      current_conversation_.get(), delegate_);
+  conversation_client_->ChangeConversation(current_conversation_.get());
 }
 
 - (bool)isAgreementAccepted {

ios/browser/api/ai_chat/ai_chat_delegate.h

@@ -28,6 +28,7 @@ OBJC_EXPORT
                              status:(AiChatSuggestionGenerationStatus)status;
 - (void)onPageHasContent:(AiChatSiteInfo*)siteInfo
        shouldSendContent:(bool)shouldSendContent;
+- (void)onServiceStateChanged:(AiChatServiceState*)state;
 @end
 
 NS_ASSUME_NONNULL_END

ios/browser/api/ai_chat/conversation_client.h

@@ -9,24 +9,29 @@
 #include <string>
 #include <vector>
 
+#include "base/memory/weak_ptr.h"
 #include "brave/components/ai_chat/core/common/mojom/ai_chat.mojom.h"
 #include "mojo/public/cpp/bindings/receiver.h"
 
 @protocol AIChatDelegate;
 
 namespace ai_chat {
 
+class AIChatService;
 class ConversationHandler;
 
 // TODO(petemill): Have AIChatViewModel.swift (aka AIChatDelegate) implement
-// mojom::ConversationUI and bind directly to ConversationHandler via ai_chat.mm
-// so that this proxy isn't neccessary.
-class ConversationClient : public mojom::ConversationUI {
+// mojom::ConversationUI and mojom::ServiceObserver and bind directly to
+// ConversationHandler and AIChatService via ai_chat.mm so that this proxy isn't
+// neccessary.
+class ConversationClient : public mojom::ConversationUI,
+                           public mojom::ServiceObserver {
  public:
-  ConversationClient(ConversationHandler* conversation,
-                     id<AIChatDelegate> bridge);
+  ConversationClient(AIChatService* ai_chat_service, id<AIChatDelegate> bridge);
   ~ConversationClient() override;
 
+  void ChangeConversation(ConversationHandler* conversation);
+
  protected:
   // mojom::ConversationUI
   void OnConversationHistoryUpdate() override;
@@ -43,11 +48,19 @@ class ConversationClient : public mojom::ConversationUI {
   void OnFaviconImageDataChanged() override;
   void OnConversationDeleted() override;
 
+  // mojom::ServiceObserver
+  void OnStateChanged(mojom::ServiceStatePtr state) override;
+  void OnConversationListChanged(
+      std::vector<mojom::ConversationPtr> conversations) override {}
+
  private:
   // The actual UI
   __weak id<AIChatDelegate> bridge_;
 
   mojo::Receiver<mojom::ConversationUI> receiver_{this};
+  mojo::Receiver<mojom::ServiceObserver> service_receiver_{this};
+
+  base::WeakPtrFactory<ConversationClient> weak_ptr_factory_{this};
 };
 
 }  // namespace ai_chat

ios/browser/api/ai_chat/conversation_client.mm

@@ -8,6 +8,7 @@
 #include "ai_chat.mojom.objc+private.h"
 #include "base/strings/sys_string_conversions.h"
 #include "brave/base/mac/conversions.h"
+#include "brave/components/ai_chat/core/browser/ai_chat_service.h"
 #include "brave/components/ai_chat/core/browser/conversation_handler.h"
 #include "brave/components/ai_chat/core/common/mojom/ai_chat.mojom-shared.h"
 #include "brave/components/ai_chat/core/common/mojom/ai_chat.mojom.h"
@@ -16,14 +17,22 @@
 
 namespace ai_chat {
 
-ConversationClient::ConversationClient(ConversationHandler* conversation,
+ConversationClient::ConversationClient(AIChatService* ai_chat_service,
                                        id<AIChatDelegate> bridge)
     : bridge_(bridge) {
-  conversation->Bind(receiver_.BindNewPipeAndPassRemote());
+  ai_chat_service->BindObserver(
+      service_receiver_.BindNewPipeAndPassRemote(),
+      base::BindOnce(&ConversationClient::OnStateChanged,
+                     weak_ptr_factory_.GetWeakPtr()));
 }
 
 ConversationClient::~ConversationClient() = default;
 
+void ConversationClient::ChangeConversation(ConversationHandler* conversation) {
+  receiver_.reset();
+  conversation->Bind(receiver_.BindNewPipeAndPassRemote());
+}
+
 // MARK: - mojom::ConversationUI
 
 void ConversationClient::OnConversationHistoryUpdate() {
@@ -75,4 +84,9 @@
   // allows deletion.
 }
 
+void ConversationClient::OnStateChanged(const mojom::ServiceStatePtr state) {
+  [bridge_ onServiceStateChanged:[[AiChatServiceState alloc]
+                                     initWithServiceStatePtr:state->Clone()]];
+}
+
 }  // namespace ai_chat