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

use crate::{key::Key, uref::URef};

#[derive(Debug, Clone, PartialEq, Eq)]
pub enum ContractRef {
    Hash([u8; 32]),
    URef(URef),
}

impl ContractRef {
    pub fn into_uref(self) -> Option<URef> {
        match self {
            ContractRef::URef(ret) => Some(ret),
            _ => None,
        }
    }
}

impl From<ContractRef> for Key {
    fn from(contract_ref: ContractRef) -> Self {
        match contract_ref {
            ContractRef::Hash(h) => Key::Hash(h),
            ContractRef::URef(uref) => uref.into(),
        }
    }
}