BanAttribute.java

/**
 * 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.policymanagement;

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;

public class BanAttribute extends PolicyManagementCLI {

    private static String[] COMMAND_NAME_VALUES = { "ban" };
    private static String DESCRIPTION = "Ban an attribute. <id> is any of the attribute ids that can be specified in the "
            + "simplified policy language. By default the attribute is banned for resource and action both with value \".*\". "
            + "Different values for resource and action can be set using options --"
            + OPT_RESOURCE_LONG
            + " and --"
            + OPT_ACTION_LONG
            + ".\nExample:\n pap-admin ban subject \"/C=IT/O=INFN/OU=Personal Certificate/L=CNAF/CN=Alberto Forti\"\n"
            + "pap-admin ban fqan /dteam/test";
    private static String USAGE = "[options] <id> <value>";
    private String alias = null;

    public BanAttribute() {
        super(COMMAND_NAME_VALUES, USAGE, DESCRIPTION, null);
    }

    @SuppressWarnings("static-access")
    @Override
    protected Options defineCommandOptions() {
        Options options = new Options();
        options.addOption(OptionBuilder.hasArg(false)
                                       .withDescription("Set the policy as public (default)")
                                       .withLongOpt(OPT_PUBLIC_LONG)
                                       .create());
        options.addOption(OptionBuilder.hasArg(false)
                                       .withDescription("Set the policy as private (it won't be distributed)")
                                       .withLongOpt(OPT_PRIVATE_LONG)
                                       .create());
        options.addOption(OptionBuilder.hasArg(true)
                                       .withDescription(OPT_ACTION_DESCRIPTION)
                                       .withLongOpt(OPT_ACTION_LONG)
                                       .create(OPT_ACTION));
        options.addOption(OptionBuilder.hasArg(true)
                                       .withDescription(OPT_RESOURCE_DESCRIPTION)
                                       .withLongOpt(OPT_RESOURCE_LONG)
                                       .create(OPT_RESOURCE));
        options.addOption(OptionBuilder.hasArg(true)
                                       .withDescription(OPT_PAPALIAS_DESCRIPTION)
                                       .withLongOpt(OPT_PAPALIAS_LONG)
                                       .create());
        return options;
    }

    @Override
    protected int executeCommand(CommandLine commandLine) throws ParseException, RemoteException {

        String[] args = commandLine.getArgs();

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

        String id = args[1];
        String value = args[2];

        if (commandLine.hasOption(OPT_PAPALIAS_LONG)) {
            alias = commandLine.getOptionValue(OPT_PAPALIAS_LONG);
        }

        boolean isPublic = true;
        if (commandLine.hasOption(OPT_PRIVATE_LONG)) {
            isPublic = false;
        }

        String resource = null;
        String action = null;

        if (commandLine.hasOption(OPT_RESOURCE)) {
            resource = commandLine.getOptionValue(OPT_RESOURCE);
        } else {
            resource = ".*";
        }

        if (commandLine.hasOption(OPT_ACTION)) {
            action = commandLine.getOptionValue(OPT_ACTION);
        } else {
            action = ".*";
        }

        if (verboseMode) {
            System.out.print("Adding deny rule... ");
        }

        String policyId = null;

        policyId = highlevelPolicyMgmtClient.ban(alias, id, value, resource, action, isPublic);

        if (policyId == null) {
            printOutputMessage(String.format("error (ban rule already exists)."));
        } else {
            if (verboseMode) {
                System.out.println("ok");
            }
        }
        return ExitStatus.SUCCESS.ordinal();
    }
}