#!/usr/bin/env run-cargo-script
extern crate markup5ever;
extern crate xml5ever;
use std::default::Default;
use std::io;
use markup5ever::buffer_queue::BufferQueue;
use xml5ever::tendril::{ByteTendril, ReadExt};
use xml5ever::tokenizer::{CharacterTokens, NullCharacterToken, TagToken};
use xml5ever::tokenizer::{EmptyTag, EndTag, ShortTag, StartTag};
use xml5ever::tokenizer::{PIToken, Pi};
use xml5ever::tokenizer::{ParseError, Token, TokenSink, XmlTokenizer, XmlTokenizerOpts};
#[derive(Copy, Clone)]
struct TokenPrinter {
in_char_run: bool,
}
impl TokenPrinter {
fn is_char(&mut self, is_char: bool) {
match (self.in_char_run, is_char) {
(false, true) => print!("CHAR : \""),
(true, false) => println!("\""),
_ => (),
}
self.in_char_run = is_char;
}
fn do_char(&mut self, c: char) {
self.is_char(true);
print!("{}", c.escape_default().collect::<String>());
}
}
impl TokenSink for TokenPrinter {
fn process_token(&mut self, token: Token) {
match token {
CharacterTokens(b) => {
for c in b.chars() {
self.do_char(c);
}
},
NullCharacterToken => self.do_char('\0'),
TagToken(tag) => {
self.is_char(false);
match tag.kind {
StartTag => print!("TAG : <\x1b[32m{}\x1b[0m", tag.name.local),
EndTag => print!("END TAG : <\x1b[31m/{}\x1b[0m", tag.name.local),
ShortTag => print!("Short TAG : <\x1b[31m/{}\x1b[0m", tag.name.local),
EmptyTag => print!("Empty TAG : <\x1b[31m{}\x1b[0m", tag.name.local),
}
for attr in tag.attrs.iter() {
print!(
" \x1b[36m{}\x1b[0m='\x1b[34m{}\x1b[0m'",
attr.name.local, attr.value
);
}
if tag.kind == EmptyTag {
print!("/");
}
println!(">");
},
ParseError(err) => {
self.is_char(false);
println!("ERROR: {}", err);
},
PIToken(Pi { target, data }) => {
self.is_char(false);
println!("PI : <?{:?} {:?}?>", target, data);
},
_ => {
self.is_char(false);
println!("OTHER: {:?}", token);
},
}
}
}
fn main() {
let mut sink = TokenPrinter { in_char_run: false };
let mut input = ByteTendril::new();
io::stdin().read_to_tendril(&mut input).unwrap();
let mut input_buffer = BufferQueue::new();
input_buffer.push_back(input.try_reinterpret().unwrap());
let mut tok = XmlTokenizer::new(
sink,
XmlTokenizerOpts {
profile: true,
exact_errors: true,
..Default::default()
},
);
tok.feed(&mut input_buffer);
tok.end();
sink.is_char(false);
}