✨ Added icons implementation

module/colors.go

@@ -0,0 +1,19 @@
+// Package module - module/colors.go
+package module
+
+const (
+	// Bold format
+	Bold string = "\033[1m"
+	// Reset format color code
+	Reset string = "\033[0m"
+	// LightPurple color
+	LightPurple string = "\033[1;35m"
+	// LightBlue color
+	LightBlue string = "\033[1;36m"
+	// Grey color
+	Grey string = "\033[90m"
+	// Pink color
+	Pink string = "\033[38;5;206m"
+	// Orange color
+	Orange string = "\033[38;5;208m"
+)

module/extension.go

@@ -0,0 +1,19 @@
+// Package module - module/extension .go
+package module
+
+const (
+	// Go extension
+	Go string = ".go"
+	// Mod extension
+	Mod string = ".mod"
+	// Sum extension
+	Sum string = ".sum"
+	// Md Markdown extension
+	Md string = ".md"
+	// Gitignore file
+	Gitignore string = ".gitignore"
+	// GitFolder folder
+	GitFolder string = ".git"
+	// Github folder
+	Github string = ".github"
+)

module/icons.go

@@ -0,0 +1,15 @@
+// Package module - module/icons.go
+package module
+
+const (
+	// File icon
+	File string = "\uf15b"
+	// Folder icon
+	Folder string = "\ue5ff"
+	// GoLang icon
+	GoLang string = "\ue65e"
+	// Markdown icon
+	Markdown string = "\uf48a"
+	// Git icon
+	Git string = "\uf1d3"
+)

service/service.go

@@ -20,6 +20,7 @@ const (
 
 // Dispatcher func - executes function based on flags
 func Dispatcher(options module.Options) {
+	fmt.Println(".")
 	if options.Flags.ShowTreeView {
 		treeDirectory(options, "", true)
 	} else {

service/util.go

@@ -6,6 +6,7 @@ import (
 	"github.com/oussamaM1/treels/module"
 	"log"
 	"os"
+	"strings"
 )
 
 // CheckDefaultDirectory func - returns current working directory if no directory is specified
@@ -51,10 +52,54 @@ func isHidden(name string) bool {
 // printWithIconAndPrefix func - prints files and folder with icons and prefix
 func printWithIconAndPrefix(prefix string, file os.FileInfo) string {
 	var format string
+	// Directory icon logic
 	if file.IsDir() {
-		format = fmt.Sprintf("%s%s%s%s %s%s", prefix, module.LightPurple, module.Bold, module.Folder, file.Name(), module.Reset)
-	} else {
+		switch {
+		case strings.HasSuffix(file.Name(), module.Github), strings.HasSuffix(file.Name(), module.GitFolder):
+			format = printIconFolders(prefix, file, module.GitFolder)
+		default:
+			format = printIconFolders(prefix, file, module.Folder)
+		}
+		return format
+	}
+
+	// File icon logic
+	switch {
+	case strings.HasSuffix(file.Name(), module.Go), strings.HasSuffix(file.Name(), module.Mod), strings.HasSuffix(file.Name(), module.Sum):
+		format = printIconFiles(prefix, file, module.Go)
+	case strings.HasSuffix(file.Name(), module.Md):
+		format = printIconFiles(prefix, file, module.Md)
+	case strings.HasSuffix(file.Name(), module.Gitignore):
+		format = printIconFiles(prefix, file, module.Gitignore)
+	default:
+		// Default file icon
+		format = printIconFiles(prefix, file, module.File)
+	}
+	return format
+}
+
+func printIconFiles(prefix string, file os.FileInfo, extension string) string {
+	var format string
+	switch extension {
+	case module.Go, module.Mod, module.Sum:
+		format = fmt.Sprintf("%s%s%s %s%s", prefix, module.LightBlue, module.GoLang, module.Reset, file.Name())
+	case module.Md:
+		format = fmt.Sprintf("%s%s%s %s%s", prefix, module.Grey, module.Markdown, module.Reset, file.Name())
+	case module.Gitignore:
+		format = fmt.Sprintf("%s%s%s %s%s", prefix, module.Orange, module.Git, module.Reset, file.Name())
+	default:
 		format = fmt.Sprintf("%s%s %s", prefix, module.File, file.Name())
 	}
 	return format
 }
+
+func printIconFolders(prefix string, file os.FileInfo, extension string) string {
+	var format string
+	switch extension {
+	case module.Github, module.GitFolder:
+		format = fmt.Sprintf("%s%s%s%s %s%s", prefix, module.Bold, module.Orange, module.Git, file.Name(), module.Reset)
+	default:
+		format = fmt.Sprintf("%s%s%s%s %s%s", prefix, module.Bold, module.Pink, module.Folder, file.Name(), module.Reset)
+	}
+	return format
+}