From aa8c4a6eb7b74b3e689a4cc7e0c83436f8b907c0 Mon Sep 17 00:00:00 2001
From: Ian <ViViDboarder@gmail.com>
Date: Sun, 8 Sep 2024 06:42:26 -0700
Subject: [PATCH] Add ability to play plex media as the non-primary user
 (#122039)

* Adds ability to play media as the non-primary user

* Add return type for set function
---
 homeassistant/components/plex/server.py    | 12 +++++++++++
 homeassistant/components/plex/services.py  |  5 +++++
 tests/components/plex/test_media_search.py | 25 ++++++++++++++++++++++
 3 files changed, 42 insertions(+)

diff --git a/homeassistant/components/plex/server.py b/homeassistant/components/plex/server.py
index fbb98e8e19f..0716b3606af 100644
--- a/homeassistant/components/plex/server.py
+++ b/homeassistant/components/plex/server.py
@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+from copy import copy
 import logging
 import ssl
 import time
@@ -664,3 +665,14 @@ class PlexServer:
     def sensor_attributes(self):
         """Return active session information for use in activity sensor."""
         return {x.sensor_user: x.sensor_title for x in self.active_sessions.values()}
+
+    def set_plex_server(self, plex_server: PlexServer) -> None:
+        """Set the PlexServer instance."""
+        self._plex_server = plex_server
+
+    def switch_user(self, username: str) -> PlexServer:
+        """Return a shallow copy of a PlexServer as the provided user."""
+        new_server = copy(self)
+        new_server.set_plex_server(self.plex_server.switchUser(username))
+
+        return new_server
diff --git a/homeassistant/components/plex/services.py b/homeassistant/components/plex/services.py
index e0fe79be182..cbf72966413 100644
--- a/homeassistant/components/plex/services.py
+++ b/homeassistant/components/plex/services.py
@@ -161,6 +161,11 @@ def process_plex_payload(
     if not plex_server:
         plex_server = get_plex_server(hass)
 
+    if isinstance(content, dict):
+        if plex_user := content.pop("username", None):
+            _LOGGER.debug("Switching to Plex user: %s", plex_user)
+            plex_server = plex_server.switch_user(plex_user)
+
     if content_type == "station":
         if not supports_playqueues:
             raise HomeAssistantError("Plex stations are not supported on this device")
diff --git a/tests/components/plex/test_media_search.py b/tests/components/plex/test_media_search.py
index 8219cbe27b6..04d91e8825c 100644
--- a/tests/components/plex/test_media_search.py
+++ b/tests/components/plex/test_media_search.py
@@ -57,6 +57,31 @@ async def test_media_lookups(
         )
     assert "Media for key 123 not found" in str(excinfo.value)
 
+    # Search with a different specified username
+    with (
+        patch(
+            "plexapi.library.LibrarySection.search",
+            __qualname__="search",
+        ) as search,
+        patch(
+            "plexapi.myplex.MyPlexAccount.user",
+            __qualname__="user",
+        ) as plex_account_user,
+    ):
+        plex_account_user.return_value.get_token.return_value = "token"
+        await hass.services.async_call(
+            MEDIA_PLAYER_DOMAIN,
+            SERVICE_PLAY_MEDIA,
+            {
+                ATTR_ENTITY_ID: media_player_id,
+                ATTR_MEDIA_CONTENT_TYPE: MediaType.EPISODE,
+                ATTR_MEDIA_CONTENT_ID: '{"library_name": "TV Shows", "show_name": "TV Show", "username": "Kids"}',
+            },
+            True,
+        )
+        search.assert_called_with(**{"show.title": "TV Show", "libtype": "show"})
+        plex_account_user.assert_called_with("Kids")
+
     # TV show searches
     with pytest.raises(MediaNotFound) as excinfo:
         await hass.services.async_call(