Merge pull request #37 from xiaokaixuan/main

Add Env: `LIVE_PREFIX`

Dockerfile

@@ -17,5 +17,6 @@ FROM alpine:3.14
 COPY --from=build /allinone /allinone
 
 EXPOSE 35455
+ENV LIVE_PREFIX=https://www.goodiptv.club
 
 CMD [ "/allinone" ]

Golang/main.go

@@ -17,6 +17,7 @@ import (
 	"github.com/gin-gonic/gin"
 	"net/http"
 	"strconv"
+	"os"
 )
 
 func duanyan(adurl string, realurl any) string {
@@ -29,6 +30,14 @@ func duanyan(adurl string, realurl any) string {
 	return liveurl
 }
 
+func getLivePrefix(c *gin.Context) string {
+	prefix := os.Getenv("LIVE_PREFIX")
+	if len(prefix) > 0 {
+		return prefix
+	}
+	return fmt.Sprintf("http://%s", c.Request.Host)
+}
+
 func setupRouter(adurl string) *gin.Engine {
 	gin.SetMode(gin.ReleaseMode)
 	r := gin.Default()
@@ -63,7 +72,7 @@ func setupRouter(adurl string) *gin.Engine {
 			data := res.VList
 			for _, value := range data {
 				fmt.Fprintf(c.Writer, "#EXTINF:-1 tvg-logo=\"%s\" group-title=\"%s\", %s\n", value.SAvatar180, value.SGameFullName, value.SNick)
-				fmt.Fprintf(c.Writer, "https://www.goodiptv.club/huya/%v\n", value.LProfileRoom)
+				fmt.Fprintf(c.Writer, "%s/huya/%v\n", getLivePrefix(c), value.LProfileRoom)
 			}
 		}
 	})
@@ -94,7 +103,7 @@ func setupRouter(adurl string) *gin.Engine {
 
 			for _, value := range data {
 				fmt.Fprintf(c.Writer, "#EXTINF:-1 tvg-logo=\"https://apic.douyucdn.cn/upload/%s_big.jpg\" group-title=\"%s\", %s\n", value.Av, value.C2name, value.Nn)
-				fmt.Fprintf(c.Writer, "https://www.goodiptv.club/douyu/%v\n", value.Rid)
+				fmt.Fprintf(c.Writer, "%s/douyu/%v\n", getLivePrefix(c), value.Rid)
 			}
 		}
 	})