use std::{collections::HashMap, str::FromStr};

use crate::{
    types::{
        serde_helpers::{
            deserialize_stringified_eth_u64, deserialize_stringified_eth_u64_opt,
            deserialize_stringified_numeric, deserialize_stringified_numeric_opt,
            deserialize_stringified_u64_opt, Numeric,
        },
        Address, Bytes, H256, U256, U64,
    },
    utils::from_unformatted_hex_map,
};
use serde::{Deserialize, Serialize};

/// This represents the chain configuration, specifying the genesis block, header fields, and hard
/// fork switch blocks.
#[derive(Clone, Debug, Default, Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all = "camelCase")]
pub struct Genesis {
    /// The fork configuration for this network.
    #[serde(default)]
    pub config: ChainConfig,

    /// The genesis header nonce.
    #[serde(default, deserialize_with = "deserialize_stringified_eth_u64")]
    pub nonce: U64,

    /// The genesis header timestamp.
    #[serde(default, deserialize_with = "deserialize_stringified_eth_u64")]
    pub timestamp: U64,

    /// The genesis header extra data.
    #[serde(default)]
    pub extra_data: Bytes,

    /// The genesis header gas limit.
    #[serde(default, deserialize_with = "deserialize_stringified_eth_u64")]
    pub gas_limit: U64,

    /// The genesis header difficulty.
    #[serde(deserialize_with = "deserialize_stringified_numeric")]
    pub difficulty: U256,

    /// The genesis header mix hash.
    #[serde(default)]
    pub mix_hash: H256,

    /// The genesis header coinbase address.
    #[serde(default)]
    pub coinbase: Address,

    /// The initial state of the genesis block.
    pub alloc: HashMap<Address, GenesisAccount>,

    // The following fields are only included for tests, and should not be used in real genesis
    // blocks.
    /// The block number
    #[serde(
        skip_serializing_if = "Option::is_none",
        deserialize_with = "deserialize_stringified_eth_u64_opt",
        default
    )]
    pub number: Option<U64>,

    /// The block gas gasUsed
    #[serde(
        skip_serializing_if = "Option::is_none",
        deserialize_with = "deserialize_stringified_eth_u64_opt",
        default
    )]
    pub gas_used: Option<U64>,

    /// The block parent hash
    #[serde(skip_serializing_if = "Option::is_none", default)]
    pub parent_hash: Option<H256>,

    /// The base fee
    #[serde(
        skip_serializing_if = "Option::is_none",
        deserialize_with = "deserialize_stringified_numeric_opt",
        default
    )]
    pub base_fee_per_gas: Option<U256>,
}

impl Genesis {
    /// Creates a chain config using the given chain id.
    /// and funds the given address with max coins.
    ///
    /// Enables all hard forks up to London at genesis.
    pub fn new(chain_id: u64, signer_addr: Address) -> Genesis {
        // set up a clique config with an instant sealing period and short (8 block) epoch
        let clique_config = CliqueConfig { period: Some(0), epoch: Some(8) };

        let config = ChainConfig {
            chain_id,
            eip155_block: Some(0),
            eip150_block: Some(0),
            eip158_block: Some(0),

            homestead_block: Some(0),
            byzantium_block: Some(0),
            constantinople_block: Some(0),
            petersburg_block: Some(0),
            istanbul_block: Some(0),
            muir_glacier_block: Some(0),
            berlin_block: Some(0),
            london_block: Some(0),
            clique: Some(clique_config),
            ..Default::default()
        };

        // fund account
        let mut alloc = HashMap::new();
        alloc.insert(
            signer_addr,
            GenesisAccount { balance: U256::MAX, nonce: None, code: None, storage: None },
        );

        // put signer address in the extra data, padded by the required amount of zeros
        // Clique issue: https://github.com/ethereum/EIPs/issues/225
        // Clique EIP: https://eips.ethereum.org/EIPS/eip-225
        //
        // The first 32 bytes are vanity data, so we will populate it with zeros
        // This is followed by the signer address, which is 20 bytes
        // There are 65 bytes of zeros after the signer address, which is usually populated with the
        // proposer signature. Because the genesis does not have a proposer signature, it will be
        // populated with zeros.
        let extra_data_bytes = [&[0u8; 32][..], signer_addr.as_bytes(), &[0u8; 65][..]].concat();
        let extra_data = Bytes::from(extra_data_bytes);

        Genesis {
            config,
            alloc,
            difficulty: U256::one(),
            gas_limit: U64::from(5000000),
            extra_data,
            ..Default::default()
        }
    }
}

/// An account in the state of the genesis block.
#[derive(Clone, Debug, Default, PartialEq, Eq, Serialize, Deserialize)]
pub struct GenesisAccount {
    #[serde(
        skip_serializing_if = "Option::is_none",
        deserialize_with = "deserialize_stringified_u64_opt",
        default
    )]
    pub nonce: Option<u64>,
    #[serde(deserialize_with = "deserialize_stringified_numeric")]
    pub balance: U256,
    #[serde(skip_serializing_if = "Option::is_none", default)]
    pub code: Option<Bytes>,
    #[serde(
        skip_serializing_if = "Option::is_none",
        deserialize_with = "from_unformatted_hex_map",
        default
    )]
    pub storage: Option<HashMap<H256, H256>>,
}

/// Represents a node's chain configuration.
///
/// See [geth's `ChainConfig`
/// struct](https://github.com/ethereum/go-ethereum/blob/64dccf7aa411c5c7cd36090c3d9b9892945ae813/params/config.go#L349)
/// for the source of each field.
#[derive(Clone, Debug, Deserialize, Serialize, Default, PartialEq, Eq)]
#[serde(default, rename_all = "camelCase")]
pub struct ChainConfig {
    /// The network's chain ID.
    #[serde(default = "one")]
    pub chain_id: u64,

    /// The homestead switch block (None = no fork, 0 = already homestead).
    #[serde(
        skip_serializing_if = "Option::is_none",
        deserialize_with = "deserialize_stringified_u64_opt"
    )]
    pub homestead_block: Option<u64>,

    /// The DAO fork switch block (None = no fork).
    #[serde(
        skip_serializing_if = "Option::is_none",
        deserialize_with = "deserialize_stringified_u64_opt"
    )]
    pub dao_fork_block: Option<u64>,

    /// Whether or not the node supports the DAO hard-fork.
    pub dao_fork_support: bool,

    /// The EIP-150 hard fork block (None = no fork).
    #[serde(
        skip_serializing_if = "Option::is_none",
        deserialize_with = "deserialize_stringified_u64_opt"
    )]
    pub eip150_block: Option<u64>,

    /// The EIP-150 hard fork hash.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub eip150_hash: Option<H256>,

    /// The EIP-155 hard fork block.
    #[serde(
        skip_serializing_if = "Option::is_none",
        deserialize_with = "deserialize_stringified_u64_opt"
    )]
    pub eip155_block: Option<u64>,

    /// The EIP-158 hard fork block.
    #[serde(
        skip_serializing_if = "Option::is_none",
        deserialize_with = "deserialize_stringified_u64_opt"
    )]
    pub eip158_block: Option<u64>,

    /// The Byzantium hard fork block.
    #[serde(
        skip_serializing_if = "Option::is_none",
        deserialize_with = "deserialize_stringified_u64_opt"
    )]
    pub byzantium_block: Option<u64>,

    /// The Constantinople hard fork block.
    #[serde(
        skip_serializing_if = "Option::is_none",
        deserialize_with = "deserialize_stringified_u64_opt"
    )]
    pub constantinople_block: Option<u64>,

    /// The Petersburg hard fork block.
    #[serde(
        skip_serializing_if = "Option::is_none",
        deserialize_with = "deserialize_stringified_u64_opt"
    )]
    pub petersburg_block: Option<u64>,

    /// The Istanbul hard fork block.
    #[serde(
        skip_serializing_if = "Option::is_none",
        deserialize_with = "deserialize_stringified_u64_opt"
    )]
    pub istanbul_block: Option<u64>,

    /// The Muir Glacier hard fork block.
    #[serde(
        skip_serializing_if = "Option::is_none",
        deserialize_with = "deserialize_stringified_u64_opt"
    )]
    pub muir_glacier_block: Option<u64>,

    /// The Berlin hard fork block.
    #[serde(
        skip_serializing_if = "Option::is_none",
        deserialize_with = "deserialize_stringified_u64_opt"
    )]
    pub berlin_block: Option<u64>,

    /// The London hard fork block.
    #[serde(
        skip_serializing_if = "Option::is_none",
        deserialize_with = "deserialize_stringified_u64_opt"
    )]
    pub london_block: Option<u64>,

    /// The Arrow Glacier hard fork block.
    #[serde(
        skip_serializing_if = "Option::is_none",
        deserialize_with = "deserialize_stringified_u64_opt"
    )]
    pub arrow_glacier_block: Option<u64>,

    /// The Gray Glacier hard fork block.
    #[serde(
        skip_serializing_if = "Option::is_none",
        deserialize_with = "deserialize_stringified_u64_opt"
    )]
    pub gray_glacier_block: Option<u64>,

    /// Virtual fork after the merge to use as a network splitter.
    #[serde(
        skip_serializing_if = "Option::is_none",
        deserialize_with = "deserialize_stringified_u64_opt"
    )]
    pub merge_netsplit_block: Option<u64>,

    /// Shanghai switch time.
    #[serde(
        skip_serializing_if = "Option::is_none",
        deserialize_with = "deserialize_stringified_u64_opt"
    )]
    pub shanghai_time: Option<u64>,

    /// Cancun switch time.
    #[serde(
        skip_serializing_if = "Option::is_none",
        deserialize_with = "deserialize_stringified_u64_opt"
    )]
    pub cancun_time: Option<u64>,

    /// Total difficulty reached that triggers the merge consensus upgrade.
    #[serde(skip_serializing_if = "Option::is_none", deserialize_with = "deserialize_ttd")]
    pub terminal_total_difficulty: Option<U256>,

    /// A flag specifying that the network already passed the terminal total difficulty. Its
    /// purpose is to disable legacy sync without having seen the TTD locally.
    pub terminal_total_difficulty_passed: bool,

    /// Ethash parameters.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub ethash: Option<EthashConfig>,

    /// Clique parameters.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub clique: Option<CliqueConfig>,
}

// used only for serde
#[inline]
const fn one() -> u64 {
    1
}

/// Empty consensus configuration for proof-of-work networks.
#[derive(Clone, Debug, Deserialize, Serialize, PartialEq, Eq)]
pub struct EthashConfig {}

/// Consensus configuration for Clique.
#[derive(Clone, Debug, Deserialize, Serialize, PartialEq, Eq)]
pub struct CliqueConfig {
    /// Number of seconds between blocks to enforce.
    #[serde(
        default,
        skip_serializing_if = "Option::is_none",
        deserialize_with = "deserialize_stringified_u64_opt"
    )]
    pub period: Option<u64>,

    /// Epoch length to reset votes and checkpoints.
    #[serde(
        default,
        skip_serializing_if = "Option::is_none",
        deserialize_with = "deserialize_stringified_u64_opt"
    )]
    pub epoch: Option<u64>,
}

fn deserialize_ttd<'de, D>(deserializer: D) -> Result<Option<U256>, D::Error>
where
    D: serde::Deserializer<'de>,
{
    match Option::<serde_json::Value>::deserialize(deserializer)? {
        None => Ok(None),
        Some(val) => {
            if let Some(num) = val.as_str() {
                return Numeric::from_str(num)
                    .map(U256::from)
                    .map(Some)
                    .map_err(serde::de::Error::custom)
            }

            if let serde_json::Value::Number(num) = val {
                // geth serializes ttd as number, for mainnet this exceeds u64 which serde is unable
                // to deserialize as integer
                if num.as_f64() == Some(5.875e22) {
                    Ok(Some(U256::from(58750000000000000000000u128)))
                } else {
                    num.as_u64()
                        .map(U256::from)
                        .ok_or_else(|| {
                            serde::de::Error::custom(format!("expected a number got {num}"))
                        })
                        .map(Some)
                }
            } else {
                Err(serde::de::Error::custom(format!("expected a number, got {:?}", val)))
            }
        }
    }
}

#[cfg(test)]
mod tests {
    use super::{ChainConfig, Genesis, GenesisAccount, H256};
    use crate::{
        types::{Address, Bytes, H160, U256},
        utils::EthashConfig,
    };
    use std::{collections::HashMap, str::FromStr};

    #[test]
    fn parse_hive_genesis() {
        let geth_genesis = r#"
        {
            "difficulty": "0x20000",
            "gasLimit": "0x1",
            "alloc": {},
            "config": {
              "ethash": {},
              "chainId": 1
            }
        }
        "#;

        let _genesis: Genesis = serde_json::from_str(geth_genesis).unwrap();
    }

    #[test]
    fn parse_hive_clique_smoke_genesis() {
        let geth_genesis = r#"
        {
          "difficulty": "0x1",
          "gasLimit": "0x400000",
          "extraData": "0x0000000000000000000000000000000000000000000000000000000000000000658bdf435d810c91414ec09147daa6db624063790000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000",
          "mixHash": "0x0000000000000000000000000000000000000000000000000000000000000000",
          "nonce": "0x0",
          "timestamp": "0x5c51a607",
          "alloc": {}
        }
        "#;

        let _genesis: Genesis = serde_json::from_str(geth_genesis).unwrap();
    }

    #[test]
    fn parse_non_hex_prefixed_balance() {
        // tests that we can parse balance / difficulty fields that are either hex or decimal
        let example_balance_json = r#"
        {
            "nonce": "0x0000000000000042",
            "difficulty": "34747478",
            "mixHash": "0x123456789abcdef123456789abcdef123456789abcdef123456789abcdef1234",
            "coinbase": "0xaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa",
            "timestamp": "0x123456",
            "parentHash": "0x0000000000000000000000000000000000000000000000000000000000000000",
            "extraData": "0xfafbfcfd",
            "gasLimit": "0x2fefd8",
            "alloc": {
                "0x3E951C9f69a06Bc3AD71fF7358DbC56bEd94b9F2": {
                  "balance": "1000000000000000000000000000"
                },
                "0xe228C30d4e5245f967ac21726d5412dA27aD071C": {
                  "balance": "1000000000000000000000000000"
                },
                "0xD59Ce7Ccc6454a2D2C2e06bbcf71D0Beb33480eD": {
                  "balance": "1000000000000000000000000000"
                },
                "0x1CF4D54414eF51b41f9B2238c57102ab2e61D1F2": {
                  "balance": "1000000000000000000000000000"
                },
                "0x249bE3fDEd872338C733cF3975af9736bdCb9D4D": {
                  "balance": "1000000000000000000000000000"
                },
                "0x3fCd1bff94513712f8cD63d1eD66776A67D5F78e": {
                  "balance": "1000000000000000000000000000"
                }
            },
            "config": {
                "ethash": {},
                "chainId": 10,
                "homesteadBlock": 0,
                "eip150Block": 0,
                "eip155Block": 0,
                "eip158Block": 0,
                "byzantiumBlock": 0,
                "constantinopleBlock": 0,
                "petersburgBlock": 0,
                "istanbulBlock": 0
            }
        }
        "#;

        let genesis: Genesis = serde_json::from_str(example_balance_json).unwrap();

        // check difficulty against hex ground truth
        let expected_difficulty = U256::from_str("0x2123456").unwrap();
        assert_eq!(expected_difficulty, genesis.difficulty);

        // check all alloc balances
        let dec_balance = U256::from_dec_str("1000000000000000000000000000").unwrap();
        for alloc in &genesis.alloc {
            assert_eq!(alloc.1.balance, dec_balance);
        }
    }

    #[test]
    fn parse_hive_rpc_genesis() {
        let geth_genesis = r#"
        {
          "config": {
            "chainId": 7,
            "homesteadBlock": 0,
            "eip150Block": 0,
            "eip150Hash": "0x5de1ee4135274003348e80b788e5afa4b18b18d320a5622218d5c493fedf5689",
            "eip155Block": 0,
            "eip158Block": 0
          },
          "coinbase": "0x0000000000000000000000000000000000000000",
          "difficulty": "0x20000",
          "extraData": "0x0000000000000000000000000000000000000000000000000000000000000000658bdf435d810c91414ec09147daa6db624063790000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000",
          "gasLimit": "0x2fefd8",
          "nonce": "0x0000000000000000",
          "timestamp": "0x1234",
          "alloc": {
            "cf49fda3be353c69b41ed96333cd24302da4556f": {
              "balance": "0x123450000000000000000"
            },
            "0161e041aad467a890839d5b08b138c1e6373072": {
              "balance": "0x123450000000000000000"
            },
            "87da6a8c6e9eff15d703fc2773e32f6af8dbe301": {
              "balance": "0x123450000000000000000"
            },
            "b97de4b8c857e4f6bc354f226dc3249aaee49209": {
              "balance": "0x123450000000000000000"
            },
            "c5065c9eeebe6df2c2284d046bfc906501846c51": {
              "balance": "0x123450000000000000000"
            },
            "0000000000000000000000000000000000000314": {
              "balance": "0x0",
              "code": "0x60606040526000357c0100000000000000000000000000000000000000000000000000000000900463ffffffff168063a223e05d1461006a578063abd1a0cf1461008d578063abfced1d146100d4578063e05c914a14610110578063e6768b451461014c575b610000565b346100005761007761019d565b6040518082815260200191505060405180910390f35b34610000576100be600480803573ffffffffffffffffffffffffffffffffffffffff169060200190919050506101a3565b6040518082815260200191505060405180910390f35b346100005761010e600480803573ffffffffffffffffffffffffffffffffffffffff169060200190919080359060200190919050506101ed565b005b346100005761014a600480803590602001909190803573ffffffffffffffffffffffffffffffffffffffff16906020019091905050610236565b005b346100005761017960048080359060200190919080359060200190919080359060200190919050506103c4565b60405180848152602001838152602001828152602001935050505060405180910390f35b60005481565b6000600160008373ffffffffffffffffffffffffffffffffffffffff1673ffffffffffffffffffffffffffffffffffffffff1681526020019081526020016000205490505b919050565b80600160008473ffffffffffffffffffffffffffffffffffffffff1673ffffffffffffffffffffffffffffffffffffffff168152602001908152602001600020819055505b5050565b7f6031a8d62d7c95988fa262657cd92107d90ed96e08d8f867d32f26edfe85502260405180905060405180910390a17f47e2689743f14e97f7dcfa5eec10ba1dff02f83b3d1d4b9c07b206cbbda66450826040518082815260200191505060405180910390a1817fa48a6b249a5084126c3da369fbc9b16827ead8cb5cdc094b717d3f1dcd995e2960405180905060405180910390a27f7890603b316f3509577afd111710f9ebeefa15e12f72347d9dffd0d65ae3bade81604051808273ffffffffffffffffffffffffffffffffffffffff1673ffffffffffffffffffffffffffffffffffffffff16815260200191505060405180910390a18073ffffffffffffffffffffffffffffffffffffffff167f7efef9ea3f60ddc038e50cccec621f86a0195894dc0520482abf8b5c6b659e4160405180905060405180910390a28181604051808381526020018273ffffffffffffffffffffffffffffffffffffffff1673ffffffffffffffffffffffffffffffffffffffff1681526020019250505060405180910390a05b5050565b6000600060008585859250925092505b935093509390505600a165627a7a72305820aaf842d0d0c35c45622c5263cbb54813d2974d3999c8c38551d7c613ea2bc1170029",
              "storage": {
                "0x0000000000000000000000000000000000000000000000000000000000000000": "0x1234",
                "0x6661e9d6d8b923d5bbaab1b96e1dd51ff6ea2a93520fdc9eb75d059238b8c5e9": "0x01"
              }
            },
            "0000000000000000000000000000000000000315": {
              "balance": "0x9999999999999999999999999999999",
              "code": "0x60606040526000357c0100000000000000000000000000000000000000000000000000000000900463ffffffff168063ef2769ca1461003e575b610000565b3461000057610078600480803573ffffffffffffffffffffffffffffffffffffffff1690602001909190803590602001909190505061007a565b005b8173ffffffffffffffffffffffffffffffffffffffff166108fc829081150290604051809050600060405180830381858888f1935050505015610106578173ffffffffffffffffffffffffffffffffffffffff167f30a3c50752f2552dcc2b93f5b96866280816a986c0c0408cb6778b9fa198288f826040518082815260200191505060405180910390a25b5b50505600a165627a7a72305820637991fabcc8abad4294bf2bb615db78fbec4edff1635a2647d3894e2daf6a610029"
            }
          }
        }
        "#;

        let _genesis: Genesis = serde_json::from_str(geth_genesis).unwrap();
    }

    #[test]
    fn parse_hive_graphql_genesis() {
        let geth_genesis = r#"
        {
            "config"     : {},
            "coinbase"   : "0x8888f1f195afa192cfee860698584c030f4c9db1",
            "difficulty" : "0x020000",
            "extraData"  : "0x42",
            "gasLimit"   : "0x2fefd8",
            "mixHash"    : "0x2c85bcbce56429100b2108254bb56906257582aeafcbd682bc9af67a9f5aee46",
            "nonce"      : "0x78cc16f7b4f65485",
            "parentHash" : "0x0000000000000000000000000000000000000000000000000000000000000000",
            "timestamp"  : "0x54c98c81",
            "alloc"      : {
                "a94f5374fce5edbc8e2a8697c15331677e6ebf0b": {
                    "balance" : "0x09184e72a000"
                }
            }
        }
        "#;

        let _genesis: Genesis = serde_json::from_str(geth_genesis).unwrap();
    }

    #[test]
    fn parse_hive_engine_genesis() {
        let geth_genesis = r#"
        {
          "config": {
            "chainId": 7,
            "homesteadBlock": 0,
            "eip150Block": 0,
            "eip150Hash": "0x5de1ee4135274003348e80b788e5afa4b18b18d320a5622218d5c493fedf5689",
            "eip155Block": 0,
            "eip158Block": 0,
            "byzantiumBlock": 0,
            "constantinopleBlock": 0,
            "petersburgBlock": 0,
            "istanbulBlock": 0,
            "muirGlacierBlock": 0,
            "berlinBlock": 0,
            "yolov2Block": 0,
            "yolov3Block": 0,
            "londonBlock": 0
          },
          "coinbase": "0x0000000000000000000000000000000000000000",
          "difficulty": "0x30000",
          "extraData": "0x0000000000000000000000000000000000000000000000000000000000000000658bdf435d810c91414ec09147daa6db624063790000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000",
          "gasLimit": "0x2fefd8",
          "nonce": "0x0000000000000000",
          "timestamp": "0x1234",
          "alloc": {
            "cf49fda3be353c69b41ed96333cd24302da4556f": {
              "balance": "0x123450000000000000000"
            },
            "0161e041aad467a890839d5b08b138c1e6373072": {
              "balance": "0x123450000000000000000"
            },
            "87da6a8c6e9eff15d703fc2773e32f6af8dbe301": {
              "balance": "0x123450000000000000000"
            },
            "b97de4b8c857e4f6bc354f226dc3249aaee49209": {
              "balance": "0x123450000000000000000"
            },
            "c5065c9eeebe6df2c2284d046bfc906501846c51": {
              "balance": "0x123450000000000000000"
            },
            "0000000000000000000000000000000000000314": {
              "balance": "0x0",
              "code": "0x60606040526000357c0100000000000000000000000000000000000000000000000000000000900463ffffffff168063a223e05d1461006a578063abd1a0cf1461008d578063abfced1d146100d4578063e05c914a14610110578063e6768b451461014c575b610000565b346100005761007761019d565b6040518082815260200191505060405180910390f35b34610000576100be600480803573ffffffffffffffffffffffffffffffffffffffff169060200190919050506101a3565b6040518082815260200191505060405180910390f35b346100005761010e600480803573ffffffffffffffffffffffffffffffffffffffff169060200190919080359060200190919050506101ed565b005b346100005761014a600480803590602001909190803573ffffffffffffffffffffffffffffffffffffffff16906020019091905050610236565b005b346100005761017960048080359060200190919080359060200190919080359060200190919050506103c4565b60405180848152602001838152602001828152602001935050505060405180910390f35b60005481565b6000600160008373ffffffffffffffffffffffffffffffffffffffff1673ffffffffffffffffffffffffffffffffffffffff1681526020019081526020016000205490505b919050565b80600160008473ffffffffffffffffffffffffffffffffffffffff1673ffffffffffffffffffffffffffffffffffffffff168152602001908152602001600020819055505b5050565b7f6031a8d62d7c95988fa262657cd92107d90ed96e08d8f867d32f26edfe85502260405180905060405180910390a17f47e2689743f14e97f7dcfa5eec10ba1dff02f83b3d1d4b9c07b206cbbda66450826040518082815260200191505060405180910390a1817fa48a6b249a5084126c3da369fbc9b16827ead8cb5cdc094b717d3f1dcd995e2960405180905060405180910390a27f7890603b316f3509577afd111710f9ebeefa15e12f72347d9dffd0d65ae3bade81604051808273ffffffffffffffffffffffffffffffffffffffff1673ffffffffffffffffffffffffffffffffffffffff16815260200191505060405180910390a18073ffffffffffffffffffffffffffffffffffffffff167f7efef9ea3f60ddc038e50cccec621f86a0195894dc0520482abf8b5c6b659e4160405180905060405180910390a28181604051808381526020018273ffffffffffffffffffffffffffffffffffffffff1673ffffffffffffffffffffffffffffffffffffffff1681526020019250505060405180910390a05b5050565b6000600060008585859250925092505b935093509390505600a165627a7a72305820aaf842d0d0c35c45622c5263cbb54813d2974d3999c8c38551d7c613ea2bc1170029",
              "storage": {
                "0x0000000000000000000000000000000000000000000000000000000000000000": "0x1234",
                "0x6661e9d6d8b923d5bbaab1b96e1dd51ff6ea2a93520fdc9eb75d059238b8c5e9": "0x01"
              }
            },
            "0000000000000000000000000000000000000315": {
              "balance": "0x9999999999999999999999999999999",
              "code": "0x60606040526000357c0100000000000000000000000000000000000000000000000000000000900463ffffffff168063ef2769ca1461003e575b610000565b3461000057610078600480803573ffffffffffffffffffffffffffffffffffffffff1690602001909190803590602001909190505061007a565b005b8173ffffffffffffffffffffffffffffffffffffffff166108fc829081150290604051809050600060405180830381858888f1935050505015610106578173ffffffffffffffffffffffffffffffffffffffff167f30a3c50752f2552dcc2b93f5b96866280816a986c0c0408cb6778b9fa198288f826040518082815260200191505060405180910390a25b5b50505600a165627a7a72305820637991fabcc8abad4294bf2bb615db78fbec4edff1635a2647d3894e2daf6a610029"
            },
            "0000000000000000000000000000000000000316": {
              "balance": "0x0",
              "code": "0x444355"
            },
            "0000000000000000000000000000000000000317": {
              "balance": "0x0",
              "code": "0x600160003555"
            }
          }
        }
        "#;

        let _genesis: Genesis = serde_json::from_str(geth_genesis).unwrap();
    }

    #[test]
    fn parse_hive_devp2p_genesis() {
        let geth_genesis = r#"
        {
            "config": {
                "chainId": 19763,
                "homesteadBlock": 0,
                "eip150Block": 0,
                "eip155Block": 0,
                "eip158Block": 0,
                "byzantiumBlock": 0,
                "ethash": {}
            },
            "nonce": "0xdeadbeefdeadbeef",
            "timestamp": "0x0",
            "extraData": "0x0000000000000000000000000000000000000000000000000000000000000000",
            "gasLimit": "0x80000000",
            "difficulty": "0x20000",
            "mixHash": "0x0000000000000000000000000000000000000000000000000000000000000000",
            "coinbase": "0x0000000000000000000000000000000000000000",
            "alloc": {
                "71562b71999873db5b286df957af199ec94617f7": {
                    "balance": "0xffffffffffffffffffffffffff"
                }
            },
            "number": "0x0",
            "gasUsed": "0x0",
            "parentHash": "0x0000000000000000000000000000000000000000000000000000000000000000"
        }
        "#;

        let _genesis: Genesis = serde_json::from_str(geth_genesis).unwrap();
    }

    #[test]
    fn parse_execution_apis_genesis() {
        let geth_genesis = r#"
        {
          "config": {
            "chainId": 1337,
            "homesteadBlock": 0,
            "eip150Block": 0,
            "eip150Hash": "0x0000000000000000000000000000000000000000000000000000000000000000",
            "eip155Block": 0,
            "eip158Block": 0,
            "byzantiumBlock": 0,
            "constantinopleBlock": 0,
            "petersburgBlock": 0,
            "istanbulBlock": 0,
            "muirGlacierBlock": 0,
            "berlinBlock": 0,
            "londonBlock": 0,
            "arrowGlacierBlock": 0,
            "grayGlacierBlock": 0,
            "shanghaiTime": 0,
            "terminalTotalDifficulty": 0,
            "terminalTotalDifficultyPassed": true,
            "ethash": {}
          },
          "nonce": "0x0",
          "timestamp": "0x0",
          "extraData": "0x",
          "gasLimit": "0x4c4b40",
          "difficulty": "0x1",
          "mixHash": "0x0000000000000000000000000000000000000000000000000000000000000000",
          "coinbase": "0x0000000000000000000000000000000000000000",
          "alloc": {
            "658bdf435d810c91414ec09147daa6db62406379": {
              "balance": "0x487a9a304539440000"
            },
            "aa00000000000000000000000000000000000000": {
              "code": "0x6042",
              "storage": {
                "0x0000000000000000000000000000000000000000000000000000000000000000": "0x0000000000000000000000000000000000000000000000000000000000000000",
                "0x0100000000000000000000000000000000000000000000000000000000000000": "0x0100000000000000000000000000000000000000000000000000000000000000",
                "0x0200000000000000000000000000000000000000000000000000000000000000": "0x0200000000000000000000000000000000000000000000000000000000000000",
                "0x0300000000000000000000000000000000000000000000000000000000000000": "0x0000000000000000000000000000000000000000000000000000000000000303"
              },
              "balance": "0x1",
              "nonce": "0x1"
            },
            "bb00000000000000000000000000000000000000": {
              "code": "0x600154600354",
              "storage": {
                "0x0000000000000000000000000000000000000000000000000000000000000000": "0x0000000000000000000000000000000000000000000000000000000000000000",
                "0x0100000000000000000000000000000000000000000000000000000000000000": "0x0100000000000000000000000000000000000000000000000000000000000000",
                "0x0200000000000000000000000000000000000000000000000000000000000000": "0x0200000000000000000000000000000000000000000000000000000000000000",
                "0x0300000000000000000000000000000000000000000000000000000000000000": "0x0000000000000000000000000000000000000000000000000000000000000303"
              },
              "balance": "0x2",
              "nonce": "0x1"
            }
          },
          "number": "0x0",
          "gasUsed": "0x0",
          "parentHash": "0x0000000000000000000000000000000000000000000000000000000000000000",
          "baseFeePerGas": "0x3b9aca00"
        }
        "#;

        let genesis: Genesis = serde_json::from_str(geth_genesis).unwrap();

        // ensure the test fields are parsed correctly
        assert_eq!(genesis.base_fee_per_gas, Some(1000000000.into()));
        assert_eq!(genesis.number, Some(0.into()));
        assert_eq!(genesis.gas_used, Some(0.into()));
        assert_eq!(genesis.parent_hash, Some(H256::zero()));
    }

    #[test]
    fn parse_hive_rpc_genesis_full() {
        let geth_genesis = r#"
        {
          "config": {
            "clique": {
              "period": 1
            },
            "chainId": 7,
            "homesteadBlock": 0,
            "eip150Block": 0,
            "eip155Block": 0,
            "eip158Block": 0
          },
          "coinbase": "0x0000000000000000000000000000000000000000",
          "difficulty": "0x020000",
          "extraData": "0x0000000000000000000000000000000000000000000000000000000000000000658bdf435d810c91414ec09147daa6db624063790000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000",
          "gasLimit": "0x2fefd8",
          "nonce": "0x0000000000000000",
          "timestamp": "0x1234",
          "alloc": {
            "cf49fda3be353c69b41ed96333cd24302da4556f": {
              "balance": "0x123450000000000000000"
            },
            "0161e041aad467a890839d5b08b138c1e6373072": {
              "balance": "0x123450000000000000000"
            },
            "87da6a8c6e9eff15d703fc2773e32f6af8dbe301": {
              "balance": "0x123450000000000000000"
            },
            "b97de4b8c857e4f6bc354f226dc3249aaee49209": {
              "balance": "0x123450000000000000000"
            },
            "c5065c9eeebe6df2c2284d046bfc906501846c51": {
              "balance": "0x123450000000000000000"
            },
            "0000000000000000000000000000000000000314": {
              "balance": "0x0",
              "code": "0x60606040526000357c0100000000000000000000000000000000000000000000000000000000900463ffffffff168063a223e05d1461006a578063abd1a0cf1461008d578063abfced1d146100d4578063e05c914a14610110578063e6768b451461014c575b610000565b346100005761007761019d565b6040518082815260200191505060405180910390f35b34610000576100be600480803573ffffffffffffffffffffffffffffffffffffffff169060200190919050506101a3565b6040518082815260200191505060405180910390f35b346100005761010e600480803573ffffffffffffffffffffffffffffffffffffffff169060200190919080359060200190919050506101ed565b005b346100005761014a600480803590602001909190803573ffffffffffffffffffffffffffffffffffffffff16906020019091905050610236565b005b346100005761017960048080359060200190919080359060200190919080359060200190919050506103c4565b60405180848152602001838152602001828152602001935050505060405180910390f35b60005481565b6000600160008373ffffffffffffffffffffffffffffffffffffffff1673ffffffffffffffffffffffffffffffffffffffff1681526020019081526020016000205490505b919050565b80600160008473ffffffffffffffffffffffffffffffffffffffff1673ffffffffffffffffffffffffffffffffffffffff168152602001908152602001600020819055505b5050565b7f6031a8d62d7c95988fa262657cd92107d90ed96e08d8f867d32f26edfe85502260405180905060405180910390a17f47e2689743f14e97f7dcfa5eec10ba1dff02f83b3d1d4b9c07b206cbbda66450826040518082815260200191505060405180910390a1817fa48a6b249a5084126c3da369fbc9b16827ead8cb5cdc094b717d3f1dcd995e2960405180905060405180910390a27f7890603b316f3509577afd111710f9ebeefa15e12f72347d9dffd0d65ae3bade81604051808273ffffffffffffffffffffffffffffffffffffffff1673ffffffffffffffffffffffffffffffffffffffff16815260200191505060405180910390a18073ffffffffffffffffffffffffffffffffffffffff167f7efef9ea3f60ddc038e50cccec621f86a0195894dc0520482abf8b5c6b659e4160405180905060405180910390a28181604051808381526020018273ffffffffffffffffffffffffffffffffffffffff1673ffffffffffffffffffffffffffffffffffffffff1681526020019250505060405180910390a05b5050565b6000600060008585859250925092505b935093509390505600a165627a7a72305820aaf842d0d0c35c45622c5263cbb54813d2974d3999c8c38551d7c613ea2bc1170029",
              "storage": {
                "0x0000000000000000000000000000000000000000000000000000000000000000": "0x1234",
                "0x6661e9d6d8b923d5bbaab1b96e1dd51ff6ea2a93520fdc9eb75d059238b8c5e9": "0x01"
              }
            },
            "0000000000000000000000000000000000000315": {
              "balance": "0x9999999999999999999999999999999",
              "code": "0x60606040526000357c0100000000000000000000000000000000000000000000000000000000900463ffffffff168063ef2769ca1461003e575b610000565b3461000057610078600480803573ffffffffffffffffffffffffffffffffffffffff1690602001909190803590602001909190505061007a565b005b8173ffffffffffffffffffffffffffffffffffffffff166108fc829081150290604051809050600060405180830381858888f1935050505015610106578173ffffffffffffffffffffffffffffffffffffffff167f30a3c50752f2552dcc2b93f5b96866280816a986c0c0408cb6778b9fa198288f826040518082815260200191505060405180910390a25b5b50505600a165627a7a72305820637991fabcc8abad4294bf2bb615db78fbec4edff1635a2647d3894e2daf6a610029"
            }
          },
          "mixHash": "0x0000000000000000000000000000000000000000000000000000000000000000",
          "parentHash": "0x0000000000000000000000000000000000000000000000000000000000000000"
        }
        "#;

        let genesis: Genesis = serde_json::from_str(geth_genesis).unwrap();
        let alloc_entry = genesis
            .alloc
            .get(&H160::from_str("0000000000000000000000000000000000000314").unwrap())
            .expect("missing account for parsed genesis");
        let storage = alloc_entry.storage.as_ref().expect("missing storage for parsed genesis");
        let expected_storage = HashMap::from_iter(vec![
            (
                H256::from_str(
                    "0x0000000000000000000000000000000000000000000000000000000000000000",
                )
                .unwrap(),
                H256::from_str(
                    "0x0000000000000000000000000000000000000000000000000000000000001234",
                )
                .unwrap(),
            ),
            (
                H256::from_str(
                    "0x6661e9d6d8b923d5bbaab1b96e1dd51ff6ea2a93520fdc9eb75d059238b8c5e9",
                )
                .unwrap(),
                H256::from_str(
                    "0x0000000000000000000000000000000000000000000000000000000000000001",
                )
                .unwrap(),
            ),
        ]);
        assert_eq!(storage, &expected_storage);

        let expected_code = Bytes::from_str("0x60606040526000357c0100000000000000000000000000000000000000000000000000000000900463ffffffff168063a223e05d1461006a578063abd1a0cf1461008d578063abfced1d146100d4578063e05c914a14610110578063e6768b451461014c575b610000565b346100005761007761019d565b6040518082815260200191505060405180910390f35b34610000576100be600480803573ffffffffffffffffffffffffffffffffffffffff169060200190919050506101a3565b6040518082815260200191505060405180910390f35b346100005761010e600480803573ffffffffffffffffffffffffffffffffffffffff169060200190919080359060200190919050506101ed565b005b346100005761014a600480803590602001909190803573ffffffffffffffffffffffffffffffffffffffff16906020019091905050610236565b005b346100005761017960048080359060200190919080359060200190919080359060200190919050506103c4565b60405180848152602001838152602001828152602001935050505060405180910390f35b60005481565b6000600160008373ffffffffffffffffffffffffffffffffffffffff1673ffffffffffffffffffffffffffffffffffffffff1681526020019081526020016000205490505b919050565b80600160008473ffffffffffffffffffffffffffffffffffffffff1673ffffffffffffffffffffffffffffffffffffffff168152602001908152602001600020819055505b5050565b7f6031a8d62d7c95988fa262657cd92107d90ed96e08d8f867d32f26edfe85502260405180905060405180910390a17f47e2689743f14e97f7dcfa5eec10ba1dff02f83b3d1d4b9c07b206cbbda66450826040518082815260200191505060405180910390a1817fa48a6b249a5084126c3da369fbc9b16827ead8cb5cdc094b717d3f1dcd995e2960405180905060405180910390a27f7890603b316f3509577afd111710f9ebeefa15e12f72347d9dffd0d65ae3bade81604051808273ffffffffffffffffffffffffffffffffffffffff1673ffffffffffffffffffffffffffffffffffffffff16815260200191505060405180910390a18073ffffffffffffffffffffffffffffffffffffffff167f7efef9ea3f60ddc038e50cccec621f86a0195894dc0520482abf8b5c6b659e4160405180905060405180910390a28181604051808381526020018273ffffffffffffffffffffffffffffffffffffffff1673ffffffffffffffffffffffffffffffffffffffff1681526020019250505060405180910390a05b5050565b6000600060008585859250925092505b935093509390505600a165627a7a72305820aaf842d0d0c35c45622c5263cbb54813d2974d3999c8c38551d7c613ea2bc1170029").unwrap();
        let code = alloc_entry.code.as_ref().expect("missing code for parsed genesis");
        assert_eq!(code, &expected_code);
    }

    #[test]
    fn test_hive_smoke_alloc_deserialize() {
        let hive_genesis = r#"
        {
            "nonce": "0x0000000000000042",
            "difficulty": "0x2123456",
            "mixHash": "0x123456789abcdef123456789abcdef123456789abcdef123456789abcdef1234",
            "coinbase": "0xaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa",
            "timestamp": "0x123456",
            "parentHash": "0x0000000000000000000000000000000000000000000000000000000000000000",
            "extraData": "0xfafbfcfd",
            "gasLimit": "0x2fefd8",
            "alloc": {
                "dbdbdb2cbd23b783741e8d7fcf51e459b497e4a6": {
                    "balance": "0xffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff"
                },
                "e6716f9544a56c530d868e4bfbacb172315bdead": {
                    "balance": "0x11",
                    "code": "0x12"
                },
                "b9c015918bdaba24b4ff057a92a3873d6eb201be": {
                    "balance": "0x21",
                    "storage": {
                        "0x0000000000000000000000000000000000000000000000000000000000000001": "0x22"
                    }
                },
                "1a26338f0d905e295fccb71fa9ea849ffa12aaf4": {
                    "balance": "0x31",
                    "nonce": "0x32"
                },
                "0000000000000000000000000000000000000001": {
                    "balance": "0x41"
                },
                "0000000000000000000000000000000000000002": {
                    "balance": "0x51"
                },
                "0000000000000000000000000000000000000003": {
                    "balance": "0x61"
                },
                "0000000000000000000000000000000000000004": {
                    "balance": "0x71"
                }
            },
            "config": {
                "ethash": {},
                "chainId": 10,
                "homesteadBlock": 0,
                "eip150Block": 0,
                "eip155Block": 0,
                "eip158Block": 0,
                "byzantiumBlock": 0,
                "constantinopleBlock": 0,
                "petersburgBlock": 0,
                "istanbulBlock": 0
            }
        }
        "#;

        let expected_genesis = Genesis {
            nonce: 0x0000000000000042.into(),
            difficulty: 0x2123456.into(),
            mix_hash: H256::from_str("0x123456789abcdef123456789abcdef123456789abcdef123456789abcdef1234").unwrap(),
            coinbase: Address::from_str("0xaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa").unwrap(),
            timestamp: 0x123456.into(),
            parent_hash: Some(H256::from_str("0x0000000000000000000000000000000000000000000000000000000000000000").unwrap()),
            extra_data: Bytes::from_str("0xfafbfcfd").unwrap(),
            gas_limit: 0x2fefd8.into(),
            alloc: HashMap::from_iter(vec![
                (
                    Address::from_str("0xdbdbdb2cbd23b783741e8d7fcf51e459b497e4a6").unwrap(),
                    GenesisAccount {
                        balance: U256::from_str("0xffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff").unwrap(),
                        nonce: None,
                        code: None,
                        storage: None,
                    },
                ),
                (
                    Address::from_str("0xe6716f9544a56c530d868e4bfbacb172315bdead").unwrap(),
                    GenesisAccount {
                        balance: U256::from_str("0x11").unwrap(),
                        nonce: None,
                        code: Some(Bytes::from_str("0x12").unwrap()),
                        storage: None,
                    },
                ),
                (
                    Address::from_str("0xb9c015918bdaba24b4ff057a92a3873d6eb201be").unwrap(),
                    GenesisAccount {
                        balance: U256::from_str("0x21").unwrap(),
                        nonce: None,
                        code: None,
                        storage: Some(HashMap::from_iter(vec![
                            (
                                H256::from_str("0x0000000000000000000000000000000000000000000000000000000000000001").unwrap(),
                                H256::from_str("0x0000000000000000000000000000000000000000000000000000000000000022").unwrap(),
                            ),
                        ])),
                    },
                ),
                (
                    Address::from_str("0x1a26338f0d905e295fccb71fa9ea849ffa12aaf4").unwrap(),
                    GenesisAccount {
                        balance: U256::from_str("0x31").unwrap(),
                        nonce: Some(0x32u64),
                        code: None,
                        storage: None,
                    },
                ),
                (
                    Address::from_str("0x0000000000000000000000000000000000000001").unwrap(),
                    GenesisAccount {
                        balance: U256::from_str("0x41").unwrap(),
                        nonce: None,
                        code: None,
                        storage: None,
                    },
                ),
                (
                    Address::from_str("0x0000000000000000000000000000000000000002").unwrap(),
                    GenesisAccount {
                        balance: U256::from_str("0x51").unwrap(),
                        nonce: None,
                        code: None,
                        storage: None,
                    },
                ),
                (
                    Address::from_str("0x0000000000000000000000000000000000000003").unwrap(),
                    GenesisAccount {
                        balance: U256::from_str("0x61").unwrap(),
                        nonce: None,
                        code: None,
                        storage: None,
                    },
                ),
                (
                    Address::from_str("0x0000000000000000000000000000000000000004").unwrap(),
                    GenesisAccount {
                        balance: U256::from_str("0x71").unwrap(),
                        nonce: None,
                        code: None,
                        storage: None,
                    },
                ),
            ]),
            config: ChainConfig {
                ethash: Some(EthashConfig{}),
                chain_id: 10,
                homestead_block: Some(0),
                eip150_block: Some(0),
                eip155_block: Some(0),
                eip158_block: Some(0),
                byzantium_block: Some(0),
                constantinople_block: Some(0),
                petersburg_block: Some(0),
                istanbul_block: Some(0),
                ..Default::default()
            },
            ..Default::default()
        };

        let deserialized_genesis: Genesis = serde_json::from_str(hive_genesis).unwrap();
        assert_eq!(deserialized_genesis, expected_genesis, "deserialized genesis {deserialized_genesis:#?} does not match expected {expected_genesis:#?}");
    }
}