use tbot::{
    markup::{inline_code, markdown_v2},
    prelude::*,
    types::{
        parameters::{Photo, Text},
        shipping, LabeledPrice,
    },
    Bot,
};

const PAYLOAD: &str = "crab";
const START_PARAMETER: &str = "crab";
const TITLE: &str = "A crab";
const PHOTO: Photo =
    Photo::new("https://www.rustacean.net/assets/rustacean-flat-happy.png");
const DESCRIPTION: &str = "Have you ever come across a heisenbug in your \
program? No more! Our crab will take all bugs out of your program for you.";
const CURRENCY: &str = "USD";
const PRICE: &[LabeledPrice] = &[LabeledPrice::new(TITLE, 1_00)];
const DELIVERY: &[shipping::Option] = &[shipping::Option::new(
    "crab",
    "Delivery Crab",
    &[LabeledPrice::new("At your home", 1_00)],
)];
const SUCCESS: &str = "Thanks! Your crab is already on its way.";

#[tokio::main]
async fn main() {
    // The one that you'd get from BotFather after connecting a payment provider
    // to your bot.
    let provider_token: &'static str =
        Box::leak(Box::new(std::env::var("PROVIDER_TOKEN").unwrap()));
    let mut bot = Bot::from_env("BOT_TOKEN").event_loop();

    bot.start(move |context| async move {
        let call_result = if context.text.value == START_PARAMETER {
            let mut invoice = context.send_invoice(
                TITLE,
                DESCRIPTION,
                PAYLOAD,
                provider_token,
                START_PARAMETER,
                CURRENCY,
                PRICE,
            );
            invoice = invoice.photo(PHOTO).is_flexible(true);

            invoice.call().await
        } else {
            let error_message = markdown_v2((
                "Send ",
                inline_code(["/start ", START_PARAMETER]),
                " to get started",
            ))
            .to_string();
            let text = Text::with_markdown_v2(&error_message);
            context.send_message(text).call().await
        };

        if let Err(err) = call_result {
            dbg!(err);
        }
    });

    bot.shipping(|context| async move {
        let call_result = context.ok(DELIVERY).call().await;
        if let Err(err) = call_result {
            dbg!(err);
        }
    });

    bot.pre_checkout(|context| async move {
        let call_result = context.ok().call().await;
        if let Err(err) = call_result {
            dbg!(err);
        }
    });

    bot.payment(|context| async move {
        let call_result = context.send_message(SUCCESS).call().await;
        if let Err(err) = call_result {
            dbg!(err);
        }
    });

    bot.polling().start().await.unwrap();
}