aboutsummaryrefslogtreecommitdiff
path: root/cmd/thumber/main.go
blob: bd6713067a44da273575824569eeda6222356169 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
// Command thumber creates 256x256 thumbnails for a folder/pile of them.
package main

import (
	"flag"
	"log"
	"os"
	"path/filepath"
	"strings"

	"github.com/disintegration/imaging"
	"within.website/x/internal"
)

var (
	dirToWalk = flag.String("walkdir", "./in", "directory to walk and generate thumbnails for")
)

func main() {
	internal.HandleStartup()

	err := filepath.Walk(*dirToWalk, makeThumbnail)
	if err != nil {
		log.Fatal(err)
	}
}

func makeThumbnail(fname string, info os.FileInfo, err error) error {
	if info.IsDir() {
		return nil
	}

	if strings.HasSuffix(fname, ".thumb.png") {
		return nil
	}

	if strings.HasSuffix(fname, ".html") {
		return nil
	}

	_, err = os.Stat("thumbs/" + filepath.Base(fname) + ".thumb.png")
	if err == nil {
		log.Printf("skipping %s", fname)
		return nil
	}

	log.Printf("Starting to open %s", fname)

	img, err := imaging.Open(fname)
	if err != nil {
		return err
	}

	croppedImage := imaging.Thumbnail(img, 256, 256, imaging.Lanczos)
	err = imaging.Save(croppedImage, "thumbs/"+filepath.Base(fname)+".thumb.png")

	return err
}