mirror of https://github.com/q191201771/lal.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
97 lines
2.4 KiB
Go
97 lines
2.4 KiB
Go
// Copyright 2020, Chef. All rights reserved.
|
|
// https://github.com/q191201771/lal
|
|
//
|
|
// Use of this source code is governed by a MIT-style license
|
|
// that can be found in the License file.
|
|
//
|
|
// Author: Chef (191201771@qq.com)
|
|
|
|
package hls
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/q191201771/lal/pkg/base"
|
|
|
|
"github.com/q191201771/naza/pkg/nazalog"
|
|
)
|
|
|
|
type ServerHandler struct {
|
|
outPath string
|
|
//addr string
|
|
//ln net.Listener
|
|
//httpSrv *http.Server
|
|
}
|
|
|
|
func NewServerHandler(outPath string) *ServerHandler {
|
|
return &ServerHandler{
|
|
outPath: outPath,
|
|
}
|
|
}
|
|
|
|
//
|
|
//func (s *Server) Listen() (err error) {
|
|
// if s.ln, err = net.Listen("tcp", s.addr); err != nil {
|
|
// return
|
|
// }
|
|
// s.httpSrv = &http.Server{Addr: s.addr, Handler: s}
|
|
// nazalog.Infof("start hls server listen. addr=%s", s.addr)
|
|
// return
|
|
//}
|
|
//
|
|
//func (s *Server) RunLoop() error {
|
|
// return s.httpSrv.Serve(s.ln)
|
|
//}
|
|
//
|
|
//func (s *Server) Dispose() {
|
|
// if err := s.httpSrv.Close(); err != nil {
|
|
// nazalog.Error(err)
|
|
// }
|
|
//}
|
|
|
|
func (s *ServerHandler) ServeHTTP(resp http.ResponseWriter, req *http.Request) {
|
|
//nazalog.Debugf("%+v", req)
|
|
|
|
// TODO chef:
|
|
// - check appname in URI path
|
|
|
|
ri := PathStrategy.GetRequestInfo(req.RequestURI, s.outPath)
|
|
//nazalog.Debugf("%+v", ri)
|
|
|
|
if ri.FileName == "" || ri.StreamName == "" || ri.FileNameWithPath == "" || (ri.FileType != "m3u8" && ri.FileType != "ts") {
|
|
nazalog.Warnf("invalid hls request. uri=%s, request=%+v", req.RequestURI, ri)
|
|
resp.WriteHeader(404)
|
|
return
|
|
}
|
|
|
|
content, err := ReadFile(ri.FileNameWithPath)
|
|
if err != nil {
|
|
nazalog.Warnf("read hls file failed. request=%+v, err=%+v", ri, err)
|
|
resp.WriteHeader(404)
|
|
return
|
|
}
|
|
|
|
switch ri.FileType {
|
|
case "m3u8":
|
|
resp.Header().Add("Content-Type", "application/x-mpegurl")
|
|
resp.Header().Add("Server", base.LalHlsM3u8Server)
|
|
case "ts":
|
|
resp.Header().Add("Content-Type", "video/mp2t")
|
|
resp.Header().Add("Server", base.LalHlsTsServer)
|
|
}
|
|
resp.Header().Add("Cache-Control", "no-cache")
|
|
resp.Header().Add("Access-Control-Allow-Origin", "*")
|
|
|
|
_, _ = resp.Write(content)
|
|
return
|
|
}
|
|
|
|
// m3u8文件用这个也行
|
|
//resp.Header().Add("Content-Type", "application/vnd.apple.mpegurl")
|
|
|
|
//resp.Header().Add("Access-Control-Allow-Origin", "*")
|
|
//resp.Header().Add("Access-Control-Allow-Credentials", "true")
|
|
//resp.Header().Add("Access-Control-Allow-Methods", "*")
|
|
//resp.Header().Add("Access-Control-Allow-Headers", "Content-Type,Access-Token")
|
|
//resp.Header().Add("Access-Control-Allow-Expose-Headers", "*")
|