import * as HtmlWebpackPlugin from 'html-webpack-plugin';
import { load } from 'cheerio';
import { readFileSync } from 'fs';
import { join, dirname } from 'path';
import { Compiler, Plugin } from 'webpack';
import { getTemplates, extractParts, setEntries } from './helpers';

export interface Html5EntryWebpackPluginOptions {}

export class Html5EntryWebpackPlugin implements Plugin {
  constructor(private options: Html5EntryWebpackPluginOptions = {}) {}

  apply(compiler: Compiler) {
    const entry = compiler.options.entry;
    const [template] = getTemplates(entry);

    if (template) {
      const src = dirname(template);
      const templateContent = load(readFileSync(template, 'utf8'));
      const entries = extractParts(templateContent).map(entry => join(src, entry));
      const plugins = [
        new HtmlWebpackPlugin({
          templateContent: templateContent.html(),
        }),
      ];

      setEntries(compiler.options, template, entries);
      plugins.forEach(plugin => plugin.apply(compiler));
    }
  }
}