From 3a788bcfae46e8dae1b3d34fe464124ebf0f004a Mon Sep 17 00:00:00 2001
From: Johan Andersson <j@i19.se>
Date: Sat, 20 Aug 2016 16:32:12 +0200
Subject: [PATCH] twitch: Add support for clips

fixes #426
---
 lib/svtplay_dl/service/twitch.py | 27 +++++++++++++++++++++++++--
 1 file changed, 25 insertions(+), 2 deletions(-)

diff --git a/lib/svtplay_dl/service/twitch.py b/lib/svtplay_dl/service/twitch.py
index 0261ab9..200e144 100644
--- a/lib/svtplay_dl/service/twitch.py
+++ b/lib/svtplay_dl/service/twitch.py
@@ -8,12 +8,14 @@ from __future__ import absolute_import
 import re
 import json
 import os
+import copy
 
 from svtplay_dl.utils.urllib import urlparse, quote_plus
 from svtplay_dl.service import Service
 from svtplay_dl.utils import filenamify
 from svtplay_dl.log import log
 from svtplay_dl.fetcher.hls import hlsparse
+from svtplay_dl.fetcher.http import HTTP
 from svtplay_dl.error import ServiceError
 
 
@@ -38,7 +40,7 @@ class Twitch(Service):
     # are usually two characters, but may have a country suffix as well (e.g.
     # zh-tw, zh-cn and pt-br.
     supported_domains_re = [
-        r'^(?:(?:[a-z]{2}-)?[a-z]{2}\.)?(www\.)?twitch\.tv$',
+        r'^(?:(?:[a-z]{2}-)?[a-z]{2}\.)?(www\.|clips\.)?twitch\.tv$',
     ]
 
     api_base_url = 'https://api.twitch.tv'
@@ -53,7 +55,10 @@ class Twitch(Service):
 
         match = re.match(r'/(\w+)/([bcv])/(\d+)', urlp.path)
         if not match:
-            data = self._get_channel(self.options, urlp)
+            if re.search("clips.twitch.tv", urlp.netloc):
+                data = self._get_clips(self.options)
+            else:
+                data = self._get_channel(self.options, urlp)
         else:
             if match.group(2) in ["b", "c"]:
                 yield ServiceError("This twitch video type is unsupported")
@@ -158,3 +163,21 @@ class Twitch(Service):
         streams = hlsparse(options, data, hls_url)
         for n in list(streams.keys()):
             yield streams[n]
+
+    def _get_clips(self, options):
+        match = re.search("quality_options: (\[[^\]]+\])", self.get_urldata())
+        if not match:
+            yield ServiceError("Can't find the video clip")
+            return
+        if options.output_auto:
+            name = re.search('slug: "([^"]+)"', self.get_urldata()).group(1)
+            brodcaster = re.search('broadcaster_login: "([^"]+)"', self.get_urldata()).group(1)
+            name = "twitch-%s-%s" % (brodcaster, name)
+            directory = os.path.dirname(options.output)
+            if os.path.isdir(directory):
+                name = os.path.join(directory, name)
+            options.output = name
+
+        dataj = json.loads(match.group(1))
+        for i in dataj:
+            yield HTTP(copy.copy(options), i["source"], i["quality"])