332 lines
11 KiB
Rust
332 lines
11 KiB
Rust
#![feature(iter_advance_by)]
|
|
|
|
use {
|
|
cgmath::Point2,
|
|
nested::terminal::{TerminalAtom, TerminalStyle},
|
|
std::{
|
|
fs::File,
|
|
io::{stdin, Read, Write},
|
|
os::unix::io::FromRawFd,
|
|
},
|
|
vte::{Params, Parser, Perform},
|
|
};
|
|
|
|
struct ColorPalett {
|
|
black: (u8, u8, u8),
|
|
red: (u8, u8, u8),
|
|
green: (u8, u8, u8),
|
|
yellow: (u8, u8, u8),
|
|
blue: (u8, u8, u8),
|
|
magenta: (u8, u8, u8),
|
|
cyan: (u8, u8, u8),
|
|
white: (u8, u8, u8),
|
|
}
|
|
|
|
struct PerfAtom {
|
|
colors: ColorPalett,
|
|
term_width: i16,
|
|
|
|
cursor: Point2<i16>,
|
|
style: TerminalStyle,
|
|
|
|
out: File,
|
|
}
|
|
|
|
impl PerfAtom {
|
|
fn write_atom(&mut self, pos: Point2<i16>, atom: Option<TerminalAtom>) {
|
|
self.out
|
|
.write(&bincode::serialize(&(pos, atom)).unwrap())
|
|
.expect("");
|
|
}
|
|
}
|
|
|
|
impl Perform for PerfAtom {
|
|
fn print(&mut self, c: char) {
|
|
//eprintln!("[print] {:?}", c);
|
|
self.write_atom(self.cursor, Some(TerminalAtom::new(c, self.style)));
|
|
|
|
self.cursor.x += 1;
|
|
if self.cursor.x > self.term_width {
|
|
self.cursor.x = 0;
|
|
self.cursor.y += 1;
|
|
}
|
|
}
|
|
|
|
fn execute(&mut self, byte: u8) {
|
|
//eprintln!("[execute] {:02x}", byte);
|
|
match byte {
|
|
b'\n' => {
|
|
self.cursor.x = 0;
|
|
self.cursor.y += 1;
|
|
}
|
|
_ => {}
|
|
}
|
|
}
|
|
|
|
fn hook(&mut self, params: &Params, intermediates: &[u8], ignore: bool, c: char) {
|
|
eprintln!(
|
|
"[hook] params={:?}, intermediates={:?}, ignore={:?}, char={:?}",
|
|
params, intermediates, ignore, c
|
|
);
|
|
}
|
|
|
|
fn put(&mut self, byte: u8) {
|
|
eprintln!("[put] {:02x}", byte);
|
|
}
|
|
|
|
fn unhook(&mut self) {
|
|
eprintln!("[unhook]");
|
|
}
|
|
|
|
fn osc_dispatch(&mut self, params: &[&[u8]], bell_terminated: bool) {
|
|
eprintln!(
|
|
"[osc_dispatch] params={:?} bell_terminated={}",
|
|
params, bell_terminated
|
|
);
|
|
}
|
|
|
|
fn csi_dispatch(&mut self, params: &Params, intermediates: &[u8], ignore: bool, c: char) {
|
|
eprintln!(
|
|
"[csi_dispatch] params={:#?}, intermediates={:?}, ignore={:?}, char={:?}",
|
|
params, intermediates, ignore, c
|
|
);
|
|
|
|
let mut piter = params.into_iter();
|
|
|
|
match c {
|
|
// Set SGR
|
|
'm' => {
|
|
while let Some(n) = piter.next() {
|
|
match n[0] {
|
|
0 => self.style = TerminalStyle::default(),
|
|
1 => self.style = self.style.add(TerminalStyle::bold(true)),
|
|
3 => self.style = self.style.add(TerminalStyle::italic(true)),
|
|
4 => self.style = self.style.add(TerminalStyle::underline(true)),
|
|
|
|
30 => {
|
|
self.style = self.style.add(TerminalStyle::fg_color(self.colors.black))
|
|
}
|
|
40 => {
|
|
self.style = self.style.add(TerminalStyle::bg_color(self.colors.black))
|
|
}
|
|
31 => self.style = self.style.add(TerminalStyle::fg_color(self.colors.red)),
|
|
41 => self.style = self.style.add(TerminalStyle::bg_color(self.colors.red)),
|
|
32 => {
|
|
self.style = self.style.add(TerminalStyle::fg_color(self.colors.green))
|
|
}
|
|
42 => {
|
|
self.style = self.style.add(TerminalStyle::bg_color(self.colors.green))
|
|
}
|
|
33 => {
|
|
self.style = self.style.add(TerminalStyle::fg_color(self.colors.yellow))
|
|
}
|
|
43 => {
|
|
self.style = self.style.add(TerminalStyle::bg_color(self.colors.yellow))
|
|
}
|
|
34 => {
|
|
self.style = self.style.add(TerminalStyle::fg_color(self.colors.blue))
|
|
}
|
|
44 => {
|
|
self.style = self.style.add(TerminalStyle::bg_color(self.colors.blue))
|
|
}
|
|
35 => {
|
|
self.style =
|
|
self.style.add(TerminalStyle::fg_color(self.colors.magenta))
|
|
}
|
|
45 => {
|
|
self.style =
|
|
self.style.add(TerminalStyle::bg_color(self.colors.magenta))
|
|
}
|
|
36 => {
|
|
self.style = self.style.add(TerminalStyle::fg_color(self.colors.cyan))
|
|
}
|
|
46 => {
|
|
self.style = self.style.add(TerminalStyle::bg_color(self.colors.cyan))
|
|
}
|
|
37 => {
|
|
self.style = self.style.add(TerminalStyle::fg_color(self.colors.white))
|
|
}
|
|
47 => {
|
|
self.style = self.style.add(TerminalStyle::bg_color(self.colors.white))
|
|
}
|
|
|
|
38 => {
|
|
let x = piter.next().unwrap();
|
|
match x[0] {
|
|
2 => {
|
|
let r = piter.next().unwrap();
|
|
let g = piter.next().unwrap();
|
|
let b = piter.next().unwrap();
|
|
self.style = self.style.add(TerminalStyle::fg_color((
|
|
r[0] as u8,
|
|
g[0] as u8,
|
|
b[30] as u8,
|
|
)))
|
|
}
|
|
5 => {
|
|
let v = piter.next().unwrap();
|
|
self.style = self.style.add(TerminalStyle::fg_color(
|
|
ansi_colours::rgb_from_ansi256(v[0] as u8),
|
|
))
|
|
}
|
|
_ => {}
|
|
}
|
|
}
|
|
|
|
48 => {
|
|
let x = piter.next().unwrap();
|
|
match x[0] {
|
|
2 => {
|
|
let r = piter.next().unwrap();
|
|
let g = piter.next().unwrap();
|
|
let b = piter.next().unwrap();
|
|
self.style = self.style.add(TerminalStyle::bg_color((
|
|
r[0] as u8,
|
|
g[0] as u8,
|
|
b[30] as u8,
|
|
)))
|
|
}
|
|
5 => {
|
|
let v = piter.next().unwrap();
|
|
self.style = self.style.add(TerminalStyle::bg_color(
|
|
ansi_colours::rgb_from_ansi256(v[0] as u8),
|
|
))
|
|
}
|
|
_ => {}
|
|
}
|
|
}
|
|
|
|
_ => {}
|
|
}
|
|
}
|
|
}
|
|
|
|
'H' => {
|
|
if let Some(y) = piter.next() {
|
|
self.cursor.y = y[0] as i16 - 1
|
|
};
|
|
if let Some(x) = piter.next() {
|
|
self.cursor.x = x[0] as i16 - 1
|
|
};
|
|
|
|
eprintln!("cursor at {:?}", self.cursor);
|
|
}
|
|
|
|
'A' => {
|
|
self.cursor.y -= piter.next().unwrap()[0] as i16;
|
|
}
|
|
'B' => {
|
|
self.cursor.y += piter.next().unwrap()[0] as i16;
|
|
}
|
|
'C' => {
|
|
self.cursor.x += piter.next().unwrap()[0] as i16;
|
|
}
|
|
'D' => {
|
|
self.cursor.x -= piter.next().unwrap()[0] as i16;
|
|
}
|
|
'E' => {
|
|
self.cursor.x = 0;
|
|
self.cursor.y += piter.next().unwrap()[0] as i16;
|
|
}
|
|
|
|
'J' => {
|
|
let x = piter.next().unwrap_or(&[0 as u16; 1]);
|
|
match x[0] {
|
|
0 => {}
|
|
1 => {}
|
|
2 => {
|
|
for y in 0..100 {
|
|
for x in 0..self.term_width {
|
|
self.write_atom(Point2::new(x, y), None);
|
|
}
|
|
}
|
|
}
|
|
|
|
// invalid
|
|
_ => {}
|
|
}
|
|
}
|
|
|
|
'K' => {
|
|
let x = piter.next().unwrap();
|
|
match x[0] {
|
|
// clear cursor until end
|
|
0 => {
|
|
for x in self.cursor.x..self.term_width {
|
|
self.write_atom(Point2::new(x, self.cursor.y), None);
|
|
}
|
|
}
|
|
|
|
// clear start until cursor
|
|
1 => {
|
|
for x in 0..self.cursor.x {
|
|
self.write_atom(Point2::new(x, self.cursor.y), None);
|
|
}
|
|
}
|
|
|
|
// clear entire line
|
|
2 => {
|
|
for x in 0..self.term_width {
|
|
self.write_atom(Point2::new(x, self.cursor.y), None);
|
|
}
|
|
}
|
|
|
|
// invalid
|
|
_ => {}
|
|
}
|
|
}
|
|
|
|
_ => {}
|
|
}
|
|
}
|
|
|
|
fn esc_dispatch(&mut self, intermediates: &[u8], ignore: bool, byte: u8) {
|
|
eprintln!(
|
|
"[esc_dispatch] intermediates={:?}, ignore={:?}, byte={:02x}",
|
|
intermediates, ignore, byte
|
|
);
|
|
}
|
|
}
|
|
|
|
fn main() {
|
|
let input = stdin();
|
|
let mut handle = input.lock();
|
|
|
|
let mut statemachine = Parser::new();
|
|
let mut performer = PerfAtom {
|
|
cursor: Point2::new(0, 0),
|
|
style: TerminalStyle::default(),
|
|
term_width: 200,
|
|
out: unsafe { File::from_raw_fd(1) },
|
|
|
|
colors: ColorPalett {
|
|
black: (1, 1, 1),
|
|
red: (222, 56, 43),
|
|
green: (0, 64, 0),
|
|
yellow: (255, 199, 6),
|
|
blue: (0, 111, 184),
|
|
magenta: (118, 38, 113),
|
|
cyan: (44, 181, 233),
|
|
white: (204, 204, 204),
|
|
},
|
|
};
|
|
|
|
let mut buf = [0; 2048];
|
|
|
|
loop {
|
|
match handle.read(&mut buf) {
|
|
Ok(0) => break,
|
|
Ok(n) => {
|
|
for byte in &buf[..n] {
|
|
statemachine.advance(&mut performer, *byte);
|
|
performer.out.flush().unwrap();
|
|
}
|
|
}
|
|
Err(err) => {
|
|
println!("err: {}", err);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|