|
| 1 | +// Copyright 2026 Leo Cheng |
| 2 | +// SPDX-License-Identifier: Apache-2.0 |
| 3 | +// |
| 4 | +// The five things a terminal UI cannot do without the operating system: switch |
| 5 | +// raw mode, write, read what has arrived, ask for the size, and read a clock. |
| 6 | +// Everything else in moonetui is pure MoonBit. |
| 7 | + |
| 8 | +#include <stdint.h> |
| 9 | +#include <string.h> |
| 10 | +#include <moonbit.h> |
| 11 | + |
| 12 | +#ifdef _WIN32 |
| 13 | + |
| 14 | +#include <windows.h> |
| 15 | + |
| 16 | +static DWORD saved_in_mode = 0; |
| 17 | +static DWORD saved_out_mode = 0; |
| 18 | +static int saved = 0; |
| 19 | + |
| 20 | +MOONBIT_FFI_EXPORT |
| 21 | +int32_t moonetui_tty_raw(int32_t on) { |
| 22 | + HANDLE in = GetStdHandle(STD_INPUT_HANDLE); |
| 23 | + HANDLE out = GetStdHandle(STD_OUTPUT_HANDLE); |
| 24 | + if (in == INVALID_HANDLE_VALUE || out == INVALID_HANDLE_VALUE) { |
| 25 | + return -1; |
| 26 | + } |
| 27 | + if (on) { |
| 28 | + if (!saved) { |
| 29 | + if (!GetConsoleMode(in, &saved_in_mode) || |
| 30 | + !GetConsoleMode(out, &saved_out_mode)) { |
| 31 | + return -1; |
| 32 | + } |
| 33 | + saved = 1; |
| 34 | + } |
| 35 | + // Virtual terminal input makes the console send the same escape sequences a |
| 36 | + // Unix terminal does, which is what lets one parser serve both. |
| 37 | + DWORD in_mode = saved_in_mode; |
| 38 | + in_mode &= ~(ENABLE_LINE_INPUT | ENABLE_ECHO_INPUT | ENABLE_PROCESSED_INPUT); |
| 39 | + in_mode |= ENABLE_VIRTUAL_TERMINAL_INPUT | ENABLE_WINDOW_INPUT; |
| 40 | + DWORD out_mode = saved_out_mode | ENABLE_VIRTUAL_TERMINAL_PROCESSING | |
| 41 | + DISABLE_NEWLINE_AUTO_RETURN; |
| 42 | + if (!SetConsoleMode(in, in_mode) || !SetConsoleMode(out, out_mode)) { |
| 43 | + return -1; |
| 44 | + } |
| 45 | + return 0; |
| 46 | + } |
| 47 | + if (saved) { |
| 48 | + SetConsoleMode(in, saved_in_mode); |
| 49 | + SetConsoleMode(out, saved_out_mode); |
| 50 | + } |
| 51 | + return 0; |
| 52 | +} |
| 53 | + |
| 54 | +MOONBIT_FFI_EXPORT |
| 55 | +int32_t moonetui_tty_write(const uint8_t *text, int32_t len) { |
| 56 | + HANDLE out = GetStdHandle(STD_OUTPUT_HANDLE); |
| 57 | + DWORD written = 0; |
| 58 | + int32_t at = 0; |
| 59 | + while (at < len) { |
| 60 | + if (!WriteFile(out, text + at, (DWORD)(len - at), &written, NULL)) { |
| 61 | + return -1; |
| 62 | + } |
| 63 | + at += (int32_t)written; |
| 64 | + } |
| 65 | + return at; |
| 66 | +} |
| 67 | + |
| 68 | +MOONBIT_FFI_EXPORT |
| 69 | +int32_t moonetui_tty_read(uint8_t *buf, int32_t len, int32_t timeout_ms) { |
| 70 | + HANDLE in = GetStdHandle(STD_INPUT_HANDLE); |
| 71 | + DWORD ready = WaitForSingleObject(in, (DWORD)(timeout_ms < 0 ? 0 : timeout_ms)); |
| 72 | + if (ready != WAIT_OBJECT_0) { |
| 73 | + return 0; |
| 74 | + } |
| 75 | + DWORD got = 0; |
| 76 | + if (!ReadFile(in, buf, (DWORD)len, &got, NULL)) { |
| 77 | + return -1; |
| 78 | + } |
| 79 | + return (int32_t)got; |
| 80 | +} |
| 81 | + |
| 82 | +MOONBIT_FFI_EXPORT |
| 83 | +int32_t moonetui_tty_size(int32_t *out) { |
| 84 | + CONSOLE_SCREEN_BUFFER_INFO info; |
| 85 | + HANDLE handle = GetStdHandle(STD_OUTPUT_HANDLE); |
| 86 | + if (!GetConsoleScreenBufferInfo(handle, &info)) { |
| 87 | + return -1; |
| 88 | + } |
| 89 | + out[0] = info.srWindow.Right - info.srWindow.Left + 1; |
| 90 | + out[1] = info.srWindow.Bottom - info.srWindow.Top + 1; |
| 91 | + return 0; |
| 92 | +} |
| 93 | + |
| 94 | +MOONBIT_FFI_EXPORT |
| 95 | +int32_t moonetui_tty_now(void) { |
| 96 | + return (int32_t)(GetTickCount64() & 0x7fffffff); |
| 97 | +} |
| 98 | + |
| 99 | +#else |
| 100 | + |
| 101 | +#include <errno.h> |
| 102 | +#include <poll.h> |
| 103 | +#include <sys/ioctl.h> |
| 104 | +#include <termios.h> |
| 105 | +#include <time.h> |
| 106 | +#include <unistd.h> |
| 107 | + |
| 108 | +static struct termios saved_termios; |
| 109 | +static int saved = 0; |
| 110 | + |
| 111 | +MOONBIT_FFI_EXPORT |
| 112 | +int32_t moonetui_tty_raw(int32_t on) { |
| 113 | + if (on) { |
| 114 | + if (!saved) { |
| 115 | + if (tcgetattr(STDIN_FILENO, &saved_termios) != 0) { |
| 116 | + return -1; |
| 117 | + } |
| 118 | + saved = 1; |
| 119 | + } |
| 120 | + struct termios raw = saved_termios; |
| 121 | + // Characters arrive as they are typed, without echo, and without the |
| 122 | + // terminal turning any of them into signals or flow control. |
| 123 | + raw.c_lflag &= ~(ECHO | ICANON | IEXTEN | ISIG); |
| 124 | + raw.c_iflag &= ~(IXON | IXOFF | ICRNL | INLCR | IGNCR | BRKINT | ISTRIP); |
| 125 | + raw.c_oflag &= ~(OPOST); |
| 126 | + raw.c_cc[VMIN] = 0; |
| 127 | + raw.c_cc[VTIME] = 0; |
| 128 | + return tcsetattr(STDIN_FILENO, TCSANOW, &raw) == 0 ? 0 : -1; |
| 129 | + } |
| 130 | + if (saved) { |
| 131 | + return tcsetattr(STDIN_FILENO, TCSANOW, &saved_termios) == 0 ? 0 : -1; |
| 132 | + } |
| 133 | + return 0; |
| 134 | +} |
| 135 | + |
| 136 | +MOONBIT_FFI_EXPORT |
| 137 | +int32_t moonetui_tty_write(const uint8_t *text, int32_t len) { |
| 138 | + int32_t at = 0; |
| 139 | + while (at < len) { |
| 140 | + ssize_t wrote = write(STDOUT_FILENO, text + at, (size_t)(len - at)); |
| 141 | + if (wrote < 0) { |
| 142 | + if (errno == EINTR) { |
| 143 | + continue; |
| 144 | + } |
| 145 | + return -1; |
| 146 | + } |
| 147 | + at += (int32_t)wrote; |
| 148 | + } |
| 149 | + return at; |
| 150 | +} |
| 151 | + |
| 152 | +MOONBIT_FFI_EXPORT |
| 153 | +int32_t moonetui_tty_read(uint8_t *buf, int32_t len, int32_t timeout_ms) { |
| 154 | + struct pollfd waiting; |
| 155 | + waiting.fd = STDIN_FILENO; |
| 156 | + waiting.events = POLLIN; |
| 157 | + waiting.revents = 0; |
| 158 | + int ready = poll(&waiting, 1, timeout_ms < 0 ? 0 : timeout_ms); |
| 159 | + if (ready <= 0) { |
| 160 | + return 0; |
| 161 | + } |
| 162 | + ssize_t got = read(STDIN_FILENO, buf, (size_t)len); |
| 163 | + if (got < 0) { |
| 164 | + return errno == EAGAIN || errno == EINTR ? 0 : -1; |
| 165 | + } |
| 166 | + return (int32_t)got; |
| 167 | +} |
| 168 | + |
| 169 | +MOONBIT_FFI_EXPORT |
| 170 | +int32_t moonetui_tty_size(int32_t *out) { |
| 171 | + struct winsize size; |
| 172 | + if (ioctl(STDOUT_FILENO, TIOCGWINSZ, &size) != 0) { |
| 173 | + return -1; |
| 174 | + } |
| 175 | + out[0] = size.ws_col; |
| 176 | + out[1] = size.ws_row; |
| 177 | + return 0; |
| 178 | +} |
| 179 | + |
| 180 | +MOONBIT_FFI_EXPORT |
| 181 | +int32_t moonetui_tty_now(void) { |
| 182 | + struct timespec now; |
| 183 | + clock_gettime(CLOCK_MONOTONIC, &now); |
| 184 | + int64_t ms = (int64_t)now.tv_sec * 1000 + now.tv_nsec / 1000000; |
| 185 | + return (int32_t)(ms & 0x7fffffff); |
| 186 | +} |
| 187 | + |
| 188 | +#endif |
0 commit comments