Refactor to make the library reusable

Signed-off-by: Hadi <[email protected]>

utils/print.go

@@ -0,0 +1,173 @@
+package utils
+
+import (
+	"fmt"
+	"io"
+	"net/http"
+	"os"
+	"reflect"
+	"sort"
+	"strings"
+
+	github_recon_settings "github.com/anotherhadi/github-recon/settings"
+	"github.com/charmbracelet/lipgloss"
+	gopixels "github.com/saran13raj/go-pixels"
+)
+
+var (
+	grey  = lipgloss.Color("#7d7d7d")
+	green = lipgloss.Color("#a6e3a1")
+	blue  = lipgloss.Color("#7287fd")
+
+	greyStyle  = lipgloss.NewStyle().Foreground(grey)
+	greenStyle = lipgloss.NewStyle().Foreground(green)
+	titleStyle = lipgloss.NewStyle().Bold(true).Foreground(blue)
+)
+
+func PrintStruct(settings github_recon_settings.Settings, s any, indent int) {
+	if settings.Silent {
+		return
+	}
+
+	prefix := strings.Repeat("  ", indent)
+
+	v := reflect.ValueOf(s)
+	if !v.IsValid() {
+		return
+	}
+	t := reflect.TypeOf(s)
+
+	for v.Kind() == reflect.Ptr || v.Kind() == reflect.Interface {
+		if v.IsNil() {
+			return
+		}
+		v = v.Elem()
+		t = v.Type()
+	}
+
+	switch v.Kind() {
+	case reflect.Struct:
+		if v.NumField() == 0 {
+			fmt.Println(prefix + greyStyle.Render("No data found"))
+			fmt.Println("")
+			return
+		}
+
+		printed := 0
+		for i := 0; i < v.NumField(); i++ {
+			field := t.Field(i).Name
+			value := v.Field(i)
+
+			if !value.IsValid() || (value.Kind() == reflect.String && value.String() == "") {
+				continue
+			}
+			if value.Kind() == reflect.String && value.String() == "0001-01-01 00:00:00 +0000 UTC" {
+				continue
+			}
+			if (field == "FirstFoundIn" || field == "FoundIn") && !settings.ShowSource {
+				continue
+			}
+			printed++
+
+			switch value.Kind() {
+			case reflect.Struct, reflect.Slice, reflect.Array, reflect.Ptr, reflect.Map, reflect.Interface:
+				fmt.Println(prefix + greyStyle.Render(field+":"))
+				PrintStruct(settings, value.Interface(), indent+1)
+
+			case reflect.String:
+				fmt.Printf("%s%s %s\n", prefix, greyStyle.Render(field+":"), greenStyle.Render(fmt.Sprintf("%q", value.Interface())))
+
+			default:
+				fmt.Printf("%s%s %s\n", prefix, greyStyle.Render(field+":"), greenStyle.Render(fmt.Sprintf("%v", value.Interface())))
+			}
+		}
+		if printed == 0 {
+			fmt.Println(prefix + greyStyle.Render("No data found"))
+		}
+		fmt.Println("")
+
+	case reflect.Slice, reflect.Array:
+		if v.Len() == 0 {
+			fmt.Println(prefix + greyStyle.Render("No data found"))
+			fmt.Println("")
+			return
+		}
+		for i := 0; i < v.Len(); i++ {
+			PrintStruct(settings, v.Index(i).Interface(), indent)
+		}
+
+	case reflect.Map:
+		if v.Len() == 0 {
+			fmt.Println(prefix + greyStyle.Render("No data found"))
+			return
+		}
+
+		keys := v.MapKeys()
+		keyStrs := make([]string, len(keys))
+		for i, k := range keys {
+			keyStrs[i] = fmt.Sprintf("%v", k.Interface())
+		}
+		sort.Strings(keyStrs)
+
+		for _, keyStr := range keyStrs {
+			for _, k := range keys {
+				if fmt.Sprintf("%v", k.Interface()) == keyStr {
+					val := v.MapIndex(k)
+					fmt.Println(prefix + greyStyle.Render(fmt.Sprintf("%v:", k.Interface())))
+					PrintStruct(settings, val.Interface(), indent+1)
+				}
+			}
+		}
+
+	default:
+		fmt.Println(prefix + greenStyle.Render(fmt.Sprintf("%v", v.Interface())))
+	}
+}
+
+func Header() {
+	asciiArt := "        __                       \n  ___ _/ /  _______ _______  ___ \n / _ `/ _ \\/ __/ -_) __/ _ \\/ _ \\\n \\_, /_//_/_/  \\__/\\__/\\___/_//_/\n/___/                            "
+
+	grey := lipgloss.Color("#7d7d7d")
+
+	greyStyle := lipgloss.NewStyle().Foreground(grey)
+	fmt.Println(
+		greyStyle.Render(lipgloss.JoinVertical(lipgloss.Right, asciiArt, "@anotherhadi\n")),
+	)
+}
+
+func PrintTitle(silent bool, title string) {
+	if silent {
+		return
+	}
+	fmt.Println(titleStyle.Render(title) + "\n")
+}
+
+func PrintAvatar(settings github_recon_settings.Settings, url string) {
+	if settings.HideAvatar || url == "" || settings.Silent {
+		return
+	}
+
+	resp, err := http.Get(url)
+	if err != nil {
+		return
+	}
+	defer resp.Body.Close()
+
+	tmpfile, err := os.CreateTemp("", "avatar-*.png")
+	if err != nil {
+		return
+	}
+	defer os.Remove(tmpfile.Name())
+
+	_, err = io.Copy(tmpfile, resp.Body)
+	if err != nil {
+		return
+	}
+
+	output, err := gopixels.FromImagePath(tmpfile.Name(), 30, 25, "halfcell", true)
+
+	if err != nil {
+		return
+	}
+	fmt.Println(output + "\n")
+}