gobang/src/ui/scrollbar.rs

use easy_cast::CastFloat;
use std::convert::TryFrom;
use tui::{
    backend::Backend,
    buffer::Buffer,
    layout::{Margin, Rect},
    style::Style,
    symbols::{
        block::FULL,
        line::{DOUBLE_HORIZONTAL, DOUBLE_VERTICAL},
    },
    widgets::Widget,
    Frame,
};

///
struct Scrollbar {
    max: u16,
    pos: u16,
    style_bar: Style,
    style_pos: Style,
    vertical: bool,
}

impl Scrollbar {
    fn new(max: usize, pos: usize, vertical: bool) -> Self {
        Self {
            max: u16::try_from(max).unwrap_or_default(),
            pos: u16::try_from(pos).unwrap_or_default(),
            style_pos: Style::default(),
            style_bar: Style::default(),
            vertical,
        }
    }
}

impl Widget for Scrollbar {
    fn render(self, area: Rect, buf: &mut Buffer) {
        if self.vertical && area.height <= 2 {
            return;
        }

        if !self.vertical && area.width <= 2 {
            return;
        }

        if self.max == 0 {
            return;
        }

        let right = area.right().saturating_sub(1);
        if right <= area.left() {
            return;
        };

        let bottom = area.bottom().saturating_sub(1);
        if bottom <= area.top() {
            return;
        };

        let (bar_top, bar_height) = {
            let scrollbar_area = if self.vertical {
                area.inner(&Margin {
                    horizontal: 0,
                    vertical: 1,
                })
            } else {
                area.inner(&Margin {
                    horizontal: 1,
                    vertical: 0,
                })
            };

            if self.vertical {
                (scrollbar_area.top(), scrollbar_area.height)
            } else {
                (scrollbar_area.left(), scrollbar_area.width)
            }
        };

        if self.vertical {
            for y in bar_top..(bar_top + bar_height) {
                buf.set_string(right, y, DOUBLE_VERTICAL, self.style_bar)
            }
        } else {
            for x in bar_top..(bar_top + bar_height) {
                buf.set_string(x, bottom, DOUBLE_HORIZONTAL, self.style_bar)
            }
        }

        let progress = f32::from(self.pos) / f32::from(self.max);
        let progress = if progress > 1.0 { 1.0 } else { progress };
        let pos = f32::from(bar_height) * progress;

        let pos: u16 = pos.cast_nearest();
        let pos = pos.saturating_sub(1);

        if self.vertical {
            buf.set_string(right, bar_top + pos, FULL, self.style_pos);
        } else {
            buf.set_string(bar_top + pos, bottom, "▆", self.style_pos);
        }
    }
}

pub fn draw_scrollbar<B: Backend>(
    f: &mut Frame<B>,
    r: Rect,
    max: usize,
    pos: usize,
    vertical: bool,
) {
    let mut widget = Scrollbar::new(max, pos, vertical);
    widget.style_pos = Style::default();
    f.render_widget(widget, r);
}