////////////////////////////////////////////////////////////////////////////////
/// 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_HTTP_RESPONSE_H
#define ARANGODB_REST_HTTP_RESPONSE_H 1

#include "Rest/GeneralResponse.h"

#include "Basics/StringBuffer.h"

namespace arangodb {
class RestBatchHandler;

namespace rest {
class HttpCommTask;
class GeneralCommTask;
}

class HttpResponse : public GeneralResponse {
  friend class rest::GeneralCommTask;
  friend class rest::HttpCommTask;
  friend class RestBatchHandler;  // TODO must be removed

 public:
  static bool HIDE_PRODUCT_HEADER;

 private:
  explicit HttpResponse(ResponseCode code);

 public:
  bool isHeadResponse() const { return _isHeadResponse; }

 public:
  void setCookie(std::string const& name, std::string const& value,
                 int lifeTimeSeconds, std::string const& path,
                 std::string const& domain, bool secure, bool httpOnly);

  // In case of HEAD request, no body must be defined. However, the response
  // needs to know the size of body.
  void headResponse(size_t);

  // Returns a reference to the body. This reference is only valid as long as
  // http response exists. You can add data to the body by appending
  // information to the string buffer. Note that adding data to the body
  // invalidates any previously returned header. You must call header
  // again.
  basics::StringBuffer& body() override { return _body; }
  size_t bodySize() const;

  /// @brief set type of connection
  void setConnectionType(ConnectionType type) override {
    _connectionType = type;
  }

  /// @brief set content-type
  void setContentType(ContentType type) override { _contentType = type; }

  /// @brief set content-type from a string. this should only be used in
  /// cases when the content-type is user-defined
  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;
  }

  // you should call writeHeader only after the body has been created
  void writeHeader(basics::StringBuffer*) override;

 public:
  void reset(ResponseCode code) override final;

  void setPayload(GeneralRequest const*, arangodb::velocypack::Slice const&,
                  bool generateBody,
                  arangodb::velocypack::Options const&) override final;

 private:
  // the body must already be set. deflate is then run on the existing body
  int deflate(size_t = 16384);

 private:
  ConnectionType _connectionType;
  ContentType _contentType;
  bool _isHeadResponse;
  std::vector<std::string> _cookies;
  basics::StringBuffer _body;
  size_t _bodySize;
};
}

#endif