# Path: lib/pirateplay/lib/rerequest.py
# class TemplateRequest:
# 	def __init__(self, re,
# 				encode_vars = lambda x: x, decode_content = lambda c, v: c,
# 				handlerchain = None):
# 		self.re = re
# 		self.encode_vars = encode_vars
# 		self.decode_content = decode_content
# 		self.handlerchain = handlerchain
# 	
# 	def create_vars(self, content, cumulated_vars):
# 		content = self.decode_content(content, cumulated_vars)
# 		
# 		#Make sure req_data and req_headers are empty
# 		if 'req_data' in cumulated_vars:
# 			del cumulated_vars['req_data']
# 		if 'req_headers' in cumulated_vars:
# 			del cumulated_vars['req_headers']
# 		
# 		for match in re.finditer(self.re, content, re.DOTALL):
# 			d = dict(cumulated_vars, **match.groupdict())
# 			d.update(self.encode_vars(d))
# 			yield d
# 	
# 	def create_req(self, cumulated_vars):
# 		url = cumulated_vars['req_url']
# 		if not url.startswith('http'):
# 			url = 'http://' + url
# 		req = urllib2.Request(url)
# 		
# 		if 'req_data' in cumulated_vars:
# 			debug_print('Adding post data to request: ' + cumulated_vars['req_data'])
# 			req.add_data(cumulated_vars['req_data'])
# 		
# 		for header, value in cumulated_vars.get('req_headers', {}).items():
# 			debug_print('Adding header to request: %s = %s' % (header, value))
# 			req.add_header(header, value)
# 		return req
# 	
# 	def create_content(self, req):
# 		debug_print('Opening URL: ' + req.get_full_url())
# 		try:
# 			try:
# 				f = urllib2.build_opener(self.handlerchain).open(req)
# 			except TypeError:
# 				f = urllib2.urlopen(req)
# 			
# 			content = f.read()
# 			f.close()
# 		except (urllib2.HTTPError, urllib2.URLError):
# 			content = ''
# 		
# 		return content
# 	
# 	def to_dict(self):
# 		return { 'test': self.re }
# 
# def debug_print(s):
# 	if DEBUG:
# 		print(' DEBUG ' + s)


from ..rerequest import TemplateRequest, debug_print
from random import randint
from re import search
from urllib2 import HTTPError, urlopen

#Request chains

rtmp = { 'title': 'SVT-play', 'url': 'http://svtplay.se/',
		'items': [init_req,
				TemplateRequest(
					re = stream_re2 % 'rtmp',
					encode_vars = lambda v: { 'final_url': '%(url)s swfVfy=1 swfUrl=http://www.svtplay.se/public/swf/video/svtplayer-2013.23.swf' % v,
											'quality': 'quality=%(bitrate)s kbps' % v,
											'suffix-hint': 'flv' })] }

hls = { 'items': [init_req,
				TemplateRequest(
					re = stream_re % ('http://', 'm3u8'),
					decode_content = download_clip,
					encode_vars = lambda v: { 'req_url': '%(url)s' % v }),
				TemplateRequest(
					re = r'BANDWIDTH=(?P<bitrate>\d+).*?RESOLUTION=(?P<resolution>\d+x\d+).*?(?P<url>https?://[^\n]+)',
					encode_vars = lambda v: { 'final_url': '%(url)s' % v,
												'quality': '%s kbps' % (str(int(v['bitrate'])/1000)),
												'suffix-hint': 'mp4' })] }

hds_fake = { 'items': [init_req,
					TemplateRequest(
						re = stream_re % ('http://', 'f4m'),
						encode_vars = lambda v: { 'req_url': '%s' % v['url'].replace('manifest.f4m', 'master.m3u8').replace('akamaihd.net/z/', 'akamaihd.net/i/') }),
					hls['items'][2]] }

hds = { 'startvars': { 'guid': ''.join(chr(65 + randint(0, 25)) for i in range(12)) }, 
		'items': [init_req,

# Path: lib/pirateplay/lib/rerequest.py
# class TemplateRequest:
# 	def __init__(self, re,
# 				encode_vars = lambda x: x, decode_content = lambda c, v: c,
# 				handlerchain = None):
# 		self.re = re
# 		self.encode_vars = encode_vars
# 		self.decode_content = decode_content
# 		self.handlerchain = handlerchain
# 	
# 	def create_vars(self, content, cumulated_vars):
# 		content = self.decode_content(content, cumulated_vars)
# 		
# 		#Make sure req_data and req_headers are empty
# 		if 'req_data' in cumulated_vars:
# 			del cumulated_vars['req_data']
# 		if 'req_headers' in cumulated_vars:
# 			del cumulated_vars['req_headers']
# 		
# 		for match in re.finditer(self.re, content, re.DOTALL):
# 			d = dict(cumulated_vars, **match.groupdict())
# 			d.update(self.encode_vars(d))
# 			yield d
# 	
# 	def create_req(self, cumulated_vars):
# 		url = cumulated_vars['req_url']
# 		if not url.startswith('http'):
# 			url = 'http://' + url
# 		req = urllib2.Request(url)
# 		
# 		if 'req_data' in cumulated_vars:
# 			debug_print('Adding post data to request: ' + cumulated_vars['req_data'])
# 			req.add_data(cumulated_vars['req_data'])
# 		
# 		for header, value in cumulated_vars.get('req_headers', {}).items():
# 			debug_print('Adding header to request: %s = %s' % (header, value))
# 			req.add_header(header, value)
# 		return req
# 	
# 	def create_content(self, req):
# 		debug_print('Opening URL: ' + req.get_full_url())
# 		try:
# 			try:
# 				f = urllib2.build_opener(self.handlerchain).open(req)
# 			except TypeError:
# 				f = urllib2.urlopen(req)
# 			
# 			content = f.read()
# 			f.close()
# 		except (urllib2.HTTPError, urllib2.URLError):
# 			content = ''
# 		
# 		return content
# 	
# 	def to_dict(self):
# 		return { 'test': self.re }
# 
# def debug_print(s):
# 	if DEBUG:
# 		print(' DEBUG ' + s)


# Path: lib/pirateplay/lib/services/svt.py
from ..rerequest import TemplateRequest, debug_print
from random import randint
from re import search
from urllib2 import HTTPError, urlopen


#Request chains

rtmp = { 'title': 'SVT-play', 'url': 'http://svtplay.se/',
		'items': [init_req,
				TemplateRequest(
					re = stream_re2 % 'rtmp',
					encode_vars = lambda v: { 'final_url': '%(url)s swfVfy=1 swfUrl=http://www.svtplay.se/public/swf/video/svtplayer-2013.23.swf' % v,
											'quality': 'quality=%(bitrate)s kbps' % v,
											'suffix-hint': 'flv' })] }

hls = { 'items': [init_req,
				TemplateRequest(
					re = stream_re % ('http://', 'm3u8'),
					decode_content = download_clip,
					encode_vars = lambda v: { 'req_url': '%(url)s' % v }),
				TemplateRequest(
					re = r'BANDWIDTH=(?P<bitrate>\d+).*?RESOLUTION=(?P<resolution>\d+x\d+).*?(?P<url>https?://[^\n]+)',
					encode_vars = lambda v: { 'final_url': '%(url)s' % v,
												'quality': '%s kbps' % (str(int(v['bitrate'])/1000)),
												'suffix-hint': 'mp4' })] }

hds_fake = { 'items': [init_req,
					TemplateRequest(
						re = stream_re % ('http://', 'f4m'),
						encode_vars = lambda v: { 'req_url': '%s' % v['url'].replace('manifest.f4m', 'master.m3u8').replace('akamaihd.net/z/', 'akamaihd.net/i/') }),
					hls['items'][2]] }

hds = { 'startvars': { 'guid': ''.join(chr(65 + randint(0, 25)) for i in range(12)) }, 
		'items': [init_req,

				TemplateRequest(