librespot/metadata/src/restriction.rs

use std::{
    fmt::Debug,
    ops::{Deref, DerefMut},
};

use crate::util::impl_deref_wrapped;
use crate::util::{impl_from_repeated, impl_from_repeated_copy};

use protocol::metadata::Restriction as RestrictionMessage;

use librespot_protocol as protocol;
pub use protocol::metadata::Restriction_Catalogue as RestrictionCatalogue;
pub use protocol::metadata::Restriction_Type as RestrictionType;

#[derive(Debug, Clone)]
pub struct Restriction {
    pub catalogues: RestrictionCatalogues,
    pub restriction_type: RestrictionType,
    pub catalogue_strs: Vec<String>,
    pub countries_allowed: Option<Vec<String>>,
    pub countries_forbidden: Option<Vec<String>>,
}

#[derive(Debug, Clone, Default)]
pub struct Restrictions(pub Vec<Restriction>);

impl_deref_wrapped!(Restrictions, Vec<Restriction>);

#[derive(Debug, Clone)]
pub struct RestrictionCatalogues(pub Vec<RestrictionCatalogue>);

impl_deref_wrapped!(RestrictionCatalogues, Vec<RestrictionCatalogue>);

impl Restriction {
    fn parse_country_codes(country_codes: &str) -> Vec<String> {
        country_codes
            .chunks(2)
            .map(|country_code| country_code.to_owned())
            .collect()
    }
}

impl From<&RestrictionMessage> for Restriction {
    fn from(restriction: &RestrictionMessage) -> Self {
        let countries_allowed = if restriction.has_countries_allowed() {
            Some(Self::parse_country_codes(
                restriction.get_countries_allowed(),
            ))
        } else {
            None
        };

        let countries_forbidden = if restriction.has_countries_forbidden() {
            Some(Self::parse_country_codes(
                restriction.get_countries_forbidden(),
            ))
        } else {
            None
        };

        Self {
            catalogues: restriction.get_catalogue().into(),
            restriction_type: restriction.get_field_type(),
            catalogue_strs: restriction.get_catalogue_str().to_vec(),
            countries_allowed,
            countries_forbidden,
        }
    }
}

impl_from_repeated!(RestrictionMessage, Restrictions);
impl_from_repeated_copy!(RestrictionCatalogue, RestrictionCatalogues);

struct StrChunks<'s>(&'s str, usize);

trait StrChunksExt {
    fn chunks(&self, size: usize) -> StrChunks<'_>;
}

impl StrChunksExt for str {
    fn chunks(&self, size: usize) -> StrChunks<'_> {
        StrChunks(self, size)
    }
}

impl<'s> Iterator for StrChunks<'s> {
    type Item = &'s str;
    fn next(&mut self) -> Option<&'s str> {
        let &mut StrChunks(data, size) = self;
        if data.is_empty() {
            None
        } else {
            let ret = Some(&data[..size]);
            self.0 = &data[size..];
            ret
        }
    }
}