forked from kardianos/photoview
/
editImage.go
80 lines (72 loc) · 1.5 KB
/
editImage.go
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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
package main
import (
"fmt"
"os"
"path/filepath"
"strings"
"github.com/disintegration/imaging"
)
type Rot byte
const (
RotLeft Rot = iota
RotRight
RotFlip
)
func (p *program) editImage(rot Rot, folder string, list []string) error {
for _, item := range list {
filename := filepath.Join(config.FileRoot, folder, item)
fi, err := os.Stat(filename)
if err != nil {
return err
}
fullImage, err := imaging.Open(filename)
if err != nil {
return err
}
switch rot {
case RotLeft:
fullImage = imaging.Rotate90(fullImage)
case RotRight:
fullImage = imaging.Rotate270(fullImage)
case RotFlip:
fullImage = imaging.Rotate180(fullImage)
default:
return fmt.Errorf("Unknown rot value: %d", rot)
}
err = imaging.Save(fullImage, filename)
if err != nil {
return err
}
tm := fi.ModTime()
err = os.Chtimes(filename, tm, tm)
if err != nil {
return err
}
}
return p.refreshCache(folder, list)
}
func (p *program) refreshCache(folder string, list []string) error {
cacheFolder := filepath.Join(p.cacheDir, folder)
cf, err := os.Open(cacheFolder)
if err != nil {
return err
}
defer cf.Close()
// Remove cache items.
fileList, err := cf.Readdirnames(-1)
if err != nil {
return err
}
for _, file := range fileList {
for _, item := range list {
if strings.HasPrefix(file, item) {
err = os.Remove(filepath.Join(cacheFolder, file))
if err != nil {
logger.Error("Failed to remove cached file ", file, ": ", err)
}
break
}
}
}
return nil
}