diff --git a/Dockerfile b/Dockerfile index 032b79a..ce4c322 100644 --- a/Dockerfile +++ b/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" ] diff --git a/Golang/main.go b/Golang/main.go index adebaf5..a345e62 100644 --- a/Golang/main.go +++ b/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) } } })