use std::marker::PhantomData;

use gpui2::style::StyleHelpers;
use gpui2::{elements::div, IntoElement};
use gpui2::{Element, ParentElement, ViewContext};

use crate::theme::{theme, Theme};
use crate::{icon_button, text_button, tool_divider, IconAsset};

#[derive(Default, PartialEq)]
pub enum Tool {
    #[default]
    ProjectPanel,
    CollaborationPanel,
    Terminal,
    Assistant,
    Feedback,
    Diagnostics,
}

struct ToolGroup {
    active_index: Option<usize>,
    tools: Vec<Tool>,
}

impl Default for ToolGroup {
    fn default() -> Self {
        ToolGroup {
            active_index: None,
            tools: vec![],
        }
    }
}

#[derive(Element)]
pub struct StatusBar<V: 'static> {
    view_type: PhantomData<V>,
    left_tools: Option<ToolGroup>,
    right_tools: Option<ToolGroup>,
    bottom_tools: Option<ToolGroup>,
}

pub fn status_bar<V: 'static>() -> StatusBar<V> {
    StatusBar {
        view_type: PhantomData,
        left_tools: None,
        right_tools: None,
        bottom_tools: None,
    }
}

impl<V: 'static> StatusBar<V> {
    pub fn left_tool(mut self, tool: Tool, active_index: Option<usize>) -> Self {
        self.left_tools = {
            let mut tools = vec![tool];
            tools.extend(self.left_tools.take().unwrap_or_default().tools);
            Some(ToolGroup {
                active_index,
                tools,
            })
        };
        self
    }

    pub fn right_tool(mut self, tool: Tool, active_index: Option<usize>) -> Self {
        self.right_tools = {
            let mut tools = vec![tool];
            tools.extend(self.left_tools.take().unwrap_or_default().tools);
            Some(ToolGroup {
                active_index,
                tools,
            })
        };
        self
    }

    pub fn bottom_tool(mut self, tool: Tool, active_index: Option<usize>) -> Self {
        self.bottom_tools = {
            let mut tools = vec![tool];
            tools.extend(self.left_tools.take().unwrap_or_default().tools);
            Some(ToolGroup {
                active_index,
                tools,
            })
        };
        self
    }

    fn render(&mut self, _: &mut V, cx: &mut ViewContext<V>) -> impl IntoElement<V> {
        let theme = theme(cx);

        div()
            .py_0p5()
            .px_1()
            .flex()
            .items_center()
            .justify_between()
            .w_full()
            .fill(theme.lowest.base.default.background)
            .child(self.left_tools(&theme))
            .child(self.right_tools(&theme))
    }

    fn left_tools(&self, theme: &Theme) -> impl Element<V> {
        div()
            .flex()
            .items_center()
            .gap_1()
            .child(icon_button().icon(IconAsset::FileTree))
            .child(icon_button().icon(IconAsset::Hash))
            .child(tool_divider())
            .child(icon_button().icon(IconAsset::XCircle))
    }
    fn right_tools(&self, theme: &Theme) -> impl Element<V> {
        div()
            .flex()
            .items_center()
            .gap_2()
            .child(
                div()
                    .flex()
                    .items_center()
                    .gap_1()
                    .child(text_button("116:25"))
                    .child(text_button("Rust")),
            )
            .child(tool_divider())
            .child(
                div()
                    .flex()
                    .items_center()
                    .gap_1()
                    .child(icon_button().icon(IconAsset::Copilot))
                    .child(icon_button().icon(IconAsset::Envelope)),
            )
            .child(tool_divider())
            .child(
                div()
                    .flex()
                    .items_center()
                    .gap_1()
                    .child(icon_button().icon(IconAsset::Terminal))
                    .child(icon_button().icon(IconAsset::MessageBubbles))
                    .child(icon_button().icon(IconAsset::Ai)),
            )
    }
}