// Copyright 2025 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_UI_LENS_TEST_LENS_SEARCH_CONTROLLER_H_
#define CHROME_BROWSER_UI_LENS_TEST_LENS_SEARCH_CONTROLLER_H_

#include "chrome/browser/ui/lens/lens_overlay_query_controller.h"
#include "chrome/browser/ui/lens/lens_query_flow_router.h"
#include "chrome/browser/ui/lens/lens_search_contextualization_controller.h"
#include "chrome/browser/ui/lens/lens_search_controller.h"
#include "components/contextual_search/contextual_search_context_controller.h"
#include "testing/gmock/include/gmock/gmock.h"

namespace variations {
class VariationsClient;
}  // namespace variations

namespace contextual_search {
class MockContextualSearchSessionHandle;
class MockContextualSearchContextController;
}  // namespace contextual_search

namespace lens {

class MockLensSearchController : public LensSearchController {
 public:
  explicit MockLensSearchController(tabs::TabInterface* tab);
  ~MockLensSearchController() override;

  MOCK_METHOD(lens::LensOverlayQueryController*,
              lens_overlay_query_controller,
              (),
              (override));

  MOCK_METHOD(lens::LensSearchContextualizationController*,
              lens_search_contextualization_controller,
              (),
              (override));

  MOCK_METHOD(LensOverlayController*, lens_overlay_controller, (), (override));

  MOCK_METHOD(bool, should_route_to_contextual_tasks, (), (const, override));

  MOCK_METHOD(bool, IsActive, (), (override));
  MOCK_METHOD(void,
              CloseLensSync,
              (lens::LensOverlayDismissalSource dismissal_source),
              (override));

  MOCK_METHOD(lens::LensOverlayGen204Controller*,
              gen204_controller,
              (),
              (override));

  MOCK_METHOD(std::optional<lens::LensOverlayInvocationSource>,
              invocation_source,
              (),
              (override));

  MOCK_METHOD(void,
              HandleInteractionResponse,
              (lens::mojom::TextPtr text),
              (override));
};

class FakeLensQueryFlowRouter : public LensQueryFlowRouter {
 public:
  explicit FakeLensQueryFlowRouter(
      LensSearchController* lens_search_controller);
  ~FakeLensQueryFlowRouter() override;

  contextual_search::ContextualSearchSessionHandle*
  GetContextualSearchSessionHandle() const override;

 protected:
  bool IsActiveTabContextEligible() const override;
  TabContextualizationController* GetTabContextualizationController()
      const override;

 private:
  std::unique_ptr<contextual_search::MockContextualSearchSessionHandle>
      mock_handle_;
  std::unique_ptr<contextual_search::MockContextualSearchContextController>
      mock_context_controller_;
  std::unique_ptr<tabs::TabInterface> fake_tab_interface_;
  std::unique_ptr<TabContextualizationController> mock_tab_context_controller_;
  std::unique_ptr<contextual_search::FileInfo> mock_file_info_;
};

class TestLensSearchController : public LensSearchController {
 public:
  explicit TestLensSearchController(tabs::TabInterface* tab)
      : LensSearchController(tab) {}

  std::unique_ptr<LensOverlayController> CreateLensOverlayController(
      tabs::TabInterface* tab,
      LensSearchController* lens_search_controller,
      PrefService* pref_service,
      ThemeService* theme_service) override;

  std::unique_ptr<lens::LensOverlayQueryController> CreateLensQueryController(
      lens::LensOverlayFullImageResponseCallback full_image_callback,
      lens::LensOverlayUrlResponseCallback url_callback,
      lens::LensOverlayInteractionResponseCallback interaction_callback,
      lens::LensOverlayThumbnailCreatedCallback thumbnail_created_callback,
      lens::UploadProgressCallback upload_progress_callback,
      variations::VariationsClient* variations_client,
      signin::IdentityManager* identity_manager,
      Profile* profile,
      lens::LensOverlayInvocationSource invocation_source,
      bool use_dark_mode,
      lens::LensOverlayGen204Controller* gen204_controller) override;

  std::unique_ptr<lens::LensQueryFlowRouter> CreateLensQueryFlowRouter()
      override;

  std::unique_ptr<lens::LensSearchContextualizationController>
  CreateLensSearchContextualizationController() override;
};

}  // namespace lens

#endif  // CHROME_BROWSER_UI_LENS_TEST_LENS_SEARCH_CONTROLLER_H_