// ----------------------------------------------------------------------
// File: GrpcClient.hh
// Author: Andreas-Joachim Peters - CERN
// ----------------------------------------------------------------------

/************************************************************************
 * EOS - the CERN Disk Storage System                                   *
 * Copyright (C) 2018 CERN/Switzerland                                  *
 *                                                                      *
 * This program is free software: you can redistribute it and/or modify *
 * it under the terms of the GNU General Public License as published by *
 * the Free Software Foundation, either version 3 of the License, or    *
 * (at your option) any later version.                                  *
 *                                                                      *
 * This program is distributed in the hope that it will be useful,      *
 * but WITHOUT ANY WARRANTY; without even the implied warranty of       *
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the        *
 * GNU General Public License for more details.                         *
 *                                                                      *
 * You should have received a copy of the GNU General Public License    *
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.*
 ************************************************************************/

#pragma once

/*----------------------------------------------------------------------------*/
#include "client/Namespace.hh"
#include "common/AssistedThread.hh"
/*----------------------------------------------------------------------------*/
/*----------------------------------------------------------------------------*/
#ifdef EOS_GRPC
#include <grpc++/grpc++.h>
#include "proto/Rpc.grpc.pb.h"

/*----------------------------------------------------------------------------*/

EOSCLIENTNAMESPACE_BEGIN

/**
 * @file   GrpcClient.hh
 *
 * @brief  This class implements a gRPC client for an EOS grpc server
 *
 */


class GrpcClient
{
public:

  explicit GrpcClient(std::shared_ptr<grpc::Channel> channel)
    : stub_(eos::rpc::Eos::NewStub(channel)) { }

  // convenience factory function
  static std::unique_ptr<GrpcClient> Create(std::string endpoint =
        "localhost:50051",
      std::string token = "",
      std::string keyfile = "",
      std::string certfile = "",
      std::string cafile = "");

  std::string Ping(const std::string& payload);

  std::string Md(const std::string& path, uint64_t id = 0, uint64_t ino = 0,
                 bool list = false, bool printonly = false);

  std::string Find(const std::string& path, const std::string& find_options, uint64_t id = 0, uint64_t ino = 0,
		   bool files = true, bool dirs = true, uint64_t depth=0, bool printonly = false, 
		   const std::string& exportfs="");

  int ManilaRequest(const eos::rpc::ManilaRequest& request,
		    eos::rpc::ManilaResponse& reply);

  int NsStat(const eos::rpc::NsStatRequest& request,
             eos::rpc::NsStatResponse& reply);

  int Exec(const eos::rpc::NSRequest& request,
	    eos::rpc::NSResponse& reply);

  std::string ExportFs(const eos::rpc::MDResponse& response, const std::string& exportfs);

  int FileInsert(const std::vector<std::string>& paths);
  int ContainerInsert(const std::vector<std::string>& paths);

  void set_ssl(bool onoff)
  {
    mSSL = onoff;
  }

  const bool ssl()
  {
    return mSSL;
  }

  void set_token(const std::string& _token)
  {
    mToken = _token;
  }

  std::string token()
  {
    return mToken;
  }

private:
  std::unique_ptr<eos::rpc::Eos::Stub> stub_;
  bool mSSL;
  std::string mToken;
  std::map<uint64_t, std::string> tree;
};

#endif

EOSCLIENTNAMESPACE_END