minify-html/python/src/lib.template.rs

use minify_html::{Cfg, minify as minify_html_native};
use pyo3::prelude::*;
use pyo3::wrap_pyfunction;
use std::string::String;

#[pyfunction(
    py_args = "*",
    do_not_minify_doctype = "false",
    ensure_spec_compliant_unquoted_attribute_values = "false",
    keep_closing_tags = "false",
    keep_comments = "false",
    keep_html_and_head_opening_tags = "false",
    keep_spaces_between_attributes = "false",
    minify_css = "false",
    minify_js = "false",
    remove_bangs = "false",
    remove_processing_instructions = "false",
)]
fn minify(
    code: String,
    do_not_minify_doctype: bool,
    ensure_spec_compliant_unquoted_attribute_values: bool,
    keep_closing_tags: bool,
    keep_comments: bool,
    keep_html_and_head_opening_tags: bool,
    keep_spaces_between_attributes: bool,
    minify_css: bool,
    minify_js: bool,
    remove_bangs: bool,
    remove_processing_instructions: bool,
) -> PyResult<String> {
    let code = code.into_bytes();
    let out_code = minify_html_native(&code, &Cfg {
        do_not_minify_doctype,
        ensure_spec_compliant_unquoted_attribute_values,
        keep_closing_tags,
        keep_comments,
        keep_html_and_head_opening_tags,
        keep_spaces_between_attributes,
        minify_css,
        minify_js,
        remove_bangs,
        remove_processing_instructions,
    });
    Ok(String::from_utf8(out_code).unwrap())
}

#[pymodule]
fn REPLACE_WITH_MODULE_NAME(_py: Python, m: &PyModule) -> PyResult<()> {
    m.add_wrapped(wrap_pyfunction!(minify))?;

    Ok(())
}