import sys
import xbmcgui
import xbmcplugin
import requests
import re

addon_handle = int(sys.argv[1])

url = 'https://ok.ru/videoembed/' + sys.argv[2][1:]
regex = [
'https://vd\\d+.mycdn.me/expires/\\d+.*m3u8',
'https://vd\\d+.mycdn.me/video.m3u8.*?id=\\d+'
]


def get_stream(url):
	xbmc.log(msg="OKRU: STARTING..." + url, level=xbmc.LOGINFO)


	r = requests.get(url)
	xbmc.log(msg=str(len(r.text)), level=xbmc.LOGINFO)
	if (r is None):
		xbmc.log(msg='OKRU: Error when sending request.', level=xbmc.LOGDEBUG)

	url_list = list(filter(lambda x: x is not None, map(lambda x: re.search(x, r.text), regex)))
	if (len(url_list) == 0):
		xbmc.log(msg='OKRU: Couldnt get any valid stream from html.', level=xbmc.LOGDEBUG)
	else:
		return url_list[0].group(0).replace('\\\\u0026', '&') + '&bytes='


url = get_stream(url)
li = xbmcgui.ListItem(path=url)
xbmcplugin.setResolvedUrl(addon_handle, True, listitem=li)