1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

#[macro_use]
extern crate html5ever;
extern crate toks;

use html5ever::parse_document;
use html5ever::tendril::TendrilSink;

use std::io::{self, Read};
use toks::prelude::*;

pub struct LinkTok {
    total: u32,
}

impl Tok for LinkTok {
    fn is_match(&self, qn: &QualName) -> bool {
        qn.local == local_name!("a")
    }

    fn process(&mut self, _: &mut Vec<Attribute>, _: &mut Vec<Handle>) {
        self.total += 1;
    }
}

// How to use
// $ cargo build --example count_links
// $ cat your.html | ./target/debug/examples/count_links
// Link <a> count 9
fn main() {
    let mut chunk = String::new();
    io::stdin().read_to_string(&mut chunk).unwrap();

    let dom = parse_document(RcDom::default(), Default::default()).one(chunk);

    let mut lt = LinkTok { total: 0 };

    // Dropping mut reference
    {
        recursion(&mut vec![&mut lt], dom.document);
    }

    println!("Link <a> count {}", lt.total);
}