// 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 CHROME_BROWSER_CONTEXTUAL_TASKS_CONTEXTUAL_TASKS_EXTENSION_HANDLER_H_ #define CHROME_BROWSER_CONTEXTUAL_TASKS_CONTEXTUAL_TASKS_EXTENSION_HANDLER_H_ #include #include #include #include "base/callback_list.h" #include "base/memory/weak_ptr.h" #include "base/scoped_observation.h" #include "base/uuid.h" #include "build/build_config.h" #include "chrome/browser/contextual_tasks/aim_message_poster.h" #include "chrome/browser/contextual_tasks/contextual_tasks.mojom.h" #include "chrome/browser/contextual_tasks/contextual_tasks_types.h" #include "chrome/browser/ui/views/permissions/permission_prompt_observer.h" #include "components/omnibox/browser/searchbox.mojom.h" #include "content/public/browser/document_user_data.h" #include "mojo/public/cpp/bindings/pending_receiver.h" #include "mojo/public/cpp/bindings/pending_remote.h" #include "mojo/public/cpp/bindings/receiver.h" #include "mojo/public/cpp/bindings/remote.h" #include "third_party/omnibox_proto/model_mode.pb.h" #include "third_party/omnibox_proto/tool_mode.pb.h" #include "ui/base/window_open_disposition.h" #include "ui/webui/resources/cr_components/composebox/composebox.mojom.h" #include "url/gurl.h" // Handles Mojo connection requests from the Contextual Tasks extension // document, implementing composebox and searchbox page handler interfaces. namespace omnibox { struct InputState; } // namespace omnibox namespace contextual_search { class ContextualSearchSessionHandle; class InputStateModel; } // namespace contextual_search namespace content { class WebContents; } // namespace content class LensSearchController; class ContextualTasksExtensionHandler : public content::DocumentUserData, public contextual_tasks::mojom::ExtensionPageHandlerFactory, public contextual_tasks::mojom::ExtensionPageHandler, public composebox::mojom::PageHandlerFactory, public composebox::mojom::PageHandler, public searchbox::mojom::PageHandler, public contextual_tasks::AimMessagePoster, public PermissionPromptObserver::Observer { public: ~ContextualTasksExtensionHandler() override; // PermissionPromptObserver::Observer: void OnPermissionPromptChanged(bool is_showing, const gfx::Size& prompt_size) override; // contextual_tasks::AimMessagePoster: void PostAimMessage(const lens::ClientToAimMessage& message) override; void BindContextualTasksFactory( mojo::PendingReceiver< contextual_tasks::mojom::ExtensionPageHandlerFactory> receiver); // contextual_tasks::mojom::ExtensionPageHandlerFactory: void CreateExtensionPageHandler( mojo::PendingRemote page, mojo::PendingReceiver receiver) override; // contextual_tasks::mojom::ExtensionPageHandler: void SetTaskId(const base::Uuid& uuid) override; void OnWebviewMessage(const std::vector& message) override; void GetHandshakeMessage(GetHandshakeMessageCallback callback) override; void BindComposeboxFactory( mojo::PendingReceiver receiver); // composebox::mojom::PageHandlerFactory: void CreatePageHandler( mojo::PendingReceiver receiver, mojo::PendingRemote searchbox_page, mojo::PendingReceiver searchbox_handler) override; // composebox::mojom::PageHandler: void FocusChanged(bool focused) override; void StartPlatformVoiceRecognition() override; void HandleLensButtonClick() override; void HandleFileUpload(bool is_image) override; void NavigateUrl(const GURL& url) override; void CloseLensOverlayFromWebUI( composebox::mojom::LensOverlayDismissalSource dismissal_source) override; void SetSmartTabSharingActive(bool active) override; void GetSmartTabSharingActive( GetSmartTabSharingActiveCallback callback) override; void NotifyComposeboxQuerySubmittedWithContext() override; void CanShowNextboxAnimation( CanShowNextboxAnimationCallback callback) override; void RecordNextboxAnimationImpression(bool shown) override; void OnContextMenuOpened() override; // searchbox::mojom::PageHandler: void SubmitQuery(const std::string& query_text, uint8_t mouse_button, bool alt_key, bool ctrl_key, bool meta_key, bool shift_key, bool is_voice_search) override; void SetActiveToolMode(omnibox::ToolMode tool, bool is_set_by_aim) override; void SetActiveModelMode(omnibox::ModelMode model, bool is_set_by_aim) override; // These are stubs required to implement searchbox::mojom::PageHandler // (which is shared with Realbox) but are not used by the extension // composebox. void OnFocusChanged(bool focused) override; void QueryAutocomplete(int32_t query_id, std::optional tab_id, const std::u16string& input, bool prevent_inline_autocomplete, uint32_t cursor_position, omnibox::SuggestInventory suggest_inventory, bool is_on_focus, const std::string& keyword, searchbox::mojom::InputMethod input_method) override; void StopAutocomplete(bool clear_result) override; void OpenAutocompleteMatch(uint8_t line, const GURL& url, bool are_matches_showing, uint8_t mouse_button, searchbox::mojom::ActionModifiersPtr modifiers, bool via_keyboard) override; void SetSmartComposeStats( searchbox::mojom::SmartComposeStatsPtr smart_compose_stats) override; void SetPopupSelection( searchbox::mojom::OmniboxPopupSelectionPtr selection) override; void OpenPopupSelection(uint32_t result_sequence_id, searchbox::mojom::OmniboxPopupSelectionPtr selection, WindowOpenDisposition disposition) override; void OnNavigationLikely( uint8_t line, const GURL& url, omnibox::mojom::NavigationPredictor navigation_predictor) override; void DeleteAutocompleteMatch(uint8_t line, const GURL& url) override; void ActivateKeyword(uint8_t line, const GURL& url, base::TimeTicks match_selection_timestamp, bool is_mouse_event) override; void ExecuteAction(uint8_t line, uint8_t action_index, const GURL& url, base::TimeTicks match_selection_timestamp, uint8_t mouse_button, bool alt_key, bool ctrl_key, bool meta_key, bool shift_key) override; void GetCyclingPlaceholderConfig( GetCyclingPlaceholderConfigCallback callback) override; void GetRecentTabs(GetRecentTabsCallback callback) override; void GetTabPreview(int32_t tab_id, GetTabPreviewCallback callback) override; void WaitForTabFaviconLoad(int32_t tab_id, WaitForTabFaviconLoadCallback callback) override; void GetInputState(GetInputStateCallback callback) override; void NotifySessionStarted() override; void NotifySessionAbandoned() override; void AddFileContext(searchbox::mojom::SelectedFileInfoPtr file_info, mojo_base::BigBuffer file_bytes, AddFileContextCallback callback) override; void AddTabContext(int32_t tab_id, bool delay_upload, searchbox::mojom::TabAttachmentSource source, AddTabContextCallback callback) override; void DeleteContext(const base::UnguessableToken& file_token, bool from_automatic_chip) override; void DeleteTabContext(int32_t tab_id) override; void ClearFiles(bool should_block_auto_suggested_tabs) override; void OpenLensSearch() override; void RecordToolSelectionAction(omnibox::ToolMode tool) override; void RecordModelSelectionAction(omnibox::ModelMode model) override; void ActivateMetricsFunnel(const std::string& funnel_name) override; void GetDriveDisclaimerStatus( GetDriveDisclaimerStatusCallback callback) override; void OnDriveDisclaimerAccepted() override; void OnDriveUploadClicked(OnDriveUploadClickedCallback callback) override; void OpenProfilePicker() override; void ShowScreenshotMenu(const gfx::Rect& anchor_rect) override; void GetPageClassification(GetPageClassificationCallback callback) override; void OnThumbnailRemoved() override; void StartScreenshare(bool prefer_entire_screen, StartScreenshareCallback callback) override; void CaptureRegionScreenshot( CaptureRegionScreenshotCallback callback) override; private: friend class content::DocumentUserData; explicit ContextualTasksExtensionHandler(content::RenderFrameHost* rfh); DOCUMENT_USER_DATA_KEY_DECL(); content::WebContents* GetActiveTabWebContents() const; contextual_search::ContextualSearchSessionHandle* GetOrCreateContextualSessionHandle(); std::optional GetActiveTabContextId(); std::optional GetLensOverlayToken(); #if !BUILDFLAG(IS_ANDROID) LensSearchController* GetLensSearchController() const; #endif mojo::Receiver contextual_tasks_factory_receiver_{this}; mojo::Receiver contextual_tasks_handler_receiver_{this}; mojo::Remote contextual_tasks_page_; mojo::Receiver composebox_factory_receiver_{this}; mojo::Receiver composebox_handler_receiver_{ this}; mojo::Receiver searchbox_handler_receiver_{ this}; mojo::Remote searchbox_page_; void InitializeInputStateModel(); void OnInputStateChanged(const omnibox::InputState& state); base::WeakPtr GetOrCreateInputStateModel(); base::WeakPtr input_state_model_; base::CallbackListSubscription input_state_subscription_; base::ScopedObservation permission_prompt_observation_{this}; std::optional task_id_; omnibox::ToolMode active_tool_ = omnibox::TOOL_MODE_UNSPECIFIED; omnibox::ModelMode active_model_ = omnibox::MODEL_MODE_UNSPECIFIED; }; #endif // CHROME_BROWSER_CONTEXTUAL_TASKS_CONTEXTUAL_TASKS_EXTENSION_HANDLER_H_