radicle-heartwood-lfs/crates/radicle-cli/src/commands/stats.rs

mod args;

use std::path::Path;

use localtime::LocalDuration;
use localtime::LocalTime;
use radicle::git;
use radicle::issue::cache::Issues as _;
use radicle::node::address;
use radicle::node::routing;
use radicle::patch::cache::Patches as _;
use radicle::storage::{ReadRepository, ReadStorage, WriteRepository};
use radicle_term::Element;
use serde::Serialize;

use crate::terminal as term;

pub use args::Args;
pub(crate) use args::ABOUT;

#[derive(Default, Serialize)]
#[serde(rename_all = "camelCase")]
struct NodeStats {
    all: usize,
    public_daily: usize,
    online_daily: usize,
    online_weekly: usize,
    seeding_weekly: usize,
}

#[derive(Default, Serialize)]
#[serde(rename_all = "camelCase")]
struct LocalStats {
    repos: usize,
    issues: usize,
    patches: usize,
    pushes: usize,
    forks: usize,
}

#[derive(Default, Serialize)]
#[serde(rename_all = "camelCase")]
struct RepoStats {
    unique: usize,
    replicas: usize,
}

#[derive(Default, Serialize)]
#[serde(rename_all = "camelCase")]
struct Stats {
    local: LocalStats,
    repos: RepoStats,
    nodes: NodeStats,
}

pub fn run(_args: Args, ctx: impl term::Context) -> anyhow::Result<()> {
    let profile = ctx.profile()?;
    let storage = &profile.storage;
    let mut stats = Stats::default();

    for repo in storage.repositories()? {
        let repo = storage.repository(repo.rid)?;
        let issues = term::cob::issues(&profile, &repo)?.counts()?;
        let patches = term::cob::patches(&profile, &repo)?.counts()?;

        stats.local.issues += issues.total();
        stats.local.patches += patches.total();
        stats.local.repos += 1;

        for remote in repo.remote_ids()? {
            let remote = remote?;
            let sigrefs = repo.reference_oid(&remote, &git::refs::storage::SIGREFS_BRANCH)?;
            let mut walk = repo.raw().revwalk()?;
            walk.push(sigrefs.into())?;

            stats.local.pushes += walk.count();
            stats.local.forks += 1;
        }
    }

    let now = LocalTime::now();
    let db = profile.database()?;
    stats.nodes.all = address::Store::nodes(&db)?;
    stats.repos.replicas = routing::Store::len(&db)?;

    {
        let row = db
            .db
            .prepare("SELECT COUNT(DISTINCT repo) FROM routing")?
            // SAFETY: `COUNT` always returns a row.
            .into_iter()
            .next()
            .unwrap()?;
        let count = row.try_read::<i64, _>(0)? as usize;

        stats.repos.unique = count;
    }

    {
        let since = now - LocalDuration::from_mins(60 * 24); // 1 day.
        let mut stmt = db.db.prepare(
            "SELECT COUNT(DISTINCT node) FROM announcements WHERE timestamp >= ?1 AND timestamp < ?2",
        )?;
        stmt.bind((1, since.as_millis() as i64))?;
        stmt.bind((2, now.as_millis() as i64))?;

        // SAFETY: `COUNT` always returns a row.
        let row = stmt.iter().next().unwrap()?;
        stats.nodes.online_daily = row.try_read::<i64, _>(0)? as usize;

        let since = now - LocalDuration::from_mins(60 * 24 * 7); // 1 week.
        stmt.reset()?;
        stmt.bind((1, since.as_millis() as i64))?;
        stmt.bind((2, now.as_millis() as i64))?;

        let row = stmt.iter().next().unwrap()?;
        stats.nodes.online_weekly = row.try_read::<i64, _>(0)? as usize;
    }

    {
        let since = now - LocalDuration::from_mins(60 * 24); // 1 day.
        let mut stmt = db.db.prepare(
            "SELECT COUNT(DISTINCT ann.node) FROM announcements as ann
             JOIN addresses AS addr
             ON ann.node == addr.node
             WHERE ann.timestamp >= ?1 AND ann.timestamp < ?2",
        )?;
        stmt.bind((1, since.as_millis() as i64))?;
        stmt.bind((2, now.as_millis() as i64))?;

        let row = stmt
            .into_iter()
            .next()
            // SAFETY: `COUNT` always returns a row.
            .unwrap()?;
        let count = row.try_read::<i64, _>(0)? as usize;

        stats.nodes.public_daily = count;
    }

    {
        let since = now - LocalDuration::from_mins(60 * 24 * 7); // 1 week.
        let mut stmt = db.db.prepare(
            "SELECT COUNT(DISTINCT node) FROM routing
             WHERE timestamp >= ?1 AND timestamp < ?2",
        )?;
        stmt.bind((1, since.as_millis() as i64))?;
        stmt.bind((2, now.as_millis() as i64))?;

        let row = stmt
            .into_iter()
            .next()
            // SAFETY: `COUNT` always returns a row.
            .unwrap()?;
        let count = row.try_read::<i64, _>(0)? as usize;

        stats.nodes.seeding_weekly = count;
    }

    let output = term::json::to_pretty(&stats, Path::new("stats.json"))?;
    output.print();

    Ok(())
}