⚗️ Perf

Author: soxft

.DS_Store

@@ -0,0 +1,3 @@
+qumagie
+pics/*
+!pics/.gitkeep

.gitignore

@@ -36,4 +36,9 @@
     $ chmod +x /share/Public/quexif-linux-amd64-{{version}}
 
     $ /share/Public/quexif-linux-amd64-{{version}} -p /share/Public/Photo
-```
+```
+
+
+## Thinks
+
+- [go-exif](//github.com/dsoprea/go-exif/v3)

.idea/photo_exif_do.iml

@@ -14,7 +14,7 @@ func Parse() {
 	flag.StringVar(&Mode, "m", "qumagie", "操作模式: qumagie (QuMagie 备份照片处理), dir (指定文件夹批量修改 EXIF时间), dirDate (按照上级文件夹名称修改 EXIF 时间)")
 	flag.StringVar(&Path, "p", "", "文件夹路径")
 	flag.StringVar(&DateTime, "datetime", "", "日期时间")
-	flag.StringVar(&DateTpl, "tpl", "2006-01-02 15.04.05", "日期时间模板, 默认为 '2006-01-02 15.04.05' 请参照 Golang 时间 layout 设置")
+	flag.StringVar(&DateTpl, "tpl", "2006-01-02 15.04.05", "日期时间模板, 默认为 '2006-01-02 15.04.05' 请参照 Golang 时间 layout 设置, 不适用于 QuMagie 模式")
 
 	flag.Parse()
 }

README.md

@@ -5,16 +5,24 @@ go 1.23
 require (
 	github.com/dsoprea/go-exif/v3 v3.0.1
 	github.com/dsoprea/go-jpeg-image-structure/v2 v2.0.0-20221012074422-4f3f7e934102
+	github.com/dsoprea/go-png-image-structure/v2 v2.0.0-20210512210324-29b889a6093d
 )
 
 require (
-	github.com/dsoprea/go-iptc v0.0.0-20200609062250-162ae6b44feb // indirect
+	github.com/dsoprea/go-exif-extra v0.0.0-20210512210440-c683d9263a55 // indirect
+	github.com/dsoprea/go-exif-knife v0.0.0-20210512212132-e3a47364f3e3 // indirect
+	github.com/dsoprea/go-heic-exif-extractor/v2 v2.0.0-20210512044107-62067e44c235 // indirect
+	github.com/dsoprea/go-iptc v0.0.0-20200610044640-bc9ca208b413 // indirect
 	github.com/dsoprea/go-logging v0.0.0-20200710184922-b02d349568dd // indirect
-	github.com/dsoprea/go-photoshop-info-format v0.0.0-20200609050348-3db9b63b202c // indirect
+	github.com/dsoprea/go-photoshop-info-format v0.0.0-20200610045659-121dd752914d // indirect
+	github.com/dsoprea/go-tiff-image-structure/v2 v2.0.0-20210512044046-dc78da6a809b // indirect
 	github.com/dsoprea/go-utility/v2 v2.0.0-20221003172846-a3e1774ef349 // indirect
+	github.com/dsoprea/go-webp-image-structure v0.0.0-20210512044215-f98af2b0401e // indirect
 	github.com/go-errors/errors v1.4.2 // indirect
 	github.com/go-xmlfmt/xmlfmt v0.0.0-20191208150333-d5b6f63a941b // indirect
 	github.com/golang/geo v0.0.0-20210211234256-740aa86cb551 // indirect
+	go4.org v0.0.0-20200411211856-f5505b9728dd // indirect
+	golang.org/x/image v0.0.0-20200618115811-c13761719519 // indirect
 	golang.org/x/net v0.0.0-20221002022538-bcab6841153b // indirect
 	gopkg.in/yaml.v2 v2.4.0 // indirect
 )

fg/fg.go

@@ -1,214 +1,39 @@
 package main
 
 import (
-	"bytes"
-	"fmt"
-	"github.com/dsoprea/go-exif/v3"
-	exifcommon "github.com/dsoprea/go-exif/v3/common"
-	jpeg "github.com/dsoprea/go-jpeg-image-structure/v2"
 	"log"
-	"os"
-	"path/filepath"
 	"photo_exif_do/fg"
-	"strings"
+	"photo_exif_do/qumagie"
+	"photo_exif_do/x_exif"
 	"time"
 )
 
-// getFileName 从文件路径中解析出文件名，并将其转换为时间对象
-// 时间格式为 "2006-01-02 15.04.05"
-func getFileName(filePath string) (time.Time, error) {
-	// 从路径中提取文件名
-	fileName := filepath.Base(filePath)
-
-	fileNameWithoutExt := strings.TrimSuffix(fileName, filepath.Ext(fileName))
-	fileNameWithoutExt = fileNameWithoutExt[:19]
-
-	// 将文件名解析为时间
-	layout := "2006-01-02 15.04.05"
-	parsedTime, err := time.Parse(layout, fileNameWithoutExt)
-	if err != nil {
-		return time.Time{}, fmt.Errorf("解析时间失败: %v", err)
-	}
-
-	return parsedTime, nil
-}
-
-// ReadExif 获取exif 中的 DateTimeOriginal
-func ReadExif(path string) string {
-	opt := exif.ScanOptions{}
-	dt, err := exif.SearchFileAndExtractExif(path)
-	if err != nil {
-		fmt.Println(err)
-		return ""
-	}
-	ets, _, err := exif.GetFlatExifData(dt, &opt)
-	if err != nil {
-		fmt.Println(err)
-		return ""
-	}
-	for _, et := range ets {
-		if et.TagName == "DateTimeOriginal" {
-			return et.Value.(string)
-		}
-	}
-
-	return ""
-}
-
-// setExifTag 设置exif标签
-func setExifTag(rootIB *exif.IfdBuilder, ifdPath, tagName, tagValue string) error {
-	// fmt.Printf("setTag(): ifdPath: %v, tagName: %v, tagValue: %v",
-	//	ifdPath, tagName, tagValue)
-
-	ifdIb, err := exif.GetOrCreateIbFromRootIb(rootIB, ifdPath)
-	if err != nil {
-		return fmt.Errorf("failed to get or create IB: %v", err)
-	}
-
-	if err := ifdIb.SetStandardWithName(tagName, tagValue); err != nil {
-		return fmt.Errorf("failed to set DateTime tag: %v", err)
-	}
-
-	return nil
-}
-
-// setDateIfNone 为文件设置日期 如果已经存在则跳过
-func setDateIfNone(filePath string, counter int) error {
-	parser := jpeg.NewJpegMediaParser()
-	intfc, err := parser.ParseFile(filePath)
-	if err != nil {
-		return fmt.Errorf("failed to parse JPEG file: %v", err)
-	}
-
-	sl := intfc.(*jpeg.SegmentList)
-
-	rootIb, err := sl.ConstructExifBuilder()
-	if err != nil {
-		//fmt.Println("No EXIF; creating it from scratch")
-
-		im, err := exifcommon.NewIfdMappingWithStandard()
-		if err != nil {
-			return fmt.Errorf("failed to create new IFD mapping with standard tags: %v", err)
-		}
-		ti := exif.NewTagIndex()
-		if err := exif.LoadStandardTags(ti); err != nil {
-			return fmt.Errorf("failed to load standard tags: %v", err)
-		}
-
-		rootIb = exif.NewIfdBuilder(im, ti, exifcommon.IfdStandardIfdIdentity,
-			exifcommon.EncodeDefaultByteOrder)
-	}
-
-	// 检测是否已经有日期
-	aDate := ReadExif(filePath)
-	if aDate != "" {
-		log.Printf("[SKIP](%d) %s already has date: %s\n", counter, filepath.Base(filePath), aDate)
-		return nil
-	}
-
-	// Form our timestamp string
-	// 从名称解析日期
-	t, err := getFileName(filePath)
-	if err != nil {
-		return fmt.Errorf("从文件名解析日期失败: %v", err)
-	}
-	ts := exifcommon.ExifFullTimestampString(t)
-
-	// Set DateTime
-	ifdPath := "IFD0"
-	if err := setExifTag(rootIb, ifdPath, "DateTime", ts); err != nil {
-		return fmt.Errorf("failed to set tag %v: %v", "DateTime", err)
-	}
-
-	// Set DateTimeOriginal
-	ifdPath = "IFD/Exif"
-	if err := setExifTag(rootIb, ifdPath, "DateTimeOriginal", ts); err != nil {
-		return fmt.Errorf("failed to set tag %v: %v", "DateTimeOriginal", err)
-	}
-
-	// Update the exif segment.
-	if err := sl.SetExif(rootIb); err != nil {
-		return fmt.Errorf("failed to set EXIF to jpeg: %v", err)
-	}
-
-	// Write the modified file
-	b := new(bytes.Buffer)
-	if err := sl.Write(b); err != nil {
-		return fmt.Errorf("failed to create JPEG data: %v", err)
-	}
-
-	// Save the file
-	if err := os.WriteFile(filePath, b.Bytes(), 0644); err != nil {
-		return fmt.Errorf("failed to write JPEG file: %v", err)
-	}
-
-	log.Printf("[SUCC](%d) %s -> %s\n", counter, filepath.Base(filePath), ts)
-
-	return nil
-}
-
-// removeEditStr 如果文件名中包含 (edited) 则自动重命名该文件
-// 主要用于处理 QFiling 归档文件
-func removeEditStr(path string) {
-	if strings.Contains(path, "(edited)") {
-		newPath := strings.ReplaceAll(path, "(edited)", "")
-		if err := os.Rename(path, newPath); err != nil {
-			log.Printf("重命名失败: %v\n", err)
-		} else {
-			log.Printf("重命名成功: %s -> %s\n", path, newPath)
-		}
-	}
-}
-
-// quMagie 处理 QuMagie 备份的照片
-func quMagie(path string) {
-	files, err := os.ReadDir(path)
-	if err != nil {
-		log.Fatal(err)
-	}
-
-	counter := 1
-
-	// 遍历目录下的文件
-	for _, file := range files {
-		// QuMagie 不会有子文件夹, 所以不需要递归
-		if file.IsDir() {
-			continue
-		}
-
-		filePath := filepath.Join(path, file.Name())
-		if strings.HasSuffix(strings.ToLower(filePath), ".jpg") || strings.HasSuffix(strings.ToLower(filePath), ".jpeg") {
-			if err := setDateIfNone(filePath, counter); err != nil {
-				log.Printf("[FAIL](%d) %s: %v\n", counter, filePath, err)
-			} else {
-				removeEditStr(filePath)
-			}
-		} else {
-			log.Printf("[SKIP](%d) %s\n", counter, filePath)
-		}
-
-		counter++
-	}
-}
-
 func main() {
 	fg.Parse()
 
 	// 安全 QA
-	log.Println("请确保已经设置了快照，程序将会直接修改文件的 exif 元数据, 建议在使用前选择少量照片进行测试后再使用")
-	log.Println("您确认要继续吗? (y/n)")
-	var confirm string
-
-	if _, err := fmt.Scanln(&confirm); err != nil {
-		return
-	} else if confirm != "y" {
-		log.Fatal("已取消")
-	}
+	//log.Println("请确保已经设置了快照，程序将会直接修改文件的 exif 元数据, 建议在使用前选择少量照片进行测试后再使用")
+	//log.Println("您确认要继续吗? (y/n)")
+	//var confirm string
+	//
+	//if _, err := fmt.Scanln(&confirm); err != nil {
+	//	return
+	//} else if confirm != "y" {
+	//	log.Fatal("已取消")
+	//}
 
 	// 读取目录
 	switch fg.Mode {
+	case "dir": // 指定文件夹批量修改
+		break
+	case "dirDate": //按照上级文件夹名称修改
+		break
+	case "test":
+		log.Println(x_exif.SetDate("pics/qumagie/2006-01-02 15.04.05.jpg", time.Now(), true))
+
+		log.Println(x_exif.ReadExif("pics/qumagie/2006-01-02 15.04.05.jpg"))
 	default:
-		quMagie(fg.Path)
+		qumagie.Run(fg.Path)
 	}
 
 }

go.mod

@@ -0,0 +1,73 @@
+package qumagie
+
+import (
+	"errors"
+	"fmt"
+	"log"
+	"os"
+	"path/filepath"
+	"photo_exif_do/x_exif"
+	"regexp"
+	"time"
+)
+
+// getTime 从文件路径中解析出文件名，并将其转换为时间对象
+// 时间格式为 "2006-01-02 15.04.05"
+// NOTICE 仅适用于 QuMagie 备份的文件名格式
+func getTime(filePath string) (time.Time, error) {
+	// 从路径中提取文件名
+	fileName := filepath.Base(filePath)
+
+	pattern := `\d{4}-\d{2}-\d{2} \d{2}\.\d{2}\.\d{2}`
+	re := regexp.MustCompile(pattern)
+	timeStr := re.FindString(fileName)
+
+	// 将文件名解析为时间
+	layout := "2006-01-02 15.04.05"
+	parsedTime, err := time.Parse(layout, timeStr)
+	if err != nil {
+		return time.Time{}, fmt.Errorf("解析时间失败: %v", err)
+	}
+
+	return parsedTime, nil
+}
+
+// Run 处理 QuMagie 备份的照片
+func Run(path string) {
+	files, err := os.ReadDir(path)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	counter := 1
+
+	// 遍历目录下的文件
+	for _, file := range files {
+		// QuMagie 不会有子文件夹, 所以不需要递归
+		if file.IsDir() {
+			continue
+		}
+
+		filePath := filepath.Join(path, file.Name())
+		// 从路径中提取文件名
+		filename := filepath.Base(filePath)
+
+		t, err := getTime(filePath)
+		if err != nil {
+			log.Printf("[SKIP](%d) %s -> %v\n", counter, filename, err)
+			counter++
+			continue
+		}
+
+		if err := x_exif.SetDate(filePath, t, true); err == nil {
+			log.Printf("[SUCC](%d) %s -> %s\n", counter, filename, t.Format("2006-01-02 15.04.05"))
+			x_exif.RemoveEditStr(filePath)
+		} else if errors.Is(err, x_exif.ErrAlreadyHasDate) || errors.Is(err, x_exif.ErrMediaTypeNotSupport) {
+			log.Printf("[SKIP](%d) %s -> %v\n", counter, filename, err)
+		} else {
+			log.Printf("[FAIL](%d) %s -> %v\n", counter, filename, err)
+		}
+
+		counter++
+	}
+}