Extractor is now able to detect if a video is blocked by country

Author: ram-on

services/youtube/YoutubeStreamExtractor.java

@@ -260,7 +260,7 @@ private JSONObject getPlayerConfig(String pageContent) throws ParsingException {
                     Parser.matchGroup1("ytplayer.config\\s*=\\s*(\\{.*?\\});", pageContent);
             return new JSONObject(ytPlayerConfigRaw);
         } catch (Parser.RegexException e) {
-            String errorReason = findErrorReason(doc);
+            String errorReason = getErrorMessage();
             switch(errorReason) {
                 case "GEMA":
                     throw new GemaException(errorReason);
@@ -948,15 +948,28 @@ private String decryptSignature(String encryptedSig, String decryptionCode)
         return result == null ? "" : result.toString();
     }
 
-    private String findErrorReason(Document doc) {
-        String errorMessage = doc.select("h1[id=\"unavailable-message\"]").first().text();
-        if(errorMessage.contains("GEMA")) {
+
+    /**
+     * {@inheritDoc}
+     */
+    public String getErrorMessage() {
+        String          errorMessage = doc.select("h1[id=\"unavailable-message\"]").first().text();
+        StringBuilder   errorReason;
+
+        if (errorMessage == null  ||  errorMessage.isEmpty()) {
+            errorReason = null;
+        } else if(errorMessage.contains("GEMA")) {
             // Gema sometimes blocks youtube music content in germany:
             // https://www.gema.de/en/
             // Detailed description:
             // https://en.wikipedia.org/wiki/GEMA_%28German_organization%29
-            return "GEMA";
+            errorReason = new StringBuilder("GEMA");
+        } else {
+            errorReason = new StringBuilder(errorMessage);
+            errorReason.append("  ");
+            errorReason.append(doc.select("[id=\"unavailable-submessage\"]").first().text());
         }
-        return "";
+
+        return errorReason != null  ?  errorReason.toString()  :  null;
     }
 }

stream_info/StreamExtractor.java

@@ -49,7 +49,7 @@ public ExtractorInitException(String message, Throwable cause) {
         }
     }
 
-    public class ContentNotAvailableException extends ParsingException {
+    public static class ContentNotAvailableException extends ParsingException {
         public ContentNotAvailableException(String message) {
             super(message);
         }
@@ -101,4 +101,11 @@ public UrlIdHandler getUrlIdHandler() {
     public int getServiceId() {
         return serviceId;
     }
+
+    /**
+     * Analyses the webpage's document and extracts any error message there might be.
+     *
+     * @return  Error message; null if there is no error message.
+     */
+    public abstract String getErrorMessage();
 }

stream_info/StreamInfo.java

@@ -74,19 +74,34 @@ public void addException(Exception e) {
     /**Fills out the video info fields which are common to all services.
      * Probably needs to be overridden by subclasses*/
     public static StreamInfo getVideoInfo(StreamExtractor extractor)
-            throws ExtractionException, IOException {
+            throws ExtractionException, StreamExtractor.ContentNotAvailableException {
         StreamInfo streamInfo = new StreamInfo();
 
-        streamInfo = extractImportantData(streamInfo, extractor);
-        streamInfo = extractStreams(streamInfo, extractor);
-        streamInfo = extractOptionalData(streamInfo, extractor);
+        try {
+            streamInfo = extractImportantData(streamInfo, extractor);
+            streamInfo = extractStreams(streamInfo, extractor);
+            streamInfo = extractOptionalData(streamInfo, extractor);
+        } catch (ExtractionException e) {
+	        // Currently YouTube does not distinguish between age restricted videos and videos blocked
+	        // by country.  This means that during the initialisation of the extractor, the extractor
+	        // will assume that a video is age restricted while in reality it it blocked by country.
+	        //
+	        // We will now detect whether the video is blocked by country or not.
+            String errorMsg = extractor.getErrorMessage();
+
+            if (errorMsg != null) {
+                throw new StreamExtractor.ContentNotAvailableException(errorMsg);
+            } else {
+                throw e;
+            }
+        }
 
         return streamInfo;
     }
 
     private static StreamInfo extractImportantData(
             StreamInfo streamInfo, StreamExtractor extractor)
-            throws ExtractionException, IOException {
+            throws ExtractionException {
         /* ---- important data, withoug the video can't be displayed goes here: ---- */
         // if one of these is not available an exception is meant to be thrown directly into the frontend.
 
@@ -112,7 +127,7 @@ private static StreamInfo extractImportantData(
 
     private static StreamInfo extractStreams(
             StreamInfo streamInfo, StreamExtractor extractor)
-            throws ExtractionException, IOException {
+            throws ExtractionException {
         /* ---- stream extraction goes here ---- */
         // At least one type of stream has to be available,
         // otherwise an exception will be thrown directly into the frontend.