wkd-java/wkd-java-cli/src/main/java/pgp/wkd/cli/WKDCLI.java

// SPDX-FileCopyrightText: 2022 Paul Schaub <vanitasvitae@fsfe.org>
//
// SPDX-License-Identifier: Apache-2.0

package pgp.wkd.cli;

import pgp.wkd.exception.CertNotFetchableException;
import pgp.wkd.exception.RejectedCertificateException;
import pgp.wkd.cli.command.GetCmd;
import picocli.CommandLine;

@CommandLine.Command(
        name = "wkd",
        resourceBundle = "msg_wkd",
        subcommands = {
                CommandLine.HelpCommand.class,
                GetCmd.class
        }
)
public class WKDCLI {

    public static void main(String[] args) {
        int exitCode = execute(args);
        if (exitCode != 0) {
            System.exit(exitCode);
        }
    }

    public static int execute(String[] args) {
        return new CommandLine(WKDCLI.class)
                .setExitCodeExceptionMapper(new CommandLine.IExitCodeExceptionMapper() {
                    @Override
                    public int getExitCode(Throwable exception) {
                        if (exception instanceof RejectedCertificateException) {
                            return ((RejectedCertificateException) exception).getErrorCode();
                        } else if (exception instanceof CertNotFetchableException) {
                            return CertNotFetchableException.ERROR_CODE;
                        }

                        // Others get mapped to 1
                        return 1;
                    }
                })
                .setCommandName("wkdcli")
                .execute(args);
    }
}