cid 0.5.0

use std::convert::TryFrom;
use std::fmt;

use multibase::Base;
use multihash::{Code, MultihashGeneric, MultihashRefGeneric};
use unsigned_varint::{decode as varint_decode, encode as varint_encode};

use crate::codec::Codec;
use crate::error::{Error, Result};
use crate::version::Version;

/// A CID with the default Multihash code table
pub type Cid = CidGeneric<Codec, Code>;

/// Representation of a CID.
///
/// Usually you would use `Cid` instead, unless you have a custom Multihash code table
#[derive(PartialEq, Eq, Clone, Debug, PartialOrd, Ord)]
pub struct CidGeneric<C, H>
where
    C: Into<u64> + TryFrom<u64> + Copy,
    H: Into<u64> + TryFrom<u64> + Copy,
{
    /// The version of CID.
    version: Version,
    /// The codec of CID.
    codec: C,
    /// The multihash of CID.
    hash: MultihashGeneric<H>,
}

impl<C, H> CidGeneric<C, H>
where
    C: Into<u64> + TryFrom<u64> + Copy,
    <C as TryFrom<u64>>::Error: fmt::Debug,
    H: Into<u64> + TryFrom<u64> + Copy,
    <H as TryFrom<u64>>::Error: fmt::Debug,
{
    /// Create a new CIDv0.
    pub fn new_v0(hash: MultihashGeneric<H>) -> Result<Self> {
        if hash.algorithm().into() != u64::from(Code::Sha2_256) {
            return Err(Error::InvalidCidV0Multihash);
        }
        Ok(Self {
            version: Version::V0,
            // Convert the code of `DagProtobuf` into the given code table
            codec: C::try_from(Codec::DagProtobuf.into()).map_err(|_| Error::UnknownCodec)?,
            hash,
        })
    }

    /// Create a new CIDv1.
    pub fn new_v1(codec: C, hash: MultihashGeneric<H>) -> Self {
        Self {
            version: Version::V1,
            codec,
            hash,
        }
    }

    /// Create a new CID.
    pub fn new(version: Version, codec: C, hash: MultihashGeneric<H>) -> Result<Self> {
        match version {
            Version::V0 => {
                if codec.into() != u64::from(Codec::DagProtobuf) {
                    return Err(Error::InvalidCidV0Codec);
                }
                Self::new_v0(hash)
            }
            Version::V1 => Ok(Self::new_v1(codec, hash)),
        }
    }

    /// Returns the cid version.
    pub fn version(&self) -> Version {
        self.version
    }

    /// Returns the cid codec.
    pub fn codec(&self) -> C {
        self.codec
    }

    /// Returns the cid multihash.
    pub fn hash(&self) -> MultihashRefGeneric<H> {
        self.hash.as_ref()
    }

    fn to_string_v0(&self) -> String {
        Base::Base58Btc.encode(self.hash.as_bytes())
    }

    fn to_string_v1(&self) -> String {
        multibase::encode(Base::Base32Lower, self.to_bytes().as_slice())
    }

    fn to_bytes_v0(&self) -> Vec<u8> {
        self.hash.to_vec()
    }

    fn to_bytes_v1(&self) -> Vec<u8> {
        let mut res = Vec::with_capacity(16);

        let mut buf = varint_encode::u64_buffer();
        let version = varint_encode::u64(self.version.into(), &mut buf);
        res.extend_from_slice(version);
        let mut buf = varint_encode::u64_buffer();
        let codec = varint_encode::u64(self.codec.into(), &mut buf);
        res.extend_from_slice(codec);
        res.extend_from_slice(&self.hash);

        res
    }

    /// Convert CID to encoded bytes.
    pub fn to_bytes(&self) -> Vec<u8> {
        match self.version {
            Version::V0 => self.to_bytes_v0(),
            Version::V1 => self.to_bytes_v1(),
        }
    }
}

#[allow(clippy::derive_hash_xor_eq)]
impl<C, H> std::hash::Hash for CidGeneric<C, H>
where
    C: Into<u64> + TryFrom<u64> + Copy,
    <C as TryFrom<u64>>::Error: fmt::Debug,
    H: Into<u64> + TryFrom<u64> + Copy,
    <H as TryFrom<u64>>::Error: fmt::Debug,
{
    fn hash<T: std::hash::Hasher>(&self, state: &mut T) {
        self.to_bytes().hash(state);
    }
}

impl<C, H> std::fmt::Display for CidGeneric<C, H>
where
    C: Into<u64> + TryFrom<u64> + Copy,
    <C as TryFrom<u64>>::Error: fmt::Debug,
    H: Into<u64> + TryFrom<u64> + Copy,
    <H as TryFrom<u64>>::Error: fmt::Debug,
{
    fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
        let output = match self.version {
            Version::V0 => self.to_string_v0(),
            Version::V1 => self.to_string_v1(),
        };
        write!(f, "{}", output)
    }
}

impl<C, H> std::str::FromStr for CidGeneric<C, H>
where
    C: Into<u64> + TryFrom<u64> + Copy,
    <C as TryFrom<u64>>::Error: fmt::Debug,
    H: Into<u64> + TryFrom<u64> + Copy,
    <H as TryFrom<u64>>::Error: fmt::Debug,
{
    type Err = Error;

    fn from_str(cid_str: &str) -> Result<Self> {
        CidGeneric::try_from(cid_str)
    }
}

impl<C, H> TryFrom<String> for CidGeneric<C, H>
where
    C: Into<u64> + TryFrom<u64> + Copy,
    <C as TryFrom<u64>>::Error: fmt::Debug,
    H: Into<u64> + TryFrom<u64> + Copy,
    <H as TryFrom<u64>>::Error: fmt::Debug,
{
    type Error = Error;

    fn try_from(cid_str: String) -> Result<Self> {
        Self::try_from(cid_str.as_str())
    }
}

impl<C, H> TryFrom<&str> for CidGeneric<C, H>
where
    C: Into<u64> + TryFrom<u64> + Copy,
    <C as TryFrom<u64>>::Error: fmt::Debug,
    H: Into<u64> + TryFrom<u64> + Copy,
    <H as TryFrom<u64>>::Error: fmt::Debug,
{
    type Error = Error;

    fn try_from(cid_str: &str) -> Result<Self> {
        static IPFS_DELIMETER: &str = "/ipfs/";

        let hash = match cid_str.find(IPFS_DELIMETER) {
            Some(index) => &cid_str[index + IPFS_DELIMETER.len()..],
            _ => cid_str,
        };

        if hash.len() < 2 {
            return Err(Error::InputTooShort);
        }

        let decoded = if Version::is_v0_str(hash) {
            Base::Base58Btc.decode(hash)?
        } else {
            let (_, decoded) = multibase::decode(hash)?;
            decoded
        };

        Self::try_from(decoded)
    }
}

impl<C, H> TryFrom<Vec<u8>> for CidGeneric<C, H>
where
    C: Into<u64> + TryFrom<u64> + Copy,
    <C as TryFrom<u64>>::Error: fmt::Debug,
    H: Into<u64> + TryFrom<u64> + Copy,
    <H as TryFrom<u64>>::Error: fmt::Debug,
{
    type Error = Error;

    fn try_from(bytes: Vec<u8>) -> Result<Self> {
        Self::try_from(bytes.as_slice())
    }
}

impl<C, H> TryFrom<&[u8]> for CidGeneric<C, H>
where
    C: Into<u64> + TryFrom<u64> + Copy,
    <C as TryFrom<u64>>::Error: fmt::Debug,
    H: Into<u64> + TryFrom<u64> + Copy,
    <H as TryFrom<u64>>::Error: fmt::Debug,
{
    type Error = Error;

    fn try_from(bytes: &[u8]) -> Result<Self> {
        if Version::is_v0_binary(bytes) {
            let mh = MultihashRefGeneric::from_slice(bytes)?.to_owned();
            CidGeneric::new_v0(mh)
        } else {
            let (raw_version, remain) = varint_decode::u64(&bytes)?;
            let version = Version::try_from(raw_version)?;

            let (raw_codec, hash) = varint_decode::u64(&remain)?;
            let codec = C::try_from(raw_codec).map_err(|_| Error::UnknownCodec)?;

            let mh = MultihashRefGeneric::from_slice(hash)?.to_owned();

            CidGeneric::new(version, codec, mh)
        }
    }
}

impl<C, H> From<&CidGeneric<C, H>> for CidGeneric<C, H>
where
    C: Into<u64> + TryFrom<u64> + Copy,
    <C as TryFrom<u64>>::Error: fmt::Debug,
    H: Into<u64> + TryFrom<u64> + Copy,
    <H as TryFrom<u64>>::Error: fmt::Debug,
{
    fn from(cid: &CidGeneric<C, H>) -> Self {
        cid.to_owned()
    }
}

impl<C, H> From<CidGeneric<C, H>> for Vec<u8>
where
    C: Into<u64> + TryFrom<u64> + Copy,
    <C as TryFrom<u64>>::Error: fmt::Debug,
    H: Into<u64> + TryFrom<u64> + Copy,
    <H as TryFrom<u64>>::Error: fmt::Debug,
{
    fn from(cid: CidGeneric<C, H>) -> Self {
        cid.to_bytes()
    }
}

impl<C, H> From<CidGeneric<C, H>> for String
where
    C: Into<u64> + TryFrom<u64> + Copy,
    <C as TryFrom<u64>>::Error: fmt::Debug,
    H: Into<u64> + TryFrom<u64> + Copy,
    <H as TryFrom<u64>>::Error: fmt::Debug,
{
    fn from(cid: CidGeneric<C, H>) -> Self {
        cid.to_string()
    }
}