use crate::color_config::lookup_ansi_color_style;
use nu_ansi_term::{Color, Style};
use nu_protocol::Config;

pub fn get_shape_color(shape: String, conf: &Config) -> Style {
    match conf.color_config.get(shape.as_str()) {
        Some(int_color) => match int_color.as_string() {
            Ok(int_color) => lookup_ansi_color_style(&int_color),
            Err(_) => Style::default(),
        },
        None => match shape.as_ref() {
            "shape_garbage" => Style::new().fg(Color::White).on(Color::Red).bold(),
            "shape_bool" => Style::new().fg(Color::LightCyan),
            "shape_int" => Style::new().fg(Color::Purple).bold(),
            "shape_float" => Style::new().fg(Color::Purple).bold(),
            "shape_range" => Style::new().fg(Color::Yellow).bold(),
            "shape_internalcall" => Style::new().fg(Color::Cyan).bold(),
            "shape_external" => Style::new().fg(Color::Cyan),
            "shape_externalarg" => Style::new().fg(Color::Green).bold(),
            "shape_literal" => Style::new().fg(Color::Blue),
            "shape_operator" => Style::new().fg(Color::Yellow),
            "shape_signature" => Style::new().fg(Color::Green).bold(),
            "shape_string" => Style::new().fg(Color::Green),
            "shape_string_interpolation" => Style::new().fg(Color::Cyan).bold(),
            "shape_list" => Style::new().fg(Color::Cyan).bold(),
            "shape_table" => Style::new().fg(Color::Blue).bold(),
            "shape_record" => Style::new().fg(Color::Cyan).bold(),
            "shape_block" => Style::new().fg(Color::Blue).bold(),
            "shape_filepath" => Style::new().fg(Color::Cyan),
            "shape_globpattern" => Style::new().fg(Color::Cyan).bold(),
            "shape_variable" => Style::new().fg(Color::Purple),
            "shape_flag" => Style::new().fg(Color::Blue).bold(),
            "shape_custom" => Style::new().fg(Color::Green),
            "shape_nothing" => Style::new().fg(Color::LightCyan),
            _ => Style::default(),
        },
    }
}