inline-critical/cli.js

#!/usr/bin/env node
'use strict';
const os = require('os');
const fs = require('fs');
const meow = require('meow');
const indentString = require('indent-string');
const stdin = require('get-stdin');
const css = require('css');
const _ = require('lodash');
const inlineCritical = require('.');

let ok;
const help = `
Usage: inline-critical <input> [<option>]

Options:
    -c, --css       Path to CSS file
    -h, --html      Path to HTML file
    -i, --ignore    Skip matching stylesheets
    -m, --minify    Minify the styles before inlining (default)
    -e, --extract   Remove the inlined styles from any stylesheets referenced in the HTML
    -b, --base      Is used when extracting styles to find the files references by href attributes
    -s, --selector  Optionally defines the element used by loadCSS as a reference for inlining
`;

const cli = meow(help, {
    autoHelp: true,
    autoVersion: true,
    flags: {
        css: {
            type: 'string',
            alias: 'c'
        },
        html: {
            type: 'string',
            alias: 'h'
        },
        ignore: {
            type: 'boolean',
            alias: 'i'
        },
        minify: {
            type: 'boolean',
            alias: 'm'
        },
        extract: {
            type: 'boolean',
            alias: 'e'
        },
        base: {
            type: 'string',
            alias: 'b'
        },
        selector: {
            type: 'string',
            alias: 's'
        }
    }
});

// Cleanup cli flags
cli.flags = _.reduce(cli.flags, (res, val, key) => {
    if (key.length <= 1) {
        return res;
    }

    switch (key) {
        case 'css':
        case 'html':
            try {
                res[key] = read(val);
            } catch (err) {
            }
            break;
        case 'base':
            res.basePath = val;
            break;
        case 'ignore':
            if (_.isString(val) || _.isRegExp(val)) {
                val = [val];
            }
            res.ignore = _.map(val || [], ignore => {
                // Check regex
                const match = ignore.match(/^\/(.*)\/([igmy]+)?$/);

                if (match) {
                    return new RegExp(_.escapeRegExp(match[1]), match[2]);
                }
                return ignore;
            });
            break;
        default:
            res[key] = val;
            break;
    }

    return res;
}, {});

function error(err) {
    process.stderr.write(indentString('Error: ' + (err.message || err), 4));
    process.stderr.write(os.EOL);
    process.stderr.write(indentString(help, 4));
    process.exit(1);
}

function read(file) {
    try {
        return fs.readFileSync(file, 'utf8');
    } catch (err) {
        error(err);
    }
}

function run(data) {
    const opts = _.defaults(cli.flags, {basePath: process.cwd()});
    ok = true;

    if (data) {
        // Detect html
        try {
            css.parse(data);
            opts.css = data;
        } catch (err) {
            opts.html = data;
        }
    }

    _.forEach(cli.input, file => {
        const tmp = read(file);
        try {
            css.parse(tmp);
            opts.css = tmp;
        } catch (err) {
            opts.html = tmp;
        }
    });

    if (!opts.html || !opts.css) {
        cli.showHelp();
    }

    try {
        const out = inlineCritical(opts.html, opts.css, opts);
        process.stdout.write(out.toString(), process.exit);
    } catch (err) {
        error(err);
    }
}

// Get stdin
stdin().then(run);
setTimeout(() => {
    if (ok) {
        return;
    }
    run();
}, 100);