use azure_messaging_servicebus::prelude::*;

#[tokio::main]
async fn main() {
    let service_bus_namespace = std::env::var("AZURE_SERVICE_BUS_NAMESPACE")
        .expect("Please set AZURE_SERVICE_BUS_NAMESPACE env variable first!");

    let topic_name =
        std::env::var("AZURE_TOPIC_NAME").expect("Please set AZURE_TOPIC_NAME env variable first!");

    let subscription_name =
        std::env::var("SUBSCRIPTION").expect("Please set SUBSCRIPTION env variable first!");

    let policy_name = std::env::var("AZURE_POLICY_NAME")
        .expect("Please set AZURE_POLICY_NAME env variable first!");

    let policy_key =
        std::env::var("AZURE_POLICY_KEY").expect("Please set AZURE_POLICY_KEY env variable first!");

    let http_client = azure_core::new_http_client();

    let client = TopicClient::new(
        http_client,
        service_bus_namespace,
        topic_name,
        policy_name,
        policy_key,
    )
    .expect("Failed to create client");

    let sender = client.topic_sender();
    let receiver = client.subscription_receiver(&subscription_name);

    let message_to_send = "hello, world!";

    sender
        .send_message(message_to_send, None)
        .await
        .expect("Failed to send message while testing receive");

    println!("Sent Message: {message_to_send}");

    let received_message = receiver
        .receive_and_delete_message()
        .await
        .expect("Failed to receive message");

    println!("Received Message: {received_message}");
}