Initial commit

2026-05-20 16:13:58 +08:00
commit ceffbe382e
500 changed files with 157469 additions and 0 deletions
@@ -0,0 +1,135 @@
+package m3u8
+
+import (
+	"bytes"
+	"fmt"
+	"io"
+	"stat/common/enum/redisconst"
+	"stat/common/errors"
+	"stat/common/http"
+	"stat/common/log"
+	"stat/common/redis"
+	"stat/common/utils"
+	"strings"
+
+	"github.com/grafov/m3u8"
+)
+
+const (
+	keyUri  = "/api/stat/media/enkey"
+	AuthKey = "fT5xSg4hltHpzVy6aV9rVECDJ1J1pN"
+)
+
+var headerMediaPlayList *m3u8.MediaPlaylist
+
+// TransFormWithSafely 被检查权限的播放列表
+// 主要检查用户是否有观看该视频的权限
+// apiType(1:app,2:h5app,3:web)
+func TransFormWithSafely(req *Req, cdnDomain, origDomain string, needHeader bool, apiType int) (*m3u8.MediaPlaylist, *bytes.Buffer, error) {
+	if req.Name == "" {
+		return nil, nil, errors.MakeWithMsg(errors.M3u8FileName, "file name is empty")
+	}
+	if cdnDomain == "" || origDomain == "" {
+		return nil, nil, errors.MakeWithMsg(errors.DomainErr, "cdnDomain or origDomain is empty")
+	}
+
+	return transform(req.Name, cdnDomain, origDomain, needHeader, apiType)
+}
+func AppendsMediaS3Version(name string) string {
+	// 默认使用V1
+	if name != "" && !strings.HasPrefix(name, "v2/") && !strings.HasPrefix(name, "v1/") && !strings.HasPrefix(name, "v3/") &&
+		!strings.HasPrefix(name, "/v2") && !strings.HasPrefix(name, "/v1") && !strings.HasPrefix(name, "/v3") {
+		return "/v1"
+	}
+	return ""
+}
+
+// transform m3u8文件转换，将不正确的m3u8文件，替换为正确的m3u8文件
+func transform(name string, cdnDomain, origDomain string, needHeader bool, apiType int) (media *m3u8.MediaPlaylist, buff *bytes.Buffer, err error) {
+	m3u8reader, err := readFromRedis(name, origDomain)
+	if err != nil {
+		return
+	}
+	p, t, err := m3u8.DecodeFrom(m3u8reader, false)
+	if err != nil {
+		err = errors.MakeWithErr(errors.M3u8Read, err)
+		return
+	}
+	tmptKeyurl := keyUri
+
+	fmt.Println("<<<<<<<<<<<<<<<<<<<start get media list>>>>>>>>>>>>>>>")
+	switch t {
+	case m3u8.MEDIA:
+		pl := p.(*m3u8.MediaPlaylist)
+		dirPath := utils.PathForInfo(name).Dir
+		media, err = newMediaPlayList(pl, cdnDomain, origDomain, dirPath, tmptKeyurl, needHeader)
+		if err != nil {
+			return
+		}
+		buff = media.Encode()
+	case m3u8.MASTER:
+		mpl := p.(*m3u8.MasterPlaylist)
+		buff = mpl.Encode()
+	default:
+		err = errors.MakeWithMsg(errors.M3u8FileName, fmt.Sprintf("invalid m3u8 type %d", t))
+	}
+	return
+}
+
+//func readFromCDN(name, domain string) (reader io.Reader, err error) {
+//	requestUrl := utils.JoinUrlPath(domain, name)
+//	resp, err := http.GetLargeFile(requestUrl, nil)
+//	if err != nil {
+//		return
+//	}
+//	if resp.StatusCode != 200 {
+//		msg := fmt.Sprintf("read m3u8 fail, url: %s, status code %d", requestUrl, resp.StatusCode)
+//		err = errors.MakeWithMsg(errors.M3u8Read, msg)
+//		return
+//	}
+//	reader = bytes.NewReader(resp.Data)
+//	return
+//}
+
+// 从redis缓存中读取
+func readFromRedis(name, domain string) (reader io.Reader, err error) {
+	m3u8Key := redisconst.MediaM3U8Key(name)
+	data, err := redis.StatRds.GetBytes(m3u8Key)
+	if err == nil && len(data) > 0 {
+		reader = bytes.NewReader(data)
+		return
+	}
+	requestUrl := utils.JoinUrlPath(domain, name)
+	resp, err := http.GetLargeFile(requestUrl, nil)
+	if err != nil {
+		//fmt.Println("requestUrl111", requestUrl)
+		return
+	}
+
+	if resp.StatusCode != 200 {
+		msg := fmt.Sprintf("read m3u8 fail, url: %s, status code %d", requestUrl, resp.StatusCode)
+		//fmt.Println(msg)
+		err = errors.MakeWithMsg(errors.M3u8Read, msg)
+		return
+	}
+	if err := redis.StatRds.Set(m3u8Key, resp.Data,
+		redisconst.MediaM3U8CacheExpire); err != nil {
+		log.Info("m3u8:readFromRedis redis set err", log.E(err))
+	}
+	reader = bytes.NewReader(resp.Data)
+	return
+}
+
+func GetHeaderMediaSegUrls() []string {
+	urls := make([]string, 0)
+	if headerMediaPlayList == nil {
+		return urls
+	}
+	for _, segment := range headerMediaPlayList.Segments {
+		if segment == nil {
+			continue
+		}
+		urls = append(urls, segment.URI)
+	}
+	return urls
+}