-
Notifications
You must be signed in to change notification settings - Fork 0
/
watcher.go
118 lines (103 loc) · 2.45 KB
/
watcher.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
113
114
115
116
117
118
package main
import (
"os"
"path/filepath"
"strings"
"github.com/fsnotify/fsnotify"
"github.com/rs/zerolog/log"
)
var Watcher *fsnotify.Watcher
func initWatcher() {
var err error
Watcher, err = fsnotify.NewWatcher()
if err != nil {
log.Fatal().Err(err).Msg("fsnotify.NewWatcher:")
}
// Start listening for events
go func() {
for {
select {
case event, ok := <-Watcher.Events:
if !ok {
return
}
if event.Has(fsnotify.Write) {
if strings.HasSuffix(event.Name, "/magnet.txt") {
parseMagnetsFile(event.Name)
}
}
if strings.HasSuffix(event.Name, "/this.torrent") {
if event.Has(fsnotify.Remove) {
path := filepath.Dir(event.Name)
dropTorrent(path)
continue
}
}
// Skip all other events in torrent dir
_, err := os.Stat(filepath.Dir(event.Name) + "/this.torrent")
if err == nil {
continue
}
if event.Has(fsnotify.Create) {
stat, err := os.Stat(event.Name)
if err != nil {
log.Error().Err(err).Str("Name", event.Name).Msg("Watcher Error")
continue
}
if stat.IsDir() {
recursiveScanDir(event.Name)
} else {
if strings.HasSuffix(event.Name, ".torrent") {
handleNewTorrentFile(event.Name)
}
}
}
if event.Has(fsnotify.Remove) || event.Has(fsnotify.Rename) {
_, err := os.Stat(event.Name)
if err != nil {
dropTorrent(event.Name)
}
}
case err, ok := <-Watcher.Errors:
if !ok {
return
}
log.Error().Err(err).Msg("Error in watcher:")
}
}
}()
}
func recursiveScanDir(path string) bool {
err := Watcher.Add(path)
if err != nil {
log.Error().Str("Path", path).Err(err).Msg("Cant add to watcher:")
}
_, err = os.Stat(path + "/this.torrent")
if err == nil {
log.Info().Str("Path", path).Msg("Found torrent")
AddTorrentFile(path + "/this.torrent")
return false
}
log.Info().Str("Path", path).Msg("Watching dir")
_, err = os.Stat(path + "/magnet.txt")
if err == nil {
parseMagnetsFile(path + "/magnet.txt")
}
files, err := os.ReadDir(path)
if err != nil {
log.Fatal().Str("Path", path).Err(err).Msg("Cant read dir")
return false
}
// Searching for new torrents since the server shutdown
for _, file := range files {
if strings.HasSuffix(file.Name(), ".torrent") {
handleNewTorrentFile(path + "/" + file.Name())
}
}
for _, file := range files {
if file.IsDir() {
recursiveScanDir(path + "/" + file.Name())
}
}
return true
}