net.e6tech.elements.common.util.Terminal Maven / Gradle / Ivy
/*
Copyright 2015-2019 Futeh Kao
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package net.e6tech.elements.common.util;
import net.e6tech.elements.common.logging.Logger;
import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
/**
* Created by futeh on 12/22/15.
*/
public class Terminal implements Closeable {
Emulator emulator;
@SuppressWarnings("squid:S106") // yes, we have to use System.out
public Terminal() {
Console console = System.console();
if (console != null) {
emulator = new NativeConsole(console);
} else {
emulator = new StreamConsole(new BufferedReader(new InputStreamReader(System.in)),
new PrintWriter(new OutputStreamWriter(System.out))) {
@Override
public void close() {
// do noting, we cannot close System.out!
}
};
}
}
public Terminal(InputStream in, OutputStream out) {
emulator = new StreamConsole(new BufferedReader(new InputStreamReader(in)),
new PrintWriter(new OutputStreamWriter(out)));
}
public Terminal(ServerSocket serverSocket) throws IOException {
final Socket socket = serverSocket.accept();
try {
emulator = new StreamConsole(new BufferedReader(new InputStreamReader(socket.getInputStream())),
new PrintWriter(new OutputStreamWriter(socket.getOutputStream()))) {
@Override
public void close() {
try {
super.close();
} finally {
try {
socket.close();
} catch (IOException e) {
Logger.suppress(e);
}
}
}
};
} catch (IOException ex) {
throw new SystemException(ex);
}
}
public String readLine(String text) {
return emulator.readLine(text);
}
public char[] readPassword(String text) {
return emulator.readPassword(text);
}
public void println(String text) {
emulator.println(text);
}
public void print(String text) {
emulator.print(text);
}
@Override
public void close() {
emulator.close();
}
interface Emulator {
String readLine(String text);
char[] readPassword(String text);
void println(String text);
void print(String text);
void close();
}
class StreamConsole implements Emulator {
private BufferedReader input;
private PrintWriter output;
StreamConsole(BufferedReader in, PrintWriter out) {
this.input = in;
this.output = out;
}
@Override
public String readLine(String text) {
output.print(text);
output.flush();
try {
return input.readLine().trim();
} catch (IOException e) {
throw new SystemException(e);
} catch (NullPointerException e) {
Logger.suppress(e);
return null;
}
}
@Override
@SuppressWarnings("squid:S1168")
public char[] readPassword(String text) {
output.print(text);
output.flush();
String str = null;
try {
str = input.readLine().trim();
} catch (IOException e) {
throw new SystemException(e);
} catch (NullPointerException e) {
Logger.suppress(e);
return null;
}
return str.toCharArray();
}
@Override
public void println(String text) {
output.println(text);
output.flush();
}
@Override
public void print(String text) {
output.print(text);
output.flush();
}
@Override
public void close() {
IOException exception = null;
try {
input.close();
} catch (IOException e) {
exception = e;
}
output.close();
if (exception != null)
throw new SystemException(exception);
}
}
class NativeConsole implements Emulator {
Console console;
NativeConsole(Console console) {
this.console = console;
}
@Override
public String readLine(String text) {
print(text);
try {
return console.readLine().trim();
} catch (NullPointerException e) {
Logger.suppress(e);
return "";
}
}
@Override
public char[] readPassword(String text) {
try {
char[] ret = console.readPassword(text);
if (ret == null)
return new char[0];
return ret;
} catch (NullPointerException e) {
Logger.suppress(e);
return new char[0];
}
}
@Override
public void println(String text) {
console.printf(text);
console.printf("\n");
}
@Override
public void print(String text) {
console.printf(text);
}
@Override
public void close() {
// Console uses System.out and we cannot close it.
}
}
@SuppressWarnings("squid:S106") // yes, we have to use System.out
public static void main(String ... args) {
String user = null;
char[] password = null;
try (Terminal term = new Terminal()) {
user = term.readLine("Username:");
password = term.readPassword("Password:");
}
System.out.println(user + ":" + new String(password));
}
}