use crate::{h, Result, TAG};
use windows::{
    core::HSTRING,
    UI::Notifications::{
        NotificationData, ToastNotification, ToastNotificationManager, ToastNotifier,
    },
};

pub struct ToastHandler<'a> {
    id: &'a str,
    toast: ToastNotification,
    manager: ToastNotifier,
}

impl<'a> ToastHandler<'a> {
    pub fn new(id: &'a str, toast: ToastNotification, manager: ToastNotifier) -> Self {
        Self { id, toast, manager }
    }
    /// Close Toast
    pub fn hide(&self) -> Result<()> {
        self.manager.Hide(&self.toast)?;

        Ok(())
    }
    /// Update progress state
    pub fn update_progress(
        &self,
        title: &str,
        status: &str,
        value: &str,
        value_string: &str,
    ) -> Result<()> {
        let manager = ToastNotificationManager::CreateToastNotifierWithId(h!(self.id))?;

        let notify_data = NotificationData::new()?;
        let values = notify_data.Values()?;

        for (key, value) in [
            ("title", title),
            ("status", status),
            ("value", value),
            ("value_string", value_string),
        ] {
            values.Insert(h!(key), h!(value))?;
        }

        manager.UpdateWithTagAndGroup(&notify_data, h!(TAG), h!(TAG))?;
        Ok(())
    }
}