chaturbate-dvr/main.go

package main

import (
	"embed"
	"fmt"
	"io/fs"
	"log"
	"net/http"
	"os"

	"github.com/gin-gonic/gin"
	"github.com/teacat/chaturbate-dvr/chaturbate"
	"github.com/teacat/chaturbate-dvr/handler"
	"github.com/urfave/cli/v2"
)

const logo = `
 ██████╗██╗  ██╗ █████╗ ████████╗██╗   ██╗██████╗ ██████╗  █████╗ ████████╗███████╗
██╔════╝██║  ██║██╔══██╗╚══██╔══╝██║   ██║██╔══██╗██╔══██╗██╔══██╗╚══██╔══╝██╔════╝
██║     ███████║███████║   ██║   ██║   ██║██████╔╝██████╔╝███████║   ██║   █████╗
██║     ██╔══██║██╔══██║   ██║   ██║   ██║██╔══██╗██╔══██╗██╔══██║   ██║   ██╔══╝
╚██████╗██║  ██║██║  ██║   ██║   ╚██████╔╝██║  ██║██████╔╝██║  ██║   ██║   ███████╗
 ╚═════╝╚═╝  ╚═╝╚═╝  ╚═╝   ╚═╝    ╚═════╝ ╚═╝  ╚═╝╚═════╝ ╚═╝  ╚═╝   ╚═╝   ╚══════╝
██████╗ ██╗   ██╗██████╗
██╔══██╗██║   ██║██╔══██╗
██║  ██║██║   ██║██████╔╝
██║  ██║╚██╗ ██╔╝██╔══██╗
██████╔╝ ╚████╔╝ ██║  ██║
╚═════╝   ╚═══╝  ╚═╝  ╚═╝`

func main() {
	app := &cli.App{
		Name:    "chaturbate-dvr",
		Version: "1.0.2",
		Usage:   "Records your favorite Chaturbate stream 😎🫵",
		Flags: []cli.Flag{
			&cli.StringFlag{
				Name:    "username",
				Aliases: []string{"u"},
				Usage:   "channel username to record",
				Value:   "",
			},
			&cli.StringFlag{
				Name:    "gui-username",
				Aliases: []string{"gui-u"},
				Usage:   "username for auth web (optional)",
				Value:   "",
			},
			&cli.StringFlag{
				Name:    "gui-password",
				Aliases: []string{"gui-p"},
				Usage:   "password for auth web (optional)",
				Value:   "",
			},
			&cli.IntFlag{
				Name:    "framerate",
				Aliases: []string{"f"},
				Usage:   "preferred framerate",
				Value:   30,
			},
			&cli.IntFlag{
				Name:    "resolution",
				Aliases: []string{"r"},
				Usage:   "preferred resolution",
				Value:   1080,
			},
			&cli.StringFlag{
				Name:    "resolution-fallback",
				Aliases: []string{"rf"},
				Usage:   "fallback to 'up' (larger) or 'down' (smaller) resolution if preferred resolution is not available",
				Value:   "down",
			},
			&cli.StringFlag{
				Name:    "filename-pattern",
				Aliases: []string{"fp"},
				Usage:   "filename pattern for videos",
				Value:   "videos/{{.Username}}_{{.Year}}-{{.Month}}-{{.Day}}_{{.Hour}}-{{.Minute}}-{{.Second}}{{if .Sequence}}_{{.Sequence}}{{end}}",
			},
			&cli.IntFlag{
				Name:    "split-duration",
				Aliases: []string{"sd"},
				Usage:   "minutes to split each video into segments ('0' to disable)",
				Value:   0,
			},
			&cli.IntFlag{
				Name:    "split-filesize",
				Aliases: []string{"sf"},
				Usage:   "size in MB to split each video into segments ('0' to disable)",
				Value:   0,
			},
			&cli.StringFlag{
				Name:  "log-level",
				Usage: "log level, availables: 'DEBUG', 'INFO', 'WARN', 'ERROR'",
				Value: "INFO",
			},
			&cli.StringFlag{
				Name:  "port",
				Usage: "port to expose the web interface and API",
				Value: "8080",
			},
			//&cli.StringFlag{
			//	Name:  "gui",
			//	Usage: "enabling GUI, availables: 'no', 'web'",
			//	Value: "web",
			//},
		},
		Action: start,
	}
	if err := app.Run(os.Args); err != nil {
		log.Fatal(err)
	}
}

func start(c *cli.Context) error {
	fmt.Println(logo)

	//if c.String("gui") == "web" {
	if c.String("username") == "" {
		return startWeb(c)
	}

	m := chaturbate.NewManager(c)
	if err := m.CreateChannel(&chaturbate.Config{
		Username:           c.String("username"),
		Framerate:          c.Int("framerate"),
		Resolution:         c.Int("resolution"),
		ResolutionFallback: c.String("resolution-fallback"),
		FilenamePattern:    c.String("filename-pattern"),
		SplitDuration:      c.Int("split-duration"),
		SplitFilesize:      c.Int("split-filesize"),
	}); err != nil {
		return err
	}

	select {} // block forever
}

//go:embed handler/view
var FS embed.FS

func startWeb(c *cli.Context) error {
	gin.SetMode(gin.ReleaseMode)
	r := gin.Default()

	//r.Use(cors.Default())
	m := chaturbate.NewManager(c)
	if err := m.LoadChannels(); err != nil {
		return err
	}

	fe, err := fs.Sub(FS, "handler/view")
	if err != nil {
		log.Fatalln(err)
	}

	guiUsername := c.String("gui-username")
	guiPassword := c.String("gui-password")

	var authorized = r.Group("/")
	var authorizedApi = r.Group("/api")

	if guiUsername != "" && guiPassword != "" {
		ginBasicAuth := gin.BasicAuth(gin.Accounts{
			guiUsername: guiPassword,
		})
		authorized.Use(ginBasicAuth)
		authorizedApi.Use(ginBasicAuth)
	}

	authorized.StaticFS("/static", http.FS(fe))
	authorized.StaticFileFS("/", "/", http.FS(fe))

	authorizedApi.POST("/get_channel", handler.NewGetChannelHandler(m, c).Handle)
	authorizedApi.POST("/create_channel", handler.NewCreateChannelHandler(m, c).Handle)
	authorizedApi.POST("/list_channels", handler.NewListChannelsHandler(m, c).Handle)
	authorizedApi.POST("/delete_channel", handler.NewDeleteChannelHandler(m, c).Handle)
	authorizedApi.POST("/pause_channel", handler.NewPauseChannelHandler(m, c).Handle)
	authorizedApi.POST("/resume_channel", handler.NewResumeChannelHandler(m, c).Handle)
	authorizedApi.GET("/listen_update", handler.NewListenUpdateHandler(m, c).Handle)
	authorizedApi.POST("/get_settings", handler.NewGetSettingsHandler(c).Handle)
	authorizedApi.POST("/terminate_program", handler.NewTerminateProgramHandler(c).Handle)

	fmt.Printf("👋 Visit http://localhost:%s to use the Web UI\n", c.String("port"))

	return r.Run(fmt.Sprintf(":%s", c.String("port")))
}