1 #include <stdlib.h>
2 #include <stdint.h>
3 #include <ctype.h>
4 #include <string.h>
5 #include <sys/types.h>  /* pid_t, ... */
6 #include <stdio.h>
7 #include <limits.h>     /* PATH_MAX */
8 #include <locale.h>     /* setlocale(), LC_ALL */
9 #include <unistd.h>     /* chdir(), getcwd(), read(), close(), ... */
10 #include <dirent.h>     /* DIR, struct dirent, opendir(), ... */
11 #include <sys/stat.h>
12 #include <fcntl.h>      /* open() */
13 #include <sys/wait.h>   /* waitpid() */
14 #include <signal.h>     /* struct sigaction, sigaction() */
15 #include <errno.h>
16 #include <curses.h>
17 
18 #include "config.h"
19 
20 /* String buffers. */
21 #define ROWSZ 256
22 static char ROW[ROWSZ];
23 #define STATUSSZ 256
24 static char STATUS[STATUSSZ];
25 #define INPUTSZ 256
26 static char INPUT[INPUTSZ];
27 
28 /* Argument buffers for execvp(). */
29 #define MAXARGS 256
30 static char *ARGS[MAXARGS];
31 
32 /* Listing view parameters. */
33 #define HEIGHT      (LINES-4)
34 #define STATUSPOS   (COLS-16)
35 
36 /* Listing view flags. */
37 #define SHOW_FILES      0x01u
38 #define SHOW_DIRS       0x02u
39 #define SHOW_HIDDEN     0x04u
40 
41 /* Marks parameters. */
42 #define BULK_INIT   5
43 #define BULK_THRESH 256
44 
45 /* Information associated to each entry in listing. */
46 typedef struct Row {
47     char *name;
48     off_t size;
49     mode_t mode;
50     int islink;
51     int marked;
52 } Row;
53 
54 /* Dynamic array of marked entries. */
55 typedef struct Marks {
56     char dirpath[PATH_MAX];
57     int bulk;
58     int nentries;
59     char **entries;
60 } Marks;
61 
62 /* Line editing state. */
63 typedef struct Edit {
64     char buffer[INPUTSZ-1];
65     int left, right;
66 } Edit;
67 
68 /* Global state. Some basic info is allocated for ten tabs. */
69 static struct Rover {
70     int tab;
71     int nfiles;
72     int scroll[10];
73     int esel[10];
74     uint8_t flags[10];
75     Row *rows;
76     WINDOW *window;
77     char cwd[10][PATH_MAX];
78     Marks marks;
79     Edit edit;
80     int edit_scroll;
81 } rover;
82 
83 /* Macros for accessing global state. */
84 #define ENAME(I)    rover.rows[I].name
85 #define ESIZE(I)    rover.rows[I].size
86 #define EMODE(I)    rover.rows[I].mode
87 #define ISLINK(I)   rover.rows[I].islink
88 #define MARKED(I)   rover.rows[I].marked
89 #define SCROLL      rover.scroll[rover.tab]
90 #define ESEL        rover.esel[rover.tab]
91 #define FLAGS       rover.flags[rover.tab]
92 #define CWD         rover.cwd[rover.tab]
93 
94 /* Helpers. */
95 #define MIN(A, B)   ((A) < (B) ? (A) : (B))
96 #define MAX(A, B)   ((A) > (B) ? (A) : (B))
97 #define ISDIR(E)    (strchr((E), '/') != NULL)
98 
99 /* Line Editing Macros. */
100 #define EDIT_FULL(E)       ((E).left == (E).right)
101 #define EDIT_CAN_LEFT(E)   ((E).left)
102 #define EDIT_CAN_RIGHT(E)  ((E).right < INPUTSZ-1)
103 #define EDIT_LEFT(E)       (E).buffer[(E).right--] = (E).buffer[--(E).left]
104 #define EDIT_RIGHT(E)      (E).buffer[(E).left++] = (E).buffer[++(E).right]
105 #define EDIT_INSERT(E, C)  (E).buffer[(E).left++] = (C)
106 #define EDIT_BACKSPACE(E)  (E).left--
107 #define EDIT_DELETE(E)     (E).right++
108 #define EDIT_CLEAR(E)      do { (E).left = 0; (E).right = INPUTSZ-1; } while(0)
109 
110 typedef enum EditStat {CONTINUE, CONFIRM, CANCEL} EditStat;
111 typedef enum Color {DEFAULT, RED, GREEN, YELLOW, BLUE, CYAN, MAGENTA, WHITE} Color;
112 typedef int (*PROCESS)(const char *path);
113 
114 static void
115 init_marks(Marks *marks)
116 {
117     strcpy(marks->dirpath, "");
118     marks->bulk = BULK_INIT;
119     marks->nentries = 0;
120     marks->entries = calloc(marks->bulk, sizeof *marks->entries);
121 }
122 
123 /* Unmark all entries. */
124 static void
125 mark_none(Marks *marks)
126 {
127     int i;
128 
129     strcpy(marks->dirpath, "");
130     for (i = 0; i < marks->bulk && marks->nentries; i++)
131         if (marks->entries[i]) {
132             free(marks->entries[i]);
133             marks->entries[i] = NULL;
134             marks->nentries--;
135         }
136     if (marks->bulk > BULK_THRESH) {
137         /* Reset bulk to free some memory. */
138         free(marks->entries);
139         marks->bulk = BULK_INIT;
140         marks->entries = calloc(marks->bulk, sizeof *marks->entries);
141     }
142 }
143 
144 static void
145 add_mark(Marks *marks, char *dirpath, char *entry)
146 {
147     int i;
148 
149     if (!strcmp(marks->dirpath, dirpath)) {
150         /* Append mark to directory. */
151         if (marks->nentries == marks->bulk) {
152             /* Expand bulk to accomodate new entry. */
153             int extra = marks->bulk / 2;
154             marks->bulk += extra; /* bulk *= 1.5; */
155             marks->entries = realloc(marks->entries,
156                                      marks->bulk * sizeof *marks->entries);
157             memset(&marks->entries[marks->nentries], 0,
158                    extra * sizeof *marks->entries);
159             i = marks->nentries;
160         } else {
161             /* Search for empty slot (there must be one). */
162             for (i = 0; i < marks->bulk; i++)
163                 if (!marks->entries[i])
164                     break;
165         }
166     } else {
167         /* Directory changed. Discard old marks. */
168         mark_none(marks);
169         strcpy(marks->dirpath, dirpath);
170         i = 0;
171     }
172     marks->entries[i] = malloc(strlen(entry) + 1);
173     strcpy(marks->entries[i], entry);
174     marks->nentries++;
175 }
176 
177 static void
178 del_mark(Marks *marks, char *entry)
179 {
180     int i;
181 
182     if (marks->nentries > 1) {
183         for (i = 0; i < marks->bulk; i++)
184             if (marks->entries[i] && !strcmp(marks->entries[i], entry))
185                 break;
186         free(marks->entries[i]);
187         marks->entries[i] = NULL;
188         marks->nentries--;
189     } else
190         mark_none(marks);
191 }
192 
193 static void
194 free_marks(Marks *marks)
195 {
196     int i;
197 
198     for (i = 0; i < marks->bulk && marks->nentries; i++)
199         if (marks->entries[i]) {
200             free(marks->entries[i]);
201             marks->nentries--;
202         }
203     free(marks->entries);
204 }
205 
206 static void update_view();
207 
208 /* SIGSEGV handler: clean up curses before exiting. */
209 static void
210 handle_segv(int sig)
211 {
212     (void) sig;
213     endwin();
214     fprintf(stderr, "Received SIGSEGV (segmentation fault).\n");
215     exit(1);
216 }
217 
218 /* SIGWINCH handler: resize application according to new terminal settings. */
219 static void
220 handle_winch(int sig)
221 {
222     (void) sig;
223     delwin(rover.window);
224     endwin();
225     refresh();
226     clear();
227     rover.window = subwin(stdscr, LINES - 2, COLS, 1, 0);
228     update_view();
229 }
230 
231 static void
232 enable_handlers()
233 {
234     struct sigaction sa;
235 
236     memset(&sa, 0, sizeof (struct sigaction));
237     sa.sa_handler = handle_segv;
238     sigaction(SIGSEGV, &sa, NULL);
239     sa.sa_handler = handle_winch;
240     sigaction(SIGWINCH, &sa, NULL);
241 }
242 
243 static void
244 disable_handlers()
245 {
246     struct sigaction sa;
247 
248     memset(&sa, 0, sizeof (struct sigaction));
249     sa.sa_handler = SIG_DFL;
250     sigaction(SIGSEGV, &sa, NULL);
251     sigaction(SIGWINCH, &sa, NULL);
252 }
253 
254 /* Do a fork-exec to external program (e.g. $EDITOR). */
255 static void
256 spawn()
257 {
258     pid_t pid;
259     int status;
260 
261     setenv("RVSEL", rover.nfiles ? ENAME(ESEL) : "", 1);
262     pid = fork();
263     if (pid > 0) {
264         /* fork() succeeded. */
265         disable_handlers();
266         endwin();
267         waitpid(pid, &status, 0);
268         enable_handlers();
269         kill(getpid(), SIGWINCH);
270     } else if (pid == 0) {
271         /* Child process. */
272         execvp(ARGS[0], ARGS);
273     }
274 }
275 
276 /* Curses setup. */
277 static void
278 init_term()
279 {
280     setlocale(LC_ALL, "");
281     initscr();
282     cbreak(); /* Get one character at a time. */
283     noecho();
284     nonl(); /* No NL->CR/NL on output. */
285     intrflush(stdscr, FALSE);
286     keypad(stdscr, TRUE);
287     curs_set(FALSE); /* Hide blinking cursor. */
288     if (has_colors()) {
289         short bg;
290         start_color();
291 #ifdef NCURSES_EXT_FUNCS
292         use_default_colors();
293         bg = -1;
294 #else
295         bg = COLOR_BLACK;
296 #endif
297         init_pair(RED, COLOR_RED, bg);
298         init_pair(GREEN, COLOR_GREEN, bg);
299         init_pair(YELLOW, COLOR_YELLOW, bg);
300         init_pair(BLUE, COLOR_BLUE, bg);
301         init_pair(CYAN, COLOR_CYAN, bg);
302         init_pair(MAGENTA, COLOR_MAGENTA, bg);
303         init_pair(WHITE, COLOR_WHITE, bg);
304     }
305     atexit((void (*)(void)) endwin);
306     enable_handlers();
307 }
308 
309 /* Update the listing view. */
310 static void
311 update_view()
312 {
313     int i, j;
314     int numsize;
315     int ishidden, isdir;
316     int marking;
317 
318     mvhline(0, 0, ' ', COLS);
319     attr_on(A_BOLD, NULL);
320     color_set(RVC_TABNUM, NULL);
321     mvaddch(0, COLS - 2, rover.tab + '0');
322     color_set(DEFAULT, NULL);
323     attr_off(A_BOLD, NULL);
324     if (rover.marks.nentries) {
325         numsize = snprintf(STATUS, STATUSSZ, "%d", rover.marks.nentries);
326         color_set(RVC_MARKS, NULL);
327         mvaddstr(0, COLS - 3 - numsize, STATUS);
328         color_set(DEFAULT, NULL);
329     } else
330         numsize = -1;
331     color_set(RVC_CWD, NULL);
332     mvaddnstr(0, 0, CWD, COLS - 4 - numsize);
333     color_set(DEFAULT, NULL);
334     wcolor_set(rover.window, RVC_BORDER, NULL);
335     wborder(rover.window, 0, 0, 0, 0, 0, 0, 0, 0);
336     wcolor_set(rover.window, DEFAULT, NULL);
337     /* Selection might not be visible, due to cursor wrapping or window
338        shrinking. In that case, the scroll must be moved to make it visible. */
339     SCROLL = MAX(MIN(SCROLL, ESEL), ESEL - HEIGHT + 1);
340     marking = !strcmp(CWD, rover.marks.dirpath);
341     for (i = 0, j = SCROLL; i < HEIGHT && j < rover.nfiles; i++, j++) {
342         ishidden = ENAME(j)[0] == '.';
343         isdir = S_ISDIR(EMODE(j));
344         if (j == ESEL)
345             wattr_on(rover.window, A_REVERSE, NULL);
346         if (ISLINK(j))
347             wcolor_set(rover.window, RVC_LINK, NULL);
348         else if (ishidden)
349             wcolor_set(rover.window, RVC_HIDDEN, NULL);
350         else if (isdir)
351             wcolor_set(rover.window, RVC_DIR, NULL);
352         else
353             wcolor_set(rover.window, RVC_FILE, NULL);
354         if (!isdir) {
355             char *suffix, *suffixes = "BKMGTPEZY";
356             off_t human_size = ESIZE(j) * 10;
357             for (suffix = suffixes; human_size >= 10240; suffix++)
358                 human_size = (human_size + 512) / 1024;
359             if (*suffix == 'B')
360                 snprintf(ROW, ROWSZ, "%s%*d %c", ENAME(j),
361                          (int) (COLS - strlen(ENAME(j)) - 6),
362                          (int) human_size / 10, *suffix);
363             else
364                 snprintf(ROW, ROWSZ, "%s%*d.%d %c", ENAME(j),
365                          (int) (COLS - strlen(ENAME(j)) - 8),
366                          (int) human_size / 10, (int) human_size % 10, *suffix);
367         } else
368             strcpy(ROW, ENAME(j));
369         mvwhline(rover.window, i + 1, 1, ' ', COLS - 2);
370         mvwaddnstr(rover.window, i + 1, 2, ROW, COLS - 4);
371         if (marking && MARKED(j)) {
372             wcolor_set(rover.window, RVC_MARKS, NULL);
373             mvwaddch(rover.window, i + 1, 1, RVS_MARK);
374         } else
375             mvwaddch(rover.window, i + 1, 1, ' ');
376         wcolor_set(rover.window, DEFAULT, NULL);
377         if (j == ESEL)
378             wattr_off(rover.window, A_REVERSE, NULL);
379     }
380     for (; i < HEIGHT; i++)
381         mvwhline(rover.window, i + 1, 1, ' ', COLS - 2);
382     if (rover.nfiles > HEIGHT) {
383         int center, height;
384         center = (SCROLL + (HEIGHT / 2)) * HEIGHT / rover.nfiles;
385         height = (HEIGHT-1) * HEIGHT / rover.nfiles;
386         if (!height) height = 1;
387         wcolor_set(rover.window, RVC_BORDER, NULL);
388         wborder(rover.window, 0, 0, 0, 0, 0, 0, 0, 0);
389         wcolor_set(rover.window, RVC_SCROLLBAR, NULL);
390         mvwvline(rover.window, center-(height/2)+1, COLS-1, RVS_SCROLLBAR, height);
391         wcolor_set(rover.window, DEFAULT, NULL);
392     }
393     STATUS[0] = FLAGS & SHOW_FILES  ? 'F' : ' ';
394     STATUS[1] = FLAGS & SHOW_DIRS   ? 'D' : ' ';
395     STATUS[2] = FLAGS & SHOW_HIDDEN ? 'H' : ' ';
396     if (!rover.nfiles)
397         strcpy(ROW, "0/0");
398     else
399         snprintf(ROW, ROWSZ, "%d/%d", ESEL + 1, rover.nfiles);
400     snprintf(STATUS+3, STATUSSZ-3, "%12s", ROW);
401     color_set(RVC_STATUS, NULL);
402     mvaddstr(LINES - 1, STATUSPOS, STATUS);
403     color_set(DEFAULT, NULL);
404     wrefresh(rover.window);
405 }
406 
407 /* Show a message on the status bar. */
408 static void
409 message(const char *msg, Color color)
410 {
411     int len, pos;
412 
413     len = strlen(msg);
414     pos = (STATUSPOS - len) / 2;
415     attr_on(A_BOLD, NULL);
416     color_set(color, NULL);
417     mvaddstr(LINES - 1, pos, msg);
418     color_set(DEFAULT, NULL);
419     attr_off(A_BOLD, NULL);
420 }
421 
422 /* Clear message area, leaving only status info. */
423 static void
424 clear_message()
425 {
426     mvhline(LINES - 1, 0, ' ', STATUSPOS);
427 }
428 
429 /* Comparison used to sort listing entries. */
430 static int
431 rowcmp(const void *a, const void *b)
432 {
433     int isdir1, isdir2, cmpdir;
434     const Row *r1 = a;
435     const Row *r2 = b;
436     isdir1 = S_ISDIR(r1->mode);
437     isdir2 = S_ISDIR(r2->mode);
438     cmpdir = isdir2 - isdir1;
439     return cmpdir ? cmpdir : strcoll(r1->name, r2->name);
440 }
441 
442 /* Get all entries in current working directory. */
443 static int
444 ls(Row **rowsp, uint8_t flags)
445 {
446     DIR *dp;
447     struct dirent *ep;
448     struct stat statbuf;
449     Row *rows;
450     int i, n;
451 
452     if(!(dp = opendir("."))) return -1;
453     n = -2; /* We don't want the entries "." and "..". */
454     while (readdir(dp)) n++;
455     rewinddir(dp);
456     rows = malloc(n * sizeof *rows);
457     i = 0;
458     while ((ep = readdir(dp))) {
459         if (!strcmp(ep->d_name, ".") || !strcmp(ep->d_name, ".."))
460             continue;
461         if (!(flags & SHOW_HIDDEN) && ep->d_name[0] == '.')
462             continue;
463         lstat(ep->d_name, &statbuf);
464         rows[i].islink = S_ISLNK(statbuf.st_mode);
465         stat(ep->d_name, &statbuf);
466         if (S_ISDIR(statbuf.st_mode)) {
467             if (flags & SHOW_DIRS) {
468                 rows[i].name = malloc(strlen(ep->d_name) + 2);
469                 strcpy(rows[i].name, ep->d_name);
470                 strcat(rows[i].name, "/");
471                 rows[i].mode = statbuf.st_mode;
472                 i++;
473             }
474         } else if (flags & SHOW_FILES) {
475             rows[i].name = malloc(strlen(ep->d_name) + 1);
476             strcpy(rows[i].name, ep->d_name);
477             rows[i].size = statbuf.st_size;
478             rows[i].mode = statbuf.st_mode;
479             i++;
480         }
481     }
482     n = i; /* Ignore unused space in array caused by filters. */
483     qsort(rows, n, sizeof (*rows), rowcmp);
484     closedir(dp);
485     *rowsp = rows;
486     return n;
487 }
488 
489 static void
490 free_rows(Row **rowsp, int nfiles)
491 {
492     int i;
493 
494     for (i = 0; i < nfiles; i++)
495         free((*rowsp)[i].name);
496     free(*rowsp);
497     *rowsp = NULL;
498 }
499 
500 /* Change working directory to the path in CWD. */
501 static void
502 cd(int reset)
503 {
504     int i, j;
505 
506     message("Loading...", CYAN);
507     refresh();
508     if (reset) ESEL = SCROLL = 0;
509     chdir(CWD);
510     if (rover.nfiles)
511         free_rows(&rover.rows, rover.nfiles);
512     rover.nfiles = ls(&rover.rows, FLAGS);
513     if (!strcmp(CWD, rover.marks.dirpath)) {
514         for (i = 0; i < rover.nfiles; i++) {
515             for (j = 0; j < rover.marks.bulk; j++)
516                 if (
517                     rover.marks.entries[j] &&
518                     !strcmp(rover.marks.entries[j], ENAME(i))
519                 )
520                     break;
521             MARKED(i) = j < rover.marks.bulk;
522         }
523     } else
524         for (i = 0; i < rover.nfiles; i++)
525             MARKED(i) = 0;
526     clear_message();
527     update_view();
528 }
529 
530 /* Select a target entry, if it is present. */
531 static void
532 try_to_sel(const char *target)
533 {
534     ESEL = 0;
535     if (!ISDIR(target))
536         while ((ESEL+1) < rover.nfiles && S_ISDIR(EMODE(ESEL)))
537             ESEL++;
538     while ((ESEL+1) < rover.nfiles && strcoll(ENAME(ESEL), target) < 0)
539         ESEL++;
540     if (rover.nfiles > HEIGHT) {
541         SCROLL = ESEL - (HEIGHT / 2);
542         SCROLL = MIN(MAX(SCROLL, 0), rover.nfiles - HEIGHT);
543     }
544 }
545 
546 /* Reload CWD, but try to keep selection. */
547 static void
548 reload()
549 {
550     if (rover.nfiles) {
551         strcpy(INPUT, ENAME(ESEL));
552         cd(1);
553         try_to_sel(INPUT);
554         update_view();
555     } else
556         cd(1);
557 }
558 
559 /* Recursively process a source directory using CWD as destination root.
560    For each node (i.e. directory), do the following:
561     1. call pre(destination);
562     2. call proc() on every child leaf (i.e. files);
563     3. recurse into every child node;
564     4. call pos(source).
565    E.g. to move directory /src/ (and all its contents) inside /dst/:
566     strcpy(CWD, "/dst/");
567     process_dir(adddir, movfile, deldir, "/src/"); */
568 static int
569 process_dir(PROCESS pre, PROCESS proc, PROCESS pos, const char *path)
570 {
571     int ret;
572     DIR *dp;
573     struct dirent *ep;
574     struct stat statbuf;
575     char subpath[PATH_MAX];
576 
577     ret = 0;
578     if (pre) {
579         char dstpath[PATH_MAX];
580         strcpy(dstpath, CWD);
581         strcat(dstpath, path + strlen(rover.marks.dirpath));
582         ret |= pre(dstpath);
583     }
584     if(!(dp = opendir(path))) return -1;
585     while ((ep = readdir(dp))) {
586         if (!strcmp(ep->d_name, ".") || !strcmp(ep->d_name, ".."))
587             continue;
588         snprintf(subpath, PATH_MAX, "%s%s", path, ep->d_name);
589         stat(subpath, &statbuf);
590         if (S_ISDIR(statbuf.st_mode)) {
591             strcat(subpath, "/");
592             ret |= process_dir(pre, proc, pos, subpath);
593         } else
594             ret |= proc(subpath);
595     }
596     closedir(dp);
597     if (pos) ret |= pos(path);
598     return ret;
599 }
600 
601 /* Process all marked entries using CWD as destination root.
602    All marked entries that are directories will be recursively processed.
603    See process_dir() for details on the parameters. */
604 static void
605 process_marked(PROCESS pre, PROCESS proc, PROCESS pos)
606 {
607     int i, ret;
608     char path[PATH_MAX];
609 
610     clear_message();
611     message("Processing...", CYAN);
612     refresh();
613     for (i = 0; i < rover.marks.bulk; i++)
614         if (rover.marks.entries[i]) {
615             ret = 0;
616             snprintf(path, PATH_MAX, "%s%s", rover.marks.dirpath, rover.marks.entries[i]);
617             if (ISDIR(rover.marks.entries[i])) {
618                 if (!strncmp(path, CWD, strlen(path)))
619                     ret = -1;
620                 else
621                     ret = process_dir(pre, proc, pos, path);
622             } else
623                 ret = proc(path);
624             if (!ret) del_mark(&rover.marks, rover.marks.entries[i]);
625         }
626     reload();
627     if (!rover.marks.nentries)
628         message("Done.", GREEN);
629     else
630         message("Some errors occured.", RED);
631 }
632 
633 /* Wrappers for file operations. */
634 static PROCESS delfile = unlink;
635 static PROCESS deldir = rmdir;
636 static int addfile(const char *path) {
637     /* Using creat(2) because mknod(2) doesn't seem to be portable. */
638     int ret;
639 
640     ret = creat(path, 0644);
641     if (ret < 0) return ret;
642     return close(ret);
643 }
644 static int cpyfile(const char *srcpath) {
645     int src, dst, ret;
646     size_t size;
647     struct stat st;
648     char buf[BUFSIZ];
649     char dstpath[PATH_MAX];
650 
651     ret = src = open(srcpath, O_RDONLY);
652     if (ret < 0) return ret;
653     ret = fstat(src, &st);
654     if (ret < 0) return ret;
655     strcpy(dstpath, CWD);
656     strcat(dstpath, srcpath + strlen(rover.marks.dirpath));
657     ret = dst = creat(dstpath, st.st_mode);
658     if (ret < 0) return ret;
659     while ((size = read(src, buf, BUFSIZ)) > 0)
660         write(dst, buf, size);
661     close(src);
662     close(dst);
663     return 0;
664 }
665 static int adddir(const char *path) {
666     int ret;
667     struct stat st;
668 
669     ret = stat(CWD, &st);
670     if (ret < 0) return ret;
671     return mkdir(path, st.st_mode);
672 }
673 static int movfile(const char *srcpath) {
674     int ret;
675     char dstpath[PATH_MAX];
676 
677     strcpy(dstpath, CWD);
678     strcat(dstpath, srcpath + strlen(rover.marks.dirpath));
679     ret = rename(srcpath, dstpath);
680     if (ret < 0 && errno == EXDEV) {
681         ret = cpyfile(srcpath);
682         if (ret < 0) return ret;
683         ret = delfile(srcpath);
684     }
685     return ret;
686 }
687 
688 static void
689 start_line_edit(const char *init_input)
690 {
691     curs_set(TRUE);
692     strncpy(INPUT, init_input, INPUTSZ);
693     strncpy(rover.edit.buffer, init_input, INPUTSZ);
694     rover.edit.left = strlen(init_input);
695     rover.edit.right = INPUTSZ - 1;
696     rover.edit_scroll = 0;
697 }
698 
699 /* Read input and change editing state accordingly. */
700 static EditStat
701 get_line_edit()
702 {
703     int ch = getch();
704     if (ch == '\r' || ch == '\n' || ch == KEY_ENTER) {
705         curs_set(FALSE);
706         return CONFIRM;
707     } else if (ch == '\t') {
708         curs_set(FALSE);
709         return CANCEL;
710     } else if (EDIT_CAN_LEFT(rover.edit) && ch == KEY_LEFT) {
711         EDIT_LEFT(rover.edit);
712     } else if (EDIT_CAN_RIGHT(rover.edit) && ch == KEY_RIGHT) {
713         EDIT_RIGHT(rover.edit);
714     } else if (ch == KEY_UP) {
715         while (EDIT_CAN_LEFT(rover.edit)) EDIT_LEFT(rover.edit);
716     } else if (ch == KEY_DOWN) {
717         while (EDIT_CAN_RIGHT(rover.edit)) EDIT_RIGHT(rover.edit);
718     } else if (ch == erasechar() || ch == KEY_BACKSPACE) {
719         if (EDIT_CAN_LEFT(rover.edit)) EDIT_BACKSPACE(rover.edit);
720     } else if (ch == KEY_DC) {
721         if (EDIT_CAN_RIGHT(rover.edit)) EDIT_DELETE(rover.edit);
722     } else if (ch == killchar()) {
723         EDIT_CLEAR(rover.edit);
724         clear_message();
725     } else if (!EDIT_FULL(rover.edit) && isprint(ch)) {
726         EDIT_INSERT(rover.edit, ch);
727     }
728     /* Copy edit contents to INPUT and append null character. */
729     strncpy(INPUT, rover.edit.buffer, rover.edit.left);
730     strncpy(&INPUT[rover.edit.left], &rover.edit.buffer[rover.edit.right+1],
731             INPUTSZ-rover.edit.left-1);
732     INPUT[rover.edit.left+INPUTSZ-rover.edit.right-1] = '\0';
733     return CONTINUE;
734 }
735 
736 /* Update line input on the screen. */
737 static void
738 update_input(char *prompt, Color color)
739 {
740     int plen, ilen, maxlen;
741 
742     plen = strlen(prompt);
743     ilen = strlen(INPUT);
744     maxlen = STATUSPOS - plen - 2;
745     if (ilen - rover.edit_scroll < maxlen)
746         rover.edit_scroll = MAX(ilen - maxlen, 0);
747     else if (rover.edit.left > rover.edit_scroll + maxlen - 1)
748         rover.edit_scroll = rover.edit.left - maxlen;
749     else if (rover.edit.left < rover.edit_scroll)
750         rover.edit_scroll = MAX(rover.edit.left - maxlen, 0);
751     color_set(RVC_PROMPT, NULL);
752     mvaddstr(LINES - 1, 0, prompt);
753     color_set(color, NULL);
754     mvaddnstr(LINES - 1, plen, &INPUT[rover.edit_scroll], maxlen);
755     mvaddch(LINES - 1, plen + MIN(ilen - rover.edit_scroll, maxlen + 1), ' ');
756     color_set(DEFAULT, NULL);
757     if (rover.edit_scroll)
758         mvaddch(LINES - 1, plen - 1, '<');
759     if (ilen > rover.edit_scroll + maxlen)
760         mvaddch(LINES - 1, plen + maxlen, '>');
761     move(LINES - 1, plen + rover.edit.left - rover.edit_scroll);
762 }
763 
764 int
765 main(int argc, char *argv[])
766 {
767     int i, ch;
768     char *program;
769     const char *key;
770     DIR *d;
771     EditStat edit_stat;
772     const char *save_cwd_file = NULL;
773 
774     if (argc >= 2) {
775         if (!strcmp(argv[1], "-v") || !strcmp(argv[1], "--version")) {
776             printf("rover %s\n", RV_VERSION);
777             return 0;
778         } else if (!strcmp(argv[1], "-h") || !strcmp(argv[1], "--help")) {
779             printf(
780                 "Usage: rover [DIRECTORY [DIRECTORY [DIRECTORY [...]]]]\n"
781                 "  or:  rover [OPTION]\n"
782                 "Browse current working directory or the ones specified.\n\n"
783                 "Options:\n"
784                 "  -h, --help       print this help message and exit\n"
785                 "  -v, --version    print program version and exit\n\n"
786                 "See rover(1) for more information.\n\n"
787                 "Rover homepage: <https://github.com/lecram/rover>.\n"
788             );
789             return 0;
790         } else if (argc > 2 && !strcmp(argv[1], "--save-cwd")) {
791             save_cwd_file = argv[2];
792             argc -= 2; argv += 2;
793         }
794     }
795     init_term();
796     rover.nfiles = 0;
797     for (i = 0; i < 10; i++) {
798         rover.esel[i] = rover.scroll[i] = 0;
799         rover.flags[i] = SHOW_FILES | SHOW_DIRS;
800     }
801     strcpy(rover.cwd[0], getenv("HOME"));
802     for (i = 1; i < argc && i < 10; i++) {
803         if ((d = opendir(argv[i]))) {
804             realpath(argv[i], rover.cwd[i]);
805             closedir(d);
806         } else
807             strcpy(rover.cwd[i], rover.cwd[0]);
808     }
809     getcwd(rover.cwd[i], PATH_MAX);
810     for (i++; i < 10; i++)
811         strcpy(rover.cwd[i], rover.cwd[i-1]);
812     for (i = 0; i < 10; i++)
813         if (rover.cwd[i][strlen(rover.cwd[i]) - 1] != '/')
814             strcat(rover.cwd[i], "/");
815     rover.tab = 1;
816     rover.window = subwin(stdscr, LINES - 2, COLS, 1, 0);
817     init_marks(&rover.marks);
818     cd(1);
819     while (1) {
820         ch = getch();
821         key = keyname(ch);
822         clear_message();
823         if (!strcmp(key, RVK_QUIT)) break;
824         else if (ch >= '0' && ch <= '9') {
825             rover.tab = ch - '0';
826             cd(0);
827         } else if (!strcmp(key, RVK_HELP)) {
828             ARGS[0] = "man";
829             ARGS[1] = "rover";
830             ARGS[2] = NULL;
831             spawn();
832         } else if (!strcmp(key, RVK_DOWN)) {
833             if (!rover.nfiles) continue;
834             ESEL = (ESEL + 1) % rover.nfiles;
835             update_view();
836         } else if (!strcmp(key, RVK_UP)) {
837             if (!rover.nfiles) continue;
838             ESEL = ESEL ? ESEL - 1 : rover.nfiles - 1;
839             update_view();
840         } else if (!strcmp(key, RVK_JUMP_DOWN)) {
841             if (!rover.nfiles) continue;
842             ESEL = MIN(ESEL + RV_JUMP, rover.nfiles - 1);
843             if (rover.nfiles > HEIGHT)
844                 SCROLL = MIN(SCROLL + RV_JUMP, rover.nfiles - HEIGHT);
845             update_view();
846         } else if (!strcmp(key, RVK_JUMP_UP)) {
847             if (!rover.nfiles) continue;
848             ESEL = MAX(ESEL - RV_JUMP, 0);
849             SCROLL = MAX(SCROLL - RV_JUMP, 0);
850             update_view();
851         } else if (!strcmp(key, RVK_JUMP_TOP)) {
852             if (!rover.nfiles) continue;
853             ESEL = 0;
854             SCROLL = 0;
855             update_view();
856         } else if (!strcmp(key, RVK_JUMP_BOTTOM)) {
857             if (!rover.nfiles) continue;
858             ESEL = rover.nfiles - 1;
859             SCROLL = HEIGHT - rover.nfiles - 1;
860             update_view();
861         } else if (!strcmp(key, RVK_CD_DOWN)) {
862             if (!rover.nfiles || !S_ISDIR(EMODE(ESEL))) continue;
863             strcat(CWD, ENAME(ESEL));
864             cd(1);
865         } else if (!strcmp(key, RVK_CD_UP)) {
866             char *dirname, first;
867             if (!strcmp(CWD, "/")) continue;
868             CWD[strlen(CWD) - 1] = '\0';
869             dirname = strrchr(CWD, '/') + 1;
870             first = dirname[0];
871             dirname[0] = '\0';
872             cd(1);
873             dirname[0] = first;
874             dirname[strlen(dirname)] = '/';
875             try_to_sel(dirname);
876             dirname[0] = '\0';
877             update_view();
878         } else if (!strcmp(key, RVK_HOME)) {
879             strcpy(CWD, getenv("HOME"));
880             if (CWD[strlen(CWD) - 1] != '/')
881                 strcat(CWD, "/");
882             cd(1);
883         } else if (!strcmp(key, RVK_REFRESH)) {
884             reload();
885         } else if (!strcmp(key, RVK_SHELL)) {
886             program = getenv("SHELL");
887             if (program) {
888                 ARGS[0] = program;
889                 ARGS[1] = NULL;
890                 spawn();
891                 reload();
892             }
893         } else if (!strcmp(key, RVK_VIEW)) {
894             if (!rover.nfiles || S_ISDIR(EMODE(ESEL))) continue;
895             program = getenv("PAGER");
896             if (program) {
897                 ARGS[0] = program;
898                 ARGS[1] = ENAME(ESEL);
899                 ARGS[2] = NULL;
900                 spawn();
901             }
902         } else if (!strcmp(key, RVK_EDIT)) {
903             if (!rover.nfiles || S_ISDIR(EMODE(ESEL))) continue;
904             program = getenv("EDITOR");
905             if (program) {
906                 ARGS[0] = program;
907                 ARGS[1] = ENAME(ESEL);
908                 ARGS[2] = NULL;
909                 spawn();
910                 cd(0);
911             }
912         } else if (!strcmp(key, RVK_SEARCH)) {
913             int oldsel, oldscroll, length;
914             char *prompt = "search: ";
915             if (!rover.nfiles) continue;
916             oldsel = ESEL;
917             oldscroll = SCROLL;
918             start_line_edit("");
919             update_input(prompt, DEFAULT);
920             while ((edit_stat = get_line_edit()) == CONTINUE) {
921                 int sel;
922                 Color color = RED;
923                 length = strlen(INPUT);
924                 if (length) {
925                     for (sel = 0; sel < rover.nfiles; sel++)
926                         if (!strncmp(ENAME(sel), INPUT, length))
927                             break;
928                     if (sel < rover.nfiles) {
929                         color = GREEN;
930                         ESEL = sel;
931                         if (rover.nfiles > HEIGHT) {
932                             if (sel < 3)
933                                 SCROLL = 0;
934                             else if (sel - 3 > rover.nfiles - HEIGHT)
935                                 SCROLL = rover.nfiles - HEIGHT;
936                             else
937                                 SCROLL = sel - 3;
938                         }
939                     }
940                 } else {
941                     ESEL = oldsel;
942                     SCROLL = oldscroll;
943                 }
944                 update_view();
945                 update_input(prompt, color);
946             }
947             if (edit_stat == CANCEL) {
948                 ESEL = oldsel;
949                 SCROLL = oldscroll;
950             }
951             clear_message();
952             update_view();
953         } else if (!strcmp(key, RVK_TG_FILES)) {
954             FLAGS ^= SHOW_FILES;
955             reload();
956         } else if (!strcmp(key, RVK_TG_DIRS)) {
957             FLAGS ^= SHOW_DIRS;
958             reload();
959         } else if (!strcmp(key, RVK_TG_HIDDEN)) {
960             FLAGS ^= SHOW_HIDDEN;
961             reload();
962         } else if (!strcmp(key, RVK_NEW_FILE)) {
963             int ok = 0;
964             char *prompt = "new file: ";
965             start_line_edit("");
966             update_input(prompt, DEFAULT);
967             while ((edit_stat = get_line_edit()) == CONTINUE) {
968                 int length = strlen(INPUT);
969                 ok = 1;
970                 for (i = 0; i < rover.nfiles; i++) {
971                     if (
972                         !strncmp(ENAME(i), INPUT, length) &&
973                         (!strcmp(ENAME(i) + length, "") ||
974                          !strcmp(ENAME(i) + length, "/"))
975                     ) {
976                         ok = 0;
977                         break;
978                     }
979                 }
980                 update_input(prompt, ok ? GREEN : RED);
981             }
982             clear_message();
983             if (edit_stat == CONFIRM && strlen(INPUT)) {
984                 if (ok) {
985                     addfile(INPUT);
986                     cd(1);
987                     try_to_sel(INPUT);
988                     update_view();
989                 } else
990                     message("File already exists.", RED);
991             }
992         } else if (!strcmp(key, RVK_NEW_DIR)) {
993             int ok = 0;
994             char *prompt = "new directory: ";
995             start_line_edit("");
996             update_input(prompt, DEFAULT);
997             while ((edit_stat = get_line_edit()) == CONTINUE) {
998                 int length = strlen(INPUT);
999                 ok = 1;
1000                 for (i = 0; i < rover.nfiles; i++) {
1001                     if (
1002                         !strncmp(ENAME(i), INPUT, length) &&
1003                         (!strcmp(ENAME(i) + length, "") ||
1004                          !strcmp(ENAME(i) + length, "/"))
1005                     ) {
1006                         ok = 0;
1007                         break;
1008                     }
1009                 }
1010                 update_input(prompt, ok ? GREEN : RED);
1011             }
1012             clear_message();
1013             if (edit_stat == CONFIRM && strlen(INPUT)) {
1014                 if (ok) {
1015                     adddir(INPUT);
1016                     cd(1);
1017                     try_to_sel(INPUT);
1018                     update_view();
1019                 } else
1020                     message("File already exists.", RED);
1021             }
1022         } else if (!strcmp(key, RVK_RENAME)) {
1023             int ok = 0;
1024             char *prompt = "rename: ";
1025             char *last;
1026             int isdir;
1027             strcpy(INPUT, ENAME(ESEL));
1028             last = INPUT + strlen(INPUT) - 1;
1029             if ((isdir = *last == '/'))
1030                 *last = '\0';
1031             start_line_edit(INPUT);
1032             update_input(prompt, RED);
1033             while ((edit_stat = get_line_edit()) == CONTINUE) {
1034                 int length = strlen(INPUT);
1035                 ok = 1;
1036                 for (i = 0; i < rover.nfiles; i++)
1037                     if (
1038                         !strncmp(ENAME(i), INPUT, length) &&
1039                         (!strcmp(ENAME(i) + length, "") ||
1040                          !strcmp(ENAME(i) + length, "/"))
1041                     ) {
1042                         ok = 0;
1043                         break;
1044                     }
1045                 update_input(prompt, ok ? GREEN : RED);
1046             }
1047             clear_message();
1048             if (edit_stat == CONFIRM && strlen(INPUT)) {
1049                 if (isdir)
1050                     strcat(INPUT, "/");
1051                 if (ok) {
1052                     if (!rename(ENAME(ESEL), INPUT) && MARKED(ESEL)) {
1053                         del_mark(&rover.marks, ENAME(ESEL));
1054                         add_mark(&rover.marks, CWD, INPUT);
1055                     }
1056                     cd(1);
1057                     try_to_sel(INPUT);
1058                     update_view();
1059                 } else
1060                     message("File already exists.", RED);
1061             }
1062         } else if (!strcmp(key, RVK_DELETE)) {
1063             if (rover.nfiles) {
1064                 message("Delete selected entry? (Y to confirm)", YELLOW);
1065                 if (getch() == 'Y') {
1066                     const char *name = ENAME(ESEL);
1067                     int ret = S_ISDIR(EMODE(ESEL)) ? deldir(name) : delfile(name);
1068                     reload();
1069                     if (ret)
1070                         message("Could not delete entry.", RED);
1071                 } else
1072                     clear_message();
1073             } else
1074                   message("No entry selected for deletion.", RED);
1075         } else if (!strcmp(key, RVK_TG_MARK)) {
1076             if (MARKED(ESEL))
1077                 del_mark(&rover.marks, ENAME(ESEL));
1078             else
1079                 add_mark(&rover.marks, CWD, ENAME(ESEL));
1080             MARKED(ESEL) = !MARKED(ESEL);
1081             ESEL = (ESEL + 1) % rover.nfiles;
1082             update_view();
1083         } else if (!strcmp(key, RVK_INVMARK)) {
1084             for (i = 0; i < rover.nfiles; i++) {
1085                 if (MARKED(i))
1086                     del_mark(&rover.marks, ENAME(i));
1087                 else
1088                     add_mark(&rover.marks, CWD, ENAME(i));
1089                 MARKED(i) = !MARKED(i);
1090             }
1091             update_view();
1092         } else if (!strcmp(key, RVK_MARKALL)) {
1093             for (i = 0; i < rover.nfiles; i++)
1094                 if (!MARKED(i)) {
1095                     add_mark(&rover.marks, CWD, ENAME(i));
1096                     MARKED(i) = 1;
1097                 }
1098             update_view();
1099         } else if (!strcmp(key, RVK_MARK_DELETE)) {
1100             if (rover.marks.nentries) {
1101                 message("Delete marked entries? (Y to confirm)", YELLOW);
1102                 if (getch() == 'Y')
1103                     process_marked(NULL, delfile, deldir);
1104                 else
1105                     clear_message();
1106             } else
1107                 message("No entries marked for deletion.", RED);
1108         } else if (!strcmp(key, RVK_MARK_COPY)) {
1109             if (rover.marks.nentries)
1110                 process_marked(adddir, cpyfile, NULL);
1111             else
1112                 message("No entries marked for copying.", RED);
1113         } else if (!strcmp(key, RVK_MARK_MOVE)) {
1114             if (rover.marks.nentries)
1115                 process_marked(adddir, movfile, deldir);
1116             else
1117                 message("No entries marked for moving.", RED);
1118         }
1119     }
1120     if (rover.nfiles)
1121         free_rows(&rover.rows, rover.nfiles);
1122     free_marks(&rover.marks);
1123     delwin(rover.window);
1124     if (save_cwd_file != NULL) {
1125         FILE *fd = fopen(save_cwd_file, "w");
1126         fputs(CWD, fd);
1127         fclose(fd);
1128     }
1129     return 0;
1130 }