// ----------------------------------------------------------------------
// File: RestApiResponse.hh
// Author: Cedric Caffy - CERN
// ----------------------------------------------------------------------

/************************************************************************
 * EOS - the CERN Disk Storage System                                   *
 * Copyright (C) 2013 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/>.*
 ************************************************************************/

#ifndef EOS_RESTAPIRESPONSE_HH
#define EOS_RESTAPIRESPONSE_HH

#include "common/http/HttpResponse.hh"
#include "common/http/PlainHttpResponse.hh"
#include "mgm/Namespace.hh"
#include <memory>

EOSMGMRESTNAMESPACE_BEGIN

/**
 * This class allows to create a RestAPI http response
 * from a model object
 */
template<typename Model>
class RestApiResponse
{
public:
  /**
   * Constructor with the model
   * @param model the model that will be used to create the http response
   */
  RestApiResponse();
  RestApiResponse(const common::HttpResponse::ResponseCodes retCode);
  RestApiResponse(std::shared_ptr<Model> model,
                  const common::HttpResponse::ResponseCodes retCode);
  RestApiResponse(std::shared_ptr<Model> model,
                  const common::HttpResponse::ResponseCodes retCode,
                  const common::HttpResponse::HeaderMap& responseHeader);

  /**
   * Returns the actual HttpResponse created from the model and the return code
   * of this instance
   * The body will be the JSON representation of the model
   */
  common::HttpResponse* getHttpResponse() const;

  static RestApiResponse<Model> createResponse(std::shared_ptr<Model> model,
      const common::HttpResponse::ResponseCodes code);
  static RestApiResponse<Model> createResponse(std::shared_ptr<Model> model,
      const common::HttpResponse::ResponseCodes code,
      const common::HttpResponse::HeaderMap& responseHeader);

private:
  std::shared_ptr<Model> mModel;
  common::HttpResponse::ResponseCodes mRetCode;
  mutable common::HttpResponse::HeaderMap mHeaderMap;
};

template<typename Model>
RestApiResponse<Model>::RestApiResponse() : mRetCode(
    common::HttpResponse::ResponseCodes::OK) {}

template<typename Model>
RestApiResponse<Model>::RestApiResponse(const
                                        common::HttpResponse::ResponseCodes retCode) : mRetCode(retCode) {}

template<typename Model>
RestApiResponse<Model>::RestApiResponse(std::shared_ptr<Model> model,
                                        const common::HttpResponse::ResponseCodes retCode) :
  mModel(model), mRetCode(retCode) {}

template<typename Model>
RestApiResponse<Model>::RestApiResponse(std::shared_ptr<Model> model,
                                        const common::HttpResponse::ResponseCodes retCode,
                                        const common::HttpResponse::HeaderMap& responseHeader) :
  mModel(model), mRetCode(retCode), mHeaderMap(responseHeader) {}

template<typename Model>
inline common::HttpResponse* RestApiResponse<Model>::getHttpResponse() const
{
  common::HttpResponse* response = new common::PlainHttpResponse();

  if (mModel) {
    mHeaderMap["application/type"] = "json";
    response->SetHeaders(mHeaderMap);
    std::stringstream ss;
    mModel->jsonify(ss);
    response->SetBody(ss.str());
  }

  response->SetResponseCode(mRetCode);
  return response;
}

template<>
inline common::HttpResponse* RestApiResponse<void>::getHttpResponse() const
{
  common::HttpResponse* response = new common::PlainHttpResponse();
  response->SetResponseCode(mRetCode);
  return response;
}

EOSMGMRESTNAMESPACE_END

#endif // EOS_RESTAPIRESPONSE_HH