sui-jsonrpc 0.25.0

use std::fmt::Display;
use std::str::FromStr;

use serde::{Deserialize, Deserializer, Serialize, Serializer};
use serde_with::{DeserializeAs, DisplayFromStr, SerializeAs, serde_as};
use sui_sdk_types::GasCostSummary;

fn decode_base58(base58: &str) -> Result<Vec<u8>, bs58::decode::Error> {
    bs58::decode(base58).into_vec()
}

/// Convenience method for decoding base64 bytes the way Sui expects.
fn decode_base64_default(base64: &str) -> Result<Vec<u8>, base64ct::Error> {
    // let mut result: Vec<u8> = Vec::<u8>::from(base64);
    // let s = <base64ct::Base64 as base64ct::Encoding>::decode_in_place(&mut result)?;
    // let len = s.len();
    // result.truncate(len);
    // Ok(result)

    <base64ct::Base64 as base64ct::Encoding>::decode_vec(base64)
}

/// Convenience method for encoding bytes to base64 the way Sui expects.
pub(crate) fn encode_base64_default(bytes: impl AsRef<[u8]>) -> String {
    <base64ct::Base64 as base64ct::Encoding>::encode_string(bytes.as_ref())
}

// =============================================================================
//  BigInt
// =============================================================================

#[serde_as]
#[derive(Debug, Serialize, Deserialize, Clone, PartialEq, Eq, Copy)]
pub struct BigInt<T>(#[serde_as(as = "serde_with::DisplayFromStr")] T)
where
    T: Display + FromStr,
    <T as FromStr>::Err: Display;

impl<T> BigInt<T>
where
    T: Display + FromStr,
    <T as FromStr>::Err: Display,
{
    pub fn into_inner(self) -> T {
        self.0
    }
}

impl<T> SerializeAs<T> for BigInt<T>
where
    T: Display + FromStr + Copy,
    <T as FromStr>::Err: Display,
{
    fn serialize_as<S>(value: &T, serializer: S) -> Result<S::Ok, S::Error>
    where
        S: Serializer,
    {
        Self(*value).serialize(serializer)
    }
}

impl<'de, T> DeserializeAs<'de, T> for BigInt<T>
where
    T: Display + FromStr + Copy,
    <T as FromStr>::Err: Display,
{
    fn deserialize_as<D>(deserializer: D) -> Result<T, D::Error>
    where
        D: Deserializer<'de>,
    {
        Ok(*Self::deserialize(deserializer)?)
    }
}

impl<T> From<T> for BigInt<T>
where
    T: Display + FromStr,
    <T as FromStr>::Err: Display,
{
    fn from(v: T) -> Self {
        Self(v)
    }
}

impl<T> std::ops::Deref for BigInt<T>
where
    T: Display + FromStr,
    <T as FromStr>::Err: Display,
{
    type Target = T;

    fn deref(&self) -> &Self::Target {
        &self.0
    }
}

impl<T> Display for BigInt<T>
where
    T: Display + FromStr,
    <T as FromStr>::Err: Display,
{
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "{}", self.0)
    }
}

// =============================================================================
//  Base64orBase58
// =============================================================================

/// Always serialize as base64, but deserialize from either Base64 or Base58
pub(crate) struct Base64orBase58;

impl<T> SerializeAs<T> for Base64orBase58
where
    T: AsRef<[u8]>,
{
    fn serialize_as<S>(value: &T, serializer: S) -> Result<S::Ok, S::Error>
    where
        S: Serializer,
    {
        let encoded_string = encode_base64_default(value);
        encoded_string.serialize(serializer)
    }
}

impl<'de> DeserializeAs<'de, Vec<u8>> for Base64orBase58 {
    fn deserialize_as<D>(deserializer: D) -> Result<Vec<u8>, D::Error>
    where
        D: Deserializer<'de>,
    {
        let s = String::deserialize(deserializer)?;

        decode_base64_default(&s)
            .or_else(|_| decode_base58(&s))
            .map_err(|_| serde::de::Error::custom("Deserialization failed"))
    }
}

#[serde_as]
#[derive(Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub(crate) struct GasCostSummaryJson {
    #[serde_as(as = "DisplayFromStr")]
    computation_cost: u64,
    #[serde_as(as = "DisplayFromStr")]
    storage_cost: u64,
    #[serde_as(as = "DisplayFromStr")]
    storage_rebate: u64,
    #[serde_as(as = "DisplayFromStr")]
    non_refundable_storage_fee: u64,
}

impl From<GasCostSummaryJson> for GasCostSummary {
    fn from(
        GasCostSummaryJson {
            computation_cost,
            storage_cost,
            storage_rebate,
            non_refundable_storage_fee,
        }: GasCostSummaryJson,
    ) -> Self {
        Self {
            computation_cost,
            storage_cost,
            storage_rebate,
            non_refundable_storage_fee,
        }
    }
}

impl From<GasCostSummary> for GasCostSummaryJson {
    fn from(
        GasCostSummary {
            computation_cost,
            storage_cost,
            storage_rebate,
            non_refundable_storage_fee,
        }: GasCostSummary,
    ) -> Self {
        Self {
            computation_cost,
            storage_cost,
            storage_rebate,
            non_refundable_storage_fee,
        }
    }
}

#[cfg(test)]
mod test {
    use super::*;

    #[serde_as]
    #[derive(Debug, Deserialize, Serialize, PartialEq, Eq)]
    #[serde(rename_all = "camelCase")]
    struct Bcs {
        #[serde_as(as = "Base64orBase58")]
        bcs: Vec<u8>,
    }

    #[test]
    fn new_bcs_format() {
        let bytes = vec![0, 1, 2, 3, 4];
        let untagged_base58 = r#"{"bcs":"12VfUX"}"#;
        let tagged_base58 = r#"{"bcsEncoding":"base58","bcs":"12VfUX"}"#;
        let tagged_base64 = r#"{"bcsEncoding":"base64","bcs":"AAECAwQ="}"#;

        println!(
            "{}",
            serde_json::to_string(&Bcs { bcs: bytes.clone() }).unwrap()
        );

        assert_eq!(
            bytes,
            serde_json::from_str::<Bcs>(untagged_base58).unwrap().bcs
        );
        assert_eq!(
            bytes,
            serde_json::from_str::<Bcs>(tagged_base58).unwrap().bcs
        );
        assert_eq!(
            bytes,
            serde_json::from_str::<Bcs>(tagged_base64).unwrap().bcs
        );

        // Roundtrip base64
        let name = serde_json::from_str::<Bcs>(tagged_base64).unwrap();
        let json = serde_json::to_string(&name).unwrap();
        let from_json = serde_json::from_str::<Bcs>(&json).unwrap();
        assert_eq!(name, from_json);

        // Roundtrip base58
        let name = serde_json::from_str::<Bcs>(tagged_base58).unwrap();
        let json = serde_json::to_string(&name).unwrap();
        let from_json = serde_json::from_str::<Bcs>(&json).unwrap();
        assert_eq!(name, from_json);
    }
}