oxirs-core 0.2.3

//! RDF Format Enumeration and Detection
//!
//! Extracted and adapted from OxiGraph oxrdfio with OxiRS enhancements.
//! Based on W3C RDF specifications and IANA media type registry.

use serde::{Deserialize, Serialize};
use std::fmt;

/// JSON-LD profile for enhanced features
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, Serialize, Deserialize)]
pub enum JsonLdProfile {
    /// Standard JSON-LD processing
    Standard,
    /// Expanded JSON-LD (no compaction)
    Expanded,
    /// Compacted JSON-LD
    Compacted,
    /// Flattened JSON-LD
    Flattened,
    /// Streaming JSON-LD
    Streaming,
}

impl JsonLdProfile {
    /// Get profile from IRI
    pub fn from_iri(iri: &str) -> Option<Self> {
        match iri {
            "http://www.w3.org/ns/json-ld#expanded" => Some(Self::Expanded),
            "http://www.w3.org/ns/json-ld#compacted" => Some(Self::Compacted),
            "http://www.w3.org/ns/json-ld#flattened" => Some(Self::Flattened),
            "http://www.w3.org/ns/json-ld#streaming" => Some(Self::Streaming),
            _ => None,
        }
    }

    /// Get IRI for profile
    pub fn iri(&self) -> &'static str {
        match self {
            Self::Standard => "http://www.w3.org/ns/json-ld#standard",
            Self::Expanded => "http://www.w3.org/ns/json-ld#expanded",
            Self::Compacted => "http://www.w3.org/ns/json-ld#compacted",
            Self::Flattened => "http://www.w3.org/ns/json-ld#flattened",
            Self::Streaming => "http://www.w3.org/ns/json-ld#streaming",
        }
    }
}

/// Set of JSON-LD profiles
#[derive(Debug, Clone, PartialEq, Eq, Hash, Serialize, Deserialize)]
pub struct JsonLdProfileSet {
    profiles: Vec<JsonLdProfile>,
}

impl JsonLdProfileSet {
    /// Create empty profile set
    pub const fn empty() -> Self {
        Self {
            profiles: Vec::new(),
        }
    }

    /// Create from single profile
    pub fn from_profile(profile: JsonLdProfile) -> Self {
        Self {
            profiles: vec![profile],
        }
    }

    /// Check if contains profile
    pub fn contains(&self, profile: JsonLdProfile) -> bool {
        self.profiles.contains(&profile)
    }

    /// Add profile to set
    pub fn insert(&mut self, profile: JsonLdProfile) {
        if !self.contains(profile) {
            self.profiles.push(profile);
        }
    }

    /// Get all profiles
    pub fn profiles(&self) -> &[JsonLdProfile] {
        &self.profiles
    }

    /// Convert to jsonld module's JsonLdProfileSet (bitfield-based)
    ///
    /// Maps format::JsonLdProfile to jsonld::JsonLdProfile for compatibility.
    pub fn to_jsonld_profile_set(&self) -> crate::jsonld::JsonLdProfileSet {
        use crate::jsonld;
        let mut set = jsonld::JsonLdProfileSet::empty();

        for profile in &self.profiles {
            let jsonld_profile = match profile {
                JsonLdProfile::Standard => continue, // No direct equivalent, skip
                JsonLdProfile::Expanded => jsonld::JsonLdProfile::Expanded,
                JsonLdProfile::Compacted => jsonld::JsonLdProfile::Compacted,
                JsonLdProfile::Flattened => jsonld::JsonLdProfile::Flattened,
                JsonLdProfile::Streaming => jsonld::JsonLdProfile::Streaming,
            };
            set |= jsonld_profile;
        }

        set
    }

    /// Create from jsonld module's JsonLdProfileSet
    ///
    /// Converts jsonld::JsonLdProfileSet to format::JsonLdProfileSet.
    pub fn from_jsonld_profile_set(jsonld_set: crate::jsonld::JsonLdProfileSet) -> Self {
        use crate::jsonld;
        let mut profiles = Vec::new();

        for jsonld_profile in jsonld_set {
            let profile = match jsonld_profile {
                jsonld::JsonLdProfile::Expanded => JsonLdProfile::Expanded,
                jsonld::JsonLdProfile::Compacted => JsonLdProfile::Compacted,
                jsonld::JsonLdProfile::Flattened => JsonLdProfile::Flattened,
                jsonld::JsonLdProfile::Streaming => JsonLdProfile::Streaming,
                // Context, Frame, Framed don't have direct equivalents
                jsonld::JsonLdProfile::Context => continue,
                jsonld::JsonLdProfile::Frame => continue,
                jsonld::JsonLdProfile::Framed => continue,
            };
            profiles.push(profile);
        }

        Self { profiles }
    }
}

impl From<JsonLdProfile> for JsonLdProfileSet {
    fn from(profile: JsonLdProfile) -> Self {
        Self::from_profile(profile)
    }
}

impl std::ops::BitOr for JsonLdProfile {
    type Output = JsonLdProfileSet;

    fn bitor(self, rhs: Self) -> Self::Output {
        let mut set = JsonLdProfileSet::from_profile(self);
        set.insert(rhs);
        set
    }
}

impl std::ops::BitOrAssign<JsonLdProfile> for JsonLdProfileSet {
    fn bitor_assign(&mut self, rhs: JsonLdProfile) {
        self.insert(rhs);
    }
}

/// RDF serialization formats
///
/// This enumeration covers all major RDF serialization formats supported by OxiRS.
/// Based on W3C specifications and community standards.
#[derive(Eq, PartialEq, Debug, Clone, Hash, Serialize, Deserialize, Default)]
#[non_exhaustive]
pub enum RdfFormat {
    /// [N3](https://w3c.github.io/N3/spec/)
    N3,
    /// [N-Quads](https://www.w3.org/TR/n-quads/)
    NQuads,
    /// [N-Triples](https://www.w3.org/TR/n-triples/)
    NTriples,
    /// [RDF/XML](https://www.w3.org/TR/rdf-syntax-grammar/)
    RdfXml,
    /// [TriG](https://www.w3.org/TR/trig/)
    TriG,
    /// [Turtle](https://www.w3.org/TR/turtle/)
    #[default]
    Turtle,
    /// [JSON-LD](https://www.w3.org/TR/json-ld/) with optional profiles
    JsonLd { profile: JsonLdProfileSet },
}

impl RdfFormat {
    /// The format canonical IRI according to the [Unique URIs for file formats registry](https://www.w3.org/ns/formats/).
    ///
    /// ```
    /// use oxirs_core::format::RdfFormat;
    ///
    /// assert_eq!(
    ///     RdfFormat::NTriples.iri(),
    ///     "http://www.w3.org/ns/formats/N-Triples"
    /// )
    /// ```
    pub const fn iri(&self) -> &'static str {
        match self {
            Self::JsonLd { .. } => "https://www.w3.org/ns/formats/data/JSON-LD",
            Self::N3 => "http://www.w3.org/ns/formats/N3",
            Self::NQuads => "http://www.w3.org/ns/formats/N-Quads",
            Self::NTriples => "http://www.w3.org/ns/formats/N-Triples",
            Self::RdfXml => "http://www.w3.org/ns/formats/RDF_XML",
            Self::TriG => "http://www.w3.org/ns/formats/TriG",
            Self::Turtle => "http://www.w3.org/ns/formats/Turtle",
        }
    }

    /// The format [IANA media type](https://tools.ietf.org/html/rfc2046).
    ///
    /// ```
    /// use oxirs_core::format::RdfFormat;
    ///
    /// assert_eq!(RdfFormat::NTriples.media_type(), "application/n-triples")
    /// ```
    pub fn media_type(&self) -> &'static str {
        match self {
            Self::JsonLd { profile } => {
                if profile.contains(JsonLdProfile::Streaming) {
                    "application/ld+json;profile=http://www.w3.org/ns/json-ld#streaming"
                } else {
                    "application/ld+json"
                }
            }
            Self::N3 => "text/n3",
            Self::NQuads => "application/n-quads",
            Self::NTriples => "application/n-triples",
            Self::RdfXml => "application/rdf+xml",
            Self::TriG => "application/trig",
            Self::Turtle => "text/turtle",
        }
    }

    /// The format [IANA-registered](https://tools.ietf.org/html/rfc2046) file extension.
    ///
    /// ```
    /// use oxirs_core::format::RdfFormat;
    ///
    /// assert_eq!(RdfFormat::NTriples.file_extension(), "nt")
    /// ```
    pub const fn file_extension(&self) -> &'static str {
        match self {
            Self::JsonLd { .. } => "jsonld",
            Self::N3 => "n3",
            Self::NQuads => "nq",
            Self::NTriples => "nt",
            Self::RdfXml => "rdf",
            Self::TriG => "trig",
            Self::Turtle => "ttl",
        }
    }

    /// The format name.
    ///
    /// ```
    /// use oxirs_core::format::RdfFormat;
    ///
    /// assert_eq!(RdfFormat::NTriples.name(), "N-Triples")
    /// ```
    pub fn name(&self) -> &'static str {
        match self {
            Self::JsonLd { profile } => {
                if profile.contains(JsonLdProfile::Streaming) {
                    "Streaming JSON-LD"
                } else {
                    "JSON-LD"
                }
            }
            Self::N3 => "N3",
            Self::NQuads => "N-Quads",
            Self::NTriples => "N-Triples",
            Self::RdfXml => "RDF/XML",
            Self::TriG => "TriG",
            Self::Turtle => "Turtle",
        }
    }

    /// Checks if the format supports [RDF datasets](https://www.w3.org/TR/rdf11-concepts/#dfn-rdf-dataset) and not only [RDF graphs](https://www.w3.org/TR/rdf11-concepts/#dfn-rdf-graph).
    ///
    /// ```
    /// use oxirs_core::format::RdfFormat;
    ///
    /// assert_eq!(RdfFormat::NTriples.supports_datasets(), false);
    /// assert_eq!(RdfFormat::NQuads.supports_datasets(), true);
    /// ```
    pub const fn supports_datasets(&self) -> bool {
        matches!(self, Self::JsonLd { .. } | Self::NQuads | Self::TriG)
    }

    /// Checks if the format supports RDF-star (quoted triples).
    ///
    /// ```
    /// use oxirs_core::format::RdfFormat;
    ///
    /// assert_eq!(RdfFormat::Turtle.supports_rdf_star(), true);
    /// assert_eq!(RdfFormat::RdfXml.supports_rdf_star(), false);
    /// ```
    pub const fn supports_rdf_star(&self) -> bool {
        matches!(
            self,
            Self::NTriples | Self::NQuads | Self::Turtle | Self::TriG
        )
    }

    /// Looks for a known format from a media type.
    ///
    /// It supports some media type aliases.
    /// For example, "application/xml" is going to return `RdfFormat::RdfXml` even if it is not its canonical media type.
    ///
    /// Example:
    /// ```
    /// use oxirs_core::format::{RdfFormat, JsonLdProfile};
    ///
    /// assert_eq!(
    ///     RdfFormat::from_media_type("text/turtle; charset=utf-8"),
    ///     Some(RdfFormat::Turtle)
    /// );
    /// assert_eq!(
    ///     RdfFormat::from_media_type(
    ///         "application/ld+json ; profile = http://www.w3.org/ns/json-ld#streaming"
    ///     ),
    ///     Some(RdfFormat::JsonLd {
    ///         profile: JsonLdProfile::Streaming.into()
    ///     })
    /// )
    /// ```
    pub fn from_media_type(media_type: &str) -> Option<Self> {
        const MEDIA_SUBTYPES: [(&str, RdfFormat); 14] = [
            (
                "activity+json",
                RdfFormat::JsonLd {
                    profile: JsonLdProfileSet::empty(),
                },
            ),
            (
                "json",
                RdfFormat::JsonLd {
                    profile: JsonLdProfileSet::empty(),
                },
            ),
            (
                "ld+json",
                RdfFormat::JsonLd {
                    profile: JsonLdProfileSet::empty(),
                },
            ),
            (
                "jsonld",
                RdfFormat::JsonLd {
                    profile: JsonLdProfileSet::empty(),
                },
            ),
            ("n-quads", RdfFormat::NQuads),
            ("n-triples", RdfFormat::NTriples),
            ("n3", RdfFormat::N3),
            ("nquads", RdfFormat::NQuads),
            ("ntriples", RdfFormat::NTriples),
            ("plain", RdfFormat::NTriples),
            ("rdf+xml", RdfFormat::RdfXml),
            ("trig", RdfFormat::TriG),
            ("turtle", RdfFormat::Turtle),
            ("xml", RdfFormat::RdfXml),
        ];
        const UTF8_CHARSETS: [&str; 3] = ["ascii", "utf8", "utf-8"];

        let (type_subtype, parameters) = media_type.split_once(';').unwrap_or((media_type, ""));

        let (r#type, subtype) = type_subtype.split_once('/')?;
        let r#type = r#type.trim();
        if !r#type.eq_ignore_ascii_case("application") && !r#type.eq_ignore_ascii_case("text") {
            return None;
        }
        let subtype = subtype.trim();
        let subtype = subtype.strip_prefix("x-").unwrap_or(subtype);

        let parameters = parameters.trim();
        let parameters = if parameters.is_empty() {
            Vec::new()
        } else {
            parameters
                .split(';')
                .map(|p| {
                    let (key, value) = p.split_once('=')?;
                    Some((key.trim(), value.trim()))
                })
                .collect::<Option<Vec<_>>>()?
        };

        for (candidate_subtype, mut candidate_id) in MEDIA_SUBTYPES {
            if candidate_subtype.eq_ignore_ascii_case(subtype) {
                // We have a look at parameters
                for (key, mut value) in parameters {
                    match key {
                        "charset"
                            if !UTF8_CHARSETS.iter().any(|c| c.eq_ignore_ascii_case(value)) =>
                        {
                            return None; // No other charset than UTF-8 is supported
                        }
                        "profile" => {
                            // We remove enclosing double quotes
                            if value.starts_with('"') && value.ends_with('"') {
                                value = &value[1..value.len() - 1];
                            }
                            if let RdfFormat::JsonLd { profile } = &mut candidate_id {
                                for value in value.split(' ') {
                                    if let Some(value) = JsonLdProfile::from_iri(value.trim()) {
                                        profile.insert(value);
                                    }
                                }
                            }
                        }
                        _ => (), // We ignore
                    }
                }
                return Some(candidate_id);
            }
        }
        None
    }

    /// Looks for a known format from an extension.
    ///
    /// It supports some aliases.
    ///
    /// Example:
    /// ```
    /// use oxirs_core::format::RdfFormat;
    ///
    /// assert_eq!(RdfFormat::from_extension("nt"), Some(RdfFormat::NTriples))
    /// ```
    pub fn from_extension(extension: &str) -> Option<Self> {
        const EXTENSIONS: [(&str, RdfFormat); 10] = [
            (
                "json",
                RdfFormat::JsonLd {
                    profile: JsonLdProfileSet::empty(),
                },
            ),
            (
                "jsonld",
                RdfFormat::JsonLd {
                    profile: JsonLdProfileSet::empty(),
                },
            ),
            ("n3", RdfFormat::N3),
            ("nq", RdfFormat::NQuads),
            ("nt", RdfFormat::NTriples),
            ("rdf", RdfFormat::RdfXml),
            ("trig", RdfFormat::TriG),
            ("ttl", RdfFormat::Turtle),
            ("txt", RdfFormat::NTriples),
            ("xml", RdfFormat::RdfXml),
        ];
        for (candidate_extension, candidate_id) in EXTENSIONS {
            if candidate_extension.eq_ignore_ascii_case(extension) {
                return Some(candidate_id);
            }
        }
        None
    }
}

impl fmt::Display for RdfFormat {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        f.write_str(self.name())
    }
}

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

    #[test]
    fn test_from_media_type() {
        assert_eq!(RdfFormat::from_media_type("foo/bar"), None);
        assert_eq!(RdfFormat::from_media_type("text/csv"), None);
        assert_eq!(
            RdfFormat::from_media_type("text/turtle"),
            Some(RdfFormat::Turtle)
        );
        assert_eq!(
            RdfFormat::from_media_type("application/x-turtle"),
            Some(RdfFormat::Turtle)
        );
        assert_eq!(
            RdfFormat::from_media_type("application/ld+json"),
            Some(RdfFormat::JsonLd {
                profile: JsonLdProfileSet::empty()
            })
        );
        assert_eq!(
            RdfFormat::from_media_type("application/ld+json;profile=foo"),
            Some(RdfFormat::JsonLd {
                profile: JsonLdProfileSet::empty()
            })
        );
        assert_eq!(
            RdfFormat::from_media_type(
                "application/ld+json;profile=http://www.w3.org/ns/json-ld#streaming"
            ),
            Some(RdfFormat::JsonLd {
                profile: JsonLdProfile::Streaming.into()
            })
        );
    }

    #[test]
    fn test_from_extension() {
        assert_eq!(RdfFormat::from_extension("ttl"), Some(RdfFormat::Turtle));
        assert_eq!(RdfFormat::from_extension("nt"), Some(RdfFormat::NTriples));
        assert_eq!(RdfFormat::from_extension("nq"), Some(RdfFormat::NQuads));
        assert_eq!(RdfFormat::from_extension("rdf"), Some(RdfFormat::RdfXml));
        assert_eq!(
            RdfFormat::from_extension("jsonld"),
            Some(RdfFormat::JsonLd {
                profile: JsonLdProfileSet::empty()
            })
        );
        assert_eq!(RdfFormat::from_extension("unknown"), None);
    }

    #[test]
    fn test_format_properties() {
        assert!(RdfFormat::NQuads.supports_datasets());
        assert!(!RdfFormat::NTriples.supports_datasets());

        assert!(RdfFormat::Turtle.supports_rdf_star());
        assert!(!RdfFormat::RdfXml.supports_rdf_star());

        assert_eq!(RdfFormat::Turtle.file_extension(), "ttl");
        assert_eq!(RdfFormat::NTriples.media_type(), "application/n-triples");
        assert_eq!(RdfFormat::Turtle.name(), "Turtle");
    }

    #[test]
    fn test_jsonld_profiles() {
        let mut profile_set = JsonLdProfileSet::empty();
        assert!(!profile_set.contains(JsonLdProfile::Streaming));

        profile_set.insert(JsonLdProfile::Streaming);
        assert!(profile_set.contains(JsonLdProfile::Streaming));

        let combined = JsonLdProfile::Streaming | JsonLdProfile::Expanded;
        assert!(combined.contains(JsonLdProfile::Streaming));
        assert!(combined.contains(JsonLdProfile::Expanded));
    }
}