src getter fix, local storage fix

2 files changed, 20 insertions, 0 deletions

diff --git a/src/js/plugins/vimeo.js b/src/js/plugins/vimeo.js
index f75b16aa..b9cc92c2 100644
--- a/src/js/plugins/vimeo.js
+++ b/src/js/plugins/vimeo.js
@@ -126,6 +126,19 @@ const vimeo = {
             },
         });
 
+        // Source
+        let currentSrc;
+
+        player.embed.getVideoUrl.then(value => {
+            currentSrc = value;
+        });
+
+        Object.defineProperty(player.media, 'currentSrc', {
+            get() {
+                return currentSrc;
+            },
+        });
+
         // Rebuild UI
         window.setTimeout(() => ui.build.call(player), 0);
 
diff --git a/src/js/plugins/youtube.js b/src/js/plugins/youtube.js
index 2b4c2cdc..12e9d3e6 100644
--- a/src/js/plugins/youtube.js
+++ b/src/js/plugins/youtube.js
@@ -155,6 +155,13 @@ const youtube = {
                         },
                     });
 
+                    // Source
+                    Object.defineProperty(player.media, 'currentSrc', {
+                        get() {
+                            return instance.getVideoUrl();
+                        },
+                    });
+
                     // Get available speeds
                     if (player.config.controls.includes('settings') && player.config.settings.includes('speed')) {
                         controls.setSpeedMenu.call(player, instance.getAvailablePlaybackRates());