Add endpoint for returning stream status

handler.go

@@ -38,7 +38,6 @@ func (h *Handler) OnCreateStream(timestamp uint32, cmd *rtmpmsg.NetConnectionCre
 
 func (h *Handler) OnPublish(timestamp uint32, cmd *rtmpmsg.NetStreamPublish) error {
 	// log.Printf("OnPublish: %#v", cmd)
-
 	log.Println("Incoming stream connected.")
 
 	if cmd.PublishingName != configuration.VideoSettings.StreamingKey {
@@ -68,6 +67,8 @@ func (h *Handler) OnPublish(timestamp uint32, cmd *rtmpmsg.NetStreamPublish) err
 
 	go startFfmpeg(configuration)
 
+	streamConnected()
+
 	return nil
 }
 
@@ -164,4 +165,6 @@ func (h *Handler) OnClose() {
 	if h.flvFile != nil {
 		_ = h.flvFile.Close()
 	}
+
+	streamDisconnected()
 }

main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"encoding/json"
 	"net/http"
 	"strconv"
 
@@ -10,6 +11,7 @@ import (
 
 var ipfs icore.CoreAPI
 var configuration = getConfig()
+var online = false
 
 func main() {
 	checkConfig(configuration)
@@ -49,8 +51,28 @@ func startChatServer() {
 
 	// static files
 	http.Handle("/", http.FileServer(http.Dir("webroot")))
+	http.HandleFunc("/status", getStatus)
 
 	log.Printf("Starting public web server on port %d", configuration.WebServerPort)
 
 	log.Fatal(http.ListenAndServe(":"+strconv.Itoa(configuration.WebServerPort), nil))
 }
+
+func getStatus(w http.ResponseWriter, r *http.Request) {
+	status := Status{
+		Online: online,
+	}
+	json.NewEncoder(w).Encode(status)
+}
+
+type Status struct {
+	Online bool `json:"online"`
+}
+
+func streamConnected() {
+	online = true
+}
+
+func streamDisconnected() {
+	online = false
+}

webroot/index.html

@@ -6,6 +6,7 @@
     href="https://unpkg.com/tailwindcss@^1.0/dist/tailwind.min.css"
     rel="stylesheet"
   />
+  <script src="https://cdn.jsdelivr.net/npm/vue/dist/vue.js"></script>
 
   <script src="https://cdn.jsdelivr.net/npm/hls.js@latest"></script>
 
@@ -15,73 +16,78 @@
   <script data-main="js/main" src="vendor/require.js"></script>
 </head>
 
-<div
-  class="flex"
->
-  <div class="w-4/6">
-    <video id="video" controls style="width: 100%;"></video>
-  </div>
+<div>
+  <div class="flex">
+    <div class="w-4/6">
+      <video id="video" controls style="width: 100%;"></video>
+      <div id="app">
+        {{ streamStatus }}
+      </div>
+    </div>
 
-  <div class="w-2/6">
-    <div
-      id="messages-container"
-      class="bg-white shadow-md rounded px-8 pt-6 pb-8 mb-4"
-      style="height: 60vh; overflow-y: scroll;"
-    >
-      <div data-bind="foreach: messages">
-        <div class="bg-white shadow-md rounded px-8 pt-6 pb-8 mb-4">
-          <div class="flex items-center">
-            <img data-bind="attr:{src: image}" class="w-10 h-10 rounded-full mr-4 border-black-500" style="padding: 5px; background-color:#ececec;" />
+    <div class="w-2/6">
+      <div
+        id="messages-container"
+        class="bg-white shadow-md rounded px-8 pt-6 pb-8 mb-4"
+        style="height: 60vh; overflow-y: scroll;"
+      >
+        <div data-bind="foreach: messages">
+          <div class="bg-white shadow-md rounded px-8 pt-6 pb-8 mb-4">
+            <div class="flex items-center">
+              <img
+                data-bind="attr:{src: image}"
+                class="w-10 h-10 rounded-full mr-4 border-black-500"
+                style="padding: 5px; background-color: #ececec;"
+              />
 
-            <div class="text-sm">
-							<p class="text-700" data-bind="text:author"></p>
-              <p class="text-gray-600" data-bind="text:body"></p>
+              <div class="text-sm">
+                <p class="text-700" data-bind="text:author"></p>
+                <p class="text-gray-600" data-bind="text:body"></p>
+              </div>
             </div>
           </div>
         </div>
-			</div>
-			
-				
-    </div>
-	<form
-      class="bg-white shadow-md rounded px-8 pt-6 pb-8 mb-4"
-      data-bind="with: editingMessage"
-    >
-      <!-- Author -->
-      <label class="control-label" for="inputAuthor">Author</label>
-      <input
-        id="inputAuthor"
-        type="text"
-        data-bind="value: author"
-        class="appearance-none bg-gray-200 text-gray-700 border border-black-500 rounded py-3 px-4 mb-3 leading-tight focus:outline-none focus:bg-white"
-        placeholder="Name"
-      />
+      </div>
+      <form
+        class="bg-white shadow-md rounded px-8 pt-6 pb-8 mb-4"
+        data-bind="with: editingMessage"
+      >
+        <!-- Author -->
+        <label class="control-label" for="inputAuthor">Author</label>
+        <input
+          id="inputAuthor"
+          type="text"
+          data-bind="value: author"
+          class="appearance-none bg-gray-200 text-gray-700 border border-black-500 rounded py-3 px-4 mb-3 leading-tight focus:outline-none focus:bg-white"
+          placeholder="Name"
+        />
 
-      <!-- Body -->
-      <div>
-        <label class="control-label" for="inputBody">Message</label>
-        <div class="controls">
-          <textarea
-            id="inputBody"
-            data-bind="value: body"
-            placeholder="Message"
-            class="appearance-none block w-full bg-gray-200 text-gray-700 border border-black-500 rounded py-3 px-4 mb-3 leading-tight focus:outline-none focus:bg-white"
-          >
-          </textarea>
+        <!-- Body -->
+        <div>
+          <label class="control-label" for="inputBody">Message</label>
+          <div class="controls">
+            <textarea
+              id="inputBody"
+              data-bind="value: body"
+              placeholder="Message"
+              class="appearance-none block w-full bg-gray-200 text-gray-700 border border-black-500 rounded py-3 px-4 mb-3 leading-tight focus:outline-none focus:bg-white"
+            >
+            </textarea>
+          </div>
         </div>
-      </div>
 
-      <div class="control-group">
-        <div class="controls">
-          <a
-            href="#"
-            class="bg-blue-500 hover:bg-blue-700 text-white font-bold py-2 px-4 rounded"
-            data-bind="click: $parent.send.bind($parent)"
-            >Send</a
-          >
+        <div class="control-group">
+          <div class="controls">
+            <a
+              href="#"
+              class="bg-blue-500 hover:bg-blue-700 text-white font-bold py-2 px-4 rounded"
+              data-bind="click: $parent.send.bind($parent)"
+              >Send</a
+            >
+          </div>
         </div>
-      </div>
-    </form>
+      </form>
+    </div>
   </div>
 </div>
 
@@ -116,3 +122,33 @@
     });
   }
 </script>
+
+<script>
+  var app = new Vue({
+    el: "#app",
+    data: {
+      streamStatus: "Hello Vue!",
+    },
+  });
+</script>
+
+<script>
+    setInterval(getStatus, 5000);
+
+  async function getStatus() {
+    let url = "/status";
+    try {
+      let response = await fetch(url);
+      let status = await response.json(); // read response body and parse as JSON
+          app.streamStatus = status.online
+      ? "Stream is online."
+      : "Stream is offline.";
+
+    } catch(e) {
+      app.streamStatus = "Stream server is offline."
+    }
+
+  }
+
+  getStatus();
+</script>