ADDUSER.java

/*
 * jPOS Project [http://jpos.org]
 * Copyright (C) 2000-2017 jPOS Software SRL
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package org.jpos.q2.cli;

import org.apache.commons.cli.*;
import org.jpos.ee.*;
import org.jpos.q2.CLICommand;
import org.jpos.q2.CLIContext;

import java.io.PrintWriter;
import java.util.Collections;

@SuppressWarnings("unused")
public class ADDUSER implements CLICommand {
    @Override
    public void exec(CLIContext cli, String[] args) throws Exception {
        CommandLineParser parser = new DefaultParser();
        Options options = new Options();
        options.addOption ("n", "name", true, "User's full name");
        options.addOption ("p", "password", true, "The password");
        options.addOption ("r", "role", true, "User role");
        options.addOption ("h", "help", false, "This help");
        CommandLine line = parser.parse(options, shl(shl(args)));

        if (args.length < 2 || line.hasOption('h')) {
            showHelp (cli, args, options);
            return;
        }
        if (!line.hasOption('n')) {
            cli.println ("--nick parameter is required. See --help for help");
            return;
        }
        try (DB db = new DB()) {
            db.open();
            db.beginTransaction();
            UserManager mgr = new UserManager(db);
            User u = mgr.getUserByNick(args[1], true);
            if (u != null) {
                cli.println ("User already exists" + (u.isDeleted() ? " (soft-deleted)" : ""));
                return;
            }
            RoleManager rm = new RoleManager(db);
            Role[] rr = null;
            if (line.hasOption('r')) {
                String[] roles = line.getOptionValues('r');
                rr = new Role[roles.length];
                int i=0;
                for (String s : roles) {
                    Role r = rm.getRoleByName(s);
                    if (r == null) {
                        cli.println ("Role '" + s + "' does not exist - aborting");
                        return;
                    }
                    rr[i++] = r;
                }
            }
            User user = new User();
            user.setNick (args[1]);
            user.setName (line.getOptionValue('N'));
            db.session().save(user);
            if (line.hasOption('p')) {
                mgr.setPassword(user, line.getOptionValue('p'));
            }
            if (rr != null)
                Collections.addAll(user.getRoles(), rr);
            user.setActive(true);
            db.commit();
            cli.println ("User '" + user.getNickAndId() + "' created");
        } catch (Exception e) {
            cli.println (e.getMessage());
        }
    }

    private String[] shl (String[] args) {
        switch (args.length) {
            case 0:
                return args;
            case 1:
                return new String[0];
        }
        String[] shl = new String[args.length-1];
        System.arraycopy(args,1, shl, 0, args.length-1);
        return shl;
    }

    private void showHelp (CLIContext cli, String args[], Options options) {
        PrintWriter pw = new PrintWriter (cli.getOutputStream());
        HelpFormatter helpFormatter = new HelpFormatter ();
        helpFormatter.printHelp (args[0] + "<nick>", options);
        helpFormatter.printHelp(pw, HelpFormatter.DEFAULT_WIDTH, args[0], null, options, 0, HelpFormatter.DEFAULT_DESC_PAD, null,false);
    }
}