From 0772566270b55800afc73e790d229443139a3b96 Mon Sep 17 00:00:00 2001
From: Jack Hadrill <jack@hadrill.co.uk>
Date: Tue, 5 Jan 2021 01:59:52 +0000
Subject: [PATCH] Add video output to host view

---
 src/components/HostWires.vue |  2 +-
 src/views/Host.vue           | 70 +++++++++++++++++++++++-------------
 2 files changed, 46 insertions(+), 26 deletions(-)

diff --git a/src/components/HostWires.vue b/src/components/HostWires.vue
index 76e02c0..64c2e45 100644
--- a/src/components/HostWires.vue
+++ b/src/components/HostWires.vue
@@ -4,7 +4,7 @@
       Wire statistics
     </template>
     <template #content>
-      <DataTable :value="wires" autoLayout=true>
+      <DataTable :value="wires" :autoLayout="true">
         <Column field="peerId" header="Wire ID"></Column>
         <Column field="remoteAddress" header="Remote address"></Column>
         <Column field="uploadSpeed" header="Upload speed"></Column>
diff --git a/src/views/Host.vue b/src/views/Host.vue
index 8f66510..083c6b9 100644
--- a/src/views/Host.vue
+++ b/src/views/Host.vue
@@ -1,23 +1,36 @@
 <template>
   <div class="p-p-6">
     <h1 class="p-my-0">Host</h1>
-    <FileSelect v-if="!state.active" message="Drag a H.264 encoded MP4 here to start a screen." @selected="filesSelected" />
-    <div v-else>
-      <Card>
-        <template #title>
-          Info hash
-        </template>
+    <template v-if="!state.active">
+      <FileSelect class="p-mb-4" v-if="!state.fileSelected" message="Drag a H.264 encoded MP4 here to start a screen." @selected="filesSelected" />
+      <Card class="p-mb-4" v-else>
         <template #content>
-          <InputText id="infoHash" type="text" v-model="state.infoHash" disabled />
+          <div class="p-text-center">
+            <ProgressSpinner />
+            <h2>Generating hash. Please wait...</h2>
+          </div>
         </template>
       </Card>
-      <HostWires class="p-mt-4" :wires="wires" />
-    </div>
+    </template>
+    <Card class="p-mb-4" v-show="state.active">
+      <template #content>
+        <video id="player" controls="true"></video>
+      </template>
+    </Card>
+    <Card class="p-mb-4" v-show="state.active">
+      <template #title>
+        Info hash
+      </template>
+      <template #content>
+        <InputText id="infoHash" type="text" v-model="state.infoHash" disabled />
+      </template>
+    </Card>
+    <HostWires class="p-mb-4" :wires="wires" v-show="state.active" />
   </div>
 </template>
 
 <script>
-import { inject, onBeforeUnmount, onMounted, reactive } from 'vue'
+import { inject, onBeforeUnmount, reactive } from 'vue'
 import { useToast } from 'primevue/usetoast'
 import prettyBytes from 'pretty-bytes'
 import sha1 from 'crypto-js/sha1'
@@ -29,52 +42,59 @@ export default {
     const trackers = inject('trackers')
     const state = reactive({
       active: false,
+      fileSelected: false,
       infoHash: '',
       size: 0,
       uploaded: 0,
       uploadSpeed: 0,
       wires: []
     })
+    const webTorrent = new WebTorrent()
     const wires = reactive([])
-    var webTorrent
-
-    onMounted(() => {
-      webTorrent = new WebTorrent()
-    })
 
     onBeforeUnmount(() => {
       webTorrent.destroy()
     })
 
-    const filesSelected = (files) => {
-      if (files?.length !== 1 || !files[0].name.endsWith('.mp4') || files[0].type !== 'video/mp4') {
+    const filesSelected = (f) => {
+      if (f?.length !== 1 || !f[0].name.endsWith('.mp4') || f[0].type !== 'video/mp4') {
         toast.add({ severity: 'error', summary: 'Bad file input', detail: 'You must select a single H.264 encoded MP4.', life: 3000 })
-        return
+      } else {
+        state.fileSelected = true
       }
 
-      seedFile(files[0])
+      seedFile(f[0])
     }
 
     const seedFile = (file) => {
       webTorrent.seed(file, { announce: trackers }, (torrent) => {
-        toast.add({ severity: 'success', summary: 'File added', detail: `You are now sharing ${file.name}`, life: 5000 })
+        var file = torrent.files.find(file => {
+          return file.name.endsWith('.mp4')
+        })
+
+        state.size = file.size
         state.active = true
         state.infoHash = torrent.infoHash
-        state.size = file.size
 
-        torrent.on('upload', (bytes) => {
+        file.createReadStream().on('data', () => {
+          file.renderTo('#player')
+        })
+
+        torrent.on('upload', bytes => {
           state.uploaded += bytes
         })
 
-        torrent.on('wire', (wire) => {
+        torrent.on('wire', wire => {
           state.wires.push(wire)
           toast.add({ severity: 'info', summary: 'New watcher', detail: 'Someone has joined your screen.', life: 3000 })
         })
+
+        toast.add({ severity: 'success', summary: 'File added', detail: `You are now sharing ${file.name}`, life: 5000 })
       })
     }
 
     setInterval(() => {
-      const newWires = state.wires.map((wire) => ({
+      const newWires = state.wires.map(wire => ({
         peerId: sha1(wire.peerId).toString().substring(0, 100),
         remoteAddress: wire.remoteAddress,
         uploadSpeed: prettyBytes(wire.uploadSpeed()) + 'ps',
@@ -95,7 +115,7 @@ export default {
 </script>
 
 <style>
-#infoHash {
+#infoHash, video#player {
   width: 100%;
 }
 </style>