Attempt with Joy5

core/rtmp/rtmp.go

@@ -4,51 +4,129 @@ import (
 	"fmt"
 	"io"
 	"net"
+	"os"
+	"strings"
+	"syscall"
+	"time"
+
+	"github.com/gabek/owncast/config"
+	"github.com/gabek/owncast/core"
+	"github.com/gabek/owncast/core/ffmpeg"
+	"github.com/gabek/owncast/utils"
+
+	"github.com/nareix/joy5/format/flv"
+	"github.com/nareix/joy5/format/rtmp"
 
 	log "github.com/sirupsen/logrus"
-	yutmp "github.com/yutopp/go-rtmp"
 )
 
 var (
 	//IsConnected whether there is a connection or not
 	_isConnected = false
+	pipePath     = utils.GetTemporaryPipePath()
+	filePipe     *os.File
 )
 
 //Start starts the rtmp service, listening on port 1935
 func Start() {
 	port := 1935
 
-	tcpAddr, err := net.ResolveTCPAddr("tcp", fmt.Sprintf(":%d", port))
-	if err != nil {
-		log.Panicf("Failed to resolve the tcp address for the rtmp service: %+v", err)
+	server := rtmp.NewServer()
+
+	server.LogEvent = func(conn *rtmp.Conn, nc net.Conn, e int) {
+		log.Errorln("RTMP status:", rtmp.EventString[e])
 	}
 
-	listener, err := net.ListenTCP("tcp", tcpAddr)
-	if err != nil {
-		log.Panicf("Failed to acquire the tcp listener: %+v", err)
+	server.OnNewConn = func(conn *rtmp.Conn) {
+		log.Println("OnNewConn!", conn.FlashVer)
 	}
 
-	srv := yutmp.NewServer(&yutmp.ServerConfig{
-		OnConnect: func(conn net.Conn) (io.ReadWriteCloser, *yutmp.ConnConfig) {
-			l := log.StandardLogger()
-			l.SetLevel(log.WarnLevel)
+	server.HandleConn = func(conn *rtmp.Conn, nc net.Conn) {
+		if _isConnected {
+			log.Errorln("stream already running; can not overtake an existing stream")
+			nc.Close()
+		}
 
-			return conn, &yutmp.ConnConfig{
-				Handler: &Handler{},
+		streamingKeyComponents := strings.Split(conn.URL.Path, "/")
+		streamingKey := streamingKeyComponents[len(streamingKeyComponents)-1]
 
-				ControlState: yutmp.StreamControlStateConfig{
-					DefaultBandwidthWindowSize: 6 * 1024 * 1024 / 8,
-				},
+		if streamingKey != config.Config.VideoSettings.StreamingKey {
+			log.Errorln("invalid streaming key; rejecting incoming stream")
+			nc.Close()
+			return
+		}
 
-				Logger: l,
+		// Record streams as FLV
+		syscall.Mkfifo(pipePath, 0666)
+		file, err := os.OpenFile(pipePath, os.O_RDWR, os.ModeNamedPipe)
+		if err != nil {
+			log.Panicln(err)
+		}
+
+		filePipe = file
+		fmt.Println(pipePath)
+
+		muxer := flv.NewMuxer(filePipe)
+
+		_isConnected = true
+		core.SetStreamAsConnected()
+
+		defer filePipe.Close()
+
+		if err != nil {
+			panic(err)
+		}
+
+		defer nc.Close()
+
+		transcoder := ffmpeg.NewTranscoder()
+		go transcoder.Start()
+
+		for {
+			pkt, err := conn.ReadPacket()
+			if err != nil {
+				log.Errorln(err)
+
+				if err == io.EOF {
+					_isConnected = false
+					core.SetStreamAsDisconnected()
+					break
+				}
+
+				return
 			}
-		},
-	})
 
-	log.Infof("RTMP server is listening for incoming stream on port: %d", port)
-	if err := srv.Serve(listener); err != nil {
-		log.Panicf("Failed to serve the rtmp service: %+v", err)
+			// err = muxer.WriteFileHeader()
+			// if err != nil {
+			// 	log.Errorln(err)
+			// }
+
+			err = muxer.WritePacket(pkt)
+			if err != nil {
+				log.Errorln(err)
+			}
+
+		}
 	}
+
+	var lis net.Listener
+	var err error
+	if lis, err = net.Listen("tcp", fmt.Sprintf(":%d", port)); err != nil {
+		return
+	}
+	log.Printf("RTMP server is listening for incoming stream on port: %d", port)
+
+	go func() {
+		for {
+			nc, err := lis.Accept()
+			if err != nil {
+				time.Sleep(time.Second)
+				continue
+			}
+			go server.HandleNetConn(nc)
+		}
+	}()
+
 }
 
 //IsConnected gets whether there is an rtmp connection or not