refactor: code

Author: evilebottnawi

src/Warning.js

@@ -0,0 +1,16 @@
+export default class Warning extends Error {
+  constructor(warning) {
+    super(warning);
+    const { text, line, column } = warning;
+    this.name = 'Warning';
+    this.message = `${this.name}\n\n`;
+
+    if (typeof line !== 'undefined') {
+      this.message += `(${line}:${column}) `;
+    }
+
+    this.message += `${text}`;
+
+    this.stack = false;
+  }
+}

src/index.js

@@ -1,56 +1,33 @@
-import { parse } from 'url';
-
-import { compile } from 'es6-templates';
-import { minify } from 'html-minifier-terser';
-import { getOptions, isUrlRequest } from 'loader-utils';
+import { getOptions } from 'loader-utils';
 import validateOptions from 'schema-utils';
 
+import { attributePlugin, interpolatePlugin, minimizerPlugin } from './plugins';
+import Warning from './Warning';
+
 import {
-  getLinks,
-  getUniqueIdent,
-  replaceLinkWithIdent,
+  pluginRunner,
   isProductionMode,
   getImportCode,
   getExportCode,
 } from './utils';
 
 import schema from './options.json';
 
-export const raw = true;
-
-export default function htmlLoader(source) {
+export default function htmlLoader(content) {
   const options = getOptions(this) || {};
 
   validateOptions(schema, options, {
     name: 'HTML Loader',
     baseDataPath: 'options',
   });
 
-  let content = source.toString();
-
-  const links = getLinks(content, options.attributes);
-  const replacers = new Map();
-
-  let offset = 0;
+  const plugins = [];
 
-  for (const link of links) {
-    if (link.value && isUrlRequest(link.value, options.root)) {
-      const uri = parse(link.value);
+  const attributes =
+    typeof options.attributes === 'undefined' ? true : options.attributes;
 
-      if (typeof uri.hash !== 'undefined') {
-        uri.hash = null;
-        link.value = uri.format();
-        link.length = link.value.length;
-      }
-
-      const ident = getUniqueIdent(replacers);
-
-      replacers.set(ident, link.value);
-
-      content = replaceLinkWithIdent(content, link, ident, offset);
-
-      offset += ident.length - link.length;
-    }
+  if (attributes) {
+    plugins.push(attributePlugin(options));
   }
 
   const minimize =
@@ -59,46 +36,40 @@ export default function htmlLoader(source) {
       : options.minimize;
 
   if (minimize) {
-    const minimizeOptions =
-      typeof minimize === 'boolean'
-        ? {
-            collapseWhitespace: true,
-            conservativeCollapse: true,
-            keepClosingSlash: true,
-            minifyCSS: true,
-            minifyJS: true,
-            removeAttributeQuotes: true,
-            removeComments: true,
-            removeScriptTypeAttributes: true,
-            removeStyleLinkTypeAttributes: true,
-            useShortDoctype: true,
-          }
-        : minimize;
-
-    try {
-      content = minify(content, minimizeOptions);
-    } catch (error) {
-      this.emitError(error);
-    }
+    plugins.push(minimizerPlugin(options));
+  }
+
+  const { interpolate } = options;
+
+  if (interpolate) {
+    plugins.push(interpolatePlugin(options));
+  }
+
+  const { html, messages, warnings, errors } = pluginRunner(plugins).process(
+    content
+  );
+
+  for (const warning of warnings) {
+    this.emitWarning(new Warning(warning));
+  }
+
+  for (const error of errors) {
+    this.emitError(new Error(error));
   }
 
-  if (options.interpolate) {
-    try {
-      // Double escape quotes so that they are not unescaped completely in the template string
-      content = compile(
-        `\`${content.replace(/\\"/g, '\\\\"').replace(/\\'/g, "\\\\\\'")}\``
-      ).code;
-    } catch (error) {
-      this.emitError(error);
+  const replacers = [];
 
-      content = JSON.stringify(content);
+  for (const message of messages) {
+    // eslint-disable-next-line default-case
+    switch (message.type) {
+      case 'replacer':
+        replacers.push(message.value);
+        break;
     }
-  } else {
-    content = JSON.stringify(content);
   }
 
-  const importCode = getImportCode(this, content, replacers, options);
-  const exportCode = getExportCode(content, replacers, options);
+  const importCode = getImportCode(this, html, replacers, options);
+  const exportCode = getExportCode(html, replacers, options);
 
   return `${importCode}${exportCode};`;
 }