From 0ba0c0ed43fbcf4a57f6fc91829ba00d72e3672f Mon Sep 17 00:00:00 2001
From: bachinger <bachinger@google.com>
Date: Wed, 11 Jan 2023 23:46:58 +0000
Subject: [PATCH] Add AdsLoader.focusSkipButton()

This method allows to call through to `StreamManager.focus()` of the currently playing SSAI stream.

PiperOrigin-RevId: 501399144
---
 .../ImaServerSideAdInsertionMediaSource.java  | 34 +++++++++++++++++--
 1 file changed, 31 insertions(+), 3 deletions(-)

diff --git a/extensions/ima/src/main/java/com/google/android/exoplayer2/ext/ima/ImaServerSideAdInsertionMediaSource.java b/extensions/ima/src/main/java/com/google/android/exoplayer2/ext/ima/ImaServerSideAdInsertionMediaSource.java
index 27ea2275ce..d97e107c37 100644
--- a/extensions/ima/src/main/java/com/google/android/exoplayer2/ext/ima/ImaServerSideAdInsertionMediaSource.java
+++ b/extensions/ima/src/main/java/com/google/android/exoplayer2/ext/ima/ImaServerSideAdInsertionMediaSource.java
@@ -374,8 +374,7 @@ public final class ImaServerSideAdInsertionMediaSource extends CompositeMediaSou
 
     private final ServerSideAdInsertionConfiguration configuration;
     private final Context context;
-    private final Map<ImaServerSideAdInsertionMediaSource, MediaSourceResourceHolder>
-        mediaSourceResources;
+    private final Map<String, MediaSourceResourceHolder> mediaSourceResources;
     private final Map<String, AdPlaybackState> adPlaybackStateMap;
 
     @Nullable private Player player;
@@ -401,6 +400,35 @@ public final class ImaServerSideAdInsertionMediaSource extends CompositeMediaSou
       this.player = player;
     }
 
+    /**
+     * Puts the focus on the skip button, if a skip button is present and an ad is playing.
+     *
+     * @see StreamManager#focus()
+     */
+    public void focusSkipButton() {
+      if (player == null) {
+        return;
+      }
+      if (player.getPlaybackState() != Player.STATE_IDLE
+          && player.getPlaybackState() != Player.STATE_ENDED
+          && player.getMediaItemCount() > 0) {
+        int currentPeriodIndex = player.getCurrentPeriodIndex();
+        Object adsId =
+            player
+                .getCurrentTimeline()
+                .getPeriod(currentPeriodIndex, new Timeline.Period())
+                .getAdsId();
+        if (adsId instanceof String) {
+          MediaSourceResourceHolder mediaSourceResourceHolder = mediaSourceResources.get(adsId);
+          if (mediaSourceResourceHolder != null
+              && mediaSourceResourceHolder.imaServerSideAdInsertionMediaSource.streamManager
+                  != null) {
+            mediaSourceResourceHolder.imaServerSideAdInsertionMediaSource.streamManager.focus();
+          }
+        }
+      }
+    }
+
     /**
      * Releases resources.
      *
@@ -427,7 +455,7 @@ public final class ImaServerSideAdInsertionMediaSource extends CompositeMediaSou
         StreamPlayer streamPlayer,
         com.google.ads.interactivemedia.v3.api.AdsLoader adsLoader) {
       mediaSourceResources.put(
-          mediaSource, new MediaSourceResourceHolder(mediaSource, streamPlayer, adsLoader));
+          mediaSource.adsId, new MediaSourceResourceHolder(mediaSource, streamPlayer, adsLoader));
     }
 
     private AdPlaybackState getAdPlaybackState(String adsId) {