Newer
Older
Alinga Yeung
committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
/*
************************************************************************
******************* CANADIAN ASTRONOMY DATA CENTRE *******************
************** CENTRE CANADIEN DE DONNÉES ASTRONOMIQUES **************
*
* (c) 2014. (c) 2014.
* Government of Canada Gouvernement du Canada
* National Research Council Conseil national de recherches
* Ottawa, Canada, K1A 0R6 Ottawa, Canada, K1A 0R6
* All rights reserved Tous droits réservés
*
* NRC disclaims any warranties, Le CNRC dénie toute garantie
* expressed, implied, or énoncée, implicite ou légale,
* statutory, of any kind with de quelque nature que ce
* respect to the software, soit, concernant le logiciel,
* including without limitation y compris sans restriction
* any warranty of merchantability toute garantie de valeur
* or fitness for a particular marchande ou de pertinence
* purpose. NRC shall not be pour un usage particulier.
* liable in any event for any Le CNRC ne pourra en aucun cas
* damages, whether direct or être tenu responsable de tout
* indirect, special or general, dommage, direct ou indirect,
* consequential or incidental, particulier ou général,
* arising from the use of the accessoire ou fortuit, résultant
* software. Neither the name de l'utilisation du logiciel. Ni
* of the National Research le nom du Conseil National de
* Council of Canada nor the Recherches du Canada ni les noms
* names of its contributors may de ses participants ne peuvent
* be used to endorse or promote être utilisés pour approuver ou
* products derived from this promouvoir les produits dérivés
* software without specific prior de ce logiciel sans autorisation
* written permission. préalable et particulière
* par écrit.
*
* This file is part of the Ce fichier fait partie du projet
* OpenCADC project. OpenCADC.
*
* OpenCADC is free software: OpenCADC est un logiciel libre ;
* you can redistribute it and/or vous pouvez le redistribuer ou le
* modify it under the terms of modifier suivant les termes de
* the GNU Affero General Public la “GNU Affero General Public
* License as published by the License” telle que publiée
* Free Software Foundation, par la Free Software Foundation
* either version 3 of the : soit la version 3 de cette
* License, or (at your option) licence, soit (à votre gré)
* any later version. toute version ultérieure.
*
* OpenCADC is distributed in the OpenCADC est distribué
* hope that it will be useful, dans l’espoir qu’il vous
* but WITHOUT ANY WARRANTY; sera utile, mais SANS AUCUNE
* without even the implied GARANTIE : sans même la garantie
* warranty of MERCHANTABILITY implicite de COMMERCIALISABILITÉ
* or FITNESS FOR A PARTICULAR ni d’ADÉQUATION À UN OBJECTIF
* PURPOSE. See the GNU Affero PARTICULIER. Consultez la Licence
* General Public License for Générale Publique GNU Affero
* more details. pour plus de détails.
*
* You should have received Vous devriez avoir reçu une
* a copy of the GNU Affero copie de la Licence Générale
* General Public License along Publique GNU Affero avec
* with OpenCADC. If not, see OpenCADC ; si ce n’est
* <http://www.gnu.org/licenses/>. pas le cas, consultez :
* <http://www.gnu.org/licenses/>.
*
* $Revision: 4 $
*
************************************************************************
*/
package ca.nrc.cadc.ac.admin;
Alinga Yeung
committed
import java.io.PrintStream;
Alinga Yeung
committed
import java.security.cert.CertificateException;
import org.apache.log4j.Level;
Alinga Yeung
committed
import org.apache.log4j.Logger;
import ca.nrc.cadc.util.ArgumentMap;
import ca.nrc.cadc.util.Log4jInit;
import ca.nrc.cadc.util.StringUtil;
Alinga Yeung
committed
/**
* This class parses the command line input arguments.
*/
public class CmdLineParser
Alinga Yeung
committed
{
private static Logger log = Logger.getLogger(CmdLineParser.class);
private static final String APP_NAME = "userAdmin";
private static final String[] LOG_PACKAGES =
{"ca.nrc.cadc.ac", "ca.nrc.cadc.auth", "ca.nrc.cadc.util"};
Alinga Yeung
committed
// no need to proceed further if false
private Level logLevel = Level.OFF;
Alinga Yeung
committed
private AbstractCommand command;
private boolean isHelpCommand = false;
Alinga Yeung
committed
/**
* Constructor.
* @param args Input arguments
* @throws UsageException Error in command line
* @throws CertificateException Fail to get a certificate
Alinga Yeung
committed
*/
Alinga Yeung
committed
public CmdLineParser(final String[] args, final PrintStream outStream,
final PrintStream errStream) throws UsageException, CertificateException
Alinga Yeung
committed
{
ArgumentMap am = new ArgumentMap( args );
this.setLogLevel(am);
Alinga Yeung
committed
this.parse(am, outStream, errStream);
Alinga Yeung
committed
}
/**
* Get the user admin command to be performed.
* @return user admin command
*/
public AbstractCommand getCommand()
{
return this.command;
}
* Get the logging level.
*/
public Level getLogLevel()
{
return this.logLevel;
}
/*
* Set the log level.
* @param am Input arguments
* @throws UsageException
protected void setLogLevel(final ArgumentMap am) throws UsageException
this.logLevel = Level.WARN;
// only one log level is allowed
if (am.isSet("v") || am.isSet("verbose"))
{
Alinga Yeung
committed
this.logLevel = Level.INFO;
count++;
}
if (am.isSet("d") || am.isSet("debug"))
{
Alinga Yeung
committed
this.logLevel = Level.DEBUG;
count++;
String msg = "--verbose and --debug are mutually exclusive options";
Alinga Yeung
committed
throw new UsageException(msg);
else
// set the application log level
for (String pkg : LOG_PACKAGES)
{
Log4jInit.setLevel(APP_NAME, pkg, this.logLevel);
}
}
protected boolean hasValue(final String userID) throws UsageException
{
Alinga Yeung
committed
if (!StringUtil.hasText(userID) ||userID.equalsIgnoreCase("true"))
{
String msg = "Missing userID";
throw new UsageException(msg);
}
else
{
return true;
}
protected boolean isValid(final ArgumentMap am) throws UsageException
Alinga Yeung
committed
{
Alinga Yeung
committed
// only one command is allowed per command line
if (am.isSet("list"))
{
Alinga Yeung
committed
this.command = new ListActiveUsers();
count++;
Alinga Yeung
committed
}
if (am.isSet("list-pending"))
{
Alinga Yeung
committed
this.command = new ListPendingUsers();
count++;
Alinga Yeung
committed
}
Alinga Yeung
committed
String userID = am.getValue("view");
Alinga Yeung
committed
{
Alinga Yeung
committed
if (this.hasValue(userID))
Alinga Yeung
committed
this.command = new ViewUser(userID);
Alinga Yeung
committed
count++;
Alinga Yeung
committed
}
Alinga Yeung
committed
userID = am.getValue("reject");
Alinga Yeung
committed
{
Alinga Yeung
committed
if (this.hasValue(userID))
Alinga Yeung
committed
this.command = new RejectUser(userID);
Alinga Yeung
committed
count++;
Alinga Yeung
committed
}
Alinga Yeung
committed
userID = am.getValue("approve");
Alinga Yeung
committed
{
Alinga Yeung
committed
if (this.hasValue(userID))
String dn = am.getValue("dn");
if (dn != null)
{
this.command = new ApproveUser(userID, dn);
}
else
{
throw new UsageException("Missing parameter 'dn'");
}
Alinga Yeung
committed
count++;
Alinga Yeung
committed
}
Alinga Yeung
committed
{
Alinga Yeung
committed
return true;
Alinga Yeung
committed
}
else
{
Alinga Yeung
committed
String msg;
Alinga Yeung
committed
if (count == 0)
{
msg = "Missing command or command is not supported.";
Alinga Yeung
committed
}
else
{
msg = "Only one command can be specified.";
Alinga Yeung
committed
}
Alinga Yeung
committed
throw new UsageException(msg);
Alinga Yeung
committed
}
}
/*
Alinga Yeung
committed
* Parse the command line arguments.
* @param ArgumentMap Command line arguments
Alinga Yeung
committed
* @throws UsageException Error in command line
* @throws CertificateException Fail to get a certificate
Alinga Yeung
committed
*/
Alinga Yeung
committed
protected void parse(final ArgumentMap am, final PrintStream out,
final PrintStream err) throws UsageException, CertificateException
Alinga Yeung
committed
{
if (!am.isSet("h") && !am.isSet("help") && isValid(am))
Alinga Yeung
committed
{
// the following statements are executed only when proceed is true
this.command.setSystemOut(out);
this.command.setSystemErr(err);
Alinga Yeung
committed
}
else
{
isHelpCommand = true;
}
}
public boolean isHelpCommand()
{
return isHelpCommand;
}
Alinga Yeung
committed
/**
* Provide the default command line usage.
Alinga Yeung
committed
*/
public static String getUsage()
Alinga Yeung
committed
{
StringBuilder sb = new StringBuilder();
sb.append("\n");
sb.append("Usage: " + APP_NAME + " <command> [-v|--verbose|-d|--debug] [-h|--help]\n");
Alinga Yeung
committed
sb.append("Where command is\n");
sb.append("--list : List users in the Users tree\n");
sb.append("--list-pending : List users in the UserRequests tree\n");
sb.append("--view=<userid> : Print the entire details of the user\n");
sb.append("--approve=<userid> --dn=<dn> : Approve user with userid=<userid> and set the\n");
sb.append(" : distinguished name to <dn>\n");
sb.append("--reject=<userid> : Delete this user request\n");
Alinga Yeung
committed
sb.append("\n");
sb.append("-v|--verbose : Verbose mode print progress and error messages\n");
sb.append("-d|--debug : Debug mode print all the logging messages\n");
sb.append("-h|--help : Print this message and exit\n");
sb.append("\n");
sb.append("Authentication and authorization:\n");
sb.append(" - An LdapConfig.properties file must exist in directory ~/config/\n");
sb.append(" - The corresponding host entry (devLdap or prodLdap) must exist\n");
sb.append(" in your ~/.dbrc file.");
Alinga Yeung
committed
return sb.toString();
}
}