image support!!!

.gitignore

@@ -6,3 +6,4 @@ result/
 ./pkg
 ./goGoFetch
 /gogofetch-git-r14.ff1fcf9-1-x86_64.pkg.tar.zst
+./gogofetch-git

PKGBUILD

@@ -7,7 +7,7 @@
 pkgname=gogofetch-git
 pkgver=r14.ff1fcf9
 pkgrel=2
-pkgdesc="fetch made with C++ rewritten to go for no reason"
+pkgdesc="Fetch written in Go with image support"
 arch=('x86_64')
 url="git+https://codeberg.org/nekohepott/goGoFetch.git"
 license=('MIT')

README.md

@@ -1,15 +1,33 @@
-# Build
-to build:
+# Fetch written in Go with image support (only for foot and kitty)
+
+![Fetch image](assets/screenshot.png "fetch")
+
+# Install
+you can get gogofetch on AUR!
+```bash
+yay -S gogofetch-git
 ```
-go build main.go
+
+# Usage
+basic usage:
+```bash
+gogofetch
 ```
-to run:
+to show custom image (only png supported btw):
+```bash
+gogofetch -i *absolute_path*
 ```
-./main
+to show custom image and set custom width:
+```bash
+gogofetch -i *absolute_path* -w 250
+```
+to show help message:
+```bash
+gogofetch -h
 ```
 
 # NixOS
-credits to melvi for this
+credits to melvi for flake
 
 to build with nix:
 ```bash
@@ -19,7 +37,6 @@ to run with nix:
 ```
 nix run
 ```
-
 to install with nixos:
 
 1. Add to inputs:
@@ -40,5 +57,19 @@ home.packages = [
 ```
 3. Run rebuild and then you can run fetch with *gogofetch* command
 
-# Screenshots
-![Fetch image](assets/screenshot.png "fetch")
+
+# Build
+to build:
+```
+git clone https://codeberg.org/nekohepott/goGoFetch
+cd goGoFetch
+go build main.go
+``` 
+and run:
+```
+./main
+```
+
+
+
+

go.mod

@@ -1,3 +1,10 @@
 module goGoFetch
 
 go 1.26
+
+require github.com/mattn/go-sixel v0.0.8
+
+require (
+	github.com/nfnt/resize v0.0.0-20180221191011-83c6a9932646 // indirect
+	github.com/soniakeys/quant v1.0.0 // indirect
+)

go.sum

@@ -0,0 +1,8 @@
+github.com/BurntSushi/toml v1.6.0 h1:dRaEfpa2VI55EwlIW72hMRHdWouJeRF7TPYhI+AUQjk=
+github.com/BurntSushi/toml v1.6.0/go.mod h1:ukJfTF/6rtPPRCnwkur4qwRxa8vTRFBF0uk2lLoLwho=
+github.com/mattn/go-sixel v0.0.8 h1:H0bBGQVOJoSvzvtTgCInxvg1IZiNlTcIIIx8A6uvjpQ=
+github.com/mattn/go-sixel v0.0.8/go.mod h1:wbDSbrwpykVI1qEHyjZYsDgaJTwpVg9wSwmmh2slnBw=
+github.com/nfnt/resize v0.0.0-20180221191011-83c6a9932646 h1:zYyBkD/k9seD2A7fsi6Oo2LfFZAehjjQMERAvZLEDnQ=
+github.com/nfnt/resize v0.0.0-20180221191011-83c6a9932646/go.mod h1:jpp1/29i3P1S/RLdc7JQKbRpFeM1dOBd8T9ki5s+AY8=
+github.com/soniakeys/quant v1.0.0 h1:N1um9ktjbkZVcywBVAAYpZYSHxEfJGzshHCxx/DaI0Y=
+github.com/soniakeys/quant v1.0.0/go.mod h1:HI1k023QuVbD4H8i9YdfZP2munIHU4QpjsImz6Y6zds=

main.go

@@ -2,22 +2,37 @@ package main
 
 import (
 	"bufio"
+	"bytes"
+	"encoding/base64"
+	"flag"
 	"fmt"
+	"image"
+	"image/png"
+	_ "image/png"
 	"os"
 	"os/exec"
 	"regexp"
 	"strconv"
 	"strings"
+
+	"github.com/mattn/go-sixel"
+	"github.com/nfnt/resize"
 )
 
 const (
 	Reset = "\033[0m"
-	GreenBold  = "\033[1;32m"
-	BlueBright = "\033[94m"
+	Blue  = "\033[1;34m"
 )
 
 var dist string
 
+var logoMap = map[string]string{
+	"Arch Linux": "assets/arch.png",
+	"Debian":     "assets/debian.png",
+	"NixOS":      "assets/nixos.png",
+	"default":    "assets/linux.png",
+}
+
 func getDist() string {
 	f, _ := os.Open("/etc/os-release")
 	defer func(f *os.File) {
@@ -161,24 +176,51 @@ func getPkgs() string {
 	return "unknown"
 }
 
-func getAscii() string {
-	var ascii string
-	if strings.HasPrefix(dist, "Arch") {
-		content := "                    _       _      _\n     /\\            | |     | |    (_)\n    /  \\   _ __ ___| |__   | |     _ _ __  _   ___  __\n   / /\\ \\ | '__/ __| '_ \\  | |    | | '_ \\| | | \\ \\/ /\n  / ____ \\| | | (__| | | | | |____| | | | | |_| |>  <\n /_/    \\_\\_|  \\___|_| |_| |______|_|_| |_|\\__,_/_/\\_\\\n"
-		ascii = fmt.Sprintf(content)
-		return ascii
+//func getAscii() string {
+//	var ascii string
+//	if strings.HasPrefix(dist, "Arch") {
+//		content := "                    _       _      _\n     /\\            | |     | |    (_)\n    /  \\   _ __ ___| |__   | |     _ _ __  _   ___  __\n   / /\\ \\ | '__/ __| '_ \\  | |    | | '_ \\| | | \\ \\/ /\n  / ____ \\| | | (__| | | | | |____| | | | | |_| |>  <\n /_/    \\_\\_|  \\___|_| |_| |______|_|_| |_|\\__,_/_/\\_\\\n"
+//		ascii = fmt.Sprintf(content)
+//		return ascii
+//	}
+//	if strings.HasPrefix(dist, "Debian") {
+//		content := "  _____           _       _\n |  __ \\         | |     (_)\n | |  | |   ___  | |__    _    __ _   _ __\n | |  | |  / _ \\ | '_ \\  | |  / _` | | '_ \\\n | |__| | |  __/ | |_) | | | | (_| | | | | |\n |_____/   \\___| |_.__/  |_|  \\__,_| |_| |_|\n"
+//		ascii = fmt.Sprintf(content)
+//		return ascii
+//	}
+//	if strings.HasPrefix(dist, "NixOS") {
+//		content := " _   _ _       ____   _____\n | \\ | (_)     / __ \\ / ____|\n |  \\| |___  _| |  | | (___\n | . ` | \\ \\/ / |  | |\\___ \\\n | |\\  | |>  <| |__| |____) |\n |_| \\_|_/_/\\_\\\\____/|_____/\n"
+//		ascii = fmt.Sprintf(content)
+//		return ascii
+//	}
+//	return "unknown"
+//}
+
+func printLogo(path string, Width uint) {
+	file, _ := os.Open(path)
+	defer func(file *os.File) {
+		err := file.Close()
+		if err != nil {
+			return
 		}
-	if strings.HasPrefix(dist, "Debian") {
-		content := "  _____           _       _\n |  __ \\         | |     (_)\n | |  | |   ___  | |__    _    __ _   _ __\n | |  | |  / _ \\ | '_ \\  | |  / _` | | '_ \\\n | |__| | |  __/ | |_) | | | | (_| | | | | |\n |_____/   \\___| |_.__/  |_|  \\__,_| |_| |_|\n"
-		ascii = fmt.Sprintf(content)
-		return ascii
+	}(file)
+	img, _, _ := image.Decode(file)
+	resizedImg := resize.Resize(Width, 0, img, resize.Lanczos3)
+	if os.Getenv("KITTY_PID") != "" || os.Getenv("TERM") == "xterm-kitty" {
+		var buf bytes.Buffer
+		err := png.Encode(&buf, resizedImg)
+		if err != nil {
+			return
+		}
+		enc := base64.StdEncoding.EncodeToString(buf.Bytes())
+		fmt.Printf("\033_Gf=100,a=T;%s\033\\\n", enc)
+	} else {
+		err := sixel.NewEncoder(os.Stdout).Encode(resizedImg)
+		fmt.Printf("\n")
+		if err != nil {
+			return
 		}
-	if strings.HasPrefix(dist, "NixOS") {
-		content := " _   _ _       ____   _____\n | \\ | (_)     / __ \\ / ____|\n |  \\| |___  _| |  | | (___\n | . ` | \\ \\/ / |  | |\\___ \\\n | |\\  | |>  <| |__| |____) |\n |_| \\_|_/_/\\_\\\\____/|_____/\n"
-		ascii = fmt.Sprintf(content)
-		return ascii
 	}
-	return "unknown"
 }
 
 func main() {
@@ -186,9 +228,14 @@ func main() {
 		Label string
 		Value string
 	}
+	dist := getDist()
+	defaultLogo, ok := logoMap[dist]
+	if !ok {
+		defaultLogo = logoMap["default"]
+	}
 
 	stats := []Info{
-		{"dist", getDist()},
+		{"dist", dist},
 		{"host", getHostname()},
 		{"cpu", getCpu()},
 		{"krnl", getKernel()},
@@ -196,14 +243,19 @@ func main() {
 		{"gpu", getGpu()},
 		{"pkgs", getPkgs()},
 	}
-	fmt.Print(BlueBright)
-	fmt.Printf(getAscii())
-	fmt.Print(Reset)
-	fmt.Printf("\n")
-	fmt.Println("---------------------------")
+
+	logoPath := flag.String("i", defaultLogo, "Absolute or relative path to the logo image (only png)")
+	width := flag.Int("w", 200, "Logo width in lines (400 IS MAXIMUM FOR KITTY VALUES ABOVE WILL CRASH TERMINAL)")
+	flag.Parse()
+	if _, err := os.Stat(*logoPath); os.IsNotExist(err) {
+		fmt.Printf("Image '%s' not found\n", *logoPath)
+		return
+	}
+
+	printLogo(*logoPath, uint(*width))
 
 	for _, s := range stats {
-		fmt.Printf("> %s%s%s: %s\n", GreenBold, s.Label, Reset, s.Value)
+		fmt.Printf("~ %s%s%s: %s\n", Blue, s.Label, Reset, s.Value)
 	}
 
 }