Add images to uwufetch

utils/args.go

@@ -8,6 +8,7 @@ import (
 var args []string
 var shoulduwuify bool = true
 var noascii bool = false
+var usepng bool = false
 
 func Initargs() {
 	args = os.Args[1:]
@@ -18,6 +19,8 @@ func Initargs() {
 				shoulduwuify = false
 			case "--noascii":
 				noascii = true
+			case "--usepng":
+				usepng = true
 			}
 		}
 	}

utils/asciiarts.go

@@ -15,7 +15,7 @@ func Getascii(name string) string {
 /_-''    ''-_\  `
 
 	switch name {
-	case "Arch Linux":
+	case "arch":
 		return arch
 
 	default:

utils/printing.go

@@ -1,9 +1,14 @@
 package utils
 
 import (
+	"encoding/base64"
 	"fmt"
 	"math/rand"
+	"os"
 	"strings"
+
+	"github.com/exhq/neowofetch/data"
+	"github.com/exhq/neowofetch/images"
 )
 
 func rgb(r, g, b int) string {
@@ -38,10 +43,20 @@ var isInProgressLine = false
 var logoLines []string
 var logoWidth int
 
-func CutePrintInit(logo string) {
+var pngWidth = 12
+var pngHeight = 12
+var pngData []byte
+
+func CutePrintInit() {
+	dist := data.GetDistroVariable("ID")
+	logo := Getascii(dist)
 	if noascii {
 		logo = ""
 	}
+	if usepng {
+		pngData = images.DistroImages[dist]
+		logo = strings.Repeat(" ", pngWidth) + strings.Repeat("\n", pngHeight)
+	}
 	logoLines = strings.Split(logo, "\n")
 	logoWidth = 0
 	for _, v := range logoLines {
@@ -57,13 +72,13 @@ func printLogoIfAtBeginningOfNewLine() {
 		isInProgressLine = true
 		if logoIndex < len(logoLines) {
 			logoLine := logoLines[logoIndex]
-			logoIndex += 1
 			logoLineLength := len([]rune(logoLine))
 			padding := strings.Repeat(" ", logoWidth-logoLineLength)
 			fmt.Printf("%s%s", logoLine, padding)
 		} else {
 			fmt.Printf("%s", strings.Repeat(" ", logoWidth))
 		}
+		logoIndex += 1
 	}
 }
 
@@ -102,15 +117,11 @@ func parseFormat(format string) (parsedFormat Format) {
 			switch v {
 			case "italic":
 				parsedFormat.colorFormat += "\x1b[3m"
-				break
 			case "bold":
 				parsedFormat.colorFormat += "\x1b1"
-				break
 			case "nouwu":
 				parsedFormat.noUwuOverride = true
-				break
 			case "*":
-				break
 			default:
 				//println("Unknown format code: ", v)
 			}
@@ -145,4 +156,12 @@ func CutePrintEnd() {
 	for logoIndex < len(logoLines) {
 		CuteNewLine()
 	}
+	if usepng {
+		fmt.Printf("\x1b[%dA", logoIndex)
+		fmt.Printf("\x1b]1337;File=inline=1;width=%d;height=%d:", pngWidth, pngHeight)
+		enc := base64.NewEncoder(base64.StdEncoding, os.Stdout)
+		enc.Write(images.DistroImages["arch"])
+		enc.Close()
+		fmt.Println("\a")
+	}
 }