synctv/server/handlers/init.go

package handlers

import (
	"fmt"
	"net/http"

	"github.com/gin-gonic/gin"
	log "github.com/sirupsen/logrus"
	"github.com/synctv-org/synctv/internal/conf"
	"github.com/synctv-org/synctv/public"
	"github.com/synctv-org/synctv/room"
	"github.com/synctv-org/synctv/server/middlewares"
	"github.com/synctv-org/synctv/utils"
	rtmps "github.com/zijiren233/livelib/server"
)

func Init(e *gin.Engine, s *rtmps.Server, r *room.Rooms) {
	{
		s.SetParseChannelFunc(func(ReqAppName, ReqChannelName string, IsPublisher bool) (TrueAppName string, TrueChannel string, err error) {
			if IsPublisher {
				channelName, err := AuthRtmpPublish(ReqChannelName)
				if err != nil {
					log.Errorf("rtmp: publish auth to %s error: %v", ReqAppName, err)
					return "", "", err
				}
				if !r.HasRoom(ReqAppName) {
					log.Warnf("rtmp: publish to %s/%s error: %s", ReqAppName, channelName, fmt.Sprintf("room %s not exist", ReqAppName))
					return "", "", fmt.Errorf("room %s not exist", ReqAppName)
				}
				log.Infof("rtmp: publisher login success: %s/%s", ReqAppName, channelName)
				return ReqAppName, channelName, nil
			} else if !conf.Conf.Rtmp.RtmpPlayer {
				log.Warnf("rtmp: dial to %s/%s error: %s", ReqAppName, ReqChannelName, "rtmp player is not enabled")
				return "", "", fmt.Errorf("rtmp: dial to %s/%s error: %s", ReqAppName, ReqChannelName, "rtmp player is not enabled")
			}
			return ReqAppName, ReqChannelName, nil
		})
	}

	{
		web := e.Group("/web")

		web.Use(func(ctx *gin.Context) {
			if ctx.Request.URL.Path == "/web/" {
				ctx.Header("Cache-Control", "no-store")
			} else {
				ctx.Header("Cache-Control", "public, max-age=31536000")
			}
			ctx.Next()
		})

		web.StaticFS("", http.FS(public.Public))
	}

	{
		api := e.Group("/api")

		needAuthApi := api.Group("")
		needAuthApi.Use(middlewares.AuthRoom)

		{
			public := api.Group("/public")

			public.GET("/settings", Settings)
		}

		{
			room := api.Group("/room")
			needAuthRoom := needAuthApi.Group("/room")

			room.GET("/ws", NewWebSocketHandler(utils.NewWebSocketServer()))

			room.GET("/check", CheckRoom)

			room.GET("/user", CheckUser)

			room.GET("/list", RoomList)

			room.POST("/create", NewCreateRoomHandler(s))

			room.POST("/login", LoginRoom)

			needAuthRoom.POST("/delete", DeleteRoom)

			needAuthRoom.POST("/pwd", SetPassword)

			needAuthRoom.PUT("/admin", AddAdmin)

			needAuthRoom.DELETE("/admin", DelAdmin)

			needAuthRoom.GET("/settings", RoomSettings)
		}

		{
			movie := api.Group("/movie")
			needAuthMovie := needAuthApi.Group("/movie")

			needAuthMovie.GET("/list", MovieList)

			needAuthMovie.GET("/current", CurrentMovie)

			needAuthMovie.GET("/movies", Movies)

			needAuthMovie.POST("/current", ChangeCurrentMovie)

			needAuthMovie.POST("/push", PushMovie)

			needAuthMovie.POST("/edit", EditMovie)

			needAuthMovie.POST("/swap", SwapMovie)

			needAuthMovie.POST("/delete", DelMovie)

			needAuthMovie.POST("/clear", ClearMovies)

			movie.HEAD("/proxy/:roomId/:pullKey", ProxyMovie)

			movie.GET("/proxy/:roomId/:pullKey", ProxyMovie)

			{
				live := needAuthMovie.Group("/live")

				live.POST("/publishKey", NewPublishKey)

				live.GET("/*pullKey", JoinLive)
			}
		}

		{
			// user := api.Group("/user")
			needAuthUser := needAuthApi.Group("/user")

			needAuthUser.GET("/me", Me)

			needAuthUser.POST("/pwd", SetUserPassword)
		}
	}

	e.NoRoute(func(c *gin.Context) {
		c.Redirect(http.StatusFound, "/web/")
	})
}