use serde::{Deserialize, Serialize};

use casper_types::{
    account::{Account, AccountHash},
    ProtocolVersion, PublicKey,
};

use crate::rpcs::common::BlockIdentifier;
pub(crate) use crate::rpcs::v1_4_5::get_account::GET_ACCOUNT_METHOD;

/// Identifier of an account.
#[derive(Serialize, Deserialize, PartialEq, Eq, Debug, Clone)]
#[serde(deny_unknown_fields, untagged)]
pub enum AccountIdentifier {
    /// The public key of an account
    PublicKey(PublicKey),
    /// The account hash of an account
    AccountHash(AccountHash),
}

#[derive(Serialize, Deserialize, Debug)]
#[serde(deny_unknown_fields)]
pub(crate) struct GetAccountParams {
    /// The identifier of an Account. (named public key to match the JSON-RPC API)
    account_identifier: AccountIdentifier,
    /// The block identifier.
    block_identifier: Option<BlockIdentifier>,
}

impl GetAccountParams {
    pub(crate) fn new(
        account_identifier: AccountIdentifier,
        block_identifier: Option<BlockIdentifier>,
    ) -> Self {
        GetAccountParams {
            account_identifier,
            block_identifier,
        }
    }
}

/// The `result` field of a successful JSON-RPC response to a `state_get_account_info` request.
#[derive(Serialize, Deserialize, Debug, Clone)]
#[serde(deny_unknown_fields)]
pub struct GetAccountResult {
    /// The JSON-RPC server version.
    pub api_version: ProtocolVersion,
    /// The account.
    pub account: Account,
    /// The merkle proof of the value.
    pub merkle_proof: String,
}