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

use core::{
    borrow::Borrow,
    fmt,
    hash::{BuildHasher, Hash},
};
use std::collections::{BTreeMap, HashMap};

use crate::{FormatKey, FormatKeyError};

impl<K, V, S> FormatKey for HashMap<K, V, S>
where
    K: Borrow<str> + Eq + Hash,
    V: fmt::Display,
    S: BuildHasher,
{
    fn fmt(&self, key: &str, f: &mut fmt::Formatter<'_>) -> Result<(), FormatKeyError> {
        match self.get(key) {
            Some(v) => v.fmt(f).map_err(FormatKeyError::Fmt),
            None => Err(FormatKeyError::UnknownKey),
        }
    }
}

impl<K, V> FormatKey for BTreeMap<K, V>
where
    K: Borrow<str> + Ord,
    V: fmt::Display,
{
    fn fmt(&self, key: &str, f: &mut fmt::Formatter<'_>) -> Result<(), FormatKeyError> {
        match self.get(key) {
            Some(v) => v.fmt(f).map_err(FormatKeyError::Fmt),
            None => Err(FormatKeyError::UnknownKey),
        }
    }
}