/*
 * Copyright (C) 2022 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 <sys/types.h>

#include <mutex>
#include <optional>
#include <set>
#include <string>
#include <unordered_map>
#include <vector>

#include <fruit/fruit.h>

#include "common/libs/fs/shared_fd.h"
#include "common/libs/utils/json.h"
#include "common/libs/utils/result.h"
#include "cvd_server.pb.h"
#include "host/commands/cvd/common_utils.h"
#include "host/commands/cvd/instance_lock.h"
#include "host/commands/cvd/selector/creation_analyzer.h"
#include "host/commands/cvd/selector/group_selector.h"
#include "host/commands/cvd/selector/instance_database.h"
#include "host/commands/cvd/selector/instance_database_types.h"
#include "host/commands/cvd/selector/instance_selector.h"
#include "host/commands/cvd/server_command/host_tool_target_manager.h"

namespace cuttlefish {

class InstanceManager {
 public:
  using CreationAnalyzer = selector::CreationAnalyzer;
  using CreationAnalyzerParam = CreationAnalyzer::CreationAnalyzerParam;
  using GroupCreationInfo = selector::GroupCreationInfo;
  using LocalInstanceGroup = selector::LocalInstanceGroup;
  using LocalInstance = selector::LocalInstance;
  using GroupSelector = selector::GroupSelector;
  using InstanceSelector = selector::InstanceSelector;
  using Queries = selector::Queries;
  using Query = selector::Query;
  template <typename T>
  using Set = selector::Set<T>;

  INJECT(InstanceManager(InstanceLockFileManager&, HostToolTargetManager&));

  // For cvd start
  Result<GroupCreationInfo> Analyze(const std::string& sub_cmd,
                                    const CreationAnalyzerParam& param,
                                    const ucred& credential);

  Result<LocalInstanceGroup> SelectGroup(const cvd_common::Args& selector_args,
                                         const cvd_common::Envs& envs,
                                         const uid_t uid);

  Result<LocalInstanceGroup> SelectGroup(const cvd_common::Args& selector_args,
                                         const Queries& extra_queries,
                                         const cvd_common::Envs& envs,
                                         const uid_t uid);

  Result<LocalInstance::Copy> SelectInstance(
      const cvd_common::Args& selector_args, const Queries& extra_queries,
      const cvd_common::Envs& envs, const uid_t uid);

  Result<LocalInstance::Copy> SelectInstance(
      const cvd_common::Args& selector_args, const cvd_common::Envs& envs,
      const uid_t uid);

  bool HasInstanceGroups(const uid_t uid);
  Result<void> SetInstanceGroup(const uid_t uid,
                                const selector::GroupCreationInfo& group_info);
  Result<void> SetBuildId(const uid_t uid, const std::string& group_name,
                          const std::string& build_id);
  void RemoveInstanceGroup(const uid_t uid, const std::string&);

  cvd::Status CvdClear(const SharedFD& out, const SharedFD& err);
  Result<cvd::Status> CvdFleet(const uid_t uid, const SharedFD& out,
                               const SharedFD& err,
                               const std::vector<std::string>& fleet_cmd_args);
  static Result<std::string> GetCuttlefishConfigPath(const std::string& home);

  Result<std::optional<InstanceLockFile>> TryAcquireLock(int instance_num);

  Result<std::vector<LocalInstanceGroup>> FindGroups(const uid_t uid,
                                                     const Query& query) const;
  Result<std::vector<LocalInstanceGroup>> FindGroups(
      const uid_t uid, const Queries& queries) const;
  Result<std::vector<LocalInstance::Copy>> FindInstances(
      const uid_t uid, const Query& query) const;
  Result<std::vector<LocalInstance::Copy>> FindInstances(
      const uid_t uid, const Queries& queries) const;

  Result<LocalInstanceGroup> FindGroup(const uid_t uid,
                                       const Query& query) const;
  Result<LocalInstanceGroup> FindGroup(const uid_t uid,
                                       const Queries& queries) const;
  Result<Json::Value> Serialize(const uid_t uid);
  Result<void> LoadFromJson(const uid_t uid, const Json::Value&);

 private:
  Result<cvd::Status> CvdFleetImpl(const uid_t uid, const SharedFD& out,
                                   const SharedFD& err);
  struct StatusCommandOutput {
    std::string stderr_msg;
    Json::Value stdout_json;
  };
  Result<StatusCommandOutput> IssueStatusCommand(
      const selector::LocalInstanceGroup& group, const SharedFD& err);
  Result<void> IssueStopCommand(const SharedFD& out, const SharedFD& err,
                                const std::string& config_file_path,
                                const selector::LocalInstanceGroup& group);
  Result<std::string> StopBin(const std::string& host_android_out);

  selector::InstanceDatabase& GetInstanceDB(const uid_t uid);
  InstanceLockFileManager& lock_manager_;
  HostToolTargetManager& host_tool_target_manager_;
  mutable std::mutex instance_db_mutex_;
  std::unordered_map<uid_t, selector::InstanceDatabase> instance_dbs_;
};

}  // namespace cuttlefish