use azure_iot_hub::service::ServiceClient;

use std::error::Error;

#[tokio::main]
async fn main() -> Result<(), Box<dyn Error + Send + Sync>> {
    let iot_hub_connection_string = std::env::var("IOTHUB_CONNECTION_STRING")
        .expect("Set env variable IOTHUB_CONNECTION_STRING first!");

    let device_id = std::env::args()
        .nth(1)
        .expect("Please pass the device id as the first parameter");

    let payload = std::env::args()
        .nth(2)
        .expect("Please pass the payload as the second parameter");

    println!("Updating device twin for device: {device_id}");

    let service_client = ServiceClient::new_connection_string(iot_hub_connection_string, 3600)?;
    let json: String = serde_json::from_str(&payload).expect("Invalid JSON");
    let updated_twin = service_client
        .update_device_twin(device_id)
        .desired_properties(json)
        .await?;

    println!("Received device twin: {updated_twin:?}");

    Ok(())
}