forked from add-ons/plugin.video.vrt.nu
-
Notifications
You must be signed in to change notification settings - Fork 0
/
tokenresolver.py
145 lines (125 loc) · 6.67 KB
/
tokenresolver.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
# -*- coding: utf-8 -*-
# GNU General Public License v2.0 (see COPYING or https://www.gnu.org/licenses/gpl-2.0.txt)
from resources.lib.helperobjects import helperobjects
import requests
import json
import datetime
import time
class TokenResolver:
_API_KEY = '3_qhEcPa5JGFROVwu5SWKqJ4mVOIkwlFNMSKwzPDAh8QZOtHqu6L4nD5Q7lk0eXOOG'
_LOGIN_URL = 'https://accounts.vrt.be/accounts.login'
_TOKEN_GATEWAY_URL = 'https://token.vrt.be'
_ONDEMAND_COOKIE = 'ondemand_vrtPlayerToken'
_LIVE_COOKIE = 'live_vrtPlayerToken'
_ROAMING_XVRTTOKEN_COOKIE = 'roaming_XVRTToken'
_XVRT_TOKEN_COOKIE = 'XVRTToken'
def __init__(self, kodi_wrapper):
self._kodi_wrapper = kodi_wrapper
self._proxies = self._kodi_wrapper.get_proxies()
def get_ondemand_playertoken(self, token_url, xvrttoken):
token_path = self._kodi_wrapper.get_userdata_path() + self._ONDEMAND_COOKIE
token = self._get_cached_token(token_path, 'vrtPlayerToken')
if token is None:
cookie_value = 'X-VRT-Token=' + xvrttoken
headers = {'Content-Type': 'application/json', 'Cookie': cookie_value}
token = self._get_new_playertoken(token_path, token_url, headers)
return token
def get_live_playertoken(self, token_url, xvrttoken):
token_path = self._kodi_wrapper.get_userdata_path() + self._LIVE_COOKIE
token = self._get_cached_token(token_path, 'vrtPlayerToken')
if token is None:
if xvrttoken is not None:
cookie_value = 'X-VRT-Token=' + xvrttoken
headers = {'Content-Type': 'application/json', 'Cookie' : cookie_value}
else:
headers = {'Content-Type': 'application/json'}
token = self._get_new_playertoken(token_path, token_url, headers)
return token
def get_xvrttoken(self, get_roaming_token=False):
token_filename = self._ROAMING_XVRTTOKEN_COOKIE if get_roaming_token else self._XVRT_TOKEN_COOKIE
token_path = self._kodi_wrapper.get_userdata_path() + token_filename
token = self._get_cached_token(token_path, 'X-VRT-Token')
if token is None:
token = self._get_new_xvrttoken(token_path, get_roaming_token)
return token
@staticmethod
def get_xvrttoken_from_cookiejar(cookiejar):
for cookie in cookiejar:
if cookie.name == 'X-VRT-Token':
yield cookie
def _get_new_playertoken(self, path, token_url, headers):
playertoken = requests.post(token_url, proxies=self._proxies, headers=headers).json()
json.dump(playertoken, open(path, 'w'))
return playertoken['vrtPlayerToken']
def _get_cached_token(self, path, token_name):
cached_token = None
if self._kodi_wrapper.check_if_path_exists(path):
token = json.loads(open(path, 'r').read())
now = datetime.datetime.utcnow()
exp = datetime.datetime(*(time.strptime(token['expirationDate'], '%Y-%m-%dT%H:%M:%S.%fZ')[0:6]))
if exp > now:
self._kodi_wrapper.log_notice('Got cached token')
cached_token = token[token_name]
else:
self._kodi_wrapper.log_notice('Cached token deleted')
self._kodi_wrapper.delete_path(path)
return cached_token
def _get_new_xvrttoken(self, path, get_roaming_token):
cred = helperobjects.Credentials(self._kodi_wrapper)
if not cred.are_filled_in():
self._kodi_wrapper.open_settings()
cred.reload()
data = {'loginID': cred.username, 'password': cred.password, 'sessionExpiration': '-1', 'APIKey': self._API_KEY, 'targetEnv': 'jssdk'}
logon_json = requests.post(self._LOGIN_URL, data, proxies=self._proxies).json()
token = None
if logon_json['errorCode'] == 0:
login_token = logon_json['sessionInfo']['login_token']
login_cookie = ''.join(('glt_', self._API_KEY, '=', login_token))
payload = {'uid': logon_json['UID'], 'uidsig': logon_json['UIDSignature'], 'ts': logon_json['signatureTimestamp'], 'email': cred.username}
headers = {'Content-Type': 'application/json', 'Cookie': login_cookie}
cookie_jar = requests.post(self._TOKEN_GATEWAY_URL, proxies=self._proxies, headers=headers, json=payload).cookies
xvrttoken = TokenResolver._create_token_dictionary(cookie_jar)
token = xvrttoken['X-VRT-Token']
if get_roaming_token:
xvrttoken = self._get_roaming_xvrttoken(login_cookie, xvrttoken)
token = xvrttoken['X-VRT-Token'] if xvrttoken is not None else None
json.dump(xvrttoken, open(path, 'w'))
else:
title = self._kodi_wrapper.get_localized_string(32051)
message = self._kodi_wrapper.get_localized_string(32052)
self._kodi_wrapper.show_ok_dialog(title, message)
return token
def _get_roaming_xvrttoken(self, login_cookie, xvrttoken):
url = 'https://token.vrt.be/vrtnuinitloginEU?destination=https://www.vrt.be/vrtnu/'
cookie_value = 'X-VRT-Token=' + xvrttoken['X-VRT-Token']
headers = {'Cookie': cookie_value}
r = requests.get(url, proxies=self._proxies, headers=headers, allow_redirects=False)
url = r.headers.get('Location')
r = requests.get(url, proxies=self._proxies, headers=headers, allow_redirects=False)
url = r.headers.get('Location')
headers = {'Cookie': login_cookie}
roaming_xvrttoken = None
if url is not None:
cookie_jar = requests.get(url, proxies=self._proxies, headers=headers).cookies
roaming_xvrttoken = TokenResolver._create_token_dictionary(cookie_jar)
return roaming_xvrttoken
@staticmethod
def _create_token_dictionary(cookie_jar):
token_dictionary = None
xvrttoken_cookie = next(TokenResolver.get_xvrttoken_from_cookiejar(cookie_jar))
if xvrttoken_cookie is not None:
token_dictionary = {
xvrttoken_cookie.name: xvrttoken_cookie.value,
'expirationDate': datetime.datetime.fromtimestamp(xvrttoken_cookie.expires).strftime('%Y-%m-%dT%H:%M:%S.%fZ'),
}
return token_dictionary
def reset_cookies(self):
user_data_path = self._kodi_wrapper.get_userdata_path()
ondemand = user_data_path + self._ONDEMAND_COOKIE
live = user_data_path + self._LIVE_COOKIE
xvrt = user_data_path + self._XVRT_TOKEN_COOKIE
roaming = user_data_path + self._ROAMING_XVRTTOKEN_COOKIE
self._kodi_wrapper.delete_path(ondemand)
self._kodi_wrapper.delete_path(live)
self._kodi_wrapper.delete_path(xvrt)
self._kodi_wrapper.delete_path(roaming)