atac 0.11.2

use cookie_store::Cookie;
use ratatui::layout::Constraint;
use ratatui::widgets::ListState;
use strum::{Display, FromRepr};

use crate::utils::text_input::TextInput;

#[derive(Display, FromRepr)]
pub enum CookieColumns {
    #[strum(to_string = "URL")]
    URL,
    #[strum(to_string = "Name")]
    Name,
    #[strum(to_string = "Value")]
    Value,
    #[strum(to_string = "Path")]
    Path,
    #[strum(to_string = "Expires")]
    Expires,
    #[strum(to_string = "Http\nonly")]
    HttpOnly,
    #[strum(to_string = "Secure")]
    Secure,
    #[strum(to_string = "Same\nsite")]
    SameSite
}

impl CookieColumns {
    pub fn constraints() -> [Constraint; 8]{
        [
            Constraint::Percentage(10),
            Constraint::Percentage(15),
            Constraint::Percentage(37),
            Constraint::Percentage(10),
            Constraint::Percentage(10),
            Constraint::Percentage(6),
            Constraint::Percentage(6),
            Constraint::Percentage(6)
        ]
    }
}

pub const COOKIES_COLUMNS_NUMBER: usize = 8;

pub struct StatefulCookieTable {
    pub lists_states: [ListState; COOKIES_COLUMNS_NUMBER],
    /// (x, y)
    pub selection: Option<(usize, usize)>,
    pub rows: Vec<[String; COOKIES_COLUMNS_NUMBER]>,
    pub selection_text_input: TextInput,
}

impl StatefulCookieTable {
    fn decrement_x(&self, i: usize) -> usize {
        if i == 0 {
            self.rows.len() - 1
        } else {
            i - 1
        }
    }

    fn increment_x(&self, i: usize) -> usize {
        if i >= self.rows.len() - 1 {
            0
        } else {
            i + 1
        }
    }

    fn decrement_y(&self, i: usize) -> usize {
        if i == 0 {
            COOKIES_COLUMNS_NUMBER - 1
        } else {
            i - 1
        }
    }

    pub fn increment_y(&mut self, i: usize) -> usize {
        if i >= COOKIES_COLUMNS_NUMBER - 1 {
            0
        } else {
            i + 1
        }
    }

    pub fn up(&mut self) {
        if self.rows.is_empty() || self.selection.is_none() {
            return;
        }

        let selection = self.selection.unwrap();

        let x = match self.lists_states[selection.1].selected() {
            None => 0,
            Some(i) => self.decrement_x(i)
        };

        for list_state in self.lists_states.iter_mut() {
            list_state.select(Some(x));
        }

        match self.selection.unwrap() {
            (_, y) => self.selection = Some((x, y))
        }
    }

    pub fn down(&mut self) {
        if self.rows.is_empty() || self.selection.is_none() {
            return;
        }

        let selection = self.selection.unwrap();

        let x = match self.lists_states[selection.1].selected() {
            None => 0,
            Some(i) => self.increment_x(i)
        };

        for list_state in self.lists_states.iter_mut() {
            list_state.select(Some(x));
        }

        match self.selection.unwrap() {
            (_, y) => self.selection = Some((x, y))
        }
    }

    pub fn left(&mut self) {
        if self.rows.is_empty() || self.selection.is_none() {
            return;
        }

        let selection = self.selection.unwrap();

        let y = self.decrement_y(selection.1);

        match self.selection.unwrap() {
            (x, _) => self.selection = Some((x, y))
        }
    }

    pub fn right(&mut self) {
        if self.rows.is_empty() || self.selection.is_none() {
            return;
        }

        let selection = self.selection.unwrap();

        let y = self.increment_y(selection.1);

        match self.selection.unwrap() {
            (x, _) => self.selection = Some((x, y))
        }
    }
}

impl Default for StatefulCookieTable {
    fn default() -> Self {
        StatefulCookieTable {
            lists_states: [
                ListState::default(),
                ListState::default(),
                ListState::default(),
                ListState::default(),
                ListState::default(),
                ListState::default(),
                ListState::default(),
                ListState::default()
            ],
            selection: None,
            rows: vec![],
            selection_text_input: TextInput::default(),
        }
    }
}

pub fn cookie_to_row(cookie: &Cookie) -> [String; COOKIES_COLUMNS_NUMBER]{
    [
        match cookie.domain() {
            None => String::new(),
            Some(domain) => domain.to_string()
        },
        cookie.name().to_string(),
        cookie.value().to_string(),
        match cookie.path() {
            None => String::new(),
            Some(path) => path.to_string()
        },
        match cookie.expires() {
            None => String::new(),
            Some(expiration) => match expiration.is_datetime() {
                true => expiration.datetime().unwrap().to_string(),
                false => String::from("session")
            }
        },
        match cookie.http_only() {
            None => String::new(),
            Some(http_only) => http_only.to_string()
        },
        match cookie.secure() {
            None => String::new(),
            Some(secure) => secure.to_string()
        },
        match cookie.same_site() {
            None => String::new(),
            Some(same_site) => same_site.to_string()
        }
    ]
}