// Copyright 2018 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/sys/appmgr/hub/hub.h"

#include <fbl/ref_ptr.h>

#include "src/lib/storage/vfs/cpp/pseudo_file.h"
#include "src/lib/storage/vfs/cpp/vnode.h"
#include "src/sys/appmgr/hub/hub_info.h"

namespace component {

Hub::Hub(fbl::RefPtr<fs::PseudoDir> root) : dir_(root) {}

zx_status_t Hub::AddEntry(fbl::String name, fbl::RefPtr<fs::Vnode> vn) {
  return dir_->AddEntry(std::move(name), std::move(vn));
}

zx_status_t Hub::AddEntry(fbl::String name, fbl::String value) {
  return dir_->AddEntry(std::move(name),
                        fbl::MakeRefCounted<fs::BufferedPseudoFile>([value](fbl::String* output) {
                          *output = value;
                          return ZX_OK;
                        }));
}

zx_status_t Hub::EnsureComponentDir() {
  if (component_dir_) {
    return ZX_OK;
  }
  component_dir_ = fbl::MakeRefCounted<fs::PseudoDir>();
  return AddEntry("c", component_dir_);
}

zx_status_t Hub::AddComponent(const HubInfo& hub_info) {
  if (!component_dir_) {
    return ZX_ERR_NOT_SUPPORTED;
  }
  fbl::RefPtr<fs::PseudoDir> component_instance_dir;
  fbl::RefPtr<fs::Vnode> component_instance_vnode;
  zx_status_t status = component_dir_->Lookup(hub_info.label(), &component_instance_vnode);
  if (status == ZX_ERR_NOT_FOUND) {
    component_instance_dir = fbl::MakeRefCounted<fs::PseudoDir>();
    component_dir_->AddEntry(hub_info.label(), component_instance_dir);
  } else {
    component_instance_dir =
        fbl::RefPtr<fs::PseudoDir>(static_cast<fs::PseudoDir*>(component_instance_vnode.get()));
  }
  return component_instance_dir->AddEntry(hub_info.koid(), hub_info.hub_dir());
}

zx_status_t Hub::RemoveComponent(const HubInfo& hub_info) {
  fbl::RefPtr<fs::Vnode> component_instance_vnode;
  zx_status_t status = component_dir_->Lookup(hub_info.label(), &component_instance_vnode);
  if (status == ZX_OK) {
    auto component_instance_dir =
        fbl::RefPtr<fs::PseudoDir>(static_cast<fs::PseudoDir*>(component_instance_vnode.get()));
    status = component_instance_dir->RemoveEntry(hub_info.koid());
    if (component_instance_dir->IsEmpty()) {
      component_dir_->RemoveEntry(hub_info.label());
    }
    return status;
  }
  return ZX_ERR_NOT_FOUND;
}

}  // namespace component