From 7e400c29bf14bd75154c74e2dd9ba0814f83bec7 Mon Sep 17 00:00:00 2001
From: Martin Fischer <martin@push-f.com>
Date: Tue, 30 Nov 2021 16:16:47 +0100
Subject: introduce StartTag::next_state

Closes #11.
---
 src/emitter.rs | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

(limited to 'src')

diff --git a/src/emitter.rs b/src/emitter.rs
index 8c8976d..0a80544 100644
--- a/src/emitter.rs
+++ b/src/emitter.rs
@@ -4,6 +4,7 @@ use std::collections::VecDeque;
 use std::mem;
 
 use crate::Error;
+use crate::State;
 
 /// An emitter is an object providing methods to the tokenizer to produce tokens.
 ///
@@ -431,6 +432,23 @@ pub struct StartTag {
     pub attributes: BTreeMap<String, String>,
 }
 
+impl StartTag {
+    /// Returns the next tokenizer state according to
+    /// [Parsing HTML fragments](https://html.spec.whatwg.org/multipage/parsing.html#concept-frag-parse-context).
+    /// If `scripting` is set to true [`State::RawText`] is returned if this is a `<noscript>` tag,
+    /// otherwise [`State::Data`] is returned (as with any other regular tag).
+    pub fn next_state(&self, scripting: bool) -> State {
+        match self.name.as_str() {
+            "title" | "textarea" => State::RcData,
+            "style" | "xmp" | "iframe" | "noembed" | "noframes" => State::RawText,
+            "script" => State::ScriptData,
+            "noscript" if scripting => State::RawText,
+            "plaintext" => State::PlainText,
+            _other => State::Data,
+        }
+    }
+}
+
 /// A HTML end/close tag, such as `</p>` or `</a>`.
 #[derive(Debug, Default, Eq, PartialEq)]
 pub struct EndTag {
-- 
cgit v1.2.3