sfm

simple file manager
git clone git://git.afify.dev/sfm
Log | Files | Refs | README | LICENSE

commit 042fd5157a0764b8874b20aa87d3526ecc33d805
Author: afify <hassan@afify.dev>
Date:   Tue, 30 Jun 2020 06:44:56 +0300

initial commit

Diffstat:
ALICENSE | 15+++++++++++++++
AMakefile | 50++++++++++++++++++++++++++++++++++++++++++++++++++
AREADME.md | 63+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
Abytebuffer.inl | 70++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
Aconfig.def.h | 41+++++++++++++++++++++++++++++++++++++++++
Aconfig.mk | 13+++++++++++++
Ainput.inl | 228+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
Asfm.1 | 43+++++++++++++++++++++++++++++++++++++++++++
Asfm.c | 995+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
Asfm.png | 0
Aterm.inl | 311+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
Atermbox.c | 678+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
Atermbox.h | 317+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
Autf8.c | 79+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
Autil.c | 35+++++++++++++++++++++++++++++++++++
Autil.h | 15+++++++++++++++
16 files changed, 2953 insertions(+), 0 deletions(-)

diff --git a/LICENSE b/LICENSE @@ -0,0 +1,15 @@ +ISC License + +© 2020 Hassan Afify <hassan at afify dot dev> + +Permission to use, copy, modify, and distribute this software for any +purpose with or without fee is hereby granted, provided that the above +copyright notice and this permission notice appear in all copies. + +THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES +WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF +MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR +ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES +WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN +ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF +OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. diff --git a/Makefile b/Makefile @@ -0,0 +1,50 @@ +# sfm +# See LICENSE file for copyright and license details. + +include config.mk + +SRC = sfm.c util.c termbox.c utf8.c +OBJ = ${SRC:.c=.o} + +all: options sfm + +options: + @echo sfm build options: + @echo "CFLAGS = ${CFLAGS}" + @echo "CC = ${CC}" + +.c.o: + ${CC} -c ${CFLAGS} $< + +${OBJ}: config.h config.mk + +config.h: + cp config.def.h $@ + +sfm: ${OBJ} + ${CC} -o $@ ${OBJ} + +clean: + rm -f sfm ${OBJ} sfm-${VERSION}.tar.gz + +dist: clean + mkdir -p sfm-${VERSION} + cp -R LICENSE Makefile README.md config.def.h config.mk\ + sfm.1 util.h ${SRC} sfm-${VERSION} + tar -cf sfm-${VERSION}.tar sfm-${VERSION} + gzip sfm-${VERSION}.tar + rm -rf sfm-${VERSION} + +install: sfm + mkdir -p ${DESTDIR}${PREFIX}/bin + cp -f sfm ${DESTDIR}${PREFIX}/bin + chmod 755 ${DESTDIR}${PREFIX}/bin/sfm + mkdir -p ${DESTDIR}${MANPREFIX}/man1 + sed "s/VERSION/${VERSION}/g" < sfm.1 > ${DESTDIR}${MANPREFIX}/man1/sfm.1 + chmod 644 ${DESTDIR}${MANPREFIX}/man1/sfm.1 + +uninstall: + rm -f ${DESTDIR}${PREFIX}/bin/sfm\ + ${DESTDIR}${MANPREFIX}/man1/sfm.1 + +.PHONY: all options clean dist install uninstall diff --git a/README.md b/README.md @@ -0,0 +1,63 @@ +![sfm](https://github.com/afify/sfm/blob/master/sfm.png?raw=true) + +**simple file manager for unix-like systems** + +[![Language grade: C/C++](https://img.shields.io/lgtm/grade/cpp/g/afify/sfm.svg?logo=lgtm&logoWidth=18)](https://lgtm.com/projects/g/afify/sfm/context:cpp) +[![Build status](https://ci.appveyor.com/api/projects/status/goq88ahjyvtjrui2?svg=true)](https://ci.appveyor.com/project/afify/sfm) +[![License](https://img.shields.io/github/license/afify/sfm?color=blue)](https://github.com/afify/sfm/blob/master/LICENSE) + +Features +-------- +* No dependencies, static linking of [termbox](https://github.com/nsf/termbox). +* Fast, minimal, lightweight, c99. +* Open files (videos, images, ...) +* Vim keys navigation +* Inspired by [vifm](https://vifm.info/) and [noice](https://git.2f30.org/noice/). +* Follows the suckless [philosophy](https://suckless.org/philosophy/) and [code style](https://suckless.org/coding_style/). + +<img src="https://github.com/afify/afify.github.io/raw/master/img/sfm_sc.png" alt="drawing" width="800"/> + +Options +------- +```sh +$ sfm [-v] +$ man sfm +``` + +| key | description | +|:-------:|:---------------| +| `q` | quit | +| `h` | back | +| `j` | down | +| `k` | up | +| `l` | open dir|file | +| `g` | top | +| `G` | bottom | +| `space` | switch pane | + +Installation +------------ +**current** +```sh +git clone https://github.com/afify/sfm.git +cd sfm/ +make +make install +``` +**latest release** +```sh +wget --content-disposition $(curl -s https://api.github.com/repos/afify/sfm/releases/latest | tr -d '",' | awk '/tag_name/ {print "https://github.com/afify/sfm/archive/"$2".tar.gz"}') +tar -xzf sfm-*.tar.gz && cd sfm-*/ +make +make install +``` +Run +--- +```sh +$ sfm +``` + +Configuration +------------- +The configuration of sfm is done by creating a custom config.h +and (re)compiling the source code. diff --git a/bytebuffer.inl b/bytebuffer.inl @@ -0,0 +1,70 @@ +struct bytebuffer { + char *buf; + int len; + int cap; +}; + +static void bytebuffer_reserve(struct bytebuffer *b, int cap) { + if (b->cap >= cap) { + return; + } + + // prefer doubling capacity + if (b->cap * 2 >= cap) { + cap = b->cap * 2; + } + + char *newbuf = realloc(b->buf, cap); + b->buf = newbuf; + b->cap = cap; +} + +static void bytebuffer_init(struct bytebuffer *b, int cap) { + b->cap = 0; + b->len = 0; + b->buf = 0; + + if (cap > 0) { + b->cap = cap; + b->buf = malloc(cap); // just assume malloc works always + } +} + +static void bytebuffer_free(struct bytebuffer *b) { + if (b->buf) + free(b->buf); +} + +static void bytebuffer_clear(struct bytebuffer *b) { + b->len = 0; +} + +static void bytebuffer_append(struct bytebuffer *b, const char *data, int len) { + bytebuffer_reserve(b, b->len + len); + memcpy(b->buf + b->len, data, len); + b->len += len; +} + +static void bytebuffer_puts(struct bytebuffer *b, const char *str) { + bytebuffer_append(b, str, strlen(str)); +} + +static void bytebuffer_resize(struct bytebuffer *b, int len) { + bytebuffer_reserve(b, len); + b->len = len; +} + +static void bytebuffer_flush(struct bytebuffer *b, int fd) { + write(fd, b->buf, b->len); + bytebuffer_clear(b); +} + +static void bytebuffer_truncate(struct bytebuffer *b, int n) { + if (n <= 0) + return; + if (n > b->len) + n = b->len; + const int nmove = b->len - n; + memmove(b->buf, b->buf+n, nmove); + b->len -= n; +} diff --git a/config.def.h b/config.def.h @@ -0,0 +1,41 @@ +/* See LICENSE file for copyright and license details.*/ + +#ifndef CONFIG_H +#define CONFIG_H + +#include "termbox.h" + +/* colors */ +uint16_t dir_b = TB_DEFAULT; +uint16_t dir_f = 33; +uint16_t file_b = TB_DEFAULT; +uint16_t file_f = TB_DEFAULT; +uint16_t frame_b = TB_DEFAULT; +uint16_t frame_f = TB_DEFAULT; +uint16_t other_b = TB_DEFAULT; +uint16_t other_f = 3; +uint16_t pane_l_b = TB_DEFAULT; +uint16_t pane_l_f = 5; +uint16_t pane_r_b = TB_DEFAULT; +uint16_t pane_r_f = 166; +uint16_t status_b = 234; +uint16_t status_f = TB_DEFAULT; + +/* openwith */ +char audio[] = "mpv"; +char html[] = "firefox"; +char images[] = "sxiv"; +char pdf[] = "mupdf"; +char videos[] = "mpv"; + +/* unicode chars */ +uint32_t u_cne = 0x2510; +uint32_t u_cnw = 0x250C; +uint32_t u_cse = 0x2518; +uint32_t u_csw = 0x2514; +uint32_t u_hl = 0x2500; +uint32_t u_mn = 0x252C; +uint32_t u_ms = 0x2534; +uint32_t u_vl = 0x2502; + +#endif /* CONFIG_H */ diff --git a/config.mk b/config.mk @@ -0,0 +1,13 @@ +# sfm version +VERSION = 0.1 + +# paths +PREFIX = /usr/local +MANPREFIX = ${PREFIX}/share/man + +# flags +CPPFLAGS = -D_DEFAULT_SOURCE -D_BSD_SOURCE -D_POSIX_C_SOURCE=200809L -D_XOPEN_SOURCE=700 -DVERSION=\"${VERSION}\" +CFLAGS = -std=c99 -pedantic -Wextra -Wall -Os ${CPPFLAGS} + +# compiler and linker +CC = cc diff --git a/input.inl b/input.inl @@ -0,0 +1,228 @@ +// if s1 starts with s2 returns true, else false +// len is the length of s1 +// s2 should be null-terminated +static bool starts_with(const char *s1, int len, const char *s2) +{ + int n = 0; + while (*s2 && n < len) { + if (*s1++ != *s2++) + return false; + n++; + } + return *s2 == 0; +} + +static int parse_mouse_event(struct tb_event *event, const char *buf, int len) { + if (len >= 6 && starts_with(buf, len, "\033[M")) { + // X10 mouse encoding, the simplest one + // \033 [ M Cb Cx Cy + int b = buf[3] - 32; + switch (b & 3) { + case 0: + if ((b & 64) != 0) + event->key = TB_KEY_MOUSE_WHEEL_UP; + else + event->key = TB_KEY_MOUSE_LEFT; + break; + case 1: + if ((b & 64) != 0) + event->key = TB_KEY_MOUSE_WHEEL_DOWN; + else + event->key = TB_KEY_MOUSE_MIDDLE; + break; + case 2: + event->key = TB_KEY_MOUSE_RIGHT; + break; + case 3: + event->key = TB_KEY_MOUSE_RELEASE; + break; + default: + return -6; + } + event->type = TB_EVENT_MOUSE; // TB_EVENT_KEY by default + if ((b & 32) != 0) + event->mod |= TB_MOD_MOTION; + + // the coord is 1,1 for upper left + event->x = (uint8_t)buf[4] - 1 - 32; + event->y = (uint8_t)buf[5] - 1 - 32; + + return 6; + } else if (starts_with(buf, len, "\033[<") || starts_with(buf, len, "\033[")) { + // xterm 1006 extended mode or urxvt 1015 extended mode + // xterm: \033 [ < Cb ; Cx ; Cy (M or m) + // urxvt: \033 [ Cb ; Cx ; Cy M + int i, mi = -1, starti = -1; + int isM, isU, s1 = -1, s2 = -1; + int n1 = 0, n2 = 0, n3 = 0; + + for (i = 0; i < len; i++) { + // We search the first (s1) and the last (s2) ';' + if (buf[i] == ';') { + if (s1 == -1) + s1 = i; + s2 = i; + } + + // We search for the first 'm' or 'M' + if ((buf[i] == 'm' || buf[i] == 'M') && mi == -1) { + mi = i; + break; + } + } + if (mi == -1) + return 0; + + // whether it's a capital M or not + isM = (buf[mi] == 'M'); + + if (buf[2] == '<') { + isU = 0; + starti = 3; + } else { + isU = 1; + starti = 2; + } + + if (s1 == -1 || s2 == -1 || s1 == s2) + return 0; + + n1 = strtoul(&buf[starti], NULL, 10); + n2 = strtoul(&buf[s1 + 1], NULL, 10); + n3 = strtoul(&buf[s2 + 1], NULL, 10); + + if (isU) + n1 -= 32; + + switch (n1 & 3) { + case 0: + if ((n1&64) != 0) { + event->key = TB_KEY_MOUSE_WHEEL_UP; + } else { + event->key = TB_KEY_MOUSE_LEFT; + } + break; + case 1: + if ((n1&64) != 0) { + event->key = TB_KEY_MOUSE_WHEEL_DOWN; + } else { + event->key = TB_KEY_MOUSE_MIDDLE; + } + break; + case 2: + event->key = TB_KEY_MOUSE_RIGHT; + break; + case 3: + event->key = TB_KEY_MOUSE_RELEASE; + break; + default: + return mi + 1; + } + + if (!isM) { + // on xterm mouse release is signaled by lowercase m + event->key = TB_KEY_MOUSE_RELEASE; + } + + event->type = TB_EVENT_MOUSE; // TB_EVENT_KEY by default + if ((n1&32) != 0) + event->mod |= TB_MOD_MOTION; + + event->x = (uint8_t)n2 - 1; + event->y = (uint8_t)n3 - 1; + + return mi + 1; + } + + return 0; +} + +// convert escape sequence to event, and return consumed bytes on success (failure == 0) +static int parse_escape_seq(struct tb_event *event, const char *buf, int len) +{ + int mouse_parsed = parse_mouse_event(event, buf, len); + + if (mouse_parsed != 0) + return mouse_parsed; + + // it's pretty simple here, find 'starts_with' match and return + // success, else return failure + int i; + for (i = 0; keys[i]; i++) { + if (starts_with(buf, len, keys[i])) { + event->ch = 0; + event->key = 0xFFFF-i; + return strlen(keys[i]); + } + } + return 0; +} + +static bool extract_event(struct tb_event *event, struct bytebuffer *inbuf, int inputmode) +{ + const char *buf = inbuf->buf; + const int len = inbuf->len; + if (len == 0) + return false; + + if (buf[0] == '\033') { + int n = parse_escape_seq(event, buf, len); + if (n != 0) { + bool success = true; + if (n < 0) { + success = false; + n = -n; + } + bytebuffer_truncate(inbuf, n); + return success; + } else { + // it's not escape sequence, then it's ALT or ESC, + // check inputmode + if (inputmode&TB_INPUT_ESC) { + // if we're in escape mode, fill ESC event, pop + // buffer, return success + event->ch = 0; + event->key = TB_KEY_ESC; + event->mod = 0; + bytebuffer_truncate(inbuf, 1); + return true; + } else if (inputmode&TB_INPUT_ALT) { + // if we're in alt mode, set ALT modifier to + // event and redo parsing + event->mod = TB_MOD_ALT; + bytebuffer_truncate(inbuf, 1); + return extract_event(event, inbuf, inputmode); + } + assert(!"never got here"); + } + } + + // if we're here, this is not an escape sequence and not an alt sequence + // so, it's a FUNCTIONAL KEY or a UNICODE character + + // first of all check if it's a functional key + if ((unsigned char)buf[0] <= TB_KEY_SPACE || + (unsigned char)buf[0] == TB_KEY_BACKSPACE2) + { + // fill event, pop buffer, return success */ + event->ch = 0; + event->key = (uint16_t)buf[0]; + bytebuffer_truncate(inbuf, 1); + return true; + } + + // feh... we got utf8 here + + // check if there is all bytes + if (len >= tb_utf8_char_length(buf[0])) { + /* everything ok, fill event, pop buffer, return success */ + tb_utf8_char_to_unicode(&event->ch, buf); + event->key = 0; + bytebuffer_truncate(inbuf, tb_utf8_char_length(buf[0])); + return true; + } + + // event isn't recognized, perhaps there is not enough bytes in utf8 + // sequence + return false; +} diff --git a/sfm.1 b/sfm.1 @@ -0,0 +1,43 @@ +.TH sfm 1 sfm\-VERSION +.SH NAME +sfm \- a simple file manager for unix-like systems +.SH SYNOPSIS +.B sfm +.RB [ \-v ] +.SH DESCRIPTION +sfm is a simple file manager based on termbox. +.P +.SH OPTIONS +.TP +.B \-v +print version. +.SH USAGE +.TP +.B q +quit +.TP +.B h +back +.TP +.B j +down +.TP +.B k +up +.TP +.B l +open dir|file +.TP +.B g +top +.TP +.B G +bottom +.TP +.B space +switch pane +.SH CUSTOMIZATION +sfm is customized by creating a custom config.h and (re)compiling the source +code. This keeps it fast, secure and simple. +.SH AUTHORS +Hassan Afify <hassan at afify dot dev> diff --git a/sfm.c b/sfm.c @@ -0,0 +1,995 @@ +/* See LICENSE file for copyright and license details. */ + +#include <dirent.h> +#include <errno.h> +#include <stdarg.h> +#include <stdint.h> +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <sys/stat.h> +#include <sys/types.h> +#include <sys/resource.h> +#include <sys/wait.h> +#include <unistd.h> +#include <fcntl.h> +#include <time.h> +#include <pwd.h> +#include <grp.h> + +#include "config.h" +#include "termbox.h" +#include "util.h" + +/* macros */ +#define MAX_P 4095 +#define MAX_N 255 + +/* typedef */ +typedef struct { + char name[MAX_N]; + char full[MAX_P]; + off_t size; + mode_t mode; +} Entry; + +typedef struct { + char dirn[MAX_P]; // directory name + char high_dir[MAX_P]; // highlighted_dir fullpath + int dirx; // position of title + size_t hdir; // highlighted_dir + size_t dirc; // total files in dir + uint16_t dir_bg; + uint16_t dir_fg; +} Pane; + +/* function declarations */ +static void print_tb(const char*, int, int, uint16_t, uint16_t); +static void printf_tb(int, int, uint16_t, uint16_t, const char*, ...); +static void print_status(const char*, ...); +static void print_error(const char*, ...); +static void clear(int, int, int, uint16_t); +static void clear_status(void); +static void clear_error(void); +static void clear_pane(int); +static char *get_extentions(char*); +static char *get_full_path(char*, char*); +static char *get_parent(char*); +static char *get_file_info(Entry*); +static char *get_file_size(off_t); +static char *get_file_date(struct stat); +static char *get_file_time(struct stat); +static char *get_file_userowner(struct stat); +static char *get_file_groupowner(struct stat); +static void get_file_perm(mode_t, char*); +static int check_dir(char*); +static int open_files(char*); +static int sort_name(const void *const, const void *const); +static double xfloorf(double); +static void float_to_string(float, char*); +static int get_memory_usage(void); +static void print_col(Entry*, size_t, size_t, size_t, int, int); +static int listdir(Pane*); +static void press(struct tb_event*, Pane*, Pane*); +static void t_resize(Pane*, Pane*); +static int set_panes(Pane*, Pane*, int); +static void draw_frame(void); +static int start(void); + +/* global variables */ +static int parent_row = 1; // FIX + +/* function implementations */ +static void +print_tb(const char *str, int x, int y, uint16_t fg, uint16_t bg) +{ + while (*str != '\0') { + uint32_t uni = 0; + str += tb_utf8_char_to_unicode(&uni, str); + tb_change_cell(x, y, uni, fg, bg); + x++; + } +} + +static void +printf_tb(int x, int y, uint16_t fg, uint16_t bg, const char *fmt, ...) +{ + char buf[4096]; + va_list vl; + va_start(vl, fmt); + (void)vsnprintf(buf, sizeof(buf), fmt, vl); + va_end(vl); + print_tb(buf, x, y, fg, bg); +} + +static void +print_status(const char *fmt, ...) +{ + int height, width; + width = tb_width(); + height = tb_height(); + + char buf[256]; + va_list vl; + va_start(vl, fmt); + vsnprintf(buf, sizeof(buf), fmt, vl); + va_end(vl); + clear_status(); + print_tb(buf, 2, height-2, TB_DEFAULT, status_b); + +} + +static void +print_error(const char *fmt, ...) +{ + int height, width; + width = tb_width(); + height = tb_height(); + + char buf[256]; + va_list vl; + va_start(vl, fmt); + vsnprintf(buf, sizeof(buf), fmt, vl); + va_end(vl); + clear_error(); + print_tb(buf, width-20, height-2, 124, status_b); + +} + +static void +clear(int sx, int ex, int y, uint16_t bg) +{ + /* clear line from to */ + /* x = line number vertical */ + /* y = column number horizontal */ + int i; + for (i = sx; i < ex; i++) { + tb_change_cell(i, y, 0x0000, TB_DEFAULT, bg); + } + +} + +static void +clear_status(void) +{ + int width, height; + width = tb_width(); + height = tb_height(); + clear(1, width-30, height-2, status_b); +} + +static void +clear_error(void) +{ + int width, height; + width = tb_width(); + height = tb_height(); + clear(width-30, width-1, height-2, status_b); +} + +static void +clear_pane(int pane) +{ + int i, x, ex, y, width, height; + width = tb_width(); + height = tb_height(); + y = 0; + i = 0; + + if (pane == 2){ + x = 2; + ex = (width/2) - 1; + } else if (pane == (width/2) + 2) { + x = (width/2) + 1; + ex = width -1; + } + + while (i < height-2) { + clear(x, ex, y, TB_DEFAULT); + i++; + y++; + } + /* draw top line */ + for (y = x; y < ex ; ++y) { + tb_change_cell(y, 0, u_hl, frame_f, frame_b); + } + +} + +static char * +get_extentions(char *str) +{ + char *ext; + char *dot; + size_t counter, len, i; + + dot = "."; + counter = 0; + len = strlen(str); + + for (i = len-1; i > 0; i--) { + if (str[i] == *dot) { + break; + } else { + counter++; + } + } + + ext = ecalloc(counter, sizeof(char)); + strncpy(ext, &str[len-counter], counter); + return ext; +} + +static char * +get_full_path(char *first, char *second) +{ + char *full_path; + size_t full_path_len; + + full_path_len = strlen(first) + strlen(second) + 2; + full_path = ecalloc(full_path_len, sizeof(char)); + + if (strcmp(first, "/") == 0) { + (void)snprintf(full_path, full_path_len, "/%s", second); + + } else { + (void)snprintf( + full_path, full_path_len, "%s/%s", first, second); + } + + return full_path; +} + +static char * +get_parent(char *dir) +{ + char *parent; + char *dot; + size_t counter, len, i; + + dot = "/"; + counter = 0; + len = strlen(dir); + + for (i = len-1; i > 0; i--) { + if (dir[i] == *dot) { + break; + } else { + counter++; + } + } + + i = len - counter - 1; + if (i == 0 ) + i = len - counter; + parent = ecalloc(i+1, sizeof(char)); + strncpy(parent, dir, i); + + return parent; +} + +static char * +get_file_info(Entry *cursor) +{ + char *size; + char *result; + char perm[11]; + size_t size_len = (size_t)9; + size_t perm_len = (size_t)11; + size_t result_chars = size_len + perm_len; + result = ecalloc(result_chars, sizeof(char)); + + size = get_file_size(cursor->size); + get_file_perm(cursor->mode, perm); + + strncpy(result, perm, perm_len); + strcat(result, " "); + strncat(result, size, size_len); + + free(size); + + return result; +} + +static char * +get_file_size(off_t size) +{ + /* need to be freed */ + char *Rsize; + double lsize; + int counter; + counter = 0; + + Rsize = calloc(9, sizeof(char)); + lsize = size; + + while (lsize >= 1000) + { + lsize /= 1024.0; + ++counter; + } + + float_to_string(lsize, Rsize); + + switch (counter) + { + case 0: + strcat(Rsize, "B"); + break; + case 1: + strcat(Rsize, "KB"); + break; + case 2: + strcat(Rsize, "MB"); + break; + case 3: + strcat(Rsize, "GB"); + break; + case 4: + strcat(Rsize, "TB"); + break; + } + + return Rsize; +} + +static char * +get_file_date(struct stat status) +{ + /* need to be freed */ + char *dat; + struct tm lt; + time_t timebuf; + + dat = calloc(9, sizeof(char)); + timebuf = status.st_ctime; + localtime_r(&timebuf, &lt); + strftime(dat, 9, "%d/%m", &lt); + + return dat; +} + +static char * +get_file_time(struct stat status) +{ + /* need to be freed */ + char *tim; + struct tm lt; + time_t timebuf; + + tim = calloc(9, sizeof(char)); + timebuf = status.st_ctime; + localtime_r(&timebuf, &lt); + + strftime(tim, 9, "%I:%M%p", &lt); + + return tim; +} + +static char * +get_file_userowner(struct stat status) +{ + char *user_owner; + struct passwd *pw; + + pw = getpwuid(status.st_uid); + user_owner = pw->pw_name; + + return user_owner; +} + +static char * +get_file_groupowner(struct stat status) +{ + char *group_owner; + struct group *gr; + + gr = getgrgid(status.st_gid); + group_owner = gr->gr_name; + + return group_owner; +} + +static void +get_file_perm(mode_t mode, char *buf) +{ + const char chars[] = "rwxrwxrwx"; + size_t i; + + if(S_ISDIR(mode)) + buf[0] = 'd'; + else if(S_ISREG(mode)) + buf[0] = '-'; + else if(S_ISLNK(mode)) + buf[0] = 'l'; + else if(S_ISBLK(mode)) + buf[0] = 'b'; + else if(S_ISCHR(mode)) + buf[0] = 'c'; + else if(S_ISFIFO(mode)) + buf[0] = 'p'; + else if(S_ISSOCK(mode)) + buf[0] = 's'; + else + buf[0] = '?'; + + for (i = 1; i < 10; i++) { + buf[i] = (mode & (1 << (9-i))) ? chars[i-1] : '-'; + } + buf[10] = '\0'; +} + +static int +check_dir(char *path) +{ + DIR *dir; + dir = opendir(path); + + if (dir == NULL) { + if (errno == ENOTDIR) { + return 1; + } else { + return -1; + } + } + + if (closedir(dir) < 0) + return -1; + + return 0; +} + +static int +open_files(char *filename) +{ + tb_clear(); + // TODO + /* open editor in other window */ + /* wait option */ + char *editor, *file_ex, *software, *term; + int status, r; + pid_t pid; + int wait_finish = 0; + + editor = getenv("EDITOR"); + term = getenv("TERM"); + + file_ex = get_extentions(filename); + + if (strcmp(file_ex, "png") == 0) { + software = images; + } else if (strcmp(file_ex, "mp4") == 0) { + software = videos; + wait_finish = 1; + } else if (strcmp(file_ex, "pdf") == 0) { + software = pdf; + } else { + software = editor; + wait_finish = 1; + } + + free(file_ex); + + if (software == NULL) + term = "vi"; + + char *filex[] = {software, filename, NULL}; + + pid = fork(); + switch (pid) { + case -1: + return -1; + case 0: + (void)execvp(filex[0], filex); + exit(EXIT_SUCCESS); + default: + if (wait_finish == 1) { + while ((r = waitpid(pid, &status, 0)) == -1 && errno == EINTR) + continue; + if (r == -1) + return -1; + if (WIFEXITED(status) && WEXITSTATUS(status) != 0) + return -1; + } + } + + return 0; +} + +static int +sort_name(const void *const A, const void *const B) +{ + int result; + int data1 = (*(Entry*) A).mode; + int data2 = (*(Entry*) B).mode; + + if (data1 < data2) { + return -1; + } else if (data1 == data2) { + result = strcmp((*(Entry*) A).name,(*(Entry*) B).name); + return result; + } else { + return 1; + } +} + +static double +xfloorf(double num) +{ + return (int)num; +} + +static void +float_to_string(float f, char *r) +{ + long long int length, length2, i, number, position, sign, tenth, temp; + float number2; + + f = xfloorf(f * 100) / 100; + + sign = -1; // -1 == positive number + if (f < 0) + { + sign = '-'; + f *= -1; + } + + number2 = f; + number = f; + temp = 1; + length = 0; // Size of decimal part + length2 = 0; // Size of tenth + + /* Calculate length2 tenth part */ + while ((number2 - (float)number) != 0.0 && !((number2 - (float)number) < 0.0)) + { + temp *= 10.0; + tenth = temp; + number2 = f * tenth; + number = number2; + + length2++; + } + + /* Calculate length decimal part */ + for (length = (f > 1) ? 0 : 1; f > 1; length++) + f /= 10; + + position = length; + length = length + 1 + length2; + number = number2; + + if (sign == '-') + { + length++; + position++; + } + + if (length2 > 0) + { + for (i = length; i >= 0; i--) + { + if (i == (length)) + r[i] = '\0'; + else if (i == (position)) + r[i] = '.'; + else if (sign == '-' && i == 0) + r[i] = '-'; + else + { + r[i] = (number % 10) + '0'; + number /= 10; + } + } + } + else + { + length--; + for (i = length; i >= 0; i--) + { + if (i == (length)) + r[i] = '\0'; + else if (sign == '-' && i == 0) + r[i] = '-'; + else + { + r[i] = (number % 10) + '0'; + number /= 10; + } + } + } +} + +static int +get_memory_usage(void) +{ + struct rusage myusage; + getrusage(RUSAGE_SELF, &myusage); + return myusage.ru_maxrss; +} + +static void +print_col(Entry *entry, size_t hdir, size_t x, size_t y, int dyn_y, int width) +{ + uint16_t bg, fg; + char buf[MAX_P]; + bg = file_b; + fg = file_f; + + if (S_ISDIR(entry->mode)) { + bg = dir_b; + fg = dir_f; + } else if (S_ISLNK(entry->mode)) { + realpath(entry->full, buf); + strcat(entry->name, " -> "); + strncat(entry->name, buf, MAX_N - strlen(entry->name)-1); + bg = other_b; + fg = other_f; + } + + /* highlighted (cursor) */ + if (y + dyn_y == hdir) { + bg = bg | TB_REVERSE; + fg = fg | TB_REVERSE | TB_BOLD; + } + + /* print each element in directory */ + printf_tb(x, y, fg, bg, "%*.*s", ~width, width, entry->name); + +} + +static int +listdir(Pane *cpane) +{ + DIR *dir; + struct dirent *entry; + Entry *list; + struct stat status; + size_t y, i; + char *fullpath; + char *fileinfo; + size_t height, dyn_max, dyn_y; + int width; + + height = tb_height() - 2; + width = (tb_width() / 2) - 4; + cpane->dirc = - 2; /* dont't count . .. */ + i = 0; + + dir = opendir(cpane->dirn); + if (dir == NULL) + return -1; + + /* get content sum */ + while ((entry = readdir(dir)) != 0) { + cpane->dirc++; + } + + /* empty directory */ + if (cpane->dirc == 0) { + if (closedir(dir) < 0) + return -1; + + /* print current directory title */ + printf_tb(cpane->dirx, 0, cpane->dir_fg | TB_BOLD, + cpane->dir_bg, " %.*s ", width, cpane->dirn); + return 0; + } + + rewinddir(dir); /* reset position */ + + /* create array of entries */ + i = 0; + list = ecalloc(cpane->dirc, sizeof(Entry)); + while ((entry = readdir(dir)) != 0) { + if ((strcmp(entry->d_name, ".") == 0 || + strcmp(entry->d_name, "..") == 0 )) + continue; + + strcpy(list[i].name, entry->d_name); + fullpath = get_full_path(cpane->dirn, entry->d_name); + strcpy(list[i].full, fullpath); + + if (lstat(fullpath, &status) == 0) { + list[i].size = status.st_size; + list[i].mode = status.st_mode; + } + + free(fullpath); + + i++; + } + + qsort(list, cpane->dirc, sizeof(Entry), sort_name); + + i = 0; + y = 1; + dyn_y = 0; + + /* scroll */ + if (cpane->dirc > height - 1) { + clear_pane(cpane->dirx); + if (cpane->hdir >= height - 4){ + i = (cpane->hdir - height) + 4; + dyn_y = i; + } + } + + dyn_max = MIN(cpane->dirc, height-1+i); + + /* get full path of cursor */ + fullpath = get_full_path(cpane->dirn, + list[cpane->hdir-1].name); + strncpy(cpane->high_dir, fullpath, (size_t)MAX_P); + free(fullpath); + + /* print each entry in directory */ + while (i < dyn_max) { + print_col(&list[i], cpane->hdir, + cpane->dirx, y, dyn_y, width); + i++; + y++; + } + + fileinfo = get_file_info(&list[cpane->hdir-1]); + + /* print info in statusbar */ + print_status("%lu/%lu %s", + cpane->hdir, + cpane->dirc, + fileinfo); + + /* print current directory title */ + printf_tb(cpane->dirx, 0, cpane->dir_fg | TB_BOLD, cpane->dir_bg, + " %.*s ", width, cpane->dirn); + + free(fileinfo); + free(list); + + if (closedir(dir) < 0) + return -1; + + print_error("mem (%d)", get_memory_usage()); + + return 0; +} + +static void +press(struct tb_event *ev, Pane *cpane, Pane *opane) +{ + char *parent; + int isdir; + int ret; + clear_error(); + + switch (ev->ch) { + case 'j': + if (cpane->hdir < cpane->dirc) { + cpane->hdir++; + (void)listdir(cpane); + } + break; + case 'k': + if (cpane->hdir > 1) { + cpane->hdir--; + (void)listdir(cpane); + } + break; + case 'h': + parent = get_parent(cpane->dirn); + if (check_dir(parent) < 0) { /* failed to open directory */ + print_error("%s", strerror(errno)); + } else { + strcpy(cpane->dirn, parent); + clear_pane(cpane->dirx); + cpane->hdir = parent_row; + (void)listdir(cpane); + parent_row = 1; + } + free(parent); + + break; + case 'l': + isdir = check_dir(cpane->high_dir); + switch (isdir) { + case 0: + strcpy(cpane->dirn, cpane->high_dir); + clear_pane(cpane->dirx); + parent_row = cpane->hdir; + cpane->hdir = 1; + (void)listdir(cpane); + break; + case 1: + /* is not a directory open file */ + if (open_files(cpane->high_dir) < 0) { + print_error("%s", strerror(errno)); + } + tb_shutdown(); + ret = tb_init(); + if (ret != 0) + die("tb_init() %d\n", ret); + if (cpane->dirx == 2) /* if current left pane */ + t_resize(cpane, opane); + else + t_resize(opane, cpane); + + break; + case -1: + /* failed to open directory */ + print_error("%s", strerror(errno)); + } + + break; + case 'g': + cpane->hdir = 1; + (void)listdir(cpane); + break; + case 'G': + cpane->hdir = cpane->dirc; + (void)listdir(cpane); + break; + case 'M': + cpane->hdir = (cpane->dirc/2); + (void)listdir(cpane); + break; + } +} + +static void +t_resize(Pane *cpane, Pane *opane) +{ + tb_clear(); + draw_frame(); + (void)set_panes(cpane, opane, 1); + (void)listdir(cpane); + (void)listdir(opane); + tb_present(); + +} + +static int +set_panes(Pane *pane_l, Pane *pane_r, int resize) +{ + int height, width; + char *home; + char cwd[MAX_P]; + + (void)getcwd(cwd, sizeof(cwd)); + home = getenv("HOME"); + width = tb_width(); + height = tb_height(); + + if (home == NULL) + home = "/"; + + pane_l->dirx = 2; + pane_l->dir_fg = pane_l_f; + pane_l->dir_bg = pane_l_b; + if (resize == 0) { + strcpy(pane_l->dirn, cwd); + pane_l->hdir = 1; + } + + pane_r->dirx = (width / 2) + 2; + pane_r->dir_fg = pane_r_f; + pane_r->dir_bg = pane_r_b; + + if (resize == 0) { + strcpy(pane_r->dirn, home); + pane_r->hdir = 1; + } + + return 0; +} + +static void +draw_frame(void) +{ + int height, width, i; + + width = tb_width(); + height = tb_height(); + + /* 2 horizontal lines */ + for (i = 1; i < width-1 ; ++i) { + tb_change_cell(i, 0, u_hl, frame_f, frame_b); + tb_change_cell(i, height-1, u_hl, frame_f, frame_b); + } + + /* 3 vertical lines */ + for (i = 1; i < height-1 ; ++i) { + tb_change_cell(0, i, u_vl, frame_f, frame_b); + tb_change_cell((width-1)/2, i-1, u_vl, frame_f, frame_b); + tb_change_cell(width-1, i, u_vl, frame_f, frame_b); + } + + /* 4 corners */ + tb_change_cell(0, 0, u_cnw, frame_f, frame_b); + tb_change_cell(width-1, 0, u_cne, frame_f, frame_b); + tb_change_cell(0, height-1, u_csw, frame_f, frame_b); + tb_change_cell(width-1, height-1, u_cse, frame_f, frame_b); + + /* 2 middel top and bottom */ + tb_change_cell((width-1)/2, 0, u_mn, frame_f, frame_b); + +} + +static int +start(void) +{ + int ret, mode, support_256, support_n, init_height, init_width; + struct tb_event ev; + Pane pane_r, pane_l; + int current_pane = 0; + + ret = tb_init(); + if (ret != 0) + die("tb_init() failed with error code %d\n", ret); + + init_width = tb_width(); + init_height = tb_height(); + mode = tb_select_input_mode(TB_INPUT_ESC); + support_256 = tb_select_output_mode(TB_OUTPUT_256); + + if (support_256 != TB_OUTPUT_256) + support_n = tb_select_output_mode(TB_OUTPUT_NORMAL); + + tb_clear(); + draw_frame(); + (void)set_panes(&pane_l, &pane_r, 0); + (void)listdir(&pane_r); + (void)listdir(&pane_l); + tb_present(); + + while (tb_poll_event(&ev) != 0) { + switch (ev.type) { + case TB_EVENT_KEY: + if (ev.ch == 'q') { + tb_shutdown(); + return 0; + } else if (ev.key == TB_KEY_SPACE) { + current_pane = !current_pane; + } + + if (current_pane == 0) { + press(&ev, &pane_l, &pane_r); + + } else if (current_pane == 1) { + press(&ev, &pane_r, &pane_l); + } + + tb_present(); + break; + + case TB_EVENT_RESIZE: + t_resize(&pane_l, &pane_r); + break; + + default: + break; + } + } + + tb_shutdown(); + return 0; +} + +int +main(int argc, char *argv[]) +{ +#ifdef __OpenBSD__ + if (pledge("stdio tty rpath proc exec", NULL) == -1) + die("pledge"); +#endif /* __OpenBSD__ */ + if (argc == 1) { + if (start() != 0) + die("start failed"); + } else if ( + argc == 2 && strlen(argv[1]) == (size_t)2 && + strcmp("-v", argv[1]) == 0) { + die("sfm-"VERSION); + } else { + die("usage: sfm [-v]"); + } + return 0; +} diff --git a/sfm.png b/sfm.png Binary files differ. diff --git a/term.inl b/term.inl @@ -0,0 +1,311 @@ +enum { + T_ENTER_CA, + T_EXIT_CA, + T_SHOW_CURSOR, + T_HIDE_CURSOR, + T_CLEAR_SCREEN, + T_SGR0, + T_UNDERLINE, + T_BOLD, + T_BLINK, + T_REVERSE, + T_ENTER_KEYPAD, + T_EXIT_KEYPAD, + T_ENTER_MOUSE, + T_EXIT_MOUSE, + T_FUNCS_NUM, +}; + +#define ENTER_MOUSE_SEQ "\x1b[?1000h\x1b[?1002h\x1b[?1015h\x1b[?1006h" +#define EXIT_MOUSE_SEQ "\x1b[?1006l\x1b[?1015l\x1b[?1002l\x1b[?1000l" + +#define EUNSUPPORTED_TERM -1 + +// rxvt-256color +static const char *rxvt_256color_keys[] = { + "\033[11~","\033[12~","\033[13~","\033[14~","\033[15~","\033[17~","\033[18~","\033[19~","\033[20~","\033[21~","\033[23~","\033[24~","\033[2~","\033[3~","\033[7~","\033[8~","\033[5~","\033[6~","\033[A","\033[B","\033[D","\033[C", 0 +}; +static const char *rxvt_256color_funcs[] = { + "\0337\033[?47h", "\033[2J\033[?47l\0338", "\033[?25h", "\033[?25l", "\033[H\033[2J", "\033[m", "\033[4m", "\033[1m", "\033[5m", "\033[7m", "\033=", "\033>", ENTER_MOUSE_SEQ, EXIT_MOUSE_SEQ, +}; + +// Eterm +static const char *eterm_keys[] = { + "\033[11~","\033[12~","\033[13~","\033[14~","\033[15~","\033[17~","\033[18~","\033[19~","\033[20~","\033[21~","\033[23~","\033[24~","\033[2~","\033[3~","\033[7~","\033[8~","\033[5~","\033[6~","\033[A","\033[B","\033[D","\033[C", 0 +}; +static const char *eterm_funcs[] = { + "\0337\033[?47h", "\033[2J\033[?47l\0338", "\033[?25h", "\033[?25l", "\033[H\033[2J", "\033[m", "\033[4m", "\033[1m", "\033[5m", "\033[7m", "", "", "", "", +}; + +// screen +static const char *screen_keys[] = { + "\033OP","\033OQ","\033OR","\033OS","\033[15~","\033[17~","\033[18~","\033[19~","\033[20~","\033[21~","\033[23~","\033[24~","\033[2~","\033[3~","\033[1~","\033[4~","\033[5~","\033[6~","\033OA","\033OB","\033OD","\033OC", 0 +}; +static const char *screen_funcs[] = { + "\033[?1049h", "\033[?1049l", "\033[34h\033[?25h", "\033[?25l", "\033[H\033[J", "\033[m", "\033[4m", "\033[1m", "\033[5m", "\033[7m", "\033[?1h\033=", "\033[?1l\033>", ENTER_MOUSE_SEQ, EXIT_MOUSE_SEQ, +}; + +// rxvt-unicode +static const char *rxvt_unicode_keys[] = { + "\033[11~","\033[12~","\033[13~","\033[14~","\033[15~","\033[17~","\033[18~","\033[19~","\033[20~","\033[21~","\033[23~","\033[24~","\033[2~","\033[3~","\033[7~","\033[8~","\033[5~","\033[6~","\033[A","\033[B","\033[D","\033[C", 0 +}; +static const char *rxvt_unicode_funcs[] = { + "\033[?1049h", "\033[r\033[?1049l", "\033[?25h", "\033[?25l", "\033[H\033[2J", "\033[m\033(B", "\033[4m", "\033[1m", "\033[5m", "\033[7m", "\033=", "\033>", ENTER_MOUSE_SEQ, EXIT_MOUSE_SEQ, +}; + +// linux +static const char *linux_keys[] = { + "\033[[A","\033[[B","\033[[C","\033[[D","\033[[E","\033[17~","\033[18~","\033[19~","\033[20~","\033[21~","\033[23~","\033[24~","\033[2~","\033[3~","\033[1~","\033[4~","\033[5~","\033[6~","\033[A","\033[B","\033[D","\033[C", 0 +}; +static const char *linux_funcs[] = { + "", "", "\033[?25h\033[?0c", "\033[?25l\033[?1c", "\033[H\033[J", "\033[0;10m", "\033[4m", "\033[1m", "\033[5m", "\033[7m", "", "", "", "", +}; + +// xterm +static const char *xterm_keys[] = { + "\033OP","\033OQ","\033OR","\033OS","\033[15~","\033[17~","\033[18~","\033[19~","\033[20~","\033[21~","\033[23~","\033[24~","\033[2~","\033[3~","\033OH","\033OF","\033[5~","\033[6~","\033OA","\033OB","\033OD","\033OC", 0 +}; +static const char *xterm_funcs[] = { + "\033[?1049h", "\033[?1049l", "\033[?12l\033[?25h", "\033[?25l", "\033[H\033[2J", "\033(B\033[m", "\033[4m", "\033[1m", "\033[5m", "\033[7m", "\033[?1h\033=", "\033[?1l\033>", ENTER_MOUSE_SEQ, EXIT_MOUSE_SEQ, +}; + +static struct term { + const char *name; + const char **keys; + const char **funcs; +} terms[] = { + {"rxvt-256color", rxvt_256color_keys, rxvt_256color_funcs}, + {"Eterm", eterm_keys, eterm_funcs}, + {"screen", screen_keys, screen_funcs}, + {"rxvt-unicode", rxvt_unicode_keys, rxvt_unicode_funcs}, + {"linux", linux_keys, linux_funcs}, + {"xterm", xterm_keys, xterm_funcs}, + {0, 0, 0}, +}; + +static bool init_from_terminfo = false; +static const char **keys; +static const char **funcs; + +static int try_compatible(const char *term, const char *name, + const char **tkeys, const char **tfuncs) +{ + if (strstr(term, name)) { + keys = tkeys; + funcs = tfuncs; + return 0; + } + + return EUNSUPPORTED_TERM; +} + +static int init_term_builtin(void) +{ + int i; + const char *term = getenv("TERM"); + + if (term) { + for (i = 0; terms[i].name; i++) { + if (!strcmp(terms[i].name, term)) { + keys = terms[i].keys; + funcs = terms[i].funcs; + return 0; + } + } + + /* let's do some heuristic, maybe it's a compatible terminal */ + if (try_compatible(term, "xterm", xterm_keys, xterm_funcs) == 0) + return 0; + if (try_compatible(term, "rxvt", rxvt_unicode_keys, rxvt_unicode_funcs) == 0) + return 0; + if (try_compatible(term, "linux", linux_keys, linux_funcs) == 0) + return 0; + if (try_compatible(term, "Eterm", eterm_keys, eterm_funcs) == 0) + return 0; + if (try_compatible(term, "screen", screen_keys, screen_funcs) == 0) + return 0; + if (try_compatible(term, "tmux", screen_keys, screen_funcs) == 0) + return 0; + /* let's assume that 'cygwin' is xterm compatible */ + if (try_compatible(term, "cygwin", xterm_keys, xterm_funcs) == 0) + return 0; + } + + return EUNSUPPORTED_TERM; +} + +//---------------------------------------------------------------------- +// terminfo +//---------------------------------------------------------------------- + +static char *read_file(const char *file) { + FILE *f = fopen(file, "rb"); + if (!f) + return 0; + + struct stat st; + if (fstat(fileno(f), &st) != 0) { + fclose(f); + return 0; + } + + char *data = malloc(st.st_size); + if (!data) { + fclose(f); + return 0; + } + + if (fread(data, 1, st.st_size, f) != (size_t)st.st_size) { + fclose(f); + free(data); + return 0; + } + + fclose(f); + return data; +} + +static char *terminfo_try_path(const char *path, const char *term) { + char tmp[4096]; + snprintf(tmp, sizeof(tmp), "%s/%c/%s", path, term[0], term); + tmp[sizeof(tmp)-1] = '\0'; + char *data = read_file(tmp); + if (data) { + return data; + } + + // fallback to darwin specific dirs structure + snprintf(tmp, sizeof(tmp), "%s/%x/%s", path, term[0], term); + tmp[sizeof(tmp)-1] = '\0'; + return read_file(tmp); +} + +static char *load_terminfo(void) { + char tmp[4096]; + const char *term = getenv("TERM"); + if (!term) { + return 0; + } + + // if TERMINFO is set, no other directory should be searched + const char *terminfo = getenv("TERMINFO"); + if (terminfo) { + return terminfo_try_path(terminfo, term); + } + + // next, consider ~/.terminfo + const char *home = getenv("HOME"); + if (home) { + snprintf(tmp, sizeof(tmp), "%s/.terminfo", home); + tmp[sizeof(tmp)-1] = '\0'; + char *data = terminfo_try_path(tmp, term); + if (data) + return data; + } + + // next, TERMINFO_DIRS + const char *dirs = getenv("TERMINFO_DIRS"); + if (dirs) { + snprintf(tmp, sizeof(tmp), "%s", dirs); + tmp[sizeof(tmp)-1] = '\0'; + char *dir = strtok(tmp, ":"); + while (dir) { + const char *cdir = dir; + if (strcmp(cdir, "") == 0) { + cdir = "/usr/share/terminfo"; + } + char *data = terminfo_try_path(cdir, term); + if (data) + return data; + dir = strtok(0, ":"); + } + } + + // fallback to /usr/share/terminfo + return terminfo_try_path("/usr/share/terminfo", term); +} + +#define TI_MAGIC 0432 +#define TI_ALT_MAGIC 542 +#define TI_HEADER_LENGTH 12 +#define TB_KEYS_NUM 22 + +static const char *terminfo_copy_string(char *data, int str, int table) { + const int16_t off = *(int16_t*)(data + str); + const char *src = data + table + off; + int len = strlen(src); + char *dst = malloc(len+1); + strcpy(dst, src); + return dst; +} + +static const int16_t ti_funcs[] = { + 28, 40, 16, 13, 5, 39, 36, 27, 26, 34, 89, 88, +}; + +static const int16_t ti_keys[] = { + 66, 68 /* apparently not a typo; 67 is F10 for whatever reason */, 69, + 70, 71, 72, 73, 74, 75, 67, 216, 217, 77, 59, 76, 164, 82, 81, 87, 61, + 79, 83, +}; + +static int init_term(void) { + int i; + char *data = load_terminfo(); + if (!data) { + init_from_terminfo = false; + return init_term_builtin(); + } + + int16_t *header = (int16_t*)data; + + const int number_sec_len = header[0] == TI_ALT_MAGIC ? 4 : 2; + + if ((header[1] + header[2]) % 2) { + // old quirk to align everything on word boundaries + header[2] += 1; + } + + const int str_offset = TI_HEADER_LENGTH + + header[1] + header[2] + number_sec_len * header[3]; + const int table_offset = str_offset + 2 * header[4]; + + keys = malloc(sizeof(const char*) * (TB_KEYS_NUM+1)); + for (i = 0; i < TB_KEYS_NUM; i++) { + keys[i] = terminfo_copy_string(data, + str_offset + 2 * ti_keys[i], table_offset); + } + keys[TB_KEYS_NUM] = 0; + + funcs = malloc(sizeof(const char*) * T_FUNCS_NUM); + // the last two entries are reserved for mouse. because the table offset is + // not there, the two entries have to fill in manually + for (i = 0; i < T_FUNCS_NUM-2; i++) { + funcs[i] = terminfo_copy_string(data, + str_offset + 2 * ti_funcs[i], table_offset); + } + + funcs[T_FUNCS_NUM-2] = ENTER_MOUSE_SEQ; + funcs[T_FUNCS_NUM-1] = EXIT_MOUSE_SEQ; + + init_from_terminfo = true; + free(data); + return 0; +} + +static void shutdown_term(void) { + if (init_from_terminfo) { + int i; + for (i = 0; i < TB_KEYS_NUM; i++) { + free((void*)keys[i]); + } + // the last two entries are reserved for mouse. because the table offset + // is not there, the two entries have to fill in manually and do not + // need to be freed. + for (i = 0; i < T_FUNCS_NUM-2; i++) { + free((void*)funcs[i]); + } + free(keys); + free(funcs); + } +} diff --git a/termbox.c b/termbox.c @@ -0,0 +1,678 @@ +#include <assert.h> +#include <stdlib.h> +#include <string.h> +#include <errno.h> +#include <fcntl.h> +#include <signal.h> +#include <stdio.h> +#include <stdbool.h> +#include <sys/select.h> +#include <sys/ioctl.h> +#include <sys/time.h> +#include <sys/stat.h> +#include <termios.h> +#include <unistd.h> +#include <wchar.h> + +#include "termbox.h" + +#include "bytebuffer.inl" +#include "term.inl" +#include "input.inl" + +struct cellbuf { + int width; + int height; + struct tb_cell *cells; +}; + +#define CELL(buf, x, y) (buf)->cells[(y) * (buf)->width + (x)] +#define IS_CURSOR_HIDDEN(cx, cy) (cx == -1 || cy == -1) +#define LAST_COORD_INIT -1 + +static struct termios orig_tios; + +static struct cellbuf back_buffer; +static struct cellbuf front_buffer; +static struct bytebuffer output_buffer; +static struct bytebuffer input_buffer; + +static int termw = -1; +static int termh = -1; + +static int inputmode = TB_INPUT_ESC; +static int outputmode = TB_OUTPUT_NORMAL; + +static int inout; +static int winch_fds[2]; + +static int lastx = LAST_COORD_INIT; +static int lasty = LAST_COORD_INIT; +static int cursor_x = -1; +static int cursor_y = -1; + +static uint16_t background = TB_DEFAULT; +static uint16_t foreground = TB_DEFAULT; + +static void write_cursor(int x, int y); +static void write_sgr(uint16_t fg, uint16_t bg); + +static void cellbuf_init(struct cellbuf *buf, int width, int height); +static void cellbuf_resize(struct cellbuf *buf, int width, int height); +static void cellbuf_clear(struct cellbuf *buf); +static void cellbuf_free(struct cellbuf *buf); + +static void update_size(void); +static void update_term_size(void); +static void send_attr(uint16_t fg, uint16_t bg); +static void send_char(int x, int y, uint32_t c); +static void send_clear(void); +static void sigwinch_handler(int xxx); +static int wait_fill_event(struct tb_event *event, struct timeval *timeout); + +/* may happen in a different thread */ +static volatile int buffer_size_change_request; + +/* -------------------------------------------------------- */ + +int tb_init_fd(int inout_) +{ + inout = inout_; + if (inout == -1) { + return TB_EFAILED_TO_OPEN_TTY; + } + + if (init_term() < 0) { + close(inout); + return TB_EUNSUPPORTED_TERMINAL; + } + + if (pipe(winch_fds) < 0) { + close(inout); + return TB_EPIPE_TRAP_ERROR; + } + + struct sigaction sa; + memset(&sa, 0, sizeof(sa)); + sa.sa_handler = sigwinch_handler; + sa.sa_flags = 0; + sigaction(SIGWINCH, &sa, 0); + + tcgetattr(inout, &orig_tios); + + struct termios tios; + memcpy(&tios, &orig_tios, sizeof(tios)); + + tios.c_iflag &= ~(IGNBRK | BRKINT | PARMRK | ISTRIP + | INLCR | IGNCR | ICRNL | IXON); + tios.c_oflag &= ~OPOST; + tios.c_lflag &= ~(ECHO | ECHONL | ICANON | ISIG | IEXTEN); + tios.c_cflag &= ~(CSIZE | PARENB); + tios.c_cflag |= CS8; + tios.c_cc[VMIN] = 0; + tios.c_cc[VTIME] = 0; + tcsetattr(inout, TCSAFLUSH, &tios); + + bytebuffer_init(&input_buffer, 128); + bytebuffer_init(&output_buffer, 32 * 1024); + + bytebuffer_puts(&output_buffer, funcs[T_ENTER_CA]); + bytebuffer_puts(&output_buffer, funcs[T_ENTER_KEYPAD]); + bytebuffer_puts(&output_buffer, funcs[T_HIDE_CURSOR]); + send_clear(); + + update_term_size(); + cellbuf_init(&back_buffer, termw, termh); + cellbuf_init(&front_buffer, termw, termh); + cellbuf_clear(&back_buffer); + cellbuf_clear(&front_buffer); + + return 0; +} + +int tb_init_file(const char* name){ + return tb_init_fd(open(name, O_RDWR)); +} + +int tb_init(void) +{ + return tb_init_file("/dev/tty"); +} + +void tb_shutdown(void) +{ + if (termw == -1) { + fputs("tb_shutdown() should not be called twice.", stderr); + abort(); + } + + bytebuffer_puts(&output_buffer, funcs[T_SHOW_CURSOR]); + bytebuffer_puts(&output_buffer, funcs[T_SGR0]); + bytebuffer_puts(&output_buffer, funcs[T_CLEAR_SCREEN]); + bytebuffer_puts(&output_buffer, funcs[T_EXIT_CA]); + bytebuffer_puts(&output_buffer, funcs[T_EXIT_KEYPAD]); + bytebuffer_puts(&output_buffer, funcs[T_EXIT_MOUSE]); + bytebuffer_flush(&output_buffer, inout); + tcsetattr(inout, TCSAFLUSH, &orig_tios); + + shutdown_term(); + close(inout); + close(winch_fds[0]); + close(winch_fds[1]); + + cellbuf_free(&back_buffer); + cellbuf_free(&front_buffer); + bytebuffer_free(&output_buffer); + bytebuffer_free(&input_buffer); + termw = termh = -1; +} + +void tb_present(void) +{ + int x,y,w,i; + struct tb_cell *back, *front; + + /* invalidate cursor position */ + lastx = LAST_COORD_INIT; + lasty = LAST_COORD_INIT; + + if (buffer_size_change_request) { + update_size(); + buffer_size_change_request = 0; + } + + for (y = 0; y < front_buffer.height; ++y) { + for (x = 0; x < front_buffer.width; ) { + back = &CELL(&back_buffer, x, y); + front = &CELL(&front_buffer, x, y); + w = wcwidth(back->ch); + if (w < 1) w = 1; + if (memcmp(back, front, sizeof(struct tb_cell)) == 0) { + x += w; + continue; + } + memcpy(front, back, sizeof(struct tb_cell)); + send_attr(back->fg, back->bg); + if (w > 1 && x >= front_buffer.width - (w - 1)) { + // Not enough room for wide ch, so send spaces + for (i = x; i < front_buffer.width; ++i) { + send_char(i, y, ' '); + } + } else { + send_char(x, y, back->ch); + for (i = 1; i < w; ++i) { + front = &CELL(&front_buffer, x + i, y); + front->ch = 0; + front->fg = back->fg; + front->bg = back->bg; + } + } + x += w; + } + } + if (!IS_CURSOR_HIDDEN(cursor_x, cursor_y)) + write_cursor(cursor_x, cursor_y); + bytebuffer_flush(&output_buffer, inout); +} + +void tb_set_cursor(int cx, int cy) +{ + if (IS_CURSOR_HIDDEN(cursor_x, cursor_y) && !IS_CURSOR_HIDDEN(cx, cy)) + bytebuffer_puts(&output_buffer, funcs[T_SHOW_CURSOR]); + + if (!IS_CURSOR_HIDDEN(cursor_x, cursor_y) && IS_CURSOR_HIDDEN(cx, cy)) + bytebuffer_puts(&output_buffer, funcs[T_HIDE_CURSOR]); + + cursor_x = cx; + cursor_y = cy; + if (!IS_CURSOR_HIDDEN(cursor_x, cursor_y)) + write_cursor(cursor_x, cursor_y); +} + +void tb_put_cell(int x, int y, const struct tb_cell *cell) +{ + if ((unsigned)x >= (unsigned)back_buffer.width) + return; + if ((unsigned)y >= (unsigned)back_buffer.height) + return; + CELL(&back_buffer, x, y) = *cell; +} + +void tb_change_cell(int x, int y, uint32_t ch, uint16_t fg, uint16_t bg) +{ + struct tb_cell c = {ch, fg, bg}; + tb_put_cell(x, y, &c); +} + +void tb_blit(int x, int y, int w, int h, const struct tb_cell *cells) +{ + if (x + w < 0 || x >= back_buffer.width) + return; + if (y + h < 0 || y >= back_buffer.height) + return; + int xo = 0, yo = 0, ww = w, hh = h; + if (x < 0) { + xo = -x; + ww -= xo; + x = 0; + } + if (y < 0) { + yo = -y; + hh -= yo; + y = 0; + } + if (ww > back_buffer.width - x) + ww = back_buffer.width - x; + if (hh > back_buffer.height - y) + hh = back_buffer.height - y; + + int sy; + struct tb_cell *dst = &CELL(&back_buffer, x, y); + const struct tb_cell *src = cells + yo * w + xo; + size_t size = sizeof(struct tb_cell) * ww; + + for (sy = 0; sy < hh; ++sy) { + memcpy(dst, src, size); + dst += back_buffer.width; + src += w; + } +} + +struct tb_cell *tb_cell_buffer(void) +{ + return back_buffer.cells; +} + +int tb_poll_event(struct tb_event *event) +{ + return wait_fill_event(event, 0); +} + +int tb_peek_event(struct tb_event *event, int timeout) +{ + struct timeval tv; + tv.tv_sec = timeout / 1000; + tv.tv_usec = (timeout - (tv.tv_sec * 1000)) * 1000; + return wait_fill_event(event, &tv); +} + +int tb_width(void) +{ + return termw; +} + +int tb_height(void) +{ + return termh; +} + +void tb_clear(void) +{ + if (buffer_size_change_request) { + update_size(); + buffer_size_change_request = 0; + } + cellbuf_clear(&back_buffer); +} + +int tb_select_input_mode(int mode) +{ + if (mode) { + if ((mode & (TB_INPUT_ESC | TB_INPUT_ALT)) == 0) + mode |= TB_INPUT_ESC; + + /* technically termbox can handle that, but let's be nice and show here + what mode is actually used */ + if ((mode & (TB_INPUT_ESC | TB_INPUT_ALT)) == (TB_INPUT_ESC | TB_INPUT_ALT)) + mode &= ~TB_INPUT_ALT; + + inputmode = mode; + if (mode&TB_INPUT_MOUSE) { + bytebuffer_puts(&output_buffer, funcs[T_ENTER_MOUSE]); + bytebuffer_flush(&output_buffer, inout); + } else { + bytebuffer_puts(&output_buffer, funcs[T_EXIT_MOUSE]); + bytebuffer_flush(&output_buffer, inout); + } + } + return inputmode; +} + +int tb_select_output_mode(int mode) +{ + if (mode) + outputmode = mode; + return outputmode; +} + +void tb_set_clear_attributes(uint16_t fg, uint16_t bg) +{ + foreground = fg; + background = bg; +} + +/* -------------------------------------------------------- */ + +static int convertnum(uint32_t num, char* buf) { + int i, l = 0; + int ch; + do { + buf[l++] = '0' + (num % 10); + num /= 10; + } while (num); + for(i = 0; i < l / 2; i++) { + ch = buf[i]; + buf[i] = buf[l - 1 - i]; + buf[l - 1 - i] = ch; + } + return l; +} + +#define WRITE_LITERAL(X) bytebuffer_append(&output_buffer, (X), sizeof(X)-1) +#define WRITE_INT(X) bytebuffer_append(&output_buffer, buf, convertnum((X), buf)) + +static void write_cursor(int x, int y) { + char buf[32]; + WRITE_LITERAL("\033["); + WRITE_INT(y+1); + WRITE_LITERAL(";"); + WRITE_INT(x+1); + WRITE_LITERAL("H"); +} + +static void write_sgr(uint16_t fg, uint16_t bg) { + char buf[32]; + + if (fg == TB_DEFAULT && bg == TB_DEFAULT) + return; + + switch (outputmode) { + case TB_OUTPUT_256: + case TB_OUTPUT_216: + case TB_OUTPUT_GRAYSCALE: + WRITE_LITERAL("\033["); + if (fg != TB_DEFAULT) { + WRITE_LITERAL("38;5;"); + WRITE_INT(fg); + if (bg != TB_DEFAULT) { + WRITE_LITERAL(";"); + } + } + if (bg != TB_DEFAULT) { + WRITE_LITERAL("48;5;"); + WRITE_INT(bg); + } + WRITE_LITERAL("m"); + break; + case TB_OUTPUT_NORMAL: + default: + WRITE_LITERAL("\033["); + if (fg != TB_DEFAULT) { + WRITE_LITERAL("3"); + WRITE_INT(fg - 1); + if (bg != TB_DEFAULT) { + WRITE_LITERAL(";"); + } + } + if (bg != TB_DEFAULT) { + WRITE_LITERAL("4"); + WRITE_INT(bg - 1); + } + WRITE_LITERAL("m"); + break; + } +} + +static void cellbuf_init(struct cellbuf *buf, int width, int height) +{ + buf->cells = (struct tb_cell*)malloc(sizeof(struct tb_cell) * width * height); + assert(buf->cells); + buf->width = width; + buf->height = height; +} + +static void cellbuf_resize(struct cellbuf *buf, int width, int height) +{ + if (buf->width == width && buf->height == height) + return; + + int oldw = buf->width; + int oldh = buf->height; + struct tb_cell *oldcells = buf->cells; + + cellbuf_init(buf, width, height); + cellbuf_clear(buf); + + int minw = (width < oldw) ? width : oldw; + int minh = (height < oldh) ? height : oldh; + int i; + + for (i = 0; i < minh; ++i) { + struct tb_cell *csrc = oldcells + (i * oldw); + struct tb_cell *cdst = buf->cells + (i * width); + memcpy(cdst, csrc, sizeof(struct tb_cell) * minw); + } + + free(oldcells); +} + +static void cellbuf_clear(struct cellbuf *buf) +{ + int i; + int ncells = buf->width * buf->height; + + for (i = 0; i < ncells; ++i) { + buf->cells[i].ch = ' '; + buf->cells[i].fg = foreground; + buf->cells[i].bg = background; + } +} + +static void cellbuf_free(struct cellbuf *buf) +{ + free(buf->cells); +} + +static void get_term_size(int *w, int *h) +{ + struct winsize sz; + memset(&sz, 0, sizeof(sz)); + + ioctl(inout, TIOCGWINSZ, &sz); + + if (w) *w = sz.ws_col; + if (h) *h = sz.ws_row; +} + +static void update_term_size(void) +{ + struct winsize sz; + memset(&sz, 0, sizeof(sz)); + + ioctl(inout, TIOCGWINSZ, &sz); + + termw = sz.ws_col; + termh = sz.ws_row; +} + +static void send_attr(uint16_t fg, uint16_t bg) +{ +#define LAST_ATTR_INIT 0xFFFF + static uint16_t lastfg = LAST_ATTR_INIT, lastbg = LAST_ATTR_INIT; + if (fg != lastfg || bg != lastbg) { + bytebuffer_puts(&output_buffer, funcs[T_SGR0]); + + uint16_t fgcol; + uint16_t bgcol; + + switch (outputmode) { + case TB_OUTPUT_256: + fgcol = fg & 0xFF; + bgcol = bg & 0xFF; + break; + + case TB_OUTPUT_216: + fgcol = fg & 0xFF; if (fgcol > 215) fgcol = 7; + bgcol = bg & 0xFF; if (bgcol > 215) bgcol = 0; + fgcol += 0x10; + bgcol += 0x10; + break; + + case TB_OUTPUT_GRAYSCALE: + fgcol = fg & 0xFF; if (fgcol > 23) fgcol = 23; + bgcol = bg & 0xFF; if (bgcol > 23) bgcol = 0; + fgcol += 0xe8; + bgcol += 0xe8; + break; + + case TB_OUTPUT_NORMAL: + default: + fgcol = fg & 0x0F; + bgcol = bg & 0x0F; + } + + if (fg & TB_BOLD) + bytebuffer_puts(&output_buffer, funcs[T_BOLD]); + if (bg & TB_BOLD) + bytebuffer_puts(&output_buffer, funcs[T_BLINK]); + if (fg & TB_UNDERLINE) + bytebuffer_puts(&output_buffer, funcs[T_UNDERLINE]); + if ((fg & TB_REVERSE) || (bg & TB_REVERSE)) + bytebuffer_puts(&output_buffer, funcs[T_REVERSE]); + + write_sgr(fgcol, bgcol); + + lastfg = fg; + lastbg = bg; + } +} + +static void send_char(int x, int y, uint32_t c) +{ + char buf[7]; + int bw = tb_utf8_unicode_to_char(buf, c); + if (x-1 != lastx || y != lasty) + write_cursor(x, y); + lastx = x; lasty = y; + if(!c) buf[0] = ' '; // replace 0 with whitespace + bytebuffer_append(&output_buffer, buf, bw); +} + +static void send_clear(void) +{ + send_attr(foreground, background); + bytebuffer_puts(&output_buffer, funcs[T_CLEAR_SCREEN]); + if (!IS_CURSOR_HIDDEN(cursor_x, cursor_y)) + write_cursor(cursor_x, cursor_y); + bytebuffer_flush(&output_buffer, inout); + + /* we need to invalidate cursor position too and these two vars are + * used only for simple cursor positioning optimization, cursor + * actually may be in the correct place, but we simply discard + * optimization once and it gives us simple solution for the case when + * cursor moved */ + lastx = LAST_COORD_INIT; + lasty = LAST_COORD_INIT; +} + +static void sigwinch_handler(int xxx) +{ + (void) xxx; + const int zzz = 1; + write(winch_fds[1], &zzz, sizeof(int)); +} + +static void update_size(void) +{ + update_term_size(); + cellbuf_resize(&back_buffer, termw, termh); + cellbuf_resize(&front_buffer, termw, termh); + cellbuf_clear(&front_buffer); + send_clear(); +} + +static int read_up_to(int n) { + assert(n > 0); + const int prevlen = input_buffer.len; + bytebuffer_resize(&input_buffer, prevlen + n); + + int read_n = 0; + while (read_n <= n) { + ssize_t r = 0; + if (read_n < n) { + r = read(inout, input_buffer.buf + prevlen + read_n, n - read_n); + } +#ifdef __CYGWIN__ + // While linux man for tty says when VMIN == 0 && VTIME == 0, read + // should return 0 when there is nothing to read, cygwin's read returns + // -1. Not sure why and if it's correct to ignore it, but let's pretend + // it's zero. + if (r < 0) r = 0; +#endif + if (r < 0) { + // EAGAIN / EWOULDBLOCK shouldn't occur here + assert(errno != EAGAIN && errno != EWOULDBLOCK); + return -1; + } else if (r > 0) { + read_n += r; + } else { + bytebuffer_resize(&input_buffer, prevlen + read_n); + return read_n; + } + } + assert(!"unreachable"); + return 0; +} + +static int wait_fill_event(struct tb_event *event, struct timeval *timeout) +{ + // ;-) +#define ENOUGH_DATA_FOR_PARSING 64 + fd_set events; + memset(event, 0, sizeof(struct tb_event)); + + // try to extract event from input buffer, return on success + event->type = TB_EVENT_KEY; + if (extract_event(event, &input_buffer, inputmode)) + return event->type; + + // it looks like input buffer is incomplete, let's try the short path, + // but first make sure there is enough space + int n = read_up_to(ENOUGH_DATA_FOR_PARSING); + if (n < 0) + return -1; + if (n > 0 && extract_event(event, &input_buffer, inputmode)) + return event->type; + + // n == 0, or not enough data, let's go to select + while (1) { + FD_ZERO(&events); + FD_SET(inout, &events); + FD_SET(winch_fds[0], &events); + int maxfd = (winch_fds[0] > inout) ? winch_fds[0] : inout; + int result = select(maxfd+1, &events, 0, 0, timeout); + if (!result) + return 0; + + if (FD_ISSET(inout, &events)) { + event->type = TB_EVENT_KEY; + n = read_up_to(ENOUGH_DATA_FOR_PARSING); + if (n < 0) + return -1; + + if (n == 0) + continue; + + if (extract_event(event, &input_buffer, inputmode)) + return event->type; + } + if (FD_ISSET(winch_fds[0], &events)) { + event->type = TB_EVENT_RESIZE; + int zzz = 0; + read(winch_fds[0], &zzz, sizeof(int)); + buffer_size_change_request = 1; + get_term_size(&event->w, &event->h); + return TB_EVENT_RESIZE; + } + } +} diff --git a/termbox.h b/termbox.h @@ -0,0 +1,317 @@ +#ifndef TERMBOX_H +#define TERMBOX_H + +#include <stdint.h> + +/* Key constants. See also struct tb_event's key field. + * + * These are a safe subset of terminfo keys, which exist on all popular + * terminals. Termbox uses only them to stay truly portable. + */ +#define TB_KEY_F1 (0xFFFF-0) +#define TB_KEY_F2 (0xFFFF-1) +#define TB_KEY_F3 (0xFFFF-2) +#define TB_KEY_F4 (0xFFFF-3) +#define TB_KEY_F5 (0xFFFF-4) +#define TB_KEY_F6 (0xFFFF-5) +#define TB_KEY_F7 (0xFFFF-6) +#define TB_KEY_F8 (0xFFFF-7) +#define TB_KEY_F9 (0xFFFF-8) +#define TB_KEY_F10 (0xFFFF-9) +#define TB_KEY_F11 (0xFFFF-10) +#define TB_KEY_F12 (0xFFFF-11) +#define TB_KEY_INSERT (0xFFFF-12) +#define TB_KEY_DELETE (0xFFFF-13) +#define TB_KEY_HOME (0xFFFF-14) +#define TB_KEY_END (0xFFFF-15) +#define TB_KEY_PGUP (0xFFFF-16) +#define TB_KEY_PGDN (0xFFFF-17) +#define TB_KEY_ARROW_UP (0xFFFF-18) +#define TB_KEY_ARROW_DOWN (0xFFFF-19) +#define TB_KEY_ARROW_LEFT (0xFFFF-20) +#define TB_KEY_ARROW_RIGHT (0xFFFF-21) +#define TB_KEY_MOUSE_LEFT (0xFFFF-22) +#define TB_KEY_MOUSE_RIGHT (0xFFFF-23) +#define TB_KEY_MOUSE_MIDDLE (0xFFFF-24) +#define TB_KEY_MOUSE_RELEASE (0xFFFF-25) +#define TB_KEY_MOUSE_WHEEL_UP (0xFFFF-26) +#define TB_KEY_MOUSE_WHEEL_DOWN (0xFFFF-27) + +/* These are all ASCII code points below SPACE character and a BACKSPACE key. */ +#define TB_KEY_CTRL_TILDE 0x00 +#define TB_KEY_CTRL_2 0x00 /* clash with 'CTRL_TILDE' */ +#define TB_KEY_CTRL_A 0x01 +#define TB_KEY_CTRL_B 0x02 +#define TB_KEY_CTRL_C 0x03 +#define TB_KEY_CTRL_D 0x04 +#define TB_KEY_CTRL_E 0x05 +#define TB_KEY_CTRL_F 0x06 +#define TB_KEY_CTRL_G 0x07 +#define TB_KEY_BACKSPACE 0x08 +#define TB_KEY_CTRL_H 0x08 /* clash with 'CTRL_BACKSPACE' */ +#define TB_KEY_TAB 0x09 +#define TB_KEY_CTRL_I 0x09 /* clash with 'TAB' */ +#define TB_KEY_CTRL_J 0x0A +#define TB_KEY_CTRL_K 0x0B +#define TB_KEY_CTRL_L 0x0C +#define TB_KEY_ENTER 0x0D +#define TB_KEY_CTRL_M 0x0D /* clash with 'ENTER' */ +#define TB_KEY_CTRL_N 0x0E +#define TB_KEY_CTRL_O 0x0F +#define TB_KEY_CTRL_P 0x10 +#define TB_KEY_CTRL_Q 0x11 +#define TB_KEY_CTRL_R 0x12 +#define TB_KEY_CTRL_S 0x13 +#define TB_KEY_CTRL_T 0x14 +#define TB_KEY_CTRL_U 0x15 +#define TB_KEY_CTRL_V 0x16 +#define TB_KEY_CTRL_W 0x17 +#define TB_KEY_CTRL_X 0x18 +#define TB_KEY_CTRL_Y 0x19 +#define TB_KEY_CTRL_Z 0x1A +#define TB_KEY_ESC 0x1B +#define TB_KEY_CTRL_LSQ_BRACKET 0x1B /* clash with 'ESC' */ +#define TB_KEY_CTRL_3 0x1B /* clash with 'ESC' */ +#define TB_KEY_CTRL_4 0x1C +#define TB_KEY_CTRL_BACKSLASH 0x1C /* clash with 'CTRL_4' */ +#define TB_KEY_CTRL_5 0x1D +#define TB_KEY_CTRL_RSQ_BRACKET 0x1D /* clash with 'CTRL_5' */ +#define TB_KEY_CTRL_6 0x1E +#define TB_KEY_CTRL_7 0x1F +#define TB_KEY_CTRL_SLASH 0x1F /* clash with 'CTRL_7' */ +#define TB_KEY_CTRL_UNDERSCORE 0x1F /* clash with 'CTRL_7' */ +#define TB_KEY_SPACE 0x20 +#define TB_KEY_BACKSPACE2 0x7F +#define TB_KEY_CTRL_8 0x7F /* clash with 'BACKSPACE2' */ + +/* These are non-existing ones. + * + * #define TB_KEY_CTRL_1 clash with '1' + * #define TB_KEY_CTRL_9 clash with '9' + * #define TB_KEY_CTRL_0 clash with '0' + */ + +/* + * Alt modifier constant, see tb_event.mod field and tb_select_input_mode function. + * Mouse-motion modifier + */ +#define TB_MOD_ALT 0x01 +#define TB_MOD_MOTION 0x02 + +/* Colors (see struct tb_cell's fg and bg fields). */ +#define TB_DEFAULT 0x00 +#define TB_BLACK 0x01 +#define TB_RED 0x02 +#define TB_GREEN 0x03 +#define TB_YELLOW 0x04 +#define TB_BLUE 0x05 +#define TB_MAGENTA 0x06 +#define TB_CYAN 0x07 +#define TB_WHITE 0x08 + +// #define TB_GREY 0x09 +// #define TB_RED_B 0x0A +// #define TB_GREEN_B 0x0B +// #define TB_YELLOW_B 0x0C +// #define TB_BLUE_B 0x0D +// #define TB_MAGENTA_B 0x0E +// #define TB_CYAN_B 0x0F +// #define TB_GREY_B 0x10 + +/* Attributes, it is possible to use multiple attributes by combining them + * using bitwise OR ('|'). Although, colors cannot be combined. But you can + * combine attributes and a single color. See also struct tb_cell's fg and bg + * fields. + */ +#define TB_BOLD 0x0100 +#define TB_UNDERLINE 0x0200 +#define TB_REVERSE 0x0400 + +/* A cell, single conceptual entity on the terminal screen. The terminal screen + * is basically a 2d array of cells. It has the following fields: + * - 'ch' is a unicode character + * - 'fg' foreground color and attributes + * - 'bg' background color and attributes + */ +struct tb_cell { + uint32_t ch; + uint16_t fg; + uint16_t bg; +}; + +#define TB_EVENT_KEY 1 +#define TB_EVENT_RESIZE 2 +#define TB_EVENT_MOUSE 3 + +/* An event, single interaction from the user. The 'mod' and 'ch' fields are + * valid if 'type' is TB_EVENT_KEY. The 'w' and 'h' fields are valid if 'type' + * is TB_EVENT_RESIZE. The 'x' and 'y' fields are valid if 'type' is + * TB_EVENT_MOUSE. The 'key' field is valid if 'type' is either TB_EVENT_KEY + * or TB_EVENT_MOUSE. The fields 'key' and 'ch' are mutually exclusive; only + * one of them can be non-zero at a time. + */ +struct tb_event { + uint8_t type; + uint8_t mod; /* modifiers to either 'key' or 'ch' below */ + uint16_t key; /* one of the TB_KEY_* constants */ + uint32_t ch; /* unicode character */ + int32_t w; + int32_t h; + int32_t x; + int32_t y; +}; + +/* Error codes returned by tb_init(). All of them are self-explanatory, except + * the pipe trap error. Termbox uses unix pipes in order to deliver a message + * from a signal handler (SIGWINCH) to the main event reading loop. Honestly in + * most cases you should just check the returned code as < 0. + */ +#define TB_EUNSUPPORTED_TERMINAL -1 +#define TB_EFAILED_TO_OPEN_TTY -2 +#define TB_EPIPE_TRAP_ERROR -3 + +/* Initializes the termbox library. This function should be called before any + * other functions. Function tb_init is same as tb_init_file("/dev/tty"). + * After successful initialization, the library must be + * finalized using the tb_shutdown() function. + */ +int tb_init(void); +int tb_init_file(const char* name); +int tb_init_fd(int inout); +void tb_shutdown(void); + +/* Returns the size of the internal back buffer (which is the same as + * terminal's window size in characters). The internal buffer can be resized + * after tb_clear() or tb_present() function calls. Both dimensions have an + * unspecified negative value when called before tb_init() or after + * tb_shutdown(). + */ +int tb_width(void); +int tb_height(void); + +/* Clears the internal back buffer using TB_DEFAULT color or the + * color/attributes set by tb_set_clear_attributes() function. + */ +void tb_clear(void); +void tb_set_clear_attributes(uint16_t fg, uint16_t bg); + +/* Synchronizes the internal back buffer with the terminal. */ +void tb_present(void); + +#define TB_HIDE_CURSOR -1 + +/* Sets the position of the cursor. Upper-left character is (0, 0). If you pass + * TB_HIDE_CURSOR as both coordinates, then the cursor will be hidden. Cursor + * is hidden by default. + */ +void tb_set_cursor(int cx, int cy); + +/* Changes cell's parameters in the internal back buffer at the specified + * position. + */ +void tb_put_cell(int x, int y, const struct tb_cell *cell); +void tb_change_cell(int x, int y, uint32_t ch, uint16_t fg, uint16_t bg); + +/* Copies the buffer from 'cells' at the specified position, assuming the + * buffer is a two-dimensional array of size ('w' x 'h'), represented as a + * one-dimensional buffer containing lines of cells starting from the top. + * + * (DEPRECATED: use tb_cell_buffer() instead and copy memory on your own) + */ +void tb_blit(int x, int y, int w, int h, const struct tb_cell *cells); + +/* Returns a pointer to internal cell back buffer. You can get its dimensions + * using tb_width() and tb_height() functions. The pointer stays valid as long + * as no tb_clear() and tb_present() calls are made. The buffer is + * one-dimensional buffer containing lines of cells starting from the top. + */ +struct tb_cell *tb_cell_buffer(void); + +#define TB_INPUT_CURRENT 0 /* 000 */ +#define TB_INPUT_ESC 1 /* 001 */ +#define TB_INPUT_ALT 2 /* 010 */ +#define TB_INPUT_MOUSE 4 /* 100 */ + +/* Sets the termbox input mode. Termbox has two input modes: + * 1. Esc input mode. + * When ESC sequence is in the buffer and it doesn't match any known + * ESC sequence => ESC means TB_KEY_ESC. + * 2. Alt input mode. + * When ESC sequence is in the buffer and it doesn't match any known + * sequence => ESC enables TB_MOD_ALT modifier for the next keyboard event. + * + * You can also apply TB_INPUT_MOUSE via bitwise OR operation to either of the + * modes (e.g. TB_INPUT_ESC | TB_INPUT_MOUSE). If none of the main two modes + * were set, but the mouse mode was, TB_INPUT_ESC mode is used. If for some + * reason you've decided to use (TB_INPUT_ESC | TB_INPUT_ALT) combination, it + * will behave as if only TB_INPUT_ESC was selected. + * + * If 'mode' is TB_INPUT_CURRENT, it returns the current input mode. + * + * Default termbox input mode is TB_INPUT_ESC. + */ +int tb_select_input_mode(int mode); + +#define TB_OUTPUT_CURRENT 0 +#define TB_OUTPUT_NORMAL 1 +#define TB_OUTPUT_256 2 +#define TB_OUTPUT_216 3 +#define TB_OUTPUT_GRAYSCALE 4 + +/* Sets the termbox output mode. Termbox has three output options: + * 1. TB_OUTPUT_NORMAL => [1..8] + * This mode provides 8 different colors: + * black, red, green, yellow, blue, magenta, cyan, white + * Shortcut: TB_BLACK, TB_RED, ... + * Attributes: TB_BOLD, TB_UNDERLINE, TB_REVERSE + * + * Example usage: + * tb_change_cell(x, y, '@', TB_BLACK | TB_BOLD, TB_RED); + * + * 2. TB_OUTPUT_256 => [0..256] + * In this mode you can leverage the 256 terminal mode: + * 0x00 - 0x07: the 8 colors as in TB_OUTPUT_NORMAL + * 0x08 - 0x0f: TB_* | TB_BOLD + * 0x10 - 0xe7: 216 different colors + * 0xe8 - 0xff: 24 different shades of grey + * + * Example usage: + * tb_change_cell(x, y, '@', 184, 240); + * tb_change_cell(x, y, '@', 0xb8, 0xf0); + * + * 3. TB_OUTPUT_216 => [0..216] + * This mode supports the 3rd range of the 256 mode only. + * But you don't need to provide an offset. + * + * 4. TB_OUTPUT_GRAYSCALE => [0..23] + * This mode supports the 4th range of the 256 mode only. + * But you dont need to provide an offset. + * + * Execute build/src/demo/output to see its impact on your terminal. + * + * If 'mode' is TB_OUTPUT_CURRENT, it returns the current output mode. + * + * Default termbox output mode is TB_OUTPUT_NORMAL. + */ +int tb_select_output_mode(int mode); + +/* Wait for an event up to 'timeout' milliseconds and fill the 'event' + * structure with it, when the event is available. Returns the type of the + * event (one of TB_EVENT_* constants) or -1 if there was an error or 0 in case + * there were no event during 'timeout' period. + */ +int tb_peek_event(struct tb_event *event, int timeout); + +/* Wait for an event forever and fill the 'event' structure with it, when the + * event is available. Returns the type of the event (one of TB_EVENT_* + * constants) or -1 if there was an error. + */ +int tb_poll_event(struct tb_event *event); + +/* Utility utf8 functions. */ +#define TB_EOF -1 +int tb_utf8_char_length(char c); +int tb_utf8_char_to_unicode(uint32_t *out, const char *c); +int tb_utf8_unicode_to_char(char *out, uint32_t c); + +#endif /* TERMBOX_H */ diff --git a/utf8.c b/utf8.c @@ -0,0 +1,79 @@ +#include "termbox.h" + +static const unsigned char utf8_length[256] = { + 1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1, + 1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1, + 1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1, + 1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1, + 1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1, + 1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1, + 2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2, + 3,3,3,3,3,3,3,3,3,3,3,3,3,3,3,3,4,4,4,4,4,4,4,4,5,5,5,5,6,6,1,1 +}; + +static const unsigned char utf8_mask[6] = { + 0x7F, + 0x1F, + 0x0F, + 0x07, + 0x03, + 0x01 +}; + +int tb_utf8_char_length(char c) +{ + return utf8_length[(unsigned char)c]; +} + +int tb_utf8_char_to_unicode(uint32_t *out, const char *c) +{ + if (*c == 0) + return TB_EOF; + + int i; + unsigned char len = tb_utf8_char_length(*c); + unsigned char mask = utf8_mask[len-1]; + uint32_t result = c[0] & mask; + for (i = 1; i < len; ++i) { + result <<= 6; + result |= c[i] & 0x3f; + } + + *out = result; + return (int)len; +} + +int tb_utf8_unicode_to_char(char *out, uint32_t c) +{ + int len = 0; + int first; + int i; + + if (c < 0x80) { + first = 0; + len = 1; + } else if (c < 0x800) { + first = 0xc0; + len = 2; + } else if (c < 0x10000) { + first = 0xe0; + len = 3; + } else if (c < 0x200000) { + first = 0xf0; + len = 4; + } else if (c < 0x4000000) { + first = 0xf8; + len = 5; + } else { + first = 0xfc; + len = 6; + } + + for (i = len - 1; i > 0; --i) { + out[i] = (c & 0x3f) | 0x80; + c >>= 6; + } + out[0] = c | first; + + return len; +} diff --git a/util.c b/util.c @@ -0,0 +1,35 @@ +/* See LICENSE file for copyright and license details. */ +#include <stdarg.h> +#include <stdio.h> +#include <stdlib.h> +#include <string.h> + +#include "util.h" + +void * +ecalloc(size_t nmemb, size_t size) +{ + void *p; + p = calloc(nmemb, size); + FAIL_IF(p == NULL, "calloc"); + return p; +} + +void +die(const char *fmt, ...) +{ + va_list ap; + + va_start(ap, fmt); + (void)vfprintf(stderr, fmt, ap); + va_end(ap); + + if (fmt[0] != '\0' && fmt[strlen(fmt)-1] == ':') { + (void)fputc(' ', stderr); + perror(NULL); + } else { + (void)fputc('\n', stderr); + } + + exit(EXIT_FAILURE); +} diff --git a/util.h b/util.h @@ -0,0 +1,15 @@ +/* See LICENSE file for copyright and license details. */ + +#ifndef UTIL_H +#define UTIL_H + +#define MAX(A, B) ((A) > (B) ? (A) : (B)) +#define MIN(A, B) ((A) < (B) ? (A) : (B)) +#define LEN(A) (sizeof(A)/sizeof(A[0])) +#define BETWEEN(X, A, B) ((A) <= (X) && (X) <= (B)) +#define FAIL_IF(EXP, MSG) {if(EXP){fprintf(stderr, "[\033[31mFAILED %d\033[0m] %s\n", __LINE__, MSG);exit(EXIT_FAILURE);}} + +void *ecalloc(size_t, size_t); +void die(const char *fmt, ...); + +#endif /* UTIL_H */