mirror of https://github.com/rust-lang/book
46 lines
1.3 KiB
Rust
46 lines
1.3 KiB
Rust
extern crate regex;
|
|
|
|
use std::collections::HashSet;
|
|
use std::io;
|
|
use std::io::{Read, Write};
|
|
use regex::{Regex, Captures};
|
|
|
|
fn main () {
|
|
let mut buffer = String::new();
|
|
if let Err(e) = io::stdin().read_to_string(&mut buffer) {
|
|
panic!(e);
|
|
}
|
|
|
|
let mut refs = HashSet::new();
|
|
|
|
// Capture all links and link references.
|
|
let regex = r"\[([^\]]+)\](?:(?:\[([^\]]+)\])|(?:\([^\)]+\)))(?i)<!-- ignore -->";
|
|
let link_regex = Regex::new(regex).unwrap();
|
|
let first_pass = link_regex.replace_all(&buffer, |caps: &Captures| {
|
|
|
|
// Save the link reference we want to delete.
|
|
if let Some(reference) = caps.at(2) {
|
|
refs.insert(reference.to_owned());
|
|
}
|
|
|
|
// Put the link title back.
|
|
caps.at(1).unwrap().to_owned()
|
|
});
|
|
|
|
// Search for the references we need to delete.
|
|
let ref_regex = Regex::new(r"\n\[([^\]]+)\]:\s.*\n").unwrap();
|
|
let out = ref_regex.replace_all(&first_pass, |caps: &Captures| {
|
|
let capture = caps.at(1).unwrap().to_owned();
|
|
|
|
// Check if we've marked this reference for deletion ...
|
|
if refs.contains(capture.as_str()) {
|
|
return "".to_string();
|
|
}
|
|
|
|
// ... else we put back everything we captured.
|
|
caps.at(0).unwrap().to_owned()
|
|
});
|
|
|
|
write!(io::stdout(), "{}", out).unwrap();
|
|
}
|