| // Copyright 2019 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. |
| |
| mod battery_manager; |
| mod battery_simulator; |
| mod power; |
| |
| use { |
| crate::battery_manager::{BatteryManager, BatterySimulationStateObserver}, |
| crate::battery_simulator::SimulatedBatteryInfoSource, |
| anyhow::Error, |
| fidl_fuchsia_power::BatteryManagerRequestStream, |
| fidl_fuchsia_power_test as spower, fuchsia_async as fasync, |
| fuchsia_component::server::ServiceFs, |
| fuchsia_syslog::{self as syslog, fx_log_err, fx_log_info}, |
| futures::prelude::*, |
| std::sync::{Arc, Weak}, |
| }; |
| |
| static LOG_TAG: &str = "battery_manager"; |
| static LOG_VERBOSITY: i32 = 1; |
| |
| enum IncomingService { |
| BatteryManager(BatteryManagerRequestStream), |
| BatterySimulator(spower::BatterySimulatorRequestStream), |
| } |
| |
| #[fasync::run_singlethreaded] |
| async fn main() -> Result<(), Error> { |
| syslog::init_with_tags(&[LOG_TAG]).expect("Can't init logger"); |
| fx_log_info!("starting up"); |
| |
| let mut fs = ServiceFs::new(); |
| |
| let battery_manager = Arc::new(BatteryManager::new()); |
| let battery_manager_clone = battery_manager.clone(); |
| |
| let f = power::watch_power_device(battery_manager_clone); |
| let battery_simulator = Arc::new(SimulatedBatteryInfoSource::new( |
| battery_manager.get_battery_info_copy(), |
| Arc::downgrade(&battery_manager) as Weak<dyn BatterySimulationStateObserver>, |
| )); |
| |
| fasync::Task::spawn(f.unwrap_or_else(|e| { |
| fx_log_err!("watch_power_device failed {:?}", e); |
| })) |
| .detach(); |
| |
| fs.dir("svc") |
| .add_fidl_service(IncomingService::BatteryManager) |
| .add_fidl_service(IncomingService::BatterySimulator); |
| |
| fs.take_and_serve_directory_handle()?; |
| |
| fs.for_each_concurrent(None, |request| { |
| let battery_manager = battery_manager.clone(); |
| let battery_simulator = battery_simulator.clone(); |
| |
| async move { |
| match request { |
| IncomingService::BatteryManager(stream) => { |
| let res = battery_manager.serve(stream).await; |
| if let Err(e) = res { |
| fx_log_err!("BatteryManager failed {}", e); |
| } |
| } |
| IncomingService::BatterySimulator(stream) => { |
| let res = stream |
| .err_into() |
| .try_for_each_concurrent(None, |request| { |
| let battery_simulator = battery_simulator.clone(); |
| let battery_manager = battery_manager.clone(); |
| async move { |
| match request { |
| spower::BatterySimulatorRequest::DisconnectRealBattery { |
| .. |
| } => { |
| battery_simulator |
| .update_simulation( |
| true, |
| battery_manager.get_battery_info_copy(), |
| ) |
| .await?; |
| } |
| spower::BatterySimulatorRequest::ReconnectRealBattery { |
| .. |
| } => { |
| battery_simulator |
| .update_simulation( |
| false, |
| battery_manager.get_battery_info_copy(), |
| ) |
| .await?; |
| } |
| spower::BatterySimulatorRequest::IsSimulating { |
| responder, |
| .. |
| } => { |
| let info = battery_manager.is_simulating(); |
| responder.send(info)?; |
| } |
| _ => { |
| battery_simulator.handle_request(request).await?; |
| } |
| } |
| Ok::<(), Error>(()) |
| } |
| }) |
| .await; |
| |
| if let Err(e) = res { |
| fx_log_err!("BatterySimulator failed {}", e); |
| } |
| } |
| } |
| } |
| }) |
| .await; |
| |
| fx_log_info!("stopping battery_manager"); |
| Ok(()) |
| } |