#ifndef __ETCD_WATCHER_HPP__
#define __ETCD_WATCHER_HPP__

#include <string>
#include "etcd/Response.hpp"
#include "v3/include/AsyncWatchAction.hpp"

#include <grpc++/grpc++.h>

using etcdserverpb::Watch;
using grpc::Channel;

namespace etcd
{
  class Watcher
  {
  public:
    Watcher(std::string const & etcd_url, std::string const & key, std::function<void(Response)> callback);
    void Cancel();
    void AddKey(std::string const & key);
    ~Watcher();

  protected:
    void doWatch(std::string const & key, std::function<void(Response)> callback);

    int index;
    std::function<void(Response)> callback;
    pplx::task<void> currentTask;
    std::unique_ptr<Watch::Stub> watchServiceStub;
    std::unique_ptr<etcdv3::AsyncWatchAction> call;
  };
}

#endif