/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#ifndef mozilla_dom_cache_StreamList_h
#define mozilla_dom_cache_StreamList_h

#include "mozilla/dom/SafeRefPtr.h"
#include "mozilla/dom/cache/Context.h"
#include "mozilla/dom/cache/Types.h"
#include "nsTArray.h"

class nsIInputStream;

namespace mozilla::dom::cache {

class CacheStreamControlParent;
class Manager;

class StreamList final : public Context::Activity,
                         public SafeRefCounted<StreamList> {
 public:
  StreamList(SafeRefPtr<Manager> aManager, SafeRefPtr<Context> aContext);

  Manager& GetManager() const;
  bool ShouldOpenStreamFor(const nsID& aId) const;

  void SetStreamControl(CacheStreamControlParent* aStreamControl);
  void RemoveStreamControl(CacheStreamControlParent* aStreamControl);

  void Activate(CacheId aCacheId);

  void Add(const nsID& aId, nsCOMPtr<nsIInputStream>&& aStream);
  already_AddRefed<nsIInputStream> Extract(const nsID& aId);

  void NoteClosed(const nsID& aId);
  void NoteClosedAll();
  void CloseAll();

  // Context::Activity methods
  virtual void Cancel() override;
  virtual bool MatchesCacheId(CacheId aCacheId) const override;

 private:
  void DoStringify(nsACString& aData) override;

  struct Entry {
    explicit Entry(const nsID& aId, nsCOMPtr<nsIInputStream>&& aStream)
        : mId(aId), mStream(std::move(aStream)) {}

    nsID mId;
    nsCOMPtr<nsIInputStream> mStream;
  };
  SafeRefPtr<Manager> mManager;
  SafeRefPtr<Context> mContext;
  CacheId mCacheId;
  CacheStreamControlParent* mStreamControl;
  nsTArray<Entry> mList;
  bool mActivated;

 public:
  ~StreamList();

  NS_DECL_OWNINGTHREAD
  MOZ_DECLARE_REFCOUNTED_TYPENAME(cache::StreamList)
};

}  // namespace mozilla::dom::cache

#endif  // mozilla_dom_cache_StreamList_h