// Copyright 2019 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_CONTENT_CAPTURE_RENDERER_CONTENT_CAPTURE_SENDER_H_
#define COMPONENTS_CONTENT_CAPTURE_RENDERER_CONTENT_CAPTURE_SENDER_H_

#include "components/content_capture/common/content_capture.mojom.h"
#include "content/public/renderer/render_frame_observer.h"
#include "mojo/public/cpp/bindings/associated_receiver.h"
#include "mojo/public/cpp/bindings/associated_remote.h"
#include "mojo/public/cpp/bindings/pending_associated_receiver.h"
#include "third_party/blink/public/web/web_content_capture_client.h"

namespace blink {
class AssociatedInterfaceRegistry;
}

namespace content_capture {

struct ContentCaptureData;

// This class has one instance per RenderFrame, and implements
// WebConetentCaptureClient to get the captured content and the removed
// content from blink, then forward them to browser process; it enables
// the ContentCapture in blink by setting WebContentCaptureClient to
// WebLocalFrame.
class ContentCaptureSender : public content::RenderFrameObserver,
                             public blink::WebContentCaptureClient,
                             public mojom::ContentCaptureSender {
 public:
  explicit ContentCaptureSender(content::RenderFrame* render_frame,
                                blink::AssociatedInterfaceRegistry* registry);

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

  ~ContentCaptureSender() override;

  void BindPendingReceiver(
      mojo::PendingAssociatedReceiver<mojom::ContentCaptureSender>
          pending_receiver);

  // blink::WebContentCaptureClient:
  base::TimeDelta GetTaskInitialDelay() const override;
  void DidCompleteBatchCaptureContent() override;
  void DidCaptureContent(const std::vector<blink::WebContentHolder>& data,
                         bool first_data) override;
  void DidUpdateContent(
      const std::vector<blink::WebContentHolder>& data) override;
  void DidRemoveContent(std::vector<int64_t> data) override;

  // mojom::ContentCaptureSender:
  void StartCapture() override;
  void StopCapture() override;

  // content::RenderFrameObserver:
  void OnDestruct() override;

 private:
  void FillContentCaptureData(
      const std::vector<blink::WebContentHolder>& node_holders,
      ContentCaptureData* data,
      bool set_url);
  const mojo::AssociatedRemote<mojom::ContentCaptureReceiver>&
  GetContentCaptureReceiver();

  mojo::AssociatedRemote<mojom::ContentCaptureReceiver>
      content_capture_receiver_;
  mojo::AssociatedReceiver<mojom::ContentCaptureSender> receiver_{this};
};

}  // namespace content_capture

#endif  // COMPONENTS_CONTENT_CAPTURE_RENDERER_CONTENT_CAPTURE_SENDER_H_