kerby-kerb/kerb-admin/src/main/java/org/apache/kerby/kerberos/kerb/admin/kadmin/remote/command/RemoteAddPrincipalCommand.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.kerby.kerberos.kerb.admin.kadmin.remote.command;

import org.apache.kerby.kerberos.kerb.KrbException;
import org.apache.kerby.kerberos.kerb.admin.kadmin.remote.AdminClient;

/**
 * Remote add principal command
 */
public class RemoteAddPrincipalCommand extends RemoteCommand {

    public static final String USAGE = "Usage: add_principal [options] <principal-name>\n"
        + "\toptions are:\n"
        + "\t\t[-randkey|-nokey]\n"
        + "\t\t[-pw password]"
        + "\tExample:\n"
        + "\t\tadd_principal -pw mypassword alice\n";

    public RemoteAddPrincipalCommand(AdminClient adminClient) {
        super(adminClient);
    }

    @Override
    public void execute(String input) throws KrbException {
        String[] items = input.split("\\s+");

        if (items.length < 2) {
            System.err.println(USAGE);
            return;
        }

        String adminRealm = adminClient.getAdminConfig().getAdminRealm();
        String clientPrincipal = items[items.length - 1] + "@" + adminRealm;

        if (!items[1].startsWith("-")) {
            adminClient.requestAddPrincipal(clientPrincipal);
        } else if (items[1].startsWith("-nokey")) {
            adminClient.requestAddPrincipal(clientPrincipal);
        } else if (items[1].startsWith("-pw")) {
            String password = items[2];
            adminClient.requestAddPrincipal(clientPrincipal, password);
        } else {
            System.err.println("add_principal command format error.");
            System.err.println(USAGE);
        }
    }
}