78 lines
1.8 KiB
Go
78 lines
1.8 KiB
Go
package main
|
|
|
|
import (
|
|
"embed"
|
|
"flag"
|
|
"io/fs"
|
|
"log"
|
|
"os"
|
|
"runtime"
|
|
|
|
"github.com/marcopeocchi/yt-dlp-web-ui/server"
|
|
"github.com/marcopeocchi/yt-dlp-web-ui/server/cli"
|
|
"github.com/marcopeocchi/yt-dlp-web-ui/server/config"
|
|
)
|
|
|
|
var (
|
|
port int
|
|
queueSize int
|
|
configFile string
|
|
downloadPath string
|
|
downloaderPath string
|
|
sessionFilePath string
|
|
|
|
requireAuth bool
|
|
username string
|
|
password string
|
|
|
|
userFromEnv = os.Getenv("USERNAME")
|
|
passFromEnv = os.Getenv("PASSWORD")
|
|
|
|
//go:embed frontend/dist/index.html
|
|
//go:embed frontend/dist/assets/*
|
|
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.yml", "Config file path")
|
|
flag.StringVar(&downloadPath, "out", ".", "Where files will be saved")
|
|
flag.StringVar(&downloaderPath, "driver", "yt-dlp", "yt-dlp executable path")
|
|
flag.StringVar(&sessionFilePath, "session", ".", "session file path")
|
|
|
|
flag.BoolVar(&requireAuth, "auth", false, "Enable RPC authentication")
|
|
flag.StringVar(&username, "user", userFromEnv, "Username required for auth")
|
|
flag.StringVar(&password, "pass", passFromEnv, "Password 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.Username(username)
|
|
c.Password(password)
|
|
|
|
// if config file is found it will be merged with the current config struct
|
|
if _, err := c.LoadFromFile(configFile); err != nil {
|
|
log.Println(cli.BgRed, "config", cli.Reset, "no config file found")
|
|
}
|
|
|
|
server.RunBlocking(port, frontend)
|
|
}
|