summaryrefslogtreecommitdiff
path: root/ui.c
blob: 457824029a1043fc015ae7cbb9efbabb109495d9 (plain) (blame)
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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <sys/ioctl.h>
#include <termios.h>
#include <fcntl.h>

#include "util.h"
#include "ui.h"

#define MAX_STATUS	128

static char stbuf[MAX_STATUS];

static unsigned short rows;
static unsigned short cols;
static struct termios prevstate;
static int tty_fd;

int uiinit(const State *state)
{
	struct winsize w;
	struct termios t;

	if ((tty_fd = open("/dev/tty", O_RDWR|O_NOCTTY|O_CLOEXEC)) < 0)
		die("getpass: can't open /dev/tty for input");

	tcgetattr(tty_fd, &t);
	prevstate = t;
	if (state->flag & IGNORE_ISIG)
		t.c_lflag &= ~(ISIG);
	t.c_lflag &= ~(ECHO|INLCR|IGNCR|ICANON);
	t.c_iflag |= ICRNL;
	tcsetattr(tty_fd, TCSAFLUSH, &t);
	tcdrain(tty_fd);

	if (ioctl(STDIN_FILENO, TIOCGWINSZ, &w) == -1)
		die("ioctl() failed");

	rows = w.ws_row;
	cols = w.ws_col;

	printf("%c7", ESC); /* save cursor */
	printf("%c[?47h", ESC); /* use alt buffer */
	printf("%c[?25l", ESC); /* hide cursor */
	printf("%c[2J", ESC); /* clear vt */

	return tty_fd;
}

void pheader(const char *s)
{
	unsigned short col, row;

	row = rows / 2;
	col = (cols - strlen(s)) / 2;

	printf("%c[%d;%dH", ESC, row, col); /* pos curosr */
	printf("%c[2K", ESC); /* clear line */
	puts(s);
}

void pstatus(const char *s)
{
	unsigned short col, row;

	row = rows / 2 + 1;
	col = (cols - strlen(s)) / 2;

	printf("%c[%d;%dH", ESC, row, col); /* pos cursor */
	printf("%c[2K", ESC); /* clear line */
	puts(s);
}

void pstaticstat(const char *head, const char c)
{
	size_t stlen;

	if ((stlen = strlen(head)) > MAX_STATUS - 1)
		stlen = MAX_STATUS - 1;
	stbuf[stlen] = '\0';

	memset(stbuf, c, stlen);
	pstatus(stbuf);
}

void genrandstat(const char *head, const char *statstr)
{
	int entropy;
	size_t srlen, stlen, i;

	if ((stlen = strlen(head)) > MAX_STATUS - 1)
		stlen = MAX_STATUS - 1;
	srlen = strlen(statstr);

	for (entropy = i = 0; i < stlen; ++i, entropy /= srlen) {
		if (!entropy)
			entropy = rand();

		stbuf[i] = statstr[entropy % srlen];
	}

	stbuf[stlen] = '\0';
	pstatus(stbuf);
}

void uicleanup(void)
{
	printf("%c[2J", ESC); /* clear vt */
	printf("%c[?25h", ESC); /* show cursor */
	printf("%c[?47l", ESC); /* use normal buffer */
	printf("%c8", ESC); /* restore cursor */

	tcsetattr(tty_fd, TCSAFLUSH, &prevstate);
	close(tty_fd);
}

void pui(const char *head, const char stat)
{
	size_t stlen;

	if ((stlen = strlen(head)) > MAX_STATUS - 1)
		stlen = MAX_STATUS - 1;
	stbuf[stlen] = '\0';

	pheader(head);
	pstatus(memset(stbuf, stat, stlen));
}