Initial commit: docker compose config

server/server.go

@@ -0,0 +1,146 @@
+package server
+
+import (
+	"net"
+	"os"
+	"path/filepath"
+	"strconv"
+
+	"server/tgbot"
+
+	"server/log"
+	"server/settings"
+	"server/web"
+)
+
+func Start() {
+	settings.InitSets(settings.Args.RDB, settings.Args.SearchWA)
+	// https checks
+	if settings.Args.Ssl {
+		// set settings ssl enabled
+		settings.Ssl = settings.Args.Ssl
+		if settings.Args.SslPort == "" {
+			dbSSlPort := strconv.Itoa(settings.BTsets.SslPort)
+			if dbSSlPort != "0" {
+				settings.Args.SslPort = dbSSlPort
+			} else {
+				settings.Args.SslPort = "8091"
+			}
+		} else { // store ssl port from params to DB
+			dbSSlPort, err := strconv.Atoi(settings.Args.SslPort)
+			if err == nil {
+				settings.BTsets.SslPort = dbSSlPort
+			}
+		}
+		// check if ssl cert and key files exist
+		if settings.Args.SslCert != "" && settings.Args.SslKey != "" {
+			// set settings ssl cert and key files
+			settings.BTsets.SslCert = settings.Args.SslCert
+			settings.BTsets.SslKey = settings.Args.SslKey
+		}
+		log.TLogln("Check web ssl port", settings.Args.SslPort)
+		l, err := net.Listen("tcp", settings.Args.IP+":"+settings.Args.SslPort)
+		if l != nil {
+			l.Close()
+		}
+		if err != nil {
+			log.TLogln("Port", settings.Args.SslPort, "already in use! Please set different ssl port for HTTPS. Abort")
+			os.Exit(1)
+		}
+	}
+	// http checks
+	if settings.Args.Port == "" {
+		settings.Args.Port = "8090"
+	}
+
+	log.TLogln("Check web port", settings.Args.Port)
+	l, err := net.Listen("tcp", settings.Args.IP+":"+settings.Args.Port)
+	if l != nil {
+		l.Close()
+	}
+	if err != nil {
+		log.TLogln("Port", settings.Args.Port, "already in use! Please set different port for HTTP. Abort")
+		os.Exit(1)
+	}
+	// remove old disk caches
+	go cleanCache()
+	// set settings http and https ports. Start web server.
+	settings.Port = settings.Args.Port
+	settings.SslPort = settings.Args.SslPort
+	settings.IP = settings.Args.IP
+
+	if settings.Args.TGToken != "" {
+		if err := tgbot.Start(settings.Args.TGToken); err != nil {
+			log.TLogln("tg bot start failed", err)
+		}
+	}
+	web.Start()
+}
+
+func cleanCache() {
+	if !settings.BTsets.UseDisk || settings.BTsets.TorrentsSavePath == "/" || settings.BTsets.TorrentsSavePath == "" {
+		return
+	}
+
+	dirs, err := os.ReadDir(settings.BTsets.TorrentsSavePath)
+	if err != nil {
+		return
+	}
+
+	torrs := settings.ListTorrent()
+
+	log.TLogln("Remove unused cache in dir:", settings.BTsets.TorrentsSavePath)
+	keep := map[string]bool{}
+	for _, d := range dirs {
+		if len(d.Name()) != 40 {
+			// Not a hash
+			continue
+		}
+
+		if !settings.BTsets.RemoveCacheOnDrop {
+			keep[d.Name()] = true
+			for _, t := range torrs {
+				if d.IsDir() && d.Name() == t.InfoHash.HexString() {
+					keep[d.Name()] = false
+					break
+				}
+			}
+			for hash, del := range keep {
+				if del && hash == d.Name() {
+					log.TLogln("Remove unused cache:", d.Name())
+					removeAllFiles(filepath.Join(settings.BTsets.TorrentsSavePath, d.Name()))
+				}
+			}
+		} else {
+			if d.IsDir() {
+				log.TLogln("Remove unused cache:", d.Name())
+				removeAllFiles(filepath.Join(settings.BTsets.TorrentsSavePath, d.Name()))
+			}
+		}
+	}
+}
+
+func removeAllFiles(path string) {
+	files, err := os.ReadDir(path)
+	if err != nil {
+		return
+	}
+	for _, f := range files {
+		name := filepath.Join(path, f.Name())
+		os.Remove(name)
+	}
+	os.Remove(path)
+}
+
+func WaitServer() string {
+	err := web.Wait()
+	if err != nil {
+		return err.Error()
+	}
+	return ""
+}
+
+func Stop() {
+	web.Stop()
+	settings.CloseDB()
+}