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

use std::collections::HashMap;

use super::Key;

/// Helper for setting `locale` option
pub struct Locale<'a>(pub &'a str);
/// Helper for setting `default_key` option
pub struct DefaultKey<T>(pub T);
/// Helper for setting interpolated variables
pub struct Var<T, U>(pub T, pub U);
/// Helper for setting `count` option
pub struct Count(pub i32);

pub trait OptsPart<'a> {
    fn add_to(self, opts: Opts<'a>) -> Opts<'a>;
}

impl<'a> OptsPart<'a> for Locale<'a> {
    fn add_to(self, opts: Opts<'a>) -> Opts<'a> {
        opts.locale(self.0)
    }
}

impl<'a, T> OptsPart<'a> for DefaultKey<T>
where
    T: Into<Key<'a>>,
{
    fn add_to(self, opts: Opts<'a>) -> Opts<'a> {
        opts.default_key(self.0.into())
    }
}

impl<'a, T, U> OptsPart<'a> for Var<T, U>
where
    T: Into<String>,
    U: std::fmt::Display,
{
    fn add_to(self, opts: Opts<'a>) -> Opts<'a> {
        opts.var(self.0, self.1)
    }
}

impl<'a> OptsPart<'a> for Count {
    fn add_to(self, opts: Opts<'a>) -> Opts<'a> {
        opts.count(self.0)
    }
}

impl<'a, T> From<T> for Opts<'a>
where
    T: OptsPart<'a>,
{
    fn from(t: T) -> Self {
        t.add_to(Opts::default())
    }
}

impl<'a, T> OptsPart<'a> for (T,)
where
    T: OptsPart<'a>,
{
    fn add_to(self, opts: Opts<'a>) -> Opts<'a> {
        self.0.add_to(opts)
    }
}

impl<'a, T, U> OptsPart<'a> for (T, U)
where
    T: OptsPart<'a>,
    U: OptsPart<'a>,
{
    fn add_to(self, opts: Opts<'a>) -> Opts<'a> {
        self.1.add_to(self.0.add_to(opts))
    }
}

impl<'a, T, U, V> OptsPart<'a> for (T, U, V)
where
    T: OptsPart<'a>,
    U: OptsPart<'a>,
    U: OptsPart<'a>,
    V: OptsPart<'a>,
{
    fn add_to(self, opts: Opts<'a>) -> Opts<'a> {
        self.2.add_to(self.1.add_to(self.0.add_to(opts)))
    }
}

impl<'a, T, U, V, W> OptsPart<'a> for (T, U, V, W)
where
    T: OptsPart<'a>,
    U: OptsPart<'a>,
    U: OptsPart<'a>,
    V: OptsPart<'a>,
    W: OptsPart<'a>,
{
    fn add_to(self, opts: Opts<'a>) -> Opts<'a> {
        self.3.add_to(self.2.add_to(self.1.add_to(self.0.add_to(opts))))
    }
}

/// Options (optional) for the `translate` call
#[derive(Default)]
pub struct Opts<'a> {
    pub(crate) default_key: Option<Key<'a>>,
    pub(crate) vars: Option<HashMap<String, String>>,
    pub(crate) locale: Option<&'a str>,
    pub(crate) count: Option<i32>,
}

impl<'a> Opts<'a> {
    /// If the key does not exist, fallback to using another key.
    pub fn default_key<I: Into<Key<'a>>>(mut self, default_key: I) -> Self {
        self.default_key = Some(default_key.into());
        self
    }

    /// Set the locale for this `translate` call.
    pub fn locale(mut self, locale: &'a str) -> Self {
        self.locale = Some(locale);
        self
    }

    /// Set any variables to be interpolated.
    pub fn var<I: Into<String>, J: std::fmt::Display>(mut self, key: I, value: J) -> Self {
        let mut vars = self.vars.take().unwrap_or_else(HashMap::new);
        vars.insert(key.into(), value.to_string());
        self.vars = Some(vars);
        self
    }

    /// Set the `count` for this translation.
    ///
    /// Uses Rails style pluralization options: `zero`, `one`, `other`.
    pub fn count(mut self, count: i32) -> Self {
        self.count = Some(count);
        self.var("count", count)
    }
}

impl<'a> From<Option<Opts<'a>>> for Opts<'a> {
    fn from(t: Option<Opts<'a>>) -> Self {
        t.unwrap_or_else(Opts::default)
    }
}

impl<'a> From<()> for Opts<'a> {
    fn from(_: ()) -> Self {
        Opts::default()
    }
}