Article 1
-Some text with bold and emphasis
-
- Hello & Welcome
- -
- `, ``, `
Hello
`) based on the configuration
-fn add_attributes(
- config: &HtmlTransformerConfig,
- element: &mut BytesStart,
- is_root: bool,
- captured_attributes: &mut Vec<(String, Vec)>,
-) {
- let mut added_attrs = Vec::new();
-
- // Add root attributes if this is a root element
- if is_root {
- for attr in &config.root_attributes {
- element.push_attribute((attr.as_str(), ""));
- added_attrs.push(attr.clone());
- }
- }
-
- // Add attributes that should be applied to all elements
- for attr in &config.all_attributes {
- element.push_attribute((attr.as_str(), ""));
- added_attrs.push(attr.clone());
- }
-
- // If we're watching for a specific attribute, check if this element has it
- if let Some(watch_attr) = &config.watch_on_attribute {
- if let Some(attr_value) = element
- .attributes()
- .find(|a| {
- if let Ok(attr) = a {
- String::from_utf8_lossy(attr.key.as_ref()) == *watch_attr
- } else {
- false
- }
- })
- .and_then(|a| a.ok())
- .map(|a| String::from_utf8_lossy(a.value.as_ref()).into_owned())
- {
- captured_attributes.push((attr_value, added_attrs));
- }
- }
-}
-
-/// Main entrypoint. Transform HTML by adding attributes to the elements.
-pub fn transform(
config: &HtmlTransformerConfig,
- html: &str,
) -> Result<(String, Vec<(String, Vec)>), Box> {
- let mut reader = Reader::from_str(html);
- let reader_config = reader.config_mut();
- reader_config.check_end_names = config.check_end_names;
- // Allow bare & in HTML content (e.g. "Hello & Welcome" instead of requiring "Hello & Welcome")
- // This is needed for compatibility with HTML5 which is more lenient than strict XML
- reader_config.allow_dangling_amp = true;
-
- // We transform the HTML by reading it and writing it simultaneously
- let mut writer = Writer::new(Cursor::new(Vec::new()));
- let mut captured_attributes = Vec::new();
-
- // Track the nesting depth of elements to identify root elements (depth == 0)
- let mut depth: i32 = 0;
-
- // Read the HTML event by event
- loop {
- match reader.read_event() {
- // Start tag
- Ok(Event::Start(e)) => {
- let tag_name = String::from_utf8_lossy(e.name().as_ref())
- .to_string()
- .to_lowercase();
- let mut elem = e.into_owned();
- add_attributes(config, &mut elem, depth == 0, &mut captured_attributes);
-
- // For void elements, write as Empty event
- if config.void_elements.contains(&tag_name) {
- writer.write_event(Event::Empty(elem))?;
- } else {
- writer.write_event(Event::Start(elem))?;
- depth += 1;
- }
- }
-
- // End tag
- Ok(Event::End(e)) => {
- let tag_name = String::from_utf8_lossy(e.name().as_ref())
- .to_string()
- .to_lowercase();
-
- // Skip end tags for void elements
- if !config.void_elements.contains(&tag_name) {
- writer.write_event(Event::End(e))?;
- depth -= 1;
- }
- }
-
- // Empty element (AKA void or self-closing tag, e.g. `
`) - Ok(Event::Empty(e)) => { - let mut elem = e.into_owned(); - add_attributes(config, &mut elem, depth == 0, &mut captured_attributes); - writer.write_event(Event::Empty(elem))?; - } - - // End of file - Ok(Event::Eof) => break, - // Other events (e.g. comments, processing instructions, etc.) - Ok(e) => writer.write_event(e)?, - Err(e) => return Err(Box::new(e)), - } - } - - // Convert the transformed HTML to a string - let result = String::from_utf8(writer.into_inner().into_inner())?; - Ok((result, captured_attributes)) -} - -#[cfg(test)] -mod tests { - use super::*; - - #[test] - fn test_basic_transformation() { - let config = HtmlTransformerConfig::new( - vec!["data-root".to_string()], - vec!["data-all".to_string()], - false, - None, - ); - - let input = "";
- let (result, _) = transform(&config, input).unwrap();
-
- assert!(result.contains("data-root"));
- assert!(result.contains("data-all"));
- }
-
- #[test]
- fn test_multiple_roots() {
- let config = HtmlTransformerConfig::new(
- vec!["data-root".to_string()],
- vec!["data-all".to_string()],
- false,
- None,
- );
-
- let input = "Test Page
-
-
- "#;
-
- let (result, _) = transform(&config, input).unwrap();
-
- // Check that it parsed successfully
- assert!(result.contains(r#"Test Page"#));
- assert!(result.contains(r#""));
- assert!(!result.contains(""));
- assert!(result.contains("/>"));
- }
-
- #[test]
- fn test_config_check_end_names() {
- // Test with check_end_names = false (lenient mode)
- let config = HtmlTransformerConfig::new(
- vec!["data-root".to_string()],
- vec!["data-v-123".to_string()],
- false, // Don't check end names
- None,
- );
-
- // These should parse successfully with check_end_names = false
- let lenient_cases = [
- "", // Mismatched nesting
- "
`) - Ok(Event::Empty(e)) => { - let mut elem = e.into_owned(); - add_attributes(config, &mut elem, depth == 0, &mut captured_attributes); - writer.write_event(Event::Empty(elem))?; - } - - // End of file - Ok(Event::Eof) => break, - // Other events (e.g. comments, processing instructions, etc.) - Ok(e) => writer.write_event(e)?, - Err(e) => return Err(Box::new(e)), - } - } - - // Convert the transformed HTML to a string - let result = String::from_utf8(writer.into_inner().into_inner())?; - Ok((result, captured_attributes)) -} - -#[cfg(test)] -mod tests { - use super::*; - - #[test] - fn test_basic_transformation() { - let config = HtmlTransformerConfig::new( - vec!["data-root".to_string()], - vec!["data-all".to_string()], - false, - None, - ); - - let input = "
Hello
First
Second";
- let (result, _) = transform(&config, input).unwrap();
-
- // Both root elements should have data-root
- assert_eq!(result.matches("data-root").count(), 2);
- // All elements should have data-all
- assert_eq!(result.matches("data-all").count(), 2);
- }
-
- #[test]
- fn test_complex_html() {
- let config = HtmlTransformerConfig::new(
- vec!["data-root".to_string()],
- vec!["data-all".to_string(), "data-v-123".to_string()],
- false,
- None,
- );
-
- let input = r#"
-
-
-
-
-
-
-
-
-
-
- "#;
-
- let (result, _) = transform(&config, input).unwrap();
-
- // Check root elements have root attributes
- assert!(result.contains(
- r#"
- "#
- ));
- assert!(result.contains(r#"
"));
- }
-
- #[test]
- fn test_html_head_with_meta() {
- let config = HtmlTransformerConfig::new(
- vec!["data-root".to_string()],
- vec!["data-v-123".to_string()],
- false,
- None,
- );
-
- let input = r#"
-
-
- Hello
Text", // Wrong closing tag
- "";
- assert!(transform(&config, valid_input).is_ok());
- }
-
- #[test]
- fn test_watch_attribute() {
- let config = HtmlTransformerConfig::new(
- vec!["data-root".to_string()],
- vec!["data-v-123".to_string()],
- false,
- Some("data-id".to_string()),
- );
-
- let input = r#"
-
Text", // Non-matching end tag - ]; - - for input in lenient_cases { - assert!(transform(&config, input).is_ok()); - } - - // Test with check_end_names = true (strict mode) - let config = HtmlTransformerConfig::new( - vec!["data-root".to_string()], - vec!["data-v-123".to_string()], - true, // Check end names - None, - ); - - // These should fail with check_end_names = true - for input in lenient_cases { - assert!(transform(&config, input).is_err()); - } - - // But well-formed HTML should still work - let valid_input = "
Hello
-
-
"#;
-
- let (result, captured) = transform(&config, input).unwrap();
-
- println!("result: {}", result);
- println!("captured: {:?}", captured);
-
- // Verify HTML transformation worked
- assert!(result.contains(r#"Regular element
- Nested element -
- "#));
- assert!(result.contains(r#""#));
- assert!(result.contains(r#""#));
-
- // Verify attribute capturing
- assert_eq!(captured.len(), 3);
- assert!(captured.iter().any(|(id, attrs)| id == "123"
- && attrs.contains(&"data-root".to_string())
- && attrs.contains(&"data-v-123".to_string())));
- assert!(captured
- .iter()
- .any(|(id, attrs)| id == "456" && attrs.contains(&"data-v-123".to_string())));
- assert!(captured
- .iter()
- .any(|(id, attrs)| id == "789" && attrs.contains(&"data-v-123".to_string())));
- }
+ transform(config, html)
}
diff --git a/crates/djc-html-transformer/src/transformer.rs b/crates/djc-html-transformer/src/transformer.rs
new file mode 100644
index 0000000..52c4fc3
--- /dev/null
+++ b/crates/djc-html-transformer/src/transformer.rs
@@ -0,0 +1,406 @@
+use quick_xml::events::{BytesStart, Event};
+use quick_xml::reader::Reader;
+use quick_xml::writer::Writer;
+use std::collections::HashSet;
+use std::io::Cursor;
+
+// List of HTML5 void elements. These can be written as `` or `
`, ``, `![]()
`, etc.
+const VOID_ELEMENTS: [&str; 14] = [
+ "area", "base", "br", "col", "embed", "hr", "img", "input", "link", "meta", "param", "source",
+ "track", "wbr",
+];
+
+/// Configuration for HTML transformation
+pub struct HtmlTransformerConfig {
+ root_attributes: Vec,
+ all_attributes: Vec,
+ void_elements: HashSet,
+ check_end_names: bool,
+ watch_on_attribute: Option,
+}
+
+impl HtmlTransformerConfig {
+ pub fn new(
+ root_attributes: Vec,
+ all_attributes: Vec,
+ check_end_names: bool,
+ watch_on_attribute: Option,
+ ) -> Self {
+ let void_elements = VOID_ELEMENTS.iter().map(|&s| s.to_string()).collect();
+
+ HtmlTransformerConfig {
+ root_attributes,
+ all_attributes,
+ void_elements,
+ check_end_names,
+ watch_on_attribute,
+ }
+ }
+}
+
+/// Add attributes to a HTML start tag (e.g. `
"#));
-
- // Verify attribute capturing
- assert_eq!(captured.len(), 3);
- assert!(captured.iter().any(|(id, attrs)| id == "123"
- && attrs.contains(&"data-root".to_string())
- && attrs.contains(&"data-v-123".to_string())));
- assert!(captured
- .iter()
- .any(|(id, attrs)| id == "456" && attrs.contains(&"data-v-123".to_string())));
- assert!(captured
- .iter()
- .any(|(id, attrs)| id == "789" && attrs.contains(&"data-v-123".to_string())));
- }
+ transform(config, html)
}
diff --git a/crates/djc-html-transformer/src/transformer.rs b/crates/djc-html-transformer/src/transformer.rs
new file mode 100644
index 0000000..52c4fc3
--- /dev/null
+++ b/crates/djc-html-transformer/src/transformer.rs
@@ -0,0 +1,406 @@
+use quick_xml::events::{BytesStart, Event};
+use quick_xml::reader::Reader;
+use quick_xml::writer::Writer;
+use std::collections::HashSet;
+use std::io::Cursor;
+
+// List of HTML5 void elements. These can be written as ``, ``, `
`) based on the configuration
+fn add_attributes(
+ config: &HtmlTransformerConfig,
+ element: &mut BytesStart,
+ is_root: bool,
+ captured_attributes: &mut Vec<(String, Vec)>,
+) {
+ let mut added_attrs = Vec::new();
+
+ // Add root attributes if this is a root element
+ if is_root {
+ for attr in &config.root_attributes {
+ element.push_attribute((attr.as_str(), ""));
+ added_attrs.push(attr.clone());
+ }
+ }
+
+ // Add attributes that should be applied to all elements
+ for attr in &config.all_attributes {
+ element.push_attribute((attr.as_str(), ""));
+ added_attrs.push(attr.clone());
+ }
+
+ // If we're watching for a specific attribute, check if this element has it
+ if let Some(watch_attr) = &config.watch_on_attribute {
+ if let Some(attr_value) = element
+ .attributes()
+ .find(|a| {
+ if let Ok(attr) = a {
+ String::from_utf8_lossy(attr.key.as_ref()) == *watch_attr
+ } else {
+ false
+ }
+ })
+ .and_then(|a| a.ok())
+ .map(|a| String::from_utf8_lossy(a.value.as_ref()).into_owned())
+ {
+ captured_attributes.push((attr_value, added_attrs));
+ }
+ }
+}
+
+/// Main entrypoint. Transform HTML by adding attributes to the elements.
+pub fn transform(
+ config: &HtmlTransformerConfig,
+ html: &str,
+) -> Result<(String, Vec<(String, Vec)>), Box> {
+ let mut reader = Reader::from_str(html);
+ let reader_config = reader.config_mut();
+ reader_config.check_end_names = config.check_end_names;
+ // Allow bare & in HTML content (e.g. "Hello & Welcome" instead of requiring "Hello & Welcome")
+ // This is needed for compatibility with HTML5 which is more lenient than strict XML
+ reader_config.allow_dangling_amp = true;
+
+ // We transform the HTML by reading it and writing it simultaneously
+ let mut writer = Writer::new(Cursor::new(Vec::new()));
+ let mut captured_attributes = Vec::new();
+
+ // Track the nesting depth of elements to identify root elements (depth == 0)
+ let mut depth: i32 = 0;
+
+ // Read the HTML event by event
+ loop {
+ match reader.read_event() {
+ // Start tag
+ Ok(Event::Start(e)) => {
+ let tag_name = String::from_utf8_lossy(e.name().as_ref())
+ .to_string()
+ .to_lowercase();
+ let mut elem = e.into_owned();
+ add_attributes(config, &mut elem, depth == 0, &mut captured_attributes);
+
+ // For void elements, write as Empty event
+ if config.void_elements.contains(&tag_name) {
+ writer.write_event(Event::Empty(elem))?;
+ } else {
+ writer.write_event(Event::Start(elem))?;
+ depth += 1;
+ }
+ }
+
+ // End tag
+ Ok(Event::End(e)) => {
+ let tag_name = String::from_utf8_lossy(e.name().as_ref())
+ .to_string()
+ .to_lowercase();
+
+ // Skip end tags for void elements
+ if !config.void_elements.contains(&tag_name) {
+ writer.write_event(Event::End(e))?;
+ depth -= 1;
+ }
+ }
+
+ // Empty element (AKA void or self-closing tag, e.g. `
`) + Ok(Event::Empty(e)) => { + let mut elem = e.into_owned(); + add_attributes(config, &mut elem, depth == 0, &mut captured_attributes); + writer.write_event(Event::Empty(elem))?; + } + + // End of file + Ok(Event::Eof) => break, + // Other events (e.g. comments, processing instructions, etc.) + Ok(e) => writer.write_event(e)?, + Err(e) => return Err(Box::new(e)), + } + } + + // Convert the transformed HTML to a string + let result = String::from_utf8(writer.into_inner().into_inner())?; + Ok((result, captured_attributes)) +} + +#[cfg(test)] +mod tests { + use super::*; + + #[test] + fn test_basic_transformation() { + let config = HtmlTransformerConfig::new( + vec!["data-root".to_string()], + vec!["data-all".to_string()], + false, + None, + ); + + let input = "";
+ let (result, _) = transform(&config, input).unwrap();
+
+ assert!(result.contains("data-root"));
+ assert!(result.contains("data-all"));
+ }
+
+ #[test]
+ fn test_multiple_roots() {
+ let config = HtmlTransformerConfig::new(
+ vec!["data-root".to_string()],
+ vec!["data-all".to_string()],
+ false,
+ None,
+ );
+
+ let input = "Test Page
+
+
+ "#;
+
+ let (result, _) = transform(&config, input).unwrap();
+
+ // Check that it parsed successfully
+ assert!(result.contains(r#"Test Page"#));
+ assert!(result.contains(r#""));
+ assert!(!result.contains(""));
+ assert!(result.contains("/>"));
+ }
+
+ #[test]
+ fn test_config_check_end_names() {
+ // Test with check_end_names = false (lenient mode)
+ let config = HtmlTransformerConfig::new(
+ vec!["data-root".to_string()],
+ vec!["data-v-123".to_string()],
+ false, // Don't check end names
+ None,
+ );
+
+ // These should parse successfully with check_end_names = false
+ let lenient_cases = [
+ "", // Mismatched nesting
+ "
`) + Ok(Event::Empty(e)) => { + let mut elem = e.into_owned(); + add_attributes(config, &mut elem, depth == 0, &mut captured_attributes); + writer.write_event(Event::Empty(elem))?; + } + + // End of file + Ok(Event::Eof) => break, + // Other events (e.g. comments, processing instructions, etc.) + Ok(e) => writer.write_event(e)?, + Err(e) => return Err(Box::new(e)), + } + } + + // Convert the transformed HTML to a string + let result = String::from_utf8(writer.into_inner().into_inner())?; + Ok((result, captured_attributes)) +} + +#[cfg(test)] +mod tests { + use super::*; + + #[test] + fn test_basic_transformation() { + let config = HtmlTransformerConfig::new( + vec!["data-root".to_string()], + vec!["data-all".to_string()], + false, + None, + ); + + let input = "
Hello
First
Second";
+ let (result, _) = transform(&config, input).unwrap();
+
+ // Both root elements should have data-root
+ assert_eq!(result.matches("data-root").count(), 2);
+ // All elements should have data-all
+ assert_eq!(result.matches("data-all").count(), 2);
+ }
+
+ #[test]
+ fn test_complex_html() {
+ let config = HtmlTransformerConfig::new(
+ vec!["data-root".to_string()],
+ vec!["data-all".to_string(), "data-v-123".to_string()],
+ false,
+ None,
+ );
+
+ let input = r#"
+
+
+
+
+
+
+
+
+
+
+ "#;
+
+ let (result, _) = transform(&config, input).unwrap();
+
+ // Check root elements have root attributes
+ assert!(result.contains(
+ r#"Hello & Welcome
+ +Article 1
+Some text with bold and emphasis
+
+ "#
+ ));
+ assert!(result.contains(r#"
"));
+ }
+
+ #[test]
+ fn test_html_head_with_meta() {
+ let config = HtmlTransformerConfig::new(
+ vec!["data-root".to_string()],
+ vec!["data-v-123".to_string()],
+ false,
+ None,
+ );
+
+ let input = r#"
+
+
+ Hello
Text", // Wrong closing tag
+ "";
+ assert!(transform(&config, valid_input).is_ok());
+ }
+
+ #[test]
+ fn test_watch_attribute() {
+ let config = HtmlTransformerConfig::new(
+ vec!["data-root".to_string()],
+ vec!["data-v-123".to_string()],
+ false,
+ Some("data-id".to_string()),
+ );
+
+ let input = r#"
+
Text", // Non-matching end tag + ]; + + for input in lenient_cases { + assert!(transform(&config, input).is_ok()); + } + + // Test with check_end_names = true (strict mode) + let config = HtmlTransformerConfig::new( + vec!["data-root".to_string()], + vec!["data-v-123".to_string()], + true, // Check end names + None, + ); + + // These should fail with check_end_names = true + for input in lenient_cases { + assert!(transform(&config, input).is_err()); + } + + // But well-formed HTML should still work + let valid_input = "
Hello
+
+
"#;
+
+ let (result, captured) = transform(&config, input).unwrap();
+
+ println!("result: {}", result);
+ println!("captured: {:?}", captured);
+
+ // Verify HTML transformation worked
+ assert!(result.contains(r#"Regular element
+ Nested element +
+ "#));
+ assert!(result.contains(r#""#));
+ assert!(result.contains(r#""#));
+
+ // Verify attribute capturing
+ assert_eq!(captured.len(), 3);
+ assert!(captured.iter().any(|(id, attrs)| id == "123"
+ && attrs.contains(&"data-root".to_string())
+ && attrs.contains(&"data-v-123".to_string())));
+ assert!(captured
+ .iter()
+ .any(|(id, attrs)| id == "456" && attrs.contains(&"data-v-123".to_string())));
+ assert!(captured
+ .iter()
+ .any(|(id, attrs)| id == "789" && attrs.contains(&"data-v-123".to_string())));
+ }
+}
diff --git a/djc_core/__init__.py b/djc_core/__init__.py
index 68add3b..2dfb9ee 100644
--- a/djc_core/__init__.py
+++ b/djc_core/__init__.py
@@ -1,6 +1,6 @@
-# DO NOT MODIFY, ONLY UPDATE THE MODULE NAME WHEN NEEDED!
# This file is what maturin auto-generates. But it seems maturin omits it when we have a __init__.pyi file.
# So we have to manually include it here.
+# Following block of code is what maturin would've generated
from .djc_core import *
diff --git a/djc_core/__init__.pyi b/djc_core/__init__.pyi
index eeeff7c..1774962 100644
--- a/djc_core/__init__.pyi
+++ b/djc_core/__init__.pyi
@@ -1,34 +1 @@
-from typing import List, Dict, Optional
-
-def set_html_attributes(
- html: str,
- root_attributes: List[str],
- all_attributes: List[str],
- check_end_names: Optional[bool] = None,
- watch_on_attribute: Optional[str] = None,
-) -> tuple[str, Dict[str, List[str]]]:
- """
- Transform HTML by adding attributes to root and all elements.
-
- Args:
- html (str): The HTML string to transform. Can be a fragment or full document.
- root_attributes (List[str]): List of attribute names to add to root elements only.
- all_attributes (List[str]): List of attribute names to add to all elements.
- check_end_names (Optional[bool]): Whether to validate matching of end tags. Defaults to None.
- watch_on_attribute (Optional[str]): If set, captures which attributes were added to elements with this attribute.
-
- Returns:
- A tuple containing:
- - The transformed HTML string
- - A dictionary mapping captured attribute values to lists of attributes that were added
- to those elements. Only returned if watch_on_attribute is set, otherwise empty dict.
-
- Example:
- >>> html = ''
- >>> set_html_attributes(html, ['data-root-id'], ['data-v-123'])
- ''
-
- Raises:
- ValueError: If the HTML is malformed or cannot be parsed.
- """
- ...
+from djc_core.djc_html_transformer import *
diff --git a/djc_core/djc_html_transformer.pyi b/djc_core/djc_html_transformer.pyi
new file mode 100644
index 0000000..a8e1f47
--- /dev/null
+++ b/djc_core/djc_html_transformer.pyi
@@ -0,0 +1,36 @@
+from typing import List, Dict, Optional
+
+def set_html_attributes(
+ html: str,
+ root_attributes: List[str],
+ all_attributes: List[str],
+ check_end_names: Optional[bool] = None,
+ watch_on_attribute: Optional[str] = None,
+) -> tuple[str, Dict[str, List[str]]]:
+ """
+ Transform HTML by adding attributes to root and all elements.
+
+ Args:
+ html (str): The HTML string to transform. Can be a fragment or full document.
+ root_attributes (List[str]): List of attribute names to add to root elements only.
+ all_attributes (List[str]): List of attribute names to add to all elements.
+ check_end_names (Optional[bool]): Whether to validate matching of end tags. Defaults to None.
+ watch_on_attribute (Optional[str]): If set, captures which attributes were added to elements with this attribute.
+
+ Returns:
+ A tuple containing:
+ - The transformed HTML string
+ - A dictionary mapping captured attribute values to lists of attributes that were added
+ to those elements. Only returned if watch_on_attribute is set, otherwise empty dict.
+
+ Example:
+ >>> html = ''
+ >>> set_html_attributes(html, ['data-root-id'], ['data-v-123'])
+ ''
+
+ Raises:
+ ValueError: If the HTML is malformed or cannot be parsed.
+ """
+ ...
+
+__all__ = ["set_html_attributes"]
"#));
+
+ // Verify attribute capturing
+ assert_eq!(captured.len(), 3);
+ assert!(captured.iter().any(|(id, attrs)| id == "123"
+ && attrs.contains(&"data-root".to_string())
+ && attrs.contains(&"data-v-123".to_string())));
+ assert!(captured
+ .iter()
+ .any(|(id, attrs)| id == "456" && attrs.contains(&"data-v-123".to_string())));
+ assert!(captured
+ .iter()
+ .any(|(id, attrs)| id == "789" && attrs.contains(&"data-v-123".to_string())));
+ }
+}
diff --git a/djc_core/__init__.py b/djc_core/__init__.py
index 68add3b..2dfb9ee 100644
--- a/djc_core/__init__.py
+++ b/djc_core/__init__.py
@@ -1,6 +1,6 @@
-# DO NOT MODIFY, ONLY UPDATE THE MODULE NAME WHEN NEEDED!
# This file is what maturin auto-generates. But it seems maturin omits it when we have a __init__.pyi file.
# So we have to manually include it here.
+# Following block of code is what maturin would've generated
from .djc_core import *
diff --git a/djc_core/__init__.pyi b/djc_core/__init__.pyi
index eeeff7c..1774962 100644
--- a/djc_core/__init__.pyi
+++ b/djc_core/__init__.pyi
@@ -1,34 +1 @@
-from typing import List, Dict, Optional
-
-def set_html_attributes(
- html: str,
- root_attributes: List[str],
- all_attributes: List[str],
- check_end_names: Optional[bool] = None,
- watch_on_attribute: Optional[str] = None,
-) -> tuple[str, Dict[str, List[str]]]:
- """
- Transform HTML by adding attributes to root and all elements.
-
- Args:
- html (str): The HTML string to transform. Can be a fragment or full document.
- root_attributes (List[str]): List of attribute names to add to root elements only.
- all_attributes (List[str]): List of attribute names to add to all elements.
- check_end_names (Optional[bool]): Whether to validate matching of end tags. Defaults to None.
- watch_on_attribute (Optional[str]): If set, captures which attributes were added to elements with this attribute.
-
- Returns:
- A tuple containing:
- - The transformed HTML string
- - A dictionary mapping captured attribute values to lists of attributes that were added
- to those elements. Only returned if watch_on_attribute is set, otherwise empty dict.
-
- Example:
- >>> html = 'Hello
Hello
Hello
Hello