From e640450d988c5ddcad951612f79da6aa732a8328 Mon Sep 17 00:00:00 2001
From: Yohann D'ANELLO <yohann.danello@gmail.com>
Date: Mon, 12 Oct 2020 23:39:26 +0200
Subject: [PATCH] Start telnet connection to send ASCII Art stream

---
 stream/telnet/telnet.go | 68 +++++++++++++++++++++++++++++++++++++++++
 stream/webrtc/ingest.go | 28 ++---------------
 2 files changed, 70 insertions(+), 26 deletions(-)
 create mode 100644 stream/telnet/telnet.go

diff --git a/stream/telnet/telnet.go b/stream/telnet/telnet.go
new file mode 100644
index 0000000..ac58aac
--- /dev/null
+++ b/stream/telnet/telnet.go
@@ -0,0 +1,68 @@
+// Package telnet provides some fancy tools, like an ASCII-art stream.
+package telnet
+
+import (
+	"io"
+	"log"
+	"net"
+	"time"
+)
+
+func asciiChar(pixel byte) string {
+	asciiChars := []string{"@", "#", "$", "%", "?", "*", "+", ";", ":", ",", "."}
+	return asciiChars[pixel/25]
+}
+
+// ServeAsciiArt starts a telnet server that send all packets as ASCII Art
+func ServeAsciiArt(reader io.Reader) {
+	listener, err := net.Listen("tcp", ":4242")
+	if err != nil {
+		log.Printf("Error while listening to the port 4242: %s", err)
+		return
+	}
+
+	currentMessage := ""
+
+	go func() {
+		for {
+			s, err := listener.Accept()
+			if err != nil {
+				log.Printf("Error while accepting TCP socket: %s", s)
+				continue
+			}
+			go func(s net.Conn) {
+				for {
+					n, err := s.Write([]byte(currentMessage))
+					if err != nil {
+						log.Printf("Error while sending TCP data: %s", err)
+						_ = s.Close()
+						break
+					}
+					if n == 0 {
+						_ = s.Close()
+						break
+					}
+					time.Sleep(50 * time.Millisecond)
+				}
+			}(s)
+		}
+	}()
+
+	buff := make([]byte, 2048)
+	header := "\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n"
+	for {
+		n, _ := reader.Read(buff)
+		if n == 0 {
+			break
+		}
+		imageStr := ""
+		for j := 0; j < 18; j++ {
+			for i := 0; i < 32; i++ {
+				pixel := buff[32*j+i]
+				imageStr += asciiChar(pixel) + asciiChar(pixel)
+			}
+			imageStr += "\n"
+		}
+		currentMessage = header + imageStr
+	}
+}
diff --git a/stream/webrtc/ingest.go b/stream/webrtc/ingest.go
index 9ebe430..3ba4cb5 100644
--- a/stream/webrtc/ingest.go
+++ b/stream/webrtc/ingest.go
@@ -3,7 +3,7 @@ package webrtc
 
 import (
 	"bufio"
-	"fmt"
+	"gitlab.crans.org/nounous/ghostream/stream/telnet"
 	"io"
 	"log"
 	"net"
@@ -106,7 +106,7 @@ func ingestFrom(inputChannel chan srt.Packet) {
 			}()
 
 			// Receive ascii
-			go asciiArt(output, videoTracks[srtPacket.StreamName])
+			go telnet.ServeAsciiArt(output)
 
 			// Receive audio
 			go func() {
@@ -165,27 +165,3 @@ func ingestFrom(inputChannel chan srt.Packet) {
 		}
 	}
 }
-
-func asciiChar(pixel byte) string {
-	asciiChars := []string{"@", "#", "$", "%", "?", "*", "+", ";", ":", ",", "."}
-	return asciiChars[pixel/25]
-}
-
-func asciiArt(reader io.Reader, videoTracks []*webrtc.Track) {
-
-	buff := make([]byte, 2048)
-	for {
-		n, _ := reader.Read(buff)
-		if n == 0 {
-			break
-		}
-		for j := 0; j < 18; j++ {
-			for i := 0; i < 32; i++ {
-				pixel := buff[32*j+i]
-				fmt.Print(asciiChar(pixel))
-			}
-			fmt.Println()
-		}
-		fmt.Println()
-	}
-}