use crate::config::Config;
use crate::modules::WaybarModule;
use crate::output::WaybarOutput;
use crate::state::SharedState;
use crate::utils::{format_template, TokenValue};
use anyhow::Result;
use std::process::Command;

pub struct BudsModule;

impl WaybarModule for BudsModule {
    fn run(&self, config: &Config, _state: &SharedState, args: &[&str]) -> Result<WaybarOutput> {
        let action = args.first().unwrap_or(&"show");
        let mac = &config.buds.mac;

        match *action {
            "cycle_anc" => {
                let output = Command::new("pbpctrl").args(["get", "anc"]).output()?;
                let current_mode = String::from_utf8_lossy(&output.stdout).trim().to_string();
                
                let next_mode = match current_mode.as_str() {
                    "active" => "aware",
                    "aware" => "off",
                    _ => "active",
                };
                
                Command::new("pbpctrl").args(["set", "anc", next_mode]).status()?;
                return Ok(WaybarOutput {
                    text: String::new(),
                    tooltip: None, class: None, percentage: None,
                });
            }
            "connect" => {
                Command::new("bluetoothctl").args(["connect", mac]).status()?;
                return Ok(WaybarOutput {
                    text: String::new(),
                    tooltip: None, class: None, percentage: None,
                });
            }
            "disconnect" => {
                Command::new("bluetoothctl").args(["disconnect", mac]).status()?;
                return Ok(WaybarOutput {
                    text: String::new(),
                    tooltip: None, class: None, percentage: None,
                });
            }
            "show" | _ => {}
        }

        let bt_info = Command::new("bluetoothctl").args(["info", mac]).output()?;
        let bt_str = String::from_utf8_lossy(&bt_info.stdout);
        
        if !bt_str.contains("Connected: yes") {
            return Ok(WaybarOutput {
                text: config.buds.format_disconnected.clone(),
                tooltip: Some("Pixel Buds Pro 2 not connected".to_string()),
                class: Some("disconnected".to_string()),
                percentage: None,
            });
        }

        let bat_cmd = Command::new("pbpctrl").args(["show", "battery"]).output();
        if bat_cmd.is_err() || !bat_cmd.as_ref().unwrap().status.success() {
            return Ok(WaybarOutput {
                text: config.buds.format_disconnected.clone(),
                tooltip: Some("Pixel Buds Pro 2 connected (No Data)".to_string()),
                class: Some("disconnected".to_string()),
                percentage: None,
            });
        }
        
        let bat_result = bat_cmd.unwrap();
        let bat_output = String::from_utf8_lossy(&bat_result.stdout);
        let mut left_bud = "unknown";
        let mut right_bud = "unknown";

        for line in bat_output.lines() {
            if line.contains("left bud:") {
                left_bud = line.split_whitespace().nth(2).unwrap_or("unknown");
            } else if line.contains("right bud:") {
                right_bud = line.split_whitespace().nth(2).unwrap_or("unknown");
            }
        }

        if left_bud == "unknown" && right_bud == "unknown" {
            return Ok(WaybarOutput {
                text: "{}".to_string(),
                tooltip: None, class: None, percentage: None,
            });
        }

        let left_display = if left_bud == "unknown" { "---".to_string() } else { format!("{}%", left_bud) };
        let right_display = if right_bud == "unknown" { "---".to_string() } else { format!("{}%", right_bud) };

        let anc_cmd = Command::new("pbpctrl").args(["get", "anc"]).output()?;
        let current_mode = String::from_utf8_lossy(&anc_cmd.stdout).trim().to_string();

        let (anc_icon, class) = match current_mode.as_str() {
            "active" => ("ANC", "anc-active"),
            "aware" => ("Aware", "anc-aware"),
            "off" => ("Off", "anc-off"),
            _ => ("?", "anc-unknown"),
        };

        let text = format_template(
            &config.buds.format,
            &[
                ("left", TokenValue::String(&left_display)),
                ("right", TokenValue::String(&right_display)),
                ("anc", TokenValue::String(anc_icon)),
            ]
        );

        Ok(WaybarOutput {
            text,
            tooltip: Some("Pixel Buds Pro 2".to_string()),
            class: Some(class.to_string()),
            percentage: None,
        })
    }
}