remoting/base/protobuf_http_request.h

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

#ifndef REMOTING_BASE_PROTOBUF_HTTP_REQUEST_H_
#define REMOTING_BASE_PROTOBUF_HTTP_REQUEST_H_

#include "base/bind.h"
#include "base/callback.h"
#include "remoting/base/protobuf_http_request_base.h"

namespace google {
namespace protobuf {
class MessageLite;
}  // namespace protobuf
}  // namespace google

namespace remoting {

// A simple unary request.
class ProtobufHttpRequest final : public ProtobufHttpRequestBase {
 public:
  template <typename ResponseType>
  using ResponseCallback =
      base::OnceCallback<void(const ProtobufHttpStatus& status,
                              std::unique_ptr<ResponseType> response)>;

  explicit ProtobufHttpRequest(
      std::unique_ptr<ProtobufHttpRequestConfig> config);
  ~ProtobufHttpRequest() override;

  // Sets the amount of time to wait before giving up on a given network request
  // and considering it an error. The default value is 30s. Set it to zero to
  // disable timeout.
  void SetTimeoutDuration(base::TimeDelta timeout_duration);

  // Sets the response callback. |ResponseType| needs to be a protobuf message
  // type.
  template <typename ResponseType>
  void SetResponseCallback(ResponseCallback<ResponseType> callback) {
    auto response = std::make_unique<ResponseType>();
    response_message_ = response.get();
    response_callback_ = base::BindOnce(
        [](std::unique_ptr<ResponseType> response,
           ResponseCallback<ResponseType> callback,
           const ProtobufHttpStatus& status) {
          if (!status.ok()) {
            response.reset();
          }
          std::move(callback).Run(status, std::move(response));
        },
        std::move(response), std::move(callback));
  }

 private:
  // ProtobufHttpRequestBase implementations.
  void OnAuthFailed(const ProtobufHttpStatus& status) override;
  void StartRequestInternal(
      network::mojom::URLLoaderFactory* loader_factory) override;
  base::TimeDelta GetRequestTimeoutDuration() const override;

  void OnResponse(std::unique_ptr<std::string> response_body);

  // Parses |response_body| and writes it to |response_message_|.
  ProtobufHttpStatus ParseResponse(std::unique_ptr<std::string> response_body);

  base::TimeDelta timeout_duration_ = base::TimeDelta::FromSeconds(30);

  // This is owned by |response_callback_|.
  google::protobuf::MessageLite* response_message_;
  base::OnceCallback<void(const ProtobufHttpStatus&)> response_callback_;
};

}  // namespace remoting

#endif  // REMOTING_BASE_PROTOBUF_HTTP_REQUEST_H_