1 files changed, 49 insertions, 14 deletions

diff --git a/src/emitter.rs b/src/emitter.rs
index 341d335..9fdf967 100644
--- a/src/emitter.rs
+++ b/src/emitter.rs
@@ -158,6 +158,11 @@ pub trait Emitter<O> {
     /// If the current token is not a doctype, this method may panic.
     fn init_doctype_public_id(&mut self, offset: O);
 
+    /// Called after the last [`push_doctype_public_id`] call for a DOCTYPE public identifier.
+    ///
+    /// [`push_doctype_public_id`]: Self::push_doctype_public_id
+    fn terminate_doctype_public_id(&mut self, offset: O) {}
+
     /// Assuming the _current token_ is a doctype, set its "system identifier" to the empty string.
     ///
     /// If the current token is not a doctype, this method may panic.
@@ -172,6 +177,11 @@ pub trait Emitter<O> {
     ///
     /// If the current token is not a doctype, this method may panic.
     fn push_doctype_system_id(&mut self, s: &str);
+
+    /// Called after the last [`push_doctype_system_id`] call for a DOCTYPE system identifier.
+    ///
+    /// [`push_doctype_system_id`]: Self::push_doctype_system_id
+    fn terminate_doctype_system_id(&mut self, offset: O) {}
 }
 
 /// The default implementation of [`Emitter`], used to produce tokens.
@@ -410,8 +420,8 @@ impl<O: Offset> Emitter<O> for DefaultEmitter<O> {
             public_id: None,
             system_id: None,
             span: offset..O::default(),
-            public_id_offset: O::default(),
-            system_id_offset: O::default(),
+            public_id_span: O::default()..O::default(),
+            system_id_span: O::default()..O::default(),
         }));
     }
 
@@ -459,7 +469,7 @@ impl<O: Offset> Emitter<O> for DefaultEmitter<O> {
             return;
         };
         doctype.public_id = Some("".to_owned());
-        doctype.public_id_offset = offset;
+        doctype.public_id_span.start = offset;
     }
     fn init_doctype_system_id(&mut self, offset: O) {
         let Some(Token::Doctype(doctype)) = &mut self.current_token else {
@@ -467,7 +477,7 @@ impl<O: Offset> Emitter<O> for DefaultEmitter<O> {
             return;
         };
         doctype.system_id = Some("".to_owned());
-        doctype.system_id_offset = offset;
+        doctype.system_id_span.start = offset;
     }
     fn push_doctype_public_id(&mut self, s: &str) {
         if let Some(Token::Doctype(Doctype {
@@ -480,6 +490,19 @@ impl<O: Offset> Emitter<O> for DefaultEmitter<O> {
             debug_assert!(false);
         }
     }
+
+    fn terminate_doctype_public_id(&mut self, offset: O) {
+        if let Some(Token::Doctype(Doctype {
+            ref mut public_id_span,
+            ..
+        })) = self.current_token
+        {
+            public_id_span.end = offset;
+        } else {
+            debug_assert!(false);
+        }
+    }
+
     fn push_doctype_system_id(&mut self, s: &str) {
         if let Some(Token::Doctype(Doctype {
             system_id: Some(ref mut id),
@@ -491,6 +514,18 @@ impl<O: Offset> Emitter<O> for DefaultEmitter<O> {
             debug_assert!(false);
         }
     }
+
+    fn terminate_doctype_system_id(&mut self, offset: O) {
+        if let Some(Token::Doctype(Doctype {
+            ref mut system_id_span,
+            ..
+        })) = self.current_token
+        {
+            system_id_span.end = offset;
+        } else {
+            debug_assert!(false);
+        }
+    }
 }
 
 /// An HTML start tag, such as `<p>` or `<a>`.
@@ -573,24 +608,24 @@ pub struct Doctype<O> {
     /// The source code span of the doctype.
     pub span: Range<O>,
 
-    /// The source offset of the public identifier.
-    public_id_offset: O,
+    /// The span of the public identifier.
+    public_id_span: Range<O>,
 
-    /// The source offset of the system identifier.
-    system_id_offset: O,
+    /// The span of the system identifier.
+    system_id_span: Range<O>,
 }
 
 impl<O: Offset> Doctype<O> {
-    /// Calculates the span of the public identifier and returns it.
+    /// Returns the span of the public identifier.
     pub fn public_id_span(&self) -> Option<Range<O>> {
-        let public_id = self.public_id.as_ref()?;
-        Some(self.public_id_offset..self.public_id_offset + public_id.len())
+        self.public_id.as_ref()?;
+        Some(self.public_id_span.clone())
     }
 
-    /// Calculates the span of the system identifier and returns it.
+    /// Returns the span of the system identifier.
     pub fn system_id_span(&self) -> Option<Range<O>> {
-        let system_id = self.system_id.as_ref()?;
-        Some(self.system_id_offset..self.system_id_offset + system_id.len())
+        self.system_id.as_ref()?;
+        Some(self.system_id_span.clone())
     }
 }