209 lines
6.3 KiB
Go
209 lines
6.3 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"image/jpeg"
|
|
"io/fs"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/nfnt/resize"
|
|
)
|
|
|
|
type config struct {
|
|
Recursive bool
|
|
Diff uint
|
|
Quality uint
|
|
Replace bool
|
|
StartingPath string
|
|
Quiet bool
|
|
AtLeast uint
|
|
MaxWidth uint
|
|
MaxHeight uint
|
|
IgnoreSuffix bool
|
|
|
|
}
|
|
|
|
type stats struct {
|
|
FilesProcessed uint64
|
|
FilesReplaced uint64
|
|
StartingBytes uint64
|
|
EndingBytes uint64
|
|
}
|
|
|
|
func main() {
|
|
|
|
var c config
|
|
flag.BoolVar(&c.Recursive, "recursive", false, "Process folders recursively (default false).")
|
|
flag.UintVar(&c.Diff, "diff", 25, "Percent difference required to replace original image.")
|
|
flag.UintVar(&c.Quality, "quality", 85, "JPG compression quality level.")
|
|
flag.BoolVar(&c.Quiet, "quiet", false, "Less output - don't print per-file detail")
|
|
flag.UintVar(&c.AtLeast, "atleast", 0, "Ignore images that aren't at least this many kilobytes")
|
|
flag.BoolVar(&c.IgnoreSuffix, "ignoresuffix", false, "Ignore suffix on filenames and attempt to treat everything as a JPG")
|
|
flag.UintVar(&c.MaxWidth, "maxwidth", 0, "Maximum width, scale images bigger to fit within this width. 0 ignores. Won't replace unless it meets the diff threashold.")
|
|
flag.UintVar(&c.MaxHeight, "maxheight", 0, "Maximum height, scale images bigger to fit within this height. 0 ignores. Won't replace until it meets the diff threashold.")
|
|
flag.StringVar(&c.StartingPath, "startingpath", ".", "Start from this path instead of current working dir")
|
|
flag.BoolVar(&c.Replace, "replace", false, "Replace with compressed versions if criteria are met. Otheriwse, just test and report, don't replace any images. (default false)")
|
|
flag.Parse()
|
|
|
|
fail := false
|
|
if c.Diff > 99 {
|
|
fmt.Println("Fatal: diff out of range (-99)")
|
|
fail = true
|
|
}
|
|
if c.Quality > 99 {
|
|
fmt.Println("Fatal: compresslevel out of range (1-99)")
|
|
fail = true
|
|
}
|
|
s, err := os.Stat(c.StartingPath)
|
|
if err != nil || !s.IsDir() {
|
|
fmt.Println("Starting path is invalid")
|
|
fail = true
|
|
}
|
|
if fail {
|
|
flag.Usage()
|
|
return
|
|
}
|
|
|
|
// Generate list of files to examing
|
|
var filelist []string
|
|
if !c.Recursive {
|
|
d, err := os.ReadDir(c.StartingPath)
|
|
if err != nil {
|
|
fmt.Printf("Error reading path: %s", err.Error())
|
|
}
|
|
for _, f := range d {
|
|
if !f.IsDir() {
|
|
ext := strings.ToLower(filepath.Ext(f.Name()))
|
|
if c.IgnoreSuffix || ext == ".jpg" || ext == ".jpeg" {
|
|
filelist = append(filelist, f.Name())
|
|
}
|
|
|
|
}
|
|
}
|
|
} else {
|
|
filepath.WalkDir(c.StartingPath, func(path string, d fs.DirEntry, err error) error {
|
|
ext := strings.ToLower(filepath.Ext(path))
|
|
if !d.IsDir() && (c.IgnoreSuffix || ext == ".jpg" || ext == ".jpeg") {
|
|
filelist = append(filelist, path)
|
|
}
|
|
return nil
|
|
})
|
|
}
|
|
var st stats
|
|
for _, f := range filelist {
|
|
st = doSingleImage(f, c, st)
|
|
}
|
|
fmt.Println("\nEnding stats:")
|
|
fmt.Printf("Files processed: %d\nFiles replaced: %d\nStarting Size: %s (%d bytes)\nEnding Size: %s (%d bytes)\n",
|
|
st.FilesProcessed, st.FilesReplaced, SIFormat(st.StartingBytes), st.StartingBytes, SIFormat(st.EndingBytes), st.EndingBytes)
|
|
decrease:=st.StartingBytes - st.EndingBytes
|
|
diff := uint((float64(decrease) / float64(st.StartingBytes)) * 100)
|
|
|
|
fmt.Printf("Total savings %s (%d bytes) or %d%%\n", SIFormat(decrease), decrease, diff )
|
|
|
|
}
|
|
|
|
// doSingleImage processes a single file and updateds the run stats as it goes
|
|
func doSingleImage(f string, c config, s stats) stats {
|
|
st, err := os.Stat(f)
|
|
if err != nil {
|
|
fmt.Printf("Error reading file %s: %s\n", f, err.Error())
|
|
return s
|
|
}
|
|
if st.Size() < int64(c.AtLeast)*1024 {
|
|
s.StartingBytes += uint64(st.Size())
|
|
if !c.Quiet {
|
|
fmt.Printf("Ignoring %s as %d bytes are below threshold\n", f, st.Size())
|
|
}
|
|
return s
|
|
}
|
|
in, err := os.Open(f)
|
|
if err != nil {
|
|
fmt.Printf("Error opening file %s: %s\n", f, err.Error())
|
|
}
|
|
defer in.Close()
|
|
orig, err := jpeg.Decode(in)
|
|
if err != nil {
|
|
fmt.Printf("Couldn't decode source image %s: %s\n", f, err.Error())
|
|
return s
|
|
}
|
|
in.Close()
|
|
// Check bounds and resize if necessary
|
|
ores:=orig.Bounds()
|
|
if (c.MaxWidth !=0 && ores.Dx() > int(c.MaxWidth) ) || (c.MaxHeight!=0 && ores.Dy()>int(c.MaxHeight)) {
|
|
orig=resize.Thumbnail(c.MaxWidth, c.MaxHeight, orig, resize.Lanczos3)
|
|
fmt.Printf("Resized from %dx%d - ", ores.Dx(), ores.Dy())
|
|
}
|
|
out, err := os.OpenFile("imageoptimizer.tmp", os.O_WRONLY|os.O_CREATE|os.O_TRUNC, st.Mode())
|
|
if err != nil {
|
|
fmt.Printf("Couldn't open out temprary file: %s\n", err.Error())
|
|
return s
|
|
}
|
|
err = jpeg.Encode(out, orig, &jpeg.Options{Quality: int(c.Quality)})
|
|
if err != nil {
|
|
fmt.Printf("Error re-encoding image %s: %s\n", f, err.Error())
|
|
return s
|
|
}
|
|
out.Close()
|
|
stt, err := os.Stat("imageoptimizer.tmp")
|
|
if err != nil {
|
|
fmt.Printf("Error reading re-compressed temp file: %s\n", err.Error())
|
|
return s
|
|
}
|
|
s.StartingBytes += uint64(st.Size())
|
|
decrease:=st.Size() - stt.Size()
|
|
diff := ((float64(decrease) / float64(st.Size())) * 100)
|
|
if diff > float64(c.Diff) {
|
|
// Would replace
|
|
if c.Replace {
|
|
err = os.Rename(f, "imageoptimizer.replacetmp")
|
|
if err != nil {
|
|
fmt.Printf("Fatal error moving original file %s: %s\n", f, err.Error())
|
|
os.Exit(1)
|
|
}
|
|
err = os.Rename("imageoptimizer.tmp", f)
|
|
if err != nil {
|
|
fmt.Printf("Fatal error replacing original file %s: %s\nOriginal file has been renamed to imageoptimizer.replacetmp", f, err.Error())
|
|
os.Exit(1)
|
|
}
|
|
os.Remove("imageoptimizer.replacetmp")
|
|
}
|
|
|
|
|
|
s.EndingBytes += uint64(stt.Size())
|
|
s.FilesReplaced++
|
|
if (!c.Quiet) {
|
|
fmt.Printf("Replace %s: From %s (%d bytes) to %s (%d bytes) (%.0f%%)\n",
|
|
f, SIFormat(uint64(st.Size())), st.Size(), SIFormat(uint64(stt.Size())), stt.Size(), diff)
|
|
}
|
|
} else {
|
|
if (!c.Quiet) {
|
|
fmt.Printf("Not Replacing %s: From %s (%d bytes) to %s (%d bytes) (%.0f%%)\n",
|
|
f, SIFormat(uint64(st.Size())), st.Size(), SIFormat(uint64(stt.Size())), stt.Size(), diff)
|
|
}
|
|
s.EndingBytes += uint64(st.Size())
|
|
os.Remove("imageoptimizer.tmp")
|
|
|
|
}
|
|
|
|
s.FilesProcessed++
|
|
return s
|
|
}
|
|
|
|
// SIFormat prints sizes in a nice human readable way
|
|
func SIFormat(num_in uint64) string {
|
|
suffix := "B" //just assume bytes
|
|
num := float64(num_in)
|
|
units := []string{"", "K", "M", "G", "T", "P", "E", "Z"}
|
|
for _, unit := range units {
|
|
if num < 1000.0 {
|
|
return fmt.Sprintf("%3.1f%s%s", num, unit, suffix)
|
|
}
|
|
num = (num / 1000)
|
|
}
|
|
return fmt.Sprintf("%.1f%s%s", num, "Yi", suffix)
|
|
}
|