- Adding in check for YouTube membership capabilities before performi…

…ng any queries for data related to it
SaviorXTanren · Sep 21, 2023 · e9135e8 · e9135e8
1 parent f3a9bd4
commit e9135e8
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 16 deletions.
diff --git a/MixItUp.Base/Model/User/Platform/YouTubeUserPlatformV2Model.cs b/MixItUp.Base/Model/User/Platform/YouTubeUserPlatformV2Model.cs
@@ -95,21 +95,24 @@ public void SetMessageProperties(LiveChatMessage message)
 
         public async Task RefreshMembershipDetails()
         {
-            Member membership = await ServiceManager.Get<YouTubeSessionService>().UserConnection.CheckIfMember(this.ID);
-            if (membership != null)
+            if (ServiceManager.Get<YouTubeSessionService>().HasMembershipCapabilities)
             {
-                this.Roles.Add(UserRoleEnum.YouTubeMember);
-                this.Roles.Add(UserRoleEnum.Subscriber);
-                this.SubscribeDate = DateTime.Parse(membership.Snippet.MembershipsDetails.MembershipsDuration.MemberSince);
-                this.MemberLevels.Clear();
-                this.MemberLevels.AddRange(membership.Snippet.MembershipsDetails.AccessibleLevels);
-            }
-            else
-            {
-                this.Roles.Remove(UserRoleEnum.YouTubeMember);
-                this.Roles.Remove(UserRoleEnum.Subscriber);
-                this.SubscribeDate = null;
-                this.MemberLevels.Clear();
+                Member membership = await ServiceManager.Get<YouTubeSessionService>().UserConnection.CheckIfMember(this.ID);
+                if (membership != null)
+                {
+                    this.Roles.Add(UserRoleEnum.YouTubeMember);
+                    this.Roles.Add(UserRoleEnum.Subscriber);
+                    this.SubscribeDate = DateTime.Parse(membership.Snippet.MembershipsDetails.MembershipsDuration.MemberSince);
+                    this.MemberLevels.Clear();
+                    this.MemberLevels.AddRange(membership.Snippet.MembershipsDetails.AccessibleLevels);
+                }
+                else
+                {
+                    this.Roles.Remove(UserRoleEnum.YouTubeMember);
+                    this.Roles.Remove(UserRoleEnum.Subscriber);
+                    this.SubscribeDate = null;
+                    this.MemberLevels.Clear();
+                }
             }
         }
 

diff --git a/MixItUp.Base/Services/YouTube/YouTubePlatformService.cs b/MixItUp.Base/Services/YouTube/YouTubePlatformService.cs
@@ -105,8 +105,6 @@ public YouTubePlatformService(YouTubeConnection connection)
 
         public async Task<Video> UpdateVideo(Video video, string title = null, string description = null, string categoryId = null) { return await AsyncRunner.RunAsync(this.Connection.Videos.UpdateVideo(video, title, description, categoryId)); }
 
-        public async Task<IEnumerable<Member>> GetChannelMemberships(int maxResults = 1) { return await AsyncRunner.RunAsync(this.Connection.LiveChat.GetChannelMemberships(maxResults)); }
-
         public async Task<IEnumerable<LiveChatModerator>> GetModerators(LiveBroadcast broadcast, int maxResults = 1) { return await AsyncRunner.RunAsync(this.Connection.LiveChat.GetModerators(broadcast, maxResults)); }
 
         public async Task<IEnumerable<Subscription>> GetSubscribers(string channelID, int maxResults = 1) { return await AsyncRunner.RunAsync(this.Connection.Subscriptions.GetMyRecentSubscribers(maxResults)); }
@@ -115,6 +113,8 @@ public YouTubePlatformService(YouTubeConnection connection)
 
         public async Task<IEnumerable<MembershipsLevel>> GetMembershipLevels() { return await AsyncRunner.RunAsync(this.Connection.Membership.GetMyMembershipLevels()); }
 
+        public async Task<IEnumerable<Member>> GetChannelMemberships(int maxResults = 1) { return await AsyncRunner.RunAsync(this.Connection.LiveChat.GetChannelMemberships(maxResults)); }
+
         public async Task<IEnumerable<Member>> GetMembers(int maxResults = 1) { return await AsyncRunner.RunAsync(this.Connection.Membership.GetMembers(maxResults)); }
 
         public async Task<Member> CheckIfMember(string userID) { return await AsyncRunner.RunAsync(this.Connection.Membership.CheckIfMember(userID)); }

diff --git a/MixItUp.Base/Services/YouTube/YouTubeSessionService.cs b/MixItUp.Base/Services/YouTube/YouTubeSessionService.cs
@@ -31,6 +31,8 @@ public class YouTubeSessionService : IStreamingPlatformSessionService
         public string ChannelID { get { return this.User?.Id; } }
         public string ChannelLink { get { return this.User?.Snippet?.CustomUrl; } }
 
+        public bool HasMembershipCapabilities { get { return this.MembershipLevels.Count > 0; } }
+
         private DateTime launchDateTime = DateTime.Now;
 
         public StreamingPlatformAccountModel UserAccount