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

//!
//! Module containing a helper [`Encoding`] enum use in RPC server constructors.
//!

use crate::error::Error;
use serde::{Deserialize, Serialize};
use std::{
    fmt::{Debug, Display, Formatter},
    str::FromStr,
};
use wasm_bindgen::prelude::*;

/// RPC protocol encoding: `Borsh` or `SerdeJson`
#[wasm_bindgen]
#[derive(Debug, Clone, Copy, Serialize, Deserialize, Hash, Eq, PartialEq)]
#[serde(rename_all = "kebab-case")]
pub enum Encoding {
    Borsh = 0,
    SerdeJson = 1,
}

impl Display for Encoding {
    #[inline]
    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
        let s = match self {
            Encoding::Borsh => "Borsh",
            Encoding::SerdeJson => "JSON",
        };
        f.write_str(s)
    }
}

impl FromStr for Encoding {
    type Err = Error;
    fn from_str(s: &str) -> Result<Self, Self::Err> {
        match s.to_lowercase().as_str() {
            "borsh" => Ok(Encoding::Borsh),
            "json" => Ok(Encoding::SerdeJson),
            "serde-json" => Ok(Encoding::SerdeJson),
            _ => Err(Error::Encoding(
                "invalid encoding: {s} (must be: 'borsh' or 'json')".to_string(),
            )),
        }
    }
}

impl TryFrom<u8> for Encoding {
    type Error = Error;
    fn try_from(value: u8) -> Result<Self, Self::Error> {
        match value {
            0 => Ok(Encoding::Borsh),
            1 => Ok(Encoding::SerdeJson),
            _ => Err(Error::Encoding(
                "invalid encoding: {value} (must be: Encoding.Borsh (0) or Encoding.SerdeJson (1))"
                    .to_string(),
            )),
        }
    }
}

impl TryFrom<JsValue> for Encoding {
    type Error = Error;
    fn try_from(value: JsValue) -> Result<Self, Self::Error> {
        if let Some(v) = value.as_f64() {
            Ok(Encoding::try_from(v as u8)?)
        } else if let Some(string) = value.as_string() {
            Encoding::from_str(&string)
        } else {
            Err(Error::Encoding(
                "invalid encoding value: {value:?}".to_string(),
            ))
        }
    }
}

const ENCODING: [Encoding; 2] = [Encoding::Borsh, Encoding::SerdeJson];

impl Encoding {
    pub fn iter() -> impl Iterator<Item = &'static Encoding> {
        ENCODING.iter()
    }
}