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

#ifndef MEDIA_MOJO_SERVICES_MEDIA_RESOURCE_SHIM_H_
#define MEDIA_MOJO_SERVICES_MEDIA_RESOURCE_SHIM_H_

#include <stddef.h>

#include "base/functional/callback.h"
#include "base/memory/raw_ptr.h"
#include "base/memory/weak_ptr.h"
#include "media/base/media_resource.h"
#include "media/mojo/services/mojo_demuxer_stream_adapter.h"
#include "mojo/public/cpp/bindings/pending_remote.h"

namespace media {

// MediaResource shim for mojom::DemuxerStreams.
class MediaResourceShim : public MediaResource {
 public:
  // Constructs the shim; at least a single audio or video stream must be
  // provided.  |demuxer_ready_cb| will be called once the streams have been
  // initialized.  Calling any method before then is an error.
  MediaResourceShim(
      std::vector<mojo::PendingRemote<mojom::DemuxerStream>> streams,
      base::OnceClosure demuxer_ready_cb);

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

  ~MediaResourceShim() override;

  // MediaResource interface.
  std::vector<raw_ptr<DemuxerStream>> GetAllStreams() override;

 private:
  // Called as each mojom::DemuxerStream becomes ready.  Once all streams
  // are ready it will fire the |demuxer_ready_cb_| provided during
  // construction.
  void OnStreamReady();

  // Stored copy the ready callback provided during construction; cleared once
  // all streams are ready.
  base::OnceClosure demuxer_ready_cb_;

  // Container for demuxer stream adapters which interface with the mojo level
  // demuxer streams.  |streams_ready_| tracks how many streams are ready and is
  // used by OnStreamReady() to know when |demuxer_ready_cb_| should be fired.
  std::vector<std::unique_ptr<MojoDemuxerStreamAdapter>> streams_;
  size_t streams_ready_;

  // WeakPtrFactorys must always be the last member variable.
  base::WeakPtrFactory<MediaResourceShim> weak_factory_{this};
};

}  // namespace media

#endif  // MEDIA_MOJO_SERVICES_MEDIA_RESOURCE_SHIM_H_