term: extend tty interface with more features
This commit is contained in:
461
term/sys/darwin/tty.c
Normal file
461
term/sys/darwin/tty.c
Normal file
@@ -0,0 +1,461 @@
|
||||
#include "../../../line-ed/tty.h"
|
||||
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <sys/ioctl.h>
|
||||
#include <termios.h>
|
||||
#include <unistd.h>
|
||||
|
||||
#define ANSI_BOLD_ON "1"
|
||||
#define ANSI_BOLD_OFF "22"
|
||||
#define ANSI_ITALIC_ON "3"
|
||||
#define ANSI_ITALIC_OFF "23"
|
||||
#define ANSI_UNDERLINE_ON "4"
|
||||
#define ANSI_UNDERLINE_OFF "24"
|
||||
|
||||
#define ANSI_DEFAULTCOLOUR_FG "39"
|
||||
#define ANSI_DEFAULTCOLOUR_BG "49"
|
||||
|
||||
#define ANSI_256COLOUR_FG "38;5"
|
||||
#define ANSI_256COLOUR_BG "48;5"
|
||||
|
||||
#define ANSI_TRUECOLOUR_FG "38;2"
|
||||
#define ANSI_TRUECOLOUR_BG "48;2"
|
||||
|
||||
enum tty_flags {
|
||||
TTY_INIT = 0x01u,
|
||||
TTY_INTERACTIVE = 0x02u,
|
||||
};
|
||||
|
||||
struct s_tty {
|
||||
FILE *t_in, *t_out;
|
||||
enum tty_flags t_flags;
|
||||
struct termios t_ios_raw, t_ios_default;
|
||||
struct s_tty_vmode t_vmode;
|
||||
unsigned char t_mcount;
|
||||
};
|
||||
|
||||
static struct s_tty tty = {};
|
||||
|
||||
static const char *ansi_colour16_fg[] = {
|
||||
[TTY_COLOUR16_BLACK] = "30",
|
||||
[TTY_COLOUR16_RED] = "31",
|
||||
[TTY_COLOUR16_GREEN] = "32",
|
||||
[TTY_COLOUR16_YELLOW] = "33",
|
||||
[TTY_COLOUR16_BLUE] = "34",
|
||||
[TTY_COLOUR16_MAGENTA] = "35",
|
||||
[TTY_COLOUR16_CYAN] = "36",
|
||||
[TTY_COLOUR16_WHITE] = "37",
|
||||
[TTY_COLOUR16_GREY] = "90",
|
||||
[TTY_COLOUR16_BRIGHT_RED] = "91",
|
||||
[TTY_COLOUR16_BRIGHT_GREEN] = "92",
|
||||
[TTY_COLOUR16_BRIGHT_YELLOW] = "93",
|
||||
[TTY_COLOUR16_BRIGHT_BLUE] = "94",
|
||||
[TTY_COLOUR16_BRIGHT_MAGENTA] = "95",
|
||||
[TTY_COLOUR16_BRIGHT_CYAN] = "96",
|
||||
[TTY_COLOUR16_BRIGHT_WHITE] = "97",
|
||||
};
|
||||
|
||||
static const char *ansi_colour16_bg[] = {
|
||||
[TTY_COLOUR16_BLACK] = "40",
|
||||
[TTY_COLOUR16_RED] = "41",
|
||||
[TTY_COLOUR16_GREEN] = "42",
|
||||
[TTY_COLOUR16_YELLOW] = "43",
|
||||
[TTY_COLOUR16_BLUE] = "44",
|
||||
[TTY_COLOUR16_MAGENTA] = "45",
|
||||
[TTY_COLOUR16_CYAN] = "46",
|
||||
[TTY_COLOUR16_WHITE] = "47",
|
||||
[TTY_COLOUR16_GREY] = "100",
|
||||
[TTY_COLOUR16_BRIGHT_RED] = "101",
|
||||
[TTY_COLOUR16_BRIGHT_GREEN] = "102",
|
||||
[TTY_COLOUR16_BRIGHT_YELLOW] = "103",
|
||||
[TTY_COLOUR16_BRIGHT_BLUE] = "104",
|
||||
[TTY_COLOUR16_BRIGHT_MAGENTA] = "105",
|
||||
[TTY_COLOUR16_BRIGHT_CYAN] = "106",
|
||||
[TTY_COLOUR16_BRIGHT_WHITE] = "107",
|
||||
};
|
||||
|
||||
static void init_tty(struct s_tty *tty, FILE *in, FILE *out)
|
||||
{
|
||||
tty->t_in = in;
|
||||
tty->t_out = out;
|
||||
|
||||
int fd = fileno(in);
|
||||
|
||||
if (isatty(fd)) {
|
||||
tty->t_flags |= TTY_INTERACTIVE;
|
||||
}
|
||||
|
||||
tcgetattr(fd, &tty->t_ios_default);
|
||||
memcpy(&tty->t_ios_raw, &tty->t_ios_default, sizeof tty->t_ios_raw);
|
||||
|
||||
tty->t_ios_raw.c_iflag &= ~(ICRNL | IXON);
|
||||
tty->t_ios_raw.c_oflag &= ~(OPOST);
|
||||
tty->t_ios_raw.c_lflag &= ~(ECHO | ICANON | IEXTEN);
|
||||
|
||||
tty->t_flags |= TTY_INIT;
|
||||
}
|
||||
|
||||
struct s_tty *s_get_tty(void)
|
||||
{
|
||||
if (!(tty.t_flags & TTY_INIT)) {
|
||||
init_tty(&tty, stdin, stdout);
|
||||
}
|
||||
|
||||
return &tty;
|
||||
}
|
||||
|
||||
bool s_tty_is_interactive(const struct s_tty *tty)
|
||||
{
|
||||
return (tty->t_flags & TTY_INTERACTIVE) == TTY_INTERACTIVE;
|
||||
}
|
||||
|
||||
void s_tty_set_raw(struct s_tty *tty)
|
||||
{
|
||||
tcsetattr(STDIN_FILENO, TCSAFLUSH, &tty->t_ios_raw);
|
||||
}
|
||||
|
||||
void s_tty_set_canon(struct s_tty *tty)
|
||||
{
|
||||
tcsetattr(STDIN_FILENO, TCSAFLUSH, &tty->t_ios_default);
|
||||
}
|
||||
|
||||
void s_tty_reset_vmode(struct s_tty *tty)
|
||||
{
|
||||
if (tty->t_vmode.v_fg.c_mode == TTY_COLOUR_NONE
|
||||
&& tty->t_vmode.v_bg.c_mode == TTY_COLOUR_NONE
|
||||
&& tty->t_vmode.v_attrib == TTY_ATTRIB_NORMAL) {
|
||||
return;
|
||||
}
|
||||
|
||||
fprintf(tty->t_out, "\033[0;39;49m");
|
||||
|
||||
memset(&tty->t_vmode, 0x0, sizeof tty->t_vmode);
|
||||
|
||||
tty->t_vmode.v_fg.c_mode = TTY_COLOUR_NONE;
|
||||
tty->t_vmode.v_bg.c_mode = TTY_COLOUR_NONE;
|
||||
tty->t_vmode.v_attrib = TTY_ATTRIB_NORMAL;
|
||||
}
|
||||
|
||||
static int compare_colour(
|
||||
const struct s_tty_colour *a, const struct s_tty_colour *b)
|
||||
{
|
||||
if (a->c_mode != b->c_mode) {
|
||||
return -1;
|
||||
}
|
||||
|
||||
switch (a->c_mode) {
|
||||
case TTY_COLOUR_16:
|
||||
if (a->c_16.value != b->c_16.value) {
|
||||
return -1;
|
||||
}
|
||||
break;
|
||||
case TTY_COLOUR_256:
|
||||
if (a->c_256.value != b->c_256.value) {
|
||||
return -1;
|
||||
}
|
||||
break;
|
||||
case TTY_COLOUR_TRUE:
|
||||
if (a->c_true.r != b->c_true.r) {
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (a->c_true.g != b->c_true.g) {
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (a->c_true.b != b->c_true.b) {
|
||||
return -1;
|
||||
}
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int compare_vmode(const struct s_tty_vmode *a, const struct s_tty_vmode *b)
|
||||
{
|
||||
if (a->v_attrib != b->v_attrib) {
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (compare_colour(&a->v_fg, &b->v_fg) != 0) {
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (compare_colour(&a->v_bg, &b->v_bg) != 0) {
|
||||
return -1;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static void put_ansi_attrib(struct s_tty *tty, const char *s)
|
||||
{
|
||||
if (tty->t_mcount > 0) {
|
||||
fputs(";", tty->t_out);
|
||||
}
|
||||
|
||||
fputs(s, tty->t_out);
|
||||
tty->t_mcount++;
|
||||
}
|
||||
|
||||
static void set_attrib(
|
||||
struct s_tty *tty, enum s_tty_attrib old, enum s_tty_attrib new)
|
||||
{
|
||||
if (old == new) {
|
||||
return;
|
||||
}
|
||||
|
||||
/* Bold ON */
|
||||
if (!(old & TTY_ATTRIB_BOLD) && new &TTY_ATTRIB_BOLD) {
|
||||
put_ansi_attrib(tty, ANSI_BOLD_ON);
|
||||
}
|
||||
|
||||
/* Bold OFF */
|
||||
if (old & TTY_ATTRIB_BOLD && !(new &TTY_ATTRIB_BOLD)) {
|
||||
put_ansi_attrib(tty, ANSI_BOLD_OFF);
|
||||
}
|
||||
|
||||
/* Underline ON */
|
||||
if (!(old & TTY_ATTRIB_UNDERLINE) && new &TTY_ATTRIB_UNDERLINE) {
|
||||
put_ansi_attrib(tty, ANSI_UNDERLINE_ON);
|
||||
}
|
||||
|
||||
/* Underline OFF */
|
||||
if (old & TTY_ATTRIB_UNDERLINE && !(new &TTY_ATTRIB_UNDERLINE)) {
|
||||
put_ansi_attrib(tty, ANSI_UNDERLINE_OFF);
|
||||
}
|
||||
|
||||
/* Italic ON */
|
||||
if (!(old & TTY_ATTRIB_ITALIC) && new &TTY_ATTRIB_ITALIC) {
|
||||
put_ansi_attrib(tty, ANSI_ITALIC_ON);
|
||||
}
|
||||
|
||||
/* Italic OFF */
|
||||
if (old & TTY_ATTRIB_ITALIC && !(new &TTY_ATTRIB_ITALIC)) {
|
||||
put_ansi_attrib(tty, ANSI_ITALIC_OFF);
|
||||
}
|
||||
}
|
||||
|
||||
static void set_fg(
|
||||
struct s_tty *tty, const struct s_tty_colour *old,
|
||||
const struct s_tty_colour *new)
|
||||
{
|
||||
if (compare_colour(old, new) == 0) {
|
||||
return;
|
||||
}
|
||||
|
||||
char buf[8];
|
||||
|
||||
switch (new->c_mode) {
|
||||
case TTY_COLOUR_NONE:
|
||||
put_ansi_attrib(tty, ANSI_DEFAULTCOLOUR_FG);
|
||||
break;
|
||||
case TTY_COLOUR_16:
|
||||
put_ansi_attrib(tty, ansi_colour16_fg[new->c_16.value]);
|
||||
break;
|
||||
case TTY_COLOUR_256:
|
||||
put_ansi_attrib(tty, ANSI_256COLOUR_FG);
|
||||
snprintf(buf, sizeof buf, "%u", new->c_256.value);
|
||||
put_ansi_attrib(tty, buf);
|
||||
break;
|
||||
case TTY_COLOUR_TRUE:
|
||||
put_ansi_attrib(tty, ANSI_TRUECOLOUR_FG);
|
||||
snprintf(buf, sizeof buf, "%u", new->c_true.r);
|
||||
put_ansi_attrib(tty, buf);
|
||||
snprintf(buf, sizeof buf, "%u", new->c_true.g);
|
||||
put_ansi_attrib(tty, buf);
|
||||
snprintf(buf, sizeof buf, "%u", new->c_true.b);
|
||||
put_ansi_attrib(tty, buf);
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
static void set_bg(
|
||||
struct s_tty *tty, const struct s_tty_colour *old,
|
||||
const struct s_tty_colour *new)
|
||||
{
|
||||
if (compare_colour(old, new) == 0) {
|
||||
return;
|
||||
}
|
||||
|
||||
char buf[8];
|
||||
|
||||
switch (new->c_mode) {
|
||||
case TTY_COLOUR_NONE:
|
||||
put_ansi_attrib(tty, ANSI_DEFAULTCOLOUR_BG);
|
||||
break;
|
||||
case TTY_COLOUR_16:
|
||||
put_ansi_attrib(tty, ansi_colour16_bg[new->c_16.value]);
|
||||
break;
|
||||
case TTY_COLOUR_256:
|
||||
put_ansi_attrib(tty, ANSI_256COLOUR_BG);
|
||||
snprintf(buf, sizeof buf, "%u", new->c_256.value);
|
||||
put_ansi_attrib(tty, buf);
|
||||
break;
|
||||
case TTY_COLOUR_TRUE:
|
||||
put_ansi_attrib(tty, ANSI_TRUECOLOUR_BG);
|
||||
snprintf(buf, sizeof buf, "%u", new->c_true.r);
|
||||
put_ansi_attrib(tty, buf);
|
||||
snprintf(buf, sizeof buf, "%u", new->c_true.g);
|
||||
put_ansi_attrib(tty, buf);
|
||||
snprintf(buf, sizeof buf, "%u", new->c_true.b);
|
||||
put_ansi_attrib(tty, buf);
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
void s_tty_set_vmode(struct s_tty *tty, const struct s_tty_vmode *vmode)
|
||||
{
|
||||
if (compare_vmode(&tty->t_vmode, vmode) == 0) {
|
||||
return;
|
||||
}
|
||||
|
||||
tty->t_mcount = 0;
|
||||
|
||||
fprintf(tty->t_out, "\033[");
|
||||
|
||||
set_attrib(tty, tty->t_vmode.v_attrib, vmode->v_attrib);
|
||||
set_fg(tty, &tty->t_vmode.v_fg, &vmode->v_fg);
|
||||
set_bg(tty, &tty->t_vmode.v_bg, &vmode->v_bg);
|
||||
|
||||
fprintf(tty->t_out, "m");
|
||||
|
||||
memcpy(&tty->t_vmode, vmode, sizeof *vmode);
|
||||
}
|
||||
|
||||
s_keycode s_tty_read_key(struct s_tty *tty)
|
||||
{
|
||||
char c;
|
||||
int v;
|
||||
int fd = fileno(tty->t_in);
|
||||
|
||||
while (1) {
|
||||
v = read(fd, &c, 1);
|
||||
if (v < 1) {
|
||||
return S_KEY_EOF;
|
||||
}
|
||||
|
||||
if (c == '\r' || c == '\n') {
|
||||
return S_KEY_RETURN;
|
||||
}
|
||||
|
||||
if (c == '\b' || c == 0x7F) {
|
||||
return S_KEY_BACKSPACE;
|
||||
}
|
||||
|
||||
if (c >= 1 && c <= 26) {
|
||||
return S_TTY_CTRL_KEY(c + 'a' - 1);
|
||||
}
|
||||
|
||||
if (c != 0x1b) {
|
||||
return c;
|
||||
}
|
||||
|
||||
v = read(fd, &c, 1);
|
||||
if (v < 1) {
|
||||
return S_KEY_EOF;
|
||||
}
|
||||
|
||||
if (c != '[') {
|
||||
continue;
|
||||
}
|
||||
|
||||
v = read(fd, &c, 1);
|
||||
if (v < 1) {
|
||||
return S_KEY_EOF;
|
||||
}
|
||||
|
||||
switch (c) {
|
||||
case 'A':
|
||||
return S_KEY_ARROW_UP;
|
||||
case 'B':
|
||||
return S_KEY_ARROW_DOWN;
|
||||
case 'C':
|
||||
return S_KEY_ARROW_RIGHT;
|
||||
case 'D':
|
||||
return S_KEY_ARROW_LEFT;
|
||||
default:
|
||||
continue;
|
||||
}
|
||||
}
|
||||
|
||||
return c;
|
||||
}
|
||||
|
||||
void s_tty_move_cursor_x(struct s_tty *tty, enum s_tty_position_base base, int pos)
|
||||
{
|
||||
if (base == TTY_POS_CURSOR && pos < 0 && pos >= -4) {
|
||||
for (int i = 0; i > pos; i--) {
|
||||
fputc('\b', tty->t_out);
|
||||
}
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
if (base == TTY_POS_START) {
|
||||
if (pos == 0) {
|
||||
fputs("\033[G", tty->t_out);
|
||||
} else {
|
||||
fprintf(tty->t_out, "\033[%dG", pos + 1);
|
||||
}
|
||||
} else {
|
||||
if (pos > 1) {
|
||||
fprintf(tty->t_out, "\033[%dC", pos);
|
||||
} else if (pos == 1) {
|
||||
fputs("\033[C", tty->t_out);
|
||||
} else if (pos == -1) {
|
||||
fputs("\033[D", tty->t_out);
|
||||
} else if (pos < -1) {
|
||||
fprintf(tty->t_out, "\033[%dD", -pos);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void s_tty_move_cursor_y(struct s_tty *tty, enum s_tty_position_base base, int pos)
|
||||
{
|
||||
if (base == TTY_POS_START) {
|
||||
/* we don't need this functionality right now */
|
||||
abort();
|
||||
}
|
||||
|
||||
if (pos > 1) {
|
||||
fprintf(tty->t_out, "\033[%dB", pos);
|
||||
} else if (pos == 1) {
|
||||
fputs("\033[B", tty->t_out);
|
||||
} else if (pos == -1) {
|
||||
fputs("\033[A", tty->t_out);
|
||||
} else if (pos < -1) {
|
||||
fprintf(tty->t_out, "\033[%dA", -pos);
|
||||
}
|
||||
}
|
||||
|
||||
void s_tty_clear(struct s_tty *tty, enum s_tty_clear_mode mode)
|
||||
{
|
||||
const char *arg;
|
||||
if (mode & TTY_CLEAR_ALL) {
|
||||
arg = "2";
|
||||
} else if (mode & TTY_CLEAR_TO_CURSOR) {
|
||||
arg = "1";
|
||||
} else if (mode & TTY_CLEAR_FROM_CURSOR) {
|
||||
arg = "";
|
||||
} else {
|
||||
abort();
|
||||
}
|
||||
|
||||
if (mode & TTY_CLEAR_SCREEN) {
|
||||
fprintf(tty->t_out, "\033[%sJ", arg);
|
||||
} else if (mode & TTY_CLEAR_LINE) {
|
||||
fprintf(tty->t_out, "\033[%sK", arg);
|
||||
} else {
|
||||
abort();
|
||||
}
|
||||
}
|
||||
461
term/sys/linux/tty.c
Normal file
461
term/sys/linux/tty.c
Normal file
@@ -0,0 +1,461 @@
|
||||
#include "../../../line-ed/tty.h"
|
||||
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <sys/ioctl.h>
|
||||
#include <termios.h>
|
||||
#include <unistd.h>
|
||||
|
||||
#define ANSI_BOLD_ON "1"
|
||||
#define ANSI_BOLD_OFF "22"
|
||||
#define ANSI_ITALIC_ON "3"
|
||||
#define ANSI_ITALIC_OFF "23"
|
||||
#define ANSI_UNDERLINE_ON "4"
|
||||
#define ANSI_UNDERLINE_OFF "24"
|
||||
|
||||
#define ANSI_DEFAULTCOLOUR_FG "39"
|
||||
#define ANSI_DEFAULTCOLOUR_BG "49"
|
||||
|
||||
#define ANSI_256COLOUR_FG "38;5"
|
||||
#define ANSI_256COLOUR_BG "48;5"
|
||||
|
||||
#define ANSI_TRUECOLOUR_FG "38;2"
|
||||
#define ANSI_TRUECOLOUR_BG "48;2"
|
||||
|
||||
enum tty_flags {
|
||||
TTY_INIT = 0x01u,
|
||||
TTY_INTERACTIVE = 0x02u,
|
||||
};
|
||||
|
||||
struct s_tty {
|
||||
FILE *t_in, *t_out;
|
||||
enum tty_flags t_flags;
|
||||
struct termios t_ios_raw, t_ios_default;
|
||||
struct s_tty_vmode t_vmode;
|
||||
unsigned char t_mcount;
|
||||
};
|
||||
|
||||
static struct s_tty tty = {};
|
||||
|
||||
static const char *ansi_colour16_fg[] = {
|
||||
[TTY_COLOUR16_BLACK] = "30",
|
||||
[TTY_COLOUR16_RED] = "31",
|
||||
[TTY_COLOUR16_GREEN] = "32",
|
||||
[TTY_COLOUR16_YELLOW] = "33",
|
||||
[TTY_COLOUR16_BLUE] = "34",
|
||||
[TTY_COLOUR16_MAGENTA] = "35",
|
||||
[TTY_COLOUR16_CYAN] = "36",
|
||||
[TTY_COLOUR16_WHITE] = "37",
|
||||
[TTY_COLOUR16_GREY] = "90",
|
||||
[TTY_COLOUR16_BRIGHT_RED] = "91",
|
||||
[TTY_COLOUR16_BRIGHT_GREEN] = "92",
|
||||
[TTY_COLOUR16_BRIGHT_YELLOW] = "93",
|
||||
[TTY_COLOUR16_BRIGHT_BLUE] = "94",
|
||||
[TTY_COLOUR16_BRIGHT_MAGENTA] = "95",
|
||||
[TTY_COLOUR16_BRIGHT_CYAN] = "96",
|
||||
[TTY_COLOUR16_BRIGHT_WHITE] = "97",
|
||||
};
|
||||
|
||||
static const char *ansi_colour16_bg[] = {
|
||||
[TTY_COLOUR16_BLACK] = "40",
|
||||
[TTY_COLOUR16_RED] = "41",
|
||||
[TTY_COLOUR16_GREEN] = "42",
|
||||
[TTY_COLOUR16_YELLOW] = "43",
|
||||
[TTY_COLOUR16_BLUE] = "44",
|
||||
[TTY_COLOUR16_MAGENTA] = "45",
|
||||
[TTY_COLOUR16_CYAN] = "46",
|
||||
[TTY_COLOUR16_WHITE] = "47",
|
||||
[TTY_COLOUR16_GREY] = "100",
|
||||
[TTY_COLOUR16_BRIGHT_RED] = "101",
|
||||
[TTY_COLOUR16_BRIGHT_GREEN] = "102",
|
||||
[TTY_COLOUR16_BRIGHT_YELLOW] = "103",
|
||||
[TTY_COLOUR16_BRIGHT_BLUE] = "104",
|
||||
[TTY_COLOUR16_BRIGHT_MAGENTA] = "105",
|
||||
[TTY_COLOUR16_BRIGHT_CYAN] = "106",
|
||||
[TTY_COLOUR16_BRIGHT_WHITE] = "107",
|
||||
};
|
||||
|
||||
static void init_tty(struct s_tty *tty, FILE *in, FILE *out)
|
||||
{
|
||||
tty->t_in = in;
|
||||
tty->t_out = out;
|
||||
|
||||
int fd = fileno(in);
|
||||
|
||||
if (isatty(fd)) {
|
||||
tty->t_flags |= TTY_INTERACTIVE;
|
||||
}
|
||||
|
||||
tcgetattr(fd, &tty->t_ios_default);
|
||||
memcpy(&tty->t_ios_raw, &tty->t_ios_default, sizeof tty->t_ios_raw);
|
||||
|
||||
tty->t_ios_raw.c_iflag &= ~(ICRNL | IXON);
|
||||
tty->t_ios_raw.c_oflag &= ~(OPOST);
|
||||
tty->t_ios_raw.c_lflag &= ~(ECHO | ICANON | IEXTEN);
|
||||
|
||||
tty->t_flags |= TTY_INIT;
|
||||
}
|
||||
|
||||
struct s_tty *s_get_tty(void)
|
||||
{
|
||||
if (!(tty.t_flags & TTY_INIT)) {
|
||||
init_tty(&tty, stdin, stdout);
|
||||
}
|
||||
|
||||
return &tty;
|
||||
}
|
||||
|
||||
bool s_tty_is_interactive(const struct s_tty *tty)
|
||||
{
|
||||
return (tty->t_flags & TTY_INTERACTIVE) == TTY_INTERACTIVE;
|
||||
}
|
||||
|
||||
void s_tty_set_raw(struct s_tty *tty)
|
||||
{
|
||||
tcsetattr(STDIN_FILENO, TCSAFLUSH, &tty->t_ios_raw);
|
||||
}
|
||||
|
||||
void s_tty_set_canon(struct s_tty *tty)
|
||||
{
|
||||
tcsetattr(STDIN_FILENO, TCSAFLUSH, &tty->t_ios_default);
|
||||
}
|
||||
|
||||
void s_tty_reset_vmode(struct s_tty *tty)
|
||||
{
|
||||
if (tty->t_vmode.v_fg.c_mode == TTY_COLOUR_NONE
|
||||
&& tty->t_vmode.v_bg.c_mode == TTY_COLOUR_NONE
|
||||
&& tty->t_vmode.v_attrib == TTY_ATTRIB_NORMAL) {
|
||||
return;
|
||||
}
|
||||
|
||||
fprintf(tty->t_out, "\033[0;39;49m");
|
||||
|
||||
memset(&tty->t_vmode, 0x0, sizeof tty->t_vmode);
|
||||
|
||||
tty->t_vmode.v_fg.c_mode = TTY_COLOUR_NONE;
|
||||
tty->t_vmode.v_bg.c_mode = TTY_COLOUR_NONE;
|
||||
tty->t_vmode.v_attrib = TTY_ATTRIB_NORMAL;
|
||||
}
|
||||
|
||||
static int compare_colour(
|
||||
const struct s_tty_colour *a, const struct s_tty_colour *b)
|
||||
{
|
||||
if (a->c_mode != b->c_mode) {
|
||||
return -1;
|
||||
}
|
||||
|
||||
switch (a->c_mode) {
|
||||
case TTY_COLOUR_16:
|
||||
if (a->c_16.value != b->c_16.value) {
|
||||
return -1;
|
||||
}
|
||||
break;
|
||||
case TTY_COLOUR_256:
|
||||
if (a->c_256.value != b->c_256.value) {
|
||||
return -1;
|
||||
}
|
||||
break;
|
||||
case TTY_COLOUR_TRUE:
|
||||
if (a->c_true.r != b->c_true.r) {
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (a->c_true.g != b->c_true.g) {
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (a->c_true.b != b->c_true.b) {
|
||||
return -1;
|
||||
}
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int compare_vmode(const struct s_tty_vmode *a, const struct s_tty_vmode *b)
|
||||
{
|
||||
if (a->v_attrib != b->v_attrib) {
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (compare_colour(&a->v_fg, &b->v_fg) != 0) {
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (compare_colour(&a->v_bg, &b->v_bg) != 0) {
|
||||
return -1;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static void put_ansi_attrib(struct s_tty *tty, const char *s)
|
||||
{
|
||||
if (tty->t_mcount > 0) {
|
||||
fputs(";", tty->t_out);
|
||||
}
|
||||
|
||||
fputs(s, tty->t_out);
|
||||
tty->t_mcount++;
|
||||
}
|
||||
|
||||
static void set_attrib(
|
||||
struct s_tty *tty, enum s_tty_attrib old, enum s_tty_attrib new)
|
||||
{
|
||||
if (old == new) {
|
||||
return;
|
||||
}
|
||||
|
||||
/* Bold ON */
|
||||
if (!(old & TTY_ATTRIB_BOLD) && new &TTY_ATTRIB_BOLD) {
|
||||
put_ansi_attrib(tty, ANSI_BOLD_ON);
|
||||
}
|
||||
|
||||
/* Bold OFF */
|
||||
if (old & TTY_ATTRIB_BOLD && !(new &TTY_ATTRIB_BOLD)) {
|
||||
put_ansi_attrib(tty, ANSI_BOLD_OFF);
|
||||
}
|
||||
|
||||
/* Underline ON */
|
||||
if (!(old & TTY_ATTRIB_UNDERLINE) && new &TTY_ATTRIB_UNDERLINE) {
|
||||
put_ansi_attrib(tty, ANSI_UNDERLINE_ON);
|
||||
}
|
||||
|
||||
/* Underline OFF */
|
||||
if (old & TTY_ATTRIB_UNDERLINE && !(new &TTY_ATTRIB_UNDERLINE)) {
|
||||
put_ansi_attrib(tty, ANSI_UNDERLINE_OFF);
|
||||
}
|
||||
|
||||
/* Italic ON */
|
||||
if (!(old & TTY_ATTRIB_ITALIC) && new &TTY_ATTRIB_ITALIC) {
|
||||
put_ansi_attrib(tty, ANSI_ITALIC_ON);
|
||||
}
|
||||
|
||||
/* Italic OFF */
|
||||
if (old & TTY_ATTRIB_ITALIC && !(new &TTY_ATTRIB_ITALIC)) {
|
||||
put_ansi_attrib(tty, ANSI_ITALIC_OFF);
|
||||
}
|
||||
}
|
||||
|
||||
static void set_fg(
|
||||
struct s_tty *tty, const struct s_tty_colour *old,
|
||||
const struct s_tty_colour *new)
|
||||
{
|
||||
if (compare_colour(old, new) == 0) {
|
||||
return;
|
||||
}
|
||||
|
||||
char buf[8];
|
||||
|
||||
switch (new->c_mode) {
|
||||
case TTY_COLOUR_NONE:
|
||||
put_ansi_attrib(tty, ANSI_DEFAULTCOLOUR_FG);
|
||||
break;
|
||||
case TTY_COLOUR_16:
|
||||
put_ansi_attrib(tty, ansi_colour16_fg[new->c_16.value]);
|
||||
break;
|
||||
case TTY_COLOUR_256:
|
||||
put_ansi_attrib(tty, ANSI_256COLOUR_FG);
|
||||
snprintf(buf, sizeof buf, "%u", new->c_256.value);
|
||||
put_ansi_attrib(tty, buf);
|
||||
break;
|
||||
case TTY_COLOUR_TRUE:
|
||||
put_ansi_attrib(tty, ANSI_TRUECOLOUR_FG);
|
||||
snprintf(buf, sizeof buf, "%u", new->c_true.r);
|
||||
put_ansi_attrib(tty, buf);
|
||||
snprintf(buf, sizeof buf, "%u", new->c_true.g);
|
||||
put_ansi_attrib(tty, buf);
|
||||
snprintf(buf, sizeof buf, "%u", new->c_true.b);
|
||||
put_ansi_attrib(tty, buf);
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
static void set_bg(
|
||||
struct s_tty *tty, const struct s_tty_colour *old,
|
||||
const struct s_tty_colour *new)
|
||||
{
|
||||
if (compare_colour(old, new) == 0) {
|
||||
return;
|
||||
}
|
||||
|
||||
char buf[8];
|
||||
|
||||
switch (new->c_mode) {
|
||||
case TTY_COLOUR_NONE:
|
||||
put_ansi_attrib(tty, ANSI_DEFAULTCOLOUR_BG);
|
||||
break;
|
||||
case TTY_COLOUR_16:
|
||||
put_ansi_attrib(tty, ansi_colour16_bg[new->c_16.value]);
|
||||
break;
|
||||
case TTY_COLOUR_256:
|
||||
put_ansi_attrib(tty, ANSI_256COLOUR_BG);
|
||||
snprintf(buf, sizeof buf, "%u", new->c_256.value);
|
||||
put_ansi_attrib(tty, buf);
|
||||
break;
|
||||
case TTY_COLOUR_TRUE:
|
||||
put_ansi_attrib(tty, ANSI_TRUECOLOUR_BG);
|
||||
snprintf(buf, sizeof buf, "%u", new->c_true.r);
|
||||
put_ansi_attrib(tty, buf);
|
||||
snprintf(buf, sizeof buf, "%u", new->c_true.g);
|
||||
put_ansi_attrib(tty, buf);
|
||||
snprintf(buf, sizeof buf, "%u", new->c_true.b);
|
||||
put_ansi_attrib(tty, buf);
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
void s_tty_set_vmode(struct s_tty *tty, const struct s_tty_vmode *vmode)
|
||||
{
|
||||
if (compare_vmode(&tty->t_vmode, vmode) == 0) {
|
||||
return;
|
||||
}
|
||||
|
||||
tty->t_mcount = 0;
|
||||
|
||||
fprintf(tty->t_out, "\033[");
|
||||
|
||||
set_attrib(tty, tty->t_vmode.v_attrib, vmode->v_attrib);
|
||||
set_fg(tty, &tty->t_vmode.v_fg, &vmode->v_fg);
|
||||
set_bg(tty, &tty->t_vmode.v_bg, &vmode->v_bg);
|
||||
|
||||
fprintf(tty->t_out, "m");
|
||||
|
||||
memcpy(&tty->t_vmode, vmode, sizeof *vmode);
|
||||
}
|
||||
|
||||
s_keycode s_tty_read_key(struct s_tty *tty)
|
||||
{
|
||||
char c;
|
||||
int v;
|
||||
int fd = fileno(tty->t_in);
|
||||
|
||||
while (1) {
|
||||
v = read(fd, &c, 1);
|
||||
if (v < 1) {
|
||||
return S_KEY_EOF;
|
||||
}
|
||||
|
||||
if (c == '\r' || c == '\n') {
|
||||
return S_KEY_RETURN;
|
||||
}
|
||||
|
||||
if (c == '\b' || c == 0x7F) {
|
||||
return S_KEY_BACKSPACE;
|
||||
}
|
||||
|
||||
if (c >= 1 && c <= 26) {
|
||||
return S_TTY_CTRL_KEY(c + 'a' - 1);
|
||||
}
|
||||
|
||||
if (c != 0x1b) {
|
||||
return c;
|
||||
}
|
||||
|
||||
v = read(fd, &c, 1);
|
||||
if (v < 1) {
|
||||
return S_KEY_EOF;
|
||||
}
|
||||
|
||||
if (c != '[') {
|
||||
continue;
|
||||
}
|
||||
|
||||
v = read(fd, &c, 1);
|
||||
if (v < 1) {
|
||||
return S_KEY_EOF;
|
||||
}
|
||||
|
||||
switch (c) {
|
||||
case 'A':
|
||||
return S_KEY_ARROW_UP;
|
||||
case 'B':
|
||||
return S_KEY_ARROW_DOWN;
|
||||
case 'C':
|
||||
return S_KEY_ARROW_RIGHT;
|
||||
case 'D':
|
||||
return S_KEY_ARROW_LEFT;
|
||||
default:
|
||||
continue;
|
||||
}
|
||||
}
|
||||
|
||||
return c;
|
||||
}
|
||||
|
||||
void s_tty_move_cursor_x(struct s_tty *tty, enum s_tty_position_base base, int pos)
|
||||
{
|
||||
if (base == TTY_POS_CURSOR && pos < 0 && pos >= -4) {
|
||||
for (int i = 0; i > pos; i--) {
|
||||
fputc('\b', tty->t_out);
|
||||
}
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
if (base == TTY_POS_START) {
|
||||
if (pos == 0) {
|
||||
fputs("\033[G", tty->t_out);
|
||||
} else {
|
||||
fprintf(tty->t_out, "\033[%dG", pos + 1);
|
||||
}
|
||||
} else {
|
||||
if (pos > 1) {
|
||||
fprintf(tty->t_out, "\033[%dC", pos);
|
||||
} else if (pos == 1) {
|
||||
fputs("\033[C", tty->t_out);
|
||||
} else if (pos == -1) {
|
||||
fputs("\033[D", tty->t_out);
|
||||
} else if (pos < -1) {
|
||||
fprintf(tty->t_out, "\033[%dD", -pos);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void s_tty_move_cursor_y(struct s_tty *tty, enum s_tty_position_base base, int pos)
|
||||
{
|
||||
if (base == TTY_POS_START) {
|
||||
/* we don't need this functionality right now */
|
||||
abort();
|
||||
}
|
||||
|
||||
if (pos > 1) {
|
||||
fprintf(tty->t_out, "\033[%dB", pos);
|
||||
} else if (pos == 1) {
|
||||
fputs("\033[B", tty->t_out);
|
||||
} else if (pos == -1) {
|
||||
fputs("\033[A", tty->t_out);
|
||||
} else if (pos < -1) {
|
||||
fprintf(tty->t_out, "\033[%dA", -pos);
|
||||
}
|
||||
}
|
||||
|
||||
void s_tty_clear(struct s_tty *tty, enum s_tty_clear_mode mode)
|
||||
{
|
||||
const char *arg;
|
||||
if (mode & TTY_CLEAR_ALL) {
|
||||
arg = "2";
|
||||
} else if (mode & TTY_CLEAR_TO_CURSOR) {
|
||||
arg = "1";
|
||||
} else if (mode & TTY_CLEAR_FROM_CURSOR) {
|
||||
arg = "";
|
||||
} else {
|
||||
abort();
|
||||
}
|
||||
|
||||
if (mode & TTY_CLEAR_SCREEN) {
|
||||
fprintf(tty->t_out, "\033[%sJ", arg);
|
||||
} else if (mode & TTY_CLEAR_LINE) {
|
||||
fprintf(tty->t_out, "\033[%sK", arg);
|
||||
} else {
|
||||
abort();
|
||||
}
|
||||
}
|
||||
570
term/sys/windows/tty.c
Normal file
570
term/sys/windows/tty.c
Normal file
@@ -0,0 +1,570 @@
|
||||
#include <blue/term/tty.h>
|
||||
#include "../../tty.h"
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
#include <Windows.h>
|
||||
#include <io.h>
|
||||
#include <stdbool.h>
|
||||
|
||||
#define ANSI_BOLD_ON "1"
|
||||
#define ANSI_BOLD_OFF "22"
|
||||
#define ANSI_ITALIC_ON "3"
|
||||
#define ANSI_ITALIC_OFF "23"
|
||||
#define ANSI_UNDERLINE_ON "4"
|
||||
#define ANSI_UNDERLINE_OFF "24"
|
||||
|
||||
#define ANSI_DEFAULTCOLOUR_FG "39"
|
||||
#define ANSI_DEFAULTCOLOUR_BG "49"
|
||||
|
||||
#define ANSI_256COLOUR_FG "38;5"
|
||||
#define ANSI_256COLOUR_BG "48;5"
|
||||
|
||||
#define ANSI_TRUECOLOUR_FG "38;2"
|
||||
#define ANSI_TRUECOLOUR_BG "48;2"
|
||||
|
||||
enum tty_flags {
|
||||
B_TTY_INIT = 0x01u,
|
||||
B_TTY_INTERACTIVE = 0x02u,
|
||||
};
|
||||
|
||||
struct b_tty {
|
||||
HANDLE t_in, t_out;
|
||||
DWORD t_canon_mode_in, t_canon_mode_out;
|
||||
enum s_key_code t_repeat_key;
|
||||
unsigned int t_repeat_count;
|
||||
enum tty_flags t_flags;
|
||||
struct b_tty_vmode t_vmode;
|
||||
struct tty_format_buf t_format_buf;
|
||||
};
|
||||
|
||||
static struct b_tty std = {0};
|
||||
static struct b_tty err = {0};
|
||||
|
||||
static WORD ansi_colour16_fg[] = {
|
||||
[B_TTY_COLOUR16_BLACK] = 0,
|
||||
[B_TTY_COLOUR16_RED] = FOREGROUND_RED,
|
||||
[B_TTY_COLOUR16_GREEN] = FOREGROUND_GREEN,
|
||||
[B_TTY_COLOUR16_YELLOW] = FOREGROUND_RED | FOREGROUND_GREEN,
|
||||
[B_TTY_COLOUR16_BLUE] = FOREGROUND_BLUE,
|
||||
[B_TTY_COLOUR16_MAGENTA] = FOREGROUND_RED | FOREGROUND_BLUE,
|
||||
[B_TTY_COLOUR16_CYAN] = FOREGROUND_GREEN | FOREGROUND_BLUE,
|
||||
[B_TTY_COLOUR16_WHITE] = FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE,
|
||||
[B_TTY_COLOUR16_BRIGHT_BLACK] = FOREGROUND_INTENSITY,
|
||||
[B_TTY_COLOUR16_BRIGHT_RED] = FOREGROUND_RED | FOREGROUND_INTENSITY,
|
||||
[B_TTY_COLOUR16_BRIGHT_GREEN] = FOREGROUND_GREEN | FOREGROUND_INTENSITY,
|
||||
[B_TTY_COLOUR16_BRIGHT_YELLOW] = FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_INTENSITY,
|
||||
[B_TTY_COLOUR16_BRIGHT_BLUE] = FOREGROUND_BLUE | FOREGROUND_INTENSITY,
|
||||
[B_TTY_COLOUR16_BRIGHT_MAGENTA] = FOREGROUND_RED | FOREGROUND_BLUE | FOREGROUND_INTENSITY,
|
||||
[B_TTY_COLOUR16_BRIGHT_CYAN] = FOREGROUND_GREEN | FOREGROUND_BLUE | FOREGROUND_INTENSITY,
|
||||
[B_TTY_COLOUR16_BRIGHT_WHITE] = FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE | FOREGROUND_INTENSITY,
|
||||
};
|
||||
|
||||
static WORD ansi_colour16_bg[] = {
|
||||
[B_TTY_COLOUR16_BLACK] = 0,
|
||||
[B_TTY_COLOUR16_RED] = BACKGROUND_RED,
|
||||
[B_TTY_COLOUR16_GREEN] = BACKGROUND_GREEN,
|
||||
[B_TTY_COLOUR16_YELLOW] = BACKGROUND_RED | BACKGROUND_GREEN,
|
||||
[B_TTY_COLOUR16_BLUE] = BACKGROUND_BLUE,
|
||||
[B_TTY_COLOUR16_MAGENTA] = BACKGROUND_RED | BACKGROUND_BLUE,
|
||||
[B_TTY_COLOUR16_CYAN] = BACKGROUND_GREEN | BACKGROUND_BLUE,
|
||||
[B_TTY_COLOUR16_WHITE] = BACKGROUND_RED | BACKGROUND_GREEN | BACKGROUND_BLUE,
|
||||
[B_TTY_COLOUR16_BRIGHT_BLACK] = BACKGROUND_INTENSITY,
|
||||
[B_TTY_COLOUR16_BRIGHT_RED] = BACKGROUND_RED | BACKGROUND_INTENSITY,
|
||||
[B_TTY_COLOUR16_BRIGHT_GREEN] = BACKGROUND_GREEN | BACKGROUND_INTENSITY,
|
||||
[B_TTY_COLOUR16_BRIGHT_YELLOW] = BACKGROUND_RED | BACKGROUND_GREEN | BACKGROUND_INTENSITY,
|
||||
[B_TTY_COLOUR16_BRIGHT_BLUE] = BACKGROUND_BLUE | BACKGROUND_INTENSITY,
|
||||
[B_TTY_COLOUR16_BRIGHT_MAGENTA] = BACKGROUND_RED | BACKGROUND_BLUE | BACKGROUND_INTENSITY,
|
||||
[B_TTY_COLOUR16_BRIGHT_CYAN] = BACKGROUND_GREEN | BACKGROUND_BLUE | BACKGROUND_INTENSITY,
|
||||
[B_TTY_COLOUR16_BRIGHT_WHITE] = BACKGROUND_RED | BACKGROUND_GREEN | BACKGROUND_BLUE | BACKGROUND_INTENSITY,
|
||||
};
|
||||
|
||||
static void init_tty(struct b_tty *tty, FILE *in, FILE *out)
|
||||
{
|
||||
HANDLE in_handle = INVALID_HANDLE_VALUE;
|
||||
HANDLE out_handle = INVALID_HANDLE_VALUE;
|
||||
|
||||
if (in) {
|
||||
in_handle = (HANDLE)_get_osfhandle(fileno(in));
|
||||
}
|
||||
|
||||
if (out) {
|
||||
out_handle = (HANDLE)_get_osfhandle(fileno(out));
|
||||
}
|
||||
|
||||
tty->t_in = in_handle;
|
||||
tty->t_out = out_handle;
|
||||
tty->t_flags |= B_TTY_INIT;
|
||||
|
||||
CONSOLE_SCREEN_BUFFER_INFO csbi = {0};
|
||||
DWORD mode = 0;
|
||||
|
||||
if (in && GetConsoleScreenBufferInfo(in_handle, &csbi)) {
|
||||
GetConsoleMode(in_handle, &mode);
|
||||
tty->t_canon_mode_in = mode;
|
||||
tty->t_flags |= B_TTY_INTERACTIVE;
|
||||
}
|
||||
|
||||
if (out && GetConsoleScreenBufferInfo(out_handle, &csbi)) {
|
||||
GetConsoleMode(out_handle, &mode);
|
||||
tty->t_canon_mode_out = mode;
|
||||
tty->t_flags |= B_TTY_INTERACTIVE;
|
||||
}
|
||||
}
|
||||
|
||||
struct b_tty *z__b_tty_get_std(void)
|
||||
{
|
||||
if (!(std.t_flags & B_TTY_INIT)) {
|
||||
init_tty(&std, stdin, stdout);
|
||||
}
|
||||
|
||||
return &std;
|
||||
}
|
||||
|
||||
struct b_tty *z__b_tty_get_err(void)
|
||||
{
|
||||
if (!(err.t_flags & B_TTY_INIT)) {
|
||||
init_tty(&err, NULL, stderr);
|
||||
}
|
||||
|
||||
return &err;
|
||||
}
|
||||
|
||||
struct tty_format_buf *z__b_tty_get_format_buf(struct b_tty *tty)
|
||||
{
|
||||
return &tty->t_format_buf;
|
||||
}
|
||||
|
||||
void z__b_tty_putc(struct b_tty* tty, char c)
|
||||
{
|
||||
DWORD x;
|
||||
WriteConsoleA(tty->t_out, &c, 1, &x, NULL);
|
||||
}
|
||||
|
||||
bool b_tty_is_interactive(const struct b_tty *tty)
|
||||
{
|
||||
return (tty->t_flags & B_TTY_INTERACTIVE) == B_TTY_INTERACTIVE;
|
||||
}
|
||||
|
||||
static void tty_set_raw(struct b_tty *tty)
|
||||
{
|
||||
DWORD mode = tty->t_canon_mode_in;
|
||||
mode &= ~(ENABLE_ECHO_INPUT | ENABLE_INSERT_MODE | ENABLE_LINE_INPUT | ENABLE_QUICK_EDIT_MODE | ENABLE_WINDOW_INPUT);
|
||||
mode |= (ENABLE_PROCESSED_INPUT);
|
||||
SetConsoleMode(tty->t_in, mode);
|
||||
|
||||
mode = tty->t_canon_mode_out;
|
||||
//mode &= ~(ENABLE_PROCESSED_OUTPUT);
|
||||
SetConsoleMode(tty->t_out, mode);
|
||||
}
|
||||
|
||||
static void tty_set_canon(struct b_tty *tty)
|
||||
{
|
||||
SetConsoleMode(tty->t_in, tty->t_canon_mode_in);
|
||||
SetConsoleMode(tty->t_out, tty->t_canon_mode_out);
|
||||
}
|
||||
|
||||
void b_tty_set_mode(struct b_tty* tty, enum b_tty_mode mode)
|
||||
{
|
||||
switch (mode) {
|
||||
case B_TTY_CANONICAL:
|
||||
tty_set_canon(tty);
|
||||
break;
|
||||
case B_TTY_RAW:
|
||||
tty_set_raw(tty);
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
void b_tty_reset_vmode(struct b_tty *tty)
|
||||
{
|
||||
if (tty->t_vmode.v_fg.c_mode == B_TTY_COLOUR_NONE
|
||||
&& tty->t_vmode.v_bg.c_mode == B_TTY_COLOUR_NONE
|
||||
&& tty->t_vmode.v_attrib == B_TTY_ATTRIB_NORMAL) {
|
||||
return;
|
||||
}
|
||||
|
||||
WORD attrib = FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE;
|
||||
SetConsoleTextAttribute(tty->t_out, attrib);
|
||||
|
||||
memset(&tty->t_vmode, 0x0, sizeof tty->t_vmode);
|
||||
|
||||
tty->t_vmode.v_fg.c_mode = B_TTY_COLOUR_NONE;
|
||||
tty->t_vmode.v_bg.c_mode = B_TTY_COLOUR_NONE;
|
||||
tty->t_vmode.v_attrib = B_TTY_ATTRIB_NORMAL;
|
||||
}
|
||||
|
||||
static int compare_colour(
|
||||
const struct b_tty_colour *a, const struct b_tty_colour *b)
|
||||
{
|
||||
if (a->c_mode != b->c_mode) {
|
||||
return -1;
|
||||
}
|
||||
|
||||
switch (a->c_mode) {
|
||||
case B_TTY_COLOUR_16:
|
||||
if (a->c_16.value != b->c_16.value) {
|
||||
return -1;
|
||||
}
|
||||
break;
|
||||
case B_TTY_COLOUR_256:
|
||||
if (a->c_256.value != b->c_256.value) {
|
||||
return -1;
|
||||
}
|
||||
break;
|
||||
case B_TTY_COLOUR_TRUE:
|
||||
if (a->c_true.r != b->c_true.r) {
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (a->c_true.g != b->c_true.g) {
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (a->c_true.b != b->c_true.b) {
|
||||
return -1;
|
||||
}
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int compare_vmode(const struct b_tty_vmode *a, const struct b_tty_vmode *b)
|
||||
{
|
||||
if (a->v_attrib != b->v_attrib) {
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (compare_colour(&a->v_fg, &b->v_fg) != 0) {
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (compare_colour(&a->v_bg, &b->v_bg) != 0) {
|
||||
return -1;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static void set_attrib(WORD *attrp, enum b_tty_attrib old, enum b_tty_attrib new)
|
||||
{
|
||||
if (old == new) {
|
||||
return;
|
||||
}
|
||||
|
||||
WORD attrib = *attrp;
|
||||
|
||||
/* Bold ON */
|
||||
if (!(old & B_TTY_ATTRIB_BOLD) && new & B_TTY_ATTRIB_BOLD) {
|
||||
attrib |= FOREGROUND_INTENSITY;
|
||||
}
|
||||
|
||||
/* Bold OFF */
|
||||
if (old & B_TTY_ATTRIB_BOLD && !(new & B_TTY_ATTRIB_BOLD)) {
|
||||
attrib &= ~FOREGROUND_INTENSITY;
|
||||
}
|
||||
|
||||
/* Underline ON */
|
||||
if (!(old & B_TTY_ATTRIB_UNDERLINE) && new & B_TTY_ATTRIB_UNDERLINE) {
|
||||
attrib |= COMMON_LVB_UNDERSCORE;
|
||||
}
|
||||
|
||||
/* Underline OFF */
|
||||
if (old & B_TTY_ATTRIB_UNDERLINE && !(new & B_TTY_ATTRIB_UNDERLINE)) {
|
||||
attrib &= ~COMMON_LVB_UNDERSCORE;
|
||||
}
|
||||
|
||||
/* Italic ON */
|
||||
if (!(old & B_TTY_ATTRIB_ITALIC) && new & B_TTY_ATTRIB_ITALIC) {
|
||||
/* not supported */
|
||||
}
|
||||
|
||||
/* Italic OFF */
|
||||
if (old & B_TTY_ATTRIB_ITALIC && !(new & B_TTY_ATTRIB_ITALIC)) {
|
||||
/* not supported */
|
||||
}
|
||||
|
||||
*attrp = attrib;
|
||||
}
|
||||
|
||||
static void set_fg(
|
||||
WORD *attrp, const struct b_tty_colour *old,
|
||||
const struct b_tty_colour *new)
|
||||
{
|
||||
if (compare_colour(old, new) == 0) {
|
||||
return;
|
||||
}
|
||||
|
||||
WORD attrib = *attrp;
|
||||
attrib &= ~(FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE | FOREGROUND_INTENSITY);
|
||||
|
||||
switch (new->c_mode) {
|
||||
case B_TTY_COLOUR_16:
|
||||
attrib |= ansi_colour16_fg[new->c_16.value];
|
||||
break;
|
||||
default:
|
||||
attrib |= FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE;
|
||||
break;
|
||||
}
|
||||
|
||||
*attrp = attrib;
|
||||
}
|
||||
|
||||
static void set_bg(
|
||||
WORD *attrp, const struct b_tty_colour *old,
|
||||
const struct b_tty_colour *new)
|
||||
{
|
||||
if (compare_colour(old, new) == 0) {
|
||||
return;
|
||||
}
|
||||
|
||||
WORD attrib = *attrp;
|
||||
attrib &= ~(BACKGROUND_RED | BACKGROUND_GREEN | BACKGROUND_BLUE | BACKGROUND_INTENSITY);
|
||||
|
||||
switch (new->c_mode) {
|
||||
case B_TTY_COLOUR_16:
|
||||
attrib |= ansi_colour16_bg[new->c_16.value];
|
||||
break;
|
||||
default:
|
||||
attrib |= BACKGROUND_RED | BACKGROUND_GREEN | BACKGROUND_BLUE;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
void b_tty_set_vmode(struct b_tty *tty, const struct b_tty_vmode *vmode)
|
||||
{
|
||||
if (compare_vmode(&tty->t_vmode, vmode) == 0) {
|
||||
return;
|
||||
}
|
||||
|
||||
WORD attrib = 0;
|
||||
|
||||
CONSOLE_SCREEN_BUFFER_INFO csbi = {0};
|
||||
GetConsoleScreenBufferInfo(tty->t_out, &csbi);
|
||||
|
||||
attrib = csbi.wAttributes;
|
||||
|
||||
set_attrib(&attrib, tty->t_vmode.v_attrib, vmode->v_attrib);
|
||||
set_fg(&attrib, &tty->t_vmode.v_fg, &vmode->v_fg);
|
||||
set_bg(&attrib, &tty->t_vmode.v_bg, &vmode->v_bg);
|
||||
|
||||
SetConsoleTextAttribute(tty->t_out, attrib);
|
||||
|
||||
memcpy(&tty->t_vmode, vmode, sizeof *vmode);
|
||||
}
|
||||
|
||||
enum b_status b_tty_get_dimensions(struct b_tty* tty, unsigned int* w, unsigned int* h)
|
||||
{
|
||||
if (!(tty->t_flags & B_TTY_INTERACTIVE)) {
|
||||
return B_ERR_NOT_SUPPORTED;
|
||||
}
|
||||
|
||||
CONSOLE_SCREEN_BUFFER_INFO csbi = {0};
|
||||
GetConsoleScreenBufferInfo(tty->t_out, &csbi);
|
||||
|
||||
if (w) {
|
||||
*w = csbi.srWindow.Right - csbi.srWindow.Left;
|
||||
}
|
||||
|
||||
if (h) {
|
||||
*h = csbi.srWindow.Bottom - csbi.srWindow.Top;
|
||||
}
|
||||
|
||||
return B_SUCCESS;
|
||||
}
|
||||
|
||||
enum b_status b_tty_get_cursor_position(struct b_tty *tty, unsigned int *x, unsigned int *y)
|
||||
{
|
||||
if (!(tty->t_flags & B_TTY_INTERACTIVE)) {
|
||||
return B_ERR_NOT_SUPPORTED;
|
||||
}
|
||||
|
||||
CONSOLE_SCREEN_BUFFER_INFO csbi = {0};
|
||||
GetConsoleScreenBufferInfo(tty->t_out, &csbi);
|
||||
|
||||
if (x) {
|
||||
*x = csbi.dwCursorPosition.X;
|
||||
}
|
||||
|
||||
if (y) {
|
||||
*y = csbi.dwCursorPosition.Y;
|
||||
}
|
||||
|
||||
return B_SUCCESS;
|
||||
}
|
||||
|
||||
b_keycode b_tty_read_key(struct b_tty *tty)
|
||||
{
|
||||
if (tty->t_repeat_count > 0) {
|
||||
tty->t_repeat_count--;
|
||||
return tty->t_repeat_key;
|
||||
}
|
||||
|
||||
INPUT_RECORD d;
|
||||
HANDLE in = tty->t_in;
|
||||
BOOL status = TRUE;
|
||||
CONSOLE_READCONSOLE_CONTROL ctrl = {0};
|
||||
DWORD nr_read = 0;
|
||||
|
||||
while (1) {
|
||||
status = ReadConsoleInputA(in, &d, 1, &nr_read);
|
||||
if (status == FALSE) {
|
||||
return B_KEY_EOF;
|
||||
}
|
||||
|
||||
if (d.EventType != KEY_EVENT) {
|
||||
continue;
|
||||
}
|
||||
|
||||
if (!d.Event.KeyEvent.bKeyDown) {
|
||||
continue;
|
||||
}
|
||||
|
||||
b_keycode key = 0;
|
||||
|
||||
switch (d.Event.KeyEvent.wVirtualKeyCode) {
|
||||
case VK_CONTROL:
|
||||
case VK_RCONTROL:
|
||||
case VK_SHIFT:
|
||||
case VK_RSHIFT:
|
||||
case VK_MENU:
|
||||
continue;
|
||||
case VK_UP:
|
||||
key = B_KEY_ARROW_UP;
|
||||
break;
|
||||
case VK_DOWN:
|
||||
key = B_KEY_ARROW_DOWN;
|
||||
break;
|
||||
case VK_LEFT:
|
||||
key = B_KEY_ARROW_LEFT;
|
||||
break;
|
||||
case VK_RIGHT:
|
||||
key = B_KEY_ARROW_RIGHT;
|
||||
break;
|
||||
case VK_BACK:
|
||||
key = B_KEY_BACKSPACE;
|
||||
break;
|
||||
case VK_RETURN:
|
||||
key = B_KEY_RETURN;
|
||||
break;
|
||||
default:
|
||||
if (d.Event.KeyEvent.uChar.UnicodeChar == 0) {
|
||||
continue;
|
||||
}
|
||||
|
||||
key = d.Event.KeyEvent.uChar.AsciiChar;
|
||||
break;
|
||||
}
|
||||
|
||||
if (d.Event.KeyEvent.dwControlKeyState & LEFT_CTRL_PRESSED) {
|
||||
key = B_TTY_CTRL_KEY('a' + key - 1);
|
||||
}
|
||||
|
||||
if (d.Event.KeyEvent.wRepeatCount > 1) {
|
||||
tty->t_repeat_count = d.Event.KeyEvent.wRepeatCount - 1;
|
||||
tty->t_repeat_key = key;
|
||||
}
|
||||
|
||||
return key;
|
||||
}
|
||||
|
||||
return B_KEY_EOF;
|
||||
}
|
||||
|
||||
void b_tty_move_cursor_x(
|
||||
struct b_tty* tty, enum b_tty_position_base base, int pos)
|
||||
{
|
||||
CONSOLE_SCREEN_BUFFER_INFO console = {0};
|
||||
GetConsoleScreenBufferInfo(tty->t_out, &console);
|
||||
COORD cursor_pos;
|
||||
cursor_pos.Y = console.dwCursorPosition.Y;
|
||||
|
||||
switch (base) {
|
||||
case B_TTY_POS_CURSOR:
|
||||
cursor_pos.X = console.dwCursorPosition.X + pos;
|
||||
break;
|
||||
case B_TTY_POS_START:
|
||||
cursor_pos.X = pos;
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
|
||||
SetConsoleCursorPosition(tty->t_out, cursor_pos);
|
||||
}
|
||||
|
||||
void b_tty_move_cursor_y(struct b_tty *tty, enum b_tty_position_base base, int pos)
|
||||
{
|
||||
CONSOLE_SCREEN_BUFFER_INFO console = {0};
|
||||
GetConsoleScreenBufferInfo(tty->t_out, &console);
|
||||
COORD cursor_pos;
|
||||
cursor_pos.X = console.dwCursorPosition.X;
|
||||
|
||||
switch (base) {
|
||||
case B_TTY_POS_CURSOR:
|
||||
cursor_pos.Y = console.dwCursorPosition.Y + pos;
|
||||
break;
|
||||
case B_TTY_POS_START:
|
||||
cursor_pos.Y = pos;
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
|
||||
SetConsoleCursorPosition(tty->t_out, cursor_pos);
|
||||
}
|
||||
|
||||
void b_tty_clear(struct b_tty *tty, enum b_tty_clear_mode mode)
|
||||
{
|
||||
CONSOLE_SCREEN_BUFFER_INFO console = {0};
|
||||
GetConsoleScreenBufferInfo(tty->t_out, &console);
|
||||
|
||||
WCHAR fill = L' ';
|
||||
DWORD length = 0;
|
||||
COORD start;
|
||||
|
||||
DWORD all_length = 0, line_length = 0;
|
||||
|
||||
if (mode & B_TTY_CLEAR_ALL) {
|
||||
line_length = console.dwSize.X;
|
||||
all_length = line_length * console.dwSize.Y;
|
||||
} else if (mode & B_TTY_CLEAR_FROM_CURSOR) {
|
||||
line_length = console.dwSize.X - console.dwCursorPosition.X + 1;
|
||||
all_length = line_length + ((console.dwSize.Y - console.dwCursorPosition.Y) * console.dwSize.X);
|
||||
} else if (mode & B_TTY_CLEAR_TO_CURSOR) {
|
||||
line_length = console.dwCursorPosition.X;
|
||||
all_length = line_length
|
||||
+ ((console.dwCursorPosition.Y - 1) * console.dwSize.X);
|
||||
} else {
|
||||
abort();
|
||||
}
|
||||
|
||||
if (mode & B_TTY_CLEAR_SCREEN) {
|
||||
length = all_length;
|
||||
|
||||
if ((mode & B_TTY_CLEAR_ALL) || (mode & B_TTY_CLEAR_TO_CURSOR)) {
|
||||
start.X = 0;
|
||||
start.Y = 0;
|
||||
} else if (mode & B_TTY_CLEAR_FROM_CURSOR) {
|
||||
start = console.dwCursorPosition;
|
||||
}
|
||||
} else if (mode & B_TTY_CLEAR_LINE) {
|
||||
length = line_length;
|
||||
|
||||
if ((mode & B_TTY_CLEAR_ALL) || (mode & B_TTY_CLEAR_TO_CURSOR)) {
|
||||
start.X = 0;
|
||||
start.Y = console.dwCursorPosition.Y;
|
||||
} else if (mode & B_TTY_CLEAR_FROM_CURSOR) {
|
||||
start = console.dwCursorPosition;
|
||||
}
|
||||
} else {
|
||||
abort();
|
||||
}
|
||||
|
||||
DWORD nr_written = 0;
|
||||
FillConsoleOutputCharacterW(tty->t_out, fill, length, start, &nr_written);
|
||||
}
|
||||
Reference in New Issue
Block a user