bc-xid 0.22.0

use std::collections::HashSet;

use bc_components::{
    PublicKeysProvider, Reference, ReferenceProvider, URI, XIDProvider,
};
use bc_envelope::{
    Envelope, EnvelopeEncodable,
    extension::{
        ALLOW_RAW, CAPABILITY, CAPABILITY_RAW, DELEGATE, DELEGATE_RAW, KEY,
        KEY_RAW, NAME, NAME_RAW,
    },
};

use crate::{Error, HasPermissions, Permissions, Privilege, Result};

#[derive(Debug, Clone, PartialEq, Eq)]
pub struct Service {
    uri: URI,
    key_references: HashSet<Reference>,
    delegate_references: HashSet<Reference>,
    permissions: Permissions,
    capability: String,
    name: String,
}

impl std::hash::Hash for Service {
    fn hash<H: std::hash::Hasher>(&self, state: &mut H) {
        self.uri.hash(state);
    }
}

impl Service {
    pub fn new(uri: impl AsRef<URI>) -> Self {
        Self {
            uri: uri.as_ref().clone(),
            key_references: HashSet::new(),
            delegate_references: HashSet::new(),
            permissions: Permissions::new(),
            capability: String::new(),
            name: String::new(),
        }
    }

    pub fn uri(&self) -> &URI { &self.uri }

    pub fn capability(&self) -> &str { &self.capability }

    pub fn set_capability(&mut self, capability: impl Into<String>) {
        self.capability = capability.into();
    }

    pub fn add_capability(&mut self, capability: &str) -> Result<()> {
        if !self.capability.is_empty() {
            return Err(Error::Duplicate { item: "capability".to_string() });
        }
        if capability.is_empty() {
            return Err(Error::EmptyValue { field: "capability".to_string() });
        }
        self.set_capability(capability);

        Ok(())
    }

    pub fn key_references(&self) -> &HashSet<Reference> { &self.key_references }

    pub fn key_referenecs_mut(&mut self) -> &mut HashSet<Reference> {
        &mut self.key_references
    }

    pub fn add_key_reference(
        &mut self,
        key_reference: impl AsRef<Reference>,
    ) -> Result<()> {
        if !self.key_references.contains(key_reference.as_ref()) {
            self.key_references.insert(*key_reference.as_ref());
        } else {
            return Err(Error::Duplicate { item: "key reference".to_string() });
        }

        Ok(())
    }

    pub fn add_key(&mut self, key: &dyn PublicKeysProvider) -> Result<()> {
        self.add_key_reference(key.public_keys().reference())
    }

    pub fn delegate_references(&self) -> &HashSet<Reference> {
        &self.delegate_references
    }

    pub fn delegate_references_mut(&mut self) -> &mut HashSet<Reference> {
        &mut self.delegate_references
    }

    pub fn add_delegate_reference(
        &mut self,
        delegate_reference: impl AsRef<Reference>,
    ) -> Result<()> {
        if !self
            .delegate_references
            .contains(delegate_reference.as_ref())
        {
            self.delegate_references
                .insert(*delegate_reference.as_ref());
        } else {
            return Err(Error::Duplicate {
                item: "delegate reference".to_string(),
            });
        }

        Ok(())
    }

    pub fn add_delegate(&mut self, delegate: &dyn XIDProvider) -> Result<()> {
        self.add_delegate_reference(delegate.xid().reference())
    }

    pub fn name(&self) -> &str { &self.name }

    pub fn set_name(&mut self, name: impl Into<String>) -> Result<()> {
        if !self.name.is_empty() {
            return Err(Error::Duplicate { item: "name".to_string() });
        }
        let name = name.into();
        if name.is_empty() {
            return Err(Error::EmptyValue { field: "name".to_string() });
        }
        self.name = name;
        Ok(())
    }
}

impl HasPermissions for Service {
    fn permissions(&self) -> &Permissions { &self.permissions }

    fn permissions_mut(&mut self) -> &mut Permissions { &mut self.permissions }
}

impl EnvelopeEncodable for Service {
    fn into_envelope(self) -> bc_envelope::Envelope {
        let mut envelope = Envelope::new(self.uri);

        envelope = self
            .key_references
            .iter()
            .cloned()
            .fold(envelope, |envelope, key| envelope.add_assertion(KEY, key));

        envelope = self
            .delegate_references
            .iter()
            .cloned()
            .fold(envelope, |envelope, delegate| {
                envelope.add_assertion(DELEGATE, delegate)
            });

        envelope =
            envelope.add_nonempty_string_assertion(CAPABILITY, self.capability);
        envelope = envelope.add_nonempty_string_assertion(NAME, self.name);
        self.permissions.add_to_envelope(envelope)
    }
}

impl TryFrom<Envelope> for Service {
    type Error = Error;

    fn try_from(envelope: Envelope) -> Result<Self> {
        Self::try_from(&envelope)
    }
}

impl TryFrom<&Envelope> for Service {
    type Error = Error;

    fn try_from(envelope: &Envelope) -> Result<Self> {
        let uri: URI = envelope.subject().try_leaf()?.try_into()?;

        let mut service = Service::new(uri);

        for assertion in envelope.assertions() {
            let predicate =
                assertion.try_predicate()?.try_known_value()?.value();
            let object = assertion.try_object()?;
            if object.has_assertions() {
                return Err(Error::UnexpectedNestedAssertions);
            }
            match predicate {
                KEY_RAW => {
                    let key = Reference::try_from(object.try_leaf()?)?;
                    service.add_key_reference(key)?;
                }
                DELEGATE_RAW => {
                    let delegate = Reference::try_from(object.try_leaf()?)?;
                    service.add_delegate_reference(delegate)?;
                }
                CAPABILITY_RAW => {
                    let capability = object.try_leaf()?.try_into_text()?;
                    service.add_capability(&capability)?;
                }
                NAME_RAW => {
                    let name = object.try_leaf()?.try_into_text()?;
                    service.set_name(&name)?;
                }
                ALLOW_RAW => {
                    service.add_allow(Privilege::try_from(object)?);
                }
                _ => {
                    return Err(Error::UnexpectedPredicate {
                        predicate: predicate.to_string(),
                    });
                }
            }
        }

        Ok(service)
    }
}