Search in sources :

Example 1 with AddActionConfig

use of keywhiz.cli.configs.AddActionConfig in project keywhiz by square.

the class AddActionTest method setUp.

@Before
public void setUp() {
    addActionConfig = new AddActionConfig();
    addAction = new AddAction(addActionConfig, keywhizClient, Jackson.newObjectMapper());
}
Also used : AddActionConfig(keywhiz.cli.configs.AddActionConfig) Before(org.junit.Before)

Example 2 with AddActionConfig

use of keywhiz.cli.configs.AddActionConfig 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)

Example 3 with AddActionConfig

use of keywhiz.cli.configs.AddActionConfig in project keywhiz by square.

the class CliMainTest method parsesAddCommandWithoutOwner.

@Test
public void parsesAddCommandWithoutOwner() {
    CliMain.CommandLineParsingContext context = new CliMain.CommandLineParsingContext();
    JCommander commander = context.getCommander();
    commander.parse("add", "secret", "--name", "foo");
    String parsedCommand = commander.getParsedCommand();
    assertEquals("add", parsedCommand);
    AddActionConfig config = (AddActionConfig) context.getCommands().get(parsedCommand);
    assertNull(config.getOwner());
}
Also used : JCommander(com.beust.jcommander.JCommander) AddActionConfig(keywhiz.cli.configs.AddActionConfig) Test(org.junit.Test)

Example 4 with AddActionConfig

use of keywhiz.cli.configs.AddActionConfig in project keywhiz by square.

the class CliMainTest method parsesAddCommandWithOwner.

@Test
public void parsesAddCommandWithOwner() {
    String owner = UUID.randomUUID().toString();
    CliMain.CommandLineParsingContext context = new CliMain.CommandLineParsingContext();
    JCommander commander = context.getCommander();
    commander.parse("add", "secret", "--name", "foo", "--owner", owner);
    String parsedCommand = commander.getParsedCommand();
    assertEquals("add", parsedCommand);
    AddActionConfig config = (AddActionConfig) context.getCommands().get(parsedCommand);
    assertEquals(owner, config.getOwner());
}
Also used : JCommander(com.beust.jcommander.JCommander) AddActionConfig(keywhiz.cli.configs.AddActionConfig) Test(org.junit.Test)

Aggregations

AddActionConfig (keywhiz.cli.configs.AddActionConfig)4 JCommander (com.beust.jcommander.JCommander)2 Test (org.junit.Test)2 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 AssignActionConfig (keywhiz.cli.configs.AssignActionConfig)1 DeleteActionConfig (keywhiz.cli.configs.DeleteActionConfig)1 DescribeActionConfig (keywhiz.cli.configs.DescribeActionConfig)1 ListActionConfig (keywhiz.cli.configs.ListActionConfig)1