rasn-compiler 0.5.3

use std::collections::BTreeMap;

use crate::{
    intermediate::{
        error::{GrammarError, GrammarErrorType},
        information_object::*,
        *,
    },
    lexer::asn1_value,
};

pub(crate) fn find_tld_or_enum_value_by_name(
    type_name: &String,
    name: &String,
    tlds: &BTreeMap<String, ToplevelDefinition>,
) -> Option<ASN1Value> {
    if let Some(ToplevelDefinition::Value(v)) = tlds.get(name) {
        return Some(v.value.clone());
    } else {
        for (_, tld) in tlds.iter() {
            if let Some(value) = tld.get_distinguished_or_enum_value(Some(type_name), name) {
                return Some(value);
            }
        }
        // Make second attempt without requiring a matching type name
        // This is the current best shot at linking inner subtypes
        for (_, tld) in tlds.iter() {
            if let Some(value) = tld.get_distinguished_or_enum_value(None, name) {
                return Some(value);
            }
        }
    }
    None
}

pub(crate) fn octet_string_to_bit_string(bytes: &[u8]) -> Vec<bool> {
    let mut bits = vec![];
    for byte in bytes {
        is_bit_set(*byte, 128, &mut bits);
    }
    bits
}

fn is_bit_set(rem: u8, limit: u8, bits: &mut Vec<bool>) {
    bits.push(rem >= limit);
    if limit >= 2 {
        is_bit_set(rem % limit, limit / 2, bits)
    }
}

pub(crate) fn bit_string_to_octet_string(bits: &[bool]) -> Result<Vec<u8>, GrammarError> {
    let mut octets = vec![];
    for byte in bits.chunks(8) {
        if byte.len() != 8 {
            return Err(GrammarError {
                details: "Binary octet string value needs to be a multiple of 8 bits!".into(),
                kind: GrammarErrorType::LinkerError,
            });
        }
        octets.push(byte.iter().enumerate().fold(0u8, |acc, (i, bit)| {
            acc + if *bit { 2u8.pow(7 - i as u32) } else { 0 }
        }));
    }
    Ok(octets)
}

pub(crate) fn walk_object_field_ref_path<'a>(
    fields: &'a Vec<InformationObjectClassField>,
    path: &'a Vec<ObjectFieldIdentifier>,
    mut index: usize,
) -> Option<&'a InformationObjectClassField> {
    fields
        .iter()
        .find_map(|f| {
            path.get(index).and_then(|id| {
                (&f.identifier == id).then(|| {
                    if path.len() == (index + 1) {
                        Some(f)
                    } else {
                        index += 1;
                        walk_object_field_ref_path(fields, path, index)
                    }
                })
            })
        })
        .flatten()
}

/// Resolves the custom syntax declared in an information object class' WITH SYNTAX clause
pub fn resolve_custom_syntax(
    fields: &mut InformationObjectFields,
    class: &InformationObjectClass,
) -> Result<(), GrammarError> {
    if let InformationObjectFields::CustomSyntax(application) = fields {
        let tokens = match &class.syntax {
            Some(s) => s.flatten(),
            None => {
                return Err(GrammarError {
                    details: "No syntax definition for information object class found!".into(),
                    kind: GrammarErrorType::LinkerError,
                })
            }
        };

        let mut unsorted_default_syntax = Vec::<(usize, InformationObjectField)>::new();

        let mut application_index = 0;
        'syntax_matching: for (i, (required, token)) in tokens.iter().enumerate() {
            if let Some(expr) = application.get_mut(application_index) {
                if expr.matches(token, &tokens, i) {
                    match expr {
                        SyntaxApplication::ObjectSetDeclaration(o) => {
                            if let Some(index) =
                                class.fields.iter().enumerate().find_map(|(i, v)| {
                                    (v.identifier
                                        == ObjectFieldIdentifier::MultipleValue(
                                            token.name_or_empty().to_owned(),
                                        ))
                                    .then_some(i)
                                })
                            {
                                unsorted_default_syntax.push((
                                    index,
                                    InformationObjectField::ObjectSetField(ObjectSetField {
                                        identifier: token.name_or_empty().to_owned(),
                                        value: o.clone(),
                                    }),
                                ));
                            }
                        }
                        SyntaxApplication::LiteralOrTypeReference(t) => {
                            if let Some(index) =
                                class.fields.iter().enumerate().find_map(|(i, v)| {
                                    (v.identifier
                                        == ObjectFieldIdentifier::MultipleValue(
                                            token.name_or_empty().to_owned(),
                                        ))
                                    .then_some(i)
                                })
                            {
                                unsorted_default_syntax.push((
                                    index,
                                    InformationObjectField::TypeField(TypeField {
                                        identifier: token.name_or_empty().to_owned(),
                                        ty: ASN1Type::ElsewhereDeclaredType(t.clone()),
                                    }),
                                ));
                            } else if let Some(index) =
                                class.fields.iter().enumerate().find_map(|(i, v)| {
                                    (v.identifier
                                        == ObjectFieldIdentifier::SingleValue(
                                            token.name_or_empty().to_owned(),
                                        ))
                                    .then_some(i)
                                })
                            {
                                unsorted_default_syntax.push((
                                    index,
                                    InformationObjectField::FixedValueField(FixedValueField {
                                        identifier: token.name_or_empty().to_owned(),
                                        value: match asn1_value(&t.identifier) {
                                            Ok((_, v)) => Ok(v),
                                            Err(e) => Err(GrammarError {
                                                details: format!("Syntax mismatch while resolving information object: {e:?}"),
                                                kind: GrammarErrorType::SyntaxMismatch,
                                            }),
                                        }?,
                                    }),
                                ));
                            }
                        }
                        SyntaxApplication::TypeReference(t) => {
                            if let Some(index) =
                                class.fields.iter().enumerate().find_map(|(i, v)| {
                                    (v.identifier
                                        == ObjectFieldIdentifier::MultipleValue(
                                            token.name_or_empty().to_owned(),
                                        ))
                                    .then_some(i)
                                })
                            {
                                unsorted_default_syntax.push((
                                    index,
                                    InformationObjectField::TypeField(TypeField {
                                        identifier: token.name_or_empty().to_owned(),
                                        ty: t.clone(),
                                    }),
                                ));
                            }
                        }
                        SyntaxApplication::ValueReference(v) => {
                            if let Some(index) =
                                class.fields.iter().enumerate().find_map(|(i, v)| {
                                    (v.identifier
                                        == ObjectFieldIdentifier::SingleValue(
                                            token.name_or_empty().to_owned(),
                                        ))
                                    .then_some(i)
                                })
                            {
                                unsorted_default_syntax.push((
                                    index,
                                    InformationObjectField::FixedValueField(FixedValueField {
                                        identifier: token.name_or_empty().to_owned(),
                                        value: v.clone(),
                                    }),
                                ));
                            }
                        }
                        _ => continue 'syntax_matching,
                    }
                    application_index += 1;
                } else if *required {
                    return Err(GrammarError {
                        details: "Syntax mismatch while resolving information object.".to_string(),
                        kind: GrammarErrorType::SyntaxMismatch,
                    });
                } else {
                    continue 'syntax_matching;
                }
            } else if *required {
                return Err(GrammarError {
                    details: "Syntax mismatch while resolving information object.".to_string(),
                    kind: GrammarErrorType::SyntaxMismatch,
                });
            } else {
                continue 'syntax_matching;
            }
        }
        unsorted_default_syntax.sort_by(|&(a, _), &(b, _)| a.cmp(&b));
        *fields = InformationObjectFields::DefaultSyntax(
            unsorted_default_syntax
                .into_iter()
                .map(|(_, field)| field)
                .collect(),
        );
    }
    Ok(())
}

#[cfg(test)]
mod tests {
    use crate::validator::linking::utils::octet_string_to_bit_string;

    #[test]
    fn converts_octet_to_bit_string() {
        assert_eq!(
            octet_string_to_bit_string(&[76]),
            vec![false, true, false, false, true, true, false, false]
        );
        assert_eq!(
            octet_string_to_bit_string(&[129]),
            vec![true, false, false, false, false, false, false, true]
        );
        assert_eq!(
            octet_string_to_bit_string(&[128]),
            vec![true, false, false, false, false, false, false, false]
        );
        assert_eq!(
            octet_string_to_bit_string(&[0]),
            vec![false, false, false, false, false, false, false, false]
        );
        assert_eq!(
            octet_string_to_bit_string(&[32]),
            vec![false, false, true, false, false, false, false, false]
        );
        assert_eq!(
            octet_string_to_bit_string(&[89, 45]),
            vec![
                false, true, false, true, true, false, false, true, false, false, true, false,
                true, true, false, true
            ]
        );
    }
}