use egui_extras::{Size, StripBuilder, TableBuilder};

/// Shows off a table with dynamic layout
#[cfg_attr(feature = "serde", derive(serde::Deserialize, serde::Serialize))]
#[derive(Default)]
pub struct TableDemo {
    virtual_scroll: bool,
    resizable: bool,
}

impl super::Demo for TableDemo {
    fn name(&self) -> &'static str {
        "☰ Table Demo"
    }

    fn show(&mut self, ctx: &egui::Context, open: &mut bool) {
        egui::Window::new(self.name())
            .open(open)
            .resizable(true)
            .default_width(400.0)
            .show(ctx, |ui| {
                use super::View as _;
                self.ui(ui);
            });
    }
}

impl super::View for TableDemo {
    fn ui(&mut self, ui: &mut egui::Ui) {
        ui.checkbox(&mut self.virtual_scroll, "Virtual scroll");
        ui.checkbox(&mut self.resizable, "Resizable columns");

        // Leave room for the source code link after the table demo:
        StripBuilder::new(ui)
            .size(Size::remainder()) // for the table
            .size(Size::exact(10.0)) // for the source code link
            .vertical(|mut strip| {
                strip.cell_clip(|ui| {
                    self.table_ui(ui);
                });
                strip.cell(|ui| {
                    ui.vertical_centered(|ui| {
                        ui.add(crate::__egui_github_link_file!());
                    });
                });
            });
    }
}

impl TableDemo {
    fn table_ui(&mut self, ui: &mut egui::Ui) {
        TableBuilder::new(ui)
            .striped(true)
            .column(Size::initial(60.0).at_least(40.0))
            .column(Size::remainder().at_least(60.0))
            .column(Size::initial(60.0).at_least(40.0))
            .resizable(self.resizable)
            .header(20.0, |mut header| {
                header.col_clip(|ui| {
                    ui.heading("Left");
                });
                header.col_clip(|ui| {
                    ui.heading("Middle");
                });
                header.col_clip(|ui| {
                    ui.heading("Right");
                });
            })
            .body(|mut body| {
                if self.virtual_scroll {
                    body.rows(20.0, 100_000, |index, mut row| {
                        row.col_clip(|ui| {
                            ui.label(index.to_string());
                        });
                        row.col_clip(|ui| {
                            ui.add(
                                egui::Label::new("virtual scroll, easily with thousands of rows!")
                                    .wrap(false),
                            );
                        });
                        row.col_clip(|ui| {
                            ui.label(index.to_string());
                        });
                    });
                } else {
                    for i in 0..20 {
                        let thick = i % 4 == 0;
                        let height = if thick { 25.0 } else { 15.0 };
                        body.row(height, |mut row| {
                            row.col_clip(|ui| {
                                ui.label(i.to_string());
                            });
                            row.col_clip(|ui| {
                                ui.style_mut().wrap = Some(false);
                                if thick {
                                    ui.heading("Extra thick row");
                                } else {
                                    ui.label("Normal row");
                                }
                            });
                            row.col_clip(|ui| {
                                ui.label(i.to_string());
                            });
                        });
                    }
                }
            });
    }
}