use anyhow::{Result, Context};
use btleplug::api::{BDAddr};
use tracing_subscriber;
use tokio::io::AsyncReadExt;
use std::path::Path;
use tokio::fs::File;
use tracing::Level;
use std::env;
use tracing_subscriber::fmt::format::FmtSpan;
use tokio::time;
use std::time::Duration;
use m365::{
AuthToken,
ScooterScanner,
LoginRequest,
ConnectionHelper,
TailLight
};
async fn load_token() -> Result<AuthToken> {
let path = Path::new(".mi-token");
tracing::debug!("Opening token: {:?}", path);
let mut f = File::open(path).await?;
let mut buffer : AuthToken = [0; 12];
f.read(&mut buffer).await?;
Ok(buffer)
}
#[tokio::main(flavor = "multi_thread")]
async fn main() -> Result<()>{
tracing_subscriber::fmt()
.with_max_level(Level::INFO)
.with_span_events(FmtSpan::CLOSE)
.init();
let args: Vec<String> = env::args().collect();
if args.len() < 2 || args[1].is_empty() {
panic!("First argument is scooter mac address");
}
let token = load_token().await
.with_context(|| "Could not load registration token")?;
let mac = BDAddr::from_str_delim(&args[1]).expect("Invalid mac address");
tracing::info!("Searching scooter with address: {}", mac);
let mut scanner = ScooterScanner::new().await?;
let scooter = scanner.wait_for(&mac).await?;
let device = scanner.peripheral(&scooter).await?;
let connection = ConnectionHelper::new(&device);
connection.reconnect().await?;
let mut request = LoginRequest::new(&device, &token).await?;
let mut session = request.start().await?;
tracing::info!("Logged in with success, reading data...");
tracing::info!(" Supplementary info {:?}", session.supplementary_info().await?);
tracing::info!(" Cruise enabled: {}", session.is_cruise_on().await?);
tracing::info!(" Tail light enabled: {:?}", session.tail_light().await?);
tracing::info!(" Cruise status: {}, Switching on", session.is_cruise_on().await?);
session.set_cruise(true).await?;
tracing::info!(" Supplementary info {:?}", session.supplementary_info().await?);
tracing::info!(" Cruise status: {}, Switching off", session.is_cruise_on().await?);
session.set_cruise(false).await?;
tracing::info!(" Cruise status: {}", session.is_cruise_on().await?);
tracing::info!(" Supplementary info {:?}", session.supplementary_info().await?);
tracing::info!(" Tail light enabled: {:?}, Switching on", session.tail_light().await?);
session.set_tail_light(TailLight::Always).await?;
tracing::info!(" Supplementary info {:?}", session.supplementary_info().await?);
tracing::info!(" Tail light enabled: {:?}, Switching off", session.tail_light().await?);
time::sleep(Duration::from_secs(2)).await;
session.set_tail_light(TailLight::Off).await?;
tracing::info!(" Tail light enabled: {:?}", session.tail_light().await?);
tracing::info!(" Supplementary info {:?}", session.supplementary_info().await?);
Ok(())
}