use std::convert::TryFrom;
use std::fmt::Write;
use std::time::{Duration, Instant};
use std::{iter, str};
use log::{debug, trace};
use serde::{Deserialize, Serialize};
use vte::{Params, ParamsIter};
use alacritty_config_derive::ConfigDeserialize;
use crate::index::{Column, Line};
use crate::term::cell::Hyperlink;
use crate::term::color::Rgb;
const SYNC_UPDATE_TIMEOUT: Duration = Duration::from_millis(150);
const SYNC_BUFFER_SIZE: usize = 0x20_0000;
const SYNC_ESCAPE_START_LEN: usize = 5;
const SYNC_START_ESCAPE_START: [u8; SYNC_ESCAPE_START_LEN] = [b'\x1b', b'P', b'=', b'1', b's'];
const SYNC_END_ESCAPE_START: [u8; SYNC_ESCAPE_START_LEN] = [b'\x1b', b'P', b'=', b'2', b's'];
fn xparse_color(color: &[u8]) -> Option<Rgb> {
if !color.is_empty() && color[0] == b'#' {
parse_legacy_color(&color[1..])
} else if color.len() >= 4 && &color[..4] == b"rgb:" {
parse_rgb_color(&color[4..])
} else {
None
}
}
fn parse_rgb_color(color: &[u8]) -> Option<Rgb> {
let colors = str::from_utf8(color).ok()?.split('/').collect::<Vec<_>>();
if colors.len() != 3 {
return None;
}
let scale = |input: &str| {
if input.len() > 4 {
None
} else {
let max = u32::pow(16, input.len() as u32) - 1;
let value = u32::from_str_radix(input, 16).ok()?;
Some((255 * value / max) as u8)
}
};
Some(Rgb { r: scale(colors[0])?, g: scale(colors[1])?, b: scale(colors[2])? })
}
fn parse_legacy_color(color: &[u8]) -> Option<Rgb> {
let item_len = color.len() / 3;
let color_from_slice = |slice: &[u8]| {
let col = usize::from_str_radix(str::from_utf8(slice).ok()?, 16).ok()? << 4;
Some((col >> (4 * slice.len().saturating_sub(1))) as u8)
};
Some(Rgb {
r: color_from_slice(&color[0..item_len])?,
g: color_from_slice(&color[item_len..item_len * 2])?,
b: color_from_slice(&color[item_len * 2..])?,
})
}
fn parse_number(input: &[u8]) -> Option<u8> {
if input.is_empty() {
return None;
}
let mut num: u8 = 0;
for c in input {
let c = *c as char;
if let Some(digit) = c.to_digit(10) {
num = match num.checked_mul(10).and_then(|v| v.checked_add(digit as u8)) {
Some(v) => v,
None => return None,
}
} else {
return None;
}
}
Some(num)
}
#[derive(Debug, Default)]
struct ProcessorState {
preceding_char: Option<char>,
dcs: Option<Dcs>,
sync_state: SyncState,
}
#[derive(Debug)]
struct SyncState {
timeout: Option<Instant>,
pending_dcs: Option<Dcs>,
buffer: Vec<u8>,
}
impl Default for SyncState {
fn default() -> Self {
Self { buffer: Vec::with_capacity(SYNC_BUFFER_SIZE), pending_dcs: None, timeout: None }
}
}
#[derive(Debug)]
enum Dcs {
SyncStart,
SyncEnd,
}
#[derive(Default)]
pub struct Processor {
state: ProcessorState,
parser: vte::Parser,
}
impl Processor {
#[inline]
pub fn new() -> Self {
Self::default()
}
#[inline]
pub fn advance<H>(&mut self, handler: &mut H, byte: u8)
where
H: Handler,
{
if self.state.sync_state.timeout.is_none() {
let mut performer = Performer::new(&mut self.state, handler);
self.parser.advance(&mut performer, byte);
} else {
self.advance_sync(handler, byte);
}
}
pub fn stop_sync<H>(&mut self, handler: &mut H)
where
H: Handler,
{
for i in 0..self.state.sync_state.buffer.len() {
let byte = self.state.sync_state.buffer[i];
let mut performer = Performer::new(&mut self.state, handler);
self.parser.advance(&mut performer, byte);
}
self.state.sync_state.buffer.clear();
self.state.sync_state.timeout = None;
}
#[inline]
pub fn sync_timeout(&self) -> Option<&Instant> {
self.state.sync_state.timeout.as_ref()
}
#[inline]
pub fn sync_bytes_count(&self) -> usize {
self.state.sync_state.buffer.len()
}
#[cold]
fn advance_sync<H>(&mut self, handler: &mut H, byte: u8)
where
H: Handler,
{
self.state.sync_state.buffer.push(byte);
match self.state.sync_state.pending_dcs {
Some(_) => self.advance_sync_dcs_end(handler, byte),
None => self.advance_sync_dcs_start(),
}
}
fn advance_sync_dcs_start(&mut self) {
let len = self.state.sync_state.buffer.len();
let offset = len.saturating_sub(SYNC_ESCAPE_START_LEN);
let end = &self.state.sync_state.buffer[offset..];
if end == SYNC_START_ESCAPE_START {
self.state.sync_state.pending_dcs = Some(Dcs::SyncStart);
} else if end == SYNC_END_ESCAPE_START || len >= SYNC_BUFFER_SIZE - 1 {
self.state.sync_state.pending_dcs = Some(Dcs::SyncEnd);
}
}
fn advance_sync_dcs_end<H>(&mut self, handler: &mut H, byte: u8)
where
H: Handler,
{
match byte {
0x00..=0x17 | 0x19 | 0x1c..=0x7f | 0xa0..=0xff => (),
0x18 | 0x1a | 0x80..=0x9f => self.state.sync_state.pending_dcs = None,
0x1b => match self.state.sync_state.pending_dcs.take() {
Some(Dcs::SyncStart) => {
self.state.sync_state.timeout = Some(Instant::now() + SYNC_UPDATE_TIMEOUT);
},
Some(Dcs::SyncEnd) => self.stop_sync(handler),
None => (),
},
}
}
}
struct Performer<'a, H: Handler> {
state: &'a mut ProcessorState,
handler: &'a mut H,
}
impl<'a, H: Handler + 'a> Performer<'a, H> {
#[inline]
pub fn new<'b>(state: &'b mut ProcessorState, handler: &'b mut H) -> Performer<'b, H> {
Performer { state, handler }
}
}
pub trait Handler {
fn set_title(&mut self, _: Option<String>) {}
fn set_cursor_style(&mut self, _: Option<CursorStyle>) {}
fn set_cursor_shape(&mut self, _shape: CursorShape) {}
fn input(&mut self, _c: char) {}
fn goto(&mut self, _: Line, _: Column) {}
fn goto_line(&mut self, _: Line) {}
fn goto_col(&mut self, _: Column) {}
fn insert_blank(&mut self, _: usize) {}
fn move_up(&mut self, _: usize) {}
fn move_down(&mut self, _: usize) {}
fn identify_terminal(&mut self, _intermediate: Option<char>) {}
fn device_status(&mut self, _: usize) {}
fn move_forward(&mut self, _: Column) {}
fn move_backward(&mut self, _: Column) {}
fn move_down_and_cr(&mut self, _: usize) {}
fn move_up_and_cr(&mut self, _: usize) {}
fn put_tab(&mut self, _count: u16) {}
fn backspace(&mut self) {}
fn carriage_return(&mut self) {}
fn linefeed(&mut self) {}
fn bell(&mut self) {}
fn substitute(&mut self) {}
fn newline(&mut self) {}
fn set_horizontal_tabstop(&mut self) {}
fn scroll_up(&mut self, _: usize) {}
fn scroll_down(&mut self, _: usize) {}
fn insert_blank_lines(&mut self, _: usize) {}
fn delete_lines(&mut self, _: usize) {}
fn erase_chars(&mut self, _: Column) {}
fn delete_chars(&mut self, _: usize) {}
fn move_backward_tabs(&mut self, _count: u16) {}
fn move_forward_tabs(&mut self, _count: u16) {}
fn save_cursor_position(&mut self) {}
fn restore_cursor_position(&mut self) {}
fn clear_line(&mut self, _mode: LineClearMode) {}
fn clear_screen(&mut self, _mode: ClearMode) {}
fn clear_tabs(&mut self, _mode: TabulationClearMode) {}
fn reset_state(&mut self) {}
fn reverse_index(&mut self) {}
fn terminal_attribute(&mut self, _attr: Attr) {}
fn set_mode(&mut self, _mode: Mode) {}
fn unset_mode(&mut self, _: Mode) {}
fn set_scrolling_region(&mut self, _top: usize, _bottom: Option<usize>) {}
fn set_keypad_application_mode(&mut self) {}
fn unset_keypad_application_mode(&mut self) {}
fn set_active_charset(&mut self, _: CharsetIndex) {}
fn configure_charset(&mut self, _: CharsetIndex, _: StandardCharset) {}
fn set_color(&mut self, _: usize, _: Rgb) {}
fn dynamic_color_sequence(&mut self, _: String, _: usize, _: &str) {}
fn reset_color(&mut self, _: usize) {}
fn clipboard_store(&mut self, _: u8, _: &[u8]) {}
fn clipboard_load(&mut self, _: u8, _: &str) {}
fn decaln(&mut self) {}
fn push_title(&mut self) {}
fn pop_title(&mut self) {}
fn text_area_size_pixels(&mut self) {}
fn text_area_size_chars(&mut self) {}
fn set_hyperlink(&mut self, _: Option<Hyperlink>) {}
}
#[derive(ConfigDeserialize, Default, Debug, Eq, PartialEq, Copy, Clone, Hash)]
pub struct CursorStyle {
pub shape: CursorShape,
pub blinking: bool,
}
#[derive(ConfigDeserialize, Debug, Eq, PartialEq, Copy, Clone, Hash)]
pub enum CursorShape {
Block,
Underline,
Beam,
#[config(skip)]
HollowBlock,
#[config(skip)]
Hidden,
}
impl Default for CursorShape {
fn default() -> CursorShape {
CursorShape::Block
}
}
#[derive(Debug, Eq, PartialEq)]
pub enum Mode {
CursorKeys = 1,
ColumnMode = 3,
Insert = 4,
Origin = 6,
LineWrap = 7,
BlinkingCursor = 12,
LineFeedNewLine = 20,
ShowCursor = 25,
ReportMouseClicks = 1000,
ReportCellMouseMotion = 1002,
ReportAllMouseMotion = 1003,
ReportFocusInOut = 1004,
Utf8Mouse = 1005,
SgrMouse = 1006,
AlternateScroll = 1007,
UrgencyHints = 1042,
SwapScreenAndSetRestoreCursor = 1049,
BracketedPaste = 2004,
}
impl Mode {
pub fn from_primitive(intermediate: Option<&u8>, num: u16) -> Option<Mode> {
let private = match intermediate {
Some(b'?') => true,
None => false,
_ => return None,
};
if private {
Some(match num {
1 => Mode::CursorKeys,
3 => Mode::ColumnMode,
6 => Mode::Origin,
7 => Mode::LineWrap,
12 => Mode::BlinkingCursor,
25 => Mode::ShowCursor,
1000 => Mode::ReportMouseClicks,
1002 => Mode::ReportCellMouseMotion,
1003 => Mode::ReportAllMouseMotion,
1004 => Mode::ReportFocusInOut,
1005 => Mode::Utf8Mouse,
1006 => Mode::SgrMouse,
1007 => Mode::AlternateScroll,
1042 => Mode::UrgencyHints,
1049 => Mode::SwapScreenAndSetRestoreCursor,
2004 => Mode::BracketedPaste,
_ => {
trace!("[unimplemented] primitive mode: {}", num);
return None;
},
})
} else {
Some(match num {
4 => Mode::Insert,
20 => Mode::LineFeedNewLine,
_ => return None,
})
}
}
}
#[derive(Debug)]
pub enum LineClearMode {
Right,
Left,
All,
}
#[derive(Debug)]
pub enum ClearMode {
Below,
Above,
All,
Saved,
}
#[derive(Debug)]
pub enum TabulationClearMode {
Current,
All,
}
#[derive(Serialize, Deserialize, Debug, Copy, Clone, Eq, PartialEq, PartialOrd, Ord)]
pub enum NamedColor {
Black = 0,
Red,
Green,
Yellow,
Blue,
Magenta,
Cyan,
White,
BrightBlack,
BrightRed,
BrightGreen,
BrightYellow,
BrightBlue,
BrightMagenta,
BrightCyan,
BrightWhite,
Foreground = 256,
Background,
Cursor,
DimBlack,
DimRed,
DimGreen,
DimYellow,
DimBlue,
DimMagenta,
DimCyan,
DimWhite,
BrightForeground,
DimForeground,
}
impl NamedColor {
#[must_use]
pub fn to_bright(self) -> Self {
match self {
NamedColor::Foreground => NamedColor::BrightForeground,
NamedColor::Black => NamedColor::BrightBlack,
NamedColor::Red => NamedColor::BrightRed,
NamedColor::Green => NamedColor::BrightGreen,
NamedColor::Yellow => NamedColor::BrightYellow,
NamedColor::Blue => NamedColor::BrightBlue,
NamedColor::Magenta => NamedColor::BrightMagenta,
NamedColor::Cyan => NamedColor::BrightCyan,
NamedColor::White => NamedColor::BrightWhite,
NamedColor::DimForeground => NamedColor::Foreground,
NamedColor::DimBlack => NamedColor::Black,
NamedColor::DimRed => NamedColor::Red,
NamedColor::DimGreen => NamedColor::Green,
NamedColor::DimYellow => NamedColor::Yellow,
NamedColor::DimBlue => NamedColor::Blue,
NamedColor::DimMagenta => NamedColor::Magenta,
NamedColor::DimCyan => NamedColor::Cyan,
NamedColor::DimWhite => NamedColor::White,
val => val,
}
}
#[must_use]
pub fn to_dim(self) -> Self {
match self {
NamedColor::Black => NamedColor::DimBlack,
NamedColor::Red => NamedColor::DimRed,
NamedColor::Green => NamedColor::DimGreen,
NamedColor::Yellow => NamedColor::DimYellow,
NamedColor::Blue => NamedColor::DimBlue,
NamedColor::Magenta => NamedColor::DimMagenta,
NamedColor::Cyan => NamedColor::DimCyan,
NamedColor::White => NamedColor::DimWhite,
NamedColor::Foreground => NamedColor::DimForeground,
NamedColor::BrightBlack => NamedColor::Black,
NamedColor::BrightRed => NamedColor::Red,
NamedColor::BrightGreen => NamedColor::Green,
NamedColor::BrightYellow => NamedColor::Yellow,
NamedColor::BrightBlue => NamedColor::Blue,
NamedColor::BrightMagenta => NamedColor::Magenta,
NamedColor::BrightCyan => NamedColor::Cyan,
NamedColor::BrightWhite => NamedColor::White,
NamedColor::BrightForeground => NamedColor::Foreground,
val => val,
}
}
}
#[derive(Serialize, Deserialize, Debug, Clone, Copy, PartialEq, Eq)]
pub enum Color {
Named(NamedColor),
Spec(Rgb),
Indexed(u8),
}
#[derive(Debug, Eq, PartialEq)]
pub enum Attr {
Reset,
Bold,
Dim,
Italic,
Underline,
DoubleUnderline,
Undercurl,
DottedUnderline,
DashedUnderline,
BlinkSlow,
BlinkFast,
Reverse,
Hidden,
Strike,
CancelBold,
CancelBoldDim,
CancelItalic,
CancelUnderline,
CancelBlink,
CancelReverse,
CancelHidden,
CancelStrike,
Foreground(Color),
Background(Color),
UnderlineColor(Option<Color>),
}
#[derive(Clone, Copy, Debug, Eq, PartialEq)]
pub enum CharsetIndex {
G0,
G1,
G2,
G3,
}
impl Default for CharsetIndex {
fn default() -> Self {
CharsetIndex::G0
}
}
#[derive(Clone, Copy, Debug, Eq, PartialEq)]
pub enum StandardCharset {
Ascii,
SpecialCharacterAndLineDrawing,
}
impl Default for StandardCharset {
fn default() -> Self {
StandardCharset::Ascii
}
}
impl StandardCharset {
#[inline]
pub fn map(self, c: char) -> char {
match self {
StandardCharset::Ascii => c,
StandardCharset::SpecialCharacterAndLineDrawing => match c {
'_' => ' ',
'`' => '◆',
'a' => '▒',
'b' => '\u{2409}', 'c' => '\u{240c}', 'd' => '\u{240d}', 'e' => '\u{240a}', 'f' => '°',
'g' => '±',
'h' => '\u{2424}', 'i' => '\u{240b}', 'j' => '┘',
'k' => '┐',
'l' => '┌',
'm' => '└',
'n' => '┼',
'o' => '⎺',
'p' => '⎻',
'q' => '─',
'r' => '⎼',
's' => '⎽',
't' => '├',
'u' => '┤',
'v' => '┴',
'w' => '┬',
'x' => '│',
'y' => '≤',
'z' => '≥',
'{' => 'π',
'|' => '≠',
'}' => '£',
'~' => '·',
_ => c,
},
}
}
}
impl<'a, H> vte::Perform for Performer<'a, H>
where
H: Handler + 'a,
{
#[inline]
fn print(&mut self, c: char) {
self.handler.input(c);
self.state.preceding_char = Some(c);
}
#[inline]
fn execute(&mut self, byte: u8) {
match byte {
C0::HT => self.handler.put_tab(1),
C0::BS => self.handler.backspace(),
C0::CR => self.handler.carriage_return(),
C0::LF | C0::VT | C0::FF => self.handler.linefeed(),
C0::BEL => self.handler.bell(),
C0::SUB => self.handler.substitute(),
C0::SI => self.handler.set_active_charset(CharsetIndex::G0),
C0::SO => self.handler.set_active_charset(CharsetIndex::G1),
_ => debug!("[unhandled] execute byte={:02x}", byte),
}
}
#[inline]
fn hook(&mut self, params: &Params, intermediates: &[u8], ignore: bool, action: char) {
match (action, intermediates) {
('s', [b'=']) => {
if params.iter().next().map_or(false, |param| param[0] == 1) {
self.state.dcs = Some(Dcs::SyncStart);
}
},
_ => debug!(
"[unhandled hook] params={:?}, ints: {:?}, ignore: {:?}, action: {:?}",
params, intermediates, ignore, action
),
}
}
#[inline]
fn put(&mut self, byte: u8) {
debug!("[unhandled put] byte={:?}", byte);
}
#[inline]
fn unhook(&mut self) {
match self.state.dcs {
Some(Dcs::SyncStart) => {
self.state.sync_state.timeout = Some(Instant::now() + SYNC_UPDATE_TIMEOUT);
},
Some(Dcs::SyncEnd) => (),
_ => debug!("[unhandled unhook]"),
}
}
#[inline]
fn osc_dispatch(&mut self, params: &[&[u8]], bell_terminated: bool) {
let terminator = if bell_terminated { "\x07" } else { "\x1b\\" };
fn unhandled(params: &[&[u8]]) {
let mut buf = String::new();
for items in params {
buf.push('[');
for item in *items {
let _ = write!(buf, "{:?}", *item as char);
}
buf.push_str("],");
}
debug!("[unhandled osc_dispatch]: [{}] at line {}", &buf, line!());
}
if params.is_empty() || params[0].is_empty() {
return;
}
match params[0] {
b"0" | b"2" => {
if params.len() >= 2 {
let title = params[1..]
.iter()
.flat_map(|x| str::from_utf8(x))
.collect::<Vec<&str>>()
.join(";")
.trim()
.to_owned();
self.handler.set_title(Some(title));
return;
}
unhandled(params);
},
b"4" => {
if params.len() <= 1 || params.len() % 2 == 0 {
unhandled(params);
return;
}
for chunk in params[1..].chunks(2) {
let index = match parse_number(chunk[0]) {
Some(index) => index,
None => {
unhandled(params);
continue;
},
};
if let Some(c) = xparse_color(chunk[1]) {
self.handler.set_color(index as usize, c);
} else if chunk[1] == b"?" {
let prefix = format!("4;{}", index);
self.handler.dynamic_color_sequence(prefix, index as usize, terminator);
} else {
unhandled(params);
}
}
},
b"8" if params.len() > 2 => {
let link_params = params[1];
let uri = str::from_utf8(params[2]).unwrap_or_default();
if uri.is_empty() {
self.handler.set_hyperlink(None);
return;
}
let id = link_params
.split(|&b| b == b':')
.find_map(|kv| kv.strip_prefix(b"id="))
.and_then(|kv| str::from_utf8(kv).ok());
self.handler.set_hyperlink(Some(Hyperlink::new(id, uri)));
},
b"10" | b"11" | b"12" => {
if params.len() >= 2 {
if let Some(mut dynamic_code) = parse_number(params[0]) {
for param in ¶ms[1..] {
let offset = dynamic_code as usize - 10;
let index = NamedColor::Foreground as usize + offset;
if index > NamedColor::Cursor as usize {
unhandled(params);
break;
}
if let Some(color) = xparse_color(param) {
self.handler.set_color(index, color);
} else if param == b"?" {
self.handler.dynamic_color_sequence(
dynamic_code.to_string(),
index,
terminator,
);
} else {
unhandled(params);
}
dynamic_code += 1;
}
return;
}
}
unhandled(params);
},
b"50" => {
if params.len() >= 2
&& params[1].len() >= 13
&& params[1][0..12] == *b"CursorShape="
{
let shape = match params[1][12] as char {
'0' => CursorShape::Block,
'1' => CursorShape::Beam,
'2' => CursorShape::Underline,
_ => return unhandled(params),
};
self.handler.set_cursor_shape(shape);
return;
}
unhandled(params);
},
b"52" => {
if params.len() < 3 {
return unhandled(params);
}
let clipboard = params[1].first().unwrap_or(&b'c');
match params[2] {
b"?" => self.handler.clipboard_load(*clipboard, terminator),
base64 => self.handler.clipboard_store(*clipboard, base64),
}
},
b"104" => {
if params.len() == 1 || params[1].is_empty() {
for i in 0..256 {
self.handler.reset_color(i);
}
return;
}
for param in ¶ms[1..] {
match parse_number(param) {
Some(index) => self.handler.reset_color(index as usize),
None => unhandled(params),
}
}
},
b"110" => self.handler.reset_color(NamedColor::Foreground as usize),
b"111" => self.handler.reset_color(NamedColor::Background as usize),
b"112" => self.handler.reset_color(NamedColor::Cursor as usize),
_ => unhandled(params),
}
}
#[allow(clippy::cognitive_complexity)]
#[inline]
fn csi_dispatch(
&mut self,
params: &Params,
intermediates: &[u8],
has_ignored_intermediates: bool,
action: char,
) {
macro_rules! unhandled {
() => {{
debug!(
"[Unhandled CSI] action={:?}, params={:?}, intermediates={:?}",
action, params, intermediates
);
}};
}
if has_ignored_intermediates || intermediates.len() > 1 {
unhandled!();
return;
}
let mut params_iter = params.iter();
let handler = &mut self.handler;
let mut next_param_or = |default: u16| match params_iter.next() {
Some(&[param, ..]) if param != 0 => param,
_ => default,
};
match (action, intermediates) {
('@', []) => handler.insert_blank(next_param_or(1) as usize),
('A', []) => handler.move_up(next_param_or(1) as usize),
('B', []) | ('e', []) => handler.move_down(next_param_or(1) as usize),
('b', []) => {
if let Some(c) = self.state.preceding_char {
for _ in 0..next_param_or(1) {
handler.input(c);
}
} else {
debug!("tried to repeat with no preceding char");
}
},
('C', []) | ('a', []) => handler.move_forward(Column(next_param_or(1) as usize)),
('c', intermediates) if next_param_or(0) == 0 => {
handler.identify_terminal(intermediates.first().map(|&i| i as char))
},
('D', []) => handler.move_backward(Column(next_param_or(1) as usize)),
('d', []) => handler.goto_line(Line(next_param_or(1) as i32 - 1)),
('E', []) => handler.move_down_and_cr(next_param_or(1) as usize),
('F', []) => handler.move_up_and_cr(next_param_or(1) as usize),
('G', []) | ('`', []) => handler.goto_col(Column(next_param_or(1) as usize - 1)),
('g', []) => {
let mode = match next_param_or(0) {
0 => TabulationClearMode::Current,
3 => TabulationClearMode::All,
_ => {
unhandled!();
return;
},
};
handler.clear_tabs(mode);
},
('H', []) | ('f', []) => {
let y = next_param_or(1) as i32;
let x = next_param_or(1) as usize;
handler.goto(Line(y - 1), Column(x - 1));
},
('h', intermediates) => {
for param in params_iter.map(|param| param[0]) {
match Mode::from_primitive(intermediates.first(), param) {
Some(mode) => handler.set_mode(mode),
None => unhandled!(),
}
}
},
('I', []) => handler.move_forward_tabs(next_param_or(1)),
('J', []) => {
let mode = match next_param_or(0) {
0 => ClearMode::Below,
1 => ClearMode::Above,
2 => ClearMode::All,
3 => ClearMode::Saved,
_ => {
unhandled!();
return;
},
};
handler.clear_screen(mode);
},
('K', []) => {
let mode = match next_param_or(0) {
0 => LineClearMode::Right,
1 => LineClearMode::Left,
2 => LineClearMode::All,
_ => {
unhandled!();
return;
},
};
handler.clear_line(mode);
},
('L', []) => handler.insert_blank_lines(next_param_or(1) as usize),
('l', intermediates) => {
for param in params_iter.map(|param| param[0]) {
match Mode::from_primitive(intermediates.first(), param) {
Some(mode) => handler.unset_mode(mode),
None => unhandled!(),
}
}
},
('M', []) => handler.delete_lines(next_param_or(1) as usize),
('m', []) => {
if params.is_empty() {
handler.terminal_attribute(Attr::Reset);
} else {
for attr in attrs_from_sgr_parameters(&mut params_iter) {
match attr {
Some(attr) => handler.terminal_attribute(attr),
None => unhandled!(),
}
}
}
},
('n', []) => handler.device_status(next_param_or(0) as usize),
('P', []) => handler.delete_chars(next_param_or(1) as usize),
('q', [b' ']) => {
let cursor_style_id = next_param_or(0);
let shape = match cursor_style_id {
0 => None,
1 | 2 => Some(CursorShape::Block),
3 | 4 => Some(CursorShape::Underline),
5 | 6 => Some(CursorShape::Beam),
_ => {
unhandled!();
return;
},
};
let cursor_style =
shape.map(|shape| CursorStyle { shape, blinking: cursor_style_id % 2 == 1 });
handler.set_cursor_style(cursor_style);
},
('r', []) => {
let top = next_param_or(1) as usize;
let bottom =
params_iter.next().map(|param| param[0] as usize).filter(|¶m| param != 0);
handler.set_scrolling_region(top, bottom);
},
('S', []) => handler.scroll_up(next_param_or(1) as usize),
('s', []) => handler.save_cursor_position(),
('T', []) => handler.scroll_down(next_param_or(1) as usize),
('t', []) => match next_param_or(1) as usize {
14 => handler.text_area_size_pixels(),
18 => handler.text_area_size_chars(),
22 => handler.push_title(),
23 => handler.pop_title(),
_ => unhandled!(),
},
('u', []) => handler.restore_cursor_position(),
('X', []) => handler.erase_chars(Column(next_param_or(1) as usize)),
('Z', []) => handler.move_backward_tabs(next_param_or(1)),
_ => unhandled!(),
}
}
#[inline]
fn esc_dispatch(&mut self, intermediates: &[u8], _ignore: bool, byte: u8) {
macro_rules! unhandled {
() => {{
debug!(
"[unhandled] esc_dispatch ints={:?}, byte={:?} ({:02x})",
intermediates, byte as char, byte
);
}};
}
macro_rules! configure_charset {
($charset:path, $intermediates:expr) => {{
let index: CharsetIndex = match $intermediates {
[b'('] => CharsetIndex::G0,
[b')'] => CharsetIndex::G1,
[b'*'] => CharsetIndex::G2,
[b'+'] => CharsetIndex::G3,
_ => {
unhandled!();
return;
},
};
self.handler.configure_charset(index, $charset)
}};
}
match (byte, intermediates) {
(b'B', intermediates) => configure_charset!(StandardCharset::Ascii, intermediates),
(b'D', []) => self.handler.linefeed(),
(b'E', []) => {
self.handler.linefeed();
self.handler.carriage_return();
},
(b'H', []) => self.handler.set_horizontal_tabstop(),
(b'M', []) => self.handler.reverse_index(),
(b'Z', []) => self.handler.identify_terminal(None),
(b'c', []) => self.handler.reset_state(),
(b'0', intermediates) => {
configure_charset!(StandardCharset::SpecialCharacterAndLineDrawing, intermediates)
},
(b'7', []) => self.handler.save_cursor_position(),
(b'8', [b'#']) => self.handler.decaln(),
(b'8', []) => self.handler.restore_cursor_position(),
(b'=', []) => self.handler.set_keypad_application_mode(),
(b'>', []) => self.handler.unset_keypad_application_mode(),
(b'\\', []) => (),
_ => unhandled!(),
}
}
}
#[inline]
fn attrs_from_sgr_parameters(params: &mut ParamsIter<'_>) -> Vec<Option<Attr>> {
let mut attrs = Vec::with_capacity(params.size_hint().0);
while let Some(param) = params.next() {
let attr = match param {
[0] => Some(Attr::Reset),
[1] => Some(Attr::Bold),
[2] => Some(Attr::Dim),
[3] => Some(Attr::Italic),
[4, 0] => Some(Attr::CancelUnderline),
[4, 2] => Some(Attr::DoubleUnderline),
[4, 3] => Some(Attr::Undercurl),
[4, 4] => Some(Attr::DottedUnderline),
[4, 5] => Some(Attr::DashedUnderline),
[4, ..] => Some(Attr::Underline),
[5] => Some(Attr::BlinkSlow),
[6] => Some(Attr::BlinkFast),
[7] => Some(Attr::Reverse),
[8] => Some(Attr::Hidden),
[9] => Some(Attr::Strike),
[21] => Some(Attr::CancelBold),
[22] => Some(Attr::CancelBoldDim),
[23] => Some(Attr::CancelItalic),
[24] => Some(Attr::CancelUnderline),
[25] => Some(Attr::CancelBlink),
[27] => Some(Attr::CancelReverse),
[28] => Some(Attr::CancelHidden),
[29] => Some(Attr::CancelStrike),
[30] => Some(Attr::Foreground(Color::Named(NamedColor::Black))),
[31] => Some(Attr::Foreground(Color::Named(NamedColor::Red))),
[32] => Some(Attr::Foreground(Color::Named(NamedColor::Green))),
[33] => Some(Attr::Foreground(Color::Named(NamedColor::Yellow))),
[34] => Some(Attr::Foreground(Color::Named(NamedColor::Blue))),
[35] => Some(Attr::Foreground(Color::Named(NamedColor::Magenta))),
[36] => Some(Attr::Foreground(Color::Named(NamedColor::Cyan))),
[37] => Some(Attr::Foreground(Color::Named(NamedColor::White))),
[38] => {
let mut iter = params.map(|param| param[0]);
parse_sgr_color(&mut iter).map(Attr::Foreground)
},
[38, params @ ..] => handle_colon_rgb(params).map(Attr::Foreground),
[39] => Some(Attr::Foreground(Color::Named(NamedColor::Foreground))),
[40] => Some(Attr::Background(Color::Named(NamedColor::Black))),
[41] => Some(Attr::Background(Color::Named(NamedColor::Red))),
[42] => Some(Attr::Background(Color::Named(NamedColor::Green))),
[43] => Some(Attr::Background(Color::Named(NamedColor::Yellow))),
[44] => Some(Attr::Background(Color::Named(NamedColor::Blue))),
[45] => Some(Attr::Background(Color::Named(NamedColor::Magenta))),
[46] => Some(Attr::Background(Color::Named(NamedColor::Cyan))),
[47] => Some(Attr::Background(Color::Named(NamedColor::White))),
[48] => {
let mut iter = params.map(|param| param[0]);
parse_sgr_color(&mut iter).map(Attr::Background)
},
[48, params @ ..] => handle_colon_rgb(params).map(Attr::Background),
[49] => Some(Attr::Background(Color::Named(NamedColor::Background))),
[58] => {
let mut iter = params.map(|param| param[0]);
parse_sgr_color(&mut iter).map(|color| Attr::UnderlineColor(Some(color)))
},
[58, params @ ..] => {
handle_colon_rgb(params).map(|color| Attr::UnderlineColor(Some(color)))
},
[59] => Some(Attr::UnderlineColor(None)),
[90] => Some(Attr::Foreground(Color::Named(NamedColor::BrightBlack))),
[91] => Some(Attr::Foreground(Color::Named(NamedColor::BrightRed))),
[92] => Some(Attr::Foreground(Color::Named(NamedColor::BrightGreen))),
[93] => Some(Attr::Foreground(Color::Named(NamedColor::BrightYellow))),
[94] => Some(Attr::Foreground(Color::Named(NamedColor::BrightBlue))),
[95] => Some(Attr::Foreground(Color::Named(NamedColor::BrightMagenta))),
[96] => Some(Attr::Foreground(Color::Named(NamedColor::BrightCyan))),
[97] => Some(Attr::Foreground(Color::Named(NamedColor::BrightWhite))),
[100] => Some(Attr::Background(Color::Named(NamedColor::BrightBlack))),
[101] => Some(Attr::Background(Color::Named(NamedColor::BrightRed))),
[102] => Some(Attr::Background(Color::Named(NamedColor::BrightGreen))),
[103] => Some(Attr::Background(Color::Named(NamedColor::BrightYellow))),
[104] => Some(Attr::Background(Color::Named(NamedColor::BrightBlue))),
[105] => Some(Attr::Background(Color::Named(NamedColor::BrightMagenta))),
[106] => Some(Attr::Background(Color::Named(NamedColor::BrightCyan))),
[107] => Some(Attr::Background(Color::Named(NamedColor::BrightWhite))),
_ => None,
};
attrs.push(attr);
}
attrs
}
#[inline]
fn handle_colon_rgb(params: &[u16]) -> Option<Color> {
let rgb_start = if params.len() > 4 { 2 } else { 1 };
let rgb_iter = params[rgb_start..].iter().copied();
let mut iter = iter::once(params[0]).chain(rgb_iter);
parse_sgr_color(&mut iter)
}
fn parse_sgr_color(params: &mut dyn Iterator<Item = u16>) -> Option<Color> {
match params.next() {
Some(2) => Some(Color::Spec(Rgb {
r: u8::try_from(params.next()?).ok()?,
g: u8::try_from(params.next()?).ok()?,
b: u8::try_from(params.next()?).ok()?,
})),
Some(5) => Some(Color::Indexed(u8::try_from(params.next()?).ok()?)),
_ => None,
}
}
#[allow(non_snake_case)]
pub mod C0 {
pub const NUL: u8 = 0x00;
pub const SOH: u8 = 0x01;
pub const STX: u8 = 0x02;
pub const ETX: u8 = 0x03;
pub const EOT: u8 = 0x04;
pub const ENQ: u8 = 0x05;
pub const ACK: u8 = 0x06;
pub const BEL: u8 = 0x07;
pub const BS: u8 = 0x08;
pub const HT: u8 = 0x09;
pub const LF: u8 = 0x0A;
pub const VT: u8 = 0x0B;
pub const FF: u8 = 0x0C;
pub const CR: u8 = 0x0D;
pub const SO: u8 = 0x0E;
pub const SI: u8 = 0x0F;
pub const DLE: u8 = 0x10;
pub const XON: u8 = 0x11;
pub const DC2: u8 = 0x12;
pub const XOFF: u8 = 0x13;
pub const DC4: u8 = 0x14;
pub const NAK: u8 = 0x15;
pub const SYN: u8 = 0x16;
pub const ETB: u8 = 0x17;
pub const CAN: u8 = 0x18;
pub const EM: u8 = 0x19;
pub const SUB: u8 = 0x1A;
pub const ESC: u8 = 0x1B;
pub const FS: u8 = 0x1C;
pub const GS: u8 = 0x1D;
pub const RS: u8 = 0x1E;
pub const US: u8 = 0x1F;
pub const DEL: u8 = 0x7f;
}
#[cfg(test)]
mod tests {
use super::*;
use crate::term::color::Rgb;
struct MockHandler {
index: CharsetIndex,
charset: StandardCharset,
attr: Option<Attr>,
identity_reported: bool,
color: Option<Rgb>,
reset_colors: Vec<usize>,
}
impl Handler for MockHandler {
fn terminal_attribute(&mut self, attr: Attr) {
self.attr = Some(attr);
}
fn configure_charset(&mut self, index: CharsetIndex, charset: StandardCharset) {
self.index = index;
self.charset = charset;
}
fn set_active_charset(&mut self, index: CharsetIndex) {
self.index = index;
}
fn identify_terminal(&mut self, _intermediate: Option<char>) {
self.identity_reported = true;
}
fn reset_state(&mut self) {
*self = Self::default();
}
fn set_color(&mut self, _: usize, c: Rgb) {
self.color = Some(c);
}
fn reset_color(&mut self, index: usize) {
self.reset_colors.push(index)
}
}
impl Default for MockHandler {
fn default() -> MockHandler {
MockHandler {
index: CharsetIndex::G0,
charset: StandardCharset::Ascii,
attr: None,
identity_reported: false,
color: None,
reset_colors: Vec::new(),
}
}
}
#[test]
fn parse_control_attribute() {
static BYTES: &[u8] = &[0x1b, b'[', b'1', b'm'];
let mut parser = Processor::new();
let mut handler = MockHandler::default();
for byte in BYTES {
parser.advance(&mut handler, *byte);
}
assert_eq!(handler.attr, Some(Attr::Bold));
}
#[test]
fn parse_terminal_identity_csi() {
let bytes: &[u8] = &[0x1b, b'[', b'1', b'c'];
let mut parser = Processor::new();
let mut handler = MockHandler::default();
for byte in bytes {
parser.advance(&mut handler, *byte);
}
assert!(!handler.identity_reported);
handler.reset_state();
let bytes: &[u8] = &[0x1b, b'[', b'c'];
for byte in bytes {
parser.advance(&mut handler, *byte);
}
assert!(handler.identity_reported);
handler.reset_state();
let bytes: &[u8] = &[0x1b, b'[', b'0', b'c'];
for byte in bytes {
parser.advance(&mut handler, *byte);
}
assert!(handler.identity_reported);
}
#[test]
fn parse_terminal_identity_esc() {
let bytes: &[u8] = &[0x1b, b'Z'];
let mut parser = Processor::new();
let mut handler = MockHandler::default();
for byte in bytes {
parser.advance(&mut handler, *byte);
}
assert!(handler.identity_reported);
handler.reset_state();
let bytes: &[u8] = &[0x1b, b'#', b'Z'];
let mut parser = Processor::new();
let mut handler = MockHandler::default();
for byte in bytes {
parser.advance(&mut handler, *byte);
}
assert!(!handler.identity_reported);
handler.reset_state();
}
#[test]
fn parse_truecolor_attr() {
static BYTES: &[u8] = &[
0x1b, b'[', b'3', b'8', b';', b'2', b';', b'1', b'2', b'8', b';', b'6', b'6', b';',
b'2', b'5', b'5', b'm',
];
let mut parser = Processor::new();
let mut handler = MockHandler::default();
for byte in BYTES {
parser.advance(&mut handler, *byte);
}
let spec = Rgb { r: 128, g: 66, b: 255 };
assert_eq!(handler.attr, Some(Attr::Foreground(Color::Spec(spec))));
}
#[test]
fn parse_zsh_startup() {
static BYTES: &[u8] = &[
0x1b, b'[', b'1', b'm', 0x1b, b'[', b'7', b'm', b'%', 0x1b, b'[', b'2', b'7', b'm',
0x1b, b'[', b'1', b'm', 0x1b, b'[', b'0', b'm', b' ', b' ', b' ', b' ', b' ', b' ',
b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ',
b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ',
b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ',
b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ',
b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ', b' ',
b' ', b' ', b' ', b'\r', b' ', b'\r', b'\r', 0x1b, b'[', b'0', b'm', 0x1b, b'[', b'2',
b'7', b'm', 0x1b, b'[', b'2', b'4', b'm', 0x1b, b'[', b'J', b'j', b'w', b'i', b'l',
b'm', b'@', b'j', b'w', b'i', b'l', b'm', b'-', b'd', b'e', b's', b'k', b' ', 0x1b,
b'[', b'0', b'1', b';', b'3', b'2', b'm', 0xe2, 0x9e, 0x9c, b' ', 0x1b, b'[', b'0',
b'1', b';', b'3', b'2', b'm', b' ', 0x1b, b'[', b'3', b'6', b'm', b'~', b'/', b'c',
b'o', b'd', b'e',
];
let mut handler = MockHandler::default();
let mut parser = Processor::new();
for byte in BYTES {
parser.advance(&mut handler, *byte);
}
}
#[test]
fn parse_designate_g0_as_line_drawing() {
static BYTES: &[u8] = &[0x1b, b'(', b'0'];
let mut parser = Processor::new();
let mut handler = MockHandler::default();
for byte in BYTES {
parser.advance(&mut handler, *byte);
}
assert_eq!(handler.index, CharsetIndex::G0);
assert_eq!(handler.charset, StandardCharset::SpecialCharacterAndLineDrawing);
}
#[test]
fn parse_designate_g1_as_line_drawing_and_invoke() {
static BYTES: &[u8] = &[0x1b, b')', b'0', 0x0e];
let mut parser = Processor::new();
let mut handler = MockHandler::default();
for byte in &BYTES[..3] {
parser.advance(&mut handler, *byte);
}
assert_eq!(handler.index, CharsetIndex::G1);
assert_eq!(handler.charset, StandardCharset::SpecialCharacterAndLineDrawing);
let mut handler = MockHandler::default();
parser.advance(&mut handler, BYTES[3]);
assert_eq!(handler.index, CharsetIndex::G1);
}
#[test]
fn parse_valid_rgb_colors() {
assert_eq!(xparse_color(b"rgb:f/e/d"), Some(Rgb { r: 0xff, g: 0xee, b: 0xdd }));
assert_eq!(xparse_color(b"rgb:11/aa/ff"), Some(Rgb { r: 0x11, g: 0xaa, b: 0xff }));
assert_eq!(xparse_color(b"rgb:f/ed1/cb23"), Some(Rgb { r: 0xff, g: 0xec, b: 0xca }));
assert_eq!(xparse_color(b"rgb:ffff/0/0"), Some(Rgb { r: 0xff, g: 0x0, b: 0x0 }));
}
#[test]
fn parse_valid_legacy_rgb_colors() {
assert_eq!(xparse_color(b"#1af"), Some(Rgb { r: 0x10, g: 0xa0, b: 0xf0 }));
assert_eq!(xparse_color(b"#11aaff"), Some(Rgb { r: 0x11, g: 0xaa, b: 0xff }));
assert_eq!(xparse_color(b"#110aa0ff0"), Some(Rgb { r: 0x11, g: 0xaa, b: 0xff }));
assert_eq!(xparse_color(b"#1100aa00ff00"), Some(Rgb { r: 0x11, g: 0xaa, b: 0xff }));
}
#[test]
fn parse_invalid_rgb_colors() {
assert_eq!(xparse_color(b"rgb:0//"), None);
assert_eq!(xparse_color(b"rgb://///"), None);
}
#[test]
fn parse_invalid_legacy_rgb_colors() {
assert_eq!(xparse_color(b"#"), None);
assert_eq!(xparse_color(b"#f"), None);
}
#[test]
fn parse_invalid_number() {
assert_eq!(parse_number(b"1abc"), None);
}
#[test]
fn parse_valid_number() {
assert_eq!(parse_number(b"123"), Some(123));
}
#[test]
fn parse_number_too_large() {
assert_eq!(parse_number(b"321"), None);
}
#[test]
fn parse_osc4_set_color() {
let bytes: &[u8] = b"\x1b]4;0;#fff\x1b\\";
let mut parser = Processor::new();
let mut handler = MockHandler::default();
for byte in bytes {
parser.advance(&mut handler, *byte);
}
assert_eq!(handler.color, Some(Rgb { r: 0xf0, g: 0xf0, b: 0xf0 }));
}
#[test]
fn parse_osc104_reset_color() {
let bytes: &[u8] = b"\x1b]104;1;\x1b\\";
let mut parser = Processor::new();
let mut handler = MockHandler::default();
for byte in bytes {
parser.advance(&mut handler, *byte);
}
assert_eq!(handler.reset_colors, vec![1]);
}
#[test]
fn parse_osc104_reset_all_colors() {
let bytes: &[u8] = b"\x1b]104;\x1b\\";
let mut parser = Processor::new();
let mut handler = MockHandler::default();
for byte in bytes {
parser.advance(&mut handler, *byte);
}
let expected: Vec<usize> = (0..256).collect();
assert_eq!(handler.reset_colors, expected);
}
#[test]
fn parse_osc104_reset_all_colors_no_semicolon() {
let bytes: &[u8] = b"\x1b]104\x1b\\";
let mut parser = Processor::new();
let mut handler = MockHandler::default();
for byte in bytes {
parser.advance(&mut handler, *byte);
}
let expected: Vec<usize> = (0..256).collect();
assert_eq!(handler.reset_colors, expected);
}
}