1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
#include <stdio.h>
#include <unistd.h>
#include <fcntl.h>
#include <stdlib.h>
#include <stdarg.h>
#include <termios.h>
#include <sys/ioctl.h>
#include "tty.h"
void tty_reset(tty_t *tty) {
tcsetattr(tty->fdin, TCSANOW, &tty->original_termios);
}
void tty_close(tty_t *tty) {
tty_reset(tty);
fclose(tty->fout);
close(tty->fdin);
}
void tty_init(tty_t *tty, const char *tty_filename) {
tty->fdin = open(tty_filename, O_RDONLY);
tty->fout = fopen(tty_filename, "w");
if (setvbuf(tty->fout, NULL, _IOFBF, 4096))
perror("setvbuf");
if (tcgetattr(tty->fdin, &tty->original_termios))
perror("tcgetattr");
struct termios new_termios = tty->original_termios;
/*
* Disable both of
* ICANON Canonical input (erase and kill processing).
* ECHO Enable echo.
* ISIG Enable signals from control characters
*/
new_termios.c_lflag &= ~(ICANON | ECHO | ISIG);
if (tcsetattr(tty->fdin, TCSANOW, &new_termios))
perror("tcsetattr");
tty_getwinsz(tty);
tty_setnormal(tty);
}
void tty_getwinsz(tty_t *tty) {
struct winsize ws;
if (ioctl(fileno(tty->fout), TIOCGWINSZ, &ws) == -1) {
tty->maxwidth = 80;
tty->maxheight = 25;
} else {
tty->maxwidth = ws.ws_col;
tty->maxheight = ws.ws_row;
}
}
char tty_getchar(tty_t *tty) {
char ch;
int size = read(tty->fdin, &ch, 1);
if (size < 0) {
perror("error reading from tty");
exit(EXIT_FAILURE);
} else if (size == 0) {
/* EOF */
exit(EXIT_FAILURE);
} else {
return ch;
}
}
static void tty_sgr(tty_t *tty, int code) {
tty_printf(tty, "%c%c%im", 0x1b, '[', code);
}
void tty_setfg(tty_t *tty, int fg) {
if (tty->fgcolor != fg) {
tty_sgr(tty, 30 + fg);
tty->fgcolor = fg;
}
}
void tty_setinvert(tty_t *tty) {
tty_sgr(tty, 7);
}
void tty_setnormal(tty_t *tty) {
tty_sgr(tty, 0);
tty->fgcolor = 9;
}
void tty_newline(tty_t *tty) {
tty_printf(tty, "%c%cK\n", 0x1b, '[');
}
void tty_clearline(tty_t *tty) {
tty_printf(tty, "%c%cK", 0x1b, '[');
}
void tty_setcol(tty_t *tty, int col) {
tty_printf(tty, "%c%c%iG", 0x1b, '[', col + 1);
}
void tty_moveup(tty_t *tty, int i) {
tty_printf(tty, "%c%c%iA", 0x1b, '[', i);
}
void tty_printf(tty_t *tty, const char *fmt, ...) {
va_list args;
va_start(args, fmt);
vfprintf(tty->fout, fmt, args);
va_end(args);
}
void tty_flush(tty_t *tty) {
fflush(tty->fout);
}
size_t tty_getwidth(tty_t *tty) {
return tty->maxwidth;
}
size_t tty_getheight(tty_t *tty) {
return tty->maxheight;
}
|
