131 lines
3.5 KiB
Go
131 lines
3.5 KiB
Go
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/clean"
|
|
"github.com/photoprism/photoprism/pkg/fs"
|
|
)
|
|
|
|
func RawExif(fileName string, fileFormat fs.Type, bruteForce bool) (rawExif []byte, err error) {
|
|
defer func() {
|
|
if e := recover(); e != nil {
|
|
err = fmt.Errorf("%s in %s (raw exif panic)\nstack: %s", e, clean.Log(filepath.Base(fileName)), debug.Stack())
|
|
}
|
|
}()
|
|
|
|
// Extract raw Exif block.
|
|
var parsed bool
|
|
|
|
// Sanitized and shortened file name for logs.
|
|
logName := clean.Log(filepath.Base(fileName))
|
|
|
|
// Try Exif parser for specific media file format first.
|
|
switch fileFormat {
|
|
case fs.ImageJPEG:
|
|
jpegMp := jpegstructure.NewJpegMediaParser()
|
|
|
|
sl, err := jpegMp.ParseFile(fileName)
|
|
|
|
if err != nil {
|
|
log.Infof("metadata: %s while parsing jpeg file %s", err, logName)
|
|
} else {
|
|
_, rawExif, err = sl.Exif()
|
|
|
|
if err != nil {
|
|
if !bruteForce || strings.HasPrefix(err.Error(), "no exif header") {
|
|
return rawExif, fmt.Errorf("found no exif header")
|
|
} else if strings.HasPrefix(err.Error(), "no exif data") {
|
|
log.Debugf("metadata: failed parsing %s, starting brute-force search (parse jpeg)", logName)
|
|
} else {
|
|
log.Infof("metadata: %s in %s, starting brute-force search (parse jpeg)", err, logName)
|
|
}
|
|
} else {
|
|
parsed = true
|
|
}
|
|
}
|
|
case fs.ImagePNG:
|
|
pngMp := pngstructure.NewPngMediaParser()
|
|
|
|
cs, err := pngMp.ParseFile(fileName)
|
|
|
|
if err != nil {
|
|
return rawExif, fmt.Errorf("%s while parsing png file", err)
|
|
} else {
|
|
_, rawExif, err = cs.Exif()
|
|
|
|
if err != nil {
|
|
if err.Error() == "file does not have EXIF" || strings.HasPrefix(err.Error(), "no exif data") {
|
|
return rawExif, fmt.Errorf("found no exif header")
|
|
} else {
|
|
log.Infof("metadata: %s in %s (parse png)", err, logName)
|
|
}
|
|
} else {
|
|
parsed = true
|
|
}
|
|
}
|
|
case fs.ImageHEIC:
|
|
heicMp := heicexif.NewHeicExifMediaParser()
|
|
|
|
cs, err := heicMp.ParseFile(fileName)
|
|
|
|
if err != nil {
|
|
return rawExif, fmt.Errorf("%s while parsing heic file", err)
|
|
} else {
|
|
_, rawExif, err = cs.Exif()
|
|
|
|
if err != nil {
|
|
if err.Error() == "file does not have EXIF" || strings.HasPrefix(err.Error(), "no exif data") {
|
|
return rawExif, fmt.Errorf("found no exif header")
|
|
} else {
|
|
log.Infof("metadata: %s in %s (parse heic)", err, logName)
|
|
}
|
|
} else {
|
|
parsed = true
|
|
}
|
|
}
|
|
case fs.ImageTIFF:
|
|
tiffMp := tiffstructure.NewTiffMediaParser()
|
|
|
|
cs, err := tiffMp.ParseFile(fileName)
|
|
|
|
if err != nil {
|
|
return rawExif, fmt.Errorf("%s while parsing tiff file", err)
|
|
} else {
|
|
_, rawExif, err = cs.Exif()
|
|
|
|
if err != nil {
|
|
if err.Error() == "file does not have EXIF" || strings.HasPrefix(err.Error(), "no exif data") {
|
|
return rawExif, fmt.Errorf("found no exif header")
|
|
} else {
|
|
log.Infof("metadata: %s in %s (parse tiff)", err, logName)
|
|
}
|
|
} else {
|
|
parsed = true
|
|
}
|
|
}
|
|
default:
|
|
log.Debugf("metadata: no native file format support for %s, performing brute-force exif search", logName)
|
|
bruteForce = true
|
|
}
|
|
|
|
// Start brute-force search for Exif data?
|
|
if !parsed && bruteForce {
|
|
rawExif, err = exif.SearchFileAndExtractExif(fileName)
|
|
|
|
if err != nil {
|
|
return rawExif, fmt.Errorf("found no exif data")
|
|
}
|
|
}
|
|
|
|
return rawExif, nil
|
|
}
|