Initial commit

README.md

@@ -0,0 +1 @@
+# rtmp2hls

go.mod

@@ -0,0 +1,8 @@
+module github.com/netroby/go-rtmp-server
+
+go 1.13
+
+require (
+	github.com/grafov/m3u8 v0.11.1
+	github.com/nareix/joy4 v0.0.0-20181022032202-3ddbc8f9d431
+)

go.sum

@@ -0,0 +1,4 @@
+github.com/grafov/m3u8 v0.11.1 h1:igZ7EBIB2IAsPPazKwRKdbhxcoBKO3lO1UY57PZDeNA=
+github.com/grafov/m3u8 v0.11.1/go.mod h1:nqzOkfBiZJENr52zTVd/Dcl03yzphIMbJqkXGu+u080=
+github.com/nareix/joy4 v0.0.0-20181022032202-3ddbc8f9d431 h1:nWhrOsCKdV6bivw03k7MROF2tYzCFGfYBYFrTEHyucs=
+github.com/nareix/joy4 v0.0.0-20181022032202-3ddbc8f9d431/go.mod h1:aFJ1ZwLjvHN4yEzE5Bkz8rD8/d8Vlj3UIuvz2yfET7I=

main.go

@@ -0,0 +1,110 @@
+package main
+
+import (
+	"fmt"
+	"io"
+	"log"
+	"os"
+	"strings"
+
+	"github.com/grafov/m3u8"
+	"github.com/nareix/joy4/av"
+	"github.com/nareix/joy4/format/rtmp"
+	"github.com/nareix/joy4/format/ts"
+)
+
+const addr = ":1935"
+const key = "test"
+const packetsPerSegment = 100
+
+// TODO: replace failln with println / switch to logrus to include stream name in msg
+
+func main() {
+	server := &rtmp.Server{Addr: addr}
+
+	// create hls playlist
+	playlist, err := m3u8.NewMediaPlaylist(5, 10)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	server.HandlePublish = func(conn *rtmp.Conn) {
+		log.Printf("Handling request %s", conn.URL.RequestURI())
+		if conn.URL.Query().Get("key") != key {
+			log.Println("Key mismatch, aborting request")
+			return
+		}
+
+		streamName := strings.ReplaceAll(conn.URL.Path, "/", "")
+		if streamName == "" {
+			log.Println("Invalid stream name")
+			return
+		}
+
+		streams, err := conn.Streams()
+		if err != nil {
+			log.Fatalln(err)
+		}
+
+		i := 0
+		clientConnected := true
+		for clientConnected {
+			// create new segment
+			segmentName := fmt.Sprintf("%s%04d.ts", streamName, i)
+			outFile, err := os.Create(segmentName)
+			if err != nil {
+				log.Fatalln(err)
+			}
+			tsMuxer := ts.NewMuxer(outFile)
+
+			// write header
+			if err := tsMuxer.WriteHeader(streams); err != nil {
+				log.Fatalln(err)
+			}
+			// write some data
+			packetCount := packetsPerSegment
+			for packetCount > 0 {
+				var packet av.Packet
+				if packet, err = conn.ReadPacket(); err != nil {
+					if err == io.EOF {
+						log.Println("Client disconnected")
+						clientConnected = false
+						break
+					}
+					log.Fatalln(err)
+				}
+				if err = tsMuxer.WritePacket(packet); err != nil {
+					log.Fatalln(err)
+				}
+				if packet.IsKeyFrame {
+					fmt.Println("packet is keyframe")
+					packetCount--
+				}
+			}
+			// write trailer
+			if err := tsMuxer.WriteTrailer(); err != nil {
+				log.Fatalln(err)
+			}
+			log.Printf("Successfully wrote segment %s\n", segmentName)
+
+			// update playlist
+			playlist.Append(segmentName, 1.0, "")
+			playlistFile, err := os.Create(fmt.Sprintf("%s.m3u8", streamName))
+			if err != nil {
+				log.Fatalln(err)
+			}
+			playlistFile.Write(playlist.Encode().Bytes())
+			playlistFile.Close()
+
+			// increase counter
+			i++
+		}
+
+		// todo: cleanup old segments
+
+		// cleanup stream: remove playlist and segments
+	}
+
+	log.Printf("Listening on %s", server.Addr)
+	server.ListenAndServe()
+}