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

const AMOUNT: &str = "amount";
const SPL_TOKEN: &str = "spl-token";
const REFERENCE: &str = "reference";
const LABEL: &str = "label";
const MESSAGE: &str = "message";
const MEMO: &str = "memo";

#[derive(PartialEq, Clone, Hash, Eq)]
pub enum QueryParameter {
    Amount,
    SplToken,
    Reference,
    Label,
    Message,
    Memo,
    Extension(String),
}

impl ToString for QueryParameter {
    fn to_string(&self) -> String {
        self.as_str().to_string()
    }
}

impl QueryParameter {
    pub fn as_str(&self) -> &str {
        match self {
            Self::Amount => AMOUNT,
            Self::SplToken => SPL_TOKEN,
            Self::Reference => REFERENCE,
            Self::Label => LABEL,
            Self::Message => MESSAGE,
            Self::Memo => MEMO,
            Self::Extension(extension) => extension,
        }
    }
    pub fn allows_multiple(&self) -> bool {
        matches!(
            self,
            QueryParameter::Reference | QueryParameter::Extension(_)
        )
    }
}

impl From<&str> for QueryParameter {
    fn from(token: &str) -> QueryParameter {
        match token {
            AMOUNT => QueryParameter::Amount,
            SPL_TOKEN => QueryParameter::SplToken,
            REFERENCE => QueryParameter::Reference,
            LABEL => QueryParameter::Label,
            MESSAGE => QueryParameter::Message,
            MEMO => QueryParameter::Memo,
            _ => QueryParameter::Extension(token.to_string()),
        }
    }
}