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

use std::collections::HashMap;

use crate::*;

/// Gives access to the query string restricting the view to unique keys.
///
/// Dereferences to `HashMap<String, String>` for easy key/value manipulation.
///
/// # Example
///
/// ```rust
/// use url2::prelude::*;
///
/// let mut url = Url2::parse("https://test.com/?a=1&b=2");
///
/// url.query_unique().remove("b");
///
/// assert_eq!("https://test.com/?a=1", url.as_str());
/// ```
pub struct Url2QueryUnique<'lt> {
    pub(crate) url_ref: &'lt mut Url2,
}

impl<'lt> Url2QueryUnique<'lt> {
    /// builder-style helper for chaining
    ///
    /// # Example
    ///
    /// ```rust
    /// use url2::prelude::*;
    ///
    /// let mut url = Url2::default();
    /// url.query_unique()
    ///     .set_pair("a", "1")
    ///     .set_pair("b", "2");
    ///
    /// assert!(
    ///     "none:?a=1&b=2" == url.as_str() ||
    ///     "none:?b=2&a=1" == url.as_str()
    /// );
    /// ```
    pub fn set_pair(self, name: &str, value: &str) -> Self {
        self.url_ref
            .unique_cache
            .as_mut()
            .unwrap()
            .insert(name.to_string(), value.to_string());
        self
    }
}

impl<'lt> Drop for Url2QueryUnique<'lt> {
    fn drop(&mut self) {
        self.url_ref.priv_sync_query_unique_cache();
    }
}

impl<'lt> std::ops::Deref for Url2QueryUnique<'lt> {
    type Target = HashMap<String, String>;

    fn deref(&self) -> &Self::Target {
        self.url_ref.unique_cache.as_ref().unwrap()
    }
}

impl<'lt> std::ops::DerefMut for Url2QueryUnique<'lt> {
    fn deref_mut(&mut self) -> &mut Self::Target {
        self.url_ref.unique_cache.as_mut().unwrap()
    }
}