host/commands/cvd/request_context.h

/*
 * Copyright (C) 2023 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#pragma once

#include <atomic>
#include <vector>

#include "host/commands/cvd/command_sequence.h"
#include "host/commands/cvd/instance_manager.h"
#include "host/commands/cvd/server_client.h"
#include "host/commands/cvd/server_command/host_tool_target_manager.h"
#include "host/commands/cvd/server_command/server_handler.h"
#include "host/commands/cvd/server_command/subprocess_waiter.h"
#include "host/libs/web/build_api.h"

namespace cuttlefish {

class CvdServer;

class RequestContext {
 public:
  RequestContext(CvdServer& cvd_server, InstanceManager& instance_manager,
                 BuildApi& build_api,
                 HostToolTargetManager& host_tool_target_manager,
                 std::atomic<bool>& acloud_translator_optout);

  Result<CvdServerHandler*> Handler(const RequestWithStdio& request);

 private:
  void InstantiateHandlers();

  CvdServer& cvd_server_;
  std::vector<std::unique_ptr<CvdServerHandler>> request_handlers_;
  InstanceManager& instance_manager_;
  BuildApi& build_api_;
  SubprocessWaiter subprocess_waiter_;
  InstanceLockFileManager lock_file_manager_;
  HostToolTargetManager& host_tool_target_manager_;
  CommandSequenceExecutor command_sequence_executor_;
  std::atomic<bool>& acloud_translator_optout_;
};

Result<CvdServerHandler*> RequestHandler(
    const RequestWithStdio& request,
    const std::vector<std::unique_ptr<CvdServerHandler>>& handlers);

}  // namespace cuttlefish