auths-verifier 0.1.6

//! Detect diverging rotation events on a shared identity KEL.
//!
//! With a `kt=1` controller set and no witnesses, two controllers can each
//! sign a valid `rot` at the same sequence number independently. Both
//! rotations are cryptographically valid — there is no single source of
//! truth that orders them. Verifiers treat the KEL as duplicitous and
//! surface the conflict; the user resolves it out-of-band (typically by
//! running `auths device remove` on whichever side they trust).
//!
//! This module is read-only: it inspects a replay stream and reports
//! whether divergence is present. It never mutates state or rejects
//! otherwise-valid signatures. Callers decide policy (fail-open with a
//! warning, fail-closed, etc.) — the structured report is the contract.

use std::collections::HashMap;

use crate::types::IdentityDID;

/// A descriptor of one event as seen in a local replay stream.
///
/// Two events collide if they share `prefix` + `seq` but differ in
/// `said`. Same-SAID collisions are replicas of the same event and
/// never indicate duplicity.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct KelEventRef<'a> {
    /// The shared-KEL prefix (`did:keri:E…`) the event belongs to.
    pub prefix: &'a str,
    /// The event sequence number.
    pub seq: u64,
    /// The event's self-addressing identifier (the `d` field).
    pub said: &'a str,
}

/// Output of [`detect_duplicity`].
#[derive(Debug, Clone, PartialEq, Eq, serde::Serialize, serde::Deserialize)]
#[serde(tag = "type")]
pub enum DuplicityReport {
    /// No conflicting events seen.
    Clean,
    /// Two or more events at the same `seq` have different SAIDs.
    Diverging {
        /// The shared-KEL prefix the conflict is on.
        shared_kel_prefix: IdentityDID,
        /// The sequence number where divergence starts.
        seq: u64,
        /// The conflicting event SAIDs, in the order they were observed.
        event_saids: Vec<String>,
    },
}

impl DuplicityReport {
    /// `true` when the report represents an actual divergence.
    pub fn is_diverging(&self) -> bool {
        matches!(self, DuplicityReport::Diverging { .. })
    }
}

/// Scan `events` for same-prefix same-seq events with differing SAIDs.
///
/// Returns the first divergence found (lowest `seq`). Callers that need
/// all divergences can filter the input and call repeatedly; for the
/// Stage-1 UX path — single warning banner — first-wins is enough.
///
/// Args:
/// * `events`: Events to scan. Can be any subset / order; duplicates of
///   the same SAID (i.e., replicas) are tolerated.
///
/// Usage:
/// ```
/// use auths_verifier::duplicity::{KelEventRef, detect_duplicity, DuplicityReport};
///
/// let events = vec![
///     KelEventRef { prefix: "did:keri:EShared", seq: 1, said: "Ea" },
///     KelEventRef { prefix: "did:keri:EShared", seq: 2, said: "Eb" },
///     KelEventRef { prefix: "did:keri:EShared", seq: 2, said: "Ec" },
/// ];
/// match detect_duplicity(&events) {
///     DuplicityReport::Diverging { seq, .. } => assert_eq!(seq, 2),
///     _ => panic!("expected divergence"),
/// }
/// ```
pub fn detect_duplicity(events: &[KelEventRef<'_>]) -> DuplicityReport {
    // Group observed SAIDs by (prefix, seq) while preserving first-seen
    // order so the report deterministically returns SAIDs in the order
    // they appeared in the input.
    let mut seen: HashMap<(String, u64), Vec<String>> = HashMap::new();
    let mut first_conflict: Option<(String, u64)> = None;

    for ev in events {
        let key = (ev.prefix.to_string(), ev.seq);
        let saids = seen.entry(key.clone()).or_default();
        if !saids.iter().any(|s| s == ev.said) {
            saids.push(ev.said.to_string());
            if saids.len() >= 2 && first_conflict.is_none() {
                first_conflict = Some(key);
            }
        }
    }

    if let Some((prefix, seq)) = first_conflict {
        let saids = seen.remove(&(prefix.clone(), seq)).unwrap_or_default();
        let shared_kel_prefix = {
            #[allow(clippy::disallowed_methods)]
            IdentityDID::new_unchecked(prefix)
        };
        return DuplicityReport::Diverging {
            shared_kel_prefix,
            seq,
            event_saids: saids,
        };
    }

    DuplicityReport::Clean
}

/// Seal-bounding: `true` when `candidate` builds *past* a disputed position — its
/// sequence is beyond the divergence and its predecessor is one of the conflicting
/// heads. A seal-bounded producer or verifier refuses such an extension, so an
/// equivocator gains nothing durable from a fork it created, even before global
/// detection converges. A `Clean` report seals nothing. Ported from the `timeline_proof`
/// harness (`vdti::seal_rejects_extension`).
///
/// Args:
/// * `report`: The divergence that froze the position.
/// * `candidate_seq`: The sequence number of the event being appended.
/// * `candidate_prev_said`: The SAID the candidate links back to (its `p` field).
///
/// Usage:
/// ```
/// use auths_verifier::duplicity::{KelEventRef, detect_duplicity, seal_rejects_extension};
///
/// let events = vec![
///     KelEventRef { prefix: "did:keri:EShared", seq: 2, said: "ErotA" },
///     KelEventRef { prefix: "did:keri:EShared", seq: 2, said: "ErotB" },
/// ];
/// let report = detect_duplicity(&events);
/// assert!(seal_rejects_extension(&report, 3, "ErotA"));      // builds past the fork → refused
/// assert!(!seal_rejects_extension(&report, 3, "Eunrelated")); // unrelated predecessor → allowed
/// ```
pub fn seal_rejects_extension(
    report: &DuplicityReport,
    candidate_seq: u64,
    candidate_prev_said: &str,
) -> bool {
    match report {
        DuplicityReport::Diverging {
            seq, event_saids, ..
        } => candidate_seq > *seq && event_saids.iter().any(|s| s == candidate_prev_said),
        DuplicityReport::Clean => false,
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    fn ev<'a>(prefix: &'a str, seq: u64, said: &'a str) -> KelEventRef<'a> {
        KelEventRef { prefix, seq, said }
    }

    #[test]
    fn empty_is_clean() {
        assert_eq!(detect_duplicity(&[]), DuplicityReport::Clean);
    }

    #[test]
    fn single_event_is_clean() {
        let events = vec![ev("did:keri:E1", 0, "EaaaA")];
        assert_eq!(detect_duplicity(&events), DuplicityReport::Clean);
    }

    #[test]
    fn sequential_rots_are_clean() {
        // Different seq values are not duplicity — that's a normal chain.
        let events = vec![
            ev("did:keri:E1", 0, "EincpA"),
            ev("did:keri:E1", 1, "ErotA"),
            ev("did:keri:E1", 2, "ErotB"),
        ];
        assert_eq!(detect_duplicity(&events), DuplicityReport::Clean);
    }

    #[test]
    fn identical_said_at_same_seq_is_clean() {
        // Replicated event (same SAID seen twice — e.g., network redelivery)
        // must not trigger duplicity.
        let events = vec![ev("did:keri:E1", 2, "Erot"), ev("did:keri:E1", 2, "Erot")];
        assert_eq!(detect_duplicity(&events), DuplicityReport::Clean);
    }

    #[test]
    fn two_different_saids_at_same_seq_is_diverging() {
        let events = vec![
            ev("did:keri:E1", 0, "Eincp"),
            ev("did:keri:E1", 2, "ErotA"),
            ev("did:keri:E1", 2, "ErotB"),
        ];
        match detect_duplicity(&events) {
            DuplicityReport::Diverging {
                shared_kel_prefix,
                seq,
                event_saids,
            } => {
                assert_eq!(shared_kel_prefix.as_str(), "did:keri:E1");
                assert_eq!(seq, 2);
                assert_eq!(event_saids, vec!["ErotA".to_string(), "ErotB".to_string()]);
            }
            DuplicityReport::Clean => panic!("expected Diverging"),
        }
    }

    #[test]
    fn three_way_fork_reports_all_saids() {
        let events = vec![
            ev("did:keri:E1", 2, "ErotA"),
            ev("did:keri:E1", 2, "ErotB"),
            ev("did:keri:E1", 2, "ErotC"),
        ];
        match detect_duplicity(&events) {
            DuplicityReport::Diverging { event_saids, .. } => {
                assert_eq!(event_saids.len(), 3);
            }
            _ => panic!("expected three-way divergence"),
        }
    }

    #[test]
    fn icp_only_stream_is_clean() {
        let events = vec![ev("did:keri:E1", 0, "Eincp")];
        assert_eq!(detect_duplicity(&events), DuplicityReport::Clean);
    }

    #[test]
    fn concurrent_rotation_is_detected_or_prevented() {
        // The documented kt=1 accepted risk: with a single-signer shared KEL, two
        // controllers can each author a rotation at the same sequence, forking the KEL.
        // Authoring does not prevent it, so it must be surfaced — detect_duplicity flags
        // the two competing heads rather than silently accepting both as valid.
        let events = vec![
            ev("did:keri:EShared", 0, "Eicp"),
            ev("did:keri:EShared", 1, "Erot1"),
            ev("did:keri:EShared", 2, "ErotControllerA"),
            ev("did:keri:EShared", 2, "ErotControllerB"),
        ];
        match detect_duplicity(&events) {
            DuplicityReport::Diverging {
                shared_kel_prefix,
                seq,
                event_saids,
            } => {
                assert_eq!(shared_kel_prefix.as_str(), "did:keri:EShared");
                assert_eq!(seq, 2, "divergence is at the concurrent rotation");
                assert_eq!(event_saids.len(), 2);
            }
            DuplicityReport::Clean => {
                panic!("concurrent rotations at the same sequence must be flagged, not accepted")
            }
        }
    }

    #[test]
    fn diverging_report_is_diverging() {
        let report = DuplicityReport::Diverging {
            #[allow(clippy::disallowed_methods)]
            shared_kel_prefix: IdentityDID::new_unchecked("did:keri:E1".to_string()),
            seq: 2,
            event_saids: vec!["Ea".into(), "Eb".into()],
        };
        assert!(report.is_diverging());
        assert!(!DuplicityReport::Clean.is_diverging());
    }

    fn diverging_at_seq2() -> DuplicityReport {
        DuplicityReport::Diverging {
            #[allow(clippy::disallowed_methods)]
            shared_kel_prefix: IdentityDID::new_unchecked("did:keri:EShared".to_string()),
            seq: 2,
            event_saids: vec!["ErotA".into(), "ErotB".into()],
        }
    }

    #[test]
    fn seal_rejects_extension_built_on_a_disputed_head() {
        // An equivocator forks at seq 2, then tries to advance one branch at seq 3.
        // Seal-bounding refuses it — building past a disputed position gains nothing durable.
        let report = diverging_at_seq2();
        assert!(seal_rejects_extension(&report, 3, "ErotA"));
        assert!(seal_rejects_extension(&report, 3, "ErotB"));
    }

    #[test]
    fn seal_allows_extension_not_descending_from_the_dispute() {
        let report = diverging_at_seq2();
        // Builds on some other (undisputed) predecessor → this rule does not seal it.
        assert!(!seal_rejects_extension(&report, 3, "EunrelatedHead"));
        // At or below the disputed sequence is not "past" the dispute.
        assert!(!seal_rejects_extension(&report, 2, "ErotA"));
        assert!(!seal_rejects_extension(&report, 1, "ErotA"));
    }

    #[test]
    fn seal_rejects_nothing_when_clean() {
        assert!(!seal_rejects_extension(&DuplicityReport::Clean, 3, "ErotA"));
    }
}