dnsi 0.2.0

A tool for investigating the DNS.
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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60

//! Message output formats.

mod ansi;
mod dig;
mod error;
mod friendly;
mod table;
mod table_writer;
mod ttl;

use super::client::Answer;
use clap::{Parser, ValueEnum};
use error::OutputError;
use std::io;

//------------ OutputFormat --------------------------------------------------

#[derive(Clone, Copy, Debug, ValueEnum)]
pub enum OutputFormat {
    /// Similar to dig.
    Dig,

    /// Easily readable, formatted with ANSI codes and whitespace
    Friendly,

    /// Short readable format
    Table,
}

#[derive(Clone, Debug, Parser)]
pub struct OutputOptions {
    #[arg(long = "format", default_value = "friendly")]
    pub format: OutputFormat,
}

impl OutputFormat {
    pub fn write(
        self,
        msg: &Answer,
        target: &mut impl io::Write,
    ) -> Result<(), io::Error> {
        let res = match self {
            Self::Dig => self::dig::write(msg, target),
            Self::Friendly => self::friendly::write(msg, target),
            Self::Table => self::table::write(msg, target),
        };
        match res {
            Ok(()) => Ok(()),
            Err(OutputError::Io(e)) => Err(e),
            Err(OutputError::BadRecord(e)) => {
                writeln!(target, "ERROR: malformed message: {e}")?;
                Ok(())
            }
        }
    }

    pub fn print(self, msg: &Answer) -> Result<(), io::Error> {
        self.write(msg, &mut io::stdout().lock())
    }
}