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
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
|
/* SPDX-License-Identifier: GPL-2.0-or-later */
/* Copyright (C) 2026 Richard Qin */
#include "tui.h"
#include <poll.h>
#include <stdarg.h>
#include <stdio.h>
#include <termios.h>
#include <unistd.h>
struct tui_state {
struct termios saved_termios;
bool termios_saved;
bool raw_mode;
bool alternate_screen;
};
static struct tui_state tui_state;
static int tui_read_byte(void)
{
unsigned char byte;
ssize_t count = read(STDIN_FILENO, &byte, sizeof(byte));
return count == (ssize_t)sizeof(byte) ? byte : -1;
}
bool tui_init(void)
{
if (tcgetattr(STDIN_FILENO, &tui_state.saved_termios) < 0) {
return false;
}
tui_state.termios_saved = true;
struct termios raw = tui_state.saved_termios;
cfmakeraw(&raw);
raw.c_oflag |= OPOST | ONLCR;
if (tcsetattr(STDIN_FILENO, TCSAFLUSH, &raw) < 0) {
tui_state.termios_saved = false;
return false;
}
tui_state.raw_mode = true;
/* DEC private mode 1049: 切换到备用屏幕缓冲区 */
fputs("\033[?1049h\033[H\033[?25l", stdout);
fflush(stdout);
tui_state.alternate_screen = true;
return true;
}
void tui_shutdown(void)
{
if (tui_state.alternate_screen) {
tui_show_cursor();
fputs("\033[?1049l", stdout);
fflush(stdout);
tui_state.alternate_screen = false;
}
if (tui_state.raw_mode && tui_state.termios_saved) {
tcsetattr(STDIN_FILENO, TCSAFLUSH, &tui_state.saved_termios);
tui_state.raw_mode = false;
}
}
void tui_clear(void)
{
fputs("\033[2J\033[H", stdout);
}
void tui_hide_cursor(void)
{
fputs("\033[?25l", stdout);
}
void tui_show_cursor(void)
{
fputs("\033[?25h", stdout);
}
void tui_printf(const char *format, ...)
{
va_list arguments;
va_start(arguments, format);
vprintf(format, arguments);
va_end(arguments);
}
struct tui_event tui_read_key(void)
{
int character = tui_read_byte();
if (character < 0) {
return (struct tui_event){ .kind = TUI_KEY_ESCAPE };
}
if (character == '\033') {
struct pollfd input = {
.fd = STDIN_FILENO,
.events = POLLIN,
};
if (poll(&input, 1, 25) <= 0) {
return (struct tui_event){ .kind = TUI_KEY_ESCAPE };
}
int sequence = tui_read_byte();
if (sequence == '[') {
int code = tui_read_byte();
switch (code) {
case 'A':
return (struct tui_event){ .kind = TUI_KEY_UP };
case 'B':
return (struct tui_event){ .kind = TUI_KEY_DOWN };
case 'C':
return (struct tui_event){ .kind = TUI_KEY_RIGHT };
case 'D':
return (struct tui_event){ .kind = TUI_KEY_LEFT };
default:
break;
}
}
return (struct tui_event){ .kind = TUI_KEY_ESCAPE };
}
if (character == '\r' || character == '\n') {
return (struct tui_event){ .kind = TUI_KEY_ENTER };
}
if (character == '\b' || character == 127) {
return (struct tui_event){ .kind = TUI_KEY_BACKSPACE };
}
return (struct tui_event){
.kind = TUI_KEY_CHARACTER,
.character = character,
};
}
bool tui_read_line(const char *prompt, char *buffer, size_t capacity)
{
size_t length = 0;
if (capacity == 0) {
return false;
}
tui_show_cursor();
buffer[0] = '\0';
if (prompt && *prompt) {
tui_printf("%s", prompt);
}
fflush(stdout);
for (;;) {
struct tui_event event = tui_read_key();
if (event.kind == TUI_KEY_ENTER) {
fputc('\n', stdout);
tui_hide_cursor();
return true;
}
if (event.kind == TUI_KEY_ESCAPE) {
fputc('\n', stdout);
tui_hide_cursor();
return false;
}
if (event.kind == TUI_KEY_BACKSPACE) {
if (length > 0) {
buffer[--length] = '\0';
fputs("\b \b", stdout);
fflush(stdout);
}
continue;
}
if (event.kind == TUI_KEY_CHARACTER &&
event.character >= 32 && event.character != 127 &&
length + 1 < capacity) {
buffer[length++] = (char)event.character;
buffer[length] = '\0';
fputc(event.character, stdout);
fflush(stdout);
}
}
}
void tui_wait(void)
{
fputs("\n\n按任意键返回...", stdout);
fflush(stdout);
(void)tui_read_key();
}
|
