ccc/ccc.c

#define _XOPEN_SOURCE 600

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <dirent.h>         /* directories etc. */
#include <sys/stat.h>
#include <errno.h>
#include <ftw.h>
#include <time.h>
#include <ncurses.h>

#include "file.h"
#include "util.h"
#include "config.h"

/* functions' definitions */
void show_help();
void start_ccc();
void change_dir(const char *buf, int selection);
int mkdir_p(const char *destdir);
void populate_files(const char *path, int ftype);
int get_directory_size(const char *fpath, const struct stat *sb, int typeflag, struct FTW *ftwbuf);
void add_file_stat(char *filepath, int ftype);
void highlight_current_line();
void show_file_content();
void edit_file();
int write_last_d();
void wpprintw(const char *line);
void init_windows();
void draw_border_title(WINDOW *window, bool active);

/* global variables */
unsigned int focus = 0;
long current_selection = 0;
bool dirs_size = DIRS_SIZE;
bool show_hidden = SHOW_HIDDEN;
bool file_details = SHOW_DETAILS;
char *cwd;
char *p_cwd; /* previous cwd */
int half_width;
ArrayList *files;
ArrayList *marked;
WINDOW *directory_border;
WINDOW *directory_content;
WINDOW *preview_border;
WINDOW *preview_content;
WINDOW *panel;

unsigned long total_dir_size;

int main(int argc, char** argv)
{
    if (argc > 1 && strcmp(argv[1], "-h") == 0)
        die("Usage: ccc filename");

    /* check if it is interactive shell */
    if (!isatty(STDIN_FILENO)) 
        die("ccc: No tty detected. ccc requires an interactive shell to run.\n");

    /* initialize screen, don't print special chars,
     * make ctrl + c work, don't show cursor 
     * enable arrow keys */
    initscr();
    noecho();
    cbreak();
    curs_set(0);
    keypad(stdscr, TRUE);

    /* check terminal has colors */
    if (!has_colors()) {
        endwin();
        die("ccc: Color is not supported in your terminal.\n");
    } else {
        use_default_colors();
        start_color();
    }

    /* colors */
    init_pair(1, COLOR_BLACK, -1);      /*  */
    init_pair(2, COLOR_RED, -1);        /*  */
    init_pair(3, COLOR_GREEN, -1);      /* LNK */
    init_pair(4, COLOR_YELLOW, -1);     /* BLK */
    init_pair(5, COLOR_BLUE, -1);       /* DIR */ 
    init_pair(6, COLOR_MAGENTA, -1);    /*  */
    init_pair(7, COLOR_CYAN, -1);       /* MARKED FILES */
    init_pair(8, COLOR_WHITE, -1);      /* REG */

    /* init files and marked arrays */
    files = arraylist_init(100);
    marked = arraylist_init(100);

    cwd = memalloc(PATH_MAX * sizeof(char));
    p_cwd = memalloc(PATH_MAX * sizeof(char));
    getcwd(cwd, PATH_MAX);
    start_ccc();

    int ch, ch2;
    while (1) {
        if (COLS < 80 || LINES < 24) {
            endwin();
            die("ccc: Terminal size needs to be at least 80x24\n");
        }
        ch = getch();
        /* printf("%d ",ch); */
        switch (ch) {
            /* quit */
            case 'q':
                if (write_last_d() == -1) {
                    /* prompt user so error message can be shown to user */
                    getch();
                }
                endwin();
                return 0;

            /* reload using z */
            case 'z':
                change_dir(cwd, 0); 
                break;

            /* go back by backspace or h or left arrow */
            case BACKSPACE:
            case LEFT:
            case 'h':;
                /* get parent directory */
                strcpy(p_cwd, cwd);
                char *last_slash = strrchr(cwd, '/');
                if (last_slash != NULL) {
                    *last_slash = '\0';
                    change_dir(cwd, 0);
                }
                break;

            /* enter directory/open a file using enter or l or right arrow */
            case ENTER:
            case RIGHT:
            case 'l':;
                strcpy(p_cwd, cwd);
                file c_file = files->items[current_selection];
                /* check if it is directory or a regular file */
                if (strncmp(c_file.type, "DIR", 3) == 0) {
                    /* change cwd to directory */
                    change_dir(c_file.path, 0); 
                } else if (strncmp(c_file.type, "REG", 3) == 0) {
                    edit_file();
                }
                break;

            /* jump up (ctrl u) */
            case CTRLU:
                if ((current_selection - JUMP_NUM) > 0)
                    current_selection -= JUMP_NUM;
                else
                    current_selection = 0;

                highlight_current_line();
                break;

            /* go up by k or up arrow */
            case UP:
            case 'k':
                if (current_selection > 0)
                    current_selection--;

                highlight_current_line();
                break;

            /* jump down (ctrl d) */
            case CTRLD:
                if ((current_selection + JUMP_NUM) < (files->length - 1))
                    current_selection += JUMP_NUM;
                else
                    current_selection = (files->length - 1);

                highlight_current_line();
                break;

            /* go down by j or down arrow */
            case DOWN:
            case 'j':
                if (current_selection < (files->length - 1))
                    current_selection++;

                highlight_current_line();
                break;

            /* jump to the bottom */
            case 'G':
                current_selection = (files->length - 1);
                highlight_current_line();
                break;

            /* jump to the top */
            case 'g':
                ch2 = getch();
                switch (ch2) {
                    case 'g':
                        current_selection = 0;
                        highlight_current_line();
                        break;
                    default:
                        break;
                }
                break;

            /* '~' go to $HOME */
            case TILDE:;
                char *home = getenv("HOME");
                if (home == NULL) {
                    wpprintw("$HOME is not defined");
                } else {
                    change_dir(home, 0);
                }
                break;

            /* go to the trash dir */
            case 't':;
                #ifdef TRASH_DIR
                    char *trash_dir = TRASH_DIR;
                #else
                    char *trash_dir = getenv("CCC_TRASH");
                #endif
                if (trash_dir == NULL) {
                    wpprintw("$CCC_TRASH is not defined");
                } else {
                    if (access(trash_dir, F_OK) != 0) {
                        /* create the directory with 755 perm if it doesn't exit */
                        if (mkdir_p(trash_dir) == -1) {
                            switch (errno) {
                                case EACCES:
                                    wpprintw("Parent directory does not allow write permission or one of directories does not allow search access");
                            }
                        }
                    }
                    change_dir(trash_dir, 0);
                }
                break;

            /* show directories' sizes */
            case 'A':
                dirs_size = !dirs_size;
                change_dir(cwd, 0);
                break;

            /* go to previous dir */
            case '-':
                change_dir(p_cwd, 0);
                break;

            /* show help */
            case '?':
                show_help();
                break;
            
            /* toggle hidden files */
            case '.':
                show_hidden = !show_hidden;
                change_dir(cwd, 0);
                break;

            /* toggle file details */
            case 'i':
                file_details = !file_details;
                change_dir(cwd, 0);

            /* mark one file */
            case SPACE:
                add_file_stat(files->items[current_selection].path, 1);
                highlight_current_line();
                break;

            /* mark all files in directory */
            case 'a':
                populate_files(cwd, 2);
                change_dir(cwd, current_selection); /* reload current dir */
                break;

            /* mark actions: */
            /* delete */
            case 'd':;
                if (marked->length) {
                    ;
                }
                break;

            /* move */
            case 'm':;
                if (marked->length) {
                    ;
                }
                break;

            /* copy */
            case 'c':;
                if (marked->length) {
                    ;
                }
                break;

            /* symbolic link */
            case 's':;
                if (marked->length) {
                    ;
                }
                break;

            /* bulk rename */
            case 'b':;
                if (marked->length) {
                    ;
                }
                break;
            
            /* escape */
            case ESC:
                break;

            case KEY_RESIZE:
                delwin(directory_border);
                delwin(directory_content);
                delwin(preview_border);
                delwin(preview_content);
                delwin(panel);
                endwin();
                start_ccc();
                break;
            default:
                break;
        }
    }
    arraylist_free(files);
    arraylist_free(marked);
    endwin();
    return 0;
}

void show_help()
{
    wclear(directory_content);
    wclear(preview_content);
    wprintw(directory_content,"h: go to parent dir\nj: scroll down\nk: scroll up\nl: go to child dir\n\nleft:  go to parent dir\ndown:  scroll down\nup:    scroll up\nright: go to child dir\n\nenter: go to child dir/open file\nbackspace: go to parent dir\n\ngg: go to top\nG: go to bottom\n\nctrl+u: jump up\nctrl+d: jump down\n\nt: go to trash dir\n~: go to home dir\n-: go to previous dir\nz: refresh current dir\n\nA: show directory disk usage/block size\nspace: mark file\na: mark all files in directory\n\n?: show help\nq: exit with last dir written to file\nctrl+c exit without writing last dir");
    wpprintw("Visit https://github.com/piotr-marendowski/ccc or use 'man ccc' for help");
    wrefresh(directory_content);
    wrefresh(preview_content);
}

void start_ccc()
{
    half_width = COLS / 2;
    init_windows();
    refresh();
    populate_files(cwd, 0);
    highlight_current_line();
}

/*
 * Change directory in window with selection
 */
void change_dir(const char *buf, int selection)
{
    char *buf_dup;
    if (buf == p_cwd) {
        buf_dup = strdup(p_cwd);
        if (buf_dup == NULL) {
            return;
        }
    } else {
        buf_dup = (char *) buf;
    }
    strcpy(cwd, buf_dup);
    arraylist_free(files);
    files = arraylist_init(100);
    populate_files(cwd, 0);
    current_selection = selection;
    highlight_current_line();
}

/*
 * Recursively create directory by creating each subdirectory
 * like mkdir -p
 */
int mkdir_p(const char *destdir)
{
    char *path = memalloc(PATH_MAX * sizeof(char));
    char dir_path[PATH_MAX] = "";

    if (destdir[0] == '~') {
        char *home = getenv("HOME");
        if (home == NULL) {
            wpprintw("$HOME is not defined");
            return -1;
        }
        /* replace ~ with home */
        snprintf(path, PATH_MAX, "%s%s", home, destdir + 1);
    } else {
        strcpy(path, destdir);
     }

    /* fix first / not appearing in the string */
    if (path[0] == '/')
        strcat(dir_path, "/");

    char *token = strtok(path, "/");
    while (token != NULL) {
        strcat(dir_path, token);
        strcat(dir_path, "/");

        if (mkdir(dir_path, 0755) == -1) {
            struct stat st;
            if (stat(dir_path, &st) == 0 && S_ISDIR(st.st_mode)) {
                /* Directory already exists, continue to the next dir */
                token = strtok(NULL, "/");
                continue;
            }
            
            perror("ccc");
            free(path);
            return -1;
        }
        token = strtok(NULL, "/");
    }

    free(path);
    return 0;
}

/*
 * Read the provided directory and add all files in directory to linked list
 * ftype: normal files = 0, marked = 1, marking ALL = 2
 * ep->d_name -> filename
 */
void populate_files(const char *path, int ftype)
{
    DIR *dp;
    struct dirent *ep;

    #if DRAW_BORDERS
        draw_border_title(directory_border, true);
    #endif
    if ((dp = opendir(path)) != NULL) {
        /* clear directory window to ready for printing */
        wclear(directory_content);

        while ((ep = readdir(dp)) != NULL) {
            char *filename = memalloc(PATH_MAX * sizeof(char));
            /* make filename be basename of selected item just to pass check */
            filename[0] = '\0';
            strcat(filename, ep->d_name);

            /* use strncmp to filter out dotfiles */
            if ((show_hidden && strncmp(filename, ".", 1) && strncmp(filename, "..", 2)) || (!show_hidden && strcmp(filename, ".") && strcmp(filename, ".."))) {
                /* construct full file path */
                filename[0] = '\0';
                strcat(filename, cwd);
                strcat(filename, "/");
                strcat(filename, ep->d_name);   /* add filename */

                add_file_stat(filename, ftype);
            }
            free(filename);
        }
        closedir(dp);
        wrefresh(directory_content);
    } else {
        perror("ccc");
    }
}

int get_directory_size(const char *fpath, const struct stat *sb, int typeflag, struct FTW *ftwbuf)
{
    total_dir_size += sb->st_size;
    return 0;
}

/*
 * Get file's last modified time, size, type
 * Add that file into list
 * ftype: normal file = 0, normal marked = 1, marking ALL = 2
 */
void add_file_stat(char *filepath, int ftype)
{
    struct stat file_stat;
    if (stat(filepath, &file_stat) == -1) {
        /* can't be triggered? */
        if (errno == EACCES)
            arraylist_add(files, filepath, "", "", 8, false, false);
    }
    
    /* get file type and color, 4 chars for the type */
    char *type = memalloc(4 * sizeof(char));
    int color;
    if (S_ISDIR(file_stat.st_mode)) {
        strcpy(type, "DIR");            /* directory type */
        color = 5;                      /* blue color */
    } else if (S_ISREG(file_stat.st_mode)) {
        strcpy(type, "REG");            /* regular file */
        color = 8;                      /* white color */
    } else if (S_ISLNK(file_stat.st_mode)) {
        strcpy(type, "LNK");            /* symbolic link */
        color = 3;                      /* green color */
    } else if (S_ISCHR(file_stat.st_mode)) {
        strcpy(type, "CHR");            /* character device */
        color = 8;                      /* white color */
    } else if (S_ISSOCK(file_stat.st_mode)) {
        strcpy(type, "SOC");            /* socket */
        color = 8;                      /* white color */
    } else if (S_ISBLK(file_stat.st_mode)) {
        strcpy(type, "BLK");            /* block device */
        color = 4;                      /* yellow color */
    } else if (S_ISFIFO(file_stat.st_mode)) {
        strcpy(type, "FIF");            /* FIFO */
        color = 8;                      /* white color */
    } else {
        color = 8;                      /* white color */
    }

    /* if file is to be marked */
    if (ftype == 1 || ftype == 2) {
        /* force if user is marking all files */
        bool force = ftype == 2 ? true : false;
        arraylist_add(marked, filepath, NULL, type, 8, true, force);
        /* free type and return without allocating more stuff */
        free(type);
        return;
    }

    /* get last modified time */
    char *time = memalloc(20 * sizeof(char));
    /* format last modified time to a string */
    strftime(time, 20, "%Y-%m-%d %H:%M", localtime(&file_stat.st_mtime));
    
    /* get file size */
    double bytes = file_stat.st_size;
    
    if (dirs_size) {
        /* dirs_size is true, so calculate disk usage */
        if (S_ISDIR(file_stat.st_mode)) {
            /* at most 15 fd opened */
            total_dir_size = 0;
            nftw(filepath, &get_directory_size, 15, FTW_PHYS);
            bytes = total_dir_size;
        }
    }
    /* max 25 chars due to long, space, suffix and null */
    char *size = memalloc(25 * sizeof(char));
    int unit = 0;
    const char* units[] = {"B", "KiB", "MiB", "GiB", "TiB", "PiB"};
    while (bytes > 1024) {
        bytes /= 1024;
        unit++;
    }
    /* display sizes */
    sprintf(size, "%.3g%s", bytes, units[unit]);

    char *total_stat = memalloc(45 * sizeof(char));
    snprintf(total_stat, 45, "%-18s %-8s", time, size);
    total_stat[strlen(total_stat)] = '\0';

    arraylist_add(files, filepath, total_stat, type, color, false, false);

    free(time);
    free(size);
    free(total_stat);
    free(type);
}


/*
 * Highlight current line by reversing the color
 */
void highlight_current_line()
{
    long overflow = 0;
    if (current_selection > LINES - 4) {
        /* overflown */
        overflow = current_selection - (LINES - 4);
    }

    /* calculate range of files to show */
    long range = files->length;
    /* not highlight if no files in directory */
    if (range == 0) {
        #if DRAW_PREVIEW
            wprintw(preview_content, "empty directory");
            wrefresh(preview_content);
        #endif
        return;
    }

    if (range > LINES - 3) {
        /* if there are more files than lines available to display
         * shrink range to avaiable lines to display with
         * overflow to keep the number of iterations to be constant */
        range = LINES - 3 + overflow;
    }
    
    wclear(directory_content);
    long line_count = 0;
    for (long i = overflow; i < range; i++) {
        if ((overflow == 0 && i == current_selection) || (overflow != 0 && i == current_selection)) {
            wattron(directory_content, A_REVERSE);

            /* update the panel */
            wclear(panel);
            
            /* check for marked files */
            long num_marked = marked->length;
            if (num_marked > 0) {
                /* Determine length of formatted string */
                int m_len = snprintf(NULL, 0, "[%ld] selected", num_marked);
                char *selected = memalloc((m_len + 1) * sizeof(char));

                snprintf(selected, m_len + 1, "[%ld] selected", num_marked);
                wprintw(panel, "(%ld/%ld) %s %s", current_selection + 1, files->length, selected, cwd);
            } else  {
                wprintw(panel, "(%ld/%ld) %s", current_selection + 1, files->length, cwd);
            }
        }
        /* print the actual filename and stats */
        char *line = get_line(files, i, file_details);
        int color = files->items[i].color;
        /* check is file marked for action */
        bool is_marked = arraylist_includes(marked, files->items[i].path);
        if (is_marked) {
            /* show file is selected */
            wattron(directory_content, COLOR_PAIR(7));
        } else {
            /* print the whole directory with default colors */
            wattron(directory_content, COLOR_PAIR(color));
        }
        
        if (overflow > 0)
            mvwprintw(directory_content, line_count, 0, "%s", line);
        else
            mvwprintw(directory_content, i, 0, "%s", line);

        if (is_marked) {
            wattroff(directory_content, COLOR_PAIR(7));
        } else {
            wattroff(directory_content, COLOR_PAIR(color));
        }

        wattroff(directory_content, A_REVERSE);
        free(line);
        line_count++;
    }

    wrefresh(directory_content);
    wrefresh(panel);
    /* show file content every time cursor changes */
    #if DRAW_PREVIEW
        show_file_content();
    #endif
    wrefresh(preview_content);
}

/*
 * Get file content into buffer and show it to preview window
 */
void show_file_content()
{
    wclear(preview_content);
    file current_file = files->items[current_selection];

    if (strncmp(current_file.type, "DIR", 3) == 0)
        return;

    FILE *file = fopen(current_file.path, "r");
    if (file == NULL) {
        mvwprintw(preview_content, 0, 0, "Unable to read %s", current_file.path);
        return;
    }
    #if DRAW_BORDERS
        draw_border_title(preview_border, true);
    #endif

    int c;
    /* check if its binary */
    while ((c = fgetc(file)) != EOF) {
        if (c == '\0') {
            mvwprintw(preview_content, 0, 0, "binary");
            return;
        }
    }

    fseek(file, 0, SEEK_END);
    long length = ftell(file);
    /* check if file isn't empty */
    if (length != 0) {
        fseek(file, 0, SEEK_SET); /* set cursor back to start of file */
        int max_length = (LINES - 3) * (COLS / 2 - 2);
        char *buffer = memalloc(max_length * sizeof(char));
        fread(buffer, 1, max_length, file);
        mvwprintw(preview_content, 0, 0, "%s", buffer);
        free(buffer);
    } else {
        wclear(preview_content);
    }
    fclose(file);
}

/*
 * Opens $EDITOR to edit the file
 */
void edit_file()
{
    #ifdef EDITOR
        char *editor = EDITOR;
    #else
        char *editor = getenv("EDITOR");
    #endif

    if (editor == NULL) {
        wpprintw("Cannot get EDITOR variable, is it defined?");
        return;
    } else {
        def_prog_mode();    /* save the tty modes */
        endwin();           /* end curses mode temporarily */

        char *filename = files->items[current_selection].path;
        int length = strlen(editor) + strlen(filename) + 2; /* one for space one for null */
        char command[length];

        snprintf(command, length, "%s %s", editor, filename);
        system(command);
        reset_prog_mode();  /* return to previous tty mode */
        refresh();          /* store the screen contents */
        free(filename);
    }
}

int write_last_d()
{
    #ifdef LAST_D
        char *last_d = memalloc(PATH_MAX * sizeof(char)); 
        strcpy(last_d, LAST_D);
    #else
        char *last_d = getenv("CCC_LAST_D");
    #endif
    if (last_d == NULL) {
        wpprintw("Cannot get CCC_LAST_D variable, is it defined?");
        return -1;
    } else {
        char *last_ddup = memalloc(PATH_MAX * sizeof(char));
        char *home = getenv("HOME");
        if (home == NULL) {
            wpprintw("$HOME is not defined");
            return -1;
        }
        /* replace ~ with home */
        snprintf(last_ddup, PATH_MAX, "%s%s", home, last_d + 1);

        char *last_d_dir = strrchr(last_d, '/');
        if (last_d_dir != NULL) {
            *last_d_dir = '\0'; /* truncate string */
        }
        mkdir_p(last_d);
        FILE *last_d_file = fopen(last_ddup, "w");
        if (last_d_file == NULL) {
            wpprintw("Cannot open last directory file");
            return -1;
        }
        fwrite(cwd, strlen(cwd), sizeof(char), last_d_file);
        fclose(last_d_file);
    }
    return 0;
}

/*
 * Print line to the panel
 */
void wpprintw(const char *line)
{
    wclear(panel);
    wprintw(panel, "%s", line);
    wrefresh(panel);
}

void init_windows()
{
    /* offset for width of the content 1 and 2 */
    int width_left = half_width;
    int width_right = half_width;
    #ifdef WINDOW_OFFSET
        width_left += WINDOW_OFFSET;
        width_right -= WINDOW_OFFSET;
    #endif

    /*------------------------------+
    |----border(0)--||--border(2)--||
    ||              ||             ||
    || content (1)  || content (3) ||
    || (directory)  ||  (preview)  ||
    ||              ||             ||
    |---------------||-------------||
    +==========panel (4)===========*/
    
    /*                         lines,          cols,            y,          x             */
    panel =             newwin(PH,             COLS,            LINES - PH, 0             );
    /* draw border around windows */
    #if DRAW_BORDERS
    directory_border =  newwin(LINES - PH,     width_left,      0,          0             );
    directory_content = newwin(LINES - PH - 2, width_left - 2,  1,          1             );

    preview_border =    newwin(LINES - PH,     width_right,     0,          width_left    );
    preview_content =   newwin(LINES - PH - 2, width_right - 2, 1,          width_left + 1);
    
    draw_border_title(directory_border, true);
    draw_border_title(preview_border, true);
    #else
    /* if there are no borders, then draw content in their places */
    directory_border =  newwin(0,              0,               COLS,       LINES         );
    preview_border =    newwin(0,              0,               COLS,       LINES         );
    /* -1 for the one space to the left */
    directory_content = newwin(LINES - PH - 1, width_left,      0,          1             );
    preview_content =   newwin(LINES - PH,     width_right,     0,          width_left    );
    #endif

    scrollok(directory_content, true);
}

/*
 * Draw the border of the window depending if it's active or not,
 */
void draw_border_title(WINDOW *window, bool active)
{
    /* check if the window is directory of preview */
    int width = half_width;
    if (window == directory_border) {          /* left */
        width += WINDOW_OFFSET;    
    } else if (window == preview_border) {   /* right */
        width -= WINDOW_OFFSET;
    }

    /* turn on color depends on active */
    if (active) {
        wattron(window, COLOR_PAIR(7));
    } else {
        wattron(window, COLOR_PAIR(5));
    }
    /* draw top border */
    mvwaddch(window, 0, 0, ACS_ULCORNER);  /* upper left corner */
    mvwhline(window, 0, 1, ACS_HLINE, width - 2); /* top horizontal line */
    mvwaddch(window, 0, width - 1, ACS_URCORNER); /* upper right corner */

    /* draw side border */
    mvwvline(window, 1, 0, ACS_VLINE, LINES - 2); /* left vertical line */
    mvwvline(window, 1, width - 1, ACS_VLINE, LINES - 2); /* right vertical line */

    /* draw bottom border
     * make space for the panel */
    mvwaddch(window, LINES - PH - 1, 0, ACS_LLCORNER); /* lower left corner */
    mvwhline(window, LINES - PH - 1, 1, ACS_HLINE, width - 2); /* bottom horizontal line */
    mvwaddch(window, LINES - PH - 1, width - 1, ACS_LRCORNER); /* lower right corner */

    /* turn color off after turning it on */
    if (active) {
        wattroff(window, COLOR_PAIR(7));
    } else {
        wattroff(window, COLOR_PAIR(5));
    }
    wrefresh(window);  /* Refresh the window to see the colored border and title */
}