Initial commit

main.go

@@ -0,0 +1,116 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"image-transcoder/codec"
+	"io"
+	"log"
+	"os"
+	"path/filepath"
+	"strings"
+	"sync"
+)
+
+var codecs = map[string]codec.Coder{
+	".jpg":  codec.JPGCoder{},
+	".jpeg":  codec.JPGCoder{},
+	".png":  codec.PNGCoder{},
+	".webp": codec.WebpCoder{},
+}
+
+var Error *log.Logger
+var Warning *log.Logger
+
+func init() {
+	log.SetFlags(log.Lmicroseconds | log.Lshortfile)
+	logger := io.MultiWriter(os.Stdout)
+	log.SetOutput(logger)
+
+	Error = log.New(io.MultiWriter(os.Stderr, os.Stdout),
+		fmt.Sprintf("%sERROR:%s ", "\033[0;101m", "\033[0m"),
+		log.Lmicroseconds|log.Lshortfile)
+	Warning = log.New(io.MultiWriter(os.Stdout),
+		fmt.Sprintf("%sWarning:%s ", "\033[0;93m", "\033[0m"),
+		log.Lmicroseconds|log.Lshortfile)
+}
+
+func main() {
+	quality := flag.Int("quality", 80, "Quality of the image")
+	to := flag.String("to", ".jpg", "Extension to transcode to")
+	nosafe := flag.Bool("nosafe", false, "Prevents backup of the original before encoding")
+	flag.Parse()
+
+	if _, ok := codecs[*to]; !ok {
+		Error.Println("Invalid extension")
+		return
+	}
+
+	log.Printf("Using quality: %d", *quality)
+	log.Printf("Transcoding to: %s", *to)
+	log.Printf("Nosafe mode: %v", *nosafe)
+
+	var wg sync.WaitGroup
+	for _, file := range flag.Args() {
+		wg.Add(1)
+		go func(file string) {
+			defer wg.Done()
+			file = filepath.ToSlash(file)
+			file = filepath.Clean(file)
+			ext := filepath.Ext(file)
+			coder, ok := codecs[ext]
+			if !ok {
+				Error.Printf("No encoder found for file: %s", file)
+				return
+			}
+			log.Printf("%#v", coder)
+
+			filename := filepath.Base(file)
+			if *nosafe {
+				bckp := filename + ".bak"
+				err := os.Link(file, bckp)
+				if err != nil {
+					Error.Printf("Failed to backup file: %v", err)
+					return
+				}
+			}
+
+			filehandle, err := os.Open(file)
+			if err != nil {
+				Error.Printf("Failed to open file: %v", err)
+				return
+			}
+
+			log.Printf("Decoding %s", file)
+			image, err := coder.Decode(filehandle)
+			if err != nil {
+				Error.Printf("Failed to decode file: %v", err)
+				return
+			}
+			filehandle.Close()
+
+			newfile := strings.Replace(filename, ext, *to, 1)
+			log.Printf("Encoding %s to %s", file, newfile)
+
+			buf, err := coder.Encode(image, *quality)
+			if err != nil {
+				Error.Printf("Failed to encode image: %v", err)
+				return
+			}
+
+			filehandle, err = os.Create(newfile)
+			if err != nil {
+				Error.Printf("Failed to create file %s: %v", newfile, err)
+				return
+			}
+
+			n, err := filehandle.Write(buf.Bytes())
+			if n != len(buf.Bytes()) || err != nil {
+				Error.Printf("Failed to write file %s: %v", newfile, err)
+				return
+			}
+			log.Printf("Wrote %d bytes to %s", n, newfile)
+		}(file)
+	}
+	wg.Wait()
+}