html_sanitizer 0.1.1

extern crate html_sanitizer;

use std::fs::File;
use html_sanitizer::TagParser;

fn main() {
    let input_file = "C:/Users/Victor/Documents/spain october 2018/transavia flight ticket.html";
    let mut file = File::open(input_file).unwrap();
    let mut tag_parser = TagParser::new(&mut file);
    let result = tag_parser.walk(|tag| {
        if tag.name == "html" || tag.name == "body" {
            tag.ignore_self();
        } else if tag.name == "head" || tag.name == "script" || tag.name == "style" {
            tag.ignore_self_and_contents();
        } else if tag.name == "a" {
            tag.allow_attribute(String::from("href"));
        } else if tag.name == "img" {
            if let Some(url) = tag.attrs.iter().find(|(key, _)| key == &"src").map(|(_, url)| url) {
                let name = match url.rfind('/')  {
                    Some(i) => &url[i+1..],
                    None => "Load image",
                };
                tag.rewrite_as(format!("<a href=\"{0}\" onclick=\"return replace_url_by_image(this);\" title=\"{0}\">{1}</a>", url, name));
            }
        } else {
            tag.allow_attribute(String::from("style"));
        }
    });
    println!("{}", result);
}