ytdl-web/web/serve.go

package web

import (
	"fmt"
	"net/url"
	"sort"

	"github.com/gofiber/fiber/v2"
	"github.com/samber/lo"
	"github.com/spf13/viper"
	"github.com/sujit-baniya/flash"
	"go.fifitido.net/ytdl-web/version"
	"go.fifitido.net/ytdl-web/ytdl"
	"golang.org/x/exp/slog"
)

func Serve() error {
	engine := ViewsEngine()
	app := fiber.New(fiber.Config{Views: engine})

	app.Get("/", func(c *fiber.Ctx) error {
		return c.Render("views/index", fiber.Map{
			"BasePath":     viper.GetString("base_path"),
			"Flash":        flash.Get(c),
			"Version":      version.Version,
			"Build":        version.Build,
			"YtdlpVersion": ytdl.GetVersion(),
		}, "views/layouts/main")
	})

	app.Get("/download", func(c *fiber.Ctx) error {
		urlBytes, err := url.QueryUnescape(c.Query("url"))
		if err != nil {
			return flash.WithError(c, fiber.Map{
				"error":   true,
				"message": "Invalid URL",
			}).Redirect("/")
		}
		url := string(urlBytes)
		if len(url) < 1 {
			return flash.WithError(c, fiber.Map{
				"error":   true,
				"message": "Invalid URL",
			}).Redirect("/")
		}

		meta, err := ytdl.GetMetadata(url)
		if err != nil {
			return flash.WithError(c, fiber.Map{
				"error":   true,
				"message": "Could not find a video at that url, maybe try again?",
			}).Redirect("/")
		}

		formats := lo.Filter(meta.Formats, func(item ytdl.Format, _ int) bool {
			return item.ACodec != "none" && item.VCodec != "none" && item.Protocol != "m3u8_native"
		})

		sort.Slice(formats, func(i, j int) bool {
			return formats[i].Width > formats[j].Width
		})

		return c.Render("views/download", fiber.Map{
			"BasePath":     viper.GetString("base_path"),
			"Url":          url,
			"Meta":         meta,
			"Formats":      formats,
			"Version":      version.Version,
			"Build":        version.Build,
			"YtdlpVersion": ytdl.GetVersion(),
		}, "views/layouts/main")
	})

	app.Get("/download/proxy", func(c *fiber.Ctx) error {
		urlBytes, err := url.QueryUnescape(c.Query("url"))
		if err != nil {
			slog.Error("Failed to decode url param", slog.String("error", err.Error()))
			return fiber.ErrBadRequest
		}
		url := string(urlBytes)
		if len(url) < 1 {
			return fiber.ErrBadRequest
		}

		formatId := c.Query("format")
		if len(formatId) < 1 {
			return fiber.ErrBadRequest
		}

		meta, err := ytdl.GetMetadata(url)
		if err != nil {
			slog.Error("Failed to get metadata", slog.String("error", err.Error()))
			return fiber.ErrInternalServerError
		}

		format, ok := lo.Find(meta.Formats, func(format ytdl.Format) bool {
			return format.FormatID == formatId
		})
		if !ok {
			return fiber.ErrBadRequest
		}

		c.Set("Content-Disposition", fmt.Sprintf("attachment; filename=\"%s-%s.%s\"", meta.ID, format.Resolution, format.Ext))
		if format.Filesize != nil {
			c.Set("Content-Length", fmt.Sprint(*format.Filesize))
		} else if format.FilesizeApprox != nil {
			c.Set("Content-Length", fmt.Sprint(*format.FilesizeApprox))
		}

		return ytdl.Stream(c.Response().BodyWriter(), url, format)
	})

	listenAddr := fmt.Sprintf("%s:%d", viper.GetString("listen"), viper.GetInt("port"))

	return app.Listen(listenAddr)
}