use ratatui::{
    backend::Backend,
    layout::Alignment,
    prelude::{Constraint, Direction, Layout},
    style::{Color, Style},
    text::Line,
    widgets::{Block, BorderType, Borders, Row, Table, Tabs},
    Frame,
};

use crate::app::{App, Tab};

use self::utils::Wrapper;

mod utils;

fn render_table<'a>(app: &mut App, tab: Tab) -> (Table<'a>, Vec<Constraint>) {
    let fields = tab.fields();
    let torrents = app.torrents.set_fields(None).torrents();

    let rows: Vec<Row<'_>> = torrents
        .iter()
        .map(|torrent| {
            Row::new(
                fields
                    .iter()
                    .map(|&field| field.value(torrent.clone()))
                    .collect::<Vec<_>>(),
            )
        })
        .collect();

    let widths = fields
        .iter()
        .map(|&field| Constraint::Length(field.width()))
        .collect();

    let header = Row::new(
        fields
            .iter()
            .map(|&field| field.title())
            .collect::<Vec<_>>(),
    )
    .style(Style::default().fg(Color::Yellow));
    (
        Table::new(rows)
            .block(
                Block::default()
                    .borders(Borders::ALL)
                    .border_type(BorderType::Rounded),
            )
            .header(header)
            .highlight_style(Style::default().fg(Color::Red))
            .highlight_symbol(">> ")
            .column_spacing(1),
        widths,
    )
}

/// Renders the user interface widgets.
pub fn render<'a, B: Backend>(app: &mut App, frame: &mut Frame<'_, B>) {
    // This is where you add new widgets.
    // See the following resources:
    // - https://docs.rs/ratatui/latest/ratatui/widgets/index.html
    // - https://github.com/ratatui-org/ratatui/tree/master/examples

    let size = frame.size();
    let chunks = Layout::default()
        .direction(Direction::Vertical)
        .constraints([Constraint::Length(3), Constraint::Min(0)].as_ref())
        .split(size);

    // let titles = app.tabs.iter().map(Line::from).collect();
    let titles = app
        .tabs()
        .iter()
        .map(|x| Line::from(x.to_string()))
        .collect();
    let tabs = Tabs::new(titles)
        .block(
            Block::default()
                .title_alignment(Alignment::Center)
                .borders(Borders::ALL)
                .border_type(BorderType::Rounded),
        )
        .select(app.index())
        .style(Style::default().fg(Color::Blue))
        .highlight_style(Style::default().fg(Color::Green))
        .divider("|");

    frame.render_widget(tabs, chunks[0]);
    let (inner, widths) = match app.index() {
        0 => render_table(app, Tab::All),
        1 => render_table(app, Tab::Active),
        2 => render_table(app, Tab::Downloading),
        3 => render_table(app, Tab::Settings),
        _ => unreachable!(),
    };
    frame.render_stateful_widget(inner.widths(widths.as_ref()), chunks[1], &mut app.state)
}