mirror of
https://codeberg.org/SafeTwitch/safetwitch-backend.git
synced 2024-12-22 21:23:01 -05:00
79 lines
1.7 KiB
Go
79 lines
1.7 KiB
Go
package proxy
|
|
|
|
import (
|
|
b64 "encoding/base64"
|
|
"io"
|
|
"net/http"
|
|
"safetwitch-backend/extractor/twitch"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func Routes(route *gin.Engine) {
|
|
auth := route.Group("/proxy")
|
|
|
|
auth.GET("/img/:url", func(context *gin.Context) {
|
|
decodedUrl, err := b64.StdEncoding.DecodeString(context.Param("url"))
|
|
if err != nil {
|
|
context.Error(err)
|
|
}
|
|
|
|
imageResp, err := http.Get(string(decodedUrl))
|
|
if err != nil {
|
|
context.Error(err)
|
|
}
|
|
body, err := io.ReadAll(imageResp.Body)
|
|
if err != nil {
|
|
context.Error(err)
|
|
}
|
|
|
|
contentType := imageResp.Header.Get("Content-Type")
|
|
|
|
context.Data(200, contentType, body)
|
|
context.JSON(imageResp.StatusCode, imageResp.Body)
|
|
})
|
|
|
|
auth.GET("/stream/:username/hls.m3u8", func(context *gin.Context) {
|
|
streamer := context.Param("username")
|
|
playlistFile, err := twitch.GetStream(streamer)
|
|
if err != nil {
|
|
context.Error(err)
|
|
}
|
|
|
|
context.Data(200, "application/text", []byte(playlistFile))
|
|
})
|
|
|
|
auth.GET("/stream/sub/:encodedUrl", func(context *gin.Context) {
|
|
decodedUrl, err := b64.StdEncoding.DecodeString(context.Param("encodedUrl"))
|
|
if err != nil {
|
|
context.Error(err)
|
|
}
|
|
|
|
playlistFile, err := twitch.GetSubPlaylist(string(decodedUrl))
|
|
if err != nil {
|
|
context.Error(err)
|
|
}
|
|
|
|
context.Data(200, "application/text", []byte(playlistFile))
|
|
})
|
|
|
|
auth.GET("/stream/segment/:encodedUrl", func(context *gin.Context) {
|
|
decodedUrl, err := b64.StdEncoding.DecodeString(context.Param("encodedUrl"))
|
|
if err != nil {
|
|
context.Error(err)
|
|
}
|
|
|
|
segmentData, err := http.Get(string(decodedUrl))
|
|
if err != nil {
|
|
context.Error(err)
|
|
return
|
|
}
|
|
|
|
segment, err := io.ReadAll(segmentData.Body)
|
|
if err != nil {
|
|
context.Error(err)
|
|
}
|
|
|
|
context.Data(200, "fuck", segment)
|
|
})
|
|
}
|