// 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"

namespace debug_agent {

namespace {

using debug_ipc::ProcessTreeRecord;

ProcessTreeRecord GetProcessTreeFrom(const JobHandle& job) {
  ProcessTreeRecord result;
  result.type = ProcessTreeRecord::Type::kJob;
  result.koid = job.GetKoid();
  result.name = job.GetName();

  for (const auto& child_process : job.GetChildProcesses()) {
    ProcessTreeRecord& proc_record = result.children.emplace_back();
    proc_record.type = ProcessTreeRecord::Type::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));

  return result;
}

}  // namespace

ProcessTreeRecord SystemInterface::GetProcessTree() const {
  if (std::unique_ptr<JobHandle> root_job = GetRootJob())
    return GetProcessTreeFrom(*root_job);
  return 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;
}

}  // namespace debug_agent