keyhunter 0.2.0

Check for leaked API keys and secrets on public websites
Documentation 
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

use std::io::{stdout, Stdout, Write};

use super::ReportHandler;
use crate::ApiKeyError;
use miette::{IntoDiagnostic, Result};
// use miette::JSONReportHandler
use serde::Serialize;
use serde_json::value::Serializer;

pub struct JsonReportHandler {
    writer: Stdout,
}
impl Default for JsonReportHandler {
    fn default() -> Self {
        Self { writer: stdout() }
    }
}

impl ReportHandler for JsonReportHandler {
    fn report_keys(&self, keys: std::slice::Iter<'_, ApiKeyError>) -> Result<()> {
        let mut w = self.writer.lock();
        for key in keys {
            self._report_key(&mut w, key)?;
        }
        Ok(())
    }

    fn report_key(&self, key: &ApiKeyError) -> Result<()> {
        let mut w = self.writer.lock();
        self._report_key(&mut w, key)
    }
}

impl JsonReportHandler {
    fn _report_key<W: Write>(&self, w: &mut W, key: &ApiKeyError) -> Result<()> {
        let json = key.serialize(Serializer).into_diagnostic()?;
        writeln!(w, "{json}").into_diagnostic()
    }
}