cc-haxe/src/lib/CLIAppBase.hx

package lib;

import lib.args.CLIArgs;
import lib.args.ArgType;
import kernel.ps.IProcess;
import kernel.ps.ProcessHandle;

using tink.CoreApi;

abstract class CLIAppBase implements IProcess {
	private var handle:ProcessHandle;

	private final _subcommandsSync:Map<String, (CLIArgs) -> Bool> = [];
	private final _subcommandsAsync:Map<String, (CLIArgs) -> Future<Bool>> = [];
	private final _subcommandsArgs:Map<String, Array<ArgType>> = [];

	public final function run(handle:ProcessHandle) {
		this.handle = handle;

		var subcommand = handle.args[0];

		if (subcommand == null || subcommand == "") {
			handle.writeLine("No subcommand specified");
			printHelp();
			return handle.close(false);
		}

		var args = handle.args.slice(1);

		if (_subcommandsSync.exists(subcommand)) {
			var argParser = new CLIArgs(_subcommandsArgs.get(subcommand));
			if (!argParser.parse(args)) {
				handle.writeLine(argParser.getError());
				return handle.close(false);
			}
			return handle.close(_subcommandsSync[subcommand](argParser));
		} else if (_subcommandsAsync.exists(subcommand)) {
			var argParser = new CLIArgs(_subcommandsArgs.get(subcommand));
			if (!argParser.parse(args)) {
				handle.writeLine(argParser.getError());
				return handle.close(false);
			}
			_subcommandsAsync[subcommand](argParser).handle(handle.close);
		} else {
			handle.writeLine("Unknown subcommand: " + subcommand);
			printHelp();
			return handle.close(false);
		}
	}

	private function registerSyncSubcommand(command:String, callback:(CLIArgs) -> Bool, args:Array<ArgType> = null) {
		_subcommandsSync.set(command, callback);
		_subcommandsArgs.set(command, args ?? []);
	}

	private function registerAsyncSubcommand(command:String, callback:(CLIArgs) -> Future<Bool>, args:Array<ArgType> = null) {
		_subcommandsAsync.set(command, callback);
		_subcommandsArgs.set(command, args ?? []);
	}

	private function printHelp() {
		handle.writeLine("Usage: <subcommand> [args]");
		handle.writeLine("Subcommands:");
		for (k => v in this._subcommandsArgs) {
			handle.writeLine('    $k ${CLIArgs.getSynopsis(v)}');
		}
	}
}