component/
lib.rs

Help
1// Copyright 2023 The Fuchsia Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5mod args;
6
7use crate::args::*;
8use anyhow::Result;
9use component_debug::cli::*;
10use component_debug::config::resolve_raw_config_overrides;
11use component_debug::copy::copy_cmd;
12use fuchsia_component::client::{connect_to_protocol, connect_to_protocol_at_path};
13use futures::FutureExt;
14use socket_to_stdio::Stdout;
15use {fidl_fuchsia_dash as fdash, fidl_fuchsia_sys2 as fsys};
16
17pub async fn exec() -> Result<()> {
18    let args: ComponentArgs = argh::from_env();
19
20    let writer = std::io::stdout();
21    let realm_query =
22        connect_to_protocol_at_path::<fsys::RealmQueryMarker>("/svc/fuchsia.sys2.RealmQuery.root")?;
23    let route_validator = connect_to_protocol_at_path::<fsys::RouteValidatorMarker>(
24        "/svc/fuchsia.sys2.RouteValidator.root",
25    )?;
26    let lifecycle_controller = connect_to_protocol_at_path::<fsys::LifecycleControllerMarker>(
27        "/svc/fuchsia.sys2.LifecycleController.root",
28    )?;
29    let config_override = connect_to_protocol_at_path::<fsys::ConfigOverrideMarker>(
30        "/svc/fuchsia.sys2.ConfigOverride.root",
31    )?;
32
33    match args.subcommand {
34        ComponentSubcommand::Show(args) => {
35            show_cmd_print(args.query, realm_query, writer, true).await
36        }
37        ComponentSubcommand::Create(args) => {
38            let config_overrides = resolve_raw_config_overrides(
39                &realm_query,
40                &args.moniker,
41                &args.url.to_string(),
42                &args.config,
43            )
44            .await?;
45            create_cmd(args.url, args.moniker, config_overrides, lifecycle_controller, writer).await
46        }
47        ComponentSubcommand::Destroy(args) => {
48            destroy_cmd(args.query, lifecycle_controller, realm_query, writer).await
49        }
50        ComponentSubcommand::Resolve(args) => {
51            resolve_cmd(args.query, lifecycle_controller, realm_query, writer).await
52        }
53        ComponentSubcommand::Explore(args) => {
54            // TODO(https://fxbug.dev/296283299): Verify that the optional Launcher protocol is
55            // available before connecting.
56            let dash_launcher = connect_to_protocol::<fdash::LauncherMarker>()?;
57            // TODO(https://fxbug.dev/42077838): Use Stdout::raw instead, when a command is
58            // not provided.
59            let stdout = Stdout::buffered();
60
61            #[allow(clippy::large_futures)]
62            explore_cmd(
63                args.query,
64                args.ns_layout,
65                args.command,
66                args.tools,
67                dash_launcher,
68                realm_query,
69                stdout,
70            )
71            .await
72        }
73        ComponentSubcommand::Reload(args) => {
74            reload_cmd(args.query, lifecycle_controller, realm_query, writer).await
75        }
76        ComponentSubcommand::Start(args) => {
77            start_cmd(args.query, lifecycle_controller, realm_query, writer).await
78        }
79        ComponentSubcommand::Stop(args) => {
80            stop_cmd(args.query, lifecycle_controller, realm_query, writer).await
81        }
82        ComponentSubcommand::Doctor(args) => {
83            doctor_cmd_print(args.query, route_validator, realm_query, writer).await
84        }
85        ComponentSubcommand::Capability(args) => {
86            capability_cmd(args.capability_name, realm_query, writer).await
87        }
88        ComponentSubcommand::List(args) => {
89            list_cmd_print(args.filter, args.verbose, realm_query, writer).await
90        }
91        ComponentSubcommand::Graph(args) => {
92            graph_cmd(args.filter, args.orientation, realm_query, writer).await
93        }
94        ComponentSubcommand::Run(args) => {
95            let config_overrides = resolve_raw_config_overrides(
96                &realm_query,
97                &args.moniker,
98                &args.url.to_string(),
99                &args.config,
100            )
101            .await?;
102            run_cmd(
103                args.moniker,
104                args.url,
105                args.recreate,
106                args.connect_stdio,
107                config_overrides,
108                || {
109                    async {
110                        connect_to_protocol_at_path::<fsys::LifecycleControllerMarker>(
111                            "/svc/fuchsia.sys2.LifecycleController.root",
112                        )
113                    }
114                    .boxed()
115                },
116                writer,
117            )
118            .await
119        }
120        ComponentSubcommand::Copy(args) => {
121            copy_cmd(&realm_query, args.paths, args.verbose, writer).await
122        }
123        ComponentSubcommand::Storage(args) => match args.subcommand {
124            StorageSubcommand::Copy(copy_args) => {
125                storage_copy_cmd(
126                    args.provider,
127                    args.capability,
128                    copy_args.source_path,
129                    copy_args.destination_path,
130                    realm_query,
131                )
132                .await
133            }
134            StorageSubcommand::Delete(delete_args) => {
135                storage_delete_cmd(args.provider, args.capability, delete_args.path, realm_query)
136                    .await
137            }
138            StorageSubcommand::List(list_args) => {
139                storage_list_cmd(
140                    args.provider,
141                    args.capability,
142                    list_args.path,
143                    realm_query,
144                    writer,
145                )
146                .await
147            }
148            StorageSubcommand::MakeDirectory(make_dir_args) => {
149                storage_make_directory_cmd(
150                    args.provider,
151                    args.capability,
152                    make_dir_args.path,
153                    realm_query,
154                )
155                .await
156            }
157        },
158        ComponentSubcommand::Collection(args) => match args.subcommand {
159            CollectionSubcommand::List(_) => collection_list_cmd(realm_query, writer).await,
160            CollectionSubcommand::Show(show_args) => {
161                collection_show_cmd(show_args.query, realm_query, writer).await
162            }
163        },
164        ComponentSubcommand::Config(args) => match args.subcommand {
165            ConfigSubcommand::Set(SetArgs { query, key_values, reload }) => {
166                config_set_cmd(
167                    query,
168                    key_values,
169                    reload,
170                    lifecycle_controller,
171                    realm_query,
172                    config_override,
173                    writer,
174                )
175                .await
176            }
177            ConfigSubcommand::Unset(UnsetArgs { query, reload }) => {
178                config_unset_cmd(
179                    query,
180                    reload,
181                    lifecycle_controller,
182                    realm_query,
183                    config_override,
184                    writer,
185                )
186                .await
187            }
188            ConfigSubcommand::List(ConfigListArgs { query }) => {
189                config_list_cmd(query, realm_query, writer).await
190            }
191        },
192    }
193}
component/lib.rs

component/
lib.rs
