#![deny(missing_debug_implementations)]
#![cfg_attr(
    not(any(feature = "mdbx", feature = "lmdb")),
    allow(unused, clippy::drop_non_drop)
)]

mod array;
mod attestation_queue;
mod attester_record;
mod batch_stats;
mod block_queue;
pub mod config;
mod database;
mod error;
pub mod metrics;
mod migrate;
mod slasher;
pub mod test_utils;

pub use crate::slasher::Slasher;
pub use attestation_queue::{AttestationBatch, AttestationQueue, SimpleBatch};
pub use attester_record::{AttesterRecord, CompactAttesterRecord, IndexedAttesterRecord};
pub use block_queue::BlockQueue;
pub use config::{Config, DatabaseBackend, DatabaseBackendOverride};
pub use database::{
    interface::{Database, Environment, RwTransaction},
    IndexedAttestationId, SlasherDB,
};
pub use error::Error;

use types::{AttesterSlashing, AttesterSlashingBase, AttesterSlashingElectra};
use types::{EthSpec, IndexedAttestation, ProposerSlashing};

#[derive(Debug, PartialEq)]
pub enum AttesterSlashingStatus<E: EthSpec> {
    NotSlashable,
    /// A weird outcome that can occur when we go to lookup an attestation by its target
    /// epoch for a surround slashing, but find a different attestation -- indicating that
    /// the validator has already been caught double voting.
    AlreadyDoubleVoted,
    DoubleVote(Box<IndexedAttestation<E>>),
    SurroundsExisting(Box<IndexedAttestation<E>>),
    SurroundedByExisting(Box<IndexedAttestation<E>>),
}

#[derive(Debug, PartialEq)]
pub enum ProposerSlashingStatus {
    NotSlashable,
    DoubleVote(Box<ProposerSlashing>),
}

impl<E: EthSpec> AttesterSlashingStatus<E> {
    pub fn into_slashing(
        self,
        new_attestation: &IndexedAttestation<E>,
    ) -> Option<AttesterSlashing<E>> {
        use AttesterSlashingStatus::*;

        // The surrounding attestation must be in `attestation_1` to be valid.
        match self {
            NotSlashable => None,
            AlreadyDoubleVoted => None,
            DoubleVote(existing) | SurroundedByExisting(existing) => {
                match (*existing, new_attestation) {
                    // TODO(electra) - determine when we would convert a Base attestation to Electra / how to handle mismatched attestations here
                    (IndexedAttestation::Base(existing_att), IndexedAttestation::Base(new_att)) => {
                        Some(AttesterSlashing::Base(AttesterSlashingBase {
                            attestation_1: existing_att,
                            attestation_2: new_att.clone(),
                        }))
                    }
                    (
                        IndexedAttestation::Electra(existing_att),
                        IndexedAttestation::Electra(new_att),
                    ) => Some(AttesterSlashing::Electra(AttesterSlashingElectra {
                        attestation_1: existing_att,
                        attestation_2: new_att.clone(),
                    })),
                    _ => panic!("attestations must be of the same type"),
                }
            }
            // TODO(electra): fix this once we superstruct IndexedAttestation (return the correct type)
            SurroundsExisting(existing) => match (*existing, new_attestation) {
                (IndexedAttestation::Base(existing_att), IndexedAttestation::Base(new_att)) => {
                    Some(AttesterSlashing::Base(AttesterSlashingBase {
                        attestation_1: new_att.clone(),
                        attestation_2: existing_att,
                    }))
                }
                (
                    IndexedAttestation::Electra(existing_att),
                    IndexedAttestation::Electra(new_att),
                ) => Some(AttesterSlashing::Electra(AttesterSlashingElectra {
                    attestation_1: new_att.clone(),
                    attestation_2: existing_att,
                })),
                _ => panic!("attestations must be of the same type"),
            },
            /*
            Some(AttesterSlashing::Base(AttesterSlashingBase {
                attestation_1: new_attestation.clone(),
                attestation_2: *existing,
            })),
            */
        }
    }
}