3ded768a6f
* testing message queue * better mq syncronisation * major code refactoring, concern separation. * bugfix * code refactoring * queuesize configurable via flags * code refactoring * comments * code refactoring, updated readme
65 lines
1.3 KiB
Go
65 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"embed"
|
|
"flag"
|
|
"io/fs"
|
|
"log"
|
|
"runtime"
|
|
|
|
"github.com/marcopeocchi/yt-dlp-web-ui/server"
|
|
"github.com/marcopeocchi/yt-dlp-web-ui/server/config"
|
|
)
|
|
|
|
var (
|
|
port int
|
|
queueSize int
|
|
configFile string
|
|
downloadPath string
|
|
downloaderPath string
|
|
|
|
requireAuth bool
|
|
rpcSecret string
|
|
|
|
//go:embed frontend/dist
|
|
frontend embed.FS
|
|
)
|
|
|
|
func init() {
|
|
flag.IntVar(&port, "port", 3033, "Port where server will listen at")
|
|
flag.IntVar(&queueSize, "qs", runtime.NumCPU(), "Download queue size")
|
|
|
|
flag.StringVar(&configFile, "conf", "", "Config file path")
|
|
flag.StringVar(&downloadPath, "out", ".", "Where files will be saved")
|
|
flag.StringVar(&downloaderPath, "driver", "yt-dlp", "yt-dlp executable path")
|
|
|
|
flag.BoolVar(&requireAuth, "auth", false, "Enable RPC authentication")
|
|
flag.StringVar(&rpcSecret, "secret", "", "Secret required for auth")
|
|
|
|
flag.Parse()
|
|
}
|
|
|
|
func main() {
|
|
frontend, err := fs.Sub(frontend, "frontend/dist")
|
|
|
|
if err != nil {
|
|
log.Fatalln(err)
|
|
}
|
|
|
|
c := config.Instance()
|
|
|
|
c.SetPort(port)
|
|
c.QueueSize(queueSize)
|
|
c.DownloadPath(downloadPath)
|
|
c.DownloaderPath(downloaderPath)
|
|
|
|
c.RequireAuth(requireAuth)
|
|
c.RPCSecret(rpcSecret)
|
|
|
|
if configFile != "" {
|
|
c.LoadFromFile(configFile)
|
|
}
|
|
|
|
server.RunBlocking(port, frontend)
|
|
}
|