// 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 THIRD_PARTY_BLINK_RENDERER_MODULES_AI_CREATE_MONITOR_H_
#define THIRD_PARTY_BLINK_RENDERER_MODULES_AI_CREATE_MONITOR_H_

#include "services/on_device_model/public/mojom/download_observer.mojom-blink.h"
#include "third_party/blink/renderer/core/dom/abort_signal.h"
#include "third_party/blink/renderer/core/dom/events/event_target.h"
#include "third_party/blink/renderer/core/event_type_names.h"
#include "third_party/blink/renderer/core/execution_context/execution_context_lifecycle_observer.h"
#include "third_party/blink/renderer/platform/mojo/heap_mojo_receiver.h"

namespace blink {

// The monitor class that serves as the target for the `downloadprogress` event.
class CreateMonitor final
    : public EventTarget,
      public ExecutionContextClient,
      public on_device_model::mojom::blink::DownloadObserver {
  DEFINE_WRAPPERTYPEINFO();

 public:
  CreateMonitor(ExecutionContext* context,
                AbortSignal* abort_signal,
                scoped_refptr<base::SequencedTaskRunner> task_runner);
  ~CreateMonitor() override = default;

  void Trace(Visitor* visitor) const override;

  // EventTarget implementation
  const AtomicString& InterfaceName() const override;
  ExecutionContext* GetExecutionContext() const override;

  // on_device_model::mojom::blink::DownloadObserver implementation.
  void OnDownloadProgressUpdate(uint64_t downloaded_bytes,
                                uint64_t total_bytes) override;

  mojo::PendingRemote<on_device_model::mojom::blink::DownloadObserver>
  BindRemote();

  DEFINE_ATTRIBUTE_EVENT_LISTENER(downloadprogress, kDownloadprogress)

 private:
  std::optional<uint64_t> last_downloaded_bytes_;

  Member<AbortSignal> abort_signal_;
  scoped_refptr<base::SequencedTaskRunner> task_runner_;
  HeapMojoReceiver<on_device_model::mojom::blink::DownloadObserver,
                   CreateMonitor>
      receiver_;
};

}  // namespace blink

#endif  // THIRD_PARTY_BLINK_RENDERER_MODULES_AI_CREATE_MONITOR_H_