Merge pull request #58 from lutsenkirill/media_variants_support

Media variants support
Xwilarg · Aug 15, 2023 · 6985407 · 6985407
2 parents 029d684 + 6f19ea8
commit 6985407
Show file tree

Hide file tree

Showing 5 changed files with 62 additions and 2 deletions.
diff --git a/src/JsonOption/MediaConverter.cs b/src/JsonOption/MediaConverter.cs
@@ -45,12 +45,29 @@ public override Media Read(ref Utf8JsonReader reader, Type typeToConvert, JsonSe
                     Width = TryGetProperty(elem, "width")?.GetInt32(),
                     DurationMs = TryGetProperty(elem, "duration_ms")?.GetInt32(),
                     PreviewImageUrl = TryGetProperty(elem, "preview_image_url")?.GetString(),
-                    Url = TryGetProperty(elem, "url")?.GetString()
+                    Url = TryGetProperty(elem, "url")?.GetString(),
+                    Variants = GetMediaVariants(elem)
                 };
+
                 return media;
             }
         }
 
+        private MediaVariant[] GetMediaVariants(JsonElement mediaElement)
+        {
+            var mediaVariants = Array.Empty<MediaVariant>();
+
+            if(mediaElement.TryGetProperty("variants", out var variants))
+            {
+                mediaVariants = JsonSerializer.Deserialize<MediaVariant[]>(variants.GetRawText(), new JsonSerializerOptions
+                {
+                    PropertyNamingPolicy = new SnakeCaseNamingPolicy()
+                });
+            }
+
+            return mediaVariants;
+        }
+
         public override void Write(Utf8JsonWriter writer, Media value, JsonSerializerOptions options)
         {
             throw new NotImplementedException();

diff --git a/src/Request/AdvancedSearch/MediaOption.cs b/src/Request/AdvancedSearch/MediaOption.cs
@@ -25,6 +25,12 @@ public enum MediaOption
         /// <summary>
         /// URL to the image of the content
         /// </summary>
-        Url
+        Url,
+
+        /// <summary>
+        /// Each media object may have multiple display or playback variants, with different resolutions or formats.
+        /// For videos, each variant will also contain URLs to the video in each format.
+        /// </summary>
+        Variants
     }
 }
diff --git a/src/Response/RMedia/Media.cs b/src/Response/RMedia/Media.cs
@@ -10,5 +10,6 @@ public class Media
         public int? DurationMs { init; get; }
         public string PreviewImageUrl { init; get; }
         public string Url { init; get; }
+        public MediaVariant[] Variants { init; get; }
     }
 }
diff --git a/src/Response/RMedia/MediaVariant.cs b/src/Response/RMedia/MediaVariant.cs
@@ -0,0 +1,9 @@
+namespace TwitterSharp.Response.RMedia
+{
+    public class MediaVariant
+    {
+        public int? BitRate { init; get; }
+        public string ContentType { init; get; }
+        public string Url { init; get; }
+    }
+}
diff --git a/test/TestMedia.cs b/test/TestMedia.cs
@@ -1,5 +1,6 @@
 using Microsoft.VisualStudio.TestTools.UnitTesting;
 using System;
+using System.Linq;
 using System.Threading.Tasks;
 using TwitterSharp.Client;
 using TwitterSharp.Request.AdvancedSearch;
@@ -78,5 +79,31 @@ public async Task GetTweetWithMediaPreview()
             Assert.IsNotNull(a.Attachments.Media[0].PreviewImageUrl);
             Assert.AreEqual("https://pbs.twimg.com/ext_tw_video_thumb/1237543944570847233/pu/img/kRBUlSd7M7ju_QK1.jpg", a.Attachments.Media[0].PreviewImageUrl);
         }
+
+        [TestMethod]
+        public async Task GetTweetWithVideoMediaVariants()
+        {
+            var client = new TwitterClient(Environment.GetEnvironmentVariable("TWITTER_TOKEN"));
+            var answer = await client.GetTweetsAsync(new[] { "1688848776612691968" }, new TweetSearchOptions
+            {
+                TweetOptions = new[] { TweetOption.Attachments },
+                MediaOptions = new [] { MediaOption.Variants }
+            });
+            Assert.IsTrue(answer.Length == 1);
+            var a = answer[0];
+            Assert.IsNotNull(a.Attachments);
+            Assert.IsNotNull(a.Attachments.Media);
+            Assert.AreEqual(1, a.Attachments.Media.Length);
+            Assert.AreEqual("13_1688840150338428928", a.Attachments.Media[0].Key);
+            Assert.IsNotNull(a.Attachments.Media[0].Type);
+            Assert.AreEqual(MediaType.Video, a.Attachments.Media[0].Type);
+            Assert.IsNull(a.Attachments.Media[0].PreviewImageUrl);
+
+            Assert.IsNotNull(a.Attachments.Media[0].Variants);
+            Assert.IsTrue(a.Attachments.Media[0].Variants.Length > 0);
+            Assert.IsTrue(a.Attachments.Media[0].Variants.Any(v => v.ContentType == "video/mp4"));
+            Assert.IsTrue(a.Attachments.Media[0].Variants.Any(v => v.BitRate.HasValue));
+            Assert.IsTrue(a.Attachments.Media[0].Variants.All(v => string.IsNullOrWhiteSpace(v.Url) == false));
+        }
     }
 }