buffet/http_utils.cc

// Copyright 2014 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "buffet/http_utils.h"

#include <algorithm>
#include <string.h>
#include <base/values.h>
#include <base/json/json_reader.h>
#include <base/json/json_writer.h>

#include "buffet/mime_utils.h"
#include "buffet/data_encoding.h"

namespace chromeos {
namespace http {

std::unique_ptr<Response> Get(std::string const& url) {
  Request request(url);
  return request.GetResponse();
}

std::string GetAsString(std::string const& url) {
  auto resp = Get(url);
  return resp ? resp->GetDataAsString() : std::string();
}

std::unique_ptr<Response> Head(std::string const& url) {
  Request request(url, request_type::kHead);
  return request.GetResponse();
}

std::unique_ptr<Response> PostText(std::string const& url,
                                   char const* data,
                                   char const* mime_type,
                                   HeaderList const& headers) {
  if (mime_type == nullptr) {
    mime_type = chromeos::mime::application::kWwwFormUrlEncoded;
  }

  return PostBinary(url, data, strlen(data), mime_type, headers);
}

std::unique_ptr<Response> SendRequest(char const* method,
                                      std::string const& url,
                                      void const* data,
                                      size_t data_size,
                                      char const* mime_type,
                                      HeaderList const& headers) {
  Request request(url, method);
  request.AddHeaders(headers);
  if (data_size > 0) {
    if (mime_type == nullptr) {
      mime_type = chromeos::mime::application::kOctet_stream;
    }
    request.SetContentType(mime_type);
    request.AddRequestBody(data, data_size);
  }
  return request.GetResponse();
}

std::unique_ptr<Response> PostBinary(std::string const& url, void const* data,
                                     size_t data_size, char const* mime_type,
                                     HeaderList const& headers) {
  return SendRequest(request_type::kPost, url,
                     data, data_size, mime_type, headers);
}

std::unique_ptr<Response> PostFormData(std::string const& url,
                                       FormFieldList const& data,
                                       HeaderList const& headers) {
  std::string encoded_data = chromeos::data_encoding::WebParamsEncode(data);
  return PostBinary(url, encoded_data.c_str(), encoded_data.size(),
                    chromeos::mime::application::kWwwFormUrlEncoded, headers);
}


std::unique_ptr<Response> PostJson(std::string const& url,
                                   base::Value const* json,
                                   HeaderList const& headers) {
  std::string data;
  if (json)
    base::JSONWriter::Write(json, &data);
  return PostBinary(url, data.c_str(), data.size(),
                    mime::application::kJson, headers);
}

std::unique_ptr<Response> PatchJson(std::string const& url,
                                    base::Value const* json,
                                    HeaderList const& headers) {
  std::string data;
  if (json)
    base::JSONWriter::Write(json, &data);
  return SendRequest(request_type::kPatch, url, data.c_str(), data.size(),
                     mime::application::kJson, headers);
}

std::unique_ptr<base::DictionaryValue> ParseJsonResponse(
    Response const* response, int* status_code, std::string* error_message) {
  std::unique_ptr<base::DictionaryValue> dict;
  if (response) {
    if (status_code)
      *status_code = response->GetStatusCode();

    // Make sure we have a correct content type. Do not try to parse
    // binary files, or HTML output. Limit to application/json and text/plain.
    auto content_type = mime::RemoveParameters(response->GetContentType());
    if (content_type == mime::application::kJson ||
        content_type == mime::text::kPlain) {
      std::string json = response->GetDataAsString();
      auto value = base::JSONReader::ReadAndReturnError(json,
                                                        base::JSON_PARSE_RFC,
                                                        nullptr,
                                                        error_message);
      if (value) {
        base::DictionaryValue* dict_value = nullptr;
        if (value->GetAsDictionary(&dict_value)) {
          dict.reset(dict_value);
        } else {
          delete value;
          if (error_message) {
            *error_message = "Reponse is not a valid JSON object";
          }
        }
      }
    }
    else if (error_message) {
      *error_message = "Unexpected response content type: " + content_type;
    }
  } else if (error_message) {
    *error_message = "NULL response.";
  }
  return dict;
}

} // namespace http
} // namespace chromeos