import {ArgParser} from '/scripts/lib/arg-parser';
import {maxThreads, progressBar, slowPrint} from '/scripts/lib/utils';
import {copyWithDependencies} from '/scripts/copy';

/**
 * Check if exploit is available for use.
 *
 * @param {NS} ns - BitBurner API
 * @returns {string[]} - Available exploits
 */
export function availableExploits(ns) {
	return ['BruteSSH.exe', 'FTPCrack.exe', 'relaySMTP.exe', 'HTTPWorm.exe', 'SQLInject.exe']
		.filter(e => ns.fileExists(e));
}

/**
 * Attempt to root a server.
 *
 * @param {NS} ns - BitBurner API
 * @param {string} server - Server to attempt to root
 * @returns {[boolean, string[]]} - Tuple of whether the server was successfully rooted & the exploits which where ran
 */
export function root(ns, server) {
	function runExploit(name) {
		if(name == 'BruteSSH.exe') ns.brutessh(server);
		else if(name == 'FTPCrack.exe') ns.ftpcrack(server);
		else if(name == 'relaySMTP.exe') ns.relaysmtp(server);
		else if(name == 'HTTPWorm.exe') ns.httpworm(server);
		else if(name == 'SQLInject.exe') ns.sqlinject(server);
	}

	const exploits = availableExploits(ns);
	exploits.forEach(e => runExploit(e));
	try {
		ns.nuke(server)
		return [true, exploits];
	} catch {
		return [false, exploits];
	}
}

/**
 * Automatically gain root access to a server. A file can also be uploaded & executed.
 *
 * @param {NS} ns - BitBurner API
 */
export async function main(ns) {
	// Setup
	ns.disableLog('ALL');
	const argParser = new ArgParser('rootkit.js', 'Automatically gain root access to a server. A file can also be uploaded & executed.', [
		{name: 'server', desc: 'Server to root, defaults to local server', optional: true, default: ns.getHostname()},
		{name: 'script', desc: 'Script to copy & execute', optional: true},
		{name: 'args', desc: 'Arguments for script. Forward the discovered server with: {{SERVER}}', optional: true, extras: true},
		{name: 'cpu', desc: 'Number of CPU threads to start script with, will use maximum if not specified', flags: ['-c', '--cpu'], default: false},
		{name: 'quite', desc: 'Suppress program output', flags: ['-q', '--quite'], default: false},
	]);
	const args = argParser.parse(ns.args);
	if(args['script'] && !args['cpu']) args['cpu'] =
		~~(ns.getServerMaxRam(args['server']) / ns.getScriptRam(args['script'], 'home')) || 1;

	// Help
	if(args['help'] || args['_error'].length)
		return ns.tprint(argParser.help(args['help'] ? null : args['_error'][0], args['_command']));

	// Banner
	if(!args['quite']) {
		ns.tprint('===================================================');
		ns.tprint(`Rooting: ${args['server']}`);
		ns.tprint('===================================================');
	}

	// Root server if access is restricted
	if(!ns.hasRootAccess(args['server'])) {
		const [rooted, exploits] = root(ns, args['server']);
		if(exploits.length >= 1) await slowPrint(ns, 'Exploiting SSH (*:22)...');
		if(exploits.length >= 2) await slowPrint(ns, 'Exploiting FTP (*:24)...');
		if(exploits.length >= 3) await slowPrint(ns, 'Exploiting SMTP (*:25)...');
		if(exploits.length >= 4) await slowPrint(ns, 'Exploiting HTTP (*:80)...');
		if(exploits.length >= 5) await slowPrint(ns, 'Exploiting MSQL (*:3306)...');
		ns.tprint(`Root: ${rooted ? 'Success!' : 'Failed'}`);
		if(!rooted) ns.exit();
	} else {
		ns.tprint(`Root: Skipped`);
	}
	ns.tprint('');

	// Start script if required
	if(args['script']) {
		// Copy script & it's dependencies
		const files = await copyWithDependencies(ns, args['script'], args['server']);
		if(!args['quite']) {
			await ns.sleep(500);
			ns.tprint('Copying files:');
			for(let file of files) await progressBar(ns, file);
		}

		// Start the script
		const threads = args['cpu'] || maxThreads(ns, args['file'], args['server']) || 1;
		if(!args['quite']) {
			ns.tprint('');
			ns.tprint(`Executing with ${threads} thread${threads > 1 ? 's' : ''}...`);
			await ns.sleep(500);
		}
		ns.killall(args['server']);
		const pid = ns.exec(args['file'], args['server'], threads, args['args']);
		if(!args['quite']) {
			ns.tprint(!!pid ? 'Done!' : 'Failed to start');
			ns.tprint('');
		}
	}
}

/**
 * BitBurner autocomplete.
 *
 * @param {{servers: string[], txts: string[], scripts: string[], flags: string[]}} data - Contextual information
 * @returns {string[]} - Pool of autocomplete options
 */
export function autocomplete(data) {
	return [...data.servers, ...data.scripts];
}