minidom/
prefixes.rs

1// Copyright (c) 2020 Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
2// Copyright (c) 2020 Astro <astro@spaceboyz.net>
3// Copyright (c) 2020 Maxime “pep” Buquet <pep@bouah.net>
4// Copyright (c) 2020 Xidorn Quan <me@upsuper.org>
5//
6// This Source Code Form is subject to the terms of the Mozilla Public
7// License, v. 2.0. If a copy of the MPL was not distributed with this
8// file, You can obtain one at http://mozilla.org/MPL/2.0/.
9
10use core::fmt;
11use std::collections::BTreeMap;
12
13pub type Prefix = Option<String>;
14pub type Namespace = String;
15
16#[derive(Clone, Default, PartialEq, Eq)]
17pub struct Prefixes {
18    prefixes: BTreeMap<Prefix, Namespace>,
19}
20
21impl fmt::Debug for Prefixes {
22    // TODO: Fix end character
23    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
24        write!(f, "Prefixes(")?;
25        for (prefix, namespace) in &self.prefixes {
26            write!(
27                f,
28                "xmlns{}={:?} ",
29                match prefix {
30                    None => String::new(),
31                    Some(prefix) => format!(":{}", prefix),
32                },
33                namespace
34            )?;
35        }
36        write!(f, ")")
37    }
38}
39
40impl Prefixes {
41    pub fn declared_prefixes(&self) -> &BTreeMap<Prefix, Namespace> {
42        &self.prefixes
43    }
44
45    pub fn get(&self, prefix: &Prefix) -> Option<&Namespace> {
46        self.prefixes.get(prefix)
47    }
48
49    pub(crate) fn insert<S: Into<Namespace>>(&mut self, prefix: Prefix, namespace: S) {
50        self.prefixes.insert(prefix, namespace.into());
51    }
52}
53
54impl From<BTreeMap<Prefix, Namespace>> for Prefixes {
55    fn from(prefixes: BTreeMap<Prefix, Namespace>) -> Self {
56        Prefixes { prefixes }
57    }
58}
59
60impl From<Option<String>> for Prefixes {
61    fn from(namespace: Option<String>) -> Self {
62        match namespace {
63            None => Self::default(),
64            Some(namespace) => Self::from(namespace),
65        }
66    }
67}
68
69impl From<Namespace> for Prefixes {
70    fn from(namespace: Namespace) -> Self {
71        let mut prefixes = BTreeMap::new();
72        prefixes.insert(None, namespace);
73
74        Prefixes { prefixes }
75    }
76}
77
78impl From<(Prefix, Namespace)> for Prefixes {
79    fn from(prefix_namespace: (Prefix, Namespace)) -> Self {
80        let (prefix, namespace) = prefix_namespace;
81        let mut prefixes = BTreeMap::new();
82        prefixes.insert(prefix, namespace);
83
84        Prefixes { prefixes }
85    }
86}
87
88impl From<(String, String)> for Prefixes {
89    fn from(prefix_namespace: (String, String)) -> Self {
90        let (prefix, namespace) = prefix_namespace;
91        Self::from((Some(prefix), namespace))
92    }
93}