add watch torrent dir

server/cmd/main.go

@@ -3,8 +3,11 @@ package main
 import (
 	"context"
 	"fmt"
+	"io/ioutil"
 	"net"
 	"os"
+	"path/filepath"
+	"strings"
 	"time"
 
 	"github.com/alexflint/go-arg"
@@ -13,7 +16,9 @@ import (
 	"server"
 	"server/log"
 	"server/settings"
+	"server/torr"
 	"server/version"
+	"server/web/api/utils"
 )
 
 type args struct {
@@ -25,6 +30,7 @@ type args struct {
 	HttpAuth    bool   `arg:"-a" help:"http auth on all requests"`
 	DontKill    bool   `arg:"-k" help:"dont kill server on signal"`
 	UI          bool   `arg:"-u" help:"run page torrserver in browser"`
+	TorrentsDir string `arg:"-t" help:"autoload torrent from dir"`
 }
 
 func (args) Version() string {
@@ -58,6 +64,10 @@ func main() {
 		}()
 	}
 
+	if params.TorrentsDir != "" {
+		go watchTDir(params.TorrentsDir)
+	}
+
 	server.Start(params.Port, params.RDB)
 	log.TLogln(server.WaitServer())
 	log.Close()
@@ -83,3 +93,34 @@ func dnsResolve() {
 		fmt.Println("Check new dns", addrs, err)
 	}
 }
+
+func watchTDir(dir string) {
+	time.Sleep(5 * time.Second)
+	path, err := filepath.Abs(dir)
+	if err != nil {
+		path = dir
+	}
+	for {
+		files, err := ioutil.ReadDir(path)
+		if err == nil {
+			for _, file := range files {
+				filename := filepath.Join(path, file.Name())
+				if strings.ToLower(filepath.Ext(file.Name())) == ".torrent" {
+					sp, err := utils.ParseLink("file://" + filename)
+					if err == nil {
+						tor, err := torr.AddTorrent(sp, "", "", "")
+						if err == nil {
+							if tor.GotInfo() {
+								torr.SaveTorrentToDB(tor)
+								tor.Drop()
+								os.Remove(filename)
+								time.Sleep(time.Second)
+							}
+						}
+					}
+				}
+			}
+		}
+		time.Sleep(time.Second)
+	}
+}