Infrastructure/safetwitch-backend
0
Fork 0
mirror of https://codeberg.org/SafeTwitch/safetwitch-backend.git synced 2025-01-21 20:02:32 -05:00
Code Issues Activity

URL encoding and catch requests other than 200 for clips

Browse source
This commit is contained in:
dragongoose 2023-08-25 19:43:18 -04:00
parent 283d5f21e0
commit 1e4f4df6eb
No known key found for this signature in database
GPG key ID: 01397EEC371CDAA5
1 changed files with 14 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
routes/proxy/proxy.go
View file

@ -2,6 +2,7 @@ package proxy
import ( import (
b64 "encoding/base64" b64 "encoding/base64"
"errors"
"io" "io"
"net/http" "net/http"
"safetwitch-backend/extractor/twitch" "safetwitch-backend/extractor/twitch"
@ -14,7 +15,7 @@ func Routes(route *gin.Engine) {
auth := route.Group("/proxy") auth := route.Group("/proxy")
auth.GET("/img/:url", func(context *gin.Context) { auth.GET("/img/:url", func(context *gin.Context) {
decodedUrl, err := b64.StdEncoding.DecodeString(context.Param("url")) decodedUrl, err := b64.URLEncoding.DecodeString(context.Param("url"))
if err != nil { if err != nil {
context.Error(err) context.Error(err)
return return
@ -49,7 +50,7 @@ func Routes(route *gin.Engine) {
}) })
auth.GET("/stream/sub/:encodedUrl", func(context *gin.Context) { auth.GET("/stream/sub/:encodedUrl", func(context *gin.Context) {
decodedUrl, err := b64.StdEncoding.DecodeString(context.Param("encodedUrl")) decodedUrl, err := b64.URLEncoding.DecodeString(context.Param("encodedUrl"))
if err != nil { if err != nil {
context.Error(err) context.Error(err)
return return
@ -65,7 +66,7 @@ func Routes(route *gin.Engine) {
}) })
auth.GET("/stream/segment/:encodedUrl", func(context *gin.Context) { auth.GET("/stream/segment/:encodedUrl", func(context *gin.Context) {
decodedUrl, err := b64.StdEncoding.DecodeString(context.Param("encodedUrl")) decodedUrl, err := b64.URLEncoding.DecodeString(context.Param("encodedUrl"))
if err != nil { if err != nil {
context.Error(err) context.Error(err)
return return
@ -98,7 +99,7 @@ func Routes(route *gin.Engine) {
}) })
auth.GET("/vod/sub/:encodedUrl/video.m3u8", func(context *gin.Context) { auth.GET("/vod/sub/:encodedUrl/video.m3u8", func(context *gin.Context) {
decodedUrl, err := b64.StdEncoding.DecodeString(context.Param("encodedUrl")) decodedUrl, err := b64.URLEncoding.DecodeString(context.Param("encodedUrl"))
if err != nil { if err != nil {
context.Error(err) context.Error(err)
return return
@ -115,7 +116,7 @@ func Routes(route *gin.Engine) {
auth.GET("/vod/sub/:encodedUrl/:segment", func(context *gin.Context) { auth.GET("/vod/sub/:encodedUrl/:segment", func(context *gin.Context) {
decodedUrl, err := b64.StdEncoding.DecodeString(context.Param("encodedUrl")) decodedUrl, err := b64.URLEncoding.DecodeString(context.Param("encodedUrl"))
if err != nil { if err != nil {
context.Error(err) context.Error(err)
return return
@ -141,7 +142,7 @@ func Routes(route *gin.Engine) {
}) })
auth.GET("/clip/:clipUrl", func(context *gin.Context) { auth.GET("/clip/:clipUrl", func(context *gin.Context) {
decodedUrl, err := b64.StdEncoding.DecodeString(context.Param("clipUrl")) decodedUrl, err := b64.URLEncoding.DecodeString(context.Param("clipUrl"))
if err != nil { if err != nil {
context.Error(err) context.Error(err)
return return
@ -152,6 +153,13 @@ func Routes(route *gin.Engine) {
context.Error(err) context.Error(err)
return return
} }
if resp.StatusCode != 200 {
err := errors.New("status was not 200")
context.Error(err)
return
}
defer resp.Body.Close() defer resp.Body.Close()
_, err = io.Copy(context.Writer, resp.Body) _, err = io.Copy(context.Writer, resp.Body)