use std::io::{self, Write};
#[derive(Debug, Clone, Copy, Default, PartialEq, Eq)]
pub struct TextAttr {
pub bold: bool,
pub underline: bool,
pub standout: bool,
pub blink: bool,
pub fg_color: Option<u8>,
pub bg_color: Option<u8>,
}
impl TextAttr {
pub fn to_ansi(&self) -> String {
let mut codes = Vec::new();
if self.bold {
codes.push("1".to_string());
}
if self.underline {
codes.push("4".to_string());
}
if self.standout {
codes.push("7".to_string());
}
if self.blink {
codes.push("5".to_string());
}
if let Some(fg) = self.fg_color {
codes.push(format!("38;5;{}", fg));
}
if let Some(bg) = self.bg_color {
codes.push(format!("48;5;{}", bg));
}
if codes.is_empty() {
String::new()
} else {
format!("\x1b[{}m", codes.join(";"))
}
}
}
#[derive(Debug, Clone, Default)]
pub struct RefreshElement {
pub chr: char,
pub atr: TextAttr,
pub width: u8,
}
impl RefreshElement {
pub fn new(chr: char) -> Self {
let width = unicode_width::UnicodeWidthChar::width(chr).unwrap_or(1) as u8;
RefreshElement {
chr,
atr: TextAttr::default(),
width,
}
}
pub fn with_attr(chr: char, atr: TextAttr) -> Self {
let width = unicode_width::UnicodeWidthChar::width(chr).unwrap_or(1) as u8;
RefreshElement { chr, atr, width }
}
}
#[derive(Debug, Clone)]
pub struct VideoBuffer {
pub lines: Vec<Vec<RefreshElement>>,
pub cols: usize,
pub rows: usize,
}
impl VideoBuffer {
pub fn new(cols: usize, rows: usize) -> Self {
let lines = vec![vec![RefreshElement::new(' '); cols]; rows];
VideoBuffer { lines, cols, rows }
}
pub fn clear(&mut self) {
for line in &mut self.lines {
for elem in line.iter_mut() {
*elem = RefreshElement::new(' ');
}
}
}
pub fn resize(&mut self, cols: usize, rows: usize) {
self.cols = cols;
self.rows = rows;
self.lines
.resize(rows, vec![RefreshElement::new(' '); cols]);
for line in &mut self.lines {
line.resize(cols, RefreshElement::new(' '));
}
}
pub fn set(&mut self, row: usize, col: usize, elem: RefreshElement) {
if row < self.rows && col < self.cols {
self.lines[row][col] = elem;
}
}
pub fn get(&self, row: usize, col: usize) -> Option<&RefreshElement> {
self.lines.get(row).and_then(|line| line.get(col))
}
}
#[derive(Debug, Clone, Default)]
pub struct RefreshState {
pub columns: usize, pub lines: usize, pub vln: usize, pub vcs: usize, pub lpromptw: usize, pub rpromptw: usize, pub scrolloff: usize,
pub region_highlight_start: Option<usize>,
pub region_highlight_end: Option<usize>,
pub old_video: Option<VideoBuffer>,
pub new_video: Option<VideoBuffer>,
pub lpromptbuf: String,
pub rpromptbuf: String,
pub need_full_redraw: bool,
pub predisplay: String,
pub postdisplay: String,
}
impl RefreshState {
pub fn new() -> Self {
let (cols, rows) = (
crate::ported::utils::adjustcolumns(),
crate::ported::utils::adjustlines(),
);
RefreshState {
columns: cols,
lines: rows,
old_video: Some(VideoBuffer::new(cols, rows)),
new_video: Some(VideoBuffer::new(cols, rows)),
need_full_redraw: true,
..Default::default()
}
}
pub fn reset_video(&mut self) {
let (cols, rows) = (
crate::ported::utils::adjustcolumns(),
crate::ported::utils::adjustlines(),
);
self.columns = cols;
self.lines = rows;
self.old_video = Some(VideoBuffer::new(cols, rows));
self.new_video = Some(VideoBuffer::new(cols, rows));
self.need_full_redraw = true;
}
pub fn free_video(&mut self) {
self.old_video = None;
self.new_video = None;
}
pub fn swap_buffers(&mut self) {
std::mem::swap(&mut self.old_video, &mut self.new_video);
if let Some(ref mut new) = self.new_video {
new.clear();
}
}
}
use HighlightCategory as HC;
use crate::ported::zsh_h::TXT_MULTIWORD_MASK;
#[allow(unused_imports)]
#[allow(unused_imports)]
use crate::ported::zle::zle_main::*;
#[allow(unused_imports)]
use crate::ported::zle::zle_misc::*;
#[allow(unused_imports)]
use crate::ported::zle::zle_hist::*;
#[allow(unused_imports)]
use crate::ported::zle::zle_move::*;
#[allow(unused_imports)]
use crate::ported::zle::zle_word::*;
#[allow(unused_imports)]
use crate::ported::zle::zle_params::*;
#[allow(unused_imports)]
use crate::ported::zle::zle_vi::*;
#[allow(unused_imports)]
use crate::ported::zle::zle_utils::*;
#[allow(unused_imports)]
use crate::ported::zle::zle_tricky::*;
#[allow(unused_imports)]
use crate::ported::zle::textobjects::*;
#[allow(unused_imports)]
use crate::ported::zle::deltochar::*;
pub fn zrefresh() { use std::fmt::Write as FmtWrite;
let mut handle = String::new();
let (cols, _rows) = (crate::ported::utils::adjustcolumns(), crate::ported::utils::adjustlines());
let prompt = prompt().to_string();
let rprompt = rprompt().to_string();
let cursor = crate::ported::zle::zle_main::ZLECS.load(std::sync::atomic::Ordering::SeqCst);
let prompt_width = countprompt(&prompt);
let rprompt_width = countprompt(&rprompt);
let buffer_before_cursor: String = crate::ported::zle::zle_main::ZLELINE.lock().unwrap()[..cursor.min(crate::ported::zle::zle_main::ZLELINE.lock().unwrap().len())]
.iter()
.collect();
let cursor_col = prompt_width + countprompt(&buffer_before_cursor);
let scroll_margin = 8;
let effective_cols = cols.saturating_sub(1);
let scroll_offset = if cursor_col >= effective_cols.saturating_sub(scroll_margin) {
cursor_col.saturating_sub(effective_cols / 2)
} else {
0
};
let attrs = compute_render_attrs();
let _ = write!(handle, "\r\x1b[K");
if scroll_offset < prompt_width {
let mut width = 0;
let mut byte_idx = 0;
let mut in_escape = false;
for (i, c) in prompt.char_indices() {
if width >= scroll_offset {
byte_idx = i;
break;
}
if in_escape {
if c.is_ascii_alphabetic() {
in_escape = false;
}
} else if c == '\x1b' {
in_escape = true;
} else {
width += unicode_width::UnicodeWidthChar::width(c).unwrap_or(0);
}
byte_idx = i + c.len_utf8();
}
let _ = write!(handle, "{}", &prompt[byte_idx..]);
}
let buffer_start = scroll_offset.saturating_sub(prompt_width);
let drawn_prompt_width = prompt_width.saturating_sub(scroll_offset);
let rprompt_reserve = if rprompt_width > 0 {
rprompt_width + 1
} else {
0
};
let buffer_budget = effective_cols
.saturating_sub(drawn_prompt_width)
.saturating_sub(rprompt_reserve);
let mut current_attr: Option<TextAttr> = None;
let line_snapshot = crate::ported::zle::zle_main::ZLELINE.lock().unwrap().clone();
for (written, (idx, ch)) in line_snapshot
.iter()
.enumerate()
.skip(buffer_start)
.enumerate()
{
if written >= buffer_budget {
break;
}
let want_attr = attrs.get(idx).and_then(|a| *a);
if want_attr != current_attr {
let _ = write!(handle, "\x1b[0m");
if let Some(a) = want_attr {
let _ = write!(handle, "{}", a.to_ansi());
}
current_attr = want_attr;
}
let _ = write!(handle, "{}", ch);
}
if current_attr.is_some() {
let _ = write!(handle, "\x1b[0m");
}
if rprompt_width > 0 && rprompt_width + 2 < effective_cols {
let rprompt_col = effective_cols.saturating_sub(rprompt_width);
let _ = write!(handle, "\r\x1b[{}C{}\x1b[0m", rprompt_col, rprompt);
}
let display_cursor_col = cursor_col.saturating_sub(scroll_offset);
let _ = write!(handle, "\r\x1b[{}C", display_cursor_col);
use std::sync::atomic::Ordering;
let fd = crate::ported::init::SHTTY.load(Ordering::Relaxed);
let out_fd = if fd >= 0 { fd } else { 1 };
let _ = crate::ported::utils::write_loop(out_fd, handle.as_bytes());
}
pub fn compute_render_attrs() -> Vec<Option<TextAttr>> {
let buf_len = crate::ported::zle::zle_main::ZLELINE.lock().unwrap().len();
let mut attrs: Vec<Option<TextAttr>> = vec![None; buf_len];
let visual_attr = crate::ported::zle::zle_main::highlight().lock().unwrap()
.category_attrs
.get(&HighlightCategory::Region)
.copied()
.unwrap_or(TextAttr {
standout: true,
..TextAttr::default()
});
if crate::ported::zle::zle_main::REGION_ACTIVE.load(std::sync::atomic::Ordering::SeqCst) != 0 {
let (lo, hi) = if crate::ported::zle::zle_main::MARK.load(std::sync::atomic::Ordering::SeqCst) <= crate::ported::zle::zle_main::ZLECS.load(std::sync::atomic::Ordering::SeqCst) {
(crate::ported::zle::zle_main::MARK.load(std::sync::atomic::Ordering::SeqCst), crate::ported::zle::zle_main::ZLECS.load(std::sync::atomic::Ordering::SeqCst))
} else {
(crate::ported::zle::zle_main::ZLECS.load(std::sync::atomic::Ordering::SeqCst), crate::ported::zle::zle_main::MARK.load(std::sync::atomic::Ordering::SeqCst))
};
let lo = lo.min(buf_len);
let hi = hi.min(buf_len);
for slot in attrs.iter_mut().take(hi).skip(lo) {
*slot = Some(visual_attr);
}
}
for region in &crate::ported::zle::zle_main::highlight().lock().unwrap().regions {
let start = region.start.min(buf_len);
let end = region.end.min(buf_len);
for slot in attrs.iter_mut().take(end).skip(start) {
*slot = Some(region.attr);
}
}
attrs
}
pub fn full_refresh() -> io::Result<()> {
use std::sync::atomic::Ordering;
let fd = crate::ported::init::SHTTY.load(Ordering::Relaxed);
let out = if fd >= 0 { fd } else { 1 };
let _ = crate::ported::utils::write_loop(out, b"\x1b[2J\x1b[H");
zrefresh();
Ok(())
}
pub fn partial_refresh() -> io::Result<()> {
zrefresh();
Ok(())
}
pub fn clearscreen() { let _ = crate::ported::utils::write_loop({ use std::sync::atomic::Ordering; let f = crate::ported::init::SHTTY.load(Ordering::Relaxed); if f >= 0 { f } else { 1 } }, b"\x1b[2J\x1b[H");
zrefresh();
}
pub fn redisplay() { zrefresh();
}
pub fn moveto(row: usize, col: usize) { let s = format!("\x1b[{};{}H", row + 1, col + 1);
let _ = crate::ported::utils::write_loop({ use std::sync::atomic::Ordering; let f = crate::ported::init::SHTTY.load(Ordering::Relaxed); if f >= 0 { f } else { 1 } }, s.as_bytes());
}
pub fn tc_downcurs(count: usize) {
if count > 0 {
let s = format!("\x1b[{}B", count);
let _ = crate::ported::utils::write_loop({ use std::sync::atomic::Ordering; let f = crate::ported::init::SHTTY.load(Ordering::Relaxed); if f >= 0 { f } else { 1 } }, s.as_bytes());
}
}
pub fn tc_rightcurs(count: usize) {
if count > 0 {
let s = format!("\x1b[{}C", count);
let _ = crate::ported::utils::write_loop({ use std::sync::atomic::Ordering; let f = crate::ported::init::SHTTY.load(Ordering::Relaxed); if f >= 0 { f } else { 1 } }, s.as_bytes());
}
}
pub fn scrollwindow(lines: i32) {
let s = if lines > 0 {
format!("\x1b[{}S", lines)
} else if lines < 0 {
format!("\x1b[{}T", -lines)
} else {
return;
};
let _ = crate::ported::utils::write_loop({ use std::sync::atomic::Ordering; let f = crate::ported::init::SHTTY.load(Ordering::Relaxed); if f >= 0 { f } else { 1 } }, s.as_bytes());
}
pub fn singlerefresh() { zrefresh();
}
pub fn refreshline(_line: usize) {
zrefresh();
}
pub fn zwcputc(c: char) {
let mut buf = [0u8; 4];
let s = c.encode_utf8(&mut buf);
let _ = crate::ported::utils::write_loop({ use std::sync::atomic::Ordering; let f = crate::ported::init::SHTTY.load(Ordering::Relaxed); if f >= 0 { f } else { 1 } }, s.as_bytes());
}
pub fn zwcwrite(s: &str) {
let _ = crate::ported::utils::write_loop({ use std::sync::atomic::Ordering; let f = crate::ported::init::SHTTY.load(Ordering::Relaxed); if f >= 0 { f } else { 1 } }, s.as_bytes());
}
fn countprompt(s: &str) -> usize {
let mut width = 0;
let mut in_escape = false;
for c in s.chars() {
if in_escape {
if c.is_ascii_alphabetic() {
in_escape = false;
}
} else if c == '\x1b' {
in_escape = true;
} else {
width += unicode_width::UnicodeWidthChar::width(c).unwrap_or(0);
}
}
width
}
#[derive(Debug, Clone)]
pub struct RegionHighlight {
pub start: usize,
pub end: usize,
pub attr: TextAttr,
pub memo: Option<String>,
}
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub enum HighlightCategory {
Region,
Isearch,
Suffix,
Paste,
Default,
Special,
Ellipsis,
}
#[derive(Debug, Default)]
pub struct HighlightManager {
pub regions: Vec<RegionHighlight>,
pub category_attrs: std::collections::HashMap<HighlightCategory, TextAttr>,
}
impl HighlightManager {
pub fn new() -> Self {
HighlightManager {
regions: Vec::new(),
category_attrs: std::collections::HashMap::new(),
}
}
pub fn set_region_highlight(&mut self, start: usize, end: usize, attr: TextAttr) {
self.regions.push(RegionHighlight {
start,
end,
attr,
memo: None,
});
}
pub fn get_region_highlight(&self, pos: usize) -> Option<&RegionHighlight> {
self.regions.iter().find(|r| pos >= r.start && pos < r.end)
}
pub fn unset_region_highlight(&mut self) {
self.regions.clear();
}
pub fn free(&mut self) {
self.regions.clear();
}
}
pub fn tcout(cap: &str) { use std::sync::atomic::Ordering;
let fd = crate::ported::init::SHTTY.load(Ordering::Relaxed);
if fd >= 0 {
let _ = crate::ported::utils::write_loop(fd, cap.as_bytes());
} else {
let _ = crate::ported::utils::write_loop(1, cap.as_bytes());
}
}
pub fn tcoutarg(cap: &str, arg: i32) { use std::sync::atomic::Ordering;
let s = cap.replace("%d", &arg.to_string());
let fd = crate::ported::init::SHTTY.load(Ordering::Relaxed);
let out_fd = if fd >= 0 { fd } else { 1 };
let _ = crate::ported::utils::write_loop(out_fd, s.as_bytes()); }
pub fn tcmultout(cap: &str, count: i32) { use std::sync::atomic::Ordering;
let fd = crate::ported::init::SHTTY.load(Ordering::Relaxed);
let out_fd = if fd >= 0 { fd } else { 1 };
for _ in 0..count { let _ = crate::ported::utils::write_loop(out_fd, cap.as_bytes());
}
}
pub fn tcoutclear(to_end: bool) { use std::sync::atomic::Ordering;
let bytes: &[u8] = if to_end {
b"\x1b[J" } else {
b"\x1b[2J" };
let fd = crate::ported::init::SHTTY.load(Ordering::Relaxed);
let out_fd = if fd >= 0 { fd } else { 1 };
let _ = crate::ported::utils::write_loop(out_fd, bytes);
}
pub fn zle_refresh_boot() -> RefreshState {
RefreshState::new()
}
pub fn zle_refresh_finish(state: &mut RefreshState) {
state.free_video();
}
pub fn match_highlight(spec: &str) -> TextAttr {
let mut attr = TextAttr::default();
for token in spec.split(',') {
let token = token.trim();
if token.is_empty() {
continue;
}
match token {
"none" => {
attr = TextAttr::default();
}
"bold" => attr.bold = true,
"nobold" => attr.bold = false,
"underline" => attr.underline = true,
"nounderline" => attr.underline = false,
"standout" => attr.standout = true,
"nostandout" => attr.standout = false,
"blink" => attr.blink = true,
"noblink" => attr.blink = false,
other => {
if let Some(rest) = other.strip_prefix("fg=") {
attr.fg_color = match_colour(rest);
} else if let Some(rest) = other.strip_prefix("bg=") {
attr.bg_color = match_colour(rest);
}
}
}
}
attr
}
fn match_colour(name: &str) -> Option<u8> {
match name {
"black" => Some(0),
"red" => Some(1),
"green" => Some(2),
"yellow" => Some(3),
"blue" => Some(4),
"magenta" => Some(5),
"cyan" => Some(6),
"white" => Some(7),
"default" => None,
n => n.parse::<u8>().ok(),
}
}
pub fn zle_set_highlight(manager: &mut HighlightManager, atrs: &[&str]) {
let mut seen = std::collections::HashSet::new();
for entry in atrs {
if entry.is_empty() {
continue;
}
if *entry == "none" {
for cat in [
HC::Region,
HC::Isearch,
HC::Suffix,
HC::Paste,
HC::Default,
HC::Special,
HC::Ellipsis,
] {
manager.category_attrs.insert(cat, TextAttr::default());
seen.insert(cat);
}
continue;
}
let (prefix, rest) = match entry.split_once(':') {
Some(t) => t,
None => continue,
};
let cat = match prefix {
"region" => HC::Region,
"isearch" => HC::Isearch,
"suffix" => HC::Suffix,
"paste" => HC::Paste,
"default" => HC::Default,
"special" => HC::Special,
"ellipsis" => HC::Ellipsis,
_ => continue,
};
manager.category_attrs.insert(cat, match_highlight(rest));
seen.insert(cat);
}
let default_standout = TextAttr {
standout: true,
..TextAttr::default()
};
let default_underline = TextAttr {
underline: true,
..TextAttr::default()
};
let default_bold = TextAttr {
bold: true,
..TextAttr::default()
};
if !seen.contains(&HC::Region) {
manager.category_attrs.insert(HC::Region, default_standout);
}
if !seen.contains(&HC::Isearch) {
manager.category_attrs.insert(HC::Isearch, default_underline);
}
if !seen.contains(&HC::Suffix) {
manager.category_attrs.insert(HC::Suffix, default_bold);
}
if !seen.contains(&HC::Special) {
manager.category_attrs.insert(HC::Special, default_standout);
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_countprompt() {
let _g = crate::ported::zle::zle_main::zle_test_setup();
assert_eq!(countprompt("hello"), 5);
assert_eq!(countprompt("\x1b[31mhello\x1b[0m"), 5);
assert_eq!(countprompt("日本語"), 6); }
#[test]
fn test_video_buffer() {
let _g = crate::ported::zle::zle_main::zle_test_setup();
let mut buf = VideoBuffer::new(80, 24);
assert_eq!(buf.cols, 80);
assert_eq!(buf.rows, 24);
buf.set(0, 0, RefreshElement::new('A'));
assert_eq!(buf.get(0, 0).map(|e| e.chr), Some('A'));
buf.clear();
assert_eq!(buf.get(0, 0).map(|e| e.chr), Some(' '));
}
#[test]
fn test_refresh_state() {
let _g = crate::ported::zle::zle_main::zle_test_setup();
let mut state = RefreshState::new();
assert!(state.old_video.is_some());
assert!(state.new_video.is_some());
state.swap_buffers();
state.free_video();
assert!(state.old_video.is_none());
}
#[test]
fn compute_render_attrs_empty_buffer_yields_empty_overlay() {
let _g = crate::ported::zle::zle_main::zle_test_setup();
assert!(compute_render_attrs().is_empty());
}
#[test]
fn compute_render_attrs_visual_mode_paints_mark_to_cursor_in_standout() {
let _g = crate::ported::zle::zle_main::zle_test_setup();
*crate::ported::zle::zle_main::ZLELINE.lock().unwrap() = "hello world".chars().collect();
crate::ported::zle::zle_main::ZLELL.store(crate::ported::zle::zle_main::ZLELINE.lock().unwrap().len(), std::sync::atomic::Ordering::SeqCst);
crate::ported::zle::zle_main::MARK.store(2, std::sync::atomic::Ordering::SeqCst);
crate::ported::zle::zle_main::ZLECS.store(7, std::sync::atomic::Ordering::SeqCst);
crate::ported::zle::zle_main::REGION_ACTIVE.store(1, std::sync::atomic::Ordering::SeqCst); let attrs = compute_render_attrs();
assert_eq!(attrs.len(), 11);
for slot in attrs.iter().take(2) {
assert!(slot.is_none());
}
for slot in attrs.iter().skip(7) {
assert!(slot.is_none());
}
for slot in attrs.iter().take(7).skip(2) {
let attr = slot.expect("standout");
assert!(attr.standout);
}
}
#[test]
fn compute_render_attrs_visual_mode_handles_reverse_mark_order() {
let _g = crate::ported::zle::zle_main::zle_test_setup();
*crate::ported::zle::zle_main::ZLELINE.lock().unwrap() = "abcdef".chars().collect();
crate::ported::zle::zle_main::ZLELL.store(6, std::sync::atomic::Ordering::SeqCst);
crate::ported::zle::zle_main::MARK.store(5, std::sync::atomic::Ordering::SeqCst);
crate::ported::zle::zle_main::ZLECS.store(1, std::sync::atomic::Ordering::SeqCst);
crate::ported::zle::zle_main::REGION_ACTIVE.store(2, std::sync::atomic::Ordering::SeqCst); let attrs = compute_render_attrs();
assert!(attrs[0].is_none());
for slot in attrs.iter().take(5).skip(1) {
assert!(slot.unwrap().standout);
}
assert!(attrs[5].is_none());
}
#[test]
fn match_highlight_handles_combined_attrs() {
let _g = crate::ported::zle::zle_main::zle_test_setup();
let attr = match_highlight("bold,fg=red,underline");
assert!(attr.bold);
assert!(attr.underline);
assert_eq!(attr.fg_color, Some(1));
}
#[test]
fn match_highlight_named_and_numeric_colors() {
let _g = crate::ported::zle::zle_main::zle_test_setup();
assert_eq!(match_highlight("fg=cyan").fg_color, Some(6));
assert_eq!(match_highlight("bg=42").bg_color, Some(42));
assert_eq!(match_highlight("fg=999").fg_color, None);
}
#[test]
fn match_highlight_negation_clears_attr() {
let _g = crate::ported::zle::zle_main::zle_test_setup();
let attr = match_highlight("bold,nobold,underline");
assert!(!attr.bold);
assert!(attr.underline);
}
#[test]
fn match_highlight_none_resets_everything() {
let _g = crate::ported::zle::zle_main::zle_test_setup();
let attr = match_highlight("bold,fg=red,none,underline");
assert!(!attr.bold);
assert!(attr.underline);
assert_eq!(attr.fg_color, None);
}
#[test]
fn zle_set_highlight_populates_categories_and_defaults() {
let _g = crate::ported::zle::zle_main::zle_test_setup();
let mut mgr = HighlightManager::new();
let entries = ["region:fg=red,bold", "isearch:fg=blue"];
zle_set_highlight(&mut mgr, &entries);
let region = mgr.category_attrs[&HighlightCategory::Region];
assert!(region.bold);
assert_eq!(region.fg_color, Some(1));
let isearch = mgr.category_attrs[&HighlightCategory::Isearch];
assert_eq!(isearch.fg_color, Some(4));
let suffix = mgr.category_attrs[&HighlightCategory::Suffix];
assert!(suffix.bold);
let special = mgr.category_attrs[&HighlightCategory::Special];
assert!(special.standout);
}
#[test]
fn zle_set_highlight_none_clears_every_slot() {
let _g = crate::ported::zle::zle_main::zle_test_setup();
let mut mgr = HighlightManager::new();
zle_set_highlight(&mut mgr, &["none"]);
for cat in [
HighlightCategory::Region,
HighlightCategory::Isearch,
HighlightCategory::Suffix,
HighlightCategory::Paste,
] {
let attr = mgr.category_attrs[&cat];
assert_eq!(attr, TextAttr::default());
}
}
#[test]
fn compute_render_attrs_visual_uses_zle_highlight_region_attr() {
let _g = crate::ported::zle::zle_main::zle_test_setup();
crate::ported::zle::zle_main::zle_reset();
*crate::ported::zle::zle_main::ZLELINE.lock().unwrap() = "abcde".chars().collect();
crate::ported::zle::zle_main::ZLELL.store(5, std::sync::atomic::Ordering::SeqCst);
crate::ported::zle::zle_main::MARK.store(1, std::sync::atomic::Ordering::SeqCst);
crate::ported::zle::zle_main::ZLECS.store(4, std::sync::atomic::Ordering::SeqCst);
crate::ported::zle::zle_main::REGION_ACTIVE.store(1, std::sync::atomic::Ordering::SeqCst);
zle_set_highlight(&mut crate::ported::zle::zle_main::highlight().lock().unwrap(), &["region:fg=red,bold"]);
let attrs = compute_render_attrs();
for slot in attrs.iter().take(4).skip(1) {
let a = slot.expect("region painted");
assert!(a.bold);
assert_eq!(a.fg_color, Some(1));
assert!(!a.standout);
}
}
#[test]
fn compute_render_attrs_explicit_regions_override_default() {
let _g = crate::ported::zle::zle_main::zle_test_setup();
*crate::ported::zle::zle_main::ZLELINE.lock().unwrap() = "abcde".chars().collect();
crate::ported::zle::zle_main::ZLELL.store(5, std::sync::atomic::Ordering::SeqCst);
let custom = TextAttr {
bold: true,
fg_color: Some(1),
..TextAttr::default()
};
crate::ported::zle::zle_main::highlight().lock().unwrap()
.set_region_highlight(1, 4, custom);
let attrs = compute_render_attrs();
assert!(attrs[0].is_none());
for slot in attrs.iter().take(4).skip(1) {
let a = slot.expect("custom");
assert!(a.bold);
assert_eq!(a.fg_color, Some(1));
}
assert!(attrs[4].is_none());
}
}
pub fn addmultiword(base: &mut crate::ported::zle::zle_h::REFRESH_ELEMENT, _tptr: &[char], _ichars: usize) {
base.atr |= TXT_MULTIWORD_MASK;
}
pub fn bufswap(state: &mut RefreshState) { std::mem::swap(&mut state.old_video, &mut state.new_video);
}
pub fn freevideo(state: &mut RefreshState) { state.old_video = None;
state.new_video = None;
}
#[allow(unused_variables)]
pub fn nextline(rpms: &mut RefreshState, wrapped: i32) -> i32 { rpms.vln += 1;
if rpms.vln >= rpms.lines {
return 1; }
rpms.vcs = 0;
0
}
pub fn resetvideo(state: &mut RefreshState) { let cols = crate::ported::utils::adjustcolumns();
let rows = crate::ported::utils::adjustlines();
state.columns = cols;
state.lines = rows;
state.old_video = Some(VideoBuffer::new(cols, rows));
state.new_video = Some(VideoBuffer::new(cols, rows));
state.need_full_redraw = true;
}
pub fn singmoveto(state: &mut RefreshState, pos: usize) { state.vcs = pos;
}
pub fn snextline(rpms: &mut RefreshState) -> i32 { if rpms.vln > 0 {
rpms.vln -= 1;
}
rpms.vcs = 0;
0
}
pub fn tcout_via_func(_cap: i32, _arg: i32) -> i32 { 1
}
pub fn wpfxlen(olds: &[crate::ported::zle::zle_h::REFRESH_ELEMENT],
news: &[crate::ported::zle::zle_h::REFRESH_ELEMENT]) -> usize {
let mut i = 0;
while i < olds.len() && i < news.len()
&& olds[i].chr != '\0' && olds[i] == news[i]
{
i += 1;
}
i
}
pub fn zle_free_highlight() { }
#[allow(non_snake_case)]
pub fn ZR_memset( dst: &mut [crate::ported::zle::zle_h::REFRESH_ELEMENT],
rc: crate::ported::zle::zle_h::REFRESH_ELEMENT,
len: usize,
) {
let n = len.min(dst.len());
for slot in dst.iter_mut().take(n) { *slot = rc;
}
}
#[inline]
#[allow(non_snake_case)]
pub fn ZR_equal( a: crate::ported::zle::zle_h::REFRESH_ELEMENT,
b: crate::ported::zle::zle_h::REFRESH_ELEMENT,
) -> bool {
a == b
}
#[inline]
#[allow(non_snake_case)]
pub fn ZR_memcpy( dst: &mut [crate::ported::zle::zle_h::REFRESH_ELEMENT],
src: &[crate::ported::zle::zle_h::REFRESH_ELEMENT],
l: usize,
) {
dst[..l].copy_from_slice(&src[..l]);
}
#[allow(non_snake_case)]
pub fn ZR_strcpy( dst: &mut [crate::ported::zle::zle_h::REFRESH_ELEMENT],
src: &[crate::ported::zle::zle_h::REFRESH_ELEMENT],
) {
let mut i = 0;
loop { if i >= dst.len() || i >= src.len() {
break;
}
dst[i] = src[i];
if src[i].chr == '\0' {
break;
}
i += 1;
}
}
#[allow(non_snake_case)]
pub fn ZR_strlen(wstr: &[crate::ported::zle::zle_h::REFRESH_ELEMENT]) -> usize { let mut len = 0; while len < wstr.len() && wstr[len].chr != '\0' { len += 1; }
len }
#[allow(non_snake_case)]
pub fn ZR_strncmp( oldwstr: &[crate::ported::zle::zle_h::REFRESH_ELEMENT],
newwstr: &[crate::ported::zle::zle_h::REFRESH_ELEMENT],
len: usize,
) -> i32 {
let mut i = 0;
while i < len { if i >= oldwstr.len() || i >= newwstr.len() {
return if oldwstr.get(i) == newwstr.get(i) { 0 } else { 1 };
}
let o = oldwstr[i];
let n = newwstr[i];
let old_is_nul = (o.atr & TXT_MULTIWORD_MASK) == 0 && o.chr == '\0';
let new_is_nul = (n.atr & TXT_MULTIWORD_MASK) == 0 && n.chr == '\0';
if old_is_nul || new_is_nul {
return if o == n { 0 } else { 1 }; }
if o != n { return 1;
}
i += 1; }
0 }
pub const DEF_MWBUF_ALLOC: usize = 32;
pub static ZR_END_ELLIPSIS: &[crate::ported::zle::zle_h::REFRESH_ELEMENT] = &[ crate::ported::zle::zle_h::REFRESH_ELEMENT { chr: ' ', atr: 0 },
crate::ported::zle::zle_h::REFRESH_ELEMENT { chr: '.', atr: crate::ported::zsh_h::TXT_ERROR },
crate::ported::zle::zle_h::REFRESH_ELEMENT { chr: '.', atr: crate::ported::zsh_h::TXT_ERROR },
crate::ported::zle::zle_h::REFRESH_ELEMENT { chr: '.', atr: crate::ported::zsh_h::TXT_ERROR },
crate::ported::zle::zle_h::REFRESH_ELEMENT { chr: '.', atr: crate::ported::zsh_h::TXT_ERROR },
crate::ported::zle::zle_h::REFRESH_ELEMENT { chr: '>', atr: 0 },
];
pub const ZR_END_ELLIPSIS_SIZE: usize = ZR_END_ELLIPSIS.len();
pub static ZR_MID_ELLIPSIS1: &[crate::ported::zle::zle_h::REFRESH_ELEMENT] = &[ crate::ported::zle::zle_h::REFRESH_ELEMENT { chr: ' ', atr: 0 },
crate::ported::zle::zle_h::REFRESH_ELEMENT { chr: '<', atr: 0 },
crate::ported::zle::zle_h::REFRESH_ELEMENT { chr: '.', atr: crate::ported::zsh_h::TXT_ERROR },
crate::ported::zle::zle_h::REFRESH_ELEMENT { chr: '.', atr: crate::ported::zsh_h::TXT_ERROR },
crate::ported::zle::zle_h::REFRESH_ELEMENT { chr: '.', atr: crate::ported::zsh_h::TXT_ERROR },
crate::ported::zle::zle_h::REFRESH_ELEMENT { chr: '.', atr: crate::ported::zsh_h::TXT_ERROR },
];
pub const ZR_MID_ELLIPSIS1_SIZE: usize = ZR_MID_ELLIPSIS1.len();
pub static ZR_MID_ELLIPSIS2: &[crate::ported::zle::zle_h::REFRESH_ELEMENT] = &[ crate::ported::zle::zle_h::REFRESH_ELEMENT { chr: '>', atr: crate::ported::zsh_h::TXT_ERROR },
crate::ported::zle::zle_h::REFRESH_ELEMENT { chr: ' ', atr: 0 },
];
pub const ZR_MID_ELLIPSIS2_SIZE: usize = ZR_MID_ELLIPSIS2.len();
pub static ZR_START_ELLIPSIS: &[crate::ported::zle::zle_h::REFRESH_ELEMENT] = &[ crate::ported::zle::zle_h::REFRESH_ELEMENT { chr: '>', atr: 0 },
crate::ported::zle::zle_h::REFRESH_ELEMENT { chr: '.', atr: crate::ported::zsh_h::TXT_ERROR },
crate::ported::zle::zle_h::REFRESH_ELEMENT { chr: '.', atr: crate::ported::zsh_h::TXT_ERROR },
crate::ported::zle::zle_h::REFRESH_ELEMENT { chr: '.', atr: crate::ported::zsh_h::TXT_ERROR },
crate::ported::zle::zle_h::REFRESH_ELEMENT { chr: '.', atr: crate::ported::zsh_h::TXT_ERROR },
];
pub const ZR_START_ELLIPSIS_SIZE: usize = ZR_START_ELLIPSIS.len();
#[inline] pub fn tcinscost(x: i32) -> i32 { x.max(0)
}
#[inline] pub fn tcdelcost(x: i32) -> i32 { x.max(0)
}
#[inline] pub fn tc_delchars(_x: i32) { }
#[inline] pub fn tc_inschars(_x: i32) { }
#[inline] pub fn tc_upcurs(_x: i32) { }
#[inline] pub fn tc_leftcurs(_x: i32) { }
pub static TCOUT_FUNC_NAME: std::sync::Mutex<Option<String>> = std::sync::Mutex::new(None);
pub static CLEAREOL: std::sync::atomic::AtomicI32 =
std::sync::atomic::AtomicI32::new(0);
pub static CLEARF: std::sync::atomic::AtomicI32 =
std::sync::atomic::AtomicI32::new(0);
pub static PUT_RPMPT: std::sync::atomic::AtomicI32 =
std::sync::atomic::AtomicI32::new(0);
pub static OPUT_RPMPT: std::sync::atomic::AtomicI32 =
std::sync::atomic::AtomicI32::new(0);
pub static OXTABS: std::sync::atomic::AtomicI32 =
std::sync::atomic::AtomicI32::new(0);
pub static NUMSCROLLS: std::sync::atomic::AtomicI32 =
std::sync::atomic::AtomicI32::new(0);
pub static ONUMSCROLLS: std::sync::atomic::AtomicI32 =
std::sync::atomic::AtomicI32::new(0);
pub static NLNCT: std::sync::atomic::AtomicI32 =
std::sync::atomic::AtomicI32::new(0);
pub static SHOWINGLIST: std::sync::atomic::AtomicI32 =
std::sync::atomic::AtomicI32::new(0);
pub static LISTSHOWN: std::sync::atomic::AtomicI32 =
std::sync::atomic::AtomicI32::new(0);
pub static LASTLISTLEN: std::sync::atomic::AtomicI32 =
std::sync::atomic::AtomicI32::new(0);
pub static CLEARFLAG: std::sync::atomic::AtomicI32 =
std::sync::atomic::AtomicI32::new(0);
pub static CLEARLIST: std::sync::atomic::AtomicI32 =
std::sync::atomic::AtomicI32::new(0);
#[derive(Debug, Clone, Default)]
#[allow(non_camel_case_types)]
pub struct rparams { pub canscroll: i32, pub ln: i32, pub more_status: i32, pub nvcs: i32, pub nvln: i32, pub tosln: i32, pub pos: usize, pub end: usize, }
#[cfg(test)]
mod zr_tests {
use super::*;
use crate::ported::zle::zle_h::REFRESH_ELEMENT;
use crate::ported::zsh_h::{TXT_MULTIWORD_MASK, TXTBOLDFACE};
fn re(c: char, a: u64) -> REFRESH_ELEMENT {
REFRESH_ELEMENT { chr: c, atr: a }
}
#[test]
fn zr_memset_fills_slice() {
let _g = crate::ported::zle::zle_main::zle_test_setup();
let mut buf = [REFRESH_ELEMENT::default(); 4];
let fill = re('x', 0);
ZR_memset(&mut buf, fill, 3);
assert_eq!(buf[0], fill);
assert_eq!(buf[1], fill);
assert_eq!(buf[2], fill);
assert_eq!(buf[3], REFRESH_ELEMENT::default());
}
#[test]
fn zr_memset_clamps_to_dst_len() {
let _g = crate::ported::zle::zle_main::zle_test_setup();
let mut buf = [REFRESH_ELEMENT::default(); 2];
let fill = re('y', 0);
ZR_memset(&mut buf, fill, 99); assert_eq!(buf[0], fill);
assert_eq!(buf[1], fill);
}
#[test]
fn zr_strlen_counts_to_nul() {
let _g = crate::ported::zle::zle_main::zle_test_setup();
let s = [re('h', 0), re('i', 0), re('\0', 0)];
assert_eq!(ZR_strlen(&s), 2);
}
#[test]
fn zr_strlen_empty_starts_with_nul() {
let _g = crate::ported::zle::zle_main::zle_test_setup();
let s = [re('\0', 0)];
assert_eq!(ZR_strlen(&s), 0);
}
#[test]
fn zr_strcpy_copies_through_nul() {
let _g = crate::ported::zle::zle_main::zle_test_setup();
let src = [re('a', 0), re('b', 0), re('\0', 0)];
let mut dst = [REFRESH_ELEMENT::default(); 5];
ZR_strcpy(&mut dst, &src);
assert_eq!(dst[0], re('a', 0));
assert_eq!(dst[1], re('b', 0));
assert_eq!(dst[2], re('\0', 0));
}
#[test]
fn zr_strncmp_equal_strings() {
let _g = crate::ported::zle::zle_main::zle_test_setup();
let a = [re('h', 0), re('i', 0)];
let b = [re('h', 0), re('i', 0)];
assert_eq!(ZR_strncmp(&a, &b, 2), 0);
}
#[test]
fn zr_strncmp_diff_chr_returns_1() {
let _g = crate::ported::zle::zle_main::zle_test_setup();
let a = [re('h', 0), re('i', 0)];
let b = [re('h', 0), re('o', 0)];
assert_eq!(ZR_strncmp(&a, &b, 2), 1);
}
#[test]
fn zr_strncmp_diff_atr_returns_1() {
let _g = crate::ported::zle::zle_main::zle_test_setup();
let a = [re('h', 0)];
let b = [re('h', TXTBOLDFACE)];
assert_eq!(ZR_strncmp(&a, &b, 1), 1);
}
#[test]
fn zr_strncmp_early_nul_old() {
let _g = crate::ported::zle::zle_main::zle_test_setup();
let a = [re('\0', 0)];
let b = [re('x', 0)];
assert_eq!(ZR_strncmp(&a, &b, 1), 1); let a = [re('\0', 0)];
let b = [re('\0', 0)];
assert_eq!(ZR_strncmp(&a, &b, 1), 0); }
#[test]
fn zr_strncmp_multiword_mask_skips_nul_check() {
let _g = crate::ported::zle::zle_main::zle_test_setup();
let a = [re('\0', TXT_MULTIWORD_MASK)];
let b = [re('\0', TXT_MULTIWORD_MASK)];
assert_eq!(ZR_strncmp(&a, &b, 1), 0);
}
#[test]
fn zr_equal_same_returns_true() {
let _g = crate::ported::zle::zle_main::zle_test_setup();
let a = re('a', 0);
assert!(ZR_equal(a, a));
let b = re('b', 0);
assert!(!ZR_equal(a, b));
}
#[test]
fn zr_memcpy_copies_n_elements() {
let _g = crate::ported::zle::zle_main::zle_test_setup();
let mut dst = [re('\0', 0); 5];
let src = [re('a', 0), re('b', 0), re('c', 0), re('d', 0), re('e', 0)];
ZR_memcpy(&mut dst, &src, 3);
assert_eq!(dst[0].chr, 'a');
assert_eq!(dst[1].chr, 'b');
assert_eq!(dst[2].chr, 'c');
assert_eq!(dst[3].chr, '\0');
}
#[test]
fn ellipsis_sizes_match_table_lengths() {
let _g = crate::ported::zle::zle_main::zle_test_setup();
assert_eq!(ZR_END_ELLIPSIS_SIZE, 6);
assert_eq!(ZR_MID_ELLIPSIS1_SIZE, 6);
assert_eq!(ZR_MID_ELLIPSIS2_SIZE, 2);
assert_eq!(ZR_START_ELLIPSIS_SIZE, 5);
}
#[test]
fn def_mwbuf_alloc_is_32() {
let _g = crate::ported::zle::zle_main::zle_test_setup();
assert_eq!(DEF_MWBUF_ALLOC, 32);
}
#[test]
fn tc_costs_handle_negative() {
let _g = crate::ported::zle::zle_main::zle_test_setup();
assert_eq!(tcinscost(-1), 0);
assert_eq!(tcdelcost(-1), 0);
assert_eq!(tcinscost(5), 5);
assert_eq!(tcdelcost(5), 5);
}
#[test]
fn rparams_default_zeros_all_fields() {
let _g = crate::ported::zle::zle_main::zle_test_setup();
let r = rparams::default();
assert_eq!(r.canscroll, 0);
assert_eq!(r.ln, 0);
assert_eq!(r.more_status, 0);
assert_eq!(r.nvcs, 0);
assert_eq!(r.nvln, 0);
assert_eq!(r.tosln, 0);
assert_eq!(r.pos, 0);
assert_eq!(r.end, 0);
}
}