123 lines
2.2 KiB
Go
123 lines
2.2 KiB
Go
|
package dirwatch
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bitbucket.org/anacrolix/go.torrent"
|
||
|
|
"github.com/anacrolix/libtorgo/metainfo"
|
||
|
|
"github.com/go-fsnotify/fsnotify"
|
||
|
|
"log"
|
||
|
|
"os"
|
||
|
|
"path/filepath"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Change uint
|
||
|
|
|
||
|
|
const (
|
||
|
|
Added Change = iota
|
||
|
|
Removed
|
||
|
|
)
|
||
|
|
|
||
|
|
type Event struct {
|
||
|
|
Magnet string
|
||
|
|
Change
|
||
|
|
TorrentFilePath string
|
||
|
|
InfoHash torrent.InfoHash
|
||
|
|
}
|
||
|
|
|
||
|
|
type Instance struct {
|
||
|
|
w *fsnotify.Watcher
|
||
|
|
dirName string
|
||
|
|
Events chan Event
|
||
|
|
torrentFileInfoHashes map[string]torrent.InfoHash
|
||
|
|
}
|
||
|
|
|
||
|
|
func (me *Instance) handleEvents() {
|
||
|
|
for e := range me.w.Events {
|
||
|
|
log.Printf("event: %s", e)
|
||
|
|
me.processFile(e.Name)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (me *Instance) handleErrors() {
|
||
|
|
for err := range me.w.Errors {
|
||
|
|
log.Printf("error in torrent directory watcher: %s", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func torrentFileInfoHash(fileName string) (ih torrent.InfoHash, ok bool) {
|
||
|
|
mi, _ := metainfo.LoadFromFile(fileName)
|
||
|
|
if mi == nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
if 20 != copy(ih[:], mi.Info.Hash) {
|
||
|
|
panic(mi.Info.Hash)
|
||
|
|
}
|
||
|
|
ok = true
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func (me *Instance) processFile(name string) {
|
||
|
|
name = filepath.Clean(name)
|
||
|
|
log.Print(name)
|
||
|
|
if filepath.Ext(name) != ".torrent" {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
ih, ok := me.torrentFileInfoHashes[name]
|
||
|
|
if ok {
|
||
|
|
me.Events <- Event{
|
||
|
|
TorrentFilePath: name,
|
||
|
|
Change: Removed,
|
||
|
|
InfoHash: ih,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
delete(me.torrentFileInfoHashes, name)
|
||
|
|
ih, ok = torrentFileInfoHash(name)
|
||
|
|
if ok {
|
||
|
|
me.torrentFileInfoHashes[name] = ih
|
||
|
|
me.Events <- Event{
|
||
|
|
TorrentFilePath: name,
|
||
|
|
Change: Added,
|
||
|
|
InfoHash: ih,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (me *Instance) addDir() (err error) {
|
||
|
|
f, err := os.Open(me.dirName)
|
||
|
|
if err != nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
defer f.Close()
|
||
|
|
names, err := f.Readdirnames(-1)
|
||
|
|
if err != nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
for _, n := range names {
|
||
|
|
me.processFile(filepath.Join(me.dirName, n))
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func New(dirName string) (i *Instance, err error) {
|
||
|
|
w, err := fsnotify.NewWatcher()
|
||
|
|
if err != nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
err = w.Add(dirName)
|
||
|
|
if err != nil {
|
||
|
|
w.Close()
|
||
|
|
return
|
||
|
|
}
|
||
|
|
i = &Instance{
|
||
|
|
w: w,
|
||
|
|
dirName: dirName,
|
||
|
|
Events: make(chan Event),
|
||
|
|
torrentFileInfoHashes: make(map[string]torrent.InfoHash, 20),
|
||
|
|
}
|
||
|
|
go func() {
|
||
|
|
i.addDir()
|
||
|
|
go i.handleEvents()
|
||
|
|
go i.handleErrors()
|
||
|
|
}()
|
||
|
|
return
|
||
|
|
}
|