feat: Use Last-Modified header to set the file mtime by default

config/config.go

@@ -57,6 +57,8 @@ var (
 	RetryTimes int
 	// YouTubeStream2 will use data in `url_encoded_fmt_stream_map`
 	YouTubeStream2 bool
+	// NoMTime Do not use the Last-Modified header to set the file modification time
+	NoMTime bool
 )
 
 // FakeHeaders fake http headers

downloader/downloader.go

@@ -127,6 +127,21 @@ func Save(
 		}
 	}()
 
+	// try to update mtime before renaming file
+	defer func() {
+		if err != nil || config.NoMTime {
+			return
+		}
+		resHeaders, reqError := request.Headers(urlData.URL, refer)
+		if reqError != nil {
+			return
+		}
+		t, parseError := time.Parse("Mon, 02 Jan 2006 15:04:05 GMT", resHeaders.Get("Last-Modified"))
+		if parseError == nil {
+			os.Chtimes(tempFilePath, time.Now(), t)
+		}
+	}()
+
 	if chunkSizeMB > 0 {
 		var start, end, chunkSize int64
 		chunkSize = int64(chunkSizeMB) * 1024 * 1024

main.go

@@ -72,6 +72,7 @@ func init() {
 	flag.IntVar(
 		&config.RetryTimes, "retry", 10, "How many times to retry when the download failed",
 	)
+	flag.BoolVar(&config.NoMTime, "no-mtime", false, "Do not use the Last-Modified header to set the file modification time")
 	// youku
 	flag.StringVar(&config.YoukuCcode, "ccode", "0590", "Youku ccode")
 	flag.StringVar(

utils/ffmpeg.go

@@ -5,9 +5,12 @@ import (
 	"fmt"
 	"os"
 	"os/exec"
+	"time"
+
+	"github.com/iawia002/annie/config"
 )
 
-func runMergeCmd(cmd *exec.Cmd, paths []string, mergeFilePath string) error {
+func runMergeCmd(cmd *exec.Cmd, paths []string, mergeFilePath, mergedFilePath string) error {
 	var stderr bytes.Buffer
 	cmd.Stderr = &stderr
 	err := cmd.Run()
@@ -18,6 +21,12 @@ func runMergeCmd(cmd *exec.Cmd, paths []string, mergeFilePath string) error {
 	if mergeFilePath != "" {
 		os.Remove(mergeFilePath)
 	}
+	if !config.NoMTime {
+		info, err := os.Stat(paths[0])
+		if err == nil {
+			os.Chtimes(mergedFilePath, time.Now(), info.ModTime())
+		}
+	}
 	// remove parts
 	for _, path := range paths {
 		os.Remove(path)
@@ -37,7 +46,7 @@ func MergeAudioAndVideo(paths []string, mergedFilePath string) error {
 		cmds, "-c:v", "copy", "-c:a", "aac", "-strict", "experimental",
 		mergedFilePath,
 	)
-	return runMergeCmd(exec.Command("ffmpeg", cmds...), paths, "")
+	return runMergeCmd(exec.Command("ffmpeg", cmds...), paths, "", mergedFilePath)
 }
 
 // MergeToMP4 merge video parts to MP4
@@ -55,5 +64,5 @@ func MergeToMP4(paths []string, mergedFilePath string, filename string) error {
 		"ffmpeg", "-y", "-f", "concat", "-safe", "-1",
 		"-i", mergeFilePath, "-c", "copy", "-bsf:a", "aac_adtstoasc", mergedFilePath,
 	)
-	return runMergeCmd(cmd, paths, mergeFilePath)
+	return runMergeCmd(cmd, paths, mergeFilePath, mergedFilePath)
 }