use crate::*;

pub struct Bordered<S: BorderStyle, W: Content<TuiOut>>(pub S, pub W);

render!(TuiOut: (self: Bordered<S: BorderStyle, W: Content<TuiOut>>) => {
    Fill::xy(lay!(Border(self.0), Padding::xy(1, 1, &self.1)))
});

pub struct Border<S: BorderStyle>(pub S);

render!(TuiOut: |self: Border<S: BorderStyle>, to| {
    let area = to.area();
    if area.w() > 0 && area.y() > 0 {
        to.blit(&self.0.nw(), area.x(), area.y(), self.0.style());
        to.blit(&self.0.ne(), area.x() + area.w() - 1, area.y(), self.0.style());
        to.blit(&self.0.sw(), area.x(), area.y() + area.h() - 1, self.0.style());
        to.blit(&self.0.se(), area.x() + area.w() - 1, area.y() + area.h() - 1, self.0.style());
        for x in area.x()+1..area.x()+area.w()-1 {
            to.blit(&self.0.n(), x, area.y(), self.0.style());
            to.blit(&self.0.s(), x, area.y() + area.h() - 1, self.0.style());
        }
        for y in area.y()+1..area.y()+area.h()-1 {
            to.blit(&self.0.w(), area.x(), y, self.0.style());
            to.blit(&self.0.e(), area.x() + area.w() - 1, y, self.0.style());
        }
    }
});

pub trait BorderStyle: Send + Sync + Copy {
    fn wrap <W: Content<TuiOut>> (self, w: W) -> Bordered<Self, W> {
        Bordered(self, w)
    }
    fn enclose <W: Content<TuiOut>> (self, w: W) -> impl Content<TuiOut> {
        lay!(Fill::xy(Border(self)), w)
    }
    fn enclose_bg <W: Content<TuiOut>> (self, w: W) -> impl Content<TuiOut> {
        Tui::bg(self.style().unwrap().bg.unwrap_or(Color::Reset), lay!(
            Fill::xy(Border(self)),
            w
        ))
    }
    const NW: &'static str = "";
    const N:  &'static str = "";
    const NE: &'static str = "";
    const E:  &'static str = "";
    const SE: &'static str = "";
    const S:  &'static str = "";
    const SW: &'static str = "";
    const W:  &'static str = "";

    const N0: &'static str = "";
    const S0: &'static str = "";
    const W0: &'static str = "";
    const E0: &'static str = "";

    fn n (&self) -> &str { Self::N }
    fn s (&self) -> &str { Self::S }
    fn e (&self) -> &str { Self::E }
    fn w (&self) -> &str { Self::W }
    fn nw (&self) -> &str { Self::NW }
    fn ne (&self) -> &str { Self::NE }
    fn sw (&self) -> &str { Self::SW }
    fn se (&self) -> &str { Self::SE }
    #[inline] fn draw <'a> (
        &self, to: &mut TuiOut
    ) -> Usually<()> {
        self.draw_horizontal(to, None)?;
        self.draw_vertical(to, None)?;
        self.draw_corners(to, None)?;
        Ok(())
    }
    #[inline] fn draw_horizontal (
        &self, to: &mut TuiOut, style: Option<Style>
    ) -> Usually<[u16;4]> {
        let area = to.area();
        let style = style.or_else(||self.style_horizontal());
        let [x, x2, y, y2] = area.lrtb();
        for x in x..x2.saturating_sub(1) {
            to.blit(&Self::N, x, y, style);
            to.blit(&Self::S, x, y2.saturating_sub(1), style)
        }
        Ok(area)
    }
    #[inline] fn draw_vertical (
        &self, to: &mut TuiOut, style: Option<Style>
    ) -> Usually<[u16;4]> {
        let area = to.area();
        let style = style.or_else(||self.style_vertical());
        let [x, x2, y, y2] = area.lrtb();
        let h = y2 - y;
        if h > 1 {
            for y in y..y2.saturating_sub(1) {
                to.blit(&Self::W, x, y, style);
                to.blit(&Self::E, x2.saturating_sub(1), y, style);
            }
        } else if h > 0 {
            to.blit(&Self::W0, x, y, style);
            to.blit(&Self::E0, x2.saturating_sub(1), y, style);
        }
        Ok(area)
    }
    #[inline] fn draw_corners (
        &self, to: &mut TuiOut, style: Option<Style>
    ) -> Usually<[u16;4]> {
        let area = to.area();
        let style = style.or_else(||self.style_corners());
        let [x, y, width, height] = area.xywh();
        if width > 1 && height > 1 {
            to.blit(&Self::NW, x, y, style);
            to.blit(&Self::NE, x + width - 1, y, style);
            to.blit(&Self::SW, x, y + height - 1, style);
            to.blit(&Self::SE, x + width - 1, y + height - 1, style);
        }
        Ok(area)
    }
    #[inline] fn style            (&self) -> Option<Style> { None }
    #[inline] fn style_horizontal (&self) -> Option<Style> { self.style() }
    #[inline] fn style_vertical   (&self) -> Option<Style> { self.style() }
    #[inline] fn style_corners    (&self) -> Option<Style> { self.style() }
}

macro_rules! border {
    ($($T:ident {
        $nw:literal $n:literal $ne:literal $w:literal $e:literal $sw:literal $s:literal $se:literal
        $($x:tt)*
    }),+) => {$(
        impl BorderStyle for $T {
            const NW: &'static str = $nw;
            const N:  &'static str = $n;
            const NE: &'static str = $ne;
            const W:  &'static str = $w;
            const E:  &'static str = $e;
            const SW: &'static str = $sw;
            const S:  &'static str = $s;
            const SE: &'static str = $se;
            $($x)*
        }
        #[derive(Copy, Clone)]
        pub struct $T(pub Style);
        impl Content<TuiOut> for $T {
            fn render (&self, to: &mut TuiOut) { self.draw(to); }
        }
    )+}
}

border! {
    Square {
        "┌" "─" "┐"
        "│"     "│"
        "└" "─" "┘" fn style (&self) -> Option<Style> { Some(self.0) }
    },
    SquareBold {
        "┏" "━" "┓"
        "┃"     "┃"
        "┗" "━" "┛" fn style (&self) -> Option<Style> { Some(self.0) }
    },
    TabLike {
        "╭" "─" "╮"
        "│"     "│"
        "│" " " "│" fn style (&self) -> Option<Style> { Some(self.0) }
    },
    Lozenge {
        "╭" "─" "╮"
        "│"     "│"
        "╰" "─" "╯" fn style (&self) -> Option<Style> { Some(self.0) }
    },
    Brace {
        "╭" "" "╮"
        "│"    "│"
        "╰" "" "╯" fn style (&self) -> Option<Style> { Some(self.0) }
    },
    LozengeDotted {
        "╭" "┅" "╮"
        "┇"     "┇"
        "╰" "┅" "╯" fn style (&self) -> Option<Style> { Some(self.0) }
    },
    Quarter {
        "▎" "▔" "🮇"
        "▎"     "🮇"
        "▎" "▁" "🮇" fn style (&self) -> Option<Style> { Some(self.0) }
    },
    QuarterV {
        "▎" "" "🮇"
        "▎"    "🮇"
        "▎" "" "🮇" fn style (&self) -> Option<Style> { Some(self.0) }
    },
    Chamfer {
        "🭂" "▔" "🭍"
        "▎"     "🮇"
        "🭓" "▁" "🭞" fn style (&self) -> Option<Style> { Some(self.0) }
    },
    Corners {
        "🬆" "" "🬊" // 🬴 🬸
        ""     ""
        "🬱" "" "🬵" fn style (&self) -> Option<Style> { Some(self.0) }
    },
    CornersTall {
        "🭽" "" "🭾"
        ""     ""
        "🭼" "" "🭿" fn style (&self) -> Option<Style> { Some(self.0) }
    },
    Outer {
        "🭽" "▔" "🭾"
        "▏"     "▕"
        "🭼" "▁" "🭿"
        const W0: &'static str = "[";
        const E0: &'static str = "]";
        const N0: &'static str = "⎴";
        const S0: &'static str = "⎵";
        fn style (&self) -> Option<Style> { Some(self.0) }
    },
    Brackets {
        "⎡" "" "⎤"
        ""     ""
        "⎣" "" "⎦"
        const W0: &'static str = "[";
        const E0: &'static str = "]";
        const N0: &'static str = "⎴";
        const S0: &'static str = "⎵";
        fn style (&self) -> Option<Style> { Some(self.0) }
    },
    Reticle {
        "⎡" "" "⎤"
        ""     ""
        "⎣" "" "⎦"
        const W0: &'static str = "╟";
        const E0: &'static str = "╢";
        const N0: &'static str = "┯";
        const S0: &'static str = "┷";
        fn style (&self) -> Option<Style> { Some(self.0) }
    }
}

pub const CORNERS: Brackets = Brackets(Style {
    fg:              Some(Color::Rgb(96, 255, 32)),
    bg:              None,
    underline_color: None,
    add_modifier:    Modifier::empty(),
    sub_modifier:    Modifier::DIM
});

pub const RETICLE: Reticle = Reticle(Style {
    fg:              Some(Color::Rgb(96, 255, 32)),
    bg:              None,
    underline_color: None,
    add_modifier:    Modifier::empty(),
    sub_modifier:    Modifier::DIM
});