Insert script at the end if HTML has no body tag

eush77 · eush77 · commit b2ebe126c085 · 2015-08-02T23:00:20.000+03:00
diff --git a/lib/script-injector.js b/lib/script-injector.js
@@ -34,14 +34,30 @@ function scriptInjector (script) {
       function () {
         if (needToAddScript) {
           this.push(script);
+          needToAddScript = false;
         }
         this.push(null);
       }))
     .pipe(bodyTag);
 
-  tr1.pipe(tr2);
-
-  return duplexer(tr1, tr2);
+  // If there were no <body>'s even, append the script at the end of the file
+  var tr3 = through(
+    function (data, enc, cb) {
+      this.push(data);
+      cb();
+    },
+    function () {
+      if (needToAddScript) {
+        this.push(script);
+        needToAddScript = false;
+      }
+      this.push(null);
+    }
+  );
+
+  tr1.pipe(tr2).pipe(tr3);
+
+  return duplexer(tr1, tr3);
 }
 
 module.exports = scriptInjector;
diff --git a/test/dat/test4-output.html b/test/dat/test4-output.html
@@ -0,0 +1,7 @@
+<h1>Hello World</h1>
+<p>This file is a result of X-to-HTML conversion, or not.</p>
+<script type="text/javascript">
+;(function someCode() {
+  console.log("I'm some code to be inserted inline");
+})()
+</script>
diff --git a/test/dat/test4.html b/test/dat/test4.html
@@ -0,0 +1,2 @@
+<h1>Hello World</h1>
+<p>This file is a result of X-to-HTML conversion, or not.</p>
diff --git a/test/test.js b/test/test.js
@@ -19,6 +19,7 @@ var testRunner = function (filename, t) {
 test('inject code before the closing body tag', testRunner.bind(null, 'test1'));
 test('inject code before the first script tag in the header', testRunner.bind(null, 'test2'));
 test('inject code before the first script tag in the body', testRunner.bind(null, 'test3'));
+test('inject code when there is no body tag', testRunner.bind(null, 'test4'));
 
 function someCode () {
   console.log("I'm some code to be inserted inline");

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,2 @@`
	`1`	`+<h1>Hello World</h1>`
	`2`	`+<p>This file is a result of X-to-HTML conversion, or not.</p>`