// Copyright 2020 The Fuchsia 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 "src/developer/debug/debug_agent/system_interface.h"

#include "src/developer/debug/debug_agent/component_manager.h"
#include "src/developer/debug/shared/logging/logging.h"

namespace debug_agent {

namespace {

debug_ipc::ProcessTreeRecord GetProcessTreeFrom(const JobHandle& job,
                                                const ComponentManager& component_manager) {
  debug_ipc::ProcessTreeRecord result;
  result.type = debug_ipc::TaskType::kJob;
  result.koid = job.GetKoid();
  result.name = job.GetName();

  result.components = component_manager.FindComponentInfo(job.GetKoid());

  for (const auto& child_process : job.GetChildProcesses()) {
    debug_ipc::ProcessTreeRecord& proc_record = result.children.emplace_back();
    proc_record.type = debug_ipc::TaskType::kProcess;
    proc_record.koid = child_process->GetKoid();
    proc_record.name = child_process->GetName();
  }

  for (const auto& child_job : job.GetChildJobs())
    result.children.push_back(GetProcessTreeFrom(*child_job, component_manager));

  return result;
}

}  // namespace

debug_ipc::ProcessTreeRecord SystemInterface::GetProcessTree() {
  const ComponentManager& component_manager = GetComponentManager();
  if (std::unique_ptr<JobHandle> root_job = GetRootJob())
    return GetProcessTreeFrom(*root_job, component_manager);
  return debug_ipc::ProcessTreeRecord();
}

std::unique_ptr<JobHandle> SystemInterface::GetJob(zx_koid_t job_koid) const {
  if (std::unique_ptr<JobHandle> root_job = GetRootJob())
    return root_job->FindJob(job_koid);
  return nullptr;
}

std::unique_ptr<ProcessHandle> SystemInterface::GetProcess(zx_koid_t process_koid) const {
  if (std::unique_ptr<JobHandle> root_job = GetRootJob())
    return root_job->FindProcess(process_koid);
  return nullptr;
}

zx_koid_t SystemInterface::GetParentJobKoid(zx_koid_t job) {
  if (auto it = parent_jobs_.find(job); it != parent_jobs_.end())
    return it->second;
  RefreshParentJobs();
  return parent_jobs_[job];  // Absent value becomes 0, aka ZX_KOID_INVALID.
}

void SystemInterface::RefreshParentJobs() {
  DEBUG_LOG(Agent) << "RefreshParentJobs called";
  parent_jobs_.clear();
  debug_ipc::ProcessTreeRecord record = GetProcessTree();
  std::function<void(const debug_ipc::ProcessTreeRecord&, zx_koid_t)> visit_each_record =
      [&](auto record, zx_koid_t parent_koid) {
        if (record.type == debug_ipc::TaskType::kJob) {
          parent_jobs_.emplace(record.koid, parent_koid);
          for (const auto& child : record.children) {
            visit_each_record(child, record.koid);
          }
        }
      };
  visit_each_record(record, ZX_KOID_INVALID);
}

}  // namespace debug_agent