o i @sddlZddlZddlZddlZddlZddlZddlZddlZddlZddl m Z dZ dZ dZ dZd"d ed ed ed efd dZd ejjfddZdededejjd ejfddZddZddZded ejjdBfddZdejjdBfddZdededed ejjdBfd d!ZdS)#N) log_outputz&https://www.hover.com/signin/auth.jsonz'https://www.hover.com/signin/auth2.jsonz!https://www.hover.com/api/domainsz./cookies.jsonsecret time_stepdigitsreturnc Cs|}|dd}t|d}|r|dd|7}tj|dd}tt|}td|}t ||t j  }|dd @}td |||d d d @} | d|} | d|dS)zQ Generates a one-time passcode for TFA using the provided shared secret.  =T)casefoldz>Qz>Iri 0d)upperreplacelenbase64 b32decodeinttimestructpackhmacnewhashlibsha1digestunpack) rrrmissing_paddingkey current_time time_bytes hmac_resultoffsetbinaryotpr+>/home/seanl/mnt/public/Repos/hover-dyn-dns/./authentication.py generate_totps      r-cCs$tjddd}td|j|jS)z? Initializes a session with Hover to retrieve cookies. zhttps://www.hover.com/signinT)verifyz#Init session response status code: )requestsgetr status_codecookies)responser+r+r, init_session+sr4usernamepasswordr2cCs^||dd}tjt|d|d}td|td|jtd|jtd||S) zC Logs in to Hover with the provided username and password. N)r5r6tokenT)jsonr.r2 Payload: zLogin response status code: zLogin response content: Cookies: )r/post AUTH_ENDPOINTrr1content)r5r6r2 login_payloadr3r+r+r,submit_username_password4sr?cCsZd|i}tjt||dd}td|td|jtd|jtd||S)z9 Performs 2FA login with the provided TOTP code. codeT)r8r2r.r9z Login 2FA response status code: zLogin 2FA response content: r:)r/r;AUTH_2FA_ENDPOINTrr1r=) totp_coder2r>r3r+r+r,submit_2fa_codeEsrCcCs@ttd}t||WddS1swYdS)z' Saves cookies to a JSON file. wN)open COOKIES_FILEr8dumpget_dict)r2filer+r+r, save_cookiesTs "rJcookies_filenamecCsPtj|r&ttd}t|}tj |WdS1s!wYdS)z) Loads cookies from a JSON file. rN) ospathisfilerErFr8loadr/r2cookiejar_from_dict)rKrI cookies_dictr+r+r, load_cookies\s     rScCs&|durdStjt|dd}|jdkS)za Checks if the cookies are still valid by making a request to an authenticated endpoint. NFT)r2r.)r/r0DOMAIN_CHECK_URLr1)r2r3r+r+r,check_cookies_validitygs rV totp_secretc Cstdtt}t|rtd|St}zt|||}|j}|}Wntjy9tdt dt Ynwt |t sMtdt d|dt | d}|ratdt d|t | d d krk|}ni| d }|d krt|}zt||} | j}| } Wntjytd tdt Ynwt | t stdtd| dt | ddkrtd|d| t n|dkrn td|dt t|rtdt||Std| ddt dS)NzLogging in to hover...z8Skipping login process because we are already logged in.zCould not login: the endpoint z& did not return JSON data as expected.z returned z instead of valid JSON.errorz returned the following error: status completedtypeappz.Could not submit 2fa login code: the endpoint succeededTzCould not login: totp code z2 was not accepted. The server's response was was: z#Hover is asking for a 2FA type of 'z', which we cannot handle. In your account settings, disable 2FA and re-enable as app authentication. See the readme for more details.zLogin success.zXThe login process seemed to go well, but the server is not accepting our auth cookie of hoverauth.)rrSrFrVr4r?r2r8JSONDecodeErrorr<exit isinstancedictr0r-rCrArJ) r5r6rWr2login_responselogin_response_cookies login_error auth_typetotp totp_responser+r+r,loginqsf           rj)rr)r/requests.cookiesrrrrrr8rM logging_outrr<rArUrFstrrr-r2RequestsCookieJarr4Responser?rCrJrSrVrjr+r+r+r,s,   &