edu.uiuc.ncsa.sas.admin.SASCommands Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sas-server Show documentation
There is a newer version: 5.4.3
package edu.uiuc.ncsa.sas.admin;

import edu.uiuc.ncsa.sas.SASEnvironment;
import edu.uiuc.ncsa.sas.client.ClientKeys;
import edu.uiuc.ncsa.sas.client.SASClient;
import edu.uiuc.ncsa.sas.loader.SASConfigurationLoader;
import edu.uiuc.ncsa.security.core.Identifiable;
import edu.uiuc.ncsa.security.core.Store;
import edu.uiuc.ncsa.security.core.util.Iso8601;
import edu.uiuc.ncsa.security.core.util.MyLoggingFacade;
import edu.uiuc.ncsa.security.core.util.StringUtils;
import edu.uiuc.ncsa.security.storage.cli.StoreCommands;
import edu.uiuc.ncsa.security.util.cli.CLIDriver;
import edu.uiuc.ncsa.security.util.cli.InputLine;
import edu.uiuc.ncsa.security.util.configuration.ConfigUtil;
import edu.uiuc.ncsa.security.util.crypto.KeyUtil;
import edu.uiuc.ncsa.security.util.jwk.JSONWebKeyUtil;
import org.apache.commons.configuration.tree.ConfigurationNode;

import java.io.FileReader;
import java.io.IOException;
import java.security.KeyPair;
import java.util.Vector;

/**
 * Created by Jeff Gaynor

 * on 8/15/22 at  2:48 PM
 */
public class SASCommands extends StoreCommands {
    public SASEnvironment getSatEnvironment() {
        return SASEnvironment;
    }

    public void setSatEnvironment(SASEnvironment SASEnvironment) {
        this.SASEnvironment = SASEnvironment;
    }

    SASEnvironment SASEnvironment;


    protected SASCommands(MyLoggingFacade logger, SASEnvironment SASEnvironment) throws Throwable{
        super(logger == null ? SASEnvironment.getMyLogger() : logger);
        this.SASEnvironment = SASEnvironment;
        setStore(SASEnvironment.getClientStore());
    }

    @Override
    public String getPrompt() {
        return "sas>";
    }

    public SASCommands(MyLoggingFacade logger, Store store) throws Throwable {
        super(logger, store);
    }

    @Override
    public String getName() {
        return "sas";
    }


    @Override
    public void extraUpdates(Identifiable identifiable, int magicNumber) throws IOException {
        ClientKeys keys = (ClientKeys) getMapConverter().getKeys();
        SASClient SASClient = (SASClient) identifiable;
        String newName = getPropertyHelp(keys.name(),"name of this client?", SASClient.getName());
        if (!StringUtils.isTrivial(newName)) {
            SASClient.setName(newName);
        }
        if (SASClient.getPublicKey() == null) {
            if (isOk(getPropertyHelp(keys.publicKey(), "create new keys?", "n"))) {
                KeyPair keyPair = KeyUtil.generateKeyPair();
                say(KeyUtil.toPKCS1PEM(keyPair.getPrivate()));
                SASClient.setPublicKey(keyPair.getPublic());
            } else {
                if (isOk(getInput("specify PKCS 8 file (y/n)?", "n"))) {
                    String f = getInput("enter file path", "");
                    if (StringUtils.isTrivial(f)) {
                        say("no file entered, aborted.");
                    } else {
                        try {
                            SASClient.setPublicKey(KeyUtil.fromX509PEM(new FileReader(f)));
                        } catch (Exception e) {
                            say("uh-oh, that didn't work:" + e.getMessage());
                            if (isVerbose()) {
                                e.printStackTrace();
                            }
                        }
                    }
                }
            }
        }
        if (isOk(getInput("Enter a new public key? (y/n)", "y"))) {
            String oldKey = null;
            if (SASClient.getPublicKey() == null) {
                oldKey = null;
            } else {
                oldKey = KeyUtil.toX509PEM(SASClient.getPublicKey());
            }
            String newKey = multiLineInput(oldKey, keys.publicKey());
            if (newKey == null) {
                // no changes
            } else {
                if (newKey.length() == 1) {
                    // clear it
                    SASClient.setPublicKey(null);
                } else {
                    if (-1 == newKey.indexOf("{")) {
                        SASClient.setPublicKey(KeyUtil.fromX509PEM(newKey));
                    } else {
                        // JWK
                        try {
                            SASClient.setPublicKey(JSONWebKeyUtil.getJsonWebKey(newKey).publicKey);
                        } catch (Throwable t) {
                            say("Could not process public key:" + t.getMessage());
                        }
                    }
                }
            }
        }

/*  At this point we don't use this field.r
        if(isOK(getInput("Enter a configuration? (y/n)", ""))){

        }
*/


    }


    @Override
    protected String format(Identifiable identifiable) {
        SASClient client = (SASClient) identifiable;
        String rc = "";

        String name = (client.getName() == null ? "no name" : client.getName());
        name = StringUtils.pad2(name, 20);
        rc = rc + name + " | " + StringUtils.pad2(identifiable.getIdentifierString(), 60);
        rc = rc + " | created on " + Iso8601.date2String(client.getCreationTS());
        return rc;
    }


    public static String CONFIG_NAME_FLAG = "-name";
    public static String CONFIG_FILE_FLAG = "-cfg";
    public static String CONFIG_TAG_NAME = "sas";

    public static void main(String[] args) throws Throwable {
        Vector vector = new Vector<>();
        vector.add("dummy"); // Dummy zero-th arg.
        for (String arg : args) {
            vector.add(arg);
        }
        InputLine inputLine = new InputLine(vector); // now we have a bunch of utilities for this

        String cfgname = inputLine.hasArg(CONFIG_NAME_FLAG) ? inputLine.getNextArgFor(CONFIG_NAME_FLAG) : "default";
//      Old style -- single inheritance
        ConfigurationNode node = ConfigUtil.findConfiguration(
                inputLine.getNextArgFor(CONFIG_FILE_FLAG),
                cfgname, CONFIG_TAG_NAME);

        // New style -- multi-inheritance.
        //     ConfigurationNode node = ConfigUtil.findMultiNode(inputLine.getNextArgFor(CONFIG_FILE_FLAG), cfgname, CONFIG_TAG_NAME );
        SASConfigurationLoader loader = new SASConfigurationLoader(node);

        SASEnvironment SASEnvironment1 = loader.load();
        SASCommands SASCommands = new SASCommands(null, SASEnvironment1);
        CLIDriver driver = new CLIDriver(SASCommands);

        driver.start();
    }
}