forked from koyachi/go-nude
/
util.go
38 lines (34 loc) · 791 Bytes
/
util.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
package nude
import (
"errors"
"image"
"image/gif"
"image/jpeg"
"image/png"
"os"
"strings"
)
// experimental
func DecodeImage(filePath string) (img image.Image, err error) {
return decodeImage(filePath)
}
func decodeImage(filePath string) (img image.Image, err error) {
reader, err := os.Open(filePath)
if err != nil {
return nil, err
}
defer reader.Close()
last3Strings := strings.ToLower(filePath[len(filePath)-3:])
last4Strings := strings.ToLower(filePath[len(filePath)-4:])
if last3Strings == "jpg" || last4Strings == "jpeg" {
img, err = jpeg.Decode(reader)
} else if last3Strings == "gif" {
img, err = gif.Decode(reader)
} else if last3Strings == "png" {
img, err = png.Decode(reader)
} else {
img = nil
err = errors.New("unknown format")
}
return
}