#!/usr/bin/python3 import sys import os import o11 import json import datetime import pytz import base64 import uuid import jwt from pywidevine.cdm import Cdm from pywidevine.device import Device from pywidevine.pssh import PSSH from bs4 import BeautifulSoup WVD_PATH = './WVD.wvd' user = o11.parse_params(sys.argv, 'user') password = o11.parse_params(sys.argv, 'password') id = o11.parse_params(sys.argv, 'id') action = o11.parse_params(sys.argv, 'action') bind = o11.parse_params(sys.argv, 'bind') proxy = o11.parse_params(sys.argv, 'proxy') doh = o11.parse_params(sys.argv, 'doh') worker = o11.parse_params(sys.argv, 'worker') cdm_param = o11.parse_params(sys.argv, 'cdm') challenge = o11.parse_params(sys.argv, 'challenge') o11Session = o11.session(bind=bind, proxy=proxy, worker=worker) req = o11Session.get_session() if doh != "": o11.dns(doh) if challenge == "cert": challenge = "CAQ=" authFile = '/WaipuTV_' + user + '.tokens' user_agent = 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/121.0.0.0 Safari/537.36' def do_cdm_external(pssh_data, token): pssh_obj = PSSH(pssh_data) device_obj = Device.load(WVD_PATH) cdm_obj = Cdm.from_device(device_obj) session_id = cdm_obj.open() challenge_data = cdm_obj.get_license_challenge(session_id, pssh_obj) decoded = jwt.decode(token, options={"verify_signature": False}) cdata = '{"userId":"' + decoded["userHandle"] + '","sessionId":"default","merchant":"exaring"}' lic_headers = {'dt-custom-data': base64.b64encode(cdata.encode()).decode(), 'origin': 'https://play.waipu.tv', 'referer': 'https://play.waipu.tv/', 'user-agent': user_agent, 'content-type': 'application/octet-stream'} licence = req.post('https://drm.wpstr.tv/license-proxy-widevine/cenc/?specConform=true', headers=lic_headers, data=challenge_data) cdm_obj.parse_license(session_id, licence.content) keys = [f"{key.kid.hex}:{key.key.hex()}" for key in cdm_obj.get_keys(session_id) if key.type != 'SIGNING'] cdm_obj.close(session_id) return keys def get_pssh_from_mpd(url): headers = {'User-Agent': user_agent} response = req.get(url, headers=headers) for cp in BeautifulSoup(response.content, features="xml").findAll('ContentProtection'): if cp.get('schemeIdUri', '').lower() == 'urn:uuid:edef8ba9-79d6-4ace-a3c8-27dcd51d21ed': pssh_tag = cp.find('cenc:pssh') if pssh_tag: return pssh_tag.text return None def get_channels(token): headers = {'authorization': 'Bearer ' + token, 'origin': 'https://play.waipu.tv', 'referer': 'https://play.waipu.tv/', 'user-agent': user_agent} params = {'omitted': 'false'} response = req.get('https://user-stations.waipu.tv/api/stations', params=params, headers=headers) return response.json() def get_single(token, station): headers = {'accept': 'application/vnd.streamurlprovider.newtv-stream-url-v1+json', 'authorization': 'Bearer ' + token, 'content-type': 'application/vnd.streamurlprovider.stream-url-request-v1+json', 'origin': 'https://play.waipu.tv', 'referer': 'https://play.waipu.tv/', 'user-agent': user_agent} data = json.dumps({"stream": {"station": station, "protocol": "dash", "requestMuxInstrumentation": True, "processOutcomeField": True}}) response = req.post('https://stream-url-provider.waipu.tv/api/stream-url', headers=headers, data=data) return response.json().get('streamUrl', '') def do_token_refresh(refresh_token): headers = {'content-type': 'application/x-www-form-urlencoded', 'origin': 'https://play.waipu.tv', 'referer': 'https://play.waipu.tv/', 'user-agent': user_agent} data = {'waipu_device_id': str(uuid.uuid4()), 'grant_type': 'refresh_token', 'client_id': 'waipu.web-player', 'refresh_token': refresh_token} response = req.post('https://auth.waipu.tv/oauth/token', headers=headers, data=data) resp_data = response.json() return resp_data.get('access_token', ''), resp_data.get('refresh_token', '') def login(): print("Please provide refresh_token manually in the tokens file", file=sys.stderr) print("Login to WaipuTV website, go to DevTools -> Network and find refresh_token in oauth/token response", file=sys.stderr) sys.exit(1) def do_action(): if action == "login": login() sys.exit() try: auth = json.load(open(os.path.abspath(os.path.dirname(__file__)) + authFile)) token, new_refresh = do_token_refresh(auth['refresh_token']) auth['refresh_token'] = new_refresh json.dump(auth, open(os.path.abspath(os.path.dirname(__file__)) + authFile, 'w')) except: return "error" if action == "channels": output = {'Channels': []} for c in get_channels(token): output['Channels'].append({'Name': c['displayName'], 'Mode': "live", 'SessionManifest': True, 'ManifestScript': 'id=' + c['stationId'], 'CdmType': "widevine", 'UseCdm': True, 'Cdm': 'id=' + c['stationId'], 'Video': 'best'}) print(json.dumps(output, indent=2)) elif action == "events": output = {'Events': []} for c in get_channels(token): output['Events'].append({'Name': c['displayName'], 'Mode': "live", 'SessionManifest': True, 'ManifestScript': 'id=' + c['stationId'], 'CdmType': "widevine", 'UseCdm': True, 'Cdm': 'id=' + c['stationId'], 'Video': 'best', 'Autostart': True, 'Start': int(datetime.datetime.now(pytz.UTC).timestamp()), 'End': int((datetime.datetime.now(pytz.UTC) + datetime.timedelta(hours=4)).timestamp())}) print(json.dumps(output, indent=2)) elif action == "manifest": station_id = id.replace('id=', '') if id.startswith('id=') else id url = get_single(token, station_id) pssh_data = get_pssh_from_mpd(url) output = {"Cdn": [], "ManifestUrl": url, "Headers": {"Manifest": {'User-Agent': user_agent}, "Media": {'User-Agent': user_agent}}, "Token": token} if pssh_data: output['Pssh'] = pssh_data print(json.dumps(output)) elif action == "cdm" and cdm_param == "external": station_id = id.replace('id=', '') if id.startswith('id=') else id url = get_single(token, station_id) pssh_data = get_pssh_from_mpd(url) if pssh_data: for key in do_cdm_external(pssh_data, token): print(key) if do_action() == "error": print("Error: Please create token file with refresh_token", file=sys.stderr) sys.exit(1)