fix media track turning off

Flectone · Apr 6, 2024 · e71db35 · e71db35
1 parent 0b915c1
commit e71db35
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/main/java/net/flectone/music/javafx/controller/ControllerApp.java b/src/main/java/net/flectone/music/javafx/controller/ControllerApp.java
@@ -36,6 +36,7 @@ public class ControllerApp implements Initializable {
     private static final String GET_MEDIA_NAME_SCRIPT = "document.getElementsByClassName('style-scope ytd-watch-metadata')[0].firstElementChild.textContent;";
     private static final String GET_SONG_NAME_SCRIPT = "externalAPI.getCurrentTrack().title;";
     private static final String MEDIA_INJECT_SCRIPT = "document.querySelector('video').addEventListener('ended', e => {window.media.playNext(true)});";
+    private static final String AUTO_MEDIA_INJECT_SCRIPT = "setTimeout(() => {if (document.querySelector('.ytp-time-current').textContent == \"0:00\") {document.querySelector('.ytp-cued-thumbnail-overlay').click();}}, 1500)";
 
     // UI elements
     @FXML
@@ -77,7 +78,7 @@ private void setupPlayer() {
 
         playerWebEngine.getLoadWorker().stateProperty().addListener((observable, oldValue, newValue) -> {
             if (newValue == Worker.State.SUCCEEDED) {
-                playerWebEngine.executeScript(SONG_INJECT_SCRIPT);
+                playerWebEngine.executeScript(AUTO_SONG_INJECT_SCRIPT);
                 CookieUtils.saveCookies();
             }
         });
@@ -125,6 +126,7 @@ private void injectScript(WebEngine webEngine) {
         }
 
         webEngine.executeScript(MEDIA_INJECT_SCRIPT);
+        webEngine.executeScript(AUTO_MEDIA_INJECT_SCRIPT);
     }
 
     // Inner class to bridge between Java and JavaScript