use core::time::Duration;
use std::sync::{Arc, Mutex};

use rand_core::OsRng;

use frost::Participant;

use tokio::time::timeout;

use serai_client::primitives::BlockHash;

use serai_db::MemDb;

use crate::{
  coins::{OutputType, Output, Block, Coin},
  scanner::{ScannerEvent, Scanner, ScannerHandle},
};

pub async fn test_scanner<C: Coin>(coin: C) {
  let mut keys =
    frost::tests::key_gen::<_, C::Curve>(&mut OsRng).remove(&Participant::new(1).unwrap()).unwrap();
  C::tweak_keys(&mut keys);

  // Mine blocks so there's a confirmed block
  for _ in 0 .. C::CONFIRMATIONS {
    coin.mine_block().await;
  }

  let first = Arc::new(Mutex::new(true));
  let activation_number = coin.get_latest_block_number().await.unwrap();
  let db = MemDb::new();
  let new_scanner = || async {
    let (mut scanner, active_keys) = Scanner::new(coin.clone(), db.clone());
    let mut first = first.lock().unwrap();
    if *first {
      assert!(active_keys.is_empty());
      scanner.rotate_key(activation_number, keys.group_key()).await;
      *first = false;
    } else {
      assert_eq!(active_keys.len(), 1);
    }
    scanner
  };
  let scanner = new_scanner().await;

  // Receive funds
  let block = coin.test_send(C::address(keys.group_key())).await;
  let block_id = block.id();

  // Verify the Scanner picked them up
  let verify_event = |mut scanner: ScannerHandle<C, MemDb>| async {
    let outputs =
      match timeout(Duration::from_secs(30), scanner.events.recv()).await.unwrap().unwrap() {
        ScannerEvent::Block { key, block, batch, outputs } => {
          assert_eq!(key, keys.group_key());
          assert_eq!(block, block_id);
          assert_eq!(batch, 0);
          assert_eq!(outputs.len(), 1);
          assert_eq!(outputs[0].kind(), OutputType::External);
          outputs
        }
        ScannerEvent::Completed(_, _) => {
          panic!("unexpectedly got eventuality completion");
        }
      };
    (scanner, outputs)
  };
  let (mut scanner, outputs) = verify_event(scanner).await;

  // Create a new scanner off the current DB and verify it re-emits the above events
  verify_event(new_scanner().await).await;

  // Acknowledge the block

  // Acknowledging it should yield a list of all blocks since the last acknowledged block
  let mut blocks = vec![];
  let mut curr_block = activation_number + 1;
  loop {
    let block = coin.get_block(curr_block).await.unwrap().id();
    blocks.push(BlockHash(block.as_ref().try_into().unwrap()));
    if block == block_id {
      break;
    }
    curr_block += 1;
  }
  assert_eq!(scanner.ack_up_to_block(keys.group_key(), block_id).await, (blocks, outputs));

  // There should be no more events
  assert!(timeout(Duration::from_secs(30), scanner.events.recv()).await.is_err());

  // Create a new scanner off the current DB and make sure it also does nothing
  assert!(timeout(Duration::from_secs(30), new_scanner().await.events.recv()).await.is_err());
}