Provide a method to install files and inject HTML at the end of <head…

…> section
azachar · Feb 10, 2020 · b7fc2f0 · b7fc2f0
1 parent b932d81
commit b7fc2f0
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 1 deletion.
diff --git a/index.js b/index.js
@@ -0,0 +1,30 @@
+'use strict';
+
+const fs = require('fs');
+const fse = require('fs-extra');
+const path = require('path');
+
+/**
+ * Copy files for screenshoter-report-analyzer to the specified directory.
+ *
+ * @param {String} dest The destination directory
+ * @param {Object} options Optional parameters.
+ *    injectToHead: HTML string to inject into the end of the <head> section.
+ * @param {Promise} a promise
+ */
+function setup(dest, options) {
+  const src = path.join(__dirname, 'dist');
+  return fse.copy(src, dest).then(() => {
+    if (!options || !options.injectToHead) {
+        return;
+    }
+    const indexFile = path.join(dest, 'index.html');
+    const endOfHead = /(<\/head\s*>)/i;
+    let html = fs.readFileSync(indexFile, {encoding: 'utf-8'});
+    html = html.replace(endOfHead, match => { return options.injectToHead + match; });
+    fs.writeFileSync(indexFile, html, {encoding: 'utf-8'})
+    return;
+  });
+}
+
+module.exports.setup = setup;
diff --git a/package.json b/package.json
@@ -2,7 +2,10 @@
   "name": "screenshoter-report-analyzer",
   "description": "A beautiful angular based analytics tool to visually check and fix protractor tests results.",
   "version": "0.6.0",
-  "dependencies": {},
+  "dependencies": {
+    "fs-extra": "^7.0.0"
+  },
+  "main": "index.js",
   "scripts": {
     "test": "gulp test",
     "release": "standard-version"