component-shape 0.3.2

use heck::ToSnakeCase as _;

/// Returns whether a component/prototyping suffix is a non-empty ASCII
/// identifier suffix.
pub const fn is_valid_ascii_identifier_suffix(value: &str) -> bool {
    let bytes = value.as_bytes();
    if bytes.is_empty() || (bytes.len() == 1 && bytes[0] == b'_') {
        return false;
    }
    if !is_ascii_ident_start(bytes[0]) {
        return false;
    }

    let mut idx = 1;
    while idx < bytes.len() {
        if !is_ascii_ident_continue(bytes[idx]) {
            return false;
        }
        idx += 1;
    }

    true
}

const fn is_ascii_ident_start(byte: u8) -> bool {
    byte == b'_' || byte.is_ascii_alphabetic()
}

const fn is_ascii_ident_continue(byte: u8) -> bool {
    is_ascii_ident_start(byte) || byte.is_ascii_digit()
}

/// Preferred generated component helper suffix.
#[derive(Clone, Copy, Debug, Eq, PartialEq)]
pub struct ComponentSuffix(&'static str);

#[derive(Clone, Debug, Eq, thiserror::Error, PartialEq)]
#[error("`field_suffix` must be a non-empty ASCII identifier suffix, got `{value}`")]
/// Error returned when component suffix validation rejects a value.
pub struct ComponentSuffixError {
    value: String,
}

impl ComponentSuffixError {
    /// Returns the invalid suffix value.
    pub fn value(&self) -> &str {
        &self.value
    }
}

/// Returns whether a component/prototyping suffix is valid.
pub const fn is_valid_component_suffix(value: &str) -> bool {
    is_valid_ascii_identifier_suffix(value)
}

/// Validate a component/prototyping suffix.
///
/// # Errors
///
/// Returns [`ComponentSuffixError`] when `value` is empty, is `_`, starts with a
/// non-identifier ASCII character, contains non-identifier ASCII characters, or
/// contains non-ASCII characters.
pub fn validate_component_suffix(value: &str) -> Result<(), ComponentSuffixError> {
    if is_valid_component_suffix(value) {
        Ok(())
    } else {
        Err(ComponentSuffixError {
            value: value.to_string(),
        })
    }
}

/// Derive a generated helper suffix from a field name and shape suffix source.
///
/// Both inputs are normalized to snake_case. If the suffix is the same as the
/// field name, this returns `None` so callers can use their own fallback. If the
/// suffix repeats the field name as a prefix, the duplicate prefix is removed:
/// `("email", "email_input")` becomes `Some("input")`.
pub fn component_suffix_from_suffix(field_name: &str, suffix: &str) -> Option<String> {
    let mut suffix = suffix.to_snake_case();
    let field_name = field_name.to_snake_case();

    if suffix == field_name {
        return None;
    }

    let field_prefix = format!("{field_name}_");
    if let Some(rest) = suffix.strip_prefix(&field_prefix) {
        suffix = rest.to_string();
    }

    (!suffix.is_empty()).then_some(suffix)
}

impl ComponentSuffix {
    /// Creates a component suffix from static metadata.
    ///
    /// # Panics
    ///
    /// Panics when `value` is not a valid component suffix. Use
    /// [`validate_component_suffix`] when the value comes from a recoverable
    /// runtime input boundary.
    pub const fn new(value: &'static str) -> Self {
        assert!(
            is_valid_component_suffix(value),
            "component suffix must be a non-empty ASCII identifier suffix"
        );
        Self(value)
    }

    /// Creates an optional component suffix from optional static metadata.
    ///
    /// # Panics
    ///
    /// Panics when `value` is `Some` and the contained string is not a valid
    /// component suffix.
    pub const fn new_opt(value: Option<&'static str>) -> Option<Self> {
        match value {
            Some(value) => Some(Self::new(value)),
            None => None,
        }
    }

    /// Returns the suffix text.
    pub const fn as_str(self) -> &'static str {
        self.0
    }
}

#[cfg(test)]
mod tests {
    use super::{
        ComponentSuffix, component_suffix_from_suffix, is_valid_component_suffix,
        validate_component_suffix,
    };

    #[test]
    fn component_suffix_validation_accepts_identifier_suffixes() {
        assert!(is_valid_component_suffix("input"));
        assert!(is_valid_component_suffix("number_input"));
        assert!(is_valid_component_suffix("_internal"));
        assert!(is_valid_component_suffix("input2"));
    }

    #[test]
    fn component_suffix_validation_rejects_invalid_suffixes() {
        for value in ["", "_", "2input", "input-field", "input field", "入力"] {
            let error = validate_component_suffix(value)
                .expect_err("invalid component suffix should be rejected");
            assert_eq!(error.value(), value);
        }
    }

    #[test]
    fn component_suffix_new_stores_valid_suffixes() {
        assert_eq!(ComponentSuffix::new("input").as_str(), "input");
        assert_eq!(
            ComponentSuffix::new_opt(Some("select")).map(ComponentSuffix::as_str),
            Some("select")
        );
        assert_eq!(ComponentSuffix::new_opt(None), None);
    }

    #[test]
    #[should_panic(expected = "component suffix must be a non-empty ASCII identifier suffix")]
    fn component_suffix_new_rejects_invalid_suffixes() {
        let _ = ComponentSuffix::new("input-field");
    }

    #[test]
    fn component_suffix_from_suffix_normalizes_suffixes() {
        assert_eq!(
            component_suffix_from_suffix("country", "CountrySelect"),
            Some("select".to_string())
        );
        assert_eq!(
            component_suffix_from_suffix("email", "email_input"),
            Some("input".to_string())
        );
        assert_eq!(component_suffix_from_suffix("tags", "tags"), None);
    }
}