Archived
1
Fork 0

Add fallback method to get window size

This commit is contained in:
Hadeed 2023-08-24 17:57:25 +05:00
parent 9d5d4ea8c5
commit f4a540f487

77
kilo.c
View file

@ -1,4 +1,6 @@
#include <errno.h> #include <errno.h>
#include <ctype.h>
#include <stdbool.h>
#include <stdio.h> #include <stdio.h>
#include <stdlib.h> #include <stdlib.h>
#include <sys/ioctl.h> #include <sys/ioctl.h>
@ -7,13 +9,16 @@
void enable_raw_mode(); void enable_raw_mode();
void disable_raw_mode(); void disable_raw_mode();
int get_window_size(int *, int *); int get_window_size(int *, int *);
int read_cursor_position(int *, int *);
void editor_init(); void editor_init();
void editor_redraw_screen(); void editor_redraw_screen();
void editor_clear_screen(); void editor_clear_screen();
void editor_draw_rows(); void editor_draw_rows();
void editor_process_key(); void editor_process_key();
char editor_read_key();
void die(const char *); void die(const char *);
@ -23,11 +28,11 @@ struct {
struct termios orig_termios; struct termios orig_termios;
int rows; int rows;
int cols; int cols;
} E; bool raw;
} E = { .raw = false };
int main() { int main() {
editor_init(); editor_init();
enable_raw_mode();
while (1) { while (1) {
editor_redraw_screen(); editor_redraw_screen();
@ -37,14 +42,32 @@ int main() {
return 0; return 0;
} }
void editor_init() {
if (!isatty(STDIN_FILENO)) {
printf("kilo only supports a terminal at standard in. Exiting.");
exit(1);
}
enable_raw_mode();
if (get_window_size(&E.rows, &E.cols) == -1) die("get_window_size");
}
void enable_raw_mode() { void enable_raw_mode() {
if (E.raw) return;
if (tcgetattr(STDIN_FILENO, &E.orig_termios) == -1) die("tcgetattr"); if (tcgetattr(STDIN_FILENO, &E.orig_termios) == -1) die("tcgetattr");
atexit(disable_raw_mode); atexit(disable_raw_mode);
struct termios raw = E.orig_termios; struct termios raw = E.orig_termios;
cfmakeraw(&raw); cfmakeraw(&raw);
raw.c_cc[VMIN] = 0;
raw.c_cc[VTIME] = 1;
if (tcsetattr(STDIN_FILENO, TCSAFLUSH, &raw) == -1) die("tcsetattr"); if (tcsetattr(STDIN_FILENO, TCSAFLUSH, &raw) == -1) die("tcsetattr");
E.raw = true;
} }
void disable_raw_mode() { void disable_raw_mode() {
@ -55,23 +78,37 @@ void disable_raw_mode() {
int get_window_size(int *rows, int *cols) { int get_window_size(int *rows, int *cols) {
struct winsize ws; struct winsize ws;
if (ioctl(STDIN_FILENO, TIOCGWINSZ, &ws) != -1 && ws.ws_col != 0) { if (ioctl(STDOUT_FILENO, TIOCGWINSZ, &ws) != -1 && ws.ws_col != 0) {
*rows = ws.ws_row; *rows = ws.ws_row;
*cols = ws.ws_col; *cols = ws.ws_col;
return 0; return 0;
} else {
if (write(STDOUT_FILENO, "\x1b[999C\x1b[999B", 12) != 12) return -1;
return read_cursor_position(rows, cols);
} }
return -1; return -1;
} }
void editor_init() { int read_cursor_position(int *rows, int *cols) {
if (!isatty(STDIN_FILENO)) { char buf[32];
printf("kilo only supports a terminal at standard in. Exiting."); unsigned int i = 0;
exit(1);
}
if (get_window_size(&E.rows, &E.cols) == -1) die("get_window_size"); if (write(STDOUT_FILENO, "\x1b[6n", 4) != 4) return -1;
if (read(STDIN_FILENO, buf, 1) != 1 && *buf != '\x1b') return -1;
if (read(STDIN_FILENO, buf, 1) != 1 && *buf != '[') return -1;
do {
if (i == sizeof(buf)) return -1;
if (read(STDIN_FILENO, buf+i, 1) == 0) break;
} while(buf[i++] != 'R');
buf[i-1] = '\0';
if (sscanf(buf, "%d;%d", rows, cols) != 2) return -1;
return 0;
} }
void editor_redraw_screen() { void editor_redraw_screen() {
@ -79,24 +116,22 @@ void editor_redraw_screen() {
editor_draw_rows(); editor_draw_rows();
write(STDIN_FILENO, "\x1b[H", 3); write(STDOUT_FILENO, "\x1b[H", 3);
} }
void editor_clear_screen() { void editor_clear_screen() {
write(STDIN_FILENO, "\x1b[2J", 4); write(STDOUT_FILENO, "\x1b[2J", 4);
write(STDIN_FILENO, "\x1b[H", 3); write(STDOUT_FILENO, "\x1b[H", 3);
} }
void editor_draw_rows() { void editor_draw_rows() {
for (int y = 0; y < E.rows - 1; y++) for (int y = 0; y < E.rows - 1; y++)
write(STDIN_FILENO, "~\r\n", 3); write(STDOUT_FILENO, "~\r\n", 3);
write(STDIN_FILENO, "~", 1); write(STDOUT_FILENO, "~", 1);
} }
void editor_process_key() { void editor_process_key() {
char c; char c = editor_read_key();
read(STDIN_FILENO, &c, 1);
switch (c) { switch (c) {
case CTRL_KEY('Q'): case CTRL_KEY('Q'):
editor_clear_screen(); editor_clear_screen();
@ -108,6 +143,12 @@ void editor_process_key() {
} }
} }
char editor_read_key() {
char c;
while (read(STDIN_FILENO, &c, 1) == 0);
return c;
}
void die(const char *s) { void die(const char *s) {
editor_clear_screen(); editor_clear_screen();