use rand_core::OsRng;

use curve25519_dalek::constants::ED25519_BASEPOINT_TABLE;

use serde_json::json;

use monero::{
  network::Network,
  util::{key::PublicKey, address::Address},
};

use monero_serai::{
  Protocol, random_scalar,
  rpc::{EmptyResponse, RpcError, Rpc},
};

pub async fn rpc() -> Rpc {
  let rpc = Rpc::new("http://127.0.0.1:18081".to_string());

  // Only run once
  if rpc.get_height().await.unwrap() != 1 {
    return rpc;
  }

  let addr = Address::standard(
    Network::Mainnet,
    PublicKey { point: (&random_scalar(&mut OsRng) * &ED25519_BASEPOINT_TABLE).compress() },
    PublicKey { point: (&random_scalar(&mut OsRng) * &ED25519_BASEPOINT_TABLE).compress() },
  )
  .to_string();

  // Mine 20 blocks to ensure decoy availability
  mine_block(&rpc, &addr).await.unwrap();
  mine_block(&rpc, &addr).await.unwrap();
  assert!(!matches!(rpc.get_protocol().await.unwrap(), Protocol::Unsupported));

  rpc
}

pub async fn mine_block(rpc: &Rpc, address: &str) -> Result<EmptyResponse, RpcError> {
  rpc
    .rpc_call(
      "json_rpc",
      Some(json!({
        "method": "generateblocks",
        "params": {
          "wallet_address": address,
          "amount_of_blocks": 10
        },
      })),
    )
    .await
}