nnn/noice.c

473 lines
7.8 KiB
C
Raw Normal View History

#include <sys/stat.h>
2014-10-07 06:05:30 +00:00
#include <sys/types.h>
#include <errno.h>
#include <fcntl.h>
#include <dirent.h>
#include <curses.h>
#include <libgen.h>
#include <locale.h>
#include <stdlib.h>
#include <stdio.h>
#include <signal.h>
#include <string.h>
#include <unistd.h>
2014-10-08 12:08:29 +00:00
#ifdef LINUX
#include <bsd/string.h>
#endif
2014-10-07 06:05:30 +00:00
#ifdef DEBUG
2014-10-08 08:36:17 +00:00
#define DEBUG_FD 8
#define DPRINTF_D(x) dprintf(DEBUG_FD, #x "=%d\n", x)
#define DPRINTF_U(x) dprintf(DEBUG_FD, #x "=%u\n", x)
#define DPRINTF_S(x) dprintf(DEBUG_FD, #x "=%s\n", x)
#define DPRINTF_P(x) dprintf(DEBUG_FD, #x "=0x%p\n", x)
2014-10-07 06:05:30 +00:00
#else
#define DPRINTF_D(x)
2014-10-08 08:36:17 +00:00
#define DPRINTF_U(x)
2014-10-07 06:05:30 +00:00
#define DPRINTF_S(x)
#define DPRINTF_P(x)
#endif /* DEBUG */
#define LEN(x) (sizeof(x) / sizeof(*(x)))
2014-10-07 06:49:46 +00:00
#define MIN(x, y) ((x) < (y) ? (x) : (y))
#define ISODD(x) ((x) & 1)
2014-10-07 06:05:30 +00:00
2014-10-07 15:36:29 +00:00
struct assoc {
char *ext; /* Extension */
char *bin; /* Program */
};
/* Configuration */
struct assoc assocs[] = {
{ ".avi", "mplayer" },
{ ".mp4", "mplayer" },
{ ".mkv", "mplayer" },
{ ".mp3", "mplayer" },
{ ".ogg", "mplayer" },
{ ".srt", "less" },
{ ".txt", "less" },
2014-10-08 08:43:11 +00:00
{ ".sh", "sh" },
2014-10-07 15:36:29 +00:00
{ "README", "less" },
};
#define CWD "cwd: "
#define CURSR " > "
#define EMPTY " "
2014-10-07 06:05:30 +00:00
/*
* Layout:
* .---------
* | cwd: /mnt/path
* |
2014-10-07 14:55:14 +00:00
* | file0
2014-10-07 06:05:30 +00:00
* | file1
2014-10-07 14:55:14 +00:00
* | > file2
* | file3
* | file4
2014-10-07 06:05:30 +00:00
* ...
* | filen
* |
2014-10-07 14:55:14 +00:00
* | Permission denied
2014-10-07 06:05:30 +00:00
* '------
*/
int die = 0;
char *
openwith(char *file)
2014-10-07 06:05:30 +00:00
{
char *ext = NULL;
char *bin = NULL;
2014-10-07 06:05:30 +00:00
int i;
ext = strrchr(file, '.');
if (ext == NULL)
ext = file;
DPRINTF_S(ext);
2014-10-07 06:05:30 +00:00
for (i = 0; i < LEN(assocs); i++)
if (strcmp(assocs[i].ext, ext) == 0)
bin = assocs[i].bin;
DPRINTF_S(bin);
return bin;
2014-10-07 06:05:30 +00:00
}
int
dentcmp(const void *va, const void *vb)
{
const struct dirent *a, *b;
a = (struct dirent *)va;
b = (struct dirent *)vb;
2014-10-07 06:05:30 +00:00
return strcmp(a->d_name, b->d_name);
}
2014-10-07 11:37:23 +00:00
void
initcurses(void)
{
initscr();
cbreak();
noecho();
nonl();
intrflush(stdscr, FALSE);
keypad(stdscr, TRUE);
curs_set(FALSE); /* Hide cursor */
}
void
exitcurses(void)
{
endwin(); /* Restore terminal */
}
2014-10-07 14:47:35 +00:00
/* Messages show up at the bottom */
2014-10-07 06:05:30 +00:00
void
2014-10-07 14:47:35 +00:00
printmsg(char *msg)
2014-10-07 06:05:30 +00:00
{
move(LINES - 1, 0);
2014-10-07 14:47:35 +00:00
printw("%s\n", msg);
}
/* Display warning as a message */
void
printwarn(void)
{
printmsg(strerror(errno));
2014-10-07 06:05:30 +00:00
}
/* Kill curses and display error before exiting */
void
printerr(int ret, char *prefix)
{
2014-10-07 17:59:41 +00:00
exitcurses();
2014-10-07 06:05:30 +00:00
printf("%s: %s\n", prefix, strerror(errno));
exit(ret);
}
/*
* Returns 0 normally
* On movement it updates *cur
* Returns 1 on quit
* Returns 2 on go in
* Returns 3 on go up
*/
int
nextsel(int *cur, int max)
{
int c;
c = getch();
switch (c) {
case 'q':
return 1;
/* go up */
case KEY_BACKSPACE:
case KEY_LEFT:
case 'h':
return 2;
/* go in */
case KEY_ENTER:
case '\r':
case KEY_RIGHT:
case 'l':
return 3;
/* next */
case 'j':
case KEY_DOWN:
if (*cur < max - 1)
(*cur)++;
break;
/* prev */
case 'k':
case KEY_UP:
if (*cur > 0)
(*cur)--;
break;
}
return 0;
}
int
testopen(char *path)
{
int fd;
fd = open(path, O_RDONLY);
if (fd == -1) {
return 0;
} else {
close(fd);
return 1;
}
}
int
testopendir(char *path)
{
DIR *dirp;
dirp = opendir(path);
if (dirp == NULL) {
return 0;
} else {
closedir(dirp);
return 1;
}
}
void
browse(const char *ipath)
{
DIR *dirp;
struct dirent *dp;
struct dirent *dents;
2014-10-07 06:05:30 +00:00
int i, n, cur;
int r, ret;
char *path = strdup(ipath);
char *cwd;
2014-10-07 06:05:30 +00:00
begin:
2014-10-07 06:49:46 +00:00
/* Path should be a malloc(3)-ed string at all times */
2014-10-07 06:05:30 +00:00
n = 0;
cur = 0;
dents = NULL;
dirp = opendir(path);
if (dirp == NULL) {
2014-10-07 14:47:35 +00:00
printwarn();
2014-10-07 06:05:30 +00:00
goto nochange;
}
while ((dp = readdir(dirp)) != NULL) {
/* Skip self and parent */
if (strcmp(dp->d_name, ".") == 0
|| strcmp(dp->d_name, "..") == 0)
2014-10-07 06:05:30 +00:00
continue;
dents = realloc(dents, (n + 1) * sizeof(*dents));
if (dents == NULL)
printerr(1, "realloc");
memcpy(&dents[n], dp, sizeof(*dents));
2014-10-07 06:05:30 +00:00
n++;
}
qsort(dents, n, sizeof(*dents), dentcmp);
for (;;) {
2014-10-07 11:23:44 +00:00
int nlines;
struct dirent *tmpents;
int odd;
2014-10-07 11:23:44 +00:00
2014-10-07 06:05:30 +00:00
redraw:
2014-10-07 11:23:44 +00:00
nlines = MIN(LINES - 4, n);
2014-10-07 06:05:30 +00:00
/* Clean screen */
erase();
/* Strip trailing slashes */
2014-10-07 06:05:30 +00:00
for (i = strlen(path) - 1; i > -1; i--)
if (path[i] == '/')
path[i] = '\0';
else
break;
DPRINTF_D(cur);
DPRINTF_S(path);
/* No text wrapping in cwd line */
cwd = malloc(COLS * sizeof(char));
2014-10-08 12:08:29 +00:00
strlcpy(cwd, path, COLS * sizeof(char));
cwd[COLS - strlen(CWD) - 1] = '\0';
/* No text wrapping in entries */
tmpents = malloc(n * sizeof(*tmpents));
memcpy(tmpents, dents, n * sizeof(*tmpents));
for (i = 0; i < n; i++)
tmpents[i].d_name[COLS - strlen(CURSR) - 1] = '\0';
/* Print cwd. If empty we are on the root. We store it
* as an empty string so that when we navigate in /mnt
* is doesn't come up as //mnt. */
printw(CWD "%s%s\n\n",
strcmp(cwd, "") == 0 ? "/" : "",
cwd);
2014-10-07 06:05:30 +00:00
/* Print listing */
odd = ISODD(nlines);
2014-10-07 06:49:46 +00:00
if (cur < nlines / 2) {
for (i = 0; i < nlines; i++)
printw("%s%s\n",
i == cur ? CURSR : EMPTY,
tmpents[i].d_name);
2014-10-07 06:49:46 +00:00
} else if (cur >= n - nlines / 2) {
for (i = n - nlines; i < n; i++)
printw("%s%s\n",
i == cur ? CURSR : EMPTY,
tmpents[i].d_name);
2014-10-07 06:49:46 +00:00
} else {
for (i = cur - nlines / 2;
i < cur + nlines / 2 + odd; i++)
printw("%s%s\n",
i == cur ? CURSR : EMPTY,
tmpents[i].d_name);
2014-10-07 06:49:46 +00:00
}
2014-10-07 06:05:30 +00:00
free(tmpents);
2014-10-07 06:05:30 +00:00
nochange:
ret = nextsel(&cur, n);
if (ret == 1) {
free(path);
return;
}
if (ret == 2) {
/* Handle root case */
if (strcmp(path, "") == 0) {
2014-10-07 06:05:30 +00:00
goto nochange;
} else {
2014-10-07 06:49:46 +00:00
char *dir, *tmp;
dir = dirname(path);
tmp = malloc(strlen(dir) + 1);
2014-10-08 12:08:29 +00:00
strlcpy(tmp, dir, strlen(dir) + 1);
2014-10-07 06:49:46 +00:00
free(path);
path = tmp;
2014-10-07 06:05:30 +00:00
goto out;
}
}
if (ret == 3) {
char *pathnew, *pathtmp;
size_t pathsiz;
char *name;
u_int8_t type;
char *bin;
2014-10-07 06:05:30 +00:00
pid_t pid;
struct stat sb;
2014-10-07 06:05:30 +00:00
2014-10-07 14:07:56 +00:00
/* Cannot descend in empty directories */
if (n == 0)
goto nochange;
name = dents[cur].d_name;
type = dents[cur].d_type;
pathsiz = strlen(path) + 1 + strlen(name) + 1;
pathnew = malloc(pathsiz);
snprintf(pathnew, pathsiz, "%s/%s", path, name);
2014-10-08 08:36:17 +00:00
DPRINTF_S(name);
DPRINTF_U(type);
DPRINTF_S(pathnew);
again:
switch (type) {
case DT_LNK:
/* Resolve link */
pathtmp = realpath(pathnew, NULL);
if (pathtmp == NULL) {
printwarn();
free(pathnew);
goto nochange;
} else {
r = stat(pathtmp, &sb);
free(pathtmp);
if (r == -1) {
printwarn();
free(pathnew);
goto nochange;
}
/* Directory or file */
if (S_ISDIR(sb.st_mode)) {
type = DT_DIR;
goto again;
}
if (S_ISREG(sb.st_mode)) {
type = DT_REG;
goto again;
}
/* All the rest */
printmsg("Unsupported file");
free(pathnew);
goto nochange;
}
2014-10-07 06:05:30 +00:00
case DT_DIR:
/* Change to new path */
if (testopen(pathnew)) {
2014-10-07 06:05:30 +00:00
free(path);
path = pathnew;
2014-10-07 06:05:30 +00:00
goto out;
} else {
2014-10-07 14:47:35 +00:00
printwarn();
free(pathnew);
2014-10-07 06:05:30 +00:00
goto nochange;
}
case DT_REG:
if (!testopen(pathnew)) {
printwarn();
free(pathnew);
goto nochange;
}
2014-10-07 06:05:30 +00:00
/* Open with */
bin = openwith(name);
2014-10-07 06:05:30 +00:00
if (bin == NULL) {
2014-10-07 14:47:35 +00:00
printmsg("No association");
2014-10-08 12:27:46 +00:00
free(pathnew);
2014-10-07 06:05:30 +00:00
goto nochange;
}
2014-10-07 11:37:23 +00:00
exitcurses();
2014-10-07 06:05:30 +00:00
/* Run program */
pid = fork();
if (pid == 0)
execlp(bin, bin, pathnew, NULL);
2014-10-07 06:05:30 +00:00
else
waitpid(pid, NULL, 0);
2014-10-07 11:37:23 +00:00
initcurses();
free(pathnew);
2014-10-07 06:05:30 +00:00
goto redraw;
default:
printmsg("Unsupported file");
2014-10-08 12:27:46 +00:00
free(pathnew);
goto nochange;
2014-10-07 06:05:30 +00:00
}
}
}
out:
free(dents);
r = closedir(dirp);
if (r == -1)
printerr(1, "closedir");
goto begin;
}
int
main(int argc, char *argv[])
{
char *ipath = argv[1] != NULL ? argv[1] : "/";
/* Test initial path */
if (!testopendir(ipath))
printerr(1, ipath);
/* Set locale before curses setup */
setlocale(LC_ALL, "");
2014-10-07 11:37:23 +00:00
initcurses();
2014-10-07 06:05:30 +00:00
browse(ipath);
2014-10-07 11:37:23 +00:00
exitcurses();
2014-10-07 06:05:30 +00:00
return 0;
}