/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */

/* 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 GMPStorageChild_h_

#define GMPStorageChild_h_

#include "mozilla/gmp/PGMPStorageChild.h"

#include "gmp-storage.h"

#include "nsTHashtable.h"

#include "nsRefPtrHashtable.h"

#include "gmp-platform.h"

#include <queue>

namespace mozilla {

namespace gmp {

class GMPChild;

class GMPStorageChild;

class GMPRecordImpl : public GMPRecord

{

public:

  NS_INLINE_DECL_THREADSAFE_REFCOUNTING(GMPRecordImpl)

  GMPRecordImpl(GMPStorageChild* aOwner,

                const nsCString& aName,

                GMPRecordClient* aClient);

  // GMPRecord.

  GMPErr Open() override;

  GMPErr Read() override;

  GMPErr Write(const uint8_t* aData,

               uint32_t aDataSize) override;

  GMPErr Close() override;

  const nsCString& Name() const { return mName; }

  void OpenComplete(GMPErr aStatus);

  void ReadComplete(GMPErr aStatus, const uint8_t* aBytes, uint32_t aLength);

  void WriteComplete(GMPErr aStatus);

private:

  ~GMPRecordImpl() {}

  const nsCString mName;

  GMPRecordClient* const mClient;

  GMPStorageChild* const mOwner;

};

class GMPStorageChild : public PGMPStorageChild

{

public:

  NS_INLINE_DECL_THREADSAFE_REFCOUNTING(GMPStorageChild)

  explicit GMPStorageChild(GMPChild* aPlugin);

  GMPErr CreateRecord(const nsCString& aRecordName,

                      GMPRecord** aOutRecord,

                      GMPRecordClient* aClient);

  GMPErr Open(GMPRecordImpl* aRecord);

  GMPErr Read(GMPRecordImpl* aRecord);

  GMPErr Write(GMPRecordImpl* aRecord,

               const uint8_t* aData,

               uint32_t aDataSize);

  GMPErr Close(const nsCString& aRecordName);

private:

  bool HasRecord(const nsCString& aRecordName);

  already_AddRefed<GMPRecordImpl> GetRecord(const nsCString& aRecordName);

protected:

  ~GMPStorageChild() {}

  // PGMPStorageChild

  mozilla::ipc::IPCResult RecvOpenComplete(const nsCString& aRecordName,

                                           const GMPErr& aStatus) override;

  mozilla::ipc::IPCResult RecvReadComplete(const nsCString& aRecordName,

                                           const GMPErr& aStatus,

                                           InfallibleTArray<uint8_t>&& aBytes) override;

  mozilla::ipc::IPCResult RecvWriteComplete(const nsCString& aRecordName,

                                            const GMPErr& aStatus) override;

  mozilla::ipc::IPCResult RecvShutdown() override;

private:

  Monitor mMonitor;

  nsRefPtrHashtable<nsCStringHashKey, GMPRecordImpl> mRecords;

  GMPChild* mPlugin;

  bool mShutdown;

};

} // namespace gmp

} // namespace mozilla

#endif // GMPStorageChild_h_