yantrikdb 0.15.1

//! **v0.10 Item 4a — typed provenance grammar + the anti-laundering consistency
//! matrix.**
//!
//! PURE and stateless: it parses a record's DECLARED provenance fields and
//! checks their internal consistency with no dependence on engine state, so a
//! leader and a follower reach the same verdict by construction. Wired into the
//! write paths in Item 4a.4.
//!
//! Scope + limitation (nuron/sol): the gate prevents DECLARED contradictions
//! (e.g. `source=inference` claiming `kind=fact`). It cannot verify truthful
//! provenance — a caller that lies (`source=user` for an inference) or omits
//! provenance is undetectable. Omission still cannot yield an
//! inference-claiming-fact because the matrix keys off the declared fields.

use crate::error::{Result, YantrikDbError};

/// Who/how a record was obtained. **Closed** vocabulary: an unparseable source
/// is rejected, so a caller cannot smuggle `source="inferece"` to dodge the
/// matrix. `Source` stays a `&str` at the engine API; the gate parses it.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum Source {
    User,
    Inference,
    Document,
    System,
}

impl Source {
    pub fn parse(s: &str) -> Result<Self> {
        match s.trim().to_ascii_lowercase().as_str() {
            "user" => Ok(Self::User),
            "inference" => Ok(Self::Inference),
            "document" => Ok(Self::Document),
            "system" => Ok(Self::System),
            other => Err(YantrikDbError::ProvenanceInconsistent {
                path: "source",
                reason: format!(
                    "unknown source '{other}' (expected user|inference|document|system)"
                ),
            }),
        }
    }

    /// The canonical stored form. Recall returns THIS (T06 "verbatim" means the
    /// canonical accepted value; a value that needed whitespace/case rescue is
    /// stored canonicalized — no provenance information is lost).
    pub fn as_str(self) -> &'static str {
        match self {
            Self::User => "user",
            Self::Inference => "inference",
            Self::Document => "document",
            Self::System => "system",
        }
    }
}

/// The justification tier. **Closed** vocabulary (unparseable → rejected).
/// `Asserted` — "a named source claimed it; I have not verified it" — is the
/// majority case ("user said X" / "doc states X") and its absence would force a
/// false `observation` stamp (nuron). `Learned` carries the model id.
#[derive(Debug, Clone, PartialEq, Eq)]
pub enum ConfidenceBasis {
    Observation,
    Asserted,
    Confirmation,
    Verification,
    Inference,
    Assumption,
    Learned { model: String },
}

impl ConfidenceBasis {
    pub fn parse(s: &str) -> Result<Self> {
        let t = s.trim();
        match t.to_ascii_lowercase().as_str() {
            "observation" => return Ok(Self::Observation),
            "asserted" => return Ok(Self::Asserted),
            "confirmation" => return Ok(Self::Confirmation),
            "verification" => return Ok(Self::Verification),
            "inference" => return Ok(Self::Inference),
            "assumption" => return Ok(Self::Assumption),
            _ => {}
        }
        if let Some(model) = parse_learned(t) {
            return Ok(Self::Learned { model });
        }
        Err(YantrikDbError::ProvenanceInconsistent {
            path: "confidence_basis",
            reason: format!(
                "unknown confidence_basis '{t}' (expected observation|asserted|confirmation|\
                 verification|inference|assumption|learned(<model>))"
            ),
        })
    }

    pub fn to_canonical(&self) -> String {
        match self {
            Self::Observation => "observation".into(),
            Self::Asserted => "asserted".into(),
            Self::Confirmation => "confirmation".into(),
            Self::Verification => "verification".into(),
            Self::Inference => "inference".into(),
            Self::Assumption => "assumption".into(),
            Self::Learned { model } => format!("learned({model})"),
        }
    }
}

/// `learned(<model>)` grammar: model is `[A-Za-z0-9._-]{1,64}` — non-empty,
/// bounded, ASCII only, no nested parens/unicode (sol r3 precision). Returns the
/// model string, or `None` if the shape does not match.
fn parse_learned(s: &str) -> Option<String> {
    // Case-INSENSITIVE "learned(" prefix, case-PRESERVING model. (`s` is
    // already trimmed by the caller.)
    let body = s.strip_suffix(')')?;
    if body.len() < 8 || !body.is_char_boundary(8) {
        return None;
    }
    let (prefix, inner) = body.split_at(8); // "learned(" is 8 ASCII bytes
    if !prefix.eq_ignore_ascii_case("learned(") {
        return None;
    }
    if inner.is_empty() || inner.len() > 64 {
        return None;
    }
    if !inner
        .chars()
        .all(|c| c.is_ascii_alphanumeric() || matches!(c, '.' | '_' | '-'))
    {
        return None;
    }
    Some(inner.to_string())
}

/// The record's `metadata.kind`. **Open** vocabulary: any string is accepted,
/// but only the PROTECTED tier {`fact`, `observation`, `inference`} participates
/// in the matrix. Missing / null / empty / non-string kind → `Unspecified`
/// (sol r3: cannot "reject an unparseable protected kind" under open vocab —
/// `fact.` / `inferece` are simply `Other`, not errors).
#[derive(Debug, Clone, PartialEq, Eq)]
pub enum ClaimKind {
    Fact,
    Observation,
    Inference,
    Unspecified,
    Other(String),
}

impl ClaimKind {
    /// Parse from the record's final plaintext `metadata.kind` (already merged,
    /// pre-encryption). `None` = key absent or non-string.
    pub fn parse(kind: Option<&str>) -> Self {
        match kind {
            None => Self::Unspecified,
            Some(k) => {
                let t = k.trim();
                if t.is_empty() {
                    return Self::Unspecified;
                }
                match t.to_ascii_lowercase().as_str() {
                    "fact" => Self::Fact,
                    "observation" => Self::Observation,
                    "inference" => Self::Inference,
                    _ => Self::Other(t.to_string()),
                }
            }
        }
    }

    /// The protected "authoritative claim" tier that an inference-source record
    /// may not assert without confirmation/verification or an explicit override.
    fn is_authoritative(&self) -> bool {
        matches!(self, Self::Fact | Self::Observation)
    }
}

/// Enforcement mode for the anti-laundering gate (Item 4a.4). Fresh installs
/// default to `Enforce`; migrated/legacy installs default to `Warn` (run the
/// check + nudge counter, never refuse) so upgrades don't break existing
/// callers. `Off` skips the check entirely.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum GateMode {
    Off,
    Warn,
    Enforce,
}

/// What the gate concluded about an ALLOWED write (4a.6b). `Enforce`
/// violations never produce a verdict — they are refused as `Err` before any
/// side effect. `Flagged` means warn-mode saw a violation and let it through;
/// the caller must report it via `note_flagged_write_committed` **after** the
/// write commits, so the warn→enforce nudge metric counts only writes that
/// actually landed. `#[must_use]`: dropping the verdict silently undercounts.
#[must_use = "a Flagged verdict must be ticked post-commit via note_flagged_write_committed"]
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum GateVerdict {
    Clean,
    Flagged,
}

/// Whether a `record_with_rid` call is an ORIGIN write or an already-admitted
/// APPLY (4a.6b, sol r2 finding 2). `record_with_rid` is BOTH the public
/// caller-supplied-rid origin API and the cluster commit-apply primitive, and
/// those want opposite gate behavior:
///
/// - **`Origin`** — a fresh write entering here for the first time. Runs the
///   anti-laundering gate exactly like `record()`; a declared contradiction is
///   refused in Enforce and flagged in Warn.
/// - **`Admitted`** — a consensus-committed / replicated op being applied. It
///   was already gated at the leader's origin ingress, so it MUST NOT be
///   re-gated: on the openraft apply path every follower re-applies the leader's
///   committed entry locally, and re-gating would make followers reject the
///   leader's write and wedge the cluster (yantrikdb-server's consensus-
///   admission review). The materializer drain and replication apply pass this.
///
/// This is a REQUIRED argument, not a defaulted flag, precisely so a caller
/// cannot silently inherit the wrong behavior: an apply caller that forgets to
/// pass `Admitted` gets a gated write (safe-ish), and an origin caller that
/// forgets `Origin` gets a compile error, not a silent bypass.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum WriteAdmission {
    Origin,
    Admitted,
}

impl GateMode {
    /// Parse a persisted mode. **Fail-CLOSED (sol 4a.4):** only an exact `off`
    /// yields `Off`. A malformed value (e.g. the typo `enforc`) must NOT
    /// silently disable the gate — that is the same fail-open class as a
    /// security check that quietly never fires — so it is a typed error the
    /// caller surfaces loudly.
    pub fn parse(s: &str) -> Result<Self> {
        match s.trim().to_ascii_lowercase().as_str() {
            "enforce" => Ok(Self::Enforce),
            "warn" => Ok(Self::Warn),
            "off" => Ok(Self::Off),
            other => Err(YantrikDbError::ProvenanceInconsistent {
                path: "provenance_gate_mode",
                reason: format!(
                    "malformed gate mode '{other}' (expected off|warn|enforce); refusing to \
                     silently disable the provenance gate"
                ),
            }),
        }
    }
    pub fn as_str(self) -> &'static str {
        match self {
            Self::Enforce => "enforce",
            Self::Warn => "warn",
            Self::Off => "off",
        }
    }
    pub fn as_u8(self) -> u8 {
        match self {
            Self::Off => 0,
            Self::Warn => 1,
            Self::Enforce => 2,
        }
    }
    pub fn from_u8(v: u8) -> Self {
        match v {
            2 => Self::Enforce,
            1 => Self::Warn,
            _ => Self::Off,
        }
    }
}

/// The anti-laundering consistency matrix (pure). Given already-parsed declared
/// fields, decide whether the record may be admitted. `basis = None` means the
/// caller declared no basis — which does NOT satisfy the confirmation/
/// verification allowance, so an inference still cannot claim fact/observation.
///
/// `override_kind` is the deliberate escape hatch; the DOCUMENTED escape is
/// instead to raise `confidence_basis` to confirmation/verification. **Scope of
/// the override's guarantee (sol 4a.4):** it is durably VISIBLE — it rides in
/// `metadata`, which is serialized onto the row and into the replicated oplog —
/// but it is NOT yet an engine-owned stamp carrying a reason/identity. Treat it
/// as an INTERIM AUDIT MARKER, not a privileged authorization mechanism: any
/// caller can set it, so it only carries weight where every writer is trusted
/// to invoke it honestly. A first-class stamp (engine-written, with reason and
/// actor) is follow-up work.
pub fn check_provenance_consistency_opt(
    source: Source,
    basis: Option<&ConfidenceBasis>,
    kind: &ClaimKind,
    override_kind: bool,
) -> Result<()> {
    if source == Source::Inference {
        // You cannot have OBSERVED an inference.
        if basis == Some(&ConfidenceBasis::Observation) {
            return Err(YantrikDbError::ProvenanceInconsistent {
                path: "source/confidence_basis",
                reason: "source=inference cannot have confidence_basis=observation \
                         (you did not observe an inference)"
                    .into(),
            });
        }
        // An inference cannot CLAIM to be a fact/observation unless it was
        // independently confirmed/verified (the "raise your basis" allowance)
        // or explicitly overridden (traced + stamped).
        if kind.is_authoritative()
            && !override_kind
            && !matches!(
                basis,
                Some(ConfidenceBasis::Confirmation | ConfidenceBasis::Verification)
            )
        {
            let kind_str = match kind {
                ClaimKind::Fact => "fact",
                _ => "observation",
            };
            return Err(YantrikDbError::ProvenanceInconsistent {
                path: "source/kind",
                reason: format!(
                    "source=inference cannot claim kind={kind_str} without confidence_basis in \
                     {{confirmation, verification}} or an explicit override_kind"
                ),
            });
        }
    }
    Ok(())
}

/// Convenience wrapper of [`check_provenance_consistency_opt`] for a present
/// basis (used by the exhaustive unit tests).
pub fn check_provenance_consistency(
    source: Source,
    basis: &ConfidenceBasis,
    kind: &ClaimKind,
    override_kind: bool,
) -> Result<()> {
    check_provenance_consistency_opt(source, Some(basis), kind, override_kind)
}

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

    #[test]
    fn source_parse_closed_vocab() {
        assert_eq!(Source::parse("user").unwrap(), Source::User);
        assert_eq!(Source::parse("  Inference ").unwrap(), Source::Inference);
        assert_eq!(Source::parse("DOCUMENT").unwrap(), Source::Document);
        // unparseable protected source is rejected (no laundering via typos)
        assert!(Source::parse("inferece").is_err());
        assert!(Source::parse("").is_err());
    }

    #[test]
    fn basis_parse_incl_asserted_and_learned() {
        assert_eq!(
            ConfidenceBasis::parse("asserted").unwrap(),
            ConfidenceBasis::Asserted
        );
        assert_eq!(
            ConfidenceBasis::parse(" Learned(bge-base-en-v1.5) ").unwrap(),
            ConfidenceBasis::Learned {
                model: "bge-base-en-v1.5".into()
            }
        );
        assert!(ConfidenceBasis::parse("learned()").is_err()); // empty model
        assert!(ConfidenceBasis::parse("learned(a b)").is_err()); // space illegal
        assert!(ConfidenceBasis::parse("nonsense").is_err());
        // canonical round-trip
        assert_eq!(
            ConfidenceBasis::parse("learned(m1)")
                .unwrap()
                .to_canonical(),
            "learned(m1)"
        );
    }

    #[test]
    fn kind_open_vocab_and_unspecified() {
        assert_eq!(ClaimKind::parse(Some("fact")), ClaimKind::Fact);
        assert_eq!(
            ClaimKind::parse(Some(" Observation ")),
            ClaimKind::Observation
        );
        // open vocab: non-protected is Other, not an error
        assert_eq!(
            ClaimKind::parse(Some("procedure")),
            ClaimKind::Other("procedure".into())
        );
        // fact. is NOT protected-fact (sol r3): it's Other
        assert_eq!(
            ClaimKind::parse(Some("fact.")),
            ClaimKind::Other("fact.".into())
        );
        assert_eq!(ClaimKind::parse(None), ClaimKind::Unspecified);
        assert_eq!(ClaimKind::parse(Some("")), ClaimKind::Unspecified);
        assert_eq!(ClaimKind::parse(Some("   ")), ClaimKind::Unspecified);
    }

    #[test]
    fn matrix_refuses_inference_observation_basis() {
        let e = check_provenance_consistency(
            Source::Inference,
            &ConfidenceBasis::Observation,
            &ClaimKind::Inference,
            false,
        );
        assert!(e.is_err(), "inference + observation basis must refuse");
    }

    #[test]
    fn matrix_refuses_inference_claiming_fact() {
        let e = check_provenance_consistency(
            Source::Inference,
            &ConfidenceBasis::Asserted,
            &ClaimKind::Fact,
            false,
        );
        assert!(
            e.is_err(),
            "inference claiming fact (no confirm/override) refused"
        );
    }

    #[test]
    fn matrix_confirmation_allowance() {
        // raise-your-basis: inference claiming fact IS allowed with confirmation
        assert!(check_provenance_consistency(
            Source::Inference,
            &ConfidenceBasis::Confirmation,
            &ClaimKind::Fact,
            false,
        )
        .is_ok());
        assert!(check_provenance_consistency(
            Source::Inference,
            &ConfidenceBasis::Verification,
            &ClaimKind::Observation,
            false,
        )
        .is_ok());
    }

    #[test]
    fn matrix_override_allowance() {
        assert!(check_provenance_consistency(
            Source::Inference,
            &ConfidenceBasis::Asserted,
            &ClaimKind::Fact,
            true, // explicit override
        )
        .is_ok());
    }

    #[test]
    fn matrix_allows_consistent_records() {
        // user asserting a fact — fine
        assert!(check_provenance_consistency(
            Source::User,
            &ConfidenceBasis::Asserted,
            &ClaimKind::Fact,
            false,
        )
        .is_ok());
        // inference labeled as an inference — fine
        assert!(check_provenance_consistency(
            Source::Inference,
            &ConfidenceBasis::Inference,
            &ClaimKind::Inference,
            false,
        )
        .is_ok());
        // inference with an unprotected kind — fine
        assert!(check_provenance_consistency(
            Source::Inference,
            &ConfidenceBasis::Asserted,
            &ClaimKind::Other("hypothesis".into()),
            false,
        )
        .is_ok());
        // unspecified kind never triggers the authoritative-claim rule
        assert!(check_provenance_consistency(
            Source::Inference,
            &ConfidenceBasis::Asserted,
            &ClaimKind::Unspecified,
            false,
        )
        .is_ok());
    }
}