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

Introduction

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

Source

/*
 * jPOS Project [http://jpos.org]
 * Copyright (C) 2000-2019 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.bouncycastle.util.encoders.Base64;
import org.jpos.ee.*;
import org.jpos.q2.CLICommand;
import org.jpos.q2.CLIContext;

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

@SuppressWarnings("unused")
public class ADDCONSUMER implements CLICommand {
    @Override
    public void exec(CLIContext cli, String[] args) throws Exception {
        CommandLineParser parser = new DefaultParser();
        Options options = new Options();
        options.addOption("u", "user", true, "User's nick name");
        options.addOption("U", "uuid", true, "Optional UUID");
        options.addOption("S", "secret", true, "Optional secret");
        options.addOption("r", "role", true, "User role");
        CommandLine line = parser.parse(options, shl(args));

        if (args.length < 2) {
            showHelp(cli, args, options);
            return;
        }
        if (!line.hasOption('u')) {
            cli.println("--user 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(line.getOptionValue('u'));
            if (u == null) {
                cli.println("User does not exists");
                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;
                }
            }

            String id = line.hasOption('U') ? line.getOptionValue('U') : UUID.randomUUID().toString();
            ConsumerManager cmgr = new ConsumerManager(db);
            if (cmgr.getById(id) != null) {
                cli.println("Consumer already exists");
                return;
            }
            String secret = line.hasOption('S') ? line.getOptionValue('S')
                    : Base64.toBase64String(UUID.randomUUID().toString().getBytes());
            Consumer c = new Consumer();
            c.setId(id);
            c.setUser(u);
            c.setHash(HashVersion.ONE.hash(Long.toString(c.getUser().getId(), 16), secret,
                    HashVersion.ONE.getSalt()));
            db.session().save(c);
            if (rr != null)
                Collections.addAll(c.getRoles(), rr);
            c.setActive(true);
            db.commit();
            cli.println("Consumer '" + c.getId() + "' secret '" + secret + "'");
        } 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] + "[options]", options);
        helpFormatter.printHelp(pw, HelpFormatter.DEFAULT_WIDTH, args[0], null, options, 0,
                HelpFormatter.DEFAULT_DESC_PAD, null, false);
    }
}