src/developer/ffx/plugins/component/capability/src/lib.rs

// Copyright 2022 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.

use async_trait::async_trait;
use component_debug::cli::capability_cmd;
use errors::FfxError;
use ffx_component::rcs::connect_to_realm_query;
use ffx_component_capability_args::ComponentCapabilityCommand;
use fho::{FfxMain, FfxTool, SimpleWriter};
use fidl_fuchsia_developer_remotecontrol as rc;

#[derive(FfxTool)]
pub struct CapabilityTool {
    #[command]
    cmd: ComponentCapabilityCommand,
    rcs: rc::RemoteControlProxy,
}

fho::embedded_plugin!(CapabilityTool);

#[async_trait(?Send)]
impl FfxMain for CapabilityTool {
    type Writer = SimpleWriter;
    async fn main(self, writer: Self::Writer) -> fho::Result<()> {
        let realm_query = connect_to_realm_query(&self.rcs).await?;
        // All errors from component_debug library are user-visible.
        capability_cmd(self.cmd.capability, realm_query, writer)
            .await
            .map_err(|e| FfxError::Error(e, 1))?;
        Ok(())
    }
}