oxvg_optimiser 0.0.5

use itertools::Itertools;
use oxvg_ast::{
    element::Element,
    visitor::{Context, Visitor},
};
use oxvg_collections::content_type::{ContentType, ContentTypeRef};
#[cfg(feature = "serde")]
use serde::{Deserialize, Serialize};

#[cfg(feature = "wasm")]
use tsify::Tsify;

use crate::error::JobsError;

#[cfg_attr(feature = "wasm", derive(Tsify))]
#[cfg_attr(feature = "napi", napi(object))]
#[cfg_attr(feature = "serde", derive(Deserialize, Serialize))]
#[derive(Debug, Clone)]
#[cfg_attr(feature = "serde", serde(rename_all = "camelCase"))]
/// Removes redundant whitespace from attribute values.
///
/// # Correctness
///
/// By default any whitespace is cleaned up. This shouldn't affect anything within the SVG
/// but may affect elements within `<foreignObject />`, which is treated as HTML.
///
/// For example, whitespace has an effect when between `inline` and `inline-block` elements.
/// See [MDN](https://developer.mozilla.org/en-US/docs/Web/API/Document_Object_Model/Whitespace#spaces_in_between_inline_and_inline-block_elements) for more information.
///
/// In any other case, it should never affect the appearance of the document.
///
/// # Errors
///
/// Never.
///
/// If this job produces an error or panic, please raise an [issue](https://github.com/noahbald/oxvg/issues)
pub struct CleanupAttrs {
    #[cfg_attr(feature = "serde", serde(default = "newlines_default"))]
    /// Whether to replace `'\n'` with `' '`.
    pub newlines: bool,
    #[cfg_attr(feature = "serde", serde(default = "trim_default"))]
    /// Whether to remove whitespace from each end of the value
    pub trim: bool,
    #[cfg_attr(feature = "serde", serde(default = "spaces_default"))]
    /// Whether to replace multiple whitespace characters with a single `' '`.
    pub spaces: bool,
}

impl<'input, 'arena> Visitor<'input, 'arena> for CleanupAttrs {
    type Error = JobsError<'input>;

    fn element(
        &self,
        element: &Element<'input, 'arena>,
        _context: &mut Context<'input, 'arena, '_>,
    ) -> Result<(), Self::Error> {
        for mut attr in element.attributes().into_iter_mut() {
            let ContentType::Anything(ContentTypeRef::RefMut(value)) = attr.value_mut() else {
                continue;
            };
            if self.newlines {
                *value = value.replace('\n', " ").into();
            }
            if self.trim {
                *value = value.trim().to_string().into();
            }
            if self.spaces {
                *value = value.split_whitespace().join(" ").into();
            }
        }
        Ok(())
    }
}

impl Default for CleanupAttrs {
    fn default() -> Self {
        Self {
            newlines: newlines_default(),
            trim: trim_default(),
            spaces: spaces_default(),
        }
    }
}

const fn newlines_default() -> bool {
    true
}

const fn trim_default() -> bool {
    true
}

const fn spaces_default() -> bool {
    true
}

#[test]
fn cleanup_attrs() -> anyhow::Result<()> {
    use crate::test_config;

    insta::assert_snapshot!(test_config(
        r#"{ "cleanupAttrs": {
            "newlines": true,
            "trim": true,
            "spaces": true
        } }"#,
        Some(
            r#"<svg xmlns="  http://www.w3.org/2000/svg
  " attr="a      b" attr2="a
b">
    <!-- Should remove all unnecessary whitespace from attributes -->
    test
</svg>"#
        )
    )?);

    insta::assert_snapshot!(test_config(
        r#"{ "cleanupAttrs": {
            "newlines": true,
            "trim": true,
            "spaces": true
        } }"#,
        Some(
            r#"<svg xmlns="  http://www.w3.org/2000/svg
  " attr="a      b">
    <!-- Should remove all unnecessary whitespace from attributes -->
    test &amp; &lt;&amp; &gt; &apos; &quot; &amp;
</svg>"#
        )
    )?);

    Ok(())
}