U g@@sddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z m Z Gddde ZGdd d e ZGd d d e ZGd d d ZdS)N) InfoExtractor)ExtractorError traverse_obj unescapeHTMLc @seZdZdZdZddddddd d d d d ddddddd dddddddddddddid dgZdZd d!Zd"d#Zd$d%Z dS)&GoPlayIEzGhttps?://(www\.)?goplay\.be/video/([^/]+/[^/]+/|)(?P[^/#]+)Zgoplayzjhttps://www.goplay.be/video/de-container-cup/de-container-cup-s3/de-container-cup-s3-aflevering-2#autoplayz$9c4214b8-e55d-4e4b-a446-f015f6c6f811mp4zS3 - Aflevering 2zDe Container CupzSeason 3z Episode 2r)idexttitleseriesseason season_numberepisodeepisode_numberz1This video is only available for registered users)url info_dictskipzNhttps://www.goplay.be/video/a-family-for-thr-holidays-s1-aflevering-1#autoplayz$74e3ed07-748c-49e4-85a0-393a93337dbfzA Family for the Holidays)r r r zNhttps://www.goplay.be/video/de-mol/de-mol-s11/de-mol-s11-aflevering-1#autoplayz$03eb8f2f-153e-41cb-9805-0d3a29dab656zS11 - Aflevering 1z Episode 1zDe Mol rz Season 11)r r r rrrrrZ skip_downloadT)rrparamsrNcCs.|t|ddd}|j||d\|_}dS)Nzeu-west-1_dViSsKM5YZ6s1h851s8uplco5h6mqh1jac8m)iepool_id client_id)usernamepassword)Z report_loginAwsIdp authenticate _id_token)selfrrZaws_r";/tmp/pip-unpacked-wheel-q5ljy6pj/yt_dlp/extractor/goplay.py_perform_login?szGoPlayIE._perform_logincCs|js|jdddS)Nr)method)rZraise_login_requiredr r"r"r#_real_initializeDszGoPlayIE._real_initializecs|dd\}||}|d|d}|t||d}|d}|rl|d}d|di}nLt|d d d fd d fdd} | d}| dt| d| d| dd}|jd||dd|j i| d} d| kr |j | dd|ddd\} } nzd| krt d| dd } | dd!}|jd"| d#|d$|d%d&d'id(}| |d)|}d*d+|D}||\} } ||| | d,|S)-Nr display_idzUz(GoPlayIE._real_extract..F)get_allZ episodeTitle)programr Z seasonNumberZ episodeNumber)r rrrz.https://api.goplay.be/web/v1/videos/long-form/ AuthorizationzBearer )headersZ manifestUrlsZhlsr ZHLS)r Zm3u8_idZssaizexpecting Google SSAI streamZcontentSourceIDZvideoIDz-https://dai.google.com/ondemand/dash/content/z/vid/z/streamss{"api-key":"null"}z content-typezapplication/json)r*r3Zstream_manifestcSsg|]}d|dkr|qS)z-ad-r r").0pr"r"r# xs z*GoPlayIE._real_extract..)r formats subtitles)Z_match_valid_urlgroupZ_download_webpageZ_html_search_regexZ _parse_jsonrgetr_download_jsonrZgeo_verification_headersZ#_extract_m3u8_formats_and_subtitlesrZ_extract_mpd_periodsZ_merge_mpd_periodsupdate)r rr(ZwebpageZvideo_data_jsonr)r+Zvideo_idrrapir7r8Zssai_content_source_idZ ssai_video_idZdaiZperiodsr"r-r# _real_extractHsh         zGoPlayIE._real_extract) __name__ __module__ __qualname__Z _VALID_URLZ_NETRC_MACHINEZ_TESTSrr$r'r>r"r"r"r#rsR   'rc@seZdZdZdS)InvalidLoginExceptionz# The login credentials are invalid Nr?r@rA__doc__r"r"r"r#rBsrBc@seZdZdZdS)AuthenticationExceptionz' Something went wrong while logging in NrCr"r"r"r#rEsrEc@seZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ e ddZe ddZe ddZe ddZe ddZe ddZe d d!Zd"d#Zd$S)%rz AWS Identity Provider cCs||_||_d|jkrtd||_|jdd|_d|jd|_d|_d|_t dd |_ | |j|_ | |j|_ | |d |jd |j|_||_||_d S) aM :param InfoExtrator ie: The extractor that instantiated this class. :param str pool_id: The AWS user pool to connect to (format: _). E.g.: eu-west-1_aLkOfYN3T :param str client_id: The client application ID (the ID of the application connecting) r!z4Invalid pool_id format. Should be _.rzhttps://cognito-idp.z.amazonaws.com/AFFFFFFFFFFFFFFFFC90FDAA22168C234C4C6628B80DC1CD129024E088A67CC74020BBEA63B139B22514A08798E3404DDEF9519B3CD3A431B302B0A6DF25F14374FE1356D6D51C245E485B576625E7EC6F44C42E9A637ED6B0BFF5CB6F406B7EDEE386BFB5A899FA5AE9F24117C4B1FE649286651ECE45B3DC2007CB8A163BF0598DA48361C55D39A69163FA8FD24CF5F83655D23DCA3AD961C62F356208552BB9ED529077096966D670C354E4ABC9804F1746C08CA18217C32905E462E36CE3BE39E772C180E86039B2783A2EC07A28FB5C55DF06F4C52C9DE2BCBF6955817183995497CEA956AE515D2261898FA051015728E5A8AAAC42DAD33170D04507A33A85521ABDF1CBA64ECFB850458DBEF0A8AEA71575D060C7DB3970F85A6E1E4C7ABF5AE8CDB0933D71E8C94E04A25619DCEE3D2261AD2EE6BF12FFA06D98A0864D87602733EC86A64521F2B18177B200CBBE117577A615D6C770988C0BAD946E208E24FA074E5AB3143DB5BFCE0FD108E4B82D120A93AD2CAFFFFFFFFFFFFFFFF2zCaldera Derived Keyutf-8000N)rr ValueErrorrsplitregionrZn_hexZg_hex bytearray info_bits_AwsIdp__hex_to_longbig_ng_AwsIdp__hex_hashk _AwsIdp__generate_random_small_a small_a_value_AwsIdp__calculate_a large_a_value)r rrrr"r"r#__init__s   " zAwsIdp.__init__c Cs||}t|}dddd}|jj|jd||ddd}|d }|d d krbt|d | ||}t|} d dd} |jj|jd| | ddd}d |krt |d |dd|ddfS)z, Authenticate with a username and password. z.AWSCognitoIdentityProviderService.InitiateAuthidentityzapplication/x-amz-json-1.1) X-Amz-TargetzAccept-Encoding Content-TypeNzAuthenticating usernamezInvalid username)r*r3ZnoteZerrnoteZChallengeParameters ChallengeNamePASSWORD_VERIFIERmessagez8AWSCognitoIdentityProviderService.RespondToAuthChallenge)rZr[zAuthenticating passwordzInvalid passwordZAuthenticationResultZIdTokenZ RefreshToken) #_AwsIdp__get_authentication_requestjsondumpsencoderr;rr:rE'_AwsIdp__get_challenge_response_requestrB) r rrZauth_data_dictZ auth_dataZ auth_headersZauth_response_jsonchallenge_parametersZchallenge_requestZchallenge_dataZchallenge_headersr"r"r#rsF       zAwsIdp.authenticatecCs|||jdd|jdS)z :param str username: The username to use :return: A full Authorization request. :rtype: dict )USERNAMEZSRP_AZ USER_SRP_AUTH)ZAuthParametersZAuthFlowClientId)_AwsIdp__long_to_hexrWr)r rr"r"r#Z__get_authentication_requests  z#AwsIdp.__get_authentication_requestcCs|d}|d}|d}|d}|d}|}||||||} t|} t|jdddt|dt| t|d} tj | | t j d } t | d} |||| d d |jd S) aH Create a Challenge Response Request object. :param dict[str,str|imt] challenge_parameters: The parameters for the challenge. :param str password: The password. :return: A valid and full request data object to use as a response for a challenge. :rtype: dict reZUSER_ID_FOR_SRPZSRP_BZSALTZ SECRET_BLOCKr!rrG) digestmod)re TIMESTAMPZPASSWORD_CLAIM_SECRET_BLOCKZPASSWORD_CLAIM_SIGNATUREr])ZChallengeResponsesr\rf)_AwsIdp__get_current_timestamp"_AwsIdp__get_hkdf_key_for_passwordrObase64standard_b64decoderMrrKhmacnewhashlibsha256standard_b64encodedigestdecoder)r rdrZuser_idZuser_id_for_srpZsrp_bsaltZ secret_block timestampZhkdfZsecret_block_bytesmsgZhmac_objZsignature_stringr"r"r#Z __get_challenge_response_requests>  z'AwsIdp.__get_challenge_response_requestc Cs||j|}|dkrtdd|jdd||}||}|| | ||}t |j ||j } ||j| } t | |j|||j } |t| | t| ||S)ab Calculates the final hkdf based on computed S value, and computed U value and the key. :param str username: Username. :param str password: Password. :param int server_b_value: Server B value. :param int salt: Generated salt. :return Computed HKDF value. :rtype: object rzU cannot be zero.z{}{}:{}r!r)_AwsIdp__calculate_urWrJformatrrK_AwsIdp__hash_sha256rbrOrR_AwsIdp__pad_hexpowrQrPrSrU_AwsIdp__compute_hkdfrMfromhexrg) r rrZserver_b_valueruZu_valueZusername_passwordZusername_password_hashZx_valueZ g_mod_pow_xnZ int_value2Zs_valuer"r"r#Z__get_hkdf_key_for_password$s z"AwsIdp.__get_hkdf_key_for_passwordcCsHt||tj}|jttdd}t||tj}|ddS)z Standard hkdf algorithm :param {Buffer} ikm Input key material. :param {Buffer} salt Salt value. :return {Buffer} Strong key material. rrGN)rnrorprqrsrNrMchr)r ZikmruZprkZinfo_bits_updateZ hmac_hashr"r"r#Z__compute_hkdf?szAwsIdp.__compute_hkdfcCs$|||||}||S)z Calculate the client's value U which is the hash of A and B :param int big_a: Large A value. :param int big_b: Server B value. :return Computed U value. :rtype: int )rRr{rO)r big_aZbig_bZ u_hex_hashr"r"r#Z __calculate_uLs zAwsIdp.__calculate_ucCs|d}||jS)zo Helper function to generate a random big integer :return a random value. :rtype: int )_AwsIdp__get_randomrP)r Zrandom_long_intr"r"r#Z__generate_random_small_aYs z AwsIdp.__generate_random_small_acCs,t|j|j|j}||jdkr(td|S)z Calculate the client's public value A = g^a%N with the generated random number a :return Computed large A. :rtype: int rzSafety check for A failed)r|rQrUrPrJ)r rr"r"r#Z __calculate_abszAwsIdp.__calculate_acCs|dS)Nxr")Zlong_numr"r"r#Z __long_to_hexoszAwsIdp.__long_to_hexcCs t|dS)Nr)intZ hex_stringr"r"r#Z __hex_to_longsszAwsIdp.__hex_to_longcCstt|SN)rrzrMr~rr"r"r#Z __hex_hashwszAwsIdp.__hex_hashcCs"t|}dt|d|S)zAuthenticationHelper.hash@rI)rprq hexdigestlen)bufrsr"r"r#Z __hash_sha256{szAwsIdp.__hash_sha256cCsPt|tst|}n|}t|ddkr6d|}n|ddkrLd|}|S)z Converts a Long integer (or hex string) to hex format padded with zeroes for hashing :param int|str long_int: Number or string to pad. :return Padded hex string. :rtype: str rrrIrZ89ABCDEFabcdefrH) isinstancestrrrgr)Zlong_intZhash_strr"r"r#Z __pad_hexs     zAwsIdp.__pad_hexcCstt|}t|Sr)binasciihexlifyosurandomrrO)nbytesZ random_hexr"r"r#Z __get_randomszAwsIdp.__get_randomc Cspddddddddd d d d d g }dddddddg}tjtjj}||d||jd|jd}||S)z Creates a timestamp with the correct English format. :return: timestamp in format 'Sun Jan 27 19:00:04 UTC 2019' :rtype: str NJanFebMarAprMayJunJulAugSepOctNovDecMonTueWedThuFriSatSun z %H:%M:%S UTC %Y) dtdatetimenowtimezoneutcweekdaymonthdaystrftime)monthsdaysZtime_now format_stringr"r"r#Z__get_current_timestamps &zAwsIdp.__get_current_timestampcCsd|j|jdd|jS)Nz4AWS IDP Client for: Region: {} PoolId: {} AppId: {}r!r)ryrLrrKrr&r"r"r#__str__s zAwsIdp.__str__N)r?r@rArDrXrr_rcrkr}rxrTrV staticmethodrgrOrRrzr{rrjrr"r"r"r#rs20$-           r)rlrrrrprnr`rcommonrutilsrrrrrBrErr"r"r"r#s v