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

#ifndef COMPONENTS_FACILITATED_PAYMENTS_CONTENT_BROWSER_CONTENT_FACILITATED_PAYMENTS_DRIVER_FACTORY_H_
#define COMPONENTS_FACILITATED_PAYMENTS_CONTENT_BROWSER_CONTENT_FACILITATED_PAYMENTS_DRIVER_FACTORY_H_

#include <memory>

#include "base/containers/flat_map.h"
#include "base/gtest_prod_util.h"
#include "base/memory/raw_ref.h"
#include "components/facilitated_payments/content/browser/content_facilitated_payments_driver.h"
#include "content/public/browser/web_contents_observer.h"

namespace content {
class WebContents;
class NavigationHandle;
class RenderFrameHost;
}  // namespace content

namespace payments::facilitated {

class FacilitatedPaymentsClient;

// Manages the lifetime of `ContentFacilitatedPaymentsDriver`. It is owned by
// `ContentFacilitatedPaymentsClient`. Creates one
// `ContentFacilitatedPaymentsDriver` for each outermost main `RenderFrameHost`.
class ContentFacilitatedPaymentsDriverFactory
    : public content::WebContentsObserver {
 public:
  ContentFacilitatedPaymentsDriverFactory(content::WebContents* web_contents,
                                          FacilitatedPaymentsClient* client);
  ContentFacilitatedPaymentsDriverFactory(
      const ContentFacilitatedPaymentsDriverFactory&) = delete;
  ContentFacilitatedPaymentsDriverFactory& operator=(
      const ContentFacilitatedPaymentsDriverFactory&) = delete;
  ~ContentFacilitatedPaymentsDriverFactory() override;

  // Gets or creates a dedicated `ContentFacilitatedPaymentsDriver` for the
  // `render_frame_host`. Drivers are only created for the outermost main frame.
  ContentFacilitatedPaymentsDriver& GetOrCreateForFrame(
      content::RenderFrameHost* render_frame_host);

 private:
  FRIEND_TEST_ALL_PREFIXES(
      ContentFacilitatedPaymentsDriverFactoryTest,
      OnTextCopiedToClipboard_PixCodeInIFrame_DoesNotTriggerPixDetection_PixFlowExitedReasonLogged);
  FRIEND_TEST_ALL_PREFIXES(
      ContentFacilitatedPaymentsDriverFactoryTest,
      OnTextCopiedToClipboard_PixCodeInIFrame_FlagEnabled_PixFlowExitedReasonNotLogged);
  FRIEND_TEST_ALL_PREFIXES(
      ContentFacilitatedPaymentsDriverFactoryTest,
      OnTextCopiedToClipboard_PixCodeInIFrame_FlagEnabled_CorrectIframeUrlPassedToDriver);
  FRIEND_TEST_ALL_PREFIXES(
      ContentFacilitatedPaymentsDriverFactoryTest,
      OnTextCopiedToClipboard_FrameNotActive_DoesNotTriggerPixDetection_PixFlowExitedReasonLogged);
  FRIEND_TEST_ALL_PREFIXES(
      ContentFacilitatedPaymentsDriverFactoryTest,
      OnTextCopiedToClipboard_ErrorDocument_DoesNotTriggerPixDetection_PixFlowExitedReasonLogged);
  // content::WebContentsObserver:
  void RenderFrameDeleted(content::RenderFrameHost* render_frame_host) override;
  void RenderFrameHostStateChanged(
      content::RenderFrameHost* render_frame_host,
      content::RenderFrameHost::LifecycleState old_state,
      content::RenderFrameHost::LifecycleState new_state) override;
  void DidFinishNavigation(
      content::NavigationHandle* navigation_handle) override;
  void OnTextCopiedToClipboard(content::RenderFrameHost* render_frame_host,
                               const std::u16string& copied_text) override;

  // Owns the drivers, one for each render frame host. Should be empty at
  // destruction time because its elements are erased in RenderFrameDeleted().
  base::flat_map<content::RenderFrameHost*,
                 std::unique_ptr<ContentFacilitatedPaymentsDriver>>
      driver_map_;

  // Owner.
  const raw_ref<FacilitatedPaymentsClient> client_;
};

}  // namespace payments::facilitated

#endif  // COMPONENTS_FACILITATED_PAYMENTS_CONTENT_BROWSER_CONTENT_FACILITATED_PAYMENTS_DRIVER_FACTORY_H_