summaryrefslogtreecommitdiff
path: root/tty.c
blob: 3752cd7c128dcd97fed821e26ebe39c93c360827 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
#define _XOPEN_SOURCE 700
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <termios.h>

#include "more.h"

static FILE *resetty = NULL;
static struct termios resettings = { 0 };

static void resetterm(void)
{
	tcsetattr(fileno(resetty), TCSANOW, &resettings);
}

static int query_term(const char *cap, const char *variable, int def)
{
	int n = 0;
	char cmd[64];
	snprintf(cmd, sizeof(cmd), "tput %s", cap);
	FILE *f = popen(cmd, "r");
	if (f) {
		if (fscanf(f, "%d", &n) != 1) {
			n = 0;
		}
		pclose(f);
		if (n != 0) {
			return n;
		}
	}

	char *value = getenv(variable);
	if (value) {
		n = atoi(value);
	}
	return n ? n : def;
}

struct more_tty more_open_tty(int lines)
{
	struct more_tty mt = {
		.tty = stderr,
		.lines = query_term("lines", "LINES", 24),
		.columns = query_term("cols", "COLUMNS", 80),
	};

	if (lines > 0) {
		mt.lines = lines;
	}

	/* leave room for prompts */
	lines--;

	/* FIXME: only open /dev/tty if stderr is not readable */
	// if (!(fcntl(fileno(mt.tty), F_GETFL) & (O_WRONLY | O_RDWR))) {
		mt.tty = fopen("/dev/tty", "rb+");
	// }
	if (mt.tty == NULL) {
		perror("Couldn't open tty for reading");
		exit(1);
	}

	setbuf(mt.tty, NULL);

	tcgetattr(fileno(mt.tty), &resettings);
	struct termios term = resettings;
	term.c_lflag &= ~(ECHO | ICANON);
	term.c_cc[VMIN] = 1;
	term.c_cc[VTIME] = 0;
	tcsetattr(fileno(mt.tty), TCSANOW, &term);

	resetty = mt.tty;
	atexit(resetterm);

	return mt;
}