-
Notifications
You must be signed in to change notification settings - Fork 0
/
fs.go
93 lines (80 loc) · 2.25 KB
/
fs.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
// Copyright 2024 defsub
//
// This file is part of TakeoutFM.
//
// TakeoutFM is free software: you can redistribute it and/or modify it under the
// terms of the GNU Affero General Public License as published by the Free
// Software Foundation, either version 3 of the License, or (at your option)
// any later version.
//
// TakeoutFM is distributed in the hope that it will be useful, but WITHOUT ANY
// WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
// FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for
// more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with TakeoutFM. If not, see <https://www.gnu.org/licenses/>.
// Package bucket provides support for listing S3 bucket contents and creating
// presigned URLs for fetching media. The AWS SDK is used to provide S3
// functionality, however, any S3 compatible backend service is supported.
package bucket
import (
"net/url"
"os"
"path/filepath"
"time"
"github.com/takeoutfm/takeout/lib/hash"
"github.com/takeoutfm/takeout/lib/log"
)
type FSConfig struct {
Root string
}
type fileBucket struct {
config Config
}
func newFSBucket(config Config) *fileBucket {
return &fileBucket{config: config}
}
func (f *fileBucket) IsLocal() bool {
return true //f.config.Local
}
func (f *fileBucket) List(lastSync time.Time) (objectCh chan *Object, err error) {
objectCh = make(chan *Object)
walk := func(path string, entry os.DirEntry, err error) error {
if err != nil {
return err
}
if entry.Type().IsRegular() {
info, err := entry.Info()
if err != nil {
return err
}
if info.ModTime().After(lastSync) {
var etag string
etag, err = hash.MD5Sum(path)
if err != nil {
log.Printf("etag %s: %s\n", path, err)
} else {
objectCh <- &Object{
Key: path,
Path: rewrite(f.config.RewriteRules, path),
ETag: etag,
Size: info.Size(),
LastModified: info.ModTime(),
}
}
}
}
return err
}
go func() {
defer close(objectCh)
err = filepath.WalkDir(f.config.FS.Root, walk)
}()
return
}
func (fileBucket) ObjectURL(key string) *url.URL {
url, err := url.Parse("file://" + key)
log.CheckError(err)
return url
}