////////////////////////////////////////////////////////////////////////////////
/// 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_REST_VPP_RESPONSE_H
#define ARANGODB_REST_VPP_RESPONSE_H 1

#include "Rest/GeneralResponse.h"

#include "Basics/StringBuffer.h"

namespace arangodb {
class RestBatchHandler;

namespace rest {
class VppCommTask;
class GeneralCommTask;
}

class VppResponse : public GeneralResponse {
  friend class rest::GeneralCommTask;
  friend class rest::VppCommTask;
  friend class RestBatchHandler;  // TODO must be removed

  explicit VppResponse(ResponseCode code);

 public:
  static bool HIDE_PRODUCT_HEADER;

  // required by base
  void reset(ResponseCode code) override final;
  void setPayload(GeneralRequest const*, arangodb::velocypack::Slice const&,
                  bool generateBody,
                  arangodb::velocypack::Options const&) override final;

  void writeHeader(basics::StringBuffer*) override;

  basics::StringBuffer& body() override { return _body; }

  void setConnectionType(ConnectionType type) override {
    _connectionType = type;
  }

  void setContentType(ContentType type) override { _contentType = type; }

  void setContentType(std::string const& contentType) override {
    _headers[arangodb::StaticStrings::ContentTypeHeader] = contentType;
    _contentType = ContentType::CUSTOM;
  }

  void setContentType(std::string&& contentType) override {
    _headers[arangodb::StaticStrings::ContentTypeHeader] =
        std::move(contentType);
    _contentType = ContentType::CUSTOM;
  }
  // end - required by base
 private:
  //_responseCode   - from Base
  //_headers        - from Base
  ConnectionType _connectionType;
  ContentType _contentType;
  basics::StringBuffer _body;
};
}

#endif