userspace: Implement basic `less` program
This is very simple and only supports couple of flags and scrolling
This commit is contained in:
parent
69c4940b27
commit
37dea8aee7
|
|
@ -16,6 +16,7 @@ set(USERSPACE_PROGRAMS
|
|||
id
|
||||
image
|
||||
init
|
||||
less
|
||||
ln
|
||||
loadfont
|
||||
loadkeys
|
||||
|
|
|
|||
|
|
@ -0,0 +1,9 @@
|
|||
set(SOURCES
|
||||
main.cpp
|
||||
)
|
||||
|
||||
add_executable(less ${SOURCES})
|
||||
banan_link_library(less ban)
|
||||
banan_link_library(less libc)
|
||||
|
||||
install(TARGETS less OPTIONAL)
|
||||
|
|
@ -0,0 +1,367 @@
|
|||
#include <BAN/Vector.h>
|
||||
#include <BAN/String.h>
|
||||
|
||||
#include <ctype.h>
|
||||
#include <fcntl.h>
|
||||
#include <sys/ioctl.h>
|
||||
#include <sys/select.h>
|
||||
#include <termios.h>
|
||||
#include <unistd.h>
|
||||
|
||||
struct config_t
|
||||
{
|
||||
bool raw { false };
|
||||
bool quit_if_one_screen { false };
|
||||
bool no_init { false };
|
||||
};
|
||||
static config_t config;
|
||||
|
||||
static void usage(const char* argv0, int ret)
|
||||
{
|
||||
FILE* fout = ret ? stderr : stdout;
|
||||
fprintf(fout, "usage: %s [OPTIONS]... [--] [FILE]...\n", argv0);
|
||||
fprintf(fout, " -r, -R, --raw print ANSI control sequences\n");
|
||||
fprintf(fout, " -F, -quit-if-one-screen exit immediately if output fits in one screen\n");
|
||||
fprintf(fout, " -X, -no-init don't clear screen at the startup\n");
|
||||
fprintf(fout, " -h, --help show this message and exit\n");
|
||||
exit(ret);
|
||||
}
|
||||
|
||||
static void handle_option_short(const char* argv0, char opt, bool exit_on_error)
|
||||
{
|
||||
switch (opt)
|
||||
{
|
||||
case 'r':
|
||||
case 'R':
|
||||
config.raw = true;
|
||||
break;
|
||||
case 'F':
|
||||
config.quit_if_one_screen = true;
|
||||
break;
|
||||
case 'X':
|
||||
config.no_init = true;
|
||||
break;
|
||||
case 'h':
|
||||
usage(argv0, 0);
|
||||
break;
|
||||
default:
|
||||
fprintf(stderr, "unknown option: %c\n", opt);
|
||||
fprintf(stderr, "see --help for usage\n");
|
||||
if (exit_on_error)
|
||||
exit(1);
|
||||
}
|
||||
}
|
||||
|
||||
static void handle_option_long(const char* argv0, const char* opt, bool exit_on_error)
|
||||
{
|
||||
if (!strcmp(opt, "--raw"))
|
||||
config.raw = true;
|
||||
else if (!strcmp(opt, "--quit-if-one-screen"))
|
||||
config.quit_if_one_screen = true;
|
||||
else if (!strcmp(opt, "--no-init"))
|
||||
config.no_init = true;
|
||||
else if (!strcmp(opt, "--help"))
|
||||
usage(argv0, 0);
|
||||
else
|
||||
{
|
||||
fprintf(stderr, "unknown option: %s\n", opt);
|
||||
fprintf(stderr, "see --help for usage\n");
|
||||
if (exit_on_error)
|
||||
exit(1);
|
||||
}
|
||||
}
|
||||
|
||||
static int parse_config_or_exit(int argc, char** argv)
|
||||
{
|
||||
// FIXME: long environment options
|
||||
if (const char* env = getenv("LESS"))
|
||||
for (size_t i = 0; env[i]; i++)
|
||||
(void)handle_option_short(argv[0], env[i], false);
|
||||
|
||||
int i = 1;
|
||||
for (; i < argc; i++)
|
||||
{
|
||||
if (argv[i][0] != '-' || !strcmp(argv[i], "--"))
|
||||
break;
|
||||
if (argv[i][1] == '-')
|
||||
handle_option_long(argv[0], argv[i], true);
|
||||
else for (size_t j = 1; argv[i][j]; j++)
|
||||
handle_option_short(argv[0], argv[i][j], true);
|
||||
}
|
||||
return i;
|
||||
}
|
||||
|
||||
static int get_keyboard_fd()
|
||||
{
|
||||
// if stdin is a terminal, use it
|
||||
if (isatty(STDIN_FILENO))
|
||||
return STDIN_FILENO;
|
||||
|
||||
// otherwise try to open our controlling terminal
|
||||
int fd = open("/dev/tty", O_RDONLY);
|
||||
if (fd != -1)
|
||||
return fd;
|
||||
|
||||
// if that fails, try stderr as the last fallback
|
||||
if (isatty(STDERR_FILENO))
|
||||
return STDERR_FILENO;
|
||||
|
||||
return -1;
|
||||
}
|
||||
|
||||
static int output_to_non_terminal(int fd)
|
||||
{
|
||||
for (;;)
|
||||
{
|
||||
char buffer[128];
|
||||
const ssize_t nread = read(fd, buffer, sizeof(buffer));
|
||||
if (nread == -1)
|
||||
perror("read");
|
||||
if (nread <= 0)
|
||||
break;
|
||||
|
||||
ssize_t total = 0;
|
||||
while (total < nread)
|
||||
{
|
||||
const ssize_t nwrite = write(STDOUT_FILENO, buffer + total, nread - total);
|
||||
if (nwrite < 0)
|
||||
perror("write");
|
||||
if (nwrite <= 0)
|
||||
break;
|
||||
total += nwrite;
|
||||
}
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static bool read_lines(int fd, winsize ws, BAN::Vector<BAN::String>& out)
|
||||
{
|
||||
char buffer[128];
|
||||
const ssize_t nread = read(fd, buffer, sizeof(buffer));
|
||||
if (nread < 0)
|
||||
perror("read");
|
||||
if (nread <= 0)
|
||||
return false;
|
||||
|
||||
if (out.empty())
|
||||
MUST(out.emplace_back());
|
||||
|
||||
bool in_ansi = false;
|
||||
|
||||
size_t col = 0;
|
||||
for (size_t i = 0; i < out.back().size(); i++)
|
||||
{
|
||||
if (in_ansi)
|
||||
{
|
||||
if (isalpha(out.back()[i]))
|
||||
in_ansi = false;
|
||||
}
|
||||
else if (out.back()[i] == '\e')
|
||||
in_ansi = true;
|
||||
else
|
||||
col++;
|
||||
}
|
||||
|
||||
for (ssize_t i = 0; i < nread; i++)
|
||||
{
|
||||
if (in_ansi)
|
||||
{
|
||||
if (config.raw)
|
||||
MUST(out.back().push_back(buffer[i]));
|
||||
if (isalpha(buffer[i]))
|
||||
in_ansi = false;
|
||||
continue;
|
||||
}
|
||||
|
||||
const auto append_char =
|
||||
[&col, &out, ws](char ch)
|
||||
{
|
||||
if (col >= ws.ws_col)
|
||||
{
|
||||
MUST(out.emplace_back());
|
||||
col = 0;
|
||||
}
|
||||
MUST(out.back().push_back(ch));
|
||||
col++;
|
||||
};
|
||||
|
||||
switch (buffer[i])
|
||||
{
|
||||
case '\e':
|
||||
if (config.raw)
|
||||
MUST(out.back().push_back(buffer[i]));
|
||||
in_ansi = true;
|
||||
break;
|
||||
case '\n':
|
||||
MUST(out.emplace_back());
|
||||
col = 0;
|
||||
break;
|
||||
case '\t':
|
||||
append_char(' ');
|
||||
while (col % 8)
|
||||
append_char(' ');
|
||||
break;
|
||||
default:
|
||||
append_char(buffer[i]);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
static bool less_file(int fd, int kb_fd, winsize ws)
|
||||
{
|
||||
if (!isatty(STDOUT_FILENO) || kb_fd == -1 || ws.ws_col == 0 || ws.ws_row == 0)
|
||||
return output_to_non_terminal(fd);
|
||||
|
||||
BAN::Vector<BAN::String> lines;
|
||||
size_t lines_size = 0;
|
||||
const auto update_lines_size =
|
||||
[&lines, &lines_size]
|
||||
{
|
||||
lines_size = lines.size();
|
||||
if (!lines.empty() && lines.back().empty())
|
||||
lines_size--;
|
||||
};
|
||||
|
||||
while (lines_size < ws.ws_row)
|
||||
{
|
||||
if (!read_lines(fd, ws, lines))
|
||||
break;
|
||||
update_lines_size();
|
||||
}
|
||||
|
||||
if (!config.no_init)
|
||||
{
|
||||
int y = 1;
|
||||
if (lines_size < ws.ws_row)
|
||||
y = ws.ws_row - lines_size;
|
||||
printf("\e[2J\e[%dH", y);
|
||||
}
|
||||
|
||||
for (size_t i = 0; i < lines_size && i < static_cast<size_t>(ws.ws_row - 1); i++)
|
||||
printf("%s\n", lines[i].data());
|
||||
fflush(stdout);
|
||||
|
||||
if (lines_size < ws.ws_row && config.quit_if_one_screen)
|
||||
return true;
|
||||
|
||||
printf(":");
|
||||
fflush(stdout);
|
||||
|
||||
size_t line = 0;
|
||||
for (;;)
|
||||
{
|
||||
char ch;
|
||||
if (read(kb_fd, &ch, 1) != 1)
|
||||
break;
|
||||
switch (ch)
|
||||
{
|
||||
case 'q':
|
||||
printf("\e[G\e[K");
|
||||
fflush(stdout);
|
||||
return true;
|
||||
case '\e':
|
||||
{
|
||||
if (read(kb_fd, &ch, 1) != 1 || ch != '[')
|
||||
break;
|
||||
if (read(kb_fd, &ch, 1) != 1)
|
||||
break;
|
||||
switch (ch)
|
||||
{
|
||||
case 'A':
|
||||
if (line == 0)
|
||||
break;
|
||||
line--;
|
||||
|
||||
printf("\e[H");
|
||||
for (int i = 0; i < ws.ws_row - 1; i++)
|
||||
printf("%s\e[K\n", lines[line + i].data());
|
||||
printf(":\e[K");
|
||||
fflush(stdout);
|
||||
|
||||
break;
|
||||
case 'B':
|
||||
while (lines_size - (line + 1) < ws.ws_row)
|
||||
{
|
||||
if (!read_lines(fd, ws, lines))
|
||||
break;
|
||||
update_lines_size();
|
||||
}
|
||||
if (lines_size - (line + 1) < static_cast<size_t>(ws.ws_row - 1))
|
||||
break;
|
||||
line++;
|
||||
|
||||
printf("\e[H");
|
||||
for (int i = 0; i < ws.ws_row - 1; i++)
|
||||
printf("%s\e[K\n", lines[line + i].data());
|
||||
printf(":\e[K");
|
||||
fflush(stdout);
|
||||
|
||||
break;
|
||||
}
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
int main(int argc, char** argv)
|
||||
{
|
||||
int i = parse_config_or_exit(argc, argv);
|
||||
|
||||
int kb_fd = get_keyboard_fd();
|
||||
if (kb_fd != -1)
|
||||
{
|
||||
int flags = 0;
|
||||
fcntl(kb_fd, F_GETFL, &flags);
|
||||
if (flags & O_NONBLOCK)
|
||||
fcntl(kb_fd, F_SETFL, flags & ~O_NONBLOCK);
|
||||
|
||||
termios termios;
|
||||
tcgetattr(kb_fd, &termios);
|
||||
termios.c_lflag &= ~(ECHO | ICANON);
|
||||
tcsetattr(kb_fd, TCSANOW, &termios);
|
||||
}
|
||||
|
||||
winsize ws { .ws_row = 0, .ws_col = 0 };
|
||||
if (isatty(STDOUT_FILENO))
|
||||
{
|
||||
if (ioctl(STDOUT_FILENO, TIOCGWINSZ, &ws) == 0)
|
||||
{
|
||||
// try to make stdout fully buffered to reduce flicker
|
||||
const size_t bufsize = 2 * ws.ws_row * ws.ws_col;
|
||||
if (char* buffer = static_cast<char*>(malloc(bufsize)))
|
||||
setvbuf(stdout, buffer, _IOFBF, bufsize);
|
||||
}
|
||||
}
|
||||
|
||||
if (i == argc)
|
||||
{
|
||||
if (!less_file(STDIN_FILENO, kb_fd, ws))
|
||||
return EXIT_FAILURE;
|
||||
return EXIT_SUCCESS;
|
||||
}
|
||||
|
||||
int ret = EXIT_SUCCESS;
|
||||
for (; i < argc; i++)
|
||||
{
|
||||
int fd = open(argv[i], O_RDONLY);
|
||||
if (fd == -1)
|
||||
{
|
||||
perror(argv[i]);
|
||||
continue;
|
||||
}
|
||||
|
||||
if (!less_file(fd, kb_fd, ws))
|
||||
ret = EXIT_FAILURE;
|
||||
|
||||
close(fd);
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
Loading…
Reference in New Issue