Fix elements that switch to non-data states

Closes GH-6.
Closes GH-7.
Closes GH-8.

Author: arobase-che

index.js

@@ -12,6 +12,7 @@ var xtend = require('xtend');
 module.exports = wrap;
 
 var IN_TEMPLATE_MODE = 'IN_TEMPLATE_MODE';
+var DATA_MODE = 'DATA_STATE';
 var CHARACTER_TOKEN = 'CHARACTER_TOKEN';
 var START_TAG_TOKEN = 'START_TAG_TOKEN';
 var END_TAG_TOKEN = 'END_TAG_TOKEN';
@@ -130,6 +131,12 @@ function wrap(tree, file) {
 
     if (!empty) {
       parser._processToken(endTag(node));
+
+      // Put the parser back in data mode: some elements, like
+      // textareas and iframes, change the state.
+      // See syntax-tree/hast-util-raw/issues/7.
+      // See https://github.com/inikulin/parse5/blob/2528196/packages/parse5/lib/tokenizer/index.js#L222
+      tokenizer.state = DATA_MODE;
     }
   }
 

test.js

@@ -102,6 +102,56 @@ test('raw', function (t) {
     'should pass raw nodes through (#2)'
   );
 
+  t.deepEqual(
+    raw(u('root', [
+      h('iframe', {height: 500, src: 'https://ddg.gg'}),
+      u('raw', '<img alt="foo" src="bar.jpg">')
+    ])),
+    u('root', {data: {quirksMode: false}}, [
+      h('iframe', {height: 500, src: 'https://ddg.gg'}),
+      h('img', {alt: 'foo', src: 'bar.jpg'})
+    ]),
+    'should pass raw nodes through even after iframe'
+  );
+
+  t.deepEqual(
+    raw(u('root', [
+      h('textarea', u('text', 'Some text that is <i>not</i> HTML.')),
+      u('raw', '<img alt="foo" src="bar.jpg">')
+    ])),
+    u('root', {data: {quirksMode: false}}, [
+      h('textarea', u('text', 'Some text that is <i>not</i> HTML.')),
+      h('img', {alt: 'foo', src: 'bar.jpg'})
+    ]),
+    'should pass raw nodes through even after textarea (#1)'
+  );
+
+  t.deepEqual(
+    raw(u('root', [
+      u('raw', '<textarea>Some text that is <i>not</i> HTML.</textarea>'),
+      u('raw', '<img alt="foo" src="bar.jpg">')
+    ])),
+    u('root', {data: {quirksMode: false}}, [
+      h('textarea', u('text', 'Some text that is <i>not</i> HTML.')),
+      h('img', {alt: 'foo', src: 'bar.jpg'})
+    ]),
+    'should pass raw nodes through even after textarea (#2)'
+  );
+
+  t.deepEqual(
+    raw(u('root', [
+      u('raw', '<textarea>'),
+      u('text', 'Some text that is <i>not</i> HTML.'),
+      u('raw', '</textarea>'),
+      u('raw', '<p>but this is</p>')
+    ])),
+    u('root', {data: {quirksMode: false}}, [
+      h('textarea', u('text', 'Some text that is <i>not</i> HTML.')),
+      h('p', u('text', 'but this is'))
+    ]),
+    'should pass raw nodes through even after textarea (#3)'
+  );
+
   t.end();
 });