use crate::backlight::{
    screen_backlight_subscription, ScreenBacklightRequest, ScreenBacklightUpdate,
};
use crate::config;
use crate::fl;
use crate::power_daemon::{
    power_profile_subscription, Power, PowerProfileRequest, PowerProfileUpdate,
};
use crate::upower_device::{device_subscription, DeviceDbusEvent};
use crate::upower_kbdbacklight::{
    kbd_backlight_subscription, KeyboardBacklightRequest, KeyboardBacklightUpdate,
};
use cosmic::applet::{CosmicAppletHelper, APPLET_BUTTON_THEME};
use cosmic::iced::alignment::Horizontal;
use cosmic::iced::wayland::popup::{destroy_popup, get_popup};
use cosmic::iced::wayland::SurfaceIdWrapper;
use cosmic::iced::{
    widget::{column, container, row, slider, text},
    window, Alignment, Application, Command, Length, Subscription,
};
use cosmic::iced_native::layout::Limits;
use cosmic::iced_style::application::{self, Appearance};
use cosmic::iced_style::{svg, Color};
use cosmic::theme::Svg;
use cosmic::widget::{button, horizontal_rule, icon, toggler};
use cosmic::{Element, Theme};
use log::error;
use std::time::Duration;
use tokio::sync::mpsc::UnboundedSender;

// XXX improve
// TODO: time to empty varies? needs averaging?
fn format_duration(duration: Duration) -> String {
    let secs = duration.as_secs();
    if secs > 60 {
        let min = secs / 60;
        if min > 60 {
            format!("{}:{:02}", min / 60, min % 60)
        } else {
            format!("{}{}", min, fl!("minutes"))
        }
    } else {
        format!("{}{}", secs, fl!("seconds"))
    }
}

pub fn run() -> cosmic::iced::Result {
    let helper = CosmicAppletHelper::default();
    CosmicBatteryApplet::run(helper.window_settings())
}

#[derive(Clone, Default)]
struct CosmicBatteryApplet {
    icon_name: String,
    theme: Theme,
    charging_limit: bool,
    battery_percent: f64,
    time_remaining: Duration,
    kbd_brightness: f64,
    screen_brightness: f64,
    popup: Option<window::Id>,
    id_ctr: u32,
    screen_sender: Option<UnboundedSender<ScreenBacklightRequest>>,
    kbd_sender: Option<UnboundedSender<KeyboardBacklightRequest>>,
    applet_helper: CosmicAppletHelper,
    power_profile: Power,
    power_profile_sender: Option<UnboundedSender<PowerProfileRequest>>,
}

#[derive(Debug, Clone)]
enum Message {
    TogglePopup,
    Update {
        icon_name: String,
        percent: f64,
        time_to_empty: i64,
    },
    SetKbdBrightness(i32),
    SetScreenBrightness(i32),
    SetChargingLimit(bool),
    UpdateKbdBrightness(f64),
    UpdateScreenBrightness(f64),
    OpenBatterySettings,
    InitKbdBacklight(UnboundedSender<KeyboardBacklightRequest>, f64),
    InitScreenBacklight(UnboundedSender<ScreenBacklightRequest>, f64),
    Errored(String),
    Ignore,
    InitProfile(UnboundedSender<PowerProfileRequest>, Power),
    Profile(Power),
    SelectProfile(Power),
}

impl Application for CosmicBatteryApplet {
    type Message = Message;
    type Theme = Theme;
    type Executor = cosmic::SingleThreadExecutor;
    type Flags = ();

    fn new(_flags: ()) -> (Self, Command<Message>) {
        (
            CosmicBatteryApplet {
                icon_name: "battery-symbolic".to_string(),
                ..Default::default()
            },
            Command::none(),
        )
    }

    fn title(&self) -> String {
        config::APP_ID.to_string()
    }

    fn update(&mut self, message: Message) -> Command<Message> {
        match message {
            Message::SetKbdBrightness(brightness) => {
                self.kbd_brightness = (brightness as f64 / 100.0).clamp(0., 1.);
                if let Some(tx) = &self.kbd_sender {
                    let _ = tx.send(KeyboardBacklightRequest::Set(self.kbd_brightness));
                }
            }
            Message::SetScreenBrightness(brightness) => {
                self.screen_brightness = (brightness as f64 / 100.0).clamp(0.01, 1.0);
                if let Some(tx) = &self.screen_sender {
                    let _ = tx.send(ScreenBacklightRequest::Set(self.screen_brightness));
                }
            }
            Message::SetChargingLimit(enable_charging_limit) => {
                self.charging_limit = enable_charging_limit;
            }
            Message::OpenBatterySettings => {
                // TODO Ashley
            }
            Message::Errored(e) => {
                error!("{}", e);
            }
            Message::TogglePopup => {
                if let Some(p) = self.popup.take() {
                    return destroy_popup(p);
                } else {
                    if let Some(tx) = &self.kbd_sender {
                        let _ = tx.send(KeyboardBacklightRequest::Get);
                    }
                    if let Some(tx) = &self.screen_sender {
                        let _ = tx.send(ScreenBacklightRequest::Get);
                    }

                    self.id_ctr += 1;
                    let new_id = window::Id::new(self.id_ctr);
                    self.popup.replace(new_id);

                    let mut popup_settings = self.applet_helper.get_popup_settings(
                        window::Id::new(0),
                        new_id,
                        None,
                        None,
                        None,
                    );
                    popup_settings.positioner.size_limits = Limits::NONE
                        .max_width(372)
                        .min_width(300)
                        .min_height(200)
                        .max_height(1080);
                    if let Some(tx) = self.power_profile_sender.as_ref() {
                        let _ = tx.send(PowerProfileRequest::Get);
                    }
                    return get_popup(popup_settings);
                }
            }
            Message::Update {
                icon_name,
                percent,
                time_to_empty,
            } => {
                self.icon_name = icon_name;
                self.battery_percent = percent;
                self.time_remaining = Duration::from_secs(time_to_empty as u64);
            }
            Message::UpdateKbdBrightness(b) => {
                self.kbd_brightness = b;
            }
            Message::Ignore => {}
            Message::InitKbdBacklight(tx, brightness) => {
                let _ = tx.send(KeyboardBacklightRequest::Get);
                self.kbd_sender = Some(tx);
                self.kbd_brightness = brightness;
            }
            Message::InitScreenBacklight(tx, brightness) => {
                let _ = tx.send(ScreenBacklightRequest::Get);
                self.screen_sender = Some(tx);
                self.screen_brightness = brightness;
            }
            Message::UpdateScreenBrightness(b) => {
                self.screen_brightness = b;
            }
            Message::InitProfile(tx, profile) => {
                self.power_profile_sender.replace(tx);
                self.power_profile = profile;
            }
            Message::Profile(profile) => {
                self.power_profile = profile;
                if let Some(tx) = &self.kbd_sender {
                    let _ = tx.send(KeyboardBacklightRequest::Get);
                }
                if let Some(tx) = &self.screen_sender {
                    let _ = tx.send(ScreenBacklightRequest::Get);
                }
            }
            Message::SelectProfile(profile) => {
                if let Some(tx) = self.power_profile_sender.as_ref() {
                    let _ = tx.send(PowerProfileRequest::Set(profile));
                }
            }
        }
        Command::none()
    }
    fn view(&self, id: SurfaceIdWrapper) -> Element<Message> {
        match id {
            SurfaceIdWrapper::LayerSurface(_) => unimplemented!(),
            SurfaceIdWrapper::Window(_) => self
                .applet_helper
                .icon_button(&self.icon_name)
                .on_press(Message::TogglePopup)
                .into(),
            SurfaceIdWrapper::Popup(_) => {
                let name = text(fl!("battery")).size(18);
                let description = text(
                    if "battery-full-charging-symbolic" == self.icon_name
                        || "battery-full-charged-symbolic" == self.icon_name
                    {
                        format!("{}%", self.battery_percent)
                    } else {
                        format!(
                            "{} {} ({:.0}%)",
                            format_duration(self.time_remaining),
                            fl!("until-empty"),
                            self.battery_percent
                        )
                    },
                )
                .size(12);
                self.applet_helper
                    .popup_container(
                        column![
                            row![
                                icon(&*self.icon_name, 24)
                                    .style(Svg::Custom(|theme| {
                                        svg::Appearance {
                                            color: Some(theme.palette().text),
                                        }
                                    }))
                                    .width(Length::Units(24))
                                    .height(Length::Units(24)),
                                column![name, description]
                            ]
                            .padding([0, 24])
                            .spacing(8)
                            .align_items(Alignment::Center),
                            container(horizontal_rule(1))
                                .width(Length::Fill)
                                .padding([0, 12]),
                            button(APPLET_BUTTON_THEME)
                                .custom(vec![row![
                                    column![
                                        text(fl!("battery")).size(14),
                                        text(fl!("battery-desc")).size(12)
                                    ]
                                    .width(Length::Fill),
                                    icon("emblem-ok-symbolic", 12).size(12).style(
                                        match self.power_profile {
                                            Power::Battery => Svg::SymbolicActive,
                                            _ => Svg::Default,
                                        }
                                    ),
                                ]
                                .align_items(Alignment::Center)
                                .into()])
                                .padding([8, 24])
                                .on_press(Message::SelectProfile(Power::Battery))
                                .width(Length::Fill),
                            button(APPLET_BUTTON_THEME)
                                .custom(vec![row![
                                    column![
                                        text(fl!("balanced")).size(14),
                                        text(fl!("balanced-desc")).size(12)
                                    ]
                                    .width(Length::Fill),
                                    icon("emblem-ok-symbolic", 12).size(12).style(
                                        match self.power_profile {
                                            Power::Balanced => Svg::SymbolicActive,
                                            _ => Svg::Default,
                                        }
                                    ),
                                ]
                                .align_items(Alignment::Center)
                                .into()])
                                .padding([8, 24])
                                .on_press(Message::SelectProfile(Power::Balanced))
                                .width(Length::Fill),
                            button(APPLET_BUTTON_THEME)
                                .custom(vec![row![
                                    column![
                                        text(fl!("performance")).size(14),
                                        text(fl!("performance-desc")).size(12)
                                    ]
                                    .width(Length::Fill),
                                    icon("emblem-ok-symbolic", 12).size(12).style(
                                        match self.power_profile {
                                            Power::Performance => Svg::SymbolicActive,
                                            _ => Svg::Default,
                                        }
                                    ),
                                ]
                                .align_items(Alignment::Center)
                                .into()])
                                .padding([8, 24])
                                .on_press(Message::SelectProfile(Power::Performance))
                                .width(Length::Fill),
                            container(horizontal_rule(1))
                                .width(Length::Fill)
                                .padding([0, 12]),
                            container(toggler(fl!("max-charge"), self.charging_limit, |_| {
                                Message::SetChargingLimit(!self.charging_limit)
                            }))
                            .padding([0, 24])
                            .width(Length::Fill),
                            container(horizontal_rule(1))
                                .width(Length::Fill)
                                .padding([0, 12]),
                            row![
                                icon("display-brightness-symbolic", 24)
                                    .style(Svg::Custom(|theme| {
                                        svg::Appearance {
                                            color: Some(theme.palette().text),
                                        }
                                    }))
                                    .width(Length::Units(24))
                                    .height(Length::Units(24)),
                                slider(
                                    1..=100,
                                    (self.screen_brightness * 100.0) as i32,
                                    Message::SetScreenBrightness
                                ),
                                text(format!("{:.0}%", self.screen_brightness * 100.0))
                                    .width(Length::Units(40))
                                    .horizontal_alignment(Horizontal::Right)
                            ]
                            .padding([0, 24])
                            .spacing(12),
                            row![
                                icon("keyboard-brightness-symbolic", 24)
                                    .style(Svg::Custom(|theme| {
                                        svg::Appearance {
                                            color: Some(theme.palette().text),
                                        }
                                    }))
                                    .width(Length::Units(24))
                                    .height(Length::Units(24)),
                                slider(
                                    0..=100,
                                    (self.kbd_brightness * 100.0) as i32,
                                    Message::SetKbdBrightness
                                ),
                                text(format!("{:.0}%", self.kbd_brightness * 100.0))
                                    .width(Length::Units(40))
                                    .horizontal_alignment(Horizontal::Right)
                            ]
                            .padding([0, 24])
                            .spacing(12),
                            container(horizontal_rule(1))
                                .width(Length::Fill)
                                .padding([0, 12]),
                            button(APPLET_BUTTON_THEME)
                                .custom(vec![text(fl!("power-settings"))
                                    .width(Length::Fill)
                                    .into()])
                                .on_press(Message::OpenBatterySettings)
                                .width(Length::Fill)
                                .padding([8, 24])
                        ]
                        .spacing(8)
                        .padding([8, 0]),
                    )
                    .into()
            }
        }
    }

    fn subscription(&self) -> Subscription<Message> {
        Subscription::batch(vec![
            device_subscription(0).map(|(_, event)| match event {
                DeviceDbusEvent::Update {
                    icon_name,
                    percent,
                    time_to_empty,
                } => Message::Update {
                    icon_name,
                    percent,
                    time_to_empty,
                },
            }),
            kbd_backlight_subscription(0).map(|(_, event)| match event {
                KeyboardBacklightUpdate::Update(b) => Message::UpdateKbdBrightness(b),
                KeyboardBacklightUpdate::Init(tx, b) => Message::InitKbdBacklight(tx, b),
            }),
            screen_backlight_subscription(0).map(|(_, event)| match event {
                ScreenBacklightUpdate::Update(b) => Message::UpdateScreenBrightness(b),
                ScreenBacklightUpdate::Init(tx, b) => Message::InitScreenBacklight(tx, b),
            }),
            power_profile_subscription(0).map(|(_, event)| match event {
                PowerProfileUpdate::Update { profile } => Message::Profile(profile),
                PowerProfileUpdate::Init(tx, p) => Message::InitProfile(p, tx),
                PowerProfileUpdate::Error(e) => Message::Errored(e), // TODO: handle error
            }),
        ])
    }

    fn theme(&self) -> Theme {
        self.theme
    }

    fn close_requested(&self, _id: SurfaceIdWrapper) -> Self::Message {
        Message::Ignore
    }

    fn style(&self) -> <Self::Theme as application::StyleSheet>::Style {
        <Self::Theme as application::StyleSheet>::Style::Custom(|theme| Appearance {
            background_color: Color::from_rgba(0.0, 0.0, 0.0, 0.0),
            text_color: theme.cosmic().on_bg_color().into(),
        })
    }
}