Add colored filenames based on file types

This commit is contained in:
Piotr Marendowski 2024-03-14 15:58:03 +01:00
parent 8199ec3ecb
commit fa1eb500ac
4 changed files with 78 additions and 36 deletions

View file

@ -36,5 +36,3 @@ uninstall:
clean:
$(RM) $(TARGET)
ccc.o: $(CONF)

69
ccc.c
View file

@ -71,8 +71,15 @@ int main(int argc, char** argv)
start_color();
}
init_pair(1, COLOR_BLUE, -1); /* foreground, background */
init_pair(2, COLOR_CYAN, -1); /* active 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); /* */
init_pair(8, COLOR_WHITE, -1); /* REG */
half_width = COLS / 2;
init_windows();
@ -220,7 +227,7 @@ int main(int argc, char** argv)
}
/*
* Read the provided directory and list all files to window 0
* Read the provided directory and list all files to window 1
* ep->d_name -> filename
*/
void list_files(const char *path)
@ -246,12 +253,13 @@ void list_files(const char *path)
filename[0] = '\0';
strcat(filename, cwd);
strcat(filename, "/");
strcat(filename, ep->d_name); /* add file name */
strcat(filename, ep->d_name); /* add file name */
long index = add_file_stat(filename);
char *line = get_line(index);
mvwprintw(directory_content, count, 0, "%s", line);
free(line);
count++;
}
@ -289,7 +297,6 @@ long add_file_stat(char *filepath)
/* 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 (S_ISDIR(file_stat.st_mode)) {
@ -298,7 +305,7 @@ long add_file_stat(char *filepath)
nftw(filepath, &get_directory_size, 15, FTW_PHYS);
bytes = total_dir_size;
}
/* max 25 chars due to long, space, suffix and nul */
/* 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"};
@ -309,22 +316,32 @@ long add_file_stat(char *filepath)
/* 4 sig fig, limiting characters to have better format */
sprintf(size, "%-6.4g %-3s", bytes, units[unit]);
/* get file type */
char *type = memalloc(4 * sizeof(char)); /* 3 chars for type */
/* get file type and color */
char *type = memalloc(4 * sizeof(char)); /* 4 chars for type */
int color;
if (S_ISDIR(file_stat.st_mode)) {
strcpy(type, "DIR"); /* directory */
strcpy(type, "DIR"); /* directory type */
color = 5; /* blue color */
} else if (S_ISREG(file_stat.st_mode)) {
strcpy(type, "REG"); /* regular file */
strcpy(type, "REG"); /* regular file */
color = 8; /* white color */
} else if (S_ISLNK(file_stat.st_mode)) {
strcpy(type, "LNK"); /* symbolic link */
strcpy(type, "LNK"); /* symbolic link */
color = 3; /* green color */
} else if (S_ISCHR(file_stat.st_mode)) {
strcpy(type, "CHR"); /* character device */
strcpy(type, "CHR"); /* character device */
color = 8; /* white color */
} else if (S_ISSOCK(file_stat.st_mode)) {
strcpy(type, "SOC"); /* socket */
strcpy(type, "SOC"); /* socket */
color = 8; /* white color */
} else if (S_ISBLK(file_stat.st_mode)) {
strcpy(type, "BLK"); /* block device */
strcpy(type, "BLK"); /* block device */
color = 4; /* yellow color */
} else if (S_ISFIFO(file_stat.st_mode)) {
strcpy(type, "FIF"); /* FIFO */
strcpy(type, "FIF"); /* FIFO */
color = 8; /* white color */
} else {
color = 8; /* white color */
}
/* don't know how to handle socket, block device, character device and FIFO */
@ -335,7 +352,7 @@ long add_file_stat(char *filepath)
free(time);
free(size);
long index = add_file(filepath, total_stat, type);
long index = add_file(filepath, total_stat, type, color);
free(total_stat);
free(type);
@ -367,7 +384,7 @@ void highlight_current_line()
for (long i = overflow; i < range; i++) {
if ((overflow == 0 && i == current_selection) || (overflow != 0 && i == current_selection)) {
wattron(directory_content, A_REVERSE);
wattron(directory_content, COLOR_PAIR(1));
wattron(directory_content, COLOR_PAIR(5));
/* update the panel */
wclear(panel);
@ -375,12 +392,18 @@ void highlight_current_line()
}
/* print the actual filename and stats */
char *line = get_line(i);
int color = get_color(i);
/* print the whole directory with 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);
wattroff(directory_content, A_REVERSE);
wattroff(directory_content, COLOR_PAIR(1));
wattroff(directory_content, COLOR_PAIR(color));
wattroff(directory_content, COLOR_PAIR(5));
free(line);
line_count++;
}
@ -482,9 +505,9 @@ void draw_border_title(WINDOW *window, bool active)
{
/* turn on color depends on active */
if (active) {
wattron(window, COLOR_PAIR(2));
wattron(window, COLOR_PAIR(7));
} else {
wattron(window, COLOR_PAIR(1));
wattron(window, COLOR_PAIR(5));
}
/* draw top border */
mvwaddch(window, 0, 0, ACS_ULCORNER); /* upper left corner */
@ -503,9 +526,9 @@ void draw_border_title(WINDOW *window, bool active)
/* turn color off after turning it on */
if (active) {
wattroff(window, COLOR_PAIR(2));
wattroff(window, COLOR_PAIR(7));
} else {
wattroff(window, COLOR_PAIR(1));
wattroff(window, COLOR_PAIR(5));
}
wrefresh(window); /* Refresh the window to see the colored border and title */
}

38
file.c
View file

@ -9,7 +9,7 @@ typedef struct file {
char *path;
char *stats;
char *type;
/* put some more useful stat here */
int color;
struct file *next;
} file;
@ -65,20 +65,24 @@ void clear_marked()
}
}
long add_file(char *filepath, char *stats, char *type)
long add_file(char *filepath, char *stats, char *type, int color)
{
file *current = files;
file *new_file = memalloc(sizeof(file));
char *buf = strdup(filepath);
char *buf2 = strdup(stats);
char *buf3 = strdup(type);
if (buf == NULL || buf2 == NULL || buf3 == NULL) {
int buf4 = color;
if (buf == NULL || buf2 == NULL || buf3 == NULL)
perror("ccc");
}
new_file->path = buf;
new_file->stats = buf2;
new_file->type = buf3;
new_file->color = buf4;
new_file->next = NULL;
if (current == NULL) {
files = new_file;
return 0;
@ -121,6 +125,20 @@ char *get_filepath(long index)
}
}
int get_color(long index)
{
file *file = get_file(index);
if (file != NULL) {
int color = file->color;
if (!color) {
perror("ccc");
}
return color;
} else {
return 8; /* white */
}
}
/*
* Construct a formatted line for display
*/
@ -129,14 +147,16 @@ char *get_line(long index)
file *file = get_file(index);
if (file != NULL) {
char *name = strdup(file->path);
name = basename(name);
char *stats = strdup(file->stats);
if (name == NULL || stats == NULL) {
perror("ccc");
}
size_t length = strlen(name) + strlen(stats) + 2; /* one for space and one for nul */
size_t length = strlen(name) + strlen(stats) + 2; /* one for space and one for null */
char *line = memalloc(length * sizeof(char));
name = basename(name);
if (name == NULL || stats == NULL)
perror("ccc");
snprintf(line, length, "%s %s", stats, name);
return line;
} else {
return NULL;

5
file.h
View file

@ -5,7 +5,7 @@ typedef struct file {
char *path;
char *stats;
char *type;
// put some more useful stat here
int color;
struct file *next;
} file;
@ -13,9 +13,10 @@ long files_len();
long marked_len();
void clear_files();
void clear_marked();
long add_file(char *filename, char *time, char *type);
long add_file(char *filepath, char *stats, char *type, int color);
file *get_file(long index);
char *get_filepath(long index);
int get_color(long index);
char *get_line(long index);
#endif