it all works, just need to figure out how to deal with the end of the stream
This commit is contained in:
238
src/parsers.rs
Normal file
238
src/parsers.rs
Normal file
@@ -0,0 +1,238 @@
|
||||
#[cfg(test)]
|
||||
mod tests;
|
||||
|
||||
use crate::{AnsiSequence, Output, Color, TextAttribute};
|
||||
|
||||
use num_traits::cast::FromPrimitive;
|
||||
use std::convert::TryInto;
|
||||
use nom::*;
|
||||
|
||||
named!(
|
||||
parse_int<u32>,
|
||||
map_res!(
|
||||
map_res!(
|
||||
nom::digit,
|
||||
std::str::from_utf8
|
||||
),
|
||||
|s: &str| s.parse::<u32>()
|
||||
)
|
||||
);
|
||||
|
||||
named!(
|
||||
parse_fg_color<Color>,
|
||||
do_parse!(
|
||||
val: parse_int >>
|
||||
worked: expr_opt!(FromPrimitive::from_u32(val)) >>
|
||||
(worked)
|
||||
)
|
||||
);
|
||||
|
||||
named!(
|
||||
parse_bg_color<Color>,
|
||||
do_parse!(
|
||||
val: parse_int >>
|
||||
val: expr_opt!(val.checked_sub(10)) >>
|
||||
worked: expr_opt!(FromPrimitive::from_u32(val)) >>
|
||||
(worked)
|
||||
)
|
||||
);
|
||||
|
||||
named!(
|
||||
parse_text_attr<TextAttribute>,
|
||||
map!(
|
||||
map!(
|
||||
parse_int,
|
||||
FromPrimitive::from_u32
|
||||
),
|
||||
Option::unwrap
|
||||
)
|
||||
);
|
||||
|
||||
named!(
|
||||
cursor_pos<AnsiSequence>,
|
||||
do_parse!(
|
||||
x: parse_int >>
|
||||
tag!(";") >>
|
||||
y: parse_int >>
|
||||
alt!(
|
||||
tag!("H") |
|
||||
tag!("f")
|
||||
) >>
|
||||
(AnsiSequence::CursorPos(x, y))
|
||||
)
|
||||
);
|
||||
|
||||
named!(
|
||||
cursor_up<AnsiSequence>,
|
||||
do_parse!(
|
||||
am: parse_int >>
|
||||
tag!("A") >>
|
||||
(AnsiSequence::CursorUp(am))
|
||||
)
|
||||
);
|
||||
|
||||
named!(
|
||||
cursor_down<AnsiSequence>,
|
||||
do_parse!(
|
||||
am: parse_int >>
|
||||
tag!("B") >>
|
||||
(AnsiSequence::CursorDown(am))
|
||||
)
|
||||
);
|
||||
|
||||
named!(
|
||||
cursor_forward<AnsiSequence>,
|
||||
do_parse!(
|
||||
am: parse_int >>
|
||||
tag!("C") >>
|
||||
(AnsiSequence::CursorForward(am))
|
||||
)
|
||||
);
|
||||
|
||||
named!(
|
||||
cursor_backward<AnsiSequence>,
|
||||
do_parse!(
|
||||
am: parse_int >>
|
||||
tag!("D") >>
|
||||
(AnsiSequence::CursorBackward(am))
|
||||
)
|
||||
);
|
||||
|
||||
named!(
|
||||
cursor_save<AnsiSequence>,
|
||||
do_parse!(
|
||||
tag!("s") >>
|
||||
(AnsiSequence::CursorSave)
|
||||
)
|
||||
);
|
||||
|
||||
named!(
|
||||
cursor_restore<AnsiSequence>,
|
||||
do_parse!(
|
||||
tag!("u") >>
|
||||
(AnsiSequence::CursorRestore)
|
||||
)
|
||||
);
|
||||
|
||||
named!(
|
||||
erase_display<AnsiSequence>,
|
||||
do_parse!(
|
||||
tag!("2J") >>
|
||||
(AnsiSequence::EraseDisplay)
|
||||
)
|
||||
);
|
||||
|
||||
named!(
|
||||
erase_line<AnsiSequence>,
|
||||
do_parse!(
|
||||
tag!("K") >>
|
||||
(AnsiSequence::EraseDisplay)
|
||||
)
|
||||
);
|
||||
|
||||
named!(
|
||||
graphics_mode<AnsiSequence>,
|
||||
do_parse!(
|
||||
ta: parse_text_attr >>
|
||||
tag!(";") >>
|
||||
fg: parse_fg_color >>
|
||||
tag!(";") >>
|
||||
bg: parse_bg_color >>
|
||||
tag!("m") >>
|
||||
(AnsiSequence::SetGraphicsMode{
|
||||
ta: ta,
|
||||
fg: fg,
|
||||
bg: bg
|
||||
})
|
||||
)
|
||||
);
|
||||
|
||||
named!(
|
||||
set_mode<AnsiSequence>,
|
||||
do_parse!(
|
||||
tag!("=") >>
|
||||
mode: parse_int >>
|
||||
conv: expr_res!(mode.try_into()) >>
|
||||
tag!("h") >>
|
||||
(AnsiSequence::SetMode(conv))
|
||||
)
|
||||
);
|
||||
|
||||
named!(
|
||||
reset_mode<AnsiSequence>,
|
||||
do_parse!(
|
||||
tag!("=") >>
|
||||
mode: parse_int >>
|
||||
conv: expr_res!(mode.try_into()) >>
|
||||
tag!("l") >>
|
||||
(AnsiSequence::ResetMode(conv))
|
||||
)
|
||||
);
|
||||
|
||||
named!(
|
||||
combined<AnsiSequence>,
|
||||
alt!(
|
||||
cursor_pos
|
||||
| cursor_up
|
||||
| cursor_down
|
||||
| cursor_forward
|
||||
| cursor_backward
|
||||
| cursor_save
|
||||
| cursor_restore
|
||||
| erase_display
|
||||
| erase_line
|
||||
| graphics_mode
|
||||
| set_mode
|
||||
| reset_mode
|
||||
)
|
||||
);
|
||||
|
||||
named!(
|
||||
parse_escape<Output>,
|
||||
do_parse!(
|
||||
tag_s!("\x27[") >>
|
||||
seq: combined >>
|
||||
(Output::Escape(seq))
|
||||
)
|
||||
);
|
||||
|
||||
named!(
|
||||
parse_str<Output>,
|
||||
do_parse!(
|
||||
text: map_res!(
|
||||
take_until!("\x27["),
|
||||
std::str::from_utf8
|
||||
) >>
|
||||
(Output::TextBlock(text))
|
||||
)
|
||||
);
|
||||
|
||||
named!(
|
||||
parse_output<Output>,
|
||||
do_parse!(
|
||||
out: alt!(parse_escape | parse_str) >>
|
||||
(out)
|
||||
)
|
||||
);
|
||||
|
||||
pub struct ParserIterator<'a> {
|
||||
dat: &'a[u8],
|
||||
}
|
||||
|
||||
impl<'a> Iterator for ParserIterator<'a> {
|
||||
type Item = Output<'a>;
|
||||
|
||||
fn next(&mut self) -> Option<Self::Item> {
|
||||
let parse = parse_output(self.dat.as_bytes())
|
||||
.unwrap();
|
||||
|
||||
self.dat = parse.0;
|
||||
Some(parse.1)
|
||||
}
|
||||
}
|
||||
|
||||
pub fn iterate_on<'a>(bytes: &'a[u8]) -> ParserIterator<'a> {
|
||||
ParserIterator {
|
||||
dat: bytes
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user