80 lines
2.6 KiB
Rust
80 lines
2.6 KiB
Rust
use mlua::{Result as lResult, Table};
|
|
use crossterm::style::Stylize;
|
|
use crate::vm::Vm;
|
|
|
|
macro_rules! foreground_styles_luau {
|
|
($self:expr, $style_table:expr, $($color:ident)+) => {
|
|
$(
|
|
$style_table.set(stringify!($color).to_ascii_uppercase(), $self.0.create_function(|_, text: String| -> lResult<String> {
|
|
Ok(text.$color().to_string())
|
|
})?)?;
|
|
)+
|
|
};
|
|
}
|
|
|
|
macro_rules! background_styles_luau {
|
|
($self:expr, $style_table:expr, $($color:ident)+) => {
|
|
$(
|
|
match stringify!($color).split_once("_") {
|
|
Some((_, color_name)) => $style_table.set(color_name.to_ascii_uppercase(), $self.0.create_function(|_, text: String| -> lResult<String> {
|
|
Ok(text.$color().to_string())
|
|
})?)?,
|
|
None => panic!("Luau set error: {:?}. There was nothing to split from delimiter: \"_\"", stringify!($color)),
|
|
}
|
|
)+
|
|
};
|
|
}
|
|
|
|
pub trait TerminalColors {
|
|
fn background(&self, style_table: &Table) -> lResult<()>;
|
|
fn foreground(&self, style_table: &Table) -> lResult<()>;
|
|
fn styling(&self) -> lResult<Table>;
|
|
fn terminal(&self) -> lResult<()>;
|
|
}
|
|
impl TerminalColors for Vm {
|
|
fn background(&self, style_table: &Table) -> lResult<()> {
|
|
let foreground_table = self.0.create_table()?;
|
|
foreground_styles_luau!(self, foreground_table,
|
|
dark_grey dark_red dark_green dark_cyan
|
|
dark_yellow dark_magenta dark_blue
|
|
red grey black green yellow
|
|
blue magenta cyan white
|
|
underlined
|
|
underline_dark_grey underline_dark_red underline_dark_green underline_dark_cyan
|
|
underline_dark_yellow underline_dark_magenta underline_dark_blue underline_red
|
|
underline_grey underline_black underline_green underline_yellow
|
|
underline_blue underline_magenta underline_cyan underline_white
|
|
bold
|
|
);
|
|
style_table.set("FOREGROUND", foreground_table)?;
|
|
Ok(())
|
|
}
|
|
|
|
fn foreground(&self, style_table: &Table) -> lResult<()> {
|
|
let background_table = self.0.create_table()?;
|
|
background_styles_luau!(self, background_table,
|
|
on_dark_grey on_dark_red on_dark_green on_dark_cyan
|
|
on_dark_yellow on_dark_magenta on_dark_blue
|
|
on_red on_grey on_black
|
|
on_green on_yellow
|
|
on_blue on_magenta
|
|
on_cyan on_white
|
|
);
|
|
style_table.set("BACKGROUND", background_table)?;
|
|
Ok(())
|
|
}
|
|
|
|
fn styling(&self) -> lResult<Table> {
|
|
let style_table = self.0.create_table()?;
|
|
self.foreground(&style_table)?;
|
|
self.background(&style_table)?;
|
|
Ok(style_table)
|
|
}
|
|
|
|
fn terminal(&self) -> lResult<()> {
|
|
let term_table = self.0.create_table()?;
|
|
term_table.set("OUT", self.styling()?)?;
|
|
self.0.globals().set("TERMINAL", &term_table)?;
|
|
Ok(())
|
|
}
|
|
} |