Expose itag in StreamFormat

lavalink-devs · May 27, 2024 · 22e2690 · 22e2690
1 parent 2bb3505
commit 22e2690
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/common/src/main/java/dev/lavalink/youtube/clients/skeleton/StreamingNonMusicClient.java b/common/src/main/java/dev/lavalink/youtube/clients/skeleton/StreamingNonMusicClient.java
@@ -95,6 +95,7 @@ protected boolean extractFormat(JsonBrowser formatJson,
 
             formats.add(new StreamFormat(
                 ContentType.parse(formatJson.get("mimeType").text()),
+                (int) formatJson.get("itag").asLong(-1L),
                 formatJson.get("bitrate").asLong(Units.BITRATE_UNKNOWN),
                 contentLength,
                 formatJson.get("audioChannels").asLong(2),

diff --git a/common/src/main/java/dev/lavalink/youtube/track/format/StreamFormat.java b/common/src/main/java/dev/lavalink/youtube/track/format/StreamFormat.java
@@ -13,6 +13,7 @@
 public class StreamFormat {
   private final FormatInfo info;
   private final ContentType type;
+  private final int itag;
   private final long bitrate;
   private final long contentLength;
   private final long audioChannels;
@@ -37,6 +38,7 @@ public class StreamFormat {
    */
   public StreamFormat(
       ContentType type,
+      int itag,
       long bitrate,
       long contentLength,
       long audioChannels,
@@ -49,6 +51,7 @@ public StreamFormat(
   ) {
     this.info = FormatInfo.get(type);
     this.type = type;
+    this.itag = itag;
     this.bitrate = bitrate;
     this.contentLength = contentLength;
     this.audioChannels = audioChannels;
@@ -76,6 +79,10 @@ public ContentType getType() {
     return type;
   }
 
+  public int getItag() {
+    return itag;
+  }
+
   /**
    * @return Bitrate of the format
    */
@@ -150,7 +157,8 @@ public boolean isDrc() {
   @Override
   public String toString() {
     return "YoutubeStreamFormat{" +
-        "type=" + type +
+        "itag=" + itag +
+        ", type=" + type +
         ", bitrate=" + bitrate +
         ", audioChannels=" + audioChannels +
         ", isDrc=" + isDrc +