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(); }