e d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dlm Z m
Z
d dl m Z m
Z
mZ d dlZ e j d Z edg d Zd Zd
Z G d d e Z G d
d ZdS ) N)
namedtuple)partial)BaseHTTPRequestHandler
HTTPServer)parse_qs urlencodeurlparserepo_cliConfig) client_idport
authorize_url token_url redirect_uriserver
verify_ssla~
Anaconda Server - Token received
Token received
We successfully received a token. You can close the browser window and continue to use the CLI
— Anaconda
a
Anaconda Server - Not Found
404
Not Found
c < e Zd Z fdZd Zd Zd Zd Zd Z xZ S )RequestHandlerc H || _ t j |i | d S N)configsuper__init__)selfr argskwargs __class__s @lib/python3.11/site-packages/repo_cli/utils/local_auth_server.pyr zRequestHandler.__init__W s, $)&))))) c d| j v r]t | j j d| j j d }d | j j | }| | S d| j v rIt t | j j
}|d d }| | S | dt d S )
a The main handler for the auth server.
To initiate the flow, we are openining the browser at http://localhost:PORT/get_token, which
generates the redirect header to the OpenID Connect Provider/
Once user is logged in on OpenID Connect provider, it's redirected back to the
http://localhost:PORT/?code=CODE&.... location, query string contains the authorization
code in `code` parameter.
Next we need to exchange the code for a jwt token, calling the token endpoint.
/get_tokencode)r
response_typer z{authorize_url}?{qs})r qszcode=r i N)pathr r r r formatr redirectr r query process_coderesponseHTML_404_RESPONSE)r r$ redirect_urlr" s r do_GETzRequestHandler.do_GET[ s 49 $ $!%!6%+$(K$< B 288 "k 7B 9 L == .. . di (49--344Bf:a=D$$T** *
c,-----r c | j j }d|| j j | j j d}t j ||| j j }| dt |j dk rF|
d }|| j j _ t d n"t d|||j t# j dS ) aJ Exchanges the authorization code, send back by OpenID Connect provider for the token,
calling the token endpoint with the proper client_id and redirect_uri.
Args:
code (str): authorization_code string
Returns:
None, stops the webserver by exiting and stopping the thread.
authorization_code)
grant_typer" r r verify access_tokenzReceived access_tokenz$Accessed %s with %s and got error %sN)r r r r requestspostr r* HTML_SUCCESS_RESPONSE status_codejsonr r4 loggerinfoerrortextsysexit)r r" r datarr4 s r r) zRequestHandler.process_code{ s K) . . K 4
M)T$+2H I I I
c0111 =C 6688N3L.:DK + KK/ 0 0 0 0 LL6 4
r c | d | d| | d S )Ni. Location)
send_response send_header end_headers)r locations r r' zRequestHandler.redirect sD 3 X... r c | | | dd | | j | d S )Nz Content-Typez text/html)rD rE rF wfilewriteencode)r statusr@ s r r* zRequestHandler.response sb 6""" 555
'''''r c d S r )r r& r s r log_messagezRequestHandler.log_message s r )
__name__
__module__ __qualname__r r- r) r' r* rO
__classcell__)r s @r r r V s * * * * * . . .@ <
( ( (
r r c ` e Zd Zd dZed Zej d Zd Zd Zd
dZ dS ) WebServerNc || _ || _ t j || j }|d | _ |d | _ ||| _ n| | _ d | _ d S )Nr1 authorization_endpointtoken_endpoint)
r r r5 getr9 r r r
find_unused_port
_access_token)r r openid_configuration_urlr
r openid_confs r r zWebServer.__init__ s "$ l $T_
$&& ))A B$%56 DII --//DI !r c | j S r r[ )r s r r4 zWebServer.access_token s !!r c |r || _ d S d S r r_ )r values r r4 zWebServer.access_token s" '!&D ' 'r c ( d| j f}t | j | j | j | j | d | | j }t t | }t || }t j |j }d|_
|S )N /)r
r r r r r r )targetT)r
r r r r
localhost_urlr r r r threadingThread
serve_foreverdaemon)r server_addressr handlerr threads r startzWebServer.start s di n ,n ++C00
.&11NG44!)=>>>
r c t j t j t j 5 }| d | d cddd S # 1 swxY w Y dS )zLReturns an unused port number on localhost. Will be used for webserver port.)z 127.0.0.1r N)socketAF_INET SOCK_STREAMbind getsockname)r ss r rZ zWebServer.find_unused_port s
]6>6+=
>
> &!
FF# $ $ $==??1% & & & & & & & & & & & & & & & & & &s /A&&A*-A*r! c : d | j | S )Nz http://localhost:{port}{path})r
r% )r& r
)r r% s r rf zWebServer.localhost_url s .554945PPPr )NN)r! )
rP rQ rR r propertyr4 setterrn rZ rf rN r r rU rU s
"
"
"
" " " X" ' ' ' "& & & Q Q Q Q Q Qr rU )loggingrq r> rg collectionsr functoolsr http.serverr r urllib.parser r r r5 getLoggerr: r r7 r+ r rU rN r r r sF
" " " " " " : : : : : : : : 6 6 6 6 6 6 6 6 6 6 : & &
: :O
O
O
O
O
+ O
O
O
d1Q 1Q 1Q 1Q 1Q 1Q 1Q 1Q 1Q 1Qr