use iced::widget::scrollable::{Scrollbar, Scroller};
use iced::widget::{
    button, column, container, horizontal_space, progress_bar, radio, row,
    scrollable, slider, text, vertical_space,
};
use iced::{executor, theme, Alignment, Color, Vector};
use iced::{Application, Command, Element, Length, Settings, Theme};
use lazy_static::lazy_static;

lazy_static! {
    static ref SCROLLABLE_ID: scrollable::Id = scrollable::Id::unique();
}

pub fn main() -> iced::Result {
    ScrollableDemo::run(Settings::default())
}

struct ScrollableDemo {
    scrollable_direction: Direction,
    scrollbar_width: u16,
    scrollbar_margin: u16,
    scroller_width: u16,
    current_scroll_offset: Vector<f32>,
}

#[derive(Debug, Clone, Eq, PartialEq, Copy)]
enum Direction {
    Vertical,
    Horizontal,
    Multi,
}

#[derive(Debug, Clone)]
enum Message {
    SwitchDirection(Direction),
    ScrollbarWidthChanged(u16),
    ScrollbarMarginChanged(u16),
    ScrollerWidthChanged(u16),
    ScrollToBeginning(scrollable::Direction),
    ScrollToEnd(scrollable::Direction),
    Scrolled(Vector<f32>),
}

impl Application for ScrollableDemo {
    type Executor = executor::Default;
    type Message = Message;
    type Theme = Theme;
    type Flags = ();

    fn new(_flags: Self::Flags) -> (Self, Command<Message>) {
        (
            ScrollableDemo {
                scrollable_direction: Direction::Vertical,
                scrollbar_width: 10,
                scrollbar_margin: 0,
                scroller_width: 10,
                current_scroll_offset: Vector::new(0.0, 0.0),
            },
            Command::none(),
        )
    }

    fn title(&self) -> String {
        String::from("Scrollable - Iced")
    }

    fn update(&mut self, message: Message) -> Command<Message> {
        match message {
            Message::SwitchDirection(direction) => {
                self.current_scroll_offset = Vector::new(0.0, 0.0);
                self.scrollable_direction = direction;

                Command::none()
            }
            Message::ScrollbarWidthChanged(width) => {
                self.scrollbar_width = width;

                Command::none()
            }
            Message::ScrollbarMarginChanged(margin) => {
                self.scrollbar_margin = margin;

                Command::none()
            }
            Message::ScrollerWidthChanged(width) => {
                self.scroller_width = width;

                Command::none()
            }
            Message::ScrollToBeginning(direction) => {
                match direction {
                    scrollable::Direction::Horizontal => {
                        self.current_scroll_offset.x = 0.0;
                    }
                    scrollable::Direction::Vertical => {
                        self.current_scroll_offset.y = 0.0;
                    }
                }

                scrollable::snap_to(
                    SCROLLABLE_ID.clone(),
                    Vector::new(
                        self.current_scroll_offset.x,
                        self.current_scroll_offset.y,
                    ),
                )
            }
            Message::ScrollToEnd(direction) => {
                match direction {
                    scrollable::Direction::Horizontal => {
                        self.current_scroll_offset.x = 1.0;
                    }
                    scrollable::Direction::Vertical => {
                        self.current_scroll_offset.y = 1.0;
                    }
                }

                scrollable::snap_to(
                    SCROLLABLE_ID.clone(),
                    Vector::new(
                        self.current_scroll_offset.x,
                        self.current_scroll_offset.y,
                    ),
                )
            }
            Message::Scrolled(offset) => {
                self.current_scroll_offset = offset;

                Command::none()
            }
        }
    }

    fn view(&self) -> Element<Message> {
        let scrollbar_width_slider = slider(
            0..=15,
            self.scrollbar_width,
            Message::ScrollbarWidthChanged,
        );
        let scrollbar_margin_slider = slider(
            0..=15,
            self.scrollbar_margin,
            Message::ScrollbarMarginChanged,
        );
        let scroller_width_slider =
            slider(0..=15, self.scroller_width, Message::ScrollerWidthChanged);

        let scroll_slider_controls = column![
            text("Scrollbar width:"),
            scrollbar_width_slider,
            text("Scrollbar margin:"),
            scrollbar_margin_slider,
            text("Scroller width:"),
            scroller_width_slider,
        ]
        .width(Length::Fill);

        let scroll_orientation_controls = column(vec![
            text("Scrollbar direction:").into(),
            radio(
                "Vertical",
                Direction::Vertical,
                Some(self.scrollable_direction),
                Message::SwitchDirection,
            )
            .into(),
            radio(
                "Horizontal",
                Direction::Horizontal,
                Some(self.scrollable_direction),
                Message::SwitchDirection,
            )
            .into(),
            radio(
                "Both!",
                Direction::Multi,
                Some(self.scrollable_direction),
                Message::SwitchDirection,
            )
            .into(),
        ])
        .width(Length::Fill);

        let scroll_controls =
            row![scroll_slider_controls, scroll_orientation_controls]
                .spacing(20)
                .width(Length::Fill);

        let scroll_to_end_button = |direction: scrollable::Direction| {
            button("Scroll to end")
                .padding(10)
                .width(Length::Units(120))
                .on_press(Message::ScrollToEnd(direction))
        };

        let scroll_to_beginning_button = |direction: scrollable::Direction| {
            button("Scroll to beginning")
                .padding(10)
                .width(Length::Units(120))
                .on_press(Message::ScrollToBeginning(direction))
        };

        let scrollable_content: Element<Message> =
            Element::from(match self.scrollable_direction {
                Direction::Vertical => scrollable(
                    column![
                        scroll_to_end_button(scrollable::Direction::Vertical),
                        text("Beginning!"),
                        vertical_space(Length::Units(1200)),
                        text("Middle!"),
                        vertical_space(Length::Units(1200)),
                        text("End!"),
                        scroll_to_beginning_button(
                            scrollable::Direction::Vertical
                        ),
                    ]
                    .width(Length::Fill)
                    .align_items(Alignment::Center)
                    .padding([40, 0, 40, 0])
                    .spacing(40),
                )
                .height(Length::Fill)
                .scrollbar_width(self.scrollbar_width)
                .scrollbar_margin(self.scrollbar_margin)
                .scroller_width(self.scroller_width)
                .id(SCROLLABLE_ID.clone())
                .on_scroll(Message::Scrolled),
                Direction::Horizontal => scrollable(
                    row![
                        scroll_to_end_button(scrollable::Direction::Horizontal),
                        text("Beginning!"),
                        horizontal_space(Length::Units(1200)),
                        text("Middle!"),
                        horizontal_space(Length::Units(1200)),
                        text("End!"),
                        scroll_to_beginning_button(
                            scrollable::Direction::Horizontal
                        ),
                    ]
                    .height(Length::Units(450))
                    .align_items(Alignment::Center)
                    .padding([0, 40, 0, 40])
                    .spacing(40),
                )
                .height(Length::Fill)
                .horizontal_scroll(
                    scrollable::Horizontal::new()
                        .scrollbar_height(self.scrollbar_width)
                        .scrollbar_margin(self.scrollbar_margin)
                        .scroller_height(self.scroller_width),
                )
                .style(theme::Scrollable::Custom(Box::new(
                    ScrollbarCustomStyle,
                )))
                .id(SCROLLABLE_ID.clone())
                .on_scroll(Message::Scrolled),
                Direction::Multi => scrollable(
                    //horizontal content
                    row![
                        column![
                            text("Let's do some scrolling!"),
                            vertical_space(Length::Units(2400))
                        ],
                        scroll_to_end_button(scrollable::Direction::Horizontal),
                        text("Horizontal - Beginning!"),
                        horizontal_space(Length::Units(1200)),
                        //vertical content
                        column![
                            text("Horizontal - Middle!"),
                            scroll_to_end_button(
                                scrollable::Direction::Vertical
                            ),
                            text("Vertical - Beginning!"),
                            vertical_space(Length::Units(1200)),
                            text("Vertical - Middle!"),
                            vertical_space(Length::Units(1200)),
                            text("Vertical - End!"),
                            scroll_to_beginning_button(
                                scrollable::Direction::Vertical
                            )
                        ]
                        .align_items(Alignment::Fill)
                        .spacing(40),
                        horizontal_space(Length::Units(1200)),
                        text("Horizontal - End!"),
                        scroll_to_beginning_button(
                            scrollable::Direction::Horizontal
                        ),
                    ]
                    .align_items(Alignment::Center)
                    .padding([0, 40, 0, 40])
                    .spacing(40),
                )
                .height(Length::Fill)
                .scrollbar_width(self.scrollbar_width)
                .scrollbar_margin(self.scrollbar_margin)
                .scroller_width(self.scroller_width)
                .horizontal_scroll(
                    scrollable::Horizontal::new()
                        .scrollbar_height(self.scrollbar_width)
                        .scrollbar_margin(self.scrollbar_margin)
                        .scroller_height(self.scroller_width),
                )
                .style(theme::Scrollable::Custom(Box::new(
                    ScrollbarCustomStyle,
                )))
                .id(SCROLLABLE_ID.clone())
                .on_scroll(Message::Scrolled),
            });

        let progress_bars: Element<Message> = match self.scrollable_direction {
            Direction::Vertical => {
                progress_bar(0.0..=1.0, self.current_scroll_offset.y).into()
            }
            Direction::Horizontal => {
                progress_bar(0.0..=1.0, self.current_scroll_offset.x)
                    .style(theme::ProgressBar::Custom(Box::new(
                        ProgressBarCustomStyle,
                    )))
                    .into()
            }
            Direction::Multi => column![
                progress_bar(0.0..=1.0, self.current_scroll_offset.y),
                progress_bar(0.0..=1.0, self.current_scroll_offset.x).style(
                    theme::ProgressBar::Custom(Box::new(
                        ProgressBarCustomStyle,
                    ))
                )
            ]
            .spacing(10)
            .into(),
        };

        let content: Element<Message> =
            column![scroll_controls, scrollable_content, progress_bars]
                .width(Length::Fill)
                .height(Length::Fill)
                .align_items(Alignment::Center)
                .spacing(10)
                .into();

        Element::from(
            container(content)
                .width(Length::Fill)
                .height(Length::Fill)
                .padding(40)
                .center_x()
                .center_y(),
        )
    }

    fn theme(&self) -> Self::Theme {
        Theme::Dark
    }
}

struct ScrollbarCustomStyle;

impl scrollable::StyleSheet for ScrollbarCustomStyle {
    type Style = Theme;

    fn active(&self, style: &Self::Style) -> Scrollbar {
        style.active(&theme::Scrollable::Default)
    }

    fn hovered(&self, style: &Self::Style) -> Scrollbar {
        style.hovered(&theme::Scrollable::Default)
    }

    fn hovered_horizontal(&self, style: &Self::Style) -> Scrollbar {
        Scrollbar {
            background: style.active(&theme::Scrollable::default()).background,
            border_radius: 0.0,
            border_width: 0.0,
            border_color: Default::default(),
            scroller: Scroller {
                color: Color::from_rgb8(250, 85, 134),
                border_radius: 0.0,
                border_width: 0.0,
                border_color: Default::default(),
            },
        }
    }
}

struct ProgressBarCustomStyle;

impl progress_bar::StyleSheet for ProgressBarCustomStyle {
    type Style = Theme;

    fn appearance(&self, style: &Self::Style) -> progress_bar::Appearance {
        progress_bar::Appearance {
            background: style.extended_palette().background.strong.color.into(),
            bar: Color::from_rgb8(250, 85, 134).into(),
            border_radius: 0.0,
        }
    }
}