#include "include/yellowstone_stub.h"

#include <denali/denali_client.h>

#include "include/yellowstone.h"

YellowstoneStub::YellowstoneStub(z_cap_t yellowstone_cap)
    : yellowstone_stub_(EndpointClient::AdoptEndpoint(yellowstone_cap)) {}

glcr::ErrorOr<MappedMemoryRegion> YellowstoneStub::GetAhciConfig() {
  YellowstoneGetReq req{
      .type = kYellowstoneGetAhci,
  };
  ASSIGN_OR_RETURN(
      auto resp_cap_pair,
      (yellowstone_stub_->CallEndpointGetCap<YellowstoneGetReq,
                                             YellowstoneGetAhciResp>(req)));
  return MappedMemoryRegion::FromCapability(resp_cap_pair.second());
}

glcr::ErrorOr<ScopedDenaliClient> YellowstoneStub::GetDenali() {
  YellowstoneGetReq req{
      .type = kYellowstoneGetDenali,
  };
  ASSIGN_OR_RETURN(
      auto resp_and_cap,
      (yellowstone_stub_->CallEndpointGetCap<YellowstoneGetReq,
                                             YellowstoneGetDenaliResp>(req)));
  return ScopedDenaliClient(
      EndpointClient::AdoptEndpoint(resp_and_cap.second()), 0,
      resp_and_cap.first().lba_offset);
}

glcr::ErrorCode YellowstoneStub::Register(glcr::String name,
                                          const EndpointClient& client) {
  if (register_port_.empty()) {
    YellowstoneGetReq req{
        .type = kYellowstoneGetRegistration,
    };
    ASSIGN_OR_RETURN(
        auto resp_cap,
        (yellowstone_stub_->CallEndpointGetCap<
            YellowstoneGetReq, YellowstoneGetRegistrationResp>(req)));
    register_port_ = PortClient::AdoptPort(resp_cap.second());
  }

  return register_port_.WriteString(name, client.GetCap());
}