use crate::util::{get_latest_activity_id, FederationQueueStateWithDomain};
use chrono::Local;
use lemmy_api_common::federate_retry_sleep_duration;
use lemmy_db_schema::{
  newtypes::InstanceId,
  utils::{ActualDbPool, DbPool},
};
use lemmy_utils::error::LemmyResult;
use std::{collections::HashMap, time::Duration};
use tokio::{sync::mpsc::UnboundedReceiver, time::interval};
use tracing::{debug, info, warn};

/// every 60s, print the state for every instance. exits if the receiver is done (all senders
/// dropped)
pub(crate) async fn receive_print_stats(
  pool: ActualDbPool,
  mut receiver: UnboundedReceiver<FederationQueueStateWithDomain>,
) {
  let pool = &mut DbPool::Pool(&pool);
  let mut printerval = interval(Duration::from_secs(60));
  let mut stats = HashMap::new();
  loop {
    tokio::select! {
      ele = receiver.recv() => {
        match ele {
          // update stats for instance
          Some(ele) => {stats.insert(ele.state.instance_id, ele);},
          // receiver closed, print stats and exit
          None => {
            print_stats(pool, &stats).await;
            return;
          }
        }
      },
      _ = printerval.tick() => {
        print_stats(pool, &stats).await;
      }
    }
  }
}

async fn print_stats(
  pool: &mut DbPool<'_>,
  stats: &HashMap<InstanceId, FederationQueueStateWithDomain>,
) {
  let res = print_stats_with_error(pool, stats).await;
  if let Err(e) = res {
    warn!("Failed to print stats: {e}");
  }
}

async fn print_stats_with_error(
  pool: &mut DbPool<'_>,
  stats: &HashMap<InstanceId, FederationQueueStateWithDomain>,
) -> LemmyResult<()> {
  let last_id = get_latest_activity_id(pool).await?;

  // it's expected that the values are a bit out of date, everything < SAVE_STATE_EVERY should be
  // considered up to date
  info!("Federation state as of {}:", Local::now().to_rfc3339());
  // todo: more stats (act/sec, avg http req duration)
  let mut ok_count = 0;
  let mut behind_count = 0;
  for ele in stats.values() {
    let stat = &ele.state;
    let domain = &ele.domain;
    let behind = last_id.0 - stat.last_successful_id.map(|e| e.0).unwrap_or(0);
    if stat.fail_count > 0 {
      info!(
        "{domain}: Warning. {behind} behind, {} consecutive fails, current retry delay {:.2?}",
        stat.fail_count,
        federate_retry_sleep_duration(stat.fail_count)
      );
    } else if behind > 0 {
      debug!("{}: Ok. {} activities behind", domain, behind);
      behind_count += 1;
    } else {
      ok_count += 1;
    }
  }
  info!("{ok_count} others up to date. {behind_count} instances behind.");
  Ok(())
}