-
Notifications
You must be signed in to change notification settings - Fork 0
/
utils.go
112 lines (90 loc) · 2.37 KB
/
utils.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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
package main
import (
"context"
"crypto/sha256"
"fmt"
"io"
"net/http"
"os"
"time"
"github.com/rwcarlsen/goexif/exif"
"gopkg.in/vansante/go-ffprobe.v2"
)
func doesPathExist(path string) bool {
_, err := os.Stat(path)
if err == nil {
return true
} else if os.IsNotExist(err) {
return false
} else {
return false
}
}
func getFileHash(file *os.File) (string, error) {
// Reset the file's read position because we want to use the same file object again
defer file.Seek(0, 0)
hash := sha256.New()
if _, err := io.Copy(hash, file); err != nil {
return "", err
}
return fmt.Sprintf("%x", hash.Sum(nil)), nil
}
func copyFile(sourceFile *os.File, destPath string) error {
// Reset the file's read position because we want to use the same file object again
defer sourceFile.Seek(0, 0)
desFile, err := os.Create(destPath)
if err != nil {
return err
}
defer desFile.Close()
_, err = io.Copy(desFile, sourceFile)
if err != nil {
return err
}
return nil
}
func getFileContentType(file *os.File) (string, error) {
// Reset the file's read position because we want to use the same file object again
defer file.Seek(0, 0)
// Read the first 512 bytes of the file
buffer := make([]byte, 512)
_, err := file.Read(buffer)
if err != nil {
return "", err
}
// Detect the content type based on the file header
contentType := http.DetectContentType(buffer)
return contentType, nil
}
func getImageCreationTime(file *os.File) (time.Time, error) {
// Reset the file's read position because we want to use the same file object again
defer file.Seek(0, 0)
data, err := exif.Decode(file)
if err != nil {
return time.Time{}, err
}
creationTime, err := data.DateTime()
if err != nil {
return time.Time{}, err
}
return creationTime, nil
}
func getVideoCreationTime(file *os.File) (time.Time, error) {
// Reset the file's read position because we want to use the same file object again
defer file.Seek(0, 0)
ctx, cancelFn := context.WithTimeout(context.Background(), 5*time.Second)
defer cancelFn()
data, err := ffprobe.ProbeReader(ctx, file)
if err != nil {
return time.Time{}, err
}
creationTimeString, err := data.Format.TagList.GetString("creation_time")
if err != nil {
return time.Time{}, err
}
creationTime, err := time.Parse("2006-01-02T15:04:05.000000Z", creationTimeString)
if err != nil {
return time.Time{}, err
}
return creationTime, nil
}