Package org.nimbustools.auto_config

Source Code of org.nimbustools.auto_config.GridMapAdd

/*
* Copyright 1999-2008 University of Chicago
*
* 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.nimbustools.auto_config;

import org.globus.security.gridmap.GridMap;

import java.io.File;

public class GridMapAdd {

    public void add(String path,
                    String DN,
                    String user) throws Exception {

        if (DN == null || DN.trim().length() == 0) {
            throw new Exception("no DN");
        }

        if (user == null || user.trim().length() == 0) {
            throw new Exception("no user to map to");
        }

        final String dn = DN.trim();
        final String userMapping = user.trim();

        System.out.println("Adding DN '" + dn + "'");
        System.out.println(" ... with user mapping '" + userMapping + "'");
        System.out.println(" ... to grid-mapfile '" + path + "'");

        final GridMap gridmap = new GridMap();
        gridmap.load(path);

        boolean foundOne = false;
        final String[] userids = gridmap.getUserIDs(dn);
        if (userids != null && userids.length > 0) {
            for (int i = 0; i < userids.length; i++) {
                String userid = userids[i];
                if (userid == null) {
                    throw new Exception("invalid grid-mapfile, user mapping " +
                            "is present but null?");
                }
                if (!userid.trim().equalsIgnoreCase(userMapping)) {
                    throw new Exception("There is a username mapping " +
                        "present already for this DN that does not match " +
                        "your requested userid '" + userMapping + "', exiting");
                }
                foundOne = true;
            }
        }

        if (foundOne) {
            System.out.println(
                    "Nothing to do, the DN is already added to '" + path + "'");
            return;
        }

        File file = new File(path);
        if (!file.canWrite()) {
            throw new Exception(
                    "This file is not writable: '" + path + "'");
        }

        final String newline = "\"" + dn + "\" " + userMapping;

        final TextFile textFile = new TextFile(path);
        textFile.add(newline);
        textFile.writeFile(file);
    }

    public static void mainImpl(String[] args) throws Exception {
        if (args == null || args.length != 3) {
            throw new Exception(
                    "You need to supply three and only three arguments:"
                  + "\n  1 - path to existing grid-mapfile"
                  + "\n  2 - DN"
                  + "\n  3 - user to map to");
        }

        new GridMapAdd().add(args[0], args[1], args[2]);
    }

    public static void main(String[] args) {
        try {
            mainImpl(args);
        } catch (Throwable t) {
            System.err.println("Problem: " + t.getMessage());
            System.exit(1);
        }
    }
}
TOP

Related Classes of org.nimbustools.auto_config.GridMapAdd

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.