arangodb/arangosh/Shell/V8ClientConnection.h

////////////////////////////////////////////////////////////////////////////////
/// DISCLAIMER
///
/// Copyright 2014-2016 ArangoDB GmbH, Cologne, Germany
/// Copyright 2004-2014 triAGENS GmbH, Cologne, Germany
///
/// Licensed under the Apache License, Version 2.0 (the "License");
/// you may not use this file except in compliance with the License.
/// You may obtain a copy of the License at
///
///     http://www.apache.org/licenses/LICENSE-2.0
///
/// Unless required by applicable law or agreed to in writing, software
/// distributed under the License is distributed on an "AS IS" BASIS,
/// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
/// See the License for the specific language governing permissions and
/// limitations under the License.
///
/// Copyright holder is ArangoDB GmbH, Cologne, Germany
///
/// @author Dr. Frank Celler
/// @author Achim Brandt
////////////////////////////////////////////////////////////////////////////////

#ifndef ARANGODB_SHELL_V8CLIENT_CONNECTION_H
#define ARANGODB_SHELL_V8CLIENT_CONNECTION_H 1

#include "Basics/Common.h"
#include "Basics/StringRef.h"
#include "Rest/HttpRequest.h"

#include <v8.h>

namespace arangodb {
class ClientFeature;

namespace httpclient {
class GeneralClientConnection;
class SimpleHttpClient;
class SimpleHttpResult;
}  // namespace httpclient

////////////////////////////////////////////////////////////////////////////////
/// @brief class for http requests
////////////////////////////////////////////////////////////////////////////////

class V8ClientConnection {
  V8ClientConnection(V8ClientConnection const&) = delete;
  V8ClientConnection& operator=(V8ClientConnection const&) = delete;

 public:
  static void setJwtSecret(std::string const& jwtSecret) { JWT_SECRET = std::make_shared<std::string>(jwtSecret); }
  static std::shared_ptr<std::string> jwtSecret() { return JWT_SECRET; }
  static std::string jwtToken(std::string const& secret);

 private:
  static std::shared_ptr<std::string> JWT_SECRET;

 public:
  V8ClientConnection(
      std::unique_ptr<httpclient::GeneralClientConnection>& connection,
      std::string const&, std::string const&, std::string const&, double);
  ~V8ClientConnection();

 public:
  void setInterrupted(bool value);
  bool isConnected();
  void reconnect(ClientFeature*);

  std::string const& databaseName() const { return _databaseName; }
  void setDatabaseName(std::string const& value) { _databaseName = value; }
  std::string const& username() const { return _username; }
  std::string const& password() const { return _password; }
  int lastHttpReturnCode() const { return _lastHttpReturnCode; }
  std::string lastErrorMessage() const { return _lastErrorMessage; }
  std::string const& version() const { return _version; }
  std::string const& mode() const { return _mode; }
  std::string endpointSpecification() const;

  v8::Handle<v8::Value> getData(
      v8::Isolate* isolate, StringRef const& location,
      std::unordered_map<std::string, std::string> const& headerFields, bool raw);

  v8::Handle<v8::Value> headData(
      v8::Isolate* isolate, StringRef const& location,
      std::unordered_map<std::string, std::string> const& headerFields, bool raw);

  v8::Handle<v8::Value> deleteData(
      v8::Isolate* isolate, StringRef const& location, StringRef const& body,
      std::unordered_map<std::string, std::string> const& headerFields, bool raw);

  v8::Handle<v8::Value> optionsData(
      v8::Isolate* isolate, StringRef const& location, StringRef const& body,
      std::unordered_map<std::string, std::string> const& headerFields, bool raw);

  v8::Handle<v8::Value> postData(
      v8::Isolate* isolate, StringRef const& location, StringRef const& body,
      std::unordered_map<std::string, std::string> const& headerFields, bool raw = false);

  v8::Handle<v8::Value> putData(
      v8::Isolate* isolate, StringRef const& location, StringRef const& body,
      std::unordered_map<std::string, std::string> const& headerFields, bool raw);

  v8::Handle<v8::Value> patchData(
      v8::Isolate* isolate, StringRef const& location, StringRef const& body,
      std::unordered_map<std::string, std::string> const& headerFields, bool raw);

  void initServer(v8::Isolate*, v8::Handle<v8::Context> context,
                  ClientFeature*);

 private:
  static std::string rewriteLocation(void*, std::string const&);

 private:
  void init(std::unique_ptr<httpclient::GeneralClientConnection>&,
            std::string const&, std::string const&, std::string const&);

  v8::Handle<v8::Value> requestData(
      v8::Isolate* isolate, rest::RequestType method,
      StringRef const& location, StringRef const& body,
      std::unordered_map<std::string, std::string> const& headerFields);

  v8::Handle<v8::Value> requestDataRaw(
      v8::Isolate* isolate, rest::RequestType method,
      StringRef const& location, StringRef const& body,
      std::unordered_map<std::string, std::string> const& headerFields);

  v8::Handle<v8::Value> handleResult(v8::Isolate* isolate);

 private:
  std::string _databaseName;
  std::string _username;
  std::string _password;
  double _requestTimeout;

  std::unique_ptr<httpclient::SimpleHttpClient> _client;
  int _lastHttpReturnCode;
  std::string _lastErrorMessage;
  std::unique_ptr<httpclient::SimpleHttpResult> _httpResult;

  std::string _version;
  std::string _mode;
};
}  // namespace arangodb

#endif