initial

src/main/main.go

@@ -0,0 +1,90 @@
+package main
+
+import (
+	"bytes"
+	"errors"
+	"fmt"
+	"net/http"
+	"os"
+	"time"
+
+	"github.com/alexflint/go-arg"
+	"server"
+	"server/settings"
+	"server/version"
+)
+
+type args struct {
+	Port string `arg:"-p" help:"web server port"`
+	Path string `arg:"-d" help:"database path"`
+	Add  string `arg:"-a" help:"add torrent link and exit"`
+	Kill bool   `arg:"-k" help:"dont kill program on signal"`
+}
+
+func (args) Version() string {
+	return "TorrServer " + version.Version
+}
+
+var params args
+
+func main() {
+	//test()
+	//return
+	//for _, g := range tmdb.GetMovieGenres("ru") {
+	//	fmt.Println(g.Name, g.ID)
+	//}
+	//return
+
+	//movs, _ := tmdb.DiscoverShows(map[string]string{}, 1)
+	//js, _ := json.MarshalIndent(movs, "", " ")
+	//fmt.Println(string(js))
+	//return
+
+	arg.MustParse(&params)
+
+	if params.Path == "" {
+		params.Path, _ = os.Getwd()
+	}
+
+	if params.Port == "" {
+		params.Port = "8090"
+	}
+
+	if params.Add != "" {
+		add()
+	}
+
+	Preconfig(params.Kill)
+
+	server.Start(params.Path, params.Port)
+	settings.SaveSettings()
+	fmt.Println(server.WaitServer())
+	time.Sleep(time.Second * 3)
+	os.Exit(0)
+}
+
+func add() {
+	err := addRemote()
+	if err != nil {
+		fmt.Println("Error add torrent:", err)
+		os.Exit(-1)
+	}
+
+	fmt.Println("Added ok")
+	os.Exit(0)
+}
+
+func addRemote() error {
+	url := "http://localhost:" + params.Port + "/torrent/add"
+	fmt.Println("Add torrent link:", params.Add, "\n", url)
+
+	json := `{"Link":"` + params.Add + `"}`
+	resp, err := http.Post(url, "text/html; charset=utf-8", bytes.NewBufferString(json))
+	if err != nil {
+		return err
+	}
+	if resp.StatusCode != 200 {
+		return errors.New(resp.Status)
+	}
+	return nil
+}