use stdin/stdout

for pipe in/out data

Author: typomedia

.gitignore

@@ -1,3 +1,6 @@
 .idea/
-bin/
-go.sum
+.vscode/
+dist/
+*.ico
+*.png
+*.syso

Makefile

@@ -1,13 +1,22 @@
-build:
-	go mod tidy
+build: tidy
 	go build -ldflags "-s -w" -o bin/ .
 
+run: tidy
+	go run main.go
+
+compile: tidy icon
+	GOOS=linux GOARCH=amd64 go build -ldflags "-s -w" -o dist/iconize-linux-amd64 .
+	GOOS=windows GOARCH=amd64 go build -ldflags "-s -w" -o dist/iconize-windows-amd64.exe .
+	GOOS=darwin GOARCH=amd64 go build -ldflags "-s -w" -o dist/iconize-macos-amd64 .
+	GOOS=freebsd GOARCH=amd64 go build -ldflags "-s -w" -o dist/iconize-freebsd-amd64 .
+
+tidy:
+	go mod tidy
+
 icon:
 	go install github.com/akavel/rsrc@latest
 	rsrc -ico go.ico
 
-run:
-	go mod tidy
-	go run .
-
-windows: icon build
+loc:
+	go install github.com/boyter/scc/v3@latest
+	scc --exclude-dir vendor --exclude-dir bin .

go.mod

@@ -5,5 +5,5 @@ go 1.19
 require (
 	github.com/spf13/pflag v1.0.5
 	github.com/typomedia/ico v1.0.0-beta.4
-	golang.org/x/image v0.0.0-20211028202545-6944b10bf410
+	golang.org/x/image v0.1.0
 )

main.go

@@ -3,42 +3,49 @@ package main
 import (
 	"bytes"
 	"fmt"
+	"github.com/spf13/pflag"
+	"github.com/typomedia/ico"
+	"golang.org/x/image/draw"
 	"image"
 	"image/color"
+	"io"
 	"log"
 	"os"
 	"path/filepath"
-	"time"
-
-	"github.com/spf13/pflag"
-	"github.com/typomedia/ico"
-	"golang.org/x/image/draw"
 )
 
 func main() {
 	var (
-		input  string
-		output string
+		input string
+		data  []byte
+		err   error
 	)
 
-	pflag.StringVarP(&output, "out", "o", "", "output ico file")
+	outfile := pflag.StringP("out", "o", "", "outfile ico file")
+	version := pflag.BoolP("version", "V", false, "show version")
 	pflag.Parse()
 
-	if len(pflag.Args()) < 1 {
-		os.Exit(1)
+	if *version {
+		fmt.Println("iconize 0.2.0 <[email protected]>")
+		os.Exit(0)
 	}
 
 	input = pflag.Arg(0)
-	fmt.Println("Input:", input)
-
-	start := time.Now()
-
-	file, err := os.ReadFile(input)
+	stat, _ := os.Stdin.Stat()
+
+	switch true {
+	case input != "":
+		data, err = os.ReadFile(input)
+	case (stat.Mode() & os.ModeCharDevice) == 0:
+		data, err = io.ReadAll(os.Stdin)
+	default:
+		log.Fatal("no input data given")
+	}
 	if err != nil {
 		log.Fatal(err)
 	}
 
-	img, _, err := image.Decode(bytes.NewReader(file))
+	img, _, err := image.Decode(bytes.NewReader(data))
 	if err != nil {
 		log.Fatal(err)
 	}
@@ -48,7 +55,7 @@ func main() {
 
 	alpha := image.NewRGBA(image.Rect(0, 0, size, size))
 
-	rgba := color.RGBA{0, 0, 0, 0}
+	rgba := color.RGBA{}
 	draw.Draw(alpha, alpha.Bounds(), &image.Uniform{C: rgba}, image.Point{}, draw.Src)
 
 	draw.Draw(alpha, img.Bounds().Add(
@@ -57,18 +64,6 @@ func main() {
 			Y: (size - img.Bounds().Dy()) / 2,
 		}), img, image.Point{}, draw.Over)
 
-	basename := name(input) + ".ico"
-	if output != "" {
-		basename = output
-	}
-	fmt.Println("Output:", basename)
-
-	icoFile, err := os.Create(basename)
-	if err != nil {
-		log.Fatal(err)
-	}
-	defer icoFile.Close()
-
 	icon := ico.NewIcon()
 	// https://learn.microsoft.com/en-us/windows/win32/uxguide/vis-icons
 	sizes := []int{256, 128, 64, 48, 32, 24, 16}
@@ -77,15 +72,24 @@ func main() {
 		icon.AddPng(resizedImg)
 	}
 
-	t := time.Now()
-	elapsed := t.Sub(start)
-	fmt.Println("Elapsed:", elapsed)
-
 	enc, err := icon.Encode()
 	if err != nil {
 		log.Fatal(err)
 	}
 
+	// write to stdout
+	if *outfile == "" {
+		fmt.Println(string(enc))
+		os.Exit(0)
+	}
+
+	// write to file
+	icoFile, err := os.Create(*outfile)
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer icoFile.Close()
+
 	icoFile.Write(enc)
 	os.Exit(0)