|
19 | 19 | import org.elasticsearch.core.CheckedFunction; |
20 | 20 | import org.elasticsearch.env.Environment; |
21 | 21 |
|
22 | | -import java.io.CharArrayWriter; |
23 | | -import java.io.Closeable; |
24 | 22 | import java.io.IOException; |
25 | | -import java.io.Reader; |
26 | 23 | import java.util.Arrays; |
27 | 24 | import java.util.List; |
28 | 25 |
|
@@ -53,42 +50,28 @@ protected void executeCommand(Terminal terminal, OptionSet options, Environment |
53 | 50 |
|
54 | 51 | final KeyStoreWrapper keyStore = getKeyStore(); |
55 | 52 |
|
56 | | - final Closeable closeable; |
57 | | - final CheckedFunction<String, char[], IOException> valueSupplier; |
58 | | - if (options.has(stdinOption)) { |
59 | | - final Reader stdinReader = terminal.getReader(); |
60 | | - valueSupplier = s -> { |
61 | | - try (CharArrayWriter writer = new CharArrayWriter()) { |
62 | | - int c; |
63 | | - while ((c = stdinReader.read()) != -1) { |
64 | | - if ((char) c == '\r' || (char) c == '\n') { |
65 | | - break; |
66 | | - } |
67 | | - writer.write((char) c); |
68 | | - } |
69 | | - return writer.toCharArray(); |
70 | | - } |
71 | | - }; |
72 | | - closeable = stdinReader; |
73 | | - } else { |
74 | | - valueSupplier = s -> terminal.readSecret("Enter value for " + s + ": "); |
75 | | - closeable = () -> {}; |
76 | | - } |
| 53 | + final CheckedFunction<String, char[], IOException> valueSupplier = s -> { |
| 54 | + final String prompt; |
| 55 | + if (options.has(stdinOption)) { |
| 56 | + prompt = ""; |
| 57 | + } else { |
| 58 | + prompt = "Enter value for " + s + ": "; |
| 59 | + } |
| 60 | + return terminal.readSecret(prompt); |
| 61 | + }; |
77 | 62 |
|
78 | | - try (closeable) { |
79 | | - for (final String setting : settings) { |
80 | | - if (keyStore.getSettingNames().contains(setting) && options.has(forceOption) == false) { |
81 | | - if (terminal.promptYesNo("Setting " + setting + " already exists. Overwrite?", false) == false) { |
82 | | - terminal.println("Exiting without modifying keystore."); |
83 | | - return; |
84 | | - } |
| 63 | + for (final String setting : settings) { |
| 64 | + if (keyStore.getSettingNames().contains(setting) && options.has(forceOption) == false) { |
| 65 | + if (terminal.promptYesNo("Setting " + setting + " already exists. Overwrite?", false) == false) { |
| 66 | + terminal.println("Exiting without modifying keystore."); |
| 67 | + return; |
85 | 68 | } |
| 69 | + } |
86 | 70 |
|
87 | | - try { |
88 | | - keyStore.setString(setting, valueSupplier.apply(setting)); |
89 | | - } catch (final IllegalArgumentException e) { |
90 | | - throw new UserException(ExitCodes.DATA_ERROR, e.getMessage()); |
91 | | - } |
| 71 | + try { |
| 72 | + keyStore.setString(setting, valueSupplier.apply(setting)); |
| 73 | + } catch (final IllegalArgumentException e) { |
| 74 | + throw new UserException(ExitCodes.DATA_ERROR, e.getMessage()); |
92 | 75 | } |
93 | 76 | } |
94 | 77 |
|
|
0 commit comments