inline-critical/index.js

/**
 * Module to inline styles while loading the existing stylesheets async
 *
 * @author Ben Zörb @bezoerb https://github.com/bezoerb
 * @copyright Copyright (c) 2014 Ben Zörb
 *
 * Licensed under the MIT license.
 * http://bezoerb.mit-license.org/
 * All rights reserved.
 */
'use strict';
const fs = require('fs');
const path = require('path');
const _ = require('lodash');
const UglifyJS = require('uglify-js');
const cave = require('cave');
const reaver = require('reaver');
const cheerio = require('cheerio');
const render = require('dom-serializer');
const parse = require('cheerio/lib/parse');
const CleanCSS = require('clean-css');
const slash = require('slash');
const normalizeNewline = require('normalize-newline');
const resolve = require('resolve');
const detectIndent = require('detect-indent');

/**
 * Get loadcss + cssrelpreload script
 *
 * @returns {string}
 */
function getScript() {
    const loadCssMain = resolve.sync('fg-loadcss');
    const loadCssBase = path.dirname(loadCssMain);

    const loadCSS = read(path.join(loadCssBase, 'cssrelpreload.js'));
    return UglifyJS.minify(loadCSS).code;
}

/**
 * Fixup slashes in file paths for windows
 *
 * @param {string} str
 * @return {string}
 */
function normalizePath(str) {
    return process.platform === 'win32' ? slash(str) : str;
}

/**
 * Read file *
 * @param {string} file
 * @returns {string}
 */
function read(file) {
    return fs.readFileSync(file, 'utf8');
}

/**
 * Get the indentation of the link tags
 * @param html
 * @param $el
 */
function getIndent(html, $el) {
    const regName = new RegExp(_.escapeRegExp(_.get($el, 'name')));
    const regHref = new RegExp(_.escapeRegExp(_.get($el, 'attribs.href')));
    const regRel = new RegExp(_.escapeRegExp(_.get($el, 'attribs.rel')));
    const lines = _.filter(html.split(/[\r\n]+/), line => {
        return regName.test(line) && regHref.test(line) && regRel.test(line);
    });
    return detectIndent(lines.join('\n')).indent;
}

module.exports = function (html, styles, options) {
    if (!_.isString(html)) {
        html = String(html);
    }
    const $ = cheerio.load(html, {
        decodeEntities: false
    });

    const allLinks = $('link[rel="stylesheet"], link[rel="preload"][as="style"]').filter(function () {
        return !$(this).parents('noscript').length;
    });

    let links = allLinks.filter('[rel="stylesheet"]');

    const o = _.assign({
        minify: true
    }, options || {});

    const target = o.selector || allLinks.get(0) || $('head script').get(0);
    const {indent} = detectIndent(html);
    const targetIndent = getIndent(html, target);
    const $target = $(target);

    if (_.isString(o.ignore)) {
        o.ignore = [o.ignore];
    }

    if (o.ignore) {
        links = _.filter(links, link => {
            const href = $(link).attr('href');
            return _.findIndex(options.ignore, arg => {
                return (_.isRegExp(arg) && arg.test(href)) || arg === href;
            }) === -1;
        });
    }

    // Minify if minify option is set
    if (o.minify) {
        styles = new CleanCSS().minify(styles).styles; // eslint-disable-line prefer-destructuring
    }

    if (styles) {
        const elements = [
            '<style>',
            indent + styles.replace(/(\r\n|\r|\n)/g, '$1' + targetIndent + indent).replace(/^[\s\t]+$/g, ''),
            '</style>', ''
        ].join('\n' + targetIndent).replace(/(\r\n|\r|\n)[\s\t]+(\r\n|\r|\n)/g, '$1$2');

        if ($target.length > 0) {
            // Insert inline styles right before first <link rel="stylesheet" /> or other target
            $target.before(elements);
        } else {
            // Just append to the head
            $('head').append(elements);
        }
    }

    if (links.length > 0) {
        // Modify links and ad clones to noscript block
        $(links).each(function (idx, el) {
            if (o.extract && !o.basePath) {
                throw new Error('Option `basePath` is missing and required when using `extract`!');
            }

            const $el = $(el);
            const elIndent = getIndent(html, el);

            if (o.extract) {
                const href = $el.attr('href');
                const file = path.resolve(path.join(o.basePath, href));
                if (fs.existsSync(file)) {
                    const diff = normalizeNewline(cave(file, {css: styles}));
                    fs.writeFileSync(reaver.rev(file, diff), diff);
                    $el.attr('href', normalizePath(reaver.rev(href, diff)));
                }
            }

            // Add each fallback right behind the current style to keep source order when ignoring stylesheets
            $el.after('\n' + elIndent + '<noscript>' + render(this) + '</noscript>');

            // Add preload atttibutes to actual link element
            $el.attr('rel', 'preload');
            $el.attr('as', 'style');
            $el.attr('onload', 'this.onload=null;this.rel=\'stylesheet\'');
        });

        // Add loadcss + cssrelpreload polyfill
        const scriptAnchor = $('link[rel="stylesheet"], noscript').filter(function () {
            return !$(this).parents('noscript').length;
        }).last().get(0);

        $(scriptAnchor).after('\n' + targetIndent + '<script>' + getScript() + '</script>');
    }

    const dom = parse($.html());
    const markup = render(dom);

    return Buffer.from(markup);
};