pride/src/draw.rs

use std::io::{ self, Write };

use termion::{
	terminal_size,

	clear,
	color::{ Bg, Fg, Rgb },
	cursor,
	input::TermRead,
	raw::IntoRawMode
};

use crate::{
	color::{ RESET, RESET_BG },
	flag::Flag
};

pub static BLOCK: &str = "█";
pub static UHALF: &str = "▀";

pub fn draw_lines(lines: Vec<String>, hold: bool) {
	let mut stdout = io::stdout().into_raw_mode().unwrap();

	let count = lines.len() as u16;
	for _ in 0..count { write!(stdout, "\n").ok(); }
	write!(stdout, "{}", cursor::Up(count)).ok();

	if hold { write!(stdout, "{}{}", cursor::Hide, clear::All).ok(); }

	let down = cursor::Down(1);
	for line in lines {
		let left = cursor::Left(line.len() as u16);
		write!(stdout, "{line}{left}{down}").ok();
	}

	write!(stdout, "{RESET}{RESET_BG}").ok();
	stdout.flush().ok();
	if hold {
		let stdin = io::stdin();
		for _ in stdin.keys() { break; }
		write!(stdout, "{}", clear::All).ok();
	}
	write!(stdout, "{}", cursor::Show).ok();
	stdout.flush().ok();
}

pub fn fg_stripes(colors: Vec<Fg<Rgb>>, width: u16, height: u16) -> Vec<String> {
	let width = width as usize;
	let height = height as usize;
	let count = colors.len();

	let thresh = height / count;

	let stripe = BLOCK.repeat(width);
	let mut output = Vec::new();

	//	create our color index
	let mut index = 0;
	for n in 0..height {
		if n != 0 && n % thresh == 0 {
			index += 1;
			//	and break if out of bounds
			if index >= count { break; }
		}
		let color = colors[index];
		output.push(format!("{color}{stripe}"));
	}

	output
}
pub fn bg_stripes(colors: Vec<Bg<Rgb>>, width: u16, height: u16) -> Vec<String> {
	let width = width as usize;
	let height = height as usize;
	let count = colors.len();

	let thresh = height / count;

	let stripe = " ".repeat(width);
	let mut output = Vec::new();

	let mut index = 0;
	for n in 0..height {
		if n != 0 && n % thresh == 0 {
			index += 1;
			if index >= count { break; }
		}
		let color = colors[index];
		output.push(format!("{color}{stripe}"));
	}

	output
}

impl Flag {
	pub fn draw(self, hold: bool) {
		let lines = match self {
			Flag::Stripes(colors)
				=> {
					let (width, height);
					if hold { (width, height) = terminal_size().unwrap(); }
					else {
						height = colors.len() as u16;
						width = height * 3;
					}
					fg_stripes(colors, width, height)
				},
			Flag::Lines(lines)
				=>	lines
		};
		draw_lines(lines, hold);
	}
}