25 lines
850 B
Rust
25 lines
850 B
Rust
use minify_html_onepass::{Cfg, Error, in_place as minify_html_native};
|
|
use pyo3::prelude::*;
|
|
use pyo3::exceptions::PySyntaxError;
|
|
use pyo3::wrap_pyfunction;
|
|
use std::str::from_utf8_unchecked;
|
|
|
|
#[pyfunction(py_args="*", minify_js="false", minify_css="false")]
|
|
fn minify(code: String, minify_js: bool, minify_css: bool) -> PyResult<String> {
|
|
let mut code = code.into_bytes();
|
|
match minify_html_native(&mut code, &Cfg {
|
|
minify_js,
|
|
minify_css,
|
|
}) {
|
|
Ok(out_len) => Ok(unsafe { from_utf8_unchecked(&code[0..out_len]).to_string() }),
|
|
Err(Error { error_type, position }) => Err(PySyntaxError::new_err(format!("{} [Character {}]", error_type.message(), position))),
|
|
}
|
|
}
|
|
|
|
#[pymodule]
|
|
fn minify_html_onepass(_py: Python, m: &PyModule) -> PyResult<()> {
|
|
m.add_wrapped(wrap_pyfunction!(minify))?;
|
|
|
|
Ok(())
|
|
}
|