eggsearch 0.3.2

//! Result-level types: trust labels and per-provider warnings.

use serde::{Deserialize, Serialize};

/// Trust label attached to every `SourceCard`. Live web results are
/// `ExternalUntrusted`. The `LocalTrusted` variant is reserved for
/// future local-index results; no current code path produces it.
#[derive(
    Clone, Copy, Debug, Default, Eq, PartialEq, Hash, Serialize, Deserialize, schemars::JsonSchema,
)]
#[serde(rename_all = "snake_case")]
pub enum TrustLevel {
    /// Live content fetched from the open web; treat as untrusted.
    #[default]
    ExternalUntrusted,
    /// Content originated from a configured local source (trusted by the operator).
    LocalTrusted,
    /// Trust level is unknown; default to caution.
    Unknown,
}

impl TrustLevel {
    /// Stable snake-case string form (e.g. `"external_untrusted"`).
    pub fn as_str(&self) -> &'static str {
        match self {
            Self::ExternalUntrusted => "external_untrusted",
            Self::LocalTrusted => "local_trusted",
            Self::Unknown => "unknown",
        }
    }
}

/// A warning emitted by a single provider during a search. Provider
/// failures are non-fatal: they are surfaced as warnings, not raised as
/// errors.
#[derive(Clone, Debug, PartialEq, Eq, Serialize, Deserialize, schemars::JsonSchema)]
pub struct SearchWarning {
    /// Stable provider id that emitted the warning.
    pub provider_id: String,
    /// Human-readable message.
    pub message: String,
}

impl SearchWarning {
    /// Construct a new `SearchWarning` from a provider id and message.
    pub fn new(provider_id: impl Into<String>, message: impl Into<String>) -> Self {
        Self {
            provider_id: provider_id.into(),
            message: message.into(),
        }
    }
}

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

    #[test]
    fn trust_level_as_str() {
        assert_eq!(TrustLevel::ExternalUntrusted.as_str(), "external_untrusted");
        assert_eq!(TrustLevel::LocalTrusted.as_str(), "local_trusted");
        assert_eq!(TrustLevel::Unknown.as_str(), "unknown");
    }

    #[test]
    fn trust_level_default_is_external_untrusted() {
        assert_eq!(TrustLevel::default(), TrustLevel::ExternalUntrusted);
    }

    #[test]
    fn search_warning_new() {
        let w = SearchWarning::new("brave", "rate limited");
        assert_eq!(w.provider_id, "brave");
        assert_eq!(w.message, "rate limited");
    }

    #[test]
    fn search_warning_serde_roundtrip() {
        let w = SearchWarning::new("duckduckgo", "timeout after 8s");
        let json = serde_json::to_string(&w).unwrap();
        let parsed: SearchWarning = serde_json::from_str(&json).unwrap();
        assert_eq!(parsed.provider_id, w.provider_id);
        assert_eq!(parsed.message, w.message);
    }

    #[test]
    fn trust_level_serde_snake_case() {
        let json = serde_json::to_string(&TrustLevel::ExternalUntrusted).unwrap();
        assert_eq!(json, "\"external_untrusted\"");
        let parsed: TrustLevel = serde_json::from_str(&json).unwrap();
        assert_eq!(parsed, TrustLevel::ExternalUntrusted);
    }

    #[test]
    fn trust_level_all_variants_serde() {
        for level in [
            TrustLevel::ExternalUntrusted,
            TrustLevel::LocalTrusted,
            TrustLevel::Unknown,
        ] {
            let json = serde_json::to_string(&level).unwrap();
            let parsed: TrustLevel = serde_json::from_str(&json).unwrap();
            assert_eq!(parsed, level);
        }
    }
}