org.jpos.q2.cli.ADDUSER.java Source code

Introduction

Here is the source code for org.jpos.q2.cli.ADDUSER.java

Source

/*
 * jPOS Project [http://jpos.org]
 * Copyright (C) 2000-2018 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");
        CommandLine line = parser.parse(options, shl(shl(args)));

        if (args.length < 2) {
            showHelp(cli, args, options);
            return;
        }
        if (!line.hasOption('n')) {
            cli.println("--name 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]", options);
        helpFormatter.printHelp(pw, HelpFormatter.DEFAULT_WIDTH, args[0], null, options, 0,
                HelpFormatter.DEFAULT_DESC_PAD, null, false);
    }
}