package main

import (
	"github.com/yangjiechina/live-server/flv"
	"github.com/yangjiechina/live-server/hls"
	"github.com/yangjiechina/live-server/log"
	"github.com/yangjiechina/live-server/rtc"
	"github.com/yangjiechina/live-server/rtsp"
	"go.uber.org/zap/zapcore"
	"net"
	"net/http"

	_ "net/http/pprof"

	"github.com/yangjiechina/avformat/librtmp"
	"github.com/yangjiechina/avformat/utils"
	"github.com/yangjiechina/live-server/rtmp"
	"github.com/yangjiechina/live-server/stream"
)

var rtspAddr *net.TCPAddr

func CreateTransStream(source stream.ISource, protocol stream.Protocol, streams []utils.AVStream) stream.ITransStream {
	if stream.ProtocolRtmp == protocol {
		return rtmp.NewTransStream(librtmp.ChunkSize)
	} else if stream.ProtocolHls == protocol {
		id := source.Id()

		transStream, err := hls.NewTransStream("", stream.AppConfig.Hls.M3U8Format(id), stream.AppConfig.Hls.TSFormat(id, "%d"), stream.AppConfig.Hls.Dir, stream.AppConfig.Hls.Duration, stream.AppConfig.Hls.PlaylistLength)
		if err != nil {
			panic(err)
		}

		return transStream
	} else if stream.ProtocolFlv == protocol {
		return flv.NewHttpTransStream()
	} else if stream.ProtocolRtsp == protocol {
		trackFormat := source.Id() + "?track=%d"
		return rtsp.NewTransStream(net.IPAddr{
			IP:   rtspAddr.IP,
			Zone: rtspAddr.Zone,
		}, trackFormat)
	} else if stream.ProtocolRtc == protocol {
		return rtc.NewTransStream()
	}

	return nil
}

func init() {
	stream.TransStreamFactory = CreateTransStream
}

func main() {
	//初始化日志
	log.InitLogger(zapcore.DebugLevel, "./logs/lkm.log", 10, 100, 7, false)

	stream.AppConfig.GOPCache = true
	stream.AppConfig.MergeWriteLatency = 350

	stream.AppConfig.Hls.Enable = true
	stream.AppConfig.Hls.Dir = "../tmp"
	stream.AppConfig.Hls.Duration = 2
	stream.AppConfig.Hls.PlaylistLength = 10

	rtmpAddr, err := net.ResolveTCPAddr("tcp", "0.0.0.0:1935")
	if err != nil {
		panic(err)
	}

	impl := rtmp.NewServer()
	err = impl.Start(rtmpAddr)
	if err != nil {
		panic(err)
	}

	println("启动rtmp服务成功:" + rtmpAddr.String())

	rtspAddr, err = net.ResolveTCPAddr("tcp", "0.0.0.0:554")
	if err != nil {
		panic(rtspAddr)
	}

	rtspServer := rtsp.NewServer()
	err = rtspServer.Start(rtspAddr)
	if err != nil {
		panic(err)
	}

	println("启动rtsp服务成功:" + rtspAddr.String())

	apiAddr := "0.0.0.0:8080"
	go startApiServer(apiAddr)

	loadConfigError := http.ListenAndServe(":19999", nil)
	if loadConfigError != nil {
		panic(loadConfigError)
	}
	select {}
}