TermKeyResult termkey_waitkey(TermKey *tk, TermKeyKey *key)
{
if(tk->fd == -1) {
errno = EBADF;
return TERMKEY_RES_ERROR;
}
while(1) {
TermKeyResult ret = termkey_getkey(tk, key);
switch(ret) {
case TERMKEY_RES_KEY:
case TERMKEY_RES_EOF:
case TERMKEY_RES_ERROR:
return ret;
case TERMKEY_RES_NONE:
ret = termkey_advisereadable(tk);
if(ret == TERMKEY_RES_ERROR)
return ret;
break;
case TERMKEY_RES_AGAIN:
{
if(tk->is_closed)
// We're closed now. Never going to get more bytes so just go with
// what we have
return termkey_getkey_force(tk, key);
struct pollfd fd;
retry:
fd.fd = tk->fd;
fd.events = POLLIN;
int pollret = poll(&fd, 1, tk->waittime);
if(pollret == -1) {
if(errno == EINTR && !(tk->flags & TERMKEY_FLAG_EINTR))
goto retry;
return TERMKEY_RES_ERROR;
}
if(fd.revents & (POLLIN|POLLHUP|POLLERR))
ret = termkey_advisereadable(tk);
else
ret = TERMKEY_RES_NONE;
if(ret == TERMKEY_RES_ERROR)
return ret;
if(ret == TERMKEY_RES_NONE)
return termkey_getkey_force(tk, key);
}
break;
}
}
/* UNREACHABLE */
}
static const char *ui_getkey(Ui *ui) {
UiCurses *uic = (UiCurses*)ui;
TermKeyKey key;
TermKeyResult ret = termkey_getkey(uic->termkey, &key);
if (ret == TERMKEY_RES_EOF) {
int tty = open("/dev/tty", O_RDWR);
if (tty == -1)
goto fatal;
if (tty != STDIN_FILENO && dup2(tty, STDIN_FILENO) == -1)
goto fatal;
termkey_destroy(uic->termkey);
if (!(uic->termkey = ui_termkey_new(STDIN_FILENO)))
goto fatal;
return NULL;
}
if (ret == TERMKEY_RES_AGAIN) {
struct pollfd fd;
fd.fd = STDIN_FILENO;
fd.events = POLLIN;
if (poll(&fd, 1, termkey_get_waittime(uic->termkey)) == 0)
ret = termkey_getkey_force(uic->termkey, &key);
}
if (ret != TERMKEY_RES_KEY)
return NULL;
termkey_strfkey(uic->termkey, uic->key, sizeof(uic->key), &key, TERMKEY_FORMAT_VIM);
return uic->key;
fatal:
ui_die_msg(ui, "Failed to re-open stdin as /dev/tty\n");
return NULL;
}
void CoreManager::io_input_timeout()
{
TermKeyKey key;
if (termkey_getkey_force(tk, &key) == TERMKEY_RES_KEY) {
/* This should happen only for Esc key, so no need to do locale->utf8
* conversion. */
ProcessInput(key);
}
}
int main(int argc, char *argv[])
{
TERMKEY_CHECK_VERSION;
TermKey *tk = termkey_new(0, 0);
if(!tk) {
fprintf(stderr, "Cannot allocate termkey instance\n");
exit(1);
}
struct pollfd fd;
fd.fd = 0; /* the file descriptor we passed to termkey_new() */
fd.events = POLLIN;
TermKeyResult ret;
TermKeyKey key;
int running = 1;
int nextwait = -1;
while(running) {
if(poll(&fd, 1, nextwait) == 0) {
// Timed out
if(termkey_getkey_force(tk, &key) == TERMKEY_RES_KEY)
on_key(tk, &key);
}
if(fd.revents & (POLLIN|POLLHUP|POLLERR))
termkey_advisereadable(tk);
while((ret = termkey_getkey(tk, &key)) == TERMKEY_RES_KEY) {
on_key(tk, &key);
if(key.type == TERMKEY_TYPE_UNICODE &&
key.modifiers & TERMKEY_KEYMOD_CTRL &&
(key.code.codepoint == 'C' || key.code.codepoint == 'c'))
running = 0;
}
if(ret == TERMKEY_RES_AGAIN)
nextwait = termkey_get_waittime(tk);
else
nextwait = -1;
}
termkey_destroy(tk);
}
static const char *ui_getkey(Ui *ui) {
UiCurses *uic = (UiCurses*)ui;
TermKeyKey key;
TermKeyResult ret = termkey_getkey(uic->termkey, &key);
if (ret == TERMKEY_RES_AGAIN) {
struct pollfd fd;
fd.fd = STDIN_FILENO;
fd.events = POLLIN;
if (poll(&fd, 1, termkey_get_waittime(uic->termkey)) == 0)
ret = termkey_getkey_force(uic->termkey, &key);
}
if (ret != TERMKEY_RES_KEY)
return NULL;
termkey_strfkey(uic->termkey, uic->key, sizeof(uic->key), &key, TERMKEY_FORMAT_VIM);
return uic->key;
}
int main(int argc, char *argv[])
{
TermKey *tk;
TermKeyKey key;
const char *str;
plan_tests(23);
tk = termkey_new_abstract("xterm", 0);
// 7bit DCS
termkey_push_bytes(tk, "\eP1$r1 q\e\\", 10);
is_int(termkey_getkey(tk, &key), TERMKEY_RES_KEY, "getkey yields RES_KEY for DCS");
is_int(key.type, TERMKEY_TYPE_DCS, "key.type for DCS");
is_int(key.modifiers, 0, "key.modifiers for DCS");
is_int(termkey_interpret_string(tk, &key, &str), TERMKEY_RES_KEY, "termkey_interpret_string() gives string");
is_str(str, "1$r1 q", "termkey_interpret_string() yields correct string");
is_int(termkey_getkey(tk, &key), TERMKEY_RES_NONE, "getkey again yields RES_NONE");
// 8bit DCS
termkey_push_bytes(tk, "\x90""1$r2 q""\x9c", 8);
is_int(termkey_getkey(tk, &key), TERMKEY_RES_KEY, "getkey yields RES_KEY for DCS");
is_int(key.type, TERMKEY_TYPE_DCS, "key.type for DCS");
is_int(key.modifiers, 0, "key.modifiers for DCS");
is_int(termkey_interpret_string(tk, &key, &str), TERMKEY_RES_KEY, "termkey_interpret_string() gives string");
is_str(str, "1$r2 q", "termkey_interpret_string() yields correct string");
is_int(termkey_getkey(tk, &key), TERMKEY_RES_NONE, "getkey again yields RES_NONE");
// 7bit OSC
termkey_push_bytes(tk, "\e]15;abc\e\\", 10);
is_int(termkey_getkey(tk, &key), TERMKEY_RES_KEY, "getkey yields RES_KEY for OSC");
is_int(key.type, TERMKEY_TYPE_OSC, "key.type for OSC");
is_int(key.modifiers, 0, "key.modifiers for OSC");
is_int(termkey_interpret_string(tk, &key, &str), TERMKEY_RES_KEY, "termkey_interpret_string() gives string");
is_str(str, "15;abc", "termkey_interpret_string() yields correct string");
is_int(termkey_getkey(tk, &key), TERMKEY_RES_NONE, "getkey again yields RES_NONE");
// False alarm
termkey_push_bytes(tk, "\eP", 2);
is_int(termkey_getkey(tk, &key), TERMKEY_RES_AGAIN, "getkey yields RES_AGAIN for false alarm");
is_int(termkey_getkey_force(tk, &key), TERMKEY_RES_KEY, "getkey_force yields RES_KEY for false alarm");
is_int(key.type, TERMKEY_TYPE_UNICODE, "key.type for false alarm");
is_int(key.code.codepoint, 'P', "key.code.codepoint for false alarm");
is_int(key.modifiers, TERMKEY_KEYMOD_ALT, "key.modifiers for false alarm");
termkey_destroy(tk);
return exit_status();
}
