dpp/identity/state_transition/asset_lock_proof/

mod.rs

use std::convert::{TryFrom, TryInto};

use dashcore::{OutPoint, Transaction};

use serde::{Deserialize, Deserializer, Serialize};

use bincode::{Decode, Encode};

pub use instant::*;
use platform_value::Value;
#[cfg(feature = "validation")]
use platform_version::version::PlatformVersion;
use serde::de::Error;

use crate::identity::state_transition::asset_lock_proof::chain::ChainAssetLockProof;
use crate::prelude::Identifier;
#[cfg(feature = "validation")]
use crate::validation::SimpleConsensusValidationResult;
use crate::{ProtocolError, SerdeParsingError};

pub mod chain;
pub mod instant;
pub mod validate_asset_lock_transaction_structure;

// TODO: Serialization with bincode
// TODO: Consider use Box for InstantAssetLockProof
#[derive(Clone, Debug, Eq, PartialEq, Serialize, Encode, Decode)]
#[serde(untagged)]
#[allow(clippy::large_enum_variant)]
pub enum AssetLockProof {
    Instant(#[bincode(with_serde)] InstantAssetLockProof),
    Chain(#[bincode(with_serde)] ChainAssetLockProof),
}

#[derive(Deserialize)]
#[serde(untagged)]
enum RawAssetLockProof {
    Instant(RawInstantLockProof),
    Chain(ChainAssetLockProof),
}

impl TryFrom<RawAssetLockProof> for AssetLockProof {
    type Error = ProtocolError;

    fn try_from(value: RawAssetLockProof) -> Result<Self, Self::Error> {
        match value {
            RawAssetLockProof::Instant(raw_instant_lock) => {
                let instant_lock = raw_instant_lock.try_into()?;

                Ok(AssetLockProof::Instant(instant_lock))
            }
            RawAssetLockProof::Chain(chain) => Ok(AssetLockProof::Chain(chain)),
        }
    }
}

impl<'de> Deserialize<'de> for AssetLockProof {
    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
    where
        D: Deserializer<'de>,
    {
        // Try to parse into IS Lock
        // let maybe_is_lock = RawInstantLock::deserialize(&deserializer);
        //
        // if let Ok(raw_instant_lock) = maybe_is_lock {
        //     let instant_lock = raw_instant_lock.try_into()
        //         .map_err(|e: ProtocolError| D::Error::custom(e.to_string()))?;
        //
        //     return Ok(AssetLockProof::Instant(instant_lock))
        // };
        //
        //
        // ChainAssetLockProof::deserialize(deserializer)
        //     .map(|chain| AssetLockProof::Chain(chain))
        // // Try to parse into chain lock

        let raw = RawAssetLockProof::deserialize(deserializer)?;
        raw.try_into().map_err(|e: ProtocolError| {
            D::Error::custom(format!(
                "expected to be able to deserialize asset lock proof: {}",
                e
            ))
        })
    }
}

impl Default for AssetLockProof {
    fn default() -> Self {
        Self::Instant(InstantAssetLockProof::default())
    }
}

impl AsRef<AssetLockProof> for AssetLockProof {
    fn as_ref(&self) -> &AssetLockProof {
        self
    }
}
//
// impl Serialize for AssetLockProof {
//     fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
//     where
//         S: Serializer,
//     {
//         match self {
//             AssetLockProof::Instant(instant_proof) => instant_proof.serialize(serializer),
//             AssetLockProof::Chain(chain) => chain.serialize(serializer),
//         }
//     }
// }
//
// impl<'de> Deserialize<'de> for AssetLockProof {
//     fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
//     where
//         D: Deserializer<'de>,
//     {
//         let value = platform_value::Value::deserialize(deserializer)?;
//
//         let proof_type_int: u8 = value
//             .get_integer("type")
//             .map_err(|e| D::Error::custom(e.to_string()))?;
//         let proof_type = AssetLockProofType::try_from(proof_type_int)
//             .map_err(|e| D::Error::custom(e.to_string()))?;
//
//         match proof_type {
//             AssetLockProofType::Instant => Ok(Self::Instant(
//                 platform_value::from_value(value).map_err(|e| D::Error::custom(e.to_string()))?,
//             )),
//             AssetLockProofType::Chain => Ok(Self::Chain(
//                 platform_value::from_value(value).map_err(|e| D::Error::custom(e.to_string()))?,
//             )),
//         }
//     }
// }

pub enum AssetLockProofType {
    Instant = 0,
    Chain = 1,
}

impl TryFrom<u8> for AssetLockProofType {
    type Error = SerdeParsingError;

    fn try_from(value: u8) -> Result<Self, Self::Error> {
        match value {
            0 => Ok(Self::Instant),
            1 => Ok(Self::Chain),
            _ => Err(SerdeParsingError::new("Unexpected asset lock proof type")),
        }
    }
}

impl TryFrom<u64> for AssetLockProofType {
    type Error = SerdeParsingError;

    fn try_from(value: u64) -> Result<Self, Self::Error> {
        match value {
            0 => Ok(Self::Instant),
            1 => Ok(Self::Chain),
            _ => Err(SerdeParsingError::new("Unexpected asset lock proof type")),
        }
    }
}

// TODO: Versioning
impl AssetLockProof {
    pub fn type_from_raw_value(value: &Value) -> Option<AssetLockProofType> {
        let proof_type_res = value.get_integer::<u8>("type");

        match proof_type_res {
            Ok(proof_type_int) => {
                let proof_type = AssetLockProofType::try_from(proof_type_int);
                proof_type.ok()
            }
            Err(_) => None,
        }
    }

    pub fn create_identifier(&self) -> Result<Identifier, ProtocolError> {
        match self {
            AssetLockProof::Instant(instant_proof) => instant_proof.create_identifier(),
            AssetLockProof::Chain(chain_proof) => Ok(chain_proof.create_identifier()),
        }
    }

    pub fn output_index(&self) -> u32 {
        match self {
            AssetLockProof::Instant(proof) => proof.output_index(),
            AssetLockProof::Chain(proof) => proof.out_point.vout,
        }
    }

    pub fn out_point(&self) -> Option<OutPoint> {
        match self {
            AssetLockProof::Instant(proof) => proof.out_point(),
            AssetLockProof::Chain(proof) => Some(proof.out_point),
        }
    }

    pub fn transaction(&self) -> Option<&Transaction> {
        match self {
            AssetLockProof::Instant(is_lock) => Some(is_lock.transaction()),
            AssetLockProof::Chain(_chain_lock) => None,
        }
    }

    pub fn to_raw_object(&self) -> Result<Value, ProtocolError> {
        match self {
            AssetLockProof::Instant(is) => {
                platform_value::to_value(is).map_err(ProtocolError::ValueError)
            }
            AssetLockProof::Chain(cl) => {
                platform_value::to_value(cl).map_err(ProtocolError::ValueError)
            }
        }
    }

    /// Validate the structure of the asset lock proof
    #[cfg(feature = "validation")]
    pub fn validate_structure(
        &self,
        platform_version: &PlatformVersion,
    ) -> Result<SimpleConsensusValidationResult, ProtocolError> {
        match self {
            AssetLockProof::Instant(proof) => proof.validate_structure(platform_version),
            AssetLockProof::Chain(_) => Ok(SimpleConsensusValidationResult::default()),
        }
    }
}

impl TryFrom<&Value> for AssetLockProof {
    type Error = ProtocolError;

    fn try_from(value: &Value) -> Result<Self, Self::Error> {
        //this is a complete hack for the moment
        //todo: replace with
        //  from_value(value.clone()).map_err(ProtocolError::ValueError)
        let proof_type_int: Option<u8> = value
            .get_optional_integer("type")
            .map_err(ProtocolError::ValueError)?;
        if let Some(proof_type_int) = proof_type_int {
            let proof_type = AssetLockProofType::try_from(proof_type_int)?;

            match proof_type {
                AssetLockProofType::Instant => Ok(Self::Instant(value.clone().try_into()?)),
                AssetLockProofType::Chain => Ok(Self::Chain(value.clone().try_into()?)),
            }
        } else {
            let map = value.as_map().ok_or(ProtocolError::DecodingError(
                "error decoding asset lock proof".to_string(),
            ))?;
            let (key, asset_lock_value) = map.first().ok_or(ProtocolError::DecodingError(
                "error decoding asset lock proof as it was empty".to_string(),
            ))?;
            match key.as_str().ok_or(ProtocolError::DecodingError(
                "error decoding asset lock proof".to_string(),
            ))? {
                "Instant" => Ok(Self::Instant(asset_lock_value.clone().try_into()?)),
                "Chain" => Ok(Self::Chain(asset_lock_value.clone().try_into()?)),
                _ => Err(ProtocolError::DecodingError(
                    "error decoding asset lock proof".to_string(),
                )),
            }
        }
    }
}

impl TryFrom<Value> for AssetLockProof {
    type Error = ProtocolError;

    fn try_from(value: Value) -> Result<Self, Self::Error> {
        let proof_type_int: Option<u8> = value
            .get_optional_integer("type")
            .map_err(ProtocolError::ValueError)?;
        if let Some(proof_type_int) = proof_type_int {
            let proof_type = AssetLockProofType::try_from(proof_type_int)?;

            match proof_type {
                AssetLockProofType::Instant => Ok(Self::Instant(value.try_into()?)),
                AssetLockProofType::Chain => Ok(Self::Chain(value.try_into()?)),
            }
        } else {
            let map = value.as_map().ok_or(ProtocolError::DecodingError(
                "error decoding asset lock proof".to_string(),
            ))?;
            let (key, asset_lock_value) = map.first().ok_or(ProtocolError::DecodingError(
                "error decoding asset lock proof as it was empty".to_string(),
            ))?;
            match key.as_str().ok_or(ProtocolError::DecodingError(
                "error decoding asset lock proof".to_string(),
            ))? {
                "Instant" => Ok(Self::Instant(asset_lock_value.clone().try_into()?)),
                "Chain" => Ok(Self::Chain(asset_lock_value.clone().try_into()?)),
                _ => Err(ProtocolError::DecodingError(
                    "error decoding asset lock proof".to_string(),
                )),
            }
        }
    }
}

impl TryInto<Value> for AssetLockProof {
    type Error = ProtocolError;

    fn try_into(self) -> Result<Value, Self::Error> {
        match self {
            AssetLockProof::Instant(instant_proof) => {
                platform_value::to_value(instant_proof).map_err(ProtocolError::ValueError)
            }
            AssetLockProof::Chain(chain_proof) => {
                platform_value::to_value(chain_proof).map_err(ProtocolError::ValueError)
            }
        }
    }
}

impl TryInto<Value> for &AssetLockProof {
    type Error = ProtocolError;

    fn try_into(self) -> Result<Value, Self::Error> {
        match self {
            AssetLockProof::Instant(instant_proof) => {
                platform_value::to_value(instant_proof).map_err(ProtocolError::ValueError)
            }
            AssetLockProof::Chain(chain_proof) => {
                platform_value::to_value(chain_proof).map_err(ProtocolError::ValueError)
            }
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::identity::state_transition::asset_lock_proof::chain::ChainAssetLockProof;

    mod asset_lock_proof_type_try_from {
        use super::*;

        #[test]
        fn u8_instant_type() {
            let proof_type = AssetLockProofType::try_from(0u8).expect("should parse type 0");
            assert!(matches!(proof_type, AssetLockProofType::Instant));
        }

        #[test]
        fn u8_chain_type() {
            let proof_type = AssetLockProofType::try_from(1u8).expect("should parse type 1");
            assert!(matches!(proof_type, AssetLockProofType::Chain));
        }

        #[test]
        fn u8_invalid_type() {
            let result = AssetLockProofType::try_from(2u8);
            assert!(result.is_err());
        }

        #[test]
        fn u8_max_invalid_type() {
            let result = AssetLockProofType::try_from(255u8);
            assert!(result.is_err());
        }

        #[test]
        fn u64_instant_type() {
            let proof_type = AssetLockProofType::try_from(0u64).expect("should parse type 0");
            assert!(matches!(proof_type, AssetLockProofType::Instant));
        }

        #[test]
        fn u64_chain_type() {
            let proof_type = AssetLockProofType::try_from(1u64).expect("should parse type 1");
            assert!(matches!(proof_type, AssetLockProofType::Chain));
        }

        #[test]
        fn u64_invalid_type() {
            let result = AssetLockProofType::try_from(2u64);
            assert!(result.is_err());
        }

        #[test]
        fn u64_large_invalid_type() {
            let result = AssetLockProofType::try_from(u64::MAX);
            assert!(result.is_err());
        }
    }

    mod chain_asset_lock_proof {
        use super::*;

        fn make_chain_proof() -> ChainAssetLockProof {
            ChainAssetLockProof::new(100, [0xAB; 36])
        }

        #[test]
        fn chain_proof_construction() {
            let proof = ChainAssetLockProof::new(42, [0x01; 36]);
            assert_eq!(proof.core_chain_locked_height, 42);
        }

        #[test]
        fn chain_proof_create_identifier_deterministic() {
            let proof = make_chain_proof();
            let id1 = proof.create_identifier();
            let id2 = proof.create_identifier();
            assert_eq!(id1, id2);
        }

        #[test]
        fn different_outpoints_produce_different_identifiers() {
            let proof_a = ChainAssetLockProof::new(100, [0xAA; 36]);
            let proof_b = ChainAssetLockProof::new(100, [0xBB; 36]);
            assert_ne!(proof_a.create_identifier(), proof_b.create_identifier());
        }

        #[test]
        fn chain_proof_equality() {
            let a = ChainAssetLockProof::new(10, [0x01; 36]);
            let b = ChainAssetLockProof::new(10, [0x01; 36]);
            assert_eq!(a, b);
        }

        #[test]
        fn chain_proof_inequality_height() {
            let a = ChainAssetLockProof::new(10, [0x01; 36]);
            let b = ChainAssetLockProof::new(20, [0x01; 36]);
            assert_ne!(a, b);
        }
    }

    mod asset_lock_proof_methods {
        use super::*;

        fn make_chain_lock_proof() -> AssetLockProof {
            let chain_proof = ChainAssetLockProof::new(50, [0xCC; 36]);
            AssetLockProof::Chain(chain_proof)
        }

        #[test]
        fn default_is_instant() {
            let proof = AssetLockProof::default();
            assert!(matches!(proof, AssetLockProof::Instant(_)));
        }

        #[test]
        fn as_ref_returns_self() {
            let proof = make_chain_lock_proof();
            let reference: &AssetLockProof = proof.as_ref();
            assert_eq!(&proof, reference);
        }

        #[test]
        fn chain_proof_output_index() {
            let mut out_point_bytes = [0u8; 36];
            // Set vout (last 4 bytes in little-endian) to 3
            out_point_bytes[32] = 3;
            let chain_proof = ChainAssetLockProof::new(50, out_point_bytes);
            let proof = AssetLockProof::Chain(chain_proof);
            assert_eq!(proof.output_index(), 3);
        }

        #[test]
        fn chain_proof_out_point_is_some() {
            let proof = make_chain_lock_proof();
            assert!(proof.out_point().is_some());
        }

        #[test]
        fn chain_proof_transaction_is_none() {
            let proof = make_chain_lock_proof();
            assert!(proof.transaction().is_none());
        }

        #[test]
        fn chain_proof_to_raw_object() {
            let proof = make_chain_lock_proof();
            let result = proof.to_raw_object();
            assert!(result.is_ok());
        }

        #[test]
        fn chain_proof_create_identifier() {
            let proof = make_chain_lock_proof();
            let id = proof.create_identifier();
            assert!(id.is_ok());
        }
    }

    mod try_from_value {
        use super::*;

        #[test]
        fn chain_proof_value_round_trip() {
            let chain_proof = ChainAssetLockProof::new(100, [0x42; 36]);
            let proof = AssetLockProof::Chain(chain_proof);

            // Convert to Value
            let value: Value = (&proof).try_into().expect("should convert to Value");

            // Now try to read type from value
            let _type_from_value = AssetLockProof::type_from_raw_value(&value);
            // Chain proofs serialized via serde may or may not have "type" field depending
            // on the serialization format. The untagged format may not include it.
            // What matters is that the conversion itself works.

            // Convert from Value back - this tests the TryFrom<Value> path
            // with the untagged serde format
            let raw_value = proof.to_raw_object().expect("should convert to raw object");
            assert!(!raw_value.is_null());
        }

        #[test]
        fn type_from_raw_value_returns_none_for_missing_type() {
            let value = Value::Map(vec![]);
            let result = AssetLockProof::type_from_raw_value(&value);
            assert!(result.is_none());
        }

        #[test]
        fn try_from_empty_map_fails() {
            let value = Value::Map(vec![]);
            let result = AssetLockProof::try_from(&value);
            assert!(result.is_err());
        }

        #[test]
        fn try_from_value_with_unknown_key_fails() {
            let value = Value::Map(vec![(
                Value::Text("Unknown".to_string()),
                Value::Map(vec![]),
            )]);
            let result = AssetLockProof::try_from(&value);
            assert!(result.is_err());
        }
    }

    mod try_into_value {
        use super::*;

        #[test]
        fn chain_proof_try_into_value() {
            let chain_proof = ChainAssetLockProof::new(200, [0xDD; 36]);
            let proof = AssetLockProof::Chain(chain_proof);

            let value: Result<Value, ProtocolError> = proof.try_into();
            assert!(value.is_ok());
        }

        #[test]
        fn chain_proof_ref_try_into_value() {
            let chain_proof = ChainAssetLockProof::new(200, [0xDD; 36]);
            let proof = AssetLockProof::Chain(chain_proof);

            let value: Result<Value, ProtocolError> = (&proof).try_into();
            assert!(value.is_ok());
        }
    }
}