}i'|dZddlZddlZddlmZmZddlmZmZddl m Z eje Z GddZy)a% This module contains all the logic needed to find the signature functions. YouTube's strategy to restrict downloading videos is to send a ciphered version of the signature to the client, along with the decryption algorithm obfuscated in JavaScript. For the clients to play the videos, JavaScript must take the ciphered version, cycle it through a series of "transform functions," and then signs the media URL with the output. This module is responsible for (1) finding these "transformations functions" (2) sends them to be interpreted by nodejs N)RegexMatchErrorInterpretationError) JSInterpreterextract_player_js_global_var) NodeRunnerczeZdZdedefdZdefdZdedefdZdededefd Zdedefd Ze d ed ede fd Z y)Cipherjsjs_urlc||_||_d|_d|_|j |||_|j |||_t||_ |jj|j t||_ |jj|jd|_ t||_y)N)r r _sig_param_val_nsig_param_valget_sig_function_namesig_function_nameget_nsig_function_namensig_function_namer runner_sig load_function runner_nsig calculated_nrjs_interpreter)selfr r s ;/usr/local/lib/python3.12/dist-packages/pytubefix/cipher.py__init__zCipher.__init__s "#!%!;!;B!G"&"="=b&"I$R. %%d&<&<=%b> &&t'>'>? +B/nc |jrA|jD]1}|jj||g}t|ts1nn|jj|g}dvsd|vst|tst |j||S#t $r}t |j|d}~wwxYw)a>Interpret the function that transforms the signature parameter `n`. The lack of this signature generates the 403 forbidden error. :param str n: Contains the parameter that must be transformed. :rtype: str :returns: Returns the transformed value "n". r reasonNerror_w8_)rrcall isinstancestr Exceptionrr )rrparamnsiges rget_nsigzCipher.get_nsig,s D##!11E++00%+?@ c>C!5%T[[E E  D%T[[C C DsAA?? B$BB$cgd}tjd|D]}tj|}|j |}|s,|j d}tjd|tjd|d|j vr#|j d}|rt||_|cStdd| ) a&Extract the name of the function responsible for computing the signature. :param str js: The contents of the base.js asset file. :param str js_url: Full base.js url :rtype: str :returns: Function name from regex match ) z(?P[a-zA-Z0-9_$]+)\s*=\s*function\(\s*(?P[a-zA-Z0-9_$]+)\s*\)\s*{\s*(?P=arg)\s*=\s*(?P=arg)\.split\(\s*[a-zA-Z0-9_\$\"\[\]]+\s*\)\s*;\s*[^}]+;\s*return\s+(?P=arg)\.join\(\s*[a-zA-Z0-9_\$\"\[\]]+\s*\)z(?:\b|[^a-zA-Z0-9_$])(?P[a-zA-Z0-9_$]{2,})\s*=\s*function\(\s*a\s*\)\s*{\s*a\s*=\s*a\.split\(\s*""\s*\)(?:;[a-zA-Z0-9_$]{2}\.[a-zA-Z0-9_$]{2}\(a,\d+\))?z\b(?P[a-zA-Z0-9_$]+)&&\((?P=var)=(?P[a-zA-Z0-9_$]{2,})\((?:(?P\d+),decodeURIComponent|decodeURIComponent)\((?P=var)\)\)zX\b[cs]\s*&&\s*[adf]\.set\([^,]+\s*,\s*encodeURIComponent\s*\(\s*(?P[a-zA-Z0-9$]+)\(zg\b[a-zA-Z0-9]+\s*&&\s*[a-zA-Z0-9]+\.set\([^,]+\s*,\s*encodeURIComponent\s*\(\s*(?P[a-zA-Z0-9$]+)\(z;\bm=(?P[a-zA-Z0-9$]{2,})\(decodeURIComponent\(h\.s\)\)z0("|\')signature\1\s*,\s*(?P[a-zA-Z0-9$]+)\(z!\.sig\|\|(?P[a-zA-Z0-9$]+)\(zyt\.akamaized\.net/\)\s*\|\|\s*.*?\s*[cs]\s*&&\s*[adf]\.set\([^,]+\s*,\s*(?:encodeURIComponent\s*\()?\s*(?P[a-zA-Z0-9$]+)\(z>\b[cs]\s*&&\s*[adf]\.set\([^,]+\s*,\s*(?P[a-zA-Z0-9$]+)\(zS\bc\s*&&\s*[a-zA-Z0-9]+\.set\([^,]+\s*,\s*\([^)]*\)\s*\(\s*(?P[a-zA-Z0-9$]+)\(z!looking for signature cipher namer-z"finished regex search, matched: %sz Signature cipher function name: r&get_initial_function_namez multiple in callerpattern) loggerdebugrecompilesearchgroup groupdictintr r) rr r function_patternsr3regexfunction_matchr-r&s rrzCipher.get_sig_function_name\s   89( GJJw'E"\\"-N$**51 A7K ?uEFn6688*009E.1%j+  .,vh8O  rctjd d}tj||}|r+|j d}tjd||Stjd|tjdt |\}}}|r|r|rtjd|t |j|id}tjd t|D]\} } | jd stjd | d tj|| fz}tj||}|rD|j d}tjd ||j|||_ |cStd|d|yyyy#t$r} | d} ~ wwxYw)aOExtract the name of the function that transforms the signature parameter `n`. :param str js: The contents of the base.js asset file. :param str js_url: Full base.js url :rtype: str :returns: The name of the function used to compute the throttling parameter. zlooking for nsig namez?var\s*[a-zA-Z0-9$_]{3}\s*=\s*\[(?P[a-zA-Z0-9$_]{3})\]funcnamez Nfunc name: z#Failed to get Nfunc name. Pattern: z4Extracts the function name based on the global arrayzGlobal Obj name is: dz&Successfully interpreted global objectr!z_w8_ found in index a(?xs) [;\n](?: (?Pfunction\s+)| (?:var\s+)? )(?P[a-zA-Z0-9_$]+)\s*(?(f)|=\s*function\s*) \(\s*(?:[a-zA-Z0-9_$]+\s*,\s*)?(?P[a-zA-Z0-9_$]+)(?:\s*,\s*[a-zA-Z0-9_$]+)*\s*\)\s*\{ (?:(?!(? '?G)+ '2(>I()2)D & vh-G H7;7S7STVX^7_ 4'- "1'ERYQZZ^_e^fOg#) /3':6 G s,AF2B0F2 B F2F22 G;F==GrLrHctjdtjdtj|dtj }g}|j |D]Z}|jddk(r"|jdr|jdn|jd}|j|\tjd||S)a$Extract the control parameter from the signature function `n`. :param str code: The contents of the base.js asset file. :param str func_name: function name :rtype: list :returns: The list with the extracted values. z!Looking for the control parameterz(?zh)\s*\[\w\[\d+\]\]\(\s*(?P[A-Za-z0-9_$]+)(?:\s*,\s*(?P[A-Za-z0-9_$]+))?(?:\s*,\s*[^)]*)?\s*\)arg1thisarg2zParameters found: ) r4r5r6r7rF MULTILINEfinditerr9append)rLrHr3resultsmchosens rrGzCipher._extract_nsig_param_vals  89**9-./  LL  !!$' #A()6(AaggfoQWWV_[\[b[bci[jF NN6 " #  )'34rN) __name__ __module__ __qualname__r$rr)r.rr staticmethodlistrGrrr r s0300(#4##,* * S* S* X66c6r!c!c!d!!rr )__doc__loggingr6pytubefix.exceptionsrrpytubefix.jsinterprrpytubefix.sig_nsig.node_runnerr getLoggerrYr4r r^rrres9  EJ5   8 $LLr