#include "etcd/v3/AsyncSetAction.hpp"
#include "etcd/v3/action_constants.hpp"
#include "etcd/v3/Transaction.hpp"

using etcdserverpb::Compare;

etcdv3::AsyncSetAction::AsyncSetAction(etcdv3::ActionParameters param, bool create)
  : etcdv3::Action(param) 
{
  etcdv3::Transaction transaction(parameters.key);
  isCreate = create;
  transaction.init_compare(Compare::CompareResult::Compare_CompareResult_EQUAL,
                           Compare::CompareTarget::Compare_CompareTarget_VERSION);

  transaction.setup_basic_create_sequence(parameters.key, parameters.value, parameters.lease_id);

  if(isCreate)
  {
    transaction.setup_basic_failure_operation(parameters.key);
  }
  else
  {
    transaction.setup_set_failure_operation(parameters.key, parameters.value, parameters.lease_id);
  }
  response_reader = parameters.kv_stub->AsyncTxn(&context, transaction.txn_request, &cq_);
  response_reader->Finish(&reply, &status, (void*)this);
}

etcdv3::AsyncTxnResponse etcdv3::AsyncSetAction::ParseResponse()
{

  AsyncTxnResponse txn_resp;
  
  if(!status.ok())
  {
    txn_resp.set_error_code(status.error_code());
    txn_resp.set_error_message(status.error_message());
  }
  else
  { 
    txn_resp.ParseResponse(parameters.key, parameters.withPrefix, reply);
    std::string action = isCreate? etcdv3::CREATE_ACTION:etcdv3::SET_ACTION;
    txn_resp.set_action(action);

    if(!reply.succeeded() && action == etcdv3::CREATE_ACTION)
    {
      txn_resp.set_error_code(105);
      txn_resp.set_error_message("Key already exists");
    } 
  }
  return txn_resp;
}