v/vlib/term/colors.v

132 lines
2.8 KiB
Go

// Copyright (c) 2019 Alexander Medvednikov. All rights reserved.
// Use of this source code is governed by an MIT license
// that can be found in the LICENSE file.
module term
import os
// Calling this functions enables color terminal output on windows
// Maybe implement a way to auto run an init function when a module
// is imported on a certain os. for example to run this?
pub fn enable_term_color_win() {
$if windows {
h_output := C.GetStdHandle(os.STD_OUTPUT_HANDLE)
if h_output == os.INVALID_HANDLE_VALUE
|| !C.SetConsoleMode(h_output, os.ENABLE_PROCESSED_OUTPUT|os.ENABLE_VIRTUAL_TERMINAL_PROCESSING) {
println('enable_term_color_win() Sorry, there was an error enabling terminal color.')
}
}
$else {
println('enable_term_color_win() should only be called on windows.')
}
}
pub fn format(msg, open, close string) string {
$if windows {
enable_term_color_win()
}
return '\x1b[' + open + 'm' + msg + '\x1b[' + close + 'm'
}
pub fn bg_black(msg string) string {
return format(msg, '40', '49')
}
pub fn bg_blue(msg string) string {
return format(msg, '44', '49')
}
pub fn bg_cyan(msg string) string {
return format(msg, '46', '49')
}
pub fn bg_green(msg string) string {
return format(msg, '42', '49')
}
pub fn bg_magenta(msg string) string {
return format(msg, '45', '49')
}
pub fn bg_red(msg string) string {
return format(msg, '41', '49')
}
pub fn bg_white(msg string) string {
return format(msg, '47', '49')
}
pub fn bg_yellow(msg string) string {
return format(msg, '43', '49')
}
pub fn black(msg string) string {
return format(msg, '30', '39')
}
pub fn blue(msg string) string {
return format(msg, '34', '39')
}
pub fn bold(msg string) string {
return format(msg, '1', '22')
}
pub fn cyan(msg string) string {
return format(msg, '36', '39')
}
pub fn dim(msg string) string {
return format(msg, '2', '22')
}
pub fn green(msg string) string {
return format(msg, '32', '39')
}
pub fn gray(msg string) string {
return format(msg, '90', '39')
}
pub fn hidden(msg string) string {
return format(msg, '8', '28')
}
pub fn italic(msg string) string {
return format(msg, '3', '23')
}
pub fn inverse(msg string) string {
return format(msg, '7', '27')
}
pub fn magenta(msg string) string {
return format(msg, '35', '39')
}
pub fn reset(msg string) string {
return format(msg, '0', '0')
}
pub fn red(msg string) string {
return format(msg, '31', '39')
}
pub fn strikethrough(msg string) string {
return format(msg, '9', '29')
}
pub fn underline(msg string) string {
return format(msg, '4', '24')
}
pub fn white(msg string) string {
return format(msg, '37', '39')
}
pub fn yellow(msg string) string {
return format(msg, '33', '39')
}