♻️ refactor: rustify writer utility ass_escape (#314)

Author: SigureMo

packages/biliass/rust/src/lib.rs

@@ -26,5 +26,6 @@ fn biliass_pyo3(m: &Bound<'_, PyModule>) -> PyResult<()> {
     m.add_function(wrap_pyfunction!(python::py_read_comments_from_xml, m)?)?;
     m.add_function(wrap_pyfunction!(python::py_read_comments_from_protobuf, m)?)?;
     m.add_function(wrap_pyfunction!(python::py_convert_timestamp, m)?)?;
+    m.add_function(wrap_pyfunction!(python::py_ass_escape, m)?)?;
     Ok(())
 }

packages/biliass/rust/src/python/mod.rs

@@ -6,4 +6,4 @@ mod writer;
 pub use comment::{PyComment, PyCommentPosition};
 pub use proto::{PyDanmakuElem, PyDmSegMobileReply};
 pub use reader::{py_read_comments_from_protobuf, py_read_comments_from_xml};
-pub use writer::py_convert_timestamp;
+pub use writer::{py_ass_escape, py_convert_timestamp};

packages/biliass/rust/src/python/writer.rs

@@ -6,3 +6,8 @@ use pyo3::prelude::*;
 pub fn py_convert_timestamp(timestamp: f64) -> PyResult<String> {
     Ok(writer::utils::convert_timestamp(timestamp))
 }
+
+#[pyfunction(name = "ass_escape")]
+pub fn py_ass_escape(text: &str) -> PyResult<String> {
+    Ok(writer::utils::ass_escape(text))
+}

packages/biliass/rust/src/writer/utils.rs

@@ -14,3 +14,28 @@ pub fn convert_timestamp(timestamp: f64) -> String {
 
     format!("{}:{:02}:{:02}.{:02}", hour, minute, second, centsecond)
 }
+
+pub fn ass_escape(text: &str) -> String {
+    text.replace("\\", "\\\\")
+        .replace("{", "\\{")
+        .replace("}", "\\}")
+        .split('\n')
+        .map(|line| {
+            let stripped = line.trim_matches(' ');
+            let size = line.len();
+            if stripped.len() == size {
+                line.to_owned()
+            } else {
+                let leading_spaces = line.len() - line.trim_start_matches(' ').len();
+                let trailing_spaces = line.len() - line.trim_end_matches(' ').len();
+                format!(
+                    "{}{}{}",
+                    "\u{2007}".repeat(leading_spaces),
+                    stripped,
+                    "\u{2007}".repeat(trailing_spaces)
+                )
+            }
+        })
+        .collect::<Vec<_>>()
+        .join("\\N")
+}

packages/biliass/src/biliass/_core.pyi

@@ -57,4 +57,5 @@ class Comment:
 
 def read_comments_from_xml(text: str, fontsize: float) -> list[Comment]: ...
 def read_comments_from_protobuf(data: bytes, fontsize: float) -> list[Comment]: ...
-def convert_timestamp(timestamp: int) -> float: ...
+def convert_timestamp(timestamp: float) -> float: ...
+def ass_escape(text: str) -> str: ...

packages/biliass/src/biliass/biliass.py

@@ -12,6 +12,7 @@
 from biliass._core import (
     Comment,
     CommentPosition,
+    ass_escape,
     convert_timestamp,
     read_comments_from_protobuf,
     read_comments_from_xml,
@@ -466,7 +467,7 @@ def mark_comment_row(rows: list[Comment | None], comment: Comment, row: int):
         pass
 
 
-def ass_escape(s):
+def ass_escape_py(s):
     def replace_leading_space(s):
         sstrip = s.strip(" ")
         slen = len(s)