#include "webcc/rest_request_handler.h"

#include <utility>  // for move()
#include <vector>

#include "webcc/logger.h"
#include "webcc/url.h"

namespace webcc {

bool RestRequestHandler::Bind(RestServicePtr service, const std::string& url,
                              bool is_regex) {
  return service_manager_.AddService(service, url, is_regex);
}

void RestRequestHandler::HandleConnection(HttpConnectionPtr connection) {
  const HttpRequest& request = connection->request();

  Url url(request.url(), true);

  if (!url.IsValid()) {
    connection->SendResponse(HttpStatus::kBadRequest);
    return;
  }

  std::vector<std::string> sub_matches;
  RestServicePtr service = service_manager_.GetService(url.path(),
                                                       &sub_matches);
  if (!service) {
    LOG_WARN("No service matches the URL: %s", url.path().c_str());
    connection->SendResponse(HttpStatus::kBadRequest);
    return;
  }

  UrlQuery query;
  if (request.method() == kHttpGet) {
    // Suppose URL query is only available for HTTP GET.
    Url::SplitQuery(url.query(), &query);
  }

  std::string content;
  bool ok = service->Handle(request.method(), sub_matches, query,
                            request.content(), &content);
  if (!ok) {
    connection->SendResponse(HttpStatus::kBadRequest);
    return;
  }

  if (!content.empty()) {
    connection->SetResponseContent(std::move(content), kAppJsonUtf8);
  }

  if (request.method() == kHttpPost) {
    connection->SendResponse(HttpStatus::kCreated);
  } else {
    connection->SendResponse(HttpStatus::kOK);
  }
}

}  // namespace webcc