solcandy 0.1.0

Useful utils for working with Metaplex candy machines(Solana blockchain).
Documentation
use crate::error::SolcandyError;
use crate::programs::candy_v2_program;
use solana_program::pubkey::Pubkey;
use std::str::FromStr;

/// Version of candy machine, since metaplex introduced new one.
#[derive(Copy, Clone, Debug, PartialOrd, PartialEq)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
pub enum CandyVersion {
    #[cfg_attr(feature = "serde", serde(alias = "v1"), serde(alias = "1"))]
    V1,

    #[cfg_attr(feature = "serde", serde(alias = "v2"), serde(alias = "2"))]
    V2,
}

impl FromStr for CandyVersion {
    type Err = SolcandyError;

    fn from_str(s: &str) -> Result<Self, Self::Err> {
        match s {
            "1" | "v1" | "V1" => Ok(CandyVersion::V1),
            "2" | "v2" | "V2" => Ok(CandyVersion::V2),
            _ => Err(SolcandyError::InvalidCandyVersion),
        }
    }
}

#[derive(Debug, Clone, Copy, PartialOrd, PartialEq)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
pub struct CandyMachine {
    key: Pubkey,
    version: CandyVersion,
}

impl CandyMachine {
    pub fn new(key: &Pubkey, version: CandyVersion) -> Self {
        Self { key: *key, version }
    }

    pub fn key(&self) -> &Pubkey {
        &self.key
    }

    pub fn version(&self) -> CandyVersion {
        self.version
    }

    pub fn find_creator(&self) -> Pubkey {
        match self.version {
            CandyVersion::V1 => self.key,
            CandyVersion::V2 => {
                Pubkey::find_program_address(
                    &[b"candy_machine", &self.key().to_bytes()],
                    &candy_v2_program(),
                )
                .0
            }
        }
    }
}

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

    #[cfg(feature = "serde")]
    #[test]
    fn it_deserializes_candy_version() {
        let value = json!("1");
        let version: CandyVersion = serde_json::from_value(value).unwrap();
        assert_eq!(version, CandyVersion::V1)
    }
}