src/main/java/org/casbin/Client.java

package org.casbin;

import org.apache.commons.cli.*;
import org.casbin.jcasbin.exception.CasbinEffectorException;
import org.casbin.jcasbin.main.EnforceResult;


public class Client {
    private static void configureOptions(Options options) {
        Option[] cliOptions = {
                addOption("m", "model", true, "the path of the model file"),
                addOption("p", "policy", true, "the path of the policy file"),
                addOption("e", "enforce", true, "enforce"),
                addOption("ex", "enforceEx", true, "enforceEx"),
                addOption("ap", "addPolicy", true, "Add a policy rule to the storage"),
                addOption("rp", "removePolicy", true, "Remove a policy rule from the storage")
        };
        for (Option option : cliOptions) {
            options.addOption(option);
        }
    }
    private static Option addOption(String shortOpt, String longOpt, boolean hasArg, String description) {
        return new Option(shortOpt, longOpt, hasArg, description);
    }

    public static Object run(String[] args) throws ParseException {
        Options options = new Options();
        configureOptions(options);

        CommandLineParser parser = new DefaultParser();
        CommandLine cmd = parser.parse(options, args);

        String model = cmd.getOptionValue("model");
        String policy = cmd.getOptionValue("policy");
        NewEnforcer enforcer = null;
        try {
            enforcer = new NewEnforcer(model, policy);
        }  catch (NullPointerException | CasbinEffectorException | UnsupportedOperationException e) {
            System.out.println("unsupported effect:" + e.getMessage());
            System.exit(0);
        } catch (Exception e) {
            System.out.println(e.getMessage());
            System.exit(0);
        }

        try {
            if(cmd.hasOption("enforce")) {
                String enforceArgs = cmd.getOptionValue("enforce").replace(" ","");
                boolean result = enforcer.enforce(enforceArgs.split(","));
                System.out.println(result ? "Allow" : "Ban");
                return result;
            } else if (cmd.hasOption("enforceEx")) {
                String enforceArgs = cmd.getOptionValue("enforceEx").replace(" ","");
                EnforceResult enforceResult = enforcer.enforceEx(enforceArgs.split(","));
                boolean allow = enforceResult.isAllow();
                if(allow) {
                    System.out.printf("%s Reason: %s", allow, enforceResult.getExplain());
                } else {
                    System.out.println(allow);
                }
                return allow;
            }else if (cmd.hasOption("addPolicy")){
                String policyArgs = cmd.getOptionValue("addPolicy").replace(" ","");
                boolean result = enforcer.addPolicy(policyArgs.split(","));
                System.out.println(result ? "Add Success" : "Add Failed");
                enforcer.savePolicy();
                return result;
            }else if (cmd.hasOption("removePolicy")){
                String policyArgs = cmd.getOptionValue("removePolicy").replace(" ","");
                boolean result = enforcer.removePolicy(policyArgs.split(","));
                System.out.println(result ? "Remove Success" : "Remove Failed");
                enforcer.savePolicy();
                return result;
            }else {
                System.out.println("Command Error");
                return null;
            }
        } catch (Exception e) {
            System.out.println("unsupported effect:" + e.getMessage());
            System.exit(0);
        }
        return null;
    }

    public static void main(String[] args) throws ParseException {
        Client cli = new Client();
        Object run = run(args);
    }
}