cups_proxy/mhd_http_request.cc

// Copyright 2019 The ChromiumOS Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "cups_proxy/mhd_http_request.h"

#include <algorithm>
#include <string_view>
#include <utility>
#include <vector>

namespace cups_proxy {

MHDHttpRequest::MHDHttpRequest() : chunked_(false) {}

void MHDHttpRequest::SetStatusLine(std::string_view method,
                                   std::string_view url,
                                   std::string_view version) {
  method_ = std::string(method);
  url_ = std::string(url);
  version_ = std::string(version);
}

void MHDHttpRequest::AddHeader(std::string_view key, std::string_view value) {
  // strip 100-continue message from ipp request
  if (key == "Expect" && value == "100-continue") {
    return;
  }

  // strip chunked header from ipp request
  if (key == "Transfer-Encoding" && value == "chunked") {
    chunked_ = true;
    return;
  }

  headers_[std::string(key)] = std::string(value);
}

void MHDHttpRequest::Finalize() {
  if (chunked_) {
    AddHeader("Content-Length", std::to_string(body_.size()));
  }
}

void MHDHttpRequest::PushToBody(std::string_view data) {
  body_.insert(body_.end(), data.begin(), data.end());
}

}  // namespace cups_proxy