Fixed: Plex server authentication

2026-04-21 22:04:31 -04:00 · 2014-04-14 22:07:43 -07:00
parent af75813347
commit 5bc820efed
9 changed files with 202 additions and 94 deletions
@@ -0,0 +1,121 @@
+using System;
+using System.Collections.Generic;
+using System.IO;
+using System.Linq;
+using Newtonsoft.Json.Linq;
+using NzbDrone.Common;
+using NzbDrone.Common.Cache;
+using NzbDrone.Common.Serializer;
+using RestSharp;
+
+namespace NzbDrone.Core.Notifications.Plex
+{
+    public interface IPlexServerProxy
+    {
+        List<PlexSection> GetTvSections(PlexServerSettings settings);
+        void Update(int sectionId, PlexServerSettings settings);
+    }
+
+    public class PlexServerProxy : IPlexServerProxy
+    {
+        private readonly ICached<String> _authCache;
+
+        public PlexServerProxy(ICacheManager cacheManager)
+        {
+            _authCache = cacheManager.GetCache<String>(GetType(), "authCache");
+        }
+
+        public List<PlexSection> GetTvSections(PlexServerSettings settings)
+        {
+            var request = GetPlexServerRequest("library/sections", Method.GET, settings);
+            var client = GetPlexServerClient(settings);
+
+            var response = client.Execute(request);
+
+            return Json.Deserialize<PlexMediaContainer>(response.Content)
+                       .Directories
+                       .Where(d => d.Type == "show")
+                       .SelectMany(d => d.Sections)
+                       .ToList();
+        }
+
+        public void Update(int sectionId, PlexServerSettings settings)
+        {
+            var resource = String.Format("library/sections/{2}/refresh");
+            var request = GetPlexServerRequest(resource, Method.GET, settings);
+            var client = GetPlexServerClient(settings);
+
+            var response = client.Execute(request);
+        }
+
+        private String Authenticate(string username, string password)
+        {
+            var request = GetMyPlexRequest("users/sign_in.json", Method.POST);
+            var client = GetMyPlexClient(username, password); 
+
+            var response = client.Execute(request);
+            CheckForError(response.Content);
+
+            var user = Json.Deserialize<PlexUser>(JObject.Parse(response.Content).SelectToken("user").ToString());
+
+            _authCache.Set(username, user.AuthenticationToken);
+
+            return user.AuthenticationToken;
+        }
+
+        private RestClient GetMyPlexClient(string username, string password)
+        {
+            var client = new RestClient("https://my.plexapp.com");
+            client.Authenticator = new HttpBasicAuthenticator(username, password);
+
+            return client;
+        }
+
+        private RestRequest GetMyPlexRequest(string resource, Method method)
+        {
+            var request = new RestRequest(resource, method);
+	        request.AddHeader("X-Plex-Platform", "Windows");
+	        request.AddHeader("X-Plex-Platform-Version", "7");
+	        request.AddHeader("X-Plex-Provides", "player");
+	        request.AddHeader("X-Plex-Client-Identifier", "AB6CCCC7-5CF5-4523-826A-B969E0FFD8A0");
+	        request.AddHeader("X-Plex-Product", "PlexWMC");
+	        request.AddHeader("X-Plex-Version", "0");
+
+            return request;
+
+        }
+
+        private RestClient GetPlexServerClient(PlexServerSettings settings)
+        {
+            return new RestClient(String.Format("http://{0}:{1}", settings.Host, settings.Port));
+        }
+
+        private RestRequest GetPlexServerRequest(string resource, Method method, PlexServerSettings settings)
+        {
+            var request = new RestRequest(resource, method);
+            request.AddHeader("Accept", "application/json");
+
+            if (!settings.Username.IsNullOrWhiteSpace())
+            {
+                request.AddParameter("X-Plex-Token", GetAuthenticationToken(settings.Username, settings.Password));
+            }
+
+            return request;
+        }
+
+        private string GetAuthenticationToken(string username, string password)
+        {
+            return _authCache.Get(username, () => Authenticate(username, password));
+        }
+
+        private void CheckForError(string response)
+        {
+            var error = Json.Deserialize<PlexError>(response);
+
+            if (error != null && !error.Error.IsNullOrWhiteSpace())
+            {
+                throw new PlexException(error.Error);
+            }
+        }
+    }
+}