Search in sources :

Example 1 with UpdateAction

use of keywhiz.cli.commands.UpdateAction in project keywhiz by square.

the class CommandExecutor method executeCommand.

public void executeCommand() throws IOException {
    if (command == null) {
        if (config.version) {
            System.out.println("Version: " + APP_VERSION);
        } else {
            System.err.println("Must specify a command.");
            parentCommander.usage();
        }
        return;
    }
    HttpUrl url;
    if (Strings.isNullOrEmpty(config.url)) {
        url = HttpUrl.parse("https://localhost:4444");
        System.out.println("Server URL not specified (--url flag), assuming " + url);
    } else {
        url = HttpUrl.parse(config.url);
        if (url == null) {
            System.err.print("Invalid URL " + config.url);
            return;
        }
    }
    KeywhizClient client;
    OkHttpClient httpClient;
    String user = config.getUser().orElse(USER_NAME.value());
    Path cookiePath = cookieDir.resolve(format(".keywhiz.%s.cookie", user));
    try {
        List<HttpCookie> cookieList = ClientUtils.loadCookies(cookiePath);
        httpClient = ClientUtils.sslOkHttpClient(config.getDevTrustStore(), cookieList);
        client = new KeywhizClient(mapper, httpClient, url);
        // Try a simple get request to determine whether or not the cookies are still valid
        if (!client.isLoggedIn()) {
            throw new UnauthorizedException();
        }
    } catch (IOException e) {
        // Either could not find the cookie file, or the cookies were expired -- must login manually.
        httpClient = ClientUtils.sslOkHttpClient(config.getDevTrustStore(), ImmutableList.of());
        client = new KeywhizClient(mapper, httpClient, url);
        char[] password = ClientUtils.readPassword(user);
        client.login(user, password);
        Arrays.fill(password, '\0');
    }
    // Save/update the cookies if we logged in successfully
    ClientUtils.saveCookies(cookiePath);
    Printing printing = new Printing(client);
    Command cmd = Command.valueOf(command.toUpperCase().trim());
    switch(cmd) {
        case LIST:
            new ListAction((ListActionConfig) commands.get(command), client, printing).run();
            break;
        case DESCRIBE:
            new DescribeAction((DescribeActionConfig) commands.get(command), client, printing).run();
            break;
        case ADD:
            new AddAction((AddActionConfig) commands.get(command), client, mapper).run();
            break;
        case UPDATE:
            new UpdateAction((UpdateActionConfig) commands.get(command), client, mapper).run();
            break;
        case DELETE:
            new DeleteAction((DeleteActionConfig) commands.get(command), client).run();
            break;
        case ASSIGN:
            new AssignAction((AssignActionConfig) commands.get(command), client).run();
            break;
        case UNASSIGN:
            new UnassignAction((UnassignActionConfig) commands.get(command), client).run();
            break;
        case VERSIONS:
            new ListVersionsAction((ListVersionsActionConfig) commands.get(command), client, printing).run();
            break;
        case ROLLBACK:
            new RollbackAction((RollbackActionConfig) commands.get(command), client).run();
            break;
        case RENAME:
            new RenameAction((RenameActionConfig) commands.get(command), client).run();
        case LOGIN:
            // User is already logged in at this point
            break;
        default:
            commander.usage();
    }
}
Also used : OkHttpClient(okhttp3.OkHttpClient) AssignActionConfig(keywhiz.cli.configs.AssignActionConfig) UpdateActionConfig(keywhiz.cli.configs.UpdateActionConfig) ListVersionsAction(keywhiz.cli.commands.ListVersionsAction) AddAction(keywhiz.cli.commands.AddAction) UnassignActionConfig(keywhiz.cli.configs.UnassignActionConfig) ListActionConfig(keywhiz.cli.configs.ListActionConfig) UnauthorizedException(keywhiz.client.KeywhizClient.UnauthorizedException) UnassignAction(keywhiz.cli.commands.UnassignAction) AddActionConfig(keywhiz.cli.configs.AddActionConfig) RenameAction(keywhiz.cli.commands.RenameAction) RenameActionConfig(keywhiz.cli.configs.RenameActionConfig) Path(java.nio.file.Path) KeywhizClient(keywhiz.client.KeywhizClient) AssignAction(keywhiz.cli.commands.AssignAction) UpdateAction(keywhiz.cli.commands.UpdateAction) DeleteActionConfig(keywhiz.cli.configs.DeleteActionConfig) IOException(java.io.IOException) RollbackAction(keywhiz.cli.commands.RollbackAction) HttpUrl(okhttp3.HttpUrl) ListVersionsActionConfig(keywhiz.cli.configs.ListVersionsActionConfig) RollbackActionConfig(keywhiz.cli.configs.RollbackActionConfig) DescribeActionConfig(keywhiz.cli.configs.DescribeActionConfig) ListAction(keywhiz.cli.commands.ListAction) DeleteAction(keywhiz.cli.commands.DeleteAction) HttpCookie(java.net.HttpCookie) DescribeAction(keywhiz.cli.commands.DescribeAction)

Aggregations

IOException (java.io.IOException)1 HttpCookie (java.net.HttpCookie)1 Path (java.nio.file.Path)1 AddAction (keywhiz.cli.commands.AddAction)1 AssignAction (keywhiz.cli.commands.AssignAction)1 DeleteAction (keywhiz.cli.commands.DeleteAction)1 DescribeAction (keywhiz.cli.commands.DescribeAction)1 ListAction (keywhiz.cli.commands.ListAction)1 ListVersionsAction (keywhiz.cli.commands.ListVersionsAction)1 RenameAction (keywhiz.cli.commands.RenameAction)1 RollbackAction (keywhiz.cli.commands.RollbackAction)1 UnassignAction (keywhiz.cli.commands.UnassignAction)1 UpdateAction (keywhiz.cli.commands.UpdateAction)1 AddActionConfig (keywhiz.cli.configs.AddActionConfig)1 AssignActionConfig (keywhiz.cli.configs.AssignActionConfig)1 DeleteActionConfig (keywhiz.cli.configs.DeleteActionConfig)1 DescribeActionConfig (keywhiz.cli.configs.DescribeActionConfig)1 ListActionConfig (keywhiz.cli.configs.ListActionConfig)1 ListVersionsActionConfig (keywhiz.cli.configs.ListVersionsActionConfig)1 RenameActionConfig (keywhiz.cli.configs.RenameActionConfig)1