aven-core 0.1.36

#![allow(dead_code)]

use anyhow::{Result, bail};

pub const MAX_BLOB_BYTES: usize = 25 * 1024 * 1024;
pub const MAX_FILENAME_LEN: usize = 255;
pub const MAX_ALT_TEXT_LEN: usize = 500;
pub const SUPPORTED_MEDIA_TYPES: &[&str] = &["image/png", "image/jpeg", "image/gif", "image/webp"];

pub fn validate_attachment_id(value: &str) -> Result<()> {
    if value.len() != 16 || !value.bytes().all(|byte| crate::ids::BASE32.contains(&byte)) {
        bail!("error invalid-attachment-id input={value}");
    }
    Ok(())
}

pub fn validate_sha256(value: &str) -> Result<()> {
    if value.len() != 64
        || !value
            .bytes()
            .all(|byte| byte.is_ascii_digit() || matches!(byte, b'a'..=b'f'))
    {
        bail!("error invalid-sha256");
    }
    Ok(())
}

pub fn validate_media_type(value: &str) -> Result<()> {
    if !SUPPORTED_MEDIA_TYPES.contains(&value) {
        bail!("error unsupported-attachment-media-type input={value}");
    }
    Ok(())
}

pub fn validate_blob_size(byte_size: usize) -> Result<()> {
    if byte_size == 0 || byte_size > MAX_BLOB_BYTES {
        bail!("error invalid-attachment-size bytes={byte_size}");
    }
    Ok(())
}

pub fn validate_filename(value: Option<&str>) -> Result<()> {
    let Some(value) = value else {
        return Ok(());
    };
    if value.is_empty()
        || value.len() > MAX_FILENAME_LEN
        || value
            .chars()
            .any(|ch| ch.is_control() || ch == '/' || ch == '\\')
    {
        bail!("error invalid-attachment-filename");
    }
    Ok(())
}

pub fn validate_alt_text(value: Option<&str>) -> Result<()> {
    let Some(value) = value else {
        return Ok(());
    };
    if value.len() > MAX_ALT_TEXT_LEN || value.chars().any(char::is_control) {
        bail!("error invalid-attachment-alt-text");
    }
    Ok(())
}

pub fn validate_dimensions(width: Option<i64>, height: Option<i64>) -> Result<()> {
    let (Some(width), Some(height)) = (width, height) else {
        bail!("error invalid-attachment-dimensions");
    };
    let Ok(width_u32) = u32::try_from(width) else {
        bail!("error invalid-attachment-dimensions");
    };
    let Ok(height_u32) = u32::try_from(height) else {
        bail!("error invalid-attachment-dimensions");
    };
    if width_u32 == 0
        || height_u32 == 0
        || width_u32 > crate::attachments::decode::MAX_IMAGE_EDGE
        || height_u32 > crate::attachments::decode::MAX_IMAGE_EDGE
        || u64::from(width_u32) * u64::from(height_u32)
            > crate::attachments::decode::MAX_FRAME_PIXELS
    {
        bail!("error invalid-attachment-dimensions");
    }
    Ok(())
}

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

    #[test]
    fn validates_attachment_id() {
        assert!(validate_attachment_id("7KQ9A1X4MV2P8D6R").is_ok());
        assert!(validate_attachment_id("short").is_err());
        assert!(validate_attachment_id("7KQ9A1X4MV2P8D6U").is_err()); // 'U' is not base32
        assert!(validate_attachment_id("7KQ9A1X4MV2P8D6r").is_err()); // lowercase not allowed
    }

    #[test]
    fn validates_sha256() {
        let valid = "abcdef0123456789abcdef0123456789abcdef0123456789abcdef0123456789";
        assert!(validate_sha256(valid).is_ok());
        assert!(validate_sha256("short").is_err());
        assert!(validate_sha256(&format!("{}g", &valid[..63])).is_err());
        assert!(validate_sha256(&valid.to_ascii_uppercase()).is_err());
    }

    #[test]
    fn validates_media_type() {
        assert!(validate_media_type("image/png").is_ok());
        assert!(validate_media_type("image/jpeg").is_ok());
        assert!(validate_media_type("image/gif").is_ok());
        assert!(validate_media_type("image/webp").is_ok());
        assert!(validate_media_type("image/svg+xml").is_err());
        assert!(validate_media_type("").is_err());
    }

    #[test]
    fn validates_blob_size() {
        assert!(validate_blob_size(1).is_ok());
        assert!(validate_blob_size(25 * 1024 * 1024).is_ok());
        assert!(validate_blob_size(0).is_err());
        assert!(validate_blob_size(25 * 1024 * 1024 + 1).is_err());
    }

    #[test]
    fn validates_filename() {
        assert!(validate_filename(Some("photo.png")).is_ok());
        assert!(validate_filename(None).is_ok());
        assert!(validate_filename(Some("")).is_err());
        assert!(validate_filename(Some("a/photo.png")).is_err());
        assert!(validate_filename(Some("a\\photo.png")).is_err());
        for (name, accepted) in [
            ("a".repeat(255), true),
            ("a".repeat(256), false),
            ("é".repeat(127), true),
            ("é".repeat(128), false),
        ] {
            assert_eq!(validate_filename(Some(&name)).is_ok(), accepted);
        }
        assert!(validate_filename(Some("photo\n.png")).is_err());
    }

    #[test]
    fn validates_alt_text() {
        assert!(validate_alt_text(Some("a photo of a cat")).is_ok());
        assert!(validate_alt_text(None).is_ok());
        assert!(validate_alt_text(Some("")).is_ok());
        for (text, accepted) in [
            ("a".repeat(500), true),
            ("a".repeat(501), false),
            ("é".repeat(250), true),
            ("é".repeat(251), false),
        ] {
            assert_eq!(validate_alt_text(Some(&text)).is_ok(), accepted);
        }
        assert!(validate_alt_text(Some("a\nb")).is_err());
    }

    #[test]
    fn validates_dimensions() {
        assert!(validate_dimensions(None, None).is_err());
        for (width, height, accepted) in [
            (1920, 1080, true),
            (0, 1080, false),
            (-1, 1, false),
            (i64::MAX, 1, false),
            (16384, 1, true),
            (16385, 1, false),
            (1, 16385, false),
            (10000, 4000, true),
            (10000, 4001, false),
        ] {
            assert_eq!(
                validate_dimensions(Some(width), Some(height)).is_ok(),
                accepted,
                "{width}x{height}"
            );
        }
        assert!(validate_dimensions(Some(1920), None).is_err());
        assert!(validate_dimensions(None, Some(1080)).is_err());
    }
}