From 516ec0843664fcf1513a490d181830143aec818f Mon Sep 17 00:00:00 2001
From: Mahesh Bandara Wijerathna <m4heshd@gmail.com>
Date: Tue, 21 May 2024 10:37:53 +0530
Subject: [PATCH] [ModPlayVOD, ws_util] Add support for streaming VODs

---
 backend/src/ws_util.rs                   | 29 ++++++++++++++++
 frontend/src/assets/styles/common.scss   |  1 +
 frontend/src/components/BlockVODCard.vue | 15 ++++++++-
 frontend/src/components/ModPlayVOD.vue   | 42 ++++++++++++++++++++++++
 frontend/src/modules/ws-util.js          |  5 +++
 frontend/src/pages/Landing.vue           | 22 ++++++++++++-
 frontend/src/store/modPlayVOD.js         | 20 +++++++++++
 7 files changed, 132 insertions(+), 2 deletions(-)
 create mode 100644 frontend/src/components/ModPlayVOD.vue
 create mode 100644 frontend/src/store/modPlayVOD.js

diff --git a/backend/src/ws_util.rs b/backend/src/ws_util.rs
index b12d6d4..59eb7bd 100644
--- a/backend/src/ws_util.rs
+++ b/backend/src/ws_util.rs
@@ -80,6 +80,8 @@ fn handle_ws_client(socket: &SocketRef) {
 
     socket.on("verify-url", handle_verify_url_event);
 
+    socket.on("get-playable", handle_get_playable_event);
+
     socket.on("download", handle_download_event);
 
     socket.on("cancel-download", handle_cancel_download_event);
@@ -267,6 +269,33 @@ async fn handle_verify_url_event(ack: AckSender, Data(data): Data<JSON>) {
     }
 }
 
+/// Handles the `get-playable` WS event.
+async fn handle_get_playable_event(ack: AckSender, Data(data): Data<JSON>) {
+    if let Ok(url) = serde_json::from_value::<String>(data) {
+        match get_vod_meta(url.as_str()).await {
+            Ok(mut vod) => {
+                if !vod.access {
+                    return send_error(
+                        ack,
+                        "You need a Fight Pass subscription to watch this video",
+                    );
+                }
+
+                match get_vod_stream_url(vod.id).await {
+                    Ok(hls) => {
+                        vod.hls = hls;
+                        ack.send(vod).ok();
+                    }
+                    Err(error) => send_error(ack, error),
+                }
+            }
+            Err(error) => send_error(ack, error),
+        }
+    } else {
+        send_error(ack, "Invalid video play request");
+    }
+}
+
 /// Handles the `download` WS event.
 async fn handle_download_event(ack: AckSender, Data(mut data): Data<JSON>) {
     if let (Ok(mut vod), Some(is_restart)) = (
diff --git a/frontend/src/assets/styles/common.scss b/frontend/src/assets/styles/common.scss
index d39fccf..51054f3 100644
--- a/frontend/src/assets/styles/common.scss
+++ b/frontend/src/assets/styles/common.scss
@@ -23,6 +23,7 @@ body {
   }
 
   & > i {
+    border-radius: 0;
     font-size: 26rem;
   }
 }
diff --git a/frontend/src/components/BlockVODCard.vue b/frontend/src/components/BlockVODCard.vue
index f9f01ab..ed0e8ed 100644
--- a/frontend/src/components/BlockVODCard.vue
+++ b/frontend/src/components/BlockVODCard.vue
@@ -13,6 +13,14 @@
 
         <div class="block-vod-card__details__thumbnail__actions">
           <div class="center-content block-vod-card__details__thumbnail__actions__buttons">
+            <button
+                class="circle fill medium"
+                title="Play video"
+                :disabled="vBusyState"
+                @click="$emit('play', vVODData.id)"
+            >
+              <i>play_arrow</i>
+            </button>
             <button
                 class="circle fill medium"
                 title="Download"
@@ -23,7 +31,7 @@
             </button>
             <button
                 class="circle fill medium"
-                title="Get available formats"
+                title="View available formats"
                 :disabled="vBusyState"
                 @click="$emit('getFormats', vVODData.id)"
             >
@@ -83,6 +91,7 @@ const props = defineProps({
 
 // Emits
 defineEmits([
+  'play',
   'download',
   'getFormats',
   'openExternal'
@@ -176,6 +185,10 @@ onMounted(() => nextTick(loadThumbnailImg));
           height: 55%;
           transform: translateY(20rem);
           transition: transform .2s ease-in-out;
+
+          & > button {
+            margin: 0 6rem;
+          }
         }
 
         &__bg {
diff --git a/frontend/src/components/ModPlayVOD.vue b/frontend/src/components/ModPlayVOD.vue
new file mode 100644
index 0000000..44c0177
--- /dev/null
+++ b/frontend/src/components/ModPlayVOD.vue
@@ -0,0 +1,42 @@
+<template>
+  <VModal
+      class="mod-play-vod"
+      v-model="modPlayVOD.active"
+      vIcon="live_tv"
+      :vTitle="modPlayVOD.VOD.title"
+  >
+    <template #content>
+      <VMediaPlayer
+          :vPoster="modPlayVOD.VOD.thumb"
+          :v-source="modPlayVOD.VOD.hls"
+      ></VMediaPlayer>
+    </template>
+
+    <template #controls>
+      <button @click="modPlayVOD.closeModPlayVOD()">
+        <i>close</i>
+        <span>Close</span>
+      </button>
+    </template>
+  </VModal>
+</template>
+
+<script setup>
+// Store
+import {useModPlayVODStore} from '@/store/modPlayVOD';
+// Components
+import VModal from '@/components/VModal.vue';
+import VMediaPlayer from '@/components/VMediaPlayer.vue';
+
+// Store
+const modPlayVOD = useModPlayVODStore();
+</script>
+
+<style lang="scss">
+.mod-play-vod {
+  width: 100%;
+  max-width: 950rem;
+  max-height: 80%;
+  overflow: hidden;
+}
+</style>
\ No newline at end of file
diff --git a/frontend/src/modules/ws-util.js b/frontend/src/modules/ws-util.js
index 05c3b5b..14c2391 100644
--- a/frontend/src/modules/ws-util.js
+++ b/frontend/src/modules/ws-util.js
@@ -124,6 +124,10 @@ export function useWSUtil() {
         return emitPromise('verify-url', url);
     }
 
+    function getPlayableVOD(url) {
+        return emitPromise('get-playable', url);
+    }
+
     function searchVODs(query, page) {
         return emitPromise('search-vods', query, page);
     }
@@ -168,6 +172,7 @@ export function useWSUtil() {
         saveConfig,
         login,
         verifyURL,
+        getPlayableVOD,
         getFormats,
         downloadVOD,
         cancelDownload,
diff --git a/frontend/src/pages/Landing.vue b/frontend/src/pages/Landing.vue
index 61c4a71..5c18df1 100644
--- a/frontend/src/pages/Landing.vue
+++ b/frontend/src/pages/Landing.vue
@@ -178,6 +178,7 @@
             :vShowDuration="store.config.showDuration"
             :vShowDesc="store.config.showDesc"
             :vBusyState="busy"
+            @play="(id) => playVOD(store.getFightPassURLByID(id))"
             @download="(id) => verifyVODURL(store.getFightPassURLByID(id))"
             @getFormats="(id) => getAvailableFormats(store.getFightPassURLByID(id))"
             @openExternal="(id) => store.openVODInFightPass(id)"
@@ -218,6 +219,7 @@
     <ModViewFormats
         @download="onModViewFormatsDownload"
     ></ModViewFormats>
+    <ModPlayVOD></ModPlayVOD>
     <ModSearchHelp></ModSearchHelp>
     <ModSupport></ModSupport>
     <ModMsgBox
@@ -238,10 +240,11 @@
 
 <script setup>
 // Core
-import {ref, nextTick, onMounted} from 'vue';
+import {nextTick, onMounted, ref} from 'vue';
 // Store
 import {useAppStore} from '@/store';
 import {useModViewFormatsStore} from '@/store/modViewFormats';
+import {useModPlayVODStore} from '@/store/modPlayVOD';
 // Modules
 import {useWSUtil} from '@/modules/ws-util';
 // Components
@@ -251,6 +254,7 @@ import BlockVODCard from '@/components/BlockVODCard.vue';
 import ModVODConfirm from '@/components/ModVODConfirm.vue';
 import ModConfig from '@/components/ModConfig.vue';
 import ModBinDL from '@/components/ModBinDL.vue';
+import ModPlayVOD from '@/components/ModPlayVOD.vue';
 import ModViewFormats from '@/components/ModViewFormats.vue';
 import ModSupport from '@/components/ModSupport.vue';
 import ModMsgBox from '@/components/ModMsgBox.vue';
@@ -260,6 +264,7 @@ import Overlay from '@/components/Overlay.vue';
 // Store
 const store = useAppStore();
 const modViewFormats = useModViewFormatsStore();
+const modPlayVOD = useModPlayVODStore();
 
 // Websocket
 const {
@@ -270,6 +275,7 @@ const {
   openDownloadsDir,
   searchVODs,
   verifyURL,
+  getPlayableVOD,
   getFormats
 } = useWSUtil();
 
@@ -354,6 +360,20 @@ function searchPreviousPage() {
   searchVOD(store.search.result.query, store.search.result.page - 1);
 }
 
+function playVOD(url) {
+  if (!store.isLoggedIn) return store.popError('You need to be logged in to play videos');
+
+  switchBusyState();
+
+  getPlayableVOD(url)
+      .then((VOD) => {
+        modPlayVOD.setVOD(VOD);
+        modPlayVOD.showModPlayVOD();
+      })
+      .catch(store.popError)
+      .finally(switchBusyState);
+}
+
 function verifyVODURL(url) {
   if (!store.isLoggedIn) return store.popError('You need to be logged in to download videos');
 
diff --git a/frontend/src/store/modPlayVOD.js b/frontend/src/store/modPlayVOD.js
new file mode 100644
index 0000000..48e92b5
--- /dev/null
+++ b/frontend/src/store/modPlayVOD.js
@@ -0,0 +1,20 @@
+// Store
+import {defineStore} from 'pinia';
+
+export const useModPlayVODStore = defineStore('modPlayVOD', {
+    state: () => ({
+        active: false,
+        VOD: {}
+    }),
+    actions: {
+        showModPlayVOD() {
+            this.active = true;
+        },
+        closeModPlayVOD() {
+            this.active = false;
+        },
+        setVOD(VOD) {
+            Object.assign(this.VOD, VOD);
+        }
+    }
+});