o i@svdZddlZddlZddlmZddlmZddlmZm Z ddl Z ddl m Z ddl mZd Zd ZGd d d ZdS) zDRM module is used to handle DRM operations with clock skew correction. Currently the only DRM operation is generating the Sec-MS-GEC token value used in all API requests to Microsoft Edge's online text-to-speech service.N)datetime)timezone)DictOptional)TRUSTED_CLIENT_TOKEN)SkewAdjustmentErrorl!l geAc@seZdZUdZdZeed<ededdfddZedefd d Z ed e de efd d Z ede jddfddZede fddZede fddZedee e fdee e ffddZdS)DRMzD Class to handle DRM operations with clock skew correction. gclock_skew_seconds skew_secondsreturnNcCstj|7_dS)aW Adjust the clock skew in seconds in case the system clock is off. This method updates the `clock_skew_seconds` attribute of the DRM class to the specified number of seconds. Args: skew_seconds (float): The number of seconds to adjust the clock skew to. Returns: None N)r r )r r F/home/kim/smarthome/.venv/lib/python3.10/site-packages/edge_tts/drm.pyadj_clock_skew_secondsszDRM.adj_clock_skew_secondscCsttjtjS)z Gets the current timestamp in Unix format with clock skew correction. Returns: float: The current timestamp in Unix format with clock skew correction. )dtnowtzutc timestampr r r r r rget_unix_timestamp+szDRM.get_unix_timestampdatecCs2zt|djtjdWStyYdSw)aN Parses an RFC 2616 date string into a Unix timestamp. This function parses an RFC 2616 date string into a Unix timestamp. Args: date (str): RFC 2616 date string to parse. Returns: Optional[float]: Unix timestamp of the parsed date string, or None if parsing failed. z%a, %d %b %Y %H:%M:%S %Z)tzinfoN)rstrptimereplacerrr ValueError)rr r rparse_rfc2616_date5s  zDRM.parse_rfc2616_dateecCs|jdur td||jdd}|dust|tstd|t|}|dus-t|ts5td||t}t ||dS)aS Handle a client response error. This method adjusts the clock skew based on the server date in the response headers and raises a SkewAdjustmentError if the server date is missing or invalid. Args: e (Exception): The client response error to handle. Returns: None NzNo server date in headers.DatezFailed to parse server date: ) headersrget isinstancestrr rfloatrr)rZ server_dateZserver_date_parsedZ client_dater r rhandle_client_response_errorKs    z DRM.handle_client_response_errorcCsNt}|t7}||d8}|td9}|dt}t|d S)a Generates the Sec-MS-GEC token value. This function generates a token value based on the current time in Windows file time format adjusted for clock skew, and rounded down to the nearest 5 minutes. The token is then hashed using SHA256 and returned as an uppercased hex digest. Returns: str: The generated Sec-MS-GEC token value. See Also: https://github.com/rany2/edge-tts/issues/290#issuecomment-2464956570 i,dz.0fascii) r r WIN_EPOCHS_TO_NSrhashlibsha256encode hexdigestupper)ZticksZ str_to_hashr r rgenerate_sec_ms_gecfs   zDRM.generate_sec_ms_geccCstdS)za Generates a random MUID. Returns: str: The generated MUID. )secretsZ token_hexr,r r r r generate_muidszDRM.generate_muidrcCs,|}d|vs Jdtd|d<|S)z Returns a copy of the given headers with the MUID header added. Args: headers (dict): The original headers. Returns: dict: The headers with the MUID header added. Cookiezmuid=;)copyr r0)rZcombined_headersr r rheaders_with_muids zDRM.headers_with_muid)__name__ __module__ __qualname____doc__r r"__annotations__ staticmethodrrr!rraiohttpZClientResponseErrorr#r-r0rr4r r r rr s"   ! (r )r8r(r/rrrrtypingrrr; constantsr exceptionsrr&r'r r r r rs