catgirl/ui.c

423 lines
9.2 KiB
C

/* Copyright (C) 2018 Curtis McEnroe <june@causal.agency>
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#define _XOPEN_SOURCE_EXTENDED
#include <curses.h>
#include <err.h>
#include <locale.h>
#include <stdarg.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sysexits.h>
#include <wchar.h>
#include <wctype.h>
#include "chat.h"
#undef uiFmt
#define MIN(a, b) ((a) < (b) ? (a) : (b))
#define MAX(a, b) ((a) > (b) ? (a) : (b))
#define CTRL(c) ((c) & 037)
#ifndef A_ITALIC
#define A_ITALIC A_NORMAL
#endif
static void colorInit(void) {
start_color();
use_default_colors();
if (COLORS >= 16) {
for (short pair = 0; pair < 0x100; ++pair) {
if (pair < 0x10) {
init_pair(1 + pair, pair, -1);
} else {
init_pair(1 + pair, pair & 0x0F, (pair & 0xF0) >> 4);
}
}
} else {
for (short pair = 0; pair < 0100; ++pair) {
if (pair < 010) {
init_pair(1 + pair, pair, -1);
} else {
init_pair(1 + pair, pair & 007, (pair & 070) >> 3);
}
}
}
}
static attr_t attr8(short pair) {
if (COLORS >= 16 || pair < 0) return A_NORMAL;
return (pair & 0x08) ? A_BOLD : A_NORMAL;
}
static short pair8(short pair) {
if (COLORS >= 16 || pair < 0) return pair;
return (pair & 0x70) >> 1 | (pair & 0x07);
}
static const int TOPIC_COLS = 512;
static const int INPUT_COLS = 512;
static const int LOG_LINES = 100;
static int lastLine(void) {
return LINES - 1;
}
static int lastCol(void) {
return COLS - 1;
}
static int logHeight(void) {
return LINES - 4;
}
static struct {
WINDOW *topic;
WINDOW *log;
WINDOW *input;
int scroll;
size_t cursor;
} ui;
void uiInit(void) {
setlocale(LC_CTYPE, "");
initscr();
cbreak();
noecho();
colorInit();
ui.topic = newpad(2, TOPIC_COLS);
mvwhline(ui.topic, 1, 0, ACS_HLINE, TOPIC_COLS);
ui.log = newpad(LOG_LINES, COLS);
wsetscrreg(ui.log, 0, LOG_LINES - 1);
scrollok(ui.log, true);
wmove(ui.log, LOG_LINES - logHeight() - 1, 0);
ui.scroll = LOG_LINES;
ui.input = newpad(2, INPUT_COLS);
mvwhline(ui.input, 0, 0, ACS_HLINE, INPUT_COLS);
wmove(ui.input, 1, 0);
keypad(ui.input, true);
nodelay(ui.input, true);
}
static void uiResize(void) {
wresize(ui.log, LOG_LINES, COLS);
wmove(ui.log, LOG_LINES - 1, COLS - 1);
}
void uiHide(void) {
endwin();
printf(
"This program is AGPLv3 free software!\n"
"The source is available at <" SOURCE_URL ">.\n"
);
}
void uiDraw(void) {
pnoutrefresh(
ui.topic,
0, 0,
0, 0,
1, lastCol()
);
pnoutrefresh(
ui.log,
ui.scroll - logHeight(), 0,
2, 0,
lastLine() - 2, lastCol()
);
int _, x;
getyx(ui.input, _, x);
pnoutrefresh(
ui.input,
0, MAX(0, x - lastCol() + 3),
lastLine() - 1, 0,
lastLine(), lastCol()
);
doupdate();
}
static void uiRedraw(void) {
clearok(curscr, true);
}
static const short IRC_COLORS[16] = {
8 + COLOR_WHITE, // white
0 + COLOR_BLACK, // black
0 + COLOR_BLUE, // blue
0 + COLOR_GREEN, // green
8 + COLOR_RED, // red
0 + COLOR_RED, // brown
0 + COLOR_MAGENTA, // magenta
0 + COLOR_YELLOW, // orange
8 + COLOR_YELLOW, // yellow
8 + COLOR_GREEN, // light green
0 + COLOR_CYAN, // cyan
8 + COLOR_CYAN, // light cyan
8 + COLOR_BLUE, // light blue
8 + COLOR_MAGENTA, // pink
8 + COLOR_BLACK, // gray
0 + COLOR_WHITE, // light gray
};
static const wchar_t *parseColor(short *pair, const wchar_t *str) {
short fg = 0;
size_t fgLen = MIN(wcsspn(str, L"0123456789"), 2);
if (!fgLen) { *pair = -1; return str; }
for (size_t i = 0; i < fgLen; ++i) {
fg = fg * 10 + (str[i] - L'0');
}
str = &str[fgLen];
short bg = 0;
size_t bgLen = 0;
if (str[0] == L',') bgLen = MIN(wcsspn(&str[1], L"0123456789"), 2);
for (size_t i = 0; i < bgLen; ++i) {
bg = bg * 10 + (str[1 + i] - L'0');
}
if (bgLen) str = &str[1 + bgLen];
if (*pair == -1) *pair = 0;
*pair = (*pair & 0xF0) | IRC_COLORS[fg & 0x0F];
if (bgLen) *pair = (*pair & 0x0F) | (IRC_COLORS[bg & 0x0F] << 4);
return str;
}
static void wordWrap(WINDOW *win, const wchar_t *str) {
size_t len = wcscspn(str, L" ");
size_t width = 1;
for (size_t i = 0; i < len; ++i) {
if (iswprint(str[i])) width += wcwidth(str[i]);
}
int _, x, xMax;
getyx(win, _, x);
getmaxyx(win, _, xMax);
if (width >= (size_t)(xMax - x)) {
waddch(win, '\n');
} else {
waddch(win, ' ');
}
}
enum {
IRC_BOLD = 0x02,
IRC_COLOR = 0x03,
IRC_REVERSE = 0x16,
IRC_RESET = 0x0F,
IRC_ITALIC = 0x1D,
IRC_UNDERLINE = 0x1F,
};
static const wchar_t IRC_CODES[] = {
L' ',
IRC_BOLD,
IRC_COLOR,
IRC_REVERSE,
IRC_RESET,
IRC_ITALIC,
IRC_UNDERLINE,
L'\0',
};
static void addIRC(WINDOW *win, const wchar_t *str) {
attr_t attr = A_NORMAL;
short pair = -1;
for (;;) {
size_t cc = wcscspn(str, IRC_CODES);
wattr_set(win, attr | attr8(pair), 1 + pair8(pair), NULL);
waddnwstr(win, str, cc);
if (!str[cc]) break;
str = &str[cc];
switch (*str++) {
break; case L' ': wordWrap(win, str);
break; case IRC_BOLD: attr ^= A_BOLD;
break; case IRC_ITALIC: attr ^= A_ITALIC;
break; case IRC_UNDERLINE: attr ^= A_UNDERLINE;
break; case IRC_REVERSE: attr ^= A_REVERSE;
break; case IRC_COLOR: str = parseColor(&pair, str);
break; case IRC_RESET: attr = A_NORMAL; pair = -1;
}
}
}
void uiTopic(const wchar_t *topic) {
wmove(ui.topic, 0, 0);
addIRC(ui.topic, topic);
wclrtoeol(ui.topic);
}
void uiTopicStr(const char *topic) {
size_t len = strlen(topic);
wchar_t wcs[1 + len];
len = mbstowcs(wcs, topic, 1 + len);
if (len == (size_t)-1) err(EX_DATAERR, "mbstowcs");
uiTopic(wcs);
}
void uiLog(const wchar_t *line) {
waddch(ui.log, '\n');
addIRC(ui.log, line);
}
void uiFmt(const wchar_t *format, ...) {
wchar_t *buf;
va_list ap;
va_start(ap, format);
vaswprintf(&buf, format, ap);
va_end(ap);
if (!buf) err(EX_OSERR, "vaswprintf");
uiLog(buf);
free(buf);
}
static void logUp(void) {
if (ui.scroll == logHeight()) return;
ui.scroll = MAX(ui.scroll - logHeight() / 2, logHeight());
}
static void logDown(void) {
if (ui.scroll == LOG_LINES) return;
ui.scroll = MIN(ui.scroll + logHeight() / 2, LOG_LINES);
}
static struct {
wchar_t buf[512];
wchar_t *ptr;
wchar_t *end;
} line = { .ptr = line.buf, .end = line.buf };
static const size_t BUF_LEN = sizeof(line.buf) / sizeof(line.buf[0]);
static void left(void) {
if (line.ptr > line.buf) line.ptr--;
}
static void right(void) {
if (line.ptr < line.end) line.ptr++;
}
static void home(void) {
line.ptr = line.buf;
}
static void end(void) {
line.ptr = line.end;
}
static void kill(void) {
line.end = line.ptr;
}
static void insert(wchar_t ch) {
if (line.end == &line.buf[BUF_LEN - 1]) return;
if (line.ptr != line.end) {
wmemmove(line.ptr + 1, line.ptr, line.end - line.ptr);
}
*line.ptr++ = ch;
line.end++;
}
static void backspace(void) {
if (line.ptr == line.buf) return;
if (line.ptr != line.end) {
wmemmove(line.ptr - 1, line.ptr, line.end - line.ptr);
}
line.ptr--;
line.end--;
}
static void delete(void) {
if (line.ptr == line.end) return;
right();
backspace();
}
static void enter(void) {
if (line.end == line.buf) return;
*line.end = L'\0';
input(line.buf);
line.ptr = line.buf;
line.end = line.buf;
}
static void keyChar(wint_t ch) {
switch (ch) {
break; case CTRL('L'): uiRedraw();
break; case CTRL('B'): left();
break; case CTRL('F'): right();
break; case CTRL('A'): home();
break; case CTRL('E'): end();
break; case CTRL('D'): delete();
break; case CTRL('K'): kill();
break; case '\b': backspace();
break; case '\177': backspace();
break; case '\n': enter();
break; case CTRL('C'): insert(IRC_COLOR);
break; case CTRL('N'): insert(IRC_RESET);
break; case CTRL('O'): insert(IRC_BOLD);
break; case CTRL('R'): insert(IRC_COLOR);
break; case CTRL('T'): insert(IRC_ITALIC);
break; case CTRL('U'): insert(IRC_UNDERLINE);
break; case CTRL('V'): insert(IRC_REVERSE);
break; default: if (iswprint(ch)) insert(ch);
}
}
static void keyCode(wint_t ch) {
switch (ch) {
break; case KEY_RESIZE: uiResize();
break; case KEY_PPAGE: logUp();
break; case KEY_NPAGE: logDown();
break; case KEY_LEFT: left();
break; case KEY_RIGHT: right();
break; case KEY_HOME: home();
break; case KEY_END: end();
break; case KEY_BACKSPACE: backspace();
break; case KEY_DC: delete();
break; case KEY_ENTER: enter();
}
}
void uiRead(void) {
int ret;
wint_t ch;
while (ERR != (ret = wget_wch(ui.input, &ch))) {
if (ret == KEY_CODE_YES) {
keyCode(ch);
} else {
keyChar(ch);
}
}
wmove(ui.input, 1, 0);
ch = *line.ptr;
*line.ptr = L'\0';
addIRC(ui.input, line.buf);
*line.ptr = ch;
int _, x;
getyx(ui.input, _, x);
*line.end = L'\0';
addIRC(ui.input, line.ptr);
wclrtoeol(ui.input);
wmove(ui.input, 1, x);
}