Add test web page

a6b9835c · Thomas Schneider · b6d69e58 · a6b9835c · a6b9835c
Commit a6b9835c authored 3 years ago by Thomas Schneider
--- a/index.html
+++ b/index.html
+<!doctype html>
+<html>
+  <head>
+    <meta charset="utf-8"/>
+    <title>WebSocket image server test client</title>
+    <script>
+      document.addEventListener('DOMContentLoaded', () => {
+	  var img = document.getElementById("jpeg");
+	  var ws = new WebSocket("ws://{{.}}/video");
+	  ws.onmessage = (event) => {
+	      img.src = URL.createObjectURL(event.data);
+	  };
+      });
+    </script>
+  </head>
+  <body>
+    <h1>WebSocket image server test client</h1>
+
+    <img src="" id="jpeg" />
+  </body>
+</html>
--- a/server.go
+++ b/server.go
@@ -2,6 +2,7 @@ package main

 import (
 	"flag"
+	"html/template"
 	"io"
 	"log"
 	"net/http"
@@ -26,6 +27,7 @@ func jpeg(w http.ResponseWriter, r *http.Request) {
 		log.Print("upgrade: ", err)
 		return
 	}
+	defer log.Printf("closing websocket with %v", c.RemoteAddr())
 	defer c.Close()
 	log.Printf("initialised websocket with %v", c.RemoteAddr())

@@ -50,6 +52,18 @@ func main() {
 	}
 	f.Close()

+	index, err := template.ParseFiles("index.html")
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
+		err = index.Execute(w, r.Host)
+		if err != nil {
+			log.Print(err)
+		}
+	})
+
 	http.HandleFunc("/jpeg", jpeg)

 	log.Printf("listening on %s", *listen)