#ifndef __ASYNC_LEASEACTION_HPP__
#define __ASYNC_LEASEACTION_HPP__

#include <mutex>

#include <grpc++/grpc++.h>
#include "proto/rpc.grpc.pb.h"
#include "etcd/v3/Action.hpp"
#include "etcd/v3/AsyncLeaseResponse.hpp"
#include "etcd/Response.hpp"

using grpc::ClientAsyncResponseReader;
using grpc::ClientAsyncReaderWriter;
using etcdserverpb::LeaseGrantResponse;
using etcdserverpb::LeaseRevokeResponse;
using etcdserverpb::LeaseCheckpoint;
using etcdserverpb::LeaseCheckpointResponse;
using etcdserverpb::LeaseKeepAliveRequest;
using etcdserverpb::LeaseKeepAliveResponse;
using etcdserverpb::LeaseTimeToLiveResponse;
using etcdserverpb::LeaseStatus;
using etcdserverpb::LeaseLeasesResponse;

namespace etcdv3
{
  class AsyncLeaseGrantAction : public etcdv3::Action {
    public:
      AsyncLeaseGrantAction(etcdv3::ActionParameters const &param);
      AsyncLeaseGrantResponse ParseResponse();
    private:
      LeaseGrantResponse reply;
      std::unique_ptr<ClientAsyncResponseReader<LeaseGrantResponse>> response_reader;
  };

  class AsyncLeaseRevokeAction: public etcdv3::Action {
    public:
      AsyncLeaseRevokeAction(etcdv3::ActionParameters const &param);
      AsyncLeaseRevokeResponse ParseResponse();
    private:
      LeaseRevokeResponse reply;
      std::unique_ptr<ClientAsyncResponseReader<LeaseRevokeResponse>> response_reader;
  };

  class AsyncLeaseKeepAliveAction: public etcdv3::Action {
    public:
      AsyncLeaseKeepAliveAction(etcdv3::ActionParameters const &param);
      AsyncLeaseKeepAliveResponse ParseResponse();

      etcd::Response Refresh(int ttl);
      void CancelKeepAlive();
      bool Cancelled() const;

    private:
      LeaseKeepAliveResponse reply;
      std::unique_ptr<ClientAsyncReaderWriter<LeaseKeepAliveRequest, LeaseKeepAliveResponse>> stream;

      LeaseKeepAliveRequest req;
      bool isCancelled;
      std::mutex protect_is_cancelled;
  };

  class AsyncLeaseTimeToLiveAction: public etcdv3::Action {
    public:
      AsyncLeaseTimeToLiveAction(etcdv3::ActionParameters const &param);
      AsyncLeaseTimeToLiveResponse ParseResponse();
    private:
      LeaseTimeToLiveResponse reply;
      std::unique_ptr<ClientAsyncResponseReader<LeaseTimeToLiveResponse>> response_reader;
  };

  class AsyncLeaseLeasesAction: public etcdv3::Action {
    public:
      AsyncLeaseLeasesAction(etcdv3::ActionParameters const &param);
      AsyncLeaseLeasesResponse ParseResponse();
    private:
      LeaseLeasesResponse reply;
      std::unique_ptr<ClientAsyncResponseReader<LeaseLeasesResponse>> response_reader;
  };
}

#endif