add colors to ls

1 files changed, 39 insertions, 4 deletions

diff --git a/core/ls.c b/core/ls.c
index cd96ce7..c307ece 100644
--- a/core/ls.c
+++ b/core/ls.c
@@ -5,8 +5,11 @@
 #include "../lib/arg/arg.h"
 #include "../lib/tctl/tctl.h"
 #include "../lib/cstr/cstr.h"
+#include "../lib/sys/errno.h"
+#include "../lib/aec/aec.h"
+#include "../lib/sys/stat.h"
 
-#define BUF_SIZE sizeof(struct dirent) * 32
+#define BUF_SIZE sizeof(dirent_t) * 32
 
 enum {
 	FILTER_NONE   = 0,
@@ -28,7 +31,7 @@ void add_dir(const char *path)
 }
 
 
-u8 filter_entry(struct dirent *d)
+u8 filter_entry(dirent_t *d)
 {
 	switch (filter) {
 		case FILTER_HIDDEN:
@@ -39,6 +42,29 @@ u8 filter_entry(struct dirent *d)
 }
 
 
+void set_color(int fd, const char *name)
+{
+
+	stat_t st;
+	const char *color;
+
+	fstatat(fd, name, &st, 0);
+
+	switch (st.mode & S_IFMT) {
+		case S_IFREG:  color = COLOR_WHITE; break;
+		case S_IFDIR:  color = COLOR_BLUE;  break;
+		case S_IFLNK:  color = COLOR_CYAN;  break;
+		case S_IFSOCK: color = COLOR_RED;   break;
+		case S_IFBLK:  color = COLOR_YELLOW;   break;
+		case S_IFCHR:  color = COLOR_PINK;   break;
+		case S_IFIFO:  color = COLOR_GREEN;   break;
+		default: color = COLOR_RED; break;
+	}
+
+	foreground(color);
+}
+
+
 int main(int argc, const char *argv[])
 {
 	int                  fd;
@@ -49,7 +75,7 @@ int main(int argc, const char *argv[])
 	i64 maxwidth = istty ? tctl_get_window_size().width : -1;
 	i64 current_width = 0;
 
-	struct dirent *d;
+	dirent_t *d;
 	dirs = new_list();
 
 	arg_register_flag("-a", &set_filter_none);
@@ -63,6 +89,11 @@ int main(int argc, const char *argv[])
 	for (list_node_t *node = dirs->first; node; node = node->next) {
 		fd = open(node->value, OPEN_READ_ONLY | OPEN_DIRECTORY, 0);
 
+		if (fd < 0) {
+			wff(STDERR_FD, "error: %s", errstr[-fd]);
+			continue;
+		}
+
 		while (1) {
 			nread = getdents(fd, buf, BUF_SIZE);
 
@@ -70,7 +101,7 @@ int main(int argc, const char *argv[])
 				break;
 
 			for (u64 bpos = 0; bpos < nread;) {
-				d = (struct dirent*) (buf + bpos);
+				d = (dirent_t*) (buf + bpos);
 				if (filter_entry(d)) {
 					if (!istty) {
 						wstdf("%s\n", d->d_name);
@@ -78,11 +109,15 @@ int main(int argc, const char *argv[])
 						current_width += cstr_length(d->d_name) + 2;
 						if (current_width > maxwidth) {
 							wstd("\n");
+							set_color(fd, d->d_name);
 							wstd(d->d_name);
+							wstdf("%S", SGR_RESET)
 							wstd("  ");
 							current_width = cstr_length(d->d_name) + 2;
 						} else {
+							set_color(fd, d->d_name);
 							wstdf("%s  ", d->d_name);
+							wstdf("%S", SGR_RESET)
 						}
 					}
 				}