blob: 7a7389067f2dca66f266b117b168c917e65a0aab [file] [log] [blame]
/*
* Copyright (c) 2002-2007, Marc Prud'hommeaux. All rights reserved.
*
* This software is distributable under the BSD license. See the terms of the BSD license in the
* documentation provided with this software.
*/
package org.apache.geode.management.internal.cli.shell.jline;
import java.io.BufferedReader;
import java.io.InputStreamReader;
import org.springframework.shell.support.util.OsUtils;
/**
* A buffer that can contain ANSI text.
*
* @author <a href="mailto:mwp1@cornell.edu">Marc Prud'hommeaux</a>
*/
public class ANSIBuffer {
private boolean ansiEnabled = true;
private final StringBuilder ansiBuffer = new StringBuilder();
private final StringBuilder plainBuffer = new StringBuilder();
public ANSIBuffer() {}
public ANSIBuffer(final String str) {
append(str);
}
public void setAnsiEnabled(final boolean ansi) {
ansiEnabled = ansi;
}
public boolean getAnsiEnabled() {
return ansiEnabled;
}
public String getAnsiBuffer() {
return ansiBuffer.toString();
}
public String getPlainBuffer() {
return plainBuffer.toString();
}
public String toString(final boolean ansi) {
return ansi ? getAnsiBuffer() : getPlainBuffer();
}
public String toString() {
return toString(ansiEnabled);
}
public ANSIBuffer append(final String str) {
ansiBuffer.append(str);
plainBuffer.append(str);
return this;
}
public ANSIBuffer attrib(final String str, final int code) {
ansiBuffer.append(ANSICodes.attrib(code)).append(str).append(ANSICodes.attrib(ANSICodes.OFF));
plainBuffer.append(str);
return this;
}
public ANSIBuffer red(final String str) {
return attrib(str, ANSICodes.FG_RED);
}
public ANSIBuffer blue(final String str) {
return attrib(str, ANSICodes.FG_BLUE);
}
public ANSIBuffer green(final String str) {
return attrib(str, ANSICodes.FG_GREEN);
}
public ANSIBuffer black(final String str) {
return attrib(str, ANSICodes.FG_BLACK);
}
public ANSIBuffer yellow(final String str) {
return attrib(str, ANSICodes.FG_YELLOW);
}
public ANSIBuffer magenta(final String str) {
return attrib(str, ANSICodes.FG_MAGENTA);
}
public ANSIBuffer cyan(final String str) {
return attrib(str, ANSICodes.FG_CYAN);
}
public ANSIBuffer bold(final String str) {
return attrib(str, ANSICodes.BOLD);
}
public ANSIBuffer underscore(final String str) {
return attrib(str, ANSICodes.UNDERSCORE);
}
public ANSIBuffer blink(final String str) {
return attrib(str, ANSICodes.BLINK);
}
public ANSIBuffer reverse(final String str) {
return attrib(str, ANSICodes.REVERSE);
}
public static class ANSICodes {
static final int OFF = 0;
static final int BOLD = 1;
static final int UNDERSCORE = 4;
static final int BLINK = 5;
static final int REVERSE = 7;
static final int CONCEALED = 8;
static final int FG_BLACK = 30;
static final int FG_RED = 31;
static final int FG_GREEN = 32;
static final int FG_YELLOW = 33;
static final int FG_BLUE = 34;
static final int FG_MAGENTA = 35;
static final int FG_CYAN = 36;
static final int FG_WHITE = 37;
static final char ESC = 27;
/**
* Constructor is private since this is a utility class.
*/
private ANSICodes() {}
/**
* Sets the screen mode. The mode will be one of the following values:
*
* <pre>
* mode description
* ----------------------------------------
* 0 40 x 148 x 25 monochrome (text)
* 1 40 x 148 x 25 color (text)
* 2 80 x 148 x 25 monochrome (text)
* 3 80 x 148 x 25 color (text)
* 4 320 x 148 x 200 4-color (graphics)
* 5 320 x 148 x 200 monochrome (graphics)
* 6 640 x 148 x 200 monochrome (graphics)
* 7 Enables line wrapping
* 13 320 x 148 x 200 color (graphics)
* 14 640 x 148 x 200 color (16-color graphics)
* 15 640 x 148 x 350 monochrome (2-color graphics)
* 16 640 x 148 x 350 color (16-color graphics)
* 17 640 x 148 x 480 monochrome (2-color graphics)
* 18 640 x 148 x 480 color (16-color graphics)
* 19 320 x 148 x 200 color (256-color graphics)
* </pre>
*/
public static String setmode(final int mode) {
return ESC + "[=" + mode + "h";
}
/**
* Same as setmode () except for mode = 7, which disables line wrapping (useful for writing the
* right-most column without scrolling to the next line).
*/
public static String resetmode(final int mode) {
return ESC + "[=" + mode + "l";
}
/**
* Clears the screen and moves the cursor to the home postition.
*/
public static String clrscr() {
return ESC + "[2J";
}
/**
* Removes all characters from the current cursor position until the end of the line.
*/
public static String clreol() {
return ESC + "[K";
}
/**
* Moves the cursor n positions to the left. If n is greater or equal to the current cursor
* column, the cursor is moved to the first column.
*/
public static String left(final int n) {
return ESC + "[" + n + "D";
}
/**
* Moves the cursor n positions to the right. If n plus the current cursor column is greater
* than the rightmost column, the cursor is moved to the rightmost column.
*/
public static String right(final int n) {
return ESC + "[" + n + "C";
}
/**
* Moves the cursor n rows up without changing the current column. If n is greater than or equal
* to the current row, the cursor is placed in the first row.
*/
public static String up(final int n) {
return ESC + "[" + n + "A";
}
/**
* Moves the cursor n rows down. If n plus the current row is greater than the bottom row, the
* cursor is moved to the bottom row.
*/
public static String down(final int n) {
return ESC + "[" + n + "B";
}
/*
* Moves the cursor to the given row and column. (1,1) represents the upper left corner. The
* lower right corner of a usual DOS screen is (25, 80).
*/
public static String gotoxy(final int row, final int column) {
return ESC + "[" + row + ";" + column + "H";
}
/**
* Saves the current cursor position.
*/
public static String save() {
return ESC + "[s";
}
/**
* Restores the saved cursor position.
*/
public static String restore() {
return ESC + "[u";
}
/**
* Sets the character attribute. It will be one of the following character attributes:
*
* <pre>
* Text attributes
* 0 All attributes off
* 1 Bold on
* 4 Underscore (on monochrome display adapter only)
* 5 Blink on
* 7 Reverse video on
* 8 Concealed on
*
* Foreground colors
* 30 Black
* 31 Red
* 32 Green
* 33 Yellow
* 34 Blue
* 35 Magenta
* 36 Cyan
* 37 White
*
* Background colors
* 40 Black
* 41 Red
* 42 Green
* 43 Yellow
* 44 Blue
* 45 Magenta
* 46 Cyan
* 47 White
* </pre>
*
* The attributes remain in effect until the next attribute command is sent.
*/
public static String attrib(final int attr) {
return ESC + "[" + attr + "m";
}
/**
* Sets the key with the given code to the given value. code must be derived from the following
* table, value must be any semicolon-separated combination of String (enclosed in double
* quotes) and numeric values. For example, to set F1 to the String "Hello F1", followed by a
* CRLF sequence, one can use: ANSI.setkey ("0;59", "\"Hello F1\";13;10"). Heres's the table of
* key values:
*
* <pre>
* Key Code SHIFT+code CTRL+code ALT+code
* --------------------------------------------------------------- F1 0;59 0;84 0;94 0;104 F2
* 0;60 0;85 0;95 0;105 F3 0;61 0;86 0;96 0;106 F4 0;62 0;87 0;97 0;107 F5 0;63 0;88 0;98 0;108
* F6 0;64 0;89 0;99 0;109 F7 0;65 0;90 0;100 0;110 F8 0;66 0;91 0;101 0;111 F9 0;67 0;92 0;102
* 0;112 F10 0;68 0;93 0;103 0;113 F11 0;133 0;135 0;137 0;139 F12 0;134 0;136 0;138 0;140 HOME
* (num keypad) 0;71 55 0;119 -- UP ARROW (num keypad) 0;72 56 (0;141) -- PAGE UP (num keypad)
* 0;73 57 0;132 -- LEFT ARROW (num keypad) 0;75 52 0;115 -- RIGHT ARROW (num keypad) 0;77 54
* 0;116 -- END (num keypad) 0;79 49 0;117 -- DOWN ARROW (num keypad) 0;80 50 (0;145) -- PAGE
* DOWN (num keypad) 0;81 51 0;118 -- INSERT (num keypad) 0;82 48 (0;146) -- DELETE (num keypad)
* 0;83 46 (0;147) -- HOME (224;71) (224;71) (224;119) (224;151) UP ARROW (224;72) (224;72)
* (224;141) (224;152) PAGE UP (224;73) (224;73) (224;132) (224;153) LEFT ARROW (224;75)
* (224;75) (224;115) (224;155) RIGHT ARROW (224;77) (224;77) (224;116) (224;157) END (224;79)
* (224;79) (224;117) (224;159) DOWN ARROW (224;80) (224;80) (224;145) (224;154) PAGE DOWN
* (224;81) (224;81) (224;118) (224;161) INSERT (224;82) (224;82) (224;146) (224;162) DELETE
* (224;83) (224;83) (224;147) (224;163) PRINT SCREEN -- -- 0;114 -- PAUSE/BREAK -- -- 0;0 --
* BACKSPACE 8 8 127 (0) ENTER 13 -- 10 (0 TAB 9 0;15 (0;148) (0;165) NULL 0;3 -- -- -- A 97 65
* 1 0;30 B 98 66 2 0;48 C 99 66 3 0;46 D 100 68 4 0;32 E 101 69 5 0;18 F 102 70 6 0;33 G 103 71
* 7 0;34 H 104 72 8 0;35 I 105 73 9 0;23 J 106 74 10 0;36 K 107 75 11 0;37 L 108 76 12 0;38 M
* 109 77 13 0;50 N 110 78 14 0;49 O 111 79 15 0;24 P 112 80 16 0;25 Q 113 81 17 0;16 R 114 82
* 18 0;19 S 115 83 19 0;31 T 116 84 20 0;20 U 117 85 21 0;22 V 118 86 22 0;47 W 119 87 23 0;17
* X 120 88 24 0;45 Y 121 89 25 0;21 Z 122 90 26 0;44 1 49 33 -- 0;120 2 50 64 0 0;121 3 51 35
* -- 0;122 4 52 36 -- 0;123 5 53 37 -- 0;124 6 54 94 30 0;125 7 55 38 -- 0;126 8 56 42 -- 0;126
* 9 57 40 -- 0;127 0 48 41 -- 0;129 - 45 95 31 0;130 = 61 43 --- 0;131 [ 91 123 27 0;26 ] 93
* 125 29 0;27 92 124 28 0;43 ; 59 58 -- 0;39 ' 39 34 -- 0;40 , 44 60 -- 0;51 . 46 62 -- 0;52 /
* 47 63 -- 0;53 ` 96 126 -- (0;41) ENTER (keypad) 13 -- 10 (0;166) / (keypad) 47 47 (0;142)
* (0;74) * (keypad) 42 (0;144) (0;78) -- - (keypad) 45 45 (0;149) (0;164) + (keypad) 43 43
* (0;150) (0;55) 5 (keypad) (0;76) 53 (0;143) --
*/
public static String setkey(final String code, final String value) {
return ESC + "[" + code + ";" + value + "p";
}
}
public static void main(final String[] args) throws Exception {
// sequence, one can use: ANSI.setkey ("0;59", "\"Hello F1\";13;10").
BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));
System.out.print(ANSICodes.setkey("97", "97;98;99;13") + ANSICodes.attrib(ANSICodes.OFF));
System.out.flush();
String line;
while ((line = reader.readLine()) != null) {
System.out.println("GOT: " + line);
}
}
private static final boolean ROO_BRIGHT_COLORS = Boolean.getBoolean("roo.bright");
private static final boolean SHELL_BRIGHT_COLORS = Boolean.getBoolean("spring.shell.bright");
private static final boolean BRIGHT_COLORS = ROO_BRIGHT_COLORS || SHELL_BRIGHT_COLORS;
public static ANSIBuffer getANSIBuffer() {
final char esc = (char) 27;
return new ANSIBuffer() {
@Override
public ANSIBuffer reverse(final String str) {
if (OsUtils.isWindows()) {
return super.reverse(str).append(ANSICodes.attrib(esc));
}
return super.reverse(str);
}
@Override
public ANSIBuffer attrib(final String str, final int code) {
if (BRIGHT_COLORS && 30 <= code && code <= 37) {
// This is a color code: add a 'bright' code
return append(esc + "[" + code + ";1m").append(str).append(ANSICodes.attrib(0));
}
return super.attrib(str, code);
}
};
}
}