org.glite.authz.pap.ui.cli.papmanagement.UpdatePap.java Source code

Introduction

Here is the source code for org.glite.authz.pap.ui.cli.papmanagement.UpdatePap.java

Source

/**
 * Copyright (c) Members of the EGEE Collaboration. 2006-2009.
 * See http://www.eu-egee.org/partners/ for details on the copyright holders.
 *
 * Licensed 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.glite.authz.pap.ui.cli.papmanagement;

import java.rmi.RemoteException;

import org.apache.commons.cli.CommandLine;
import org.apache.commons.cli.OptionBuilder;
import org.apache.commons.cli.Options;
import org.apache.commons.cli.ParseException;
import org.glite.authz.pap.common.Pap;

public class UpdatePap extends PAPManagementCLI {

    private static final String[] commandNameValues = { "update-pap", "upap" };
    private static final String DESCRIPTION = "Update pap information. The input is the same as for the \"add-pap\" command, "
            + "the effect is to update old information with the new one.\n"
            + "<alias> is a friendly name (it has to be unique) used to identify the pap\n"
            + "<endpoint> endpoint of the PAP in the following format: [<protocol>://]<host>[:<port>[/path]]\n"
            + "<dn> DN of the endpoint machine";
    private static final String LOPT_PRIVATE = "private";
    private static final String LOPT_PUBLIC = "public";
    private static final String USAGE = "[options] <alias> [<endpoint> <dn>]";

    public UpdatePap() {
        super(commandNameValues, USAGE, DESCRIPTION, null);
    }

    @SuppressWarnings("static-access")
    @Override
    protected Options defineCommandOptions() {
        Options options = new Options();
        options.addOption(OptionBuilder.hasArg(false)
                .withDescription("Set the remote pap as public (allow to distribute its policies)")
                .withLongOpt(LOPT_PUBLIC).create());
        options.addOption(OptionBuilder.hasArg(false).withDescription("Set the remote pap as private (default)")
                .withLongOpt(LOPT_PRIVATE).create());
        options.addOption(OptionBuilder.hasArg(false).withDescription(OPT_LOCAL_DESCRIPTION)
                .withLongOpt(OPT_LOCAL_LONG).create(OPT_LOCAL));
        options.addOption(OptionBuilder.hasArg(false).withDescription(OPT_REMOTE_DESCRIPTION)
                .withLongOpt(OPT_REMOTEL_LONG).create());
        options.addOption(OptionBuilder.hasArg(false).withDescription(OPT_NO_POLICIES_DESCRIPTION)
                .withLongOpt(OPT_NO_POLICIES_LONG).create());
        return options;
    }

    @Override
    protected int executeCommand(CommandLine commandLine) throws ParseException, RemoteException {
        String[] args = commandLine.getArgs();

        if ((args.length != 4) && (args.length != 2)) {
            throw new ParseException("Wrong number of arguments");
        }

        boolean isLocal;
        boolean isPublic = false;

        if (args.length != 2) {
            isLocal = false;
        } else {
            isLocal = true;
        }

        if (commandLine.hasOption(OPT_LOCAL)) {
            isLocal = true;
        }

        if (commandLine.hasOption(OPT_REMOTEL_LONG)) {
            isLocal = true;
        }

        if (commandLine.hasOption(LOPT_PUBLIC)) {
            isPublic = true;
        }

        String alias = args[1];
        String protocol = null;
        String dn = null;
        String host = null;
        String port = null;
        String path = null;

        if (args.length != 2) {
            protocol = AddPap.getProtocol(args[2]);
            host = AddPap.getHostname(args[2]);
            port = AddPap.getPort(args[2]);

            AddPap.validatePort(port);

            path = AddPap.getPath(args[2]);
            dn = args[3];
        }

        Pap pap = new Pap(alias, isLocal, dn, host, port, path, protocol, isPublic);

        String msg = "Updating PAP: ";

        if (verboseMode) {
            System.out.println(msg + pap.toFormattedString(0, msg.length()));
        }

        if (!(papMgmtClient.exists(pap.getAlias()))) {
            System.out.println("PAP doesn't exists.");
            return ExitStatus.FAILURE.ordinal();
        }

        papMgmtClient.updatePap(pap);

        if (pap.isRemote()) {

            if (!commandLine.hasOption(OPT_NO_POLICIES_LONG)) {

                if (verboseMode) {
                    System.out.print("Retrieving policies... ");
                }

                try {
                    papMgmtClient.refreshCache(pap.getAlias());
                } catch (RemoteException e) {
                    System.out.println("Error: pap information successfully updated but cannot retrieve policies.");
                    throw e;
                }

                if (verboseMode) {
                    System.out.println("ok.");
                }
            }
        }

        return ExitStatus.SUCCESS.ordinal();
    }
}