use std::io::{self, IsTerminal, Write};
use clap::builder::styling::{AnsiColor, Effects, Style, Styles};
use clap::CommandFactory;
use crate::cli::Args;
fn cyber_styles() -> Styles {
Styles::styled()
.header(
Style::new()
.fg_color(Some(AnsiColor::Cyan.into()))
.effects(Effects::BOLD | Effects::UNDERLINE),
)
.usage(
Style::new()
.fg_color(Some(AnsiColor::Yellow.into()))
.effects(Effects::BOLD),
)
.literal(
Style::new()
.fg_color(Some(AnsiColor::Green.into()))
.effects(Effects::BOLD),
)
.placeholder(Style::new().fg_color(Some(AnsiColor::Magenta.into())))
.valid(Style::new().fg_color(Some(AnsiColor::Green.into())))
.invalid(
Style::new()
.fg_color(Some(AnsiColor::Red.into()))
.effects(Effects::BOLD),
)
.error(
Style::new()
.fg_color(Some(AnsiColor::Red.into()))
.effects(Effects::BOLD),
)
}
const BOX_INNER: usize = 54;
fn color_on() -> bool {
std::env::var_os("NO_COLOR").is_none()
&& (io::stdout().is_terminal() || std::env::var_os("CLICOLOR_FORCE").is_some())
}
fn c_cyan(s: &str) -> String {
if color_on() {
format!("\x1b[36m{s}\x1b[0m")
} else {
s.to_string()
}
}
fn c_magenta(s: &str) -> String {
if color_on() {
format!("\x1b[35m{s}\x1b[0m")
} else {
s.to_string()
}
}
fn c_yellow(s: &str) -> String {
if color_on() {
format!("\x1b[33m{s}\x1b[0m")
} else {
s.to_string()
}
}
fn c_dim(s: &str) -> String {
if color_on() {
format!("\x1b[2m{s}\x1b[0m")
} else {
s.to_string()
}
}
fn awkrs_logo() -> String {
c_cyan(crate::banner::AWKRS_LOGO)
}
fn status_box(version: &str) -> String {
let top = format!(" ┌{}┐", "─".repeat(BOX_INNER));
let bottom = format!(" └{}┘", "─".repeat(BOX_INNER));
let mut inner = format!(" STATUS: ONLINE // SIGNAL: ████████░░ // v{version}");
while inner.chars().count() < BOX_INNER {
inner.push(' ');
}
if inner.chars().count() > BOX_INNER {
inner = inner.chars().take(BOX_INNER).collect();
}
let mid = format!(" │{inner}│");
format!("{}\n{}\n{}", c_cyan(&top), c_magenta(&mid), c_cyan(&bottom))
}
fn tagline() -> String {
let s = " >> PATTERN / ACTION ENGINE // FIELD RECORDS & TEXT HAX <<";
c_magenta(s)
}
fn section_rule(label: &str) -> String {
let prefix = format!("── {label} ");
let pad = BOX_INNER.saturating_sub(prefix.chars().count());
format!("\x1b[36m {}{}\x1b[0m", prefix, "─".repeat(pad))
}
fn help_template() -> String {
format!(
"\x1b[33mUSAGE:\x1b[0m {{usage}}\n\n{opt}\n{{options}}\n\n{pos}\n{{positionals}}",
opt = section_rule("OPTIONS"),
pos = section_rule("POSITIONAL"),
)
}
fn strip_ansi(s: &str) -> String {
let mut out = String::with_capacity(s.len());
let mut chars = s.chars();
while let Some(c) = chars.next() {
if c == '\x1b' {
for c2 in chars.by_ref() {
if c2 == 'm' {
break;
}
}
} else {
out.push(c);
}
}
out
}
fn footer(version: &str) -> String {
let rule = "─".repeat(58);
let line1 = format!(" {rule}");
let line2 = format!(" {version} // (c) MenkeTechnologies // MIT");
let line3 = " >>> PARSE THE STREAM. SPLIT THE FIELDS. JACK IN. <<<";
let dots = "░".repeat(55);
format!(
"\n{}\n{}\n{}\n{}\n",
c_dim(&line1),
c_dim(&line2),
c_yellow(line3),
c_dim(&dots)
)
}
pub fn print_cyberpunk_help(bin_name: &str) {
let version = env!("CARGO_PKG_VERSION");
let mut out = io::stdout();
let _ = writeln!(out, "{}", awkrs_logo());
let _ = writeln!(out, "{}", status_box(version));
let _ = writeln!(out, " {}", tagline());
let _ = writeln!(out);
let _ = writeln!(
out,
"{}",
c_dim(
"Pattern-directed scanning: union CLI (POSIX / gawk / mawk-style). Sequential engine."
)
);
let _ = writeln!(
out,
"{}",
c_dim("Modes: --repl (interactive REPL; also the default on a bare terminal) // --lsp // --dap")
);
let _ = writeln!(out);
let colored = color_on();
let mut cmd = Args::command()
.bin_name(bin_name)
.styles(cyber_styles())
.help_template(help_template());
cmd = cmd.color(if colored {
clap::ColorChoice::Always
} else {
clap::ColorChoice::Never
});
let body = cmd.render_help().ansi().to_string();
if colored {
let _ = write!(out, "{body}");
} else {
let _ = write!(out, "{}", strip_ansi(&body));
}
let _ = write!(out, "{}", footer(version));
let _ = writeln!(out);
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn status_box_has_fixed_inner_width_and_version() {
let s = status_box("0.0.0");
assert!(s.contains("0.0.0"));
assert!(s.contains("ONLINE"));
assert!(s.contains("SIGNAL"));
}
#[test]
fn footer_contains_version_and_license_tag() {
let f = footer("1.2.3");
assert!(f.contains("1.2.3"));
assert!(f.contains("MIT"));
}
#[test]
fn tagline_non_empty() {
assert!(!tagline().is_empty());
}
#[test]
fn status_box_middle_line_has_pipe_borders_and_version() {
let s = status_box("0.0.0-test");
assert!(
s.contains("0.0.0-test") && s.contains('│') && s.contains("ONLINE"),
"{s}"
);
let lines: Vec<_> = s.lines().collect();
assert!(
lines
.iter()
.any(|line| line.contains('│') && line.contains("STATUS")),
"{lines:?}"
);
}
#[test]
fn awkrs_logo_contains_awkrs_spelled_in_ascii_art() {
let logo = awkrs_logo();
assert!(
logo.contains("AWK") || logo.contains("awk") || logo.contains('█'),
"logo should contain banner glyphs: len={}",
logo.len()
);
}
#[test]
fn cyber_styles_instantiates_without_panic_v2() {
let _s = cyber_styles();
}
#[test]
fn color_helpers_fallback_to_string_v2() {
let s = "test_string";
let cyan = c_cyan(s);
let mag = c_magenta(s);
let yel = c_yellow(s);
let dim = c_dim(s);
assert!(cyan.contains(s));
assert!(mag.contains(s));
assert!(yel.contains(s));
assert!(dim.contains(s));
}
}