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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
// Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0
use askalono::*;
use std::collections::HashMap;
use std::fs::File;
use std::io::{stdin, Read};
enum Annotation {
Begin(String),
End,
}
fn main() {
let args: Vec<_> = std::env::args().collect();
if args.len() != 2 {
eprintln!("usage: annotate-text cache.bin.zstd < input.txt > output.html");
std::process::exit(1);
}
let cache = &args[1];
let store = Store::from_cache(File::open(cache).expect("couldn't read cache file"))
.expect("error parsing cache");
let mut buf = String::new();
stdin()
.read_to_string(&mut buf)
.expect("couldn't read stdin");
let strategy = ScanStrategy::new(&store)
.mode(ScanMode::TopDown)
.confidence_threshold(0.80);
let results = strategy
.scan(&TextData::new(&buf))
.expect("scan didn't complete successfully");
let mut annotations = HashMap::with_capacity(results.containing.len() * 2);
for result in &results.containing {
annotations.insert(
result.line_range.0,
Annotation::Begin(result.license.name.to_owned()),
);
annotations.insert(result.line_range.1, Annotation::End);
}
println!("<html><body>");
println!("<pre>{:#?}</pre>", results);
println!("<pre>");
for (i, line) in buf.lines().enumerate() {
if annotations.contains_key(&i) {
let a = annotations.get(&i).unwrap();
match a {
Annotation::Begin(license) => {
print!(
r#"<div style="background-color: rgba(50, 50, 255, 0.3)" title="{}">"#,
license
);
}
Annotation::End => {
print!("</div>");
}
}
}
println!("{}", line);
}
println!("</pre></body></html>");
}
|
