// Copyright 2026 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef IOS_CHROME_BROWSER_COBROWSE_MODEL_COBROWSE_TAB_HELPER_H_
#define IOS_CHROME_BROWSER_COBROWSE_MODEL_COBROWSE_TAB_HELPER_H_

#import "base/memory/raw_ptr.h"
#import "base/memory/weak_ptr.h"
#import "base/scoped_observation.h"
#import "ios/web/public/web_state_observer.h"
#import "ios/web/public/web_state_user_data.h"

class TemplateURLService;

@class CobrowseContext;

@protocol SceneCommands;

// Tab helper that listens for new tabs to triggers or not the cobrowse view.
class CobrowseTabHelper : public web::WebStateObserver,
                          public web::WebStateUserData<CobrowseTabHelper> {
 public:
  class Delegate {
   public:
    virtual ~Delegate() = default;
    virtual bool CanShowAssistantForWebState(web::WebState* web_state) = 0;
    virtual void ConfigureAssistantContextForWebState(
        web::WebState* web_state) = 0;

    // Returns whether a cobrowse session is currently active.
    virtual bool IsSessionActive() = 0;

    // Sets whether a cobrowse session is currently active.
    virtual void SetSessionActive(bool active) = 0;

    // Returns whether the assistant should be hidden for the given WebState.
    virtual bool ShouldHideAssistantForWebState(web::WebState* web_state) = 0;

    // Returns whether the WebState is the active WebState.
    virtual bool IsWebStateActive(web::WebState* web_state) = 0;

    // Sets the context directly.
    virtual void SetCobrowseContext(CobrowseContext* context) = 0;
  };

  CobrowseTabHelper(const CobrowseTabHelper&) = delete;
  CobrowseTabHelper& operator=(const CobrowseTabHelper&) = delete;

  ~CobrowseTabHelper() override;

  // Sets the scene commands handler.
  void SetSceneCommandsHandler(id<SceneCommands> handler);

  // Sets the delegate.
  void SetDelegate(Delegate* delegate);

  // WebStateObserver:
  void WasShown(web::WebState* web_state) override;
  void WasHidden(web::WebState* web_state) override;
  void DidStartNavigation(web::WebState* web_state,
                          web::NavigationContext* navigation_context) override;
  void WebStateDestroyed(web::WebState* web_state) override;

 private:
  friend class web::WebStateUserData<CobrowseTabHelper>;

  explicit CobrowseTabHelper(web::WebState* web_state,
                             TemplateURLService* template_url_service);

  // Returns whether the assistant should be hidden for `url`.
  bool ShouldHideAssistant(web::WebState* web_state, const GURL& url);

  // Triggers the showing of the assistant.
  void ShowAssistant();

  // The delegate for this tab helper.
  raw_ptr<Delegate> delegate_ = nullptr;

  // The template URL service used to check for search URLs.
  raw_ptr<TemplateURLService> template_url_service_;

  // The handler for scene commands.
  __weak id<SceneCommands> scene_handler_ = nil;

  // Scoped observation for the WebState.
  base::ScopedObservation<web::WebState, web::WebStateObserver> observation_{
      this};

  base::WeakPtrFactory<CobrowseTabHelper> weak_ptr_factory_{this};
};

#endif  // IOS_CHROME_BROWSER_COBROWSE_MODEL_COBROWSE_TAB_HELPER_H_