use ratatui::layout::{Constraint, Layout};

use tui::ui;

use tui::ui::layout::Spacing;

use tui::ui::widget::{

    Borders, Column, ContainerState, TableState, TextEditState, TextViewState, Window,

};

use tui::ui::{BufferedValue, Show, Ui};

    fn show(&self, ctx: &ui::Context<Message>, frame: &mut Frame) -> Result<()> {

                                ui::Layout::Expandable3 { left_only: true },

    pub fn show_browser(&self, frame: &mut Frame, ui: &mut Ui<Message>) {

    fn show_browser_footer(&self, frame: &mut Frame, ui: &mut Ui<Message>) {

    pub fn show_browser_search(&self, frame: &mut Frame, ui: &mut Ui<Message>) {

    fn show_browser_context(&self, frame: &mut Frame, ui: &mut Ui<Message>) {

    pub fn show_browser_shortcuts(&self, frame: &mut Frame, ui: &mut Ui<Message>) {

    fn show_loading_popup(&self, frame: &mut Frame, ui: &mut Ui<Message>) {

    fn show_help_text(&self, frame: &mut Frame, ui: &mut Ui<Message>) {

    fn show_help_context(&self, frame: &mut Frame, ui: &mut Ui<Message>) {

use tui::ui;

use tui::ui::layout::Spacing;

use tui::ui::span;

use tui::ui::widget::{

    Borders, Column, ContainerState, TableState, TextEditState, TextViewState, TreeState, Window,

};

use tui::ui::{BufferedValue, Show, ToRow, Ui};

    fn show(&self, ctx: &ui::Context<Message>, frame: &mut Frame) -> Result<()> {

                                ui::Layout::Expandable3 {

    pub fn show_browser(&self, frame: &mut Frame, ui: &mut Ui<Message>) {

    pub fn show_browser_search(&self, frame: &mut Frame, ui: &mut Ui<Message>) {

    fn show_browser_context(&self, frame: &mut Frame, ui: &mut Ui<Message>) {

    pub fn show_browser_shortcuts(&self, frame: &mut Frame, ui: &mut Ui<Message>) {

    pub fn show_issue(&self, frame: &mut Frame, ui: &mut Ui<Message>) {

    pub fn show_issue_context(&self, frame: &mut Frame, ui: &mut Ui<Message>) {

    pub fn show_issue_shortcuts(&self, frame: &mut Frame, ui: &mut Ui<Message>) {

    pub fn show_comment(&self, frame: &mut Frame, ui: &mut Ui<Message>) {

    pub fn show_comment_context(&self, frame: &mut Frame, ui: &mut Ui<Message>) {

    pub fn show_comment_shortcuts(&self, frame: &mut Frame, ui: &mut Ui<Message>) {

    fn show_help_text(&self, frame: &mut Frame, ui: &mut Ui<Message>) {

    fn show_help_context(&self, frame: &mut Frame, ui: &mut Ui<Message>) {

use serde::Serialize;

use tui::ui;

use tui::ui::layout::Spacing;

use tui::ui::widget::{

    Borders, Column, ContainerState, TableState, TextEditState, TextViewState, Window,

};

use tui::ui::{BufferedValue, Show, Ui};

    fn show(&self, ctx: &ui::Context<Message>, frame: &mut Frame) -> Result<()> {

                                ui::Layout::Expandable3 { left_only: true },

    pub fn show_browser(&self, frame: &mut Frame, ui: &mut Ui<Message>) {

    fn show_browser_footer(&self, frame: &mut Frame, ui: &mut Ui<Message>) {

    pub fn show_browser_search(&self, frame: &mut Frame, ui: &mut Ui<Message>) {

    fn show_browser_context(&self, frame: &mut Frame, ui: &mut Ui<Message>) {

    pub fn show_browser_shortcuts(&self, frame: &mut Frame, ui: &mut Ui<Message>) {

    fn show_help_text(&self, frame: &mut Frame, ui: &mut Ui<Message>) {

    fn show_help_context(&self, frame: &mut Frame, ui: &mut Ui<Message>) {

use tui::ui::layout::Spacing;

use tui::ui::widget::{Borders, Column, ContainerState, TableState, TextViewState, Window};

use tui::ui::{Context, Show, Ui};

use ratatui::layout::{Constraint, Layout};

use ratatui::Frame;

use radicle_tui as tui;

use tui::event::Key;

use tui::ui::layout::Spacing;

use tui::ui::widget::{Borders, Column, TableState, TextEditState, Widget};

use tui::ui::{BufferedValue, ToRow};

use tui::ui::{Response, Ui};

pub struct UiExt<'a, M>(&'a mut Ui<M>);

impl<'a, M> UiExt<'a, M> {

    pub fn new(ui: &'a mut Ui<M>) -> Self {

        Self(ui)

    }

}

impl<'a, M> From<&'a mut Ui<M>> for UiExt<'a, M> {

    fn from(ui: &'a mut Ui<M>) -> Self {

        Self::new(ui)

    }

}

#[allow(dead_code)]

impl<'a, M> UiExt<'a, M>

where

    M: Clone,

{

    #[allow(clippy::too_many_arguments)]

    pub fn browser<R, const W: usize>(

        &mut self,

        frame: &mut Frame,

        selected: &'a mut Option<usize>,

        items: &'a Vec<R>,

        header: impl IntoIterator<Item = Column<'a>>,

        footer: impl IntoIterator<Item = Column<'a>>,

        show_search: &'a mut bool,

        search: &'a mut BufferedValue<TextEditState>,

    ) -> Response

    where

        R: ToRow<W> + Clone,

    {

        Browser::<R, W>::new(selected, items, header, footer, show_search, search).ui(self.0, frame)

    }

}

#[allow(dead_code)]

#[derive(Clone, Debug)]

pub struct BrowserState {

    items: TableState,

    search: BufferedValue<TextEditState>,

    show_search: bool,

}

#[allow(dead_code)]

impl BrowserState {

    pub fn new(items: TableState, search: BufferedValue<TextEditState>, show_search: bool) -> Self {

        Self {

            items,

            search,

            show_search,

        }

    }

    pub fn selected(&self) -> Option<usize> {

        self.items.selected()

    }

}

pub struct Browser<'a, R, const W: usize> {

    items: &'a Vec<R>,

    selected: &'a mut Option<usize>,

    header: Vec<Column<'a>>,

    footer: Vec<Column<'a>>,

    show_search: &'a mut bool,

    search: &'a mut BufferedValue<TextEditState>,

}

#[allow(dead_code)]

impl<'a, R, const W: usize> Browser<'a, R, W> {

    pub fn new(

        selected: &'a mut Option<usize>,

        items: &'a Vec<R>,

        header: impl IntoIterator<Item = Column<'a>>,

        footer: impl IntoIterator<Item = Column<'a>>,

        show_search: &'a mut bool,

        search: &'a mut BufferedValue<TextEditState>,

    ) -> Self {

        Self {

            items,

            selected,

            header: header.into_iter().collect(),

            footer: footer.into_iter().collect(),

            show_search,

            search,

        }

    }

    pub fn items(&self) -> &Vec<R> {

        self.items

    }

}

/// TODO(erikli): Implement `show` that returns an `InnerResponse` such that it can

/// used like a group.

impl<R, const W: usize> Widget for Browser<'_, R, W>

where

    R: ToRow<W> + Clone,

{

    fn ui<M>(self, ui: &mut Ui<M>, frame: &mut Frame) -> Response

    where

        M: Clone,

    {

        let mut response = Response::default();

        let (mut text, mut cursor) = (self.search.read().text, self.search.read().cursor);

        ui.layout(

            Layout::vertical([

                Constraint::Length(3),

                Constraint::Min(1),

                Constraint::Length(if *self.show_search { 2 } else { 3 }),

            ]),

            Some(1),

            |ui| {

                ui.column_bar(

                    frame,

                    self.header.clone().to_vec(),

                    Spacing::default(),

                    Some(Borders::Top),

                );

                let table = ui.table(

                    frame,

                    self.selected,

                    self.items,

                    self.header.to_vec(),

                    None,

                    Spacing::from(1),

                    if *self.show_search {

                        Some(Borders::BottomSides)

                    } else {

                        Some(Borders::Sides)

                    },

                );

                response.changed |= table.changed;

                if *self.show_search {

                    let text_edit = ui.text_edit_singleline(

                        frame,

                        &mut text,

                        &mut cursor,

                        Some("Search".to_string()),

                        Some(Borders::Spacer { top: 0, left: 1 }),

                    );

                    self.search.write(TextEditState { text, cursor });

                    response.changed |= text_edit.changed;

                } else {

                    ui.column_bar(

                        frame,

                        self.footer.clone().to_vec(),

                        Spacing::from(0),

                        Some(Borders::Bottom),

                    );

                }

            },

        );

        if !*self.show_search {

            if ui.has_input(|key| key == Key::Char('/')) {

                *self.show_search = true;

            }

        } else {

            if ui.has_input(|key| key == Key::Esc) {

                *self.show_search = false;

                self.search.reset();

            }

            if ui.has_input(|key| key == Key::Enter) {

                *self.show_search = false;

                self.search.apply();

            }

        }

        response

    }

}

use tui::ui::widget::Column;

use tui::ui::ToRow;

use tui::ui::widget::{Borders, Window};

use tui::ui::{Context, Show};

use tui::store::Update;

use tui::ui::layout::Spacing;

use tui::ui::widget::{Borders, Column, TableState, Window};

use tui::ui::{Context, Show, ToRow};

use tui::Exit;

use ui::{Frontend, Show};

    let frontend = Frontend::default();

pub mod widget;

use std::collections::{HashSet, VecDeque};

use std::hash::Hash;

use std::rc::Rc;

use std::time::Duration;

use anyhow::Result;

use ratatui::layout::{Alignment, Constraint, Flex, Position, Rect};

use ratatui::prelude::*;

use ratatui::text::{Span, Text};

use ratatui::{Frame, Viewport};

use tokio::sync::broadcast;

use tokio::sync::mpsc::UnboundedReceiver;

use crate::event::{Event, Key};

use crate::store::Update;

use crate::terminal::Terminal;

use crate::ui::layout::Spacing;

use crate::ui::theme::Theme;

use crate::ui::widget::{AddContentFn, Borders, Column, Widget};

use crate::{Interrupted, Share};

const RENDERING_TICK_RATE: Duration = Duration::from_millis(250);

/// The main UI trait for the ability to render an application.

pub trait Show<M> {

    fn show(&self, ctx: &Context<M>, frame: &mut Frame) -> Result<()>;

}

#[derive(Default)]

pub struct Frontend {}

impl Frontend {

    pub async fn run<S, M, R>(

        self,

        message_tx: broadcast::Sender<M>,

        mut state_rx: UnboundedReceiver<S>,

        mut event_rx: UnboundedReceiver<Event>,

        mut interrupt_rx: broadcast::Receiver<Interrupted<R>>,

        viewport: Viewport,

    ) -> anyhow::Result<Interrupted<R>>

    where

        S: Update<M, Return = R> + Show<M>,

        M: Share,

        R: Share,

    {

        let mut ticker = tokio::time::interval(RENDERING_TICK_RATE);

        let mut terminal = Terminal::try_from(viewport)?;

        let mut state = state_rx.recv().await.unwrap();

        let mut ctx = Context::default().with_sender(message_tx);

        let result: anyhow::Result<Interrupted<R>> = loop {

            tokio::select! {

                // Tick to terminate the select every N milliseconds

                _ = ticker.tick() => (),

                // Handle input events

                Some(event) = event_rx.recv() => {

                    match event {

                        Event::Key(key) => {

                            log::debug!(target: "frontend", "Received key event: {key:?}");

                            ctx.store_input(event)

                        }

                        Event::Resize(x, y) => {

                            log::debug!(target: "frontend", "Received resize event: {x},{y}");

                            terminal.clear()?;

                        },

                        Event::Unknown => {

                            log::debug!(target: "frontend", "Received unknown event")

                        }

                    }

                },

                // Handle state updates

                Some(s) = state_rx.recv() => {

                    state = s;

                },

                // Catch and handle interrupt signal to gracefully shutdown

                Ok(interrupted) = interrupt_rx.recv() => {

                    break Ok(interrupted);

                }

            }

            terminal.draw(|frame| {

                let ctx = ctx.clone().with_frame_size(frame.area());

                if let Err(err) = state.show(&ctx, frame) {

                    log::warn!("Drawing failed: {err}");

                }

            })?;

            ctx.clear_inputs();

        };

        terminal.restore()?;

        result

    }

}

#[derive(Default, Debug)]

pub struct Response {

    pub changed: bool,

}

#[derive(Debug)]

pub struct InnerResponse<R> {

    /// What the user closure returned.

    pub inner: R,

    /// The response of the area.

    pub response: Response,

}

impl<R> InnerResponse<R> {

    #[inline]

    pub fn new(inner: R, response: Response) -> Self {

        Self { inner, response }

    }

/// A `Context` is held by the `Ui` and reflects the environment a `Ui` runs in.

#[derive(Clone, Debug)]

pub struct Context<M> {

    /// Currently captured user inputs. Inputs that where stored via `store_input`

    /// need to be cleared manually via `clear_inputs` (usually for each frame drawn).

    inputs: VecDeque<Event>,

    /// Current frame of the application.

    pub(crate) frame_size: Rect,

    /// The message sender used by the `Ui` to send application messages.

    pub(crate) sender: Option<broadcast::Sender<M>>,

}

impl<M> Default for Context<M> {

    fn default() -> Self {

        Self {

            inputs: VecDeque::default(),

            frame_size: Rect::default(),

            sender: None,

        }

    }

}

impl<M> Context<M> {

    pub fn new(frame_size: Rect) -> Self {

            frame_size,

            ..Default::default()

    pub fn with_inputs(mut self, inputs: VecDeque<Event>) -> Self {

        self.inputs = inputs;

    pub fn with_frame_size(mut self, frame_size: Rect) -> Self {

        self.frame_size = frame_size;

    pub fn with_sender(mut self, sender: broadcast::Sender<M>) -> Self {

        self.sender = Some(sender);

    pub fn frame_size(&self) -> Rect {

        self.frame_size

    }

    pub fn store_input(&mut self, event: Event) {

        self.inputs.push_back(event);

    }

    pub fn clear_inputs(&mut self) {

        self.inputs.clear();

    }

}

/// A `Layout` is used to support pre-defined layouts. It either represents

/// such a predefined layout or a wrapped `ratatui` layout. It's used internally

/// but can be build from a `ratatui` layout.

#[derive(Clone, Default, Debug)]

pub enum Layout {

    #[default]

    None,

    Wrapped {

        internal: ratatui::layout::Layout,

    },

    Expandable3 {

        left_only: bool,

    },

    Popup {

        percent_x: u16,

        percent_y: u16,

    },

}

impl From<ratatui::layout::Layout> for Layout {

    fn from(layout: ratatui::layout::Layout) -> Self {

        Layout::Wrapped { internal: layout }

    }

}

impl Layout {

    pub fn len(&self) -> usize {

        match self {

            Layout::None => 0,

            Layout::Wrapped { internal } => internal.split(Rect::default()).len(),

            Layout::Expandable3 { left_only } => {

                if *left_only {

                    1

                } else {

                    3

                }

            }

            Layout::Popup {

                percent_x: _,

                percent_y: _,

            } => 1,

        }

    }

    pub fn is_empty(&self) -> bool {

        self.len() == 0

    }

    pub fn split(&self, area: Rect) -> Rc<[Rect]> {

        match self {

            Layout::None => Rc::new([]),

            Layout::Wrapped { internal } => internal.split(area),

            Layout::Expandable3 { left_only } => {

                use ratatui::layout::Layout;

                if *left_only {

                    [area].into()

                } else if area.width <= 140 {

                    let [left, right] = Layout::horizontal([

                        Constraint::Percentage(50),

                        Constraint::Percentage(50),

                    ])

                    .areas(area);

                    let [right_top, right_bottom] =

                        Layout::vertical([Constraint::Percentage(65), Constraint::Percentage(35)])

                            .areas(right);

                    [left, right_top, right_bottom].into()

                } else {

                    Layout::horizontal([

                        Constraint::Percentage(33),

                        Constraint::Percentage(33),

                        Constraint::Percentage(33),

                    ])

                    .split(area)

                }

            }

            Layout::Popup {

                percent_x,

                percent_y,

            } => {

                use ratatui::layout::Layout;

                let vertical =

                    Layout::vertical([Constraint::Percentage(*percent_y)]).flex(Flex::Center);

                let horizontal =

                    Layout::horizontal([Constraint::Percentage(*percent_x)]).flex(Flex::Center);

                let [area] = vertical.areas(area);

                let [area] = horizontal.areas(area);

                [area].into()

            }

        }

    }

/// The `Ui` is the main frontend component that provides render and user-input capture

/// capabilities. An application consists of at least 1 root `Ui`. An `Ui` can build child

/// `Ui`s that partially inherit attributes.

pub struct Ui<M> {

    /// The context this runs in: frame sizes, captured user-inputs etc.

    ctx: Context<M>,

    /// The UI theme.

    theme: Theme,

    /// The area this can render in.

    area: Rect,

    /// The layout used to calculate the next area to draw.

    layout: Layout,

    /// Currently focused area.

    focus_area: Option<usize>,

    /// If this has focus.

    has_focus: bool,

    /// Current rendering counter that is increased whenever the next area to draw

    /// on is requested.

    count: usize,

impl<M> Ui<M> {

    pub fn has_input(&mut self, f: impl Fn(Key) -> bool) -> bool {

        self.has_focus

            && self.is_area_focused()

            && self.ctx.inputs.iter().any(|event| {

                if let Event::Key(key) = event {

                    return f(*key);

                }

                false

            })

    }

    pub fn has_global_input(&mut self, f: impl Fn(Key) -> bool) -> bool {

        self.has_focus

            && self.ctx.inputs.iter().any(|event| {

                if let Event::Key(key) = event {

                    return f(*key);

                }

                false

            })

    }

    pub fn get_input(&mut self, f: impl Fn(Key) -> bool) -> Option<Key> {

        if self.has_focus && self.is_area_focused() {

            let matches = |&event| {

                if let Event::Key(key) = event {

                    return f(key);

                }

                false

            };

            if let Some(Event::Key(key)) =

                self.ctx.inputs.iter().find(|event| matches(event)).copied()

            {

                return Some(key);

            }

            None

        } else {

            None

        }

    }

}

impl<M> Default for Ui<M> {

    fn default() -> Self {

            theme: Theme::default(),

            area: Rect::default(),

            layout: Layout::default(),

            focus_area: None,

            has_focus: true,

            count: 0,

            ctx: Context::default(),

}

impl<M> Ui<M> {

    pub fn new(area: Rect) -> Self {

        Self {

            area,

            ..Default::default()

        }

    }

    pub fn with_area(mut self, area: Rect) -> Self {

        self.area = area;

        self

    }

    pub fn with_layout(mut self, layout: Layout) -> Self {

        self.layout = layout;

        self

    }

    pub fn with_area_focus(mut self, focus: Option<usize>) -> Self {

        self.focus_area = focus;

        self

    }

    pub fn with_ctx(mut self, ctx: Context<M>) -> Self {

        self.ctx = ctx;

    pub fn with_focus(mut self) -> Self {

        self.has_focus = true;

    pub fn without_focus(mut self) -> Self {

        self.has_focus = false;

    pub fn theme(&self) -> &Theme {

        &self.theme

    }

    pub fn area(&self) -> Rect {

        self.area

    }

    pub fn next_area(&mut self) -> Option<(Rect, bool)> {

        let area_focus = self

            .focus_area

            .map(|focus| self.count == focus)

            .unwrap_or(false);

        let rect = self.layout.split(self.area).get(self.count).cloned();

        self.count += 1;

        rect.map(|rect| (rect, area_focus))