96 lines
1.9 KiB
Go
96 lines
1.9 KiB
Go
package main
|
|
|
|
import (
|
|
"io"
|
|
"net"
|
|
"net/http"
|
|
"strconv"
|
|
|
|
icore "github.com/ipfs/interface-go-ipfs-core"
|
|
"github.com/sirupsen/logrus"
|
|
log "github.com/sirupsen/logrus"
|
|
"github.com/yutopp/go-rtmp"
|
|
)
|
|
|
|
var ipfs icore.CoreAPI
|
|
var configuration = getConfig()
|
|
|
|
func main() {
|
|
checkConfig(configuration)
|
|
// resetDirectories()
|
|
|
|
var hlsDirectoryPath = configuration.PublicHLSPath
|
|
|
|
log.Println("Starting up. Please wait...")
|
|
|
|
if configuration.IPFS.Enabled {
|
|
hlsDirectoryPath = configuration.PrivateHLSPath
|
|
enableIPFS()
|
|
go monitorVideoContent(hlsDirectoryPath, configuration, &ipfs)
|
|
}
|
|
|
|
go startChatServer()
|
|
|
|
startRTMPService()
|
|
}
|
|
|
|
func enableIPFS() {
|
|
log.Println("Enabling IPFS support...")
|
|
|
|
ipfsInstance, node, _ := createIPFSInstance()
|
|
ipfs = *ipfsInstance
|
|
|
|
createIPFSDirectory(ipfsInstance, "./hls")
|
|
go startIPFSNode(ipfs, node)
|
|
}
|
|
|
|
func startChatServer() {
|
|
// log.SetFlags(log.Lshortfile)
|
|
|
|
// websocket server
|
|
server := NewServer("/entry")
|
|
go server.Listen()
|
|
|
|
// static files
|
|
http.Handle("/", http.FileServer(http.Dir("webroot")))
|
|
|
|
log.Fatal(http.ListenAndServe(":"+strconv.Itoa(configuration.WebServerPort), nil))
|
|
}
|
|
|
|
func startRTMPService() {
|
|
port := 1935
|
|
log.Printf("RTMP server is listening for incoming stream on port %d.\n", port)
|
|
|
|
tcpAddr, err := net.ResolveTCPAddr("tcp", ":"+strconv.Itoa(port))
|
|
if err != nil {
|
|
log.Panicf("Failed: %+v", err)
|
|
}
|
|
|
|
listener, err := net.ListenTCP("tcp", tcpAddr)
|
|
if err != nil {
|
|
log.Panicf("Failed: %+v", err)
|
|
}
|
|
|
|
srv := rtmp.NewServer(&rtmp.ServerConfig{
|
|
OnConnect: func(conn net.Conn) (io.ReadWriteCloser, *rtmp.ConnConfig) {
|
|
l := log.StandardLogger()
|
|
l.SetLevel(logrus.WarnLevel)
|
|
|
|
h := &Handler{}
|
|
|
|
return conn, &rtmp.ConnConfig{
|
|
Handler: h,
|
|
|
|
ControlState: rtmp.StreamControlStateConfig{
|
|
DefaultBandwidthWindowSize: 6 * 1024 * 1024 / 8,
|
|
},
|
|
|
|
Logger: l,
|
|
}
|
|
},
|
|
})
|
|
if err := srv.Serve(listener); err != nil {
|
|
log.Panicf("Failed: %+v", err)
|
|
}
|
|
}
|