69 lines
1.8 KiB
Go
69 lines
1.8 KiB
Go
package app
|
|
|
|
import (
|
|
"path/filepath"
|
|
"time"
|
|
|
|
fs "github.com/fsnotify/fsnotify"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// This is the amount of time to wait after changes before reacting to them.
|
|
// Debounce is done because moving files into the watched directories causes
|
|
// many rapid "Write" events to fire which would cause excessive Remove/Add
|
|
// method calls on the Library. To avoid this we accumulate the changes and
|
|
// only perform them once the events have stopped for this amount of time.
|
|
const debounceTimeout = time.Second * 5
|
|
|
|
// create, write, and chmod all require an add event
|
|
const addFlags = fs.Create | fs.Write | fs.Chmod
|
|
|
|
// remove, rename, write, and chmod all require a remove event
|
|
const removeFlags = fs.Remove | fs.Rename | fs.Write | fs.Chmod
|
|
|
|
// watch library paths and update Library with changes.
|
|
func startWatcher(a *App) {
|
|
timer := time.NewTimer(debounceTimeout)
|
|
addEvents := make(map[string]struct{})
|
|
removeEvents := make(map[string]struct{})
|
|
for {
|
|
select {
|
|
case e := <-a.Watcher.Events:
|
|
if filepath.Ext(e.Name) != ".mp4" {
|
|
continue
|
|
}
|
|
log.Debugf("fsnotify event: %s", e)
|
|
if e.Op&removeFlags != 0 {
|
|
removeEvents[e.Name] = struct{}{}
|
|
}
|
|
if e.Op&addFlags != 0 {
|
|
addEvents[e.Name] = struct{}{}
|
|
}
|
|
// reset timer
|
|
timer.Reset(debounceTimeout)
|
|
case <-timer.C:
|
|
eventCount := len(removeEvents) + len(addEvents)
|
|
// handle remove events first
|
|
if len(removeEvents) > 0 {
|
|
for p := range removeEvents {
|
|
a.Library.Remove(p)
|
|
}
|
|
// clear map
|
|
removeEvents = make(map[string]struct{})
|
|
}
|
|
// then handle add events
|
|
if len(addEvents) > 0 {
|
|
for p := range addEvents {
|
|
a.Library.Add(p)
|
|
}
|
|
// clear map
|
|
addEvents = make(map[string]struct{})
|
|
}
|
|
if eventCount > 0 {
|
|
buildFeed(a)
|
|
}
|
|
// reset timer
|
|
timer.Reset(debounceTimeout)
|
|
}
|
|
}
|
|
}
|