zerobyte/app/server/cli/index.ts

import { Command } from "commander";
import { assignOrganizationCommand } from "./commands/assign-organization";
import { changeEmailCommand } from "./commands/change-email";
import { changeUsernameCommand } from "./commands/change-username";
import { disable2FACommand } from "./commands/disable-2fa";
import { enablePasswordLoginCommand } from "./commands/enable-password-login";
import { rekey2FACommand } from "./commands/rekey-2fa";
import { resetPasswordCommand } from "./commands/reset-password";
import { config } from "../core/config";
import { db } from "../db/db";

const program = new Command();

program
	.name("zerobyte")
	.description("Zerobyte CLI - Backup automation tool built on top of Restic")
	.version(config.appVersion);
program.addCommand(resetPasswordCommand);
program.addCommand(disable2FACommand);
program.addCommand(enablePasswordLoginCommand);
program.addCommand(changeUsernameCommand);
program.addCommand(changeEmailCommand);
program.addCommand(rekey2FACommand);
program.addCommand(assignOrganizationCommand);

export async function runCLI(argv: string[]): Promise<boolean> {
	db.run("PRAGMA foreign_keys = ON;");

	const args = argv.slice(2);
	const isCLIMode = process.env.ZEROBYTE_CLI === "1";

	if (args.length === 0) {
		if (isCLIMode) {
			program.help();
			return true;
		}
		return false;
	}

	if (!isCLIMode && args[0].startsWith("-")) {
		return false;
	}

	await program.parseAsync(argv).catch((err) => {
		if (err.message.includes("SIGINT")) {
			process.exit(0);
		}

		console.error(err.message);
		process.exit(1);
	});

	return true;
}

export { program };