BaseCertificateAuthorityCommandLine.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.apache.nifi.toolkit.tls.service;

import org.apache.commons.cli.CommandLine;
import org.apache.nifi.toolkit.tls.commandLine.BaseCommandLine;
import org.apache.nifi.toolkit.tls.commandLine.CommandLineParseException;
import org.apache.nifi.toolkit.tls.commandLine.ExitCode;
import org.apache.nifi.toolkit.tls.configuration.TlsConfig;
import org.apache.nifi.util.StringUtils;

import java.io.File;

/**
 * Common base argument logic for the CA server and client
 */
public abstract class BaseCertificateAuthorityCommandLine extends BaseCommandLine {
    public static final String TOKEN_ARG = "token";
    public static final String CONFIG_JSON_ARG = "configJson";
    public static final String READ_CONFIG_JSON_ARG = "configJsonIn";
    public static final String USE_CONFIG_JSON_ARG = "useConfigJson";

    public static final String PORT_ARG = "PORT";
    public static final String DEFAULT_CONFIG_JSON = new File("config.json").getPath();

    private String token;
    private String configJsonOut;
    private String configJsonIn;
    private int port;
    private String dn;

    public BaseCertificateAuthorityCommandLine(String header) {
        super(header);
        addOptionWithArg("t", TOKEN_ARG, getTokenDescription());
        addOptionWithArg("f", CONFIG_JSON_ARG, "The place to write configuration info", DEFAULT_CONFIG_JSON);
        addOptionWithArg(null, READ_CONFIG_JSON_ARG, "The place to read configuration info from (defaults to the value of " + CONFIG_JSON_ARG + "), implies "
                + USE_CONFIG_JSON_ARG + " if set.", CONFIG_JSON_ARG + " value");
        addOptionNoArg("F", USE_CONFIG_JSON_ARG, "Flag specifying that all configuration is read from " + CONFIG_JSON_ARG + " to facilitate automated use (otherwise "
                + CONFIG_JSON_ARG + " will only be written to.");
        addOptionWithArg("p", PORT_ARG, getPortDescription(), TlsConfig.DEFAULT_PORT);
        addOptionWithArg("D", DN_ARG, getDnDescription(), new TlsConfig().calcDefaultDn(getDnHostname()));
    }

    protected abstract String getTokenDescription();

    protected abstract String getDnDescription();

    protected abstract String getPortDescription();

    protected abstract String getDnHostname();

    @Override
    protected CommandLine doParse(String[] args) throws CommandLineParseException {
        CommandLine commandLine = super.doParse(args);

        token = commandLine.getOptionValue(TOKEN_ARG);

        boolean useConfigJson = commandLine.hasOption(USE_CONFIG_JSON_ARG);

        configJsonOut = commandLine.getOptionValue(CONFIG_JSON_ARG, DEFAULT_CONFIG_JSON);
        configJsonIn = commandLine.getOptionValue(READ_CONFIG_JSON_ARG);
        if (StringUtils.isEmpty(configJsonIn) && useConfigJson) {
            configJsonIn = configJsonOut;
        }

        if (StringUtils.isEmpty(token) && StringUtils.isEmpty(configJsonIn)) {
            printUsageAndThrow(TOKEN_ARG + " argument must not be empty unless " + USE_CONFIG_JSON_ARG + " or " + READ_CONFIG_JSON_ARG+ " set", ExitCode.ERROR_TOKEN_ARG_EMPTY);
        }
        port = getIntValue(commandLine, PORT_ARG, TlsConfig.DEFAULT_PORT);
        dn = commandLine.getOptionValue(DN_ARG, new TlsConfig().calcDefaultDn(getDnHostname()));
        return commandLine;
    }

    public String getToken() {
        return token;
    }

    public String getConfigJsonOut() {
        return configJsonOut;
    }

    public String getConfigJsonIn() {
        return configJsonIn;
    }

    public int getPort() {
        return port;
    }

    public String getDn() {
        return dn;
    }
}