Implemented processStreamProtectionStatus

Author: felipeucelli

src/main/java/com/github/felipeucelli/javatube/sabr/core/ServerAbrStream.java

@@ -66,6 +66,27 @@ public int getValue() {
     }
 }
 
+// Reference https://github.com/coletdjnz/yt-dlp-dev/blob/5c0c2963396009a92101bc6e038b61844368409d/yt_dlp/extractor/youtube/_streaming/sabr/part.py
+enum PoTokenStatus{
+    UNKNOWN(-1),
+    OK(1),                  // PO Token is provided and valid
+    MISSING(3),             // PO Token is not provided, and is required. A PO Token should be provided ASAP
+    INVALID(3),             // PO Token is provided, but is invalid. A new one should be generated ASAP
+    PENDING(2),             // PO Token is provided, but probably only a cold start token. A full PO Token should be provided ASAP
+    NOT_REQUIRED(1),        // PO Token is not provided, and is not required
+    PENDING_MISSING(2);     // PO Token is not provided, but is pending. A full PO Token should be (probably) provided ASAP
+
+    private final int value;
+
+    PoTokenStatus(int value) {
+        this.value = value;
+    }
+
+    public int getValue() {
+        return value;
+    }
+}
+
 public class ServerAbrStream {
     private final Stream stream;
     private final BiConsumer<byte[], Long> writeChunk;
@@ -82,6 +103,7 @@ public class ServerAbrStream {
     private final Map<String, List<Integer>> previousSequences;
     private boolean RELOAD;
     private int maximumReloadAttempt;
+    private String streamProtectionStatus = PoTokenStatus.UNKNOWN.name();
     private List<Integer> sabrContextsToSend;
     private HashMap<Integer, StreamerContext.StreamerContextUpdate> sabrContextUpdates;
 
@@ -209,7 +231,7 @@ public void start() throws Exception {
                 RELOAD = false;
                 continue;
             } else if (maximumReloadAttempt <= 0) {
-                throw new SABRError("SABR Maximum reload attempts reached");
+                throw new SABRError("SABR Maximum reload attempts reached. Stream protection status: PoToken " + streamProtectionStatus);
             }
 
             if (mainFormat == null || ((Number) mainFormat.get("sequenceCount")).intValue() == ((Number) ((List<Map<String, Object>>) mainFormat.get("sequenceList")).get(((List) mainFormat.get("sequenceList")).size() - 1).get("sequenceNumber")).intValue()) {
@@ -390,7 +412,7 @@ private Map<String, Object> parseUmpResponse(byte[] response) {
                 sabrRedirect[0] = processSabrRedirect(data.get(0));
 
             } else if (partType == PART.STREAM_PROTECTION_STATUS.getValue()) {
-                StreamProtectionStatus.decode(data.get(0));
+                processStreamProtectionStatus(data.get(0));
 
             } else if (partType == PART.RELOAD_PLAYER_RESPONSE.getValue()) {
                 RELOAD = true;
@@ -520,6 +542,27 @@ private SabrRedirect processSabrRedirect(byte[] data) {
         return sabrRedirect;
     }
 
+    private void processStreamProtectionStatus(byte[] data){
+        int protectionStatus = StreamProtectionStatus.decode(data).status;
+
+        String resultStatus;
+
+        if (protectionStatus == StreamProtectionStatus.Status.OK.getValue()){
+            resultStatus = poToken != null ? PoTokenStatus.OK.name() : PoTokenStatus.NOT_REQUIRED.name();
+
+        }else if (protectionStatus == StreamProtectionStatus.Status.ATTESTATION_PENDING.getValue()){
+            resultStatus = poToken != null ? PoTokenStatus.PENDING.name() : PoTokenStatus.PENDING_MISSING.name();
+
+        }else if (protectionStatus == StreamProtectionStatus.Status.ATTESTATION_REQUIRED.getValue()){
+            resultStatus = poToken != null ? PoTokenStatus.INVALID.name() : PoTokenStatus.MISSING.name();
+        }else {
+            resultStatus = PoTokenStatus.UNKNOWN.name();
+        }
+
+        streamProtectionStatus = resultStatus;
+
+    }
+
     private void processSnackbarMessage() {
         int skip = sabrContextUpdates.get(sabrContextsToSend.get(sabrContextsToSend.size() - 1)).value.field1.skip;
         if (skip >= 60000){

src/main/java/com/github/felipeucelli/javatube/sabr/streaming/StreamProtectionStatus.java

@@ -53,6 +53,22 @@ public static StreamProtectionStatus decode(Object inputData, Integer length) {
         }
         return message;
     }
+
+    public enum Status {
+        OK(1),
+        ATTESTATION_PENDING(2),
+        ATTESTATION_REQUIRED(3);
+
+        private final int value;
+
+        Status(int value) {
+            this.value = value;
+        }
+
+        public int getValue() {
+            return value;
+        }
+    }
 }
 
 

src/main/java/com/github/felipeucelli/javatube/sabr/streaming/StreamerContext.java

@@ -1,12 +1,11 @@
 package com.github.felipeucelli.javatube.sabr.streaming;
 
-import com.github.felipeucelli.javatube.sabr.common.FormatId;
 import com.github.felipeucelli.javatube.sabr.proto.Proto.BinaryReader;
 import com.github.felipeucelli.javatube.sabr.proto.Proto.BinaryWriter;
 
 import java.util.ArrayList;
 import java.util.List;
-import java.util.Objects;
+
 
 public class StreamerContext {
     public StreamerContextClientInfo clientInfo;