mhost 0.11.3

Fast, async DNS lookup library and CLI -- modern dig/host replacement with parallel multi-server queries, DoH, DoT, subdomain discovery, and zone verification
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
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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155

// Copyright 2017-2021 Lukas Pustina <lukas@pustina.de>
//
// Licensed under the Apache License, Version 2.0, <LICENSE-APACHE or
// http://apache.org/licenses/LICENSE-2.0> or the MIT license <LICENSE-MIT or
// http://opensource.org/licenses/MIT>, at your option. This file may not be
// copied, modified, or distributed except according to those terms.

use core::sync::atomic::AtomicBool;
use std::sync::atomic::Ordering::SeqCst;

use yansi::{Color, Style};

use crate::RecordType;

// --- Output chrome styles (formerly in app/output/styles.rs) ---

static ASCII_MODE: AtomicBool = AtomicBool::new(false);

pub fn ascii_mode() {
    ASCII_MODE.store(true, SeqCst);
}

pub fn is_ascii() -> bool {
    ASCII_MODE.load(SeqCst)
}

pub fn no_color_mode() {
    yansi::disable();
}

pub static ATTENTION: Style = Style::new().fg(Color::Yellow).bold();
pub static ERROR: Style = Style::new().fg(Color::Red).bold();
pub static EMPH: Style = Style::new().fg(Color::White).bold();
pub static OK: Style = Style::new().fg(Color::Green).bold();

pub fn attention_prefix() -> &'static str {
    if ASCII_MODE.load(SeqCst) {
        "!"
    } else {
        "⚠︎"
    }
}

pub fn caption_prefix() -> &'static str {
    if ASCII_MODE.load(SeqCst) {
        ">"
    } else {
        "▶︎"
    }
}

pub fn error_prefix() -> &'static str {
    if ASCII_MODE.load(SeqCst) {
        "x"
    } else {
        "⚡︎"
    }
}

pub fn info_prefix() -> &'static str {
    if ASCII_MODE.load(SeqCst) {
        "-"
    } else {
        "▸"
    }
}

pub fn itemization_prefix() -> &'static str {
    if ASCII_MODE.load(SeqCst) {
        "*"
    } else {
        "∙"
    }
}

pub fn finished_prefix() -> &'static str {
    if ASCII_MODE.load(SeqCst) {
        "+"
    } else {
        "❖"
    }
}

pub fn ok_prefix() -> &'static str {
    if ASCII_MODE.load(SeqCst) {
        "="
    } else {
        "✓"
    }
}

// --- Record type styles (originally in app/common/styles.rs) ---

pub const fn record_type_color(rt: RecordType) -> Color {
    match rt {
        RecordType::A | RecordType::AAAA | RecordType::NULL => Color::White,
        RecordType::ANAME | RecordType::CNAME | RecordType::NS | RecordType::PTR => Color::Blue,
        RecordType::CAA => Color::Cyan,
        RecordType::DNSKEY
        | RecordType::DS
        | RecordType::RRSIG
        | RecordType::NSEC
        | RecordType::NSEC3
        | RecordType::NSEC3PARAM => Color::Green,
        RecordType::HINFO => Color::Yellow,
        RecordType::HTTPS | RecordType::SVCB | RecordType::TLSA => Color::Cyan,
        RecordType::MX => Color::Yellow,
        RecordType::SOA => Color::Green,
        RecordType::NAPTR | RecordType::SRV => Color::Red,
        RecordType::OPENPGPKEY | RecordType::TXT => Color::Magenta,
        RecordType::SSHFP => Color::Blue,
        _ => Color::White,
    }
}

pub const fn record_type_is_bold(rt: RecordType) -> bool {
    matches!(
        rt,
        RecordType::A
            | RecordType::AAAA
            | RecordType::NULL
            | RecordType::HINFO
            | RecordType::HTTPS
            | RecordType::SVCB
            | RecordType::TLSA
            | RecordType::NAPTR
            | RecordType::OPENPGPKEY
            | RecordType::SSHFP
    )
}

pub const fn record_type_style(rt: RecordType) -> Style {
    let s = Style::new().fg(record_type_color(rt));
    if record_type_is_bold(rt) {
        s.bold()
    } else {
        s
    }
}

pub static A: Style = record_type_style(RecordType::A);
pub static AAAA: Style = record_type_style(RecordType::AAAA);
pub static CAA: Style = record_type_style(RecordType::CAA);
pub static DNSSEC: Style = record_type_style(RecordType::DNSKEY);
pub static HINFO: Style = record_type_style(RecordType::HINFO);
pub static MX: Style = record_type_style(RecordType::MX);
pub static NAME: Style = record_type_style(RecordType::CNAME);
pub static NAPTR: Style = record_type_style(RecordType::NAPTR);
pub static OPENPGPKEY: Style = record_type_style(RecordType::OPENPGPKEY);
pub static SOA: Style = record_type_style(RecordType::SOA);
pub static SRV: Style = record_type_style(RecordType::SRV);
pub static SSHFP: Style = record_type_style(RecordType::SSHFP);
pub static SVCB: Style = record_type_style(RecordType::SVCB);
pub static TLSA: Style = record_type_style(RecordType::TLSA);
pub static TXT: Style = record_type_style(RecordType::TXT);