// Seems like google isn't using this setup anymore, keeping it here just in case...

use comfy_table::{
    modifiers::{UTF8_ROUND_CORNERS, UTF8_SOLID_INNER_BORDERS},
    presets::UTF8_FULL,
    Attribute, Cell, CellAlignment, ContentArrangement, Row, Table
};
use scraper::Selector;

pub fn main(data: &scraper::Html, tty: &bool, w: usize) {
    let mut y: Vec<&str> = vec![];
    for x in data.select(&Selector::parse("div.Qynugf").unwrap()) {
        y.push(x.text().next().unwrap())
    }

    match tty {
        true => {
            let mut t = Table::new();
            t.load_preset(UTF8_FULL);
            t.apply_modifier(UTF8_ROUND_CORNERS);
            t.apply_modifier(UTF8_SOLID_INNER_BORDERS);
            t.set_content_arrangement(ContentArrangement::Dynamic);
            if w > 100 {
                t.set_width(100);
            }
            for x in &y {
                t.add_row(Row::from(vec![Cell::new(x)
                    .set_alignment(CellAlignment::Center)
                    .add_attribute(Attribute::Bold)]));
            }
            println!("{}", t)
        }
        false => {
            let j = y.len();
            for i in 0..j {
                println!("{}", y[i]);
                if i != (j - 1) {
                    println!()
                }
            }
        }
    }
}