photoprism/internal/meta/exif_parser.go

124 lines
3.3 KiB
Go
Raw Normal View History

package meta
import (
"fmt"
"path/filepath"
"runtime/debug"
"strings"
"github.com/dsoprea/go-exif/v3"
heicexif "github.com/dsoprea/go-heic-exif-extractor/v2"
jpegstructure "github.com/dsoprea/go-jpeg-image-structure/v2"
pngstructure "github.com/dsoprea/go-png-image-structure/v2"
tiffstructure "github.com/dsoprea/go-tiff-image-structure/v2"
"github.com/photoprism/photoprism/pkg/fs"
"github.com/photoprism/photoprism/pkg/txt"
)
func RawExif(fileName string, fileType fs.FileFormat) (rawExif []byte, err error) {
defer func() {
if e := recover(); e != nil {
err = fmt.Errorf("metadata: %s in %s (raw exif panic)\nstack: %s", e, txt.Quote(filepath.Base(fileName)), debug.Stack())
}
}()
// Extract raw EXIF block.
var parsed bool
logName := txt.Quote(filepath.Base(fileName))
if fileType == fs.FormatJpeg {
jpegMp := jpegstructure.NewJpegMediaParser()
sl, err := jpegMp.ParseFile(fileName)
if err != nil {
log.Warnf("metadata: %s in %s (parse jpeg)", err, logName)
} else {
_, rawExif, err = sl.Exif()
if err != nil {
if strings.HasPrefix(err.Error(), "no exif header") {
return rawExif, fmt.Errorf("metadata: found no exif header in %s (parse jpeg)", logName)
} else if strings.HasPrefix(err.Error(), "no exif data") {
log.Debugf("metadata: failed parsing %s, starting brute-force search (parse jpeg)", logName)
} else {
log.Warnf("metadata: %s in %s, starting brute-force search (parse jpeg)", err, logName)
}
} else {
parsed = true
}
}
} else if fileType == fs.FormatPng {
pngMp := pngstructure.NewPngMediaParser()
cs, err := pngMp.ParseFile(fileName)
if err != nil {
return rawExif, fmt.Errorf("metadata: %s in %s (parse png)", err, logName)
}
_, rawExif, err = cs.Exif()
if err != nil {
if err.Error() == "file does not have EXIF" {
return rawExif, fmt.Errorf("metadata: found no exif header in %s (parse png)", logName)
} else {
log.Warnf("metadata: %s in %s (parse png)", err, logName)
}
} else {
parsed = true
}
} else if fileType == fs.FormatHEIF {
heicMp := heicexif.NewHeicExifMediaParser()
cs, err := heicMp.ParseFile(fileName)
if err != nil {
return rawExif, fmt.Errorf("metadata: %s in %s (parse heic)", err, logName)
}
_, rawExif, err = cs.Exif()
if err != nil {
if err.Error() == "file does not have EXIF" {
return rawExif, fmt.Errorf("metadata: found no exif header in %s (parse heic)", logName)
} else {
log.Warnf("metadata: %s in %s (parse heic)", err, logName)
}
} else {
parsed = true
}
} else if fileType == fs.FormatTiff {
tiffMp := tiffstructure.NewTiffMediaParser()
cs, err := tiffMp.ParseFile(fileName)
if err != nil {
return rawExif, fmt.Errorf("metadata: %s in %s (parse tiff)", err, logName)
}
_, rawExif, err = cs.Exif()
if err != nil {
if err.Error() == "file does not have EXIF" {
return rawExif, fmt.Errorf("metadata: found no exif header in %s (parse tiff)", logName)
} else {
log.Warnf("metadata: %s in %s (parse tiff)", err, logName)
}
} else {
parsed = true
}
}
if !parsed {
rawExif, err = exif.SearchFileAndExtractExif(fileName)
if err != nil {
return rawExif, fmt.Errorf("metadata: found no exif header in %s (search and extract)", logName)
}
}
return rawExif, nil
}