mod blake256;
mod cnaes;
mod hash_v2;
mod hash_v4;
mod slow_hash;
mod util;

use slow_hash::cn_slow_hash;

/// Calculates the `CryptoNight` v0 hash of buf.
pub fn cryptonight_hash_v0(buf: &[u8]) -> [u8; 32] {
    cn_slow_hash(buf, slow_hash::Variant::V0, 0)
}

#[derive(thiserror::Error, Debug, Copy, Clone, Eq, PartialEq)]
#[error("Data can't be hashed")]
pub struct DataCanNotBeHashed;

/// Calculates the `CryptoNight` v1 hash of buf.
///
/// This will return an error if buf is less than 43 bytes.
pub fn cryptonight_hash_v1(buf: &[u8]) -> Result<[u8; 32], DataCanNotBeHashed> {
    if buf.len() < 43 {
        return Err(DataCanNotBeHashed);
    }

    Ok(cn_slow_hash(buf, slow_hash::Variant::V1, 0))
}

/// Calculates the `CryptoNight` v2 hash of buf.
pub fn cryptonight_hash_v2(buf: &[u8]) -> [u8; 32] {
    cn_slow_hash(buf, slow_hash::Variant::V2, 0)
}

/// Calculates the `CryptoNight` R hash of buf.
pub fn cryptonight_hash_r(buf: &[u8], height: u64) -> [u8; 32] {
    cn_slow_hash(buf, slow_hash::Variant::R, height)
}

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

    #[test]
    fn slow_hash_0() {
        fn test(inp: &str, exp: &str) {
            let res = hex::encode(cryptonight_hash_v0(&hex::decode(inp).unwrap()));
            assert_eq!(&res, exp);
        }

        // https://github.com/monero-project/monero/blob/67d190ce7c33602b6a3b804f633ee1ddb7fbb4a1/tests/hash/tests-slow.txt
        test(
            "6465206f6d6e69627573206475626974616e64756d",
            "2f8e3df40bd11f9ac90c743ca8e32bb391da4fb98612aa3b6cdc639ee00b31f5",
        );
        test(
            "6162756e64616e732063617574656c61206e6f6e206e6f636574",
            "722fa8ccd594d40e4a41f3822734304c8d5eff7e1b528408e2229da38ba553c4",
        );
        test(
            "63617665617420656d70746f72",
            "bbec2cacf69866a8e740380fe7b818fc78f8571221742d729d9d02d7f8989b87",
        );
        test(
            "6578206e6968696c6f206e6968696c20666974",
            "b1257de4efc5ce28c6b40ceb1c6c8f812a64634eb3e81c5220bee9b2b76a6f05",
        );
    }

    #[test]
    fn slow_hash_1() {
        fn test(inp: &str, exp: &str) {
            let res = hex::encode(cryptonight_hash_v1(&hex::decode(inp).unwrap()).unwrap());
            assert_eq!(&res, exp);
        }

        // https://github.com/monero-project/monero/blob/67d190ce7c33602b6a3b804f633ee1ddb7fbb4a1/tests/hash/tests-slow-1.txt
        test(
            "00000000000000000000000000000000000000000000000000000000000000000000000000000000000000",
            "b5a7f63abb94d07d1a6445c36c07c7e8327fe61b1647e391b4c7edae5de57a3d",
        );
        test(
            "00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000",
            "80563c40ed46575a9e44820d93ee095e2851aa22483fd67837118c6cd951ba61",
        );
        test(
            "8519e039172b0d70e5ca7b3383d6b3167315a422747b73f019cf9528f0fde341fd0f2a63030ba6450525cf6de31837669af6f1df8131faf50aaab8d3a7405589",
            "5bb40c5880cef2f739bdb6aaaf16161eaae55530e7b10d7ea996b751a299e949",
        );
        test(
            "37a636d7dafdf259b7287eddca2f58099e98619d2f99bdb8969d7b14498102cc065201c8be90bd777323f449848b215d2977c92c4c1c2da36ab46b2e389689ed97c18fec08cd3b03235c5e4c62a37ad88c7b67932495a71090e85dd4020a9300",
            "613e638505ba1fd05f428d5c9f8e08f8165614342dac419adc6a47dce257eb3e",
        );
        test(
            "38274c97c45a172cfc97679870422e3a1ab0784960c60514d816271415c306ee3a3ed1a77e31f6a885c3cb",
            "ed082e49dbd5bbe34a3726a0d1dad981146062b39d36d62c71eb1ed8ab49459b",
        );
    }

    #[test]
    fn slow_hash_2() {
        fn test(inp: &str, exp: &str) {
            let res = hex::encode(cryptonight_hash_v2(&hex::decode(inp).unwrap()));
            assert_eq!(&res, exp);
        }

        // https://github.com/monero-project/monero/blob/67d190ce7c33602b6a3b804f633ee1ddb7fbb4a1/tests/hash/tests-slow-2.txt
        test(
            "5468697320697320612074657374205468697320697320612074657374205468697320697320612074657374",
            "353fdc068fd47b03c04b9431e005e00b68c2168a3cc7335c8b9b308156591a4f",
        );
        test(
            "4c6f72656d20697073756d20646f6c6f722073697420616d65742c20636f6e73656374657475722061646970697363696e67",
            "72f134fc50880c330fe65a2cb7896d59b2e708a0221c6a9da3f69b3a702d8682",
        );
        test(
            "656c69742c2073656420646f20656975736d6f642074656d706f7220696e6369646964756e74207574206c61626f7265",
            "410919660ec540fc49d8695ff01f974226a2a28dbbac82949c12f541b9a62d2f",
        );
        test(
            "657420646f6c6f7265206d61676e6120616c697175612e20557420656e696d206164206d696e696d2076656e69616d2c",
            "4472fecfeb371e8b7942ce0378c0ba5e6d0c6361b669c587807365c787ae652d",
        );
        test(
            "71756973206e6f737472756420657865726369746174696f6e20756c6c616d636f206c61626f726973206e697369",
            "577568395203f1f1225f2982b637f7d5e61b47a0f546ba16d46020b471b74076",
        );
        test(
            "757420616c697175697020657820656120636f6d6d6f646f20636f6e7365717561742e20447569732061757465",
            "f6fd7efe95a5c6c4bb46d9b429e3faf65b1ce439e116742d42b928e61de52385",
        );
        test(
            "697275726520646f6c6f7220696e20726570726568656e646572697420696e20766f6c7570746174652076656c6974",
            "422f8cfe8060cf6c3d9fd66f68e3c9977adb683aea2788029308bbe9bc50d728",
        );
        test(
            "657373652063696c6c756d20646f6c6f726520657520667567696174206e756c6c612070617269617475722e",
            "512e62c8c8c833cfbd9d361442cb00d63c0a3fd8964cfd2fedc17c7c25ec2d4b",
        );
        test(
            "4578636570746575722073696e74206f6363616563617420637570696461746174206e6f6e2070726f6964656e742c",
            "12a794c1aa13d561c9c6111cee631ca9d0a321718d67d3416add9de1693ba41e",
        );
        test(
            "73756e7420696e2063756c706120717569206f666669636961206465736572756e74206d6f6c6c697420616e696d20696420657374206c61626f72756d2e",
            "2659ff95fc74b6215c1dc741e85b7a9710101b30620212f80eb59c3c55993f9d",
        );
    }

    #[test]
    fn slow_hash_r() {
        fn test(inp: &str, exp: &str, height: u64) {
            let res = hex::encode(cryptonight_hash_r(&hex::decode(inp).unwrap(), height));
            assert_eq!(&res, exp);
        }

        // https://github.com/monero-project/monero/blob/67d190ce7c33602b6a3b804f633ee1ddb7fbb4a1/tests/hash/tests-slow-4.txt
        test(
            "5468697320697320612074657374205468697320697320612074657374205468697320697320612074657374",
            "f759588ad57e758467295443a9bd71490abff8e9dad1b95b6bf2f5d0d78387bc",
            1806260,
        );
        test(
            "4c6f72656d20697073756d20646f6c6f722073697420616d65742c20636f6e73656374657475722061646970697363696e67",
            "5bb833deca2bdd7252a9ccd7b4ce0b6a4854515794b56c207262f7a5b9bdb566",
            1806261,
        );
        test(
            "656c69742c2073656420646f20656975736d6f642074656d706f7220696e6369646964756e74207574206c61626f7265",
            "1ee6728da60fbd8d7d55b2b1ade487a3cf52a2c3ac6f520db12c27d8921f6cab",
            1806262,
        );
        test(
            "657420646f6c6f7265206d61676e6120616c697175612e20557420656e696d206164206d696e696d2076656e69616d2c",
            "6969fe2ddfb758438d48049f302fc2108a4fcc93e37669170e6db4b0b9b4c4cb",
            1806263,
        );
        test(
            "71756973206e6f737472756420657865726369746174696f6e20756c6c616d636f206c61626f726973206e697369",
            "7f3048b4e90d0cbe7a57c0394f37338a01fae3adfdc0e5126d863a895eb04e02",
            1806264,
        );
        test(
            "757420616c697175697020657820656120636f6d6d6f646f20636f6e7365717561742e20447569732061757465",
            "1d290443a4b542af04a82f6b2494a6ee7f20f2754c58e0849032483a56e8e2ef",
            1806265,
        );
        test(
            "757420616c697175697020657820656120636f6d6d6f646f20636f6e7365717561742e20447569732061757465",
            "1d290443a4b542af04a82f6b2494a6ee7f20f2754c58e0849032483a56e8e2ef",
            1806265,
        );
        test(
            "697275726520646f6c6f7220696e20726570726568656e646572697420696e20766f6c7570746174652076656c6974",
            "c43cc6567436a86afbd6aa9eaa7c276e9806830334b614b2bee23cc76634f6fd",
            1806266,
        );
        test(
            "657373652063696c6c756d20646f6c6f726520657520667567696174206e756c6c612070617269617475722e",
            "87be2479c0c4e8edfdfaa5603e93f4265b3f8224c1c5946feb424819d18990a4",
            1806267,
        );
        test(
            "4578636570746575722073696e74206f6363616563617420637570696461746174206e6f6e2070726f6964656e742c",
            "dd9d6a6d8e47465cceac0877ef889b93e7eba979557e3935d7f86dce11b070f3",
            1806268,
        );
        test(
            "73756e7420696e2063756c706120717569206f666669636961206465736572756e74206d6f6c6c697420616e696d20696420657374206c61626f72756d2e",
            "75c6f2ae49a20521de97285b431e717125847fb8935ed84a61e7f8d36a2c3d8e",
            1806269,
        );
    }
}