staticrypt/cli/index.js

'use strict';

var CryptoJS = require("crypto-js");
var FileSystem = require("fs");
var https = require("https");

const SCRIPT_URL = 'https://cdnjs.cloudflare.com/ajax/libs/crypto-js/3.1.9-1/crypto-js.min.js';
const SCRIPT_TAG = '<script src="' + SCRIPT_URL + '" integrity="sha384-lp4k1VRKPU9eBnPePjnJ9M2RF3i7PC30gXs70+elCVfgwLwx1tv5+ctxdtwxqZa7" crossorigin="anonymous"></script>';

const namedArgs = parseArgs(process.argv, process.argv.length);

try{
    var contents = FileSystem.readFileSync(namedArgs.input, 'utf8');
}catch(e){
    console.log("Failure: input file does not exist!");
    process.exit(1);
}

var encrypted = CryptoJS.AES.encrypt(contents, namedArgs.password);
var hmac = CryptoJS.HmacSHA256(encrypted.toString(), CryptoJS.SHA256(namedArgs.password)).toString();
var encryptedMessage = hmac + encrypted;

var data = {
    title: namedArgs.title != null ? namedArgs.title : "Protected Page",
    instructions: namedArgs.instructions != null ? namedArgs.instructions : "",
    encrypted: encryptedMessage,
    crypto_tag: SCRIPT_TAG,
    embed: namedArgs.embed != null ? namedArgs.embed : false,
    outputFilePath: namedArgs.output != null ? namedArgs.output : namedArgs.input + ".encrypted"
};

if(data.embed){
    https.get(SCRIPT_URL, (resp) => {
        let txt = '';

        resp.on('data', (chunk) => {
            txt += chunk;
        });

        resp.on('end', () => {
            data["crypto_tag"] = '<script>' + txt + '</script>';
            console.log(data);
            genFile(data);
        });

    }).on('error', (err) => {
        console.log("Failure: could not fetch embedded script");
        process.exit(1);
    });
}else{
    genFile(data);
}

function genFile(data){
    try{
        var templateContents = FileSystem.readFileSync('./password_template.html', 'utf8');
    }catch(e){
        console.log("Failure: could not read template!");
        process.exit(1);
    }

    var renderedTemplate = render(templateContents, data);

    try{
        FileSystem.writeFileSync(data.outputFilePath, renderedTemplate);
    }catch(e){
        console.log("Failure: could not generate output file!");
        process.exit(1);
    }
}

function render(tpl, data){
    return tpl.replace(/{(.*?)}/g, function (_, key) {
        return data && data[key] || '';
    });
}

function parseArgs(argv, argc){
    var argList = {
        required: {
            "input": false,
            "password": false
        },
        optional: {
            "output": false,
            "title": false,
            "instructions" : false,
            "embed": false
        }
    };

    var namedArgs = {};

    // parse through args
    for(var i = 2; i < argc; ++i){
        var pieces = argv[i].split('=');
        if(pieces.length != 2){
            console.error("Failure: invalid argument '" + argv[i] + "'");
            process.exit(1);
        }else{
            const name = pieces[0];

            if(argList.required[name] == true){
                // duplicate
                console.log("Failure: duplicate argument '" + name + "'");
                process.exit(1);
            }else if(argList.required[name] == false){
                // valid
                namedArgs[name] = pieces[1];
                argList.required[name] = true;
            }else if(argList.optional[name] == true){
                // duplicate
                console.log("Failure: duplicate argument '" + name + "'");
                process.exit(1);
            }else if(argList.optional[name] == false){
                // valid
                namedArgs[name] = pieces[1];
                argList.optional[name] = true;
            }else{
                // unknown argument
                console.log("Failure: unknown argument '" + name + "'");
                process.exit(1);
            }
        }
    }

    // make sure all required args are present
    for(var argName in argList.required){
        if(argList.required[argName] == false){
            console.log("Failure: missing required argument '" + argName + "'");
            process.exit(1);
        }
    }

    // validate password length
    if(namedArgs.password.length < 5){
        console.log("Failure: password length less than 5");
        process.exit(1);
    }

    return namedArgs;
}