#!/usr/bin/env node
'use strict';

var fs = require('fs');
var Rasha = require('../rsa');
var PEM = require('@root/pem');
var ASN1 = require('@root/asn1');

var infile = process.argv[2];
var format = process.argv[3];
var msg = process.argv[4];
var sign;
if ('sign' === format) {
	sign = true;
	format = 'pkcs8';
}

if (!infile) {
	infile = 'jwk';
}

if (
	-1 !==
	['jwk', 'pem', 'json', 'der', 'pkcs1', 'pkcs8', 'spki'].indexOf(infile)
) {
	console.info('Generating new key...');
	Rasha.generate({
		format: infile,
		modulusLength: parseInt(format, 10) || 2048,
		encoding: parseInt(format, 10) ? null : format
	})
		.then(function(key) {
			if ('der' === infile || 'der' === format) {
				key.private = key.private.toString('binary');
				key.public = key.public.toString('binary');
			}
			console.info(key.private);
			console.info(key.public);
		})
		.catch(function(err) {
			console.error(err);
			process.exit(1);
		});
	return;
}
var key = fs.readFileSync(infile, 'ascii');

try {
	key = JSON.parse(key);
} catch (e) {
	// ignore
}

var thumbprint = 'thumbprint' === format;
if (thumbprint) {
	format = 'public';
}

if ('string' === typeof key) {
	if (thumbprint) {
		Rasha.thumbprint({ pem: key }).then(console.info);
		return;
	}
	if ('tpl' === format) {
		var block = PEM.parseBlock(key);
		var asn1 = ASN1.parse(block.der);
		ASN1.tpl(asn1);
		return;
	}
	if (sign) {
		signMessage(key, msg);
		return;
	}

	var pub = -1 !== ['public', 'spki', 'pkix'].indexOf(format);
	Rasha.import({ pem: key, public: pub || format })
		.then(function(jwk) {
			console.info(JSON.stringify(jwk, null, 2));
		})
		.catch(function(err) {
			console.error(err);
			process.exit(1);
		});
} else {
	if (thumbprint) {
		Rasha.thumbprint({ jwk: key }).then(console.info);
		return;
	}
	Rasha.export({ jwk: key, format: format })
		.then(function(pem) {
			if (sign) {
				signMessage(pem, msg);
				return;
			}
			console.info(pem);
		})
		.catch(function(err) {
			console.error(err);
			process.exit(2);
		});
}

function signMessage(pem, name) {
	var msg;
	try {
		msg = fs.readFileSync(name);
	} catch (e) {
		console.warn(
			'[info] input string did not exist as a file, signing the string itself'
		);
		msg = Buffer.from(name, 'binary');
	}
	var crypto = require('crypto');
	var sign = crypto.createSign('SHA256');
	sign.write(msg);
	sign.end();
	var buf = sign.sign(pem);
	console.info(buf.toString('base64'));
	/*
  Rasha.sign({ pem: pem, message: msg, alg: 'SHA256' }).then(function (sig) {
  }).catch(function () {
    console.error(err);
    process.exit(3);
  });
  */
}