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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
use crate::errors::*;
use crate::cmd::Cmd;
use crate::db::ttl;
use crate::filters::{Target, Filter};
use crate::shell::Shell;
use serde::Serialize;
use serde_json;
use structopt::StructOpt;
use structopt::clap::AppSettings;
use crate::models::*;
#[derive(Debug, StructOpt)]
#[structopt(global_settings = &[AppSettings::ColoredHelp])]
pub struct Args {
#[structopt(subcommand)]
subcommand: Target,
#[structopt(long="json", group="output")]
json: bool,
#[structopt(long="paths", group="output")]
paths: bool,
}
enum Output {
Normal,
Json,
Paths,
}
struct Printer<'a, 'b> {
rl: &'a mut Shell<'b>,
output: Output,
}
impl<'a, 'b> Printer<'a, 'b> {
pub fn new(rl: &'a mut Shell<'b>, args: &Args) -> Printer<'a, 'b> {
let output = if args.json {
Output::Json
} else if args.paths {
Output::Paths
} else {
Output::Normal
};
Printer {
rl,
output,
}
}
pub fn select<T: Model + Detailed + Serialize>(&self, filter: &Filter) -> Result<()> {
for obj in self.rl.db().filter::<T>(&filter.parse_optional()?)? {
match self.output {
Output::Normal => println!("{}", obj.detailed(self.rl.db())?),
Output::Json => {
let v = serde_json::to_string(&obj)?;
println!("{}", v);
},
Output::Paths => {
let blob = obj.blob()
.ok_or_else(|| format_err!("This model isn't linked to blob storage"))?;
let path = self.rl.blobs()
.join(blob)?;
let path = path.to_str()
.ok_or_else(|| format_err!("Path is invalid utf-8"))?;
println!("{}", path);
},
}
}
Ok(())
}
}
impl Cmd for Args {
fn run(self, rl: &mut Shell) -> Result<()> {
ttl::reap_expired(rl.db())?;
let printer = Printer::new(rl, &self);
match &self.subcommand {
Target::Domains(filter) => printer.select::<Domain>(&filter),
Target::Subdomains(filter) => printer.select::<Subdomain>(&filter),
Target::IpAddrs(filter) => printer.select::<IpAddr>(&filter),
Target::Urls(filter) => printer.select::<Url>(&filter),
Target::Emails(filter) => printer.select::<Email>(&filter),
Target::PhoneNumbers(filter) => printer.select::<PhoneNumber>(&filter),
Target::Devices(filter) => printer.select::<Device>(&filter),
Target::Networks(filter) => printer.select::<Network>(&filter),
Target::Accounts(filter) => printer.select::<Account>(&filter),
Target::Breaches(filter) => printer.select::<Breach>(&filter),
Target::Images(filter) => printer.select::<Image>(&filter),
Target::Ports(filter) => printer.select::<Port>(&filter),
Target::Netblocks(filter) => printer.select::<Netblock>(&filter),
Target::CryptoAddrs(filter) => printer.select::<CryptoAddr>(&filter),
}
}
}