158 lines
3.8 KiB
C
158 lines
3.8 KiB
C
#include <stdint.h>
|
|
#include <stddef.h>
|
|
#include <lib/readline.h>
|
|
#include <lib/libc.h>
|
|
#include <lib/blib.h>
|
|
#include <lib/term.h>
|
|
#if defined (bios)
|
|
# include <lib/real.h>
|
|
#elif defined (uefi)
|
|
# include <efi.h>
|
|
#endif
|
|
|
|
int getchar_internal(uint8_t scancode, uint8_t ascii) {
|
|
switch (scancode) {
|
|
case 0x44:
|
|
return GETCHAR_F10;
|
|
case 0x4b:
|
|
return GETCHAR_CURSOR_LEFT;
|
|
case 0x4d:
|
|
return GETCHAR_CURSOR_RIGHT;
|
|
case 0x48:
|
|
return GETCHAR_CURSOR_UP;
|
|
case 0x50:
|
|
return GETCHAR_CURSOR_DOWN;
|
|
case 0x53:
|
|
return GETCHAR_DELETE;
|
|
case 0x4f:
|
|
return GETCHAR_END;
|
|
case 0x47:
|
|
return GETCHAR_HOME;
|
|
case 0x49:
|
|
return GETCHAR_PGUP;
|
|
case 0x51:
|
|
return GETCHAR_PGDOWN;
|
|
}
|
|
switch (ascii) {
|
|
case '\r':
|
|
return '\n';
|
|
}
|
|
return ascii;
|
|
}
|
|
|
|
#if defined (bios)
|
|
int getchar(void) {
|
|
struct rm_regs r = {0};
|
|
rm_int(0x16, &r, &r);
|
|
return getchar_internal((r.eax >> 8) & 0xff, r.eax);
|
|
}
|
|
#endif
|
|
|
|
#if defined (uefi)
|
|
int getchar(void) {
|
|
EFI_INPUT_KEY key = {0};
|
|
uefi_call_wrapper(gST->ConIn->ReadKeyStroke, 2, gST->ConIn, &key);
|
|
return getchar_internal(key.ScanCode, key.UnicodeChar);
|
|
}
|
|
|
|
int pit_sleep_and_quit_on_keypress(uint32_t pit_ticks) {
|
|
(void)pit_ticks;
|
|
return getchar();
|
|
}
|
|
#endif
|
|
|
|
static void reprint_string(int x, int y, const char *s) {
|
|
int orig_x, orig_y;
|
|
disable_cursor();
|
|
get_cursor_pos(&orig_x, &orig_y);
|
|
set_cursor_pos(x, y);
|
|
term_write(s, strlen(s));
|
|
set_cursor_pos(orig_x, orig_y);
|
|
enable_cursor();
|
|
}
|
|
|
|
static void cursor_back(void) {
|
|
int x, y;
|
|
get_cursor_pos(&x, &y);
|
|
if (x) {
|
|
x--;
|
|
} else if (y) {
|
|
y--;
|
|
x = term_cols - 1;
|
|
}
|
|
set_cursor_pos(x, y);
|
|
}
|
|
|
|
static void cursor_fwd(void) {
|
|
int x, y;
|
|
get_cursor_pos(&x, &y);
|
|
if (x < term_cols - 1) {
|
|
x++;
|
|
} else if (y < term_rows - 1) {
|
|
y++;
|
|
x = 0;
|
|
}
|
|
set_cursor_pos(x, y);
|
|
}
|
|
|
|
void readline(const char *orig_str, char *buf, size_t limit) {
|
|
size_t orig_str_len = strlen(orig_str);
|
|
memmove(buf, orig_str, orig_str_len);
|
|
buf[orig_str_len] = 0;
|
|
|
|
int orig_x, orig_y;
|
|
get_cursor_pos(&orig_x, &orig_y);
|
|
|
|
term_write(orig_str, orig_str_len);
|
|
|
|
for (size_t i = orig_str_len; ; ) {
|
|
int c = getchar();
|
|
switch (c) {
|
|
case GETCHAR_CURSOR_LEFT:
|
|
if (i) {
|
|
i--;
|
|
cursor_back();
|
|
}
|
|
continue;
|
|
case GETCHAR_CURSOR_RIGHT:
|
|
if (i < strlen(buf)) {
|
|
i++;
|
|
cursor_fwd();
|
|
}
|
|
continue;
|
|
case '\b':
|
|
if (i) {
|
|
i--;
|
|
cursor_back();
|
|
case GETCHAR_DELETE:;
|
|
size_t j;
|
|
for (j = i; ; j++) {
|
|
buf[j] = buf[j+1];
|
|
if (!buf[j]) {
|
|
buf[j] = ' ';
|
|
break;
|
|
}
|
|
}
|
|
reprint_string(orig_x, orig_y, buf);
|
|
buf[j] = 0;
|
|
}
|
|
continue;
|
|
case '\n':
|
|
term_write("\n", 1);
|
|
return;
|
|
default:
|
|
if (strlen(buf) < limit - 1) {
|
|
for (size_t j = strlen(buf); ; j--) {
|
|
buf[j+1] = buf[j];
|
|
if (j == i)
|
|
break;
|
|
}
|
|
buf[i] = c;
|
|
i++;
|
|
cursor_fwd();
|
|
reprint_string(orig_x, orig_y, buf);
|
|
}
|
|
}
|
|
}
|
|
}
|