From 5403261e8213f68633a09fc3e9bc2e6e2cd777b2 Mon Sep 17 00:00:00 2001
From: Titus Wormer <tituswormer@gmail.com>
Date: Mon, 18 Jul 2022 16:31:14 +0200
Subject: Add support for turning off constructs

---
 src/construct/html_flow.rs | 33 +++++++++++++++++++++------------
 1 file changed, 21 insertions(+), 12 deletions(-)

(limited to 'src/construct/html_flow.rs')

diff --git a/src/construct/html_flow.rs b/src/construct/html_flow.rs
index 822b9dd..1255081 100644
--- a/src/construct/html_flow.rs
+++ b/src/construct/html_flow.rs
@@ -204,18 +204,27 @@ struct Info {
 ///     ^
 /// ```
 pub fn start(tokenizer: &mut Tokenizer, code: Code) -> StateFnResult {
-    tokenizer.enter(Token::HtmlFlow);
-    // To do: allow arbitrary when code (indented) is turned off.
-    tokenizer.go(
-        space_or_tab_with_options(SpaceOrTabOptions {
-            kind: Token::HtmlFlowData,
-            min: 0,
-            max: TAB_SIZE - 1,
-            connect: false,
-            content_type: None,
-        }),
-        before,
-    )(tokenizer, code)
+    let max = if tokenizer.parse_state.constructs.code_indented {
+        TAB_SIZE - 1
+    } else {
+        usize::MAX
+    };
+
+    if tokenizer.parse_state.constructs.html_flow {
+        tokenizer.enter(Token::HtmlFlow);
+        tokenizer.go(
+            space_or_tab_with_options(SpaceOrTabOptions {
+                kind: Token::HtmlFlowData,
+                min: 0,
+                max,
+                connect: false,
+                content_type: None,
+            }),
+            before,
+        )(tokenizer, code)
+    } else {
+        (State::Nok, None)
+    }
 }
 
 /// After optional whitespace, before `<`.
-- 
cgit