o
    沪gpM                  	   @  s&  U d dl mZ d dlmZmZ d dlmZmZmZm	Z	 d dl
mZmZ d dlmZmZmZmZ d dlmZmZ d dlmZ d dlmZ d d	lmZ d d
lmZ erWd dlmZ dZde d< dZ!de d< edd"d#ddZ"edd$ddZ#d%ddZ$d&ddZ%G d d! d!ee&e	e&e'df f Z(dS )'    )annotations)IteratorMapping)TYPE_CHECKINGFinalNoReturnUnion)configruntime)encode_provider_tokenget_secrets_auth_sectionis_authlib_installedvalidate_auth_credentials)StreamlitAPIExceptionStreamlitAuthError)
ForwardMsg)gather_metrics)get_script_run_ctx)make_url_path)UserInfoz/auth/loginr   AUTH_LOGIN_ENDPOINTz/auth/logoutAUTH_LOGOUT_ENDPOINTloginNprovider
str | NonereturnNonec                 C  sT   | du rd} t  }|dur(t stdt|  t }t| |j_|| dS dS )aY%  Initiate the login flow for the given provider.

    This command redirects the user to an OpenID Connect (OIDC) provider. After
    the user authenticates their identity, they are redirected back to the
    home page of your app. Streamlit stores a cookie with the user's identity
    information in the user's browser . You can access the identity information
    through |st.experimental_user|_. Call ``st.logout()`` to remove the cookie
    and start a new session.

    You can use any OIDC provider, including Google, Microsoft, Okta, and more.
    You must configure the provider through secrets management. Although OIDC
    is an extension of OAuth 2.0, you can't use generic OAuth providers.
    Streamlit parses the user's identity token and surfaces its attributes in
    ``st.experimental_user``. If the provider returns an access token, that
    token is ignored. Therefore, this command will not allow your app to act on
    behalf of a user in a secure system.

    For all providers, there are two shared settings, ``redirect_uri`` and
    ``cookie_secret``, which you must specify in an ``[auth]`` dictionary
    in ``secrets.toml``. Other settings must be defined as described in the
    ``provider`` parameter.

    - ``redirect_uri`` is your app's absolute URL with the pathname
      ``oauth2callback``. For local development using the default port, this is
      ``http://localhost:8501/oauth2callback``.
    - ``cookie_secret`` should be a strong, randomly generated secret.

    In addition to the shared settings, the following settings are required:

    - ``client_id``
    - ``client_secret``
    - ``server_metadata_url``

    For a complete list of OIDC parameters, see `OpenID Connect Core
    <https://openid.net/specs/openid-connect-core-1_0.html#AuthRequest>`_ and
    your provider's documentation. By default, Streamlit sets
    ``scope="openid profile email"`` and ``prompt="select_account"``. You can
    change these and other OIDC parameters by passing a dictionary of settings
    to ``client_kwargs``. ``state`` and ``nonce``, which are used for
    security, are handled automatically and don't need to be specified. For
    more information, see Example 4.

    .. Important::
        - You must install ``Authlib>=1.3.2`` to use this command.
        - Your authentication configuration is dependent on your host location.
          When you deploy your app, remember to update your ``redirect_uri``
          within your app and your provider.
        - All URLs declared in the settings must be absolute (i.e., begin with
          ``http://`` or ``https://``).
        - Streamlit automatically enables CORS and XSRF protection when you
          configure authentication in ``secrets.toml``. This takes precedence
          over configuration options in ``config.toml``.
        - If a user is logged into your app and opens a new tab in the same
          browser, they will automatically be logged in to the new session with
          the same account.
        - If a user closes your app without logging out, the identity cookie
          will expire after 30 days.
        - For security reasons, authentication is not supported for embedded
          apps.

    .. |st.experimental_user| replace:: ``st.experimental_user``
    .. _st.experimental_user: https://docs.streamlit.io/develop/api-reference/utilities/st.experimental_user

    Parameters
    ----------
    provider: str or None
        The name of your provider configuration to use for login.

        If ``provider`` is ``None`` (default), Streamlit will use all settings
        in the ``[auth]`` dictionary within your app's ``secrets.toml`` file.
        Otherwise, use an ``[auth.{provider}]`` dictionary for the named
        provider, as shown in the examples that follow. When you pass a string
        to ``provider``, Streamlit will use ``redirect_uri`` and
        ``cookie_secret``, while ignoring any other values in the ``[auth]``
        dictionary.

    Examples
    --------

    **Example 1: Use an unnamed default identity provider**

    If you do not specify a name for your provider, specify all settings within
    the ``[auth]`` dictionary of your ``secrets.toml`` file. The following
    example configures Google as the default provider. For information about
    using OIDC with Google, see `Google Identity
    <https://developers.google.com/identity/openid-connect/openid-connect>`_.

    ``.streamlit/secrets.toml``:

    >>> [auth]
    >>> redirect_uri = "http://localhost:8501/oauth2callback"
    >>> cookie_secret = "xxx"
    >>> client_id = "xxx"
    >>> client_secret = "xxx"
    >>> server_metadata_url = (
    ...     "https://accounts.google.com/.well-known/openid-configuration"
    ... )

    Your app code:

    >>> import streamlit as st
    >>>
    >>> if not st.experimental_user.is_logged_in:
    >>>     if st.button("Log in"):
    >>>         st.login()
    >>> else:
    >>>     if st.button("Log out"):
    >>>         st.logout()
    >>>     st.write(f"Hello, {st.experimental_user.name}!")

    **Example 2: Use a named identity provider**

    If you specify a name for your provider, save the shared settings in the
    ``[auth]`` dictionary of your ``secrets.toml`` file, and save the other
    settings in an ``[auth.{provider}]`` dictionary, where ``{provider}`` is
    the name of your provider. The following example configures Microsoft as
    the provider. The example uses ``provider="microsoft"``, but you can use
    any name. This name is internal to Streamlit and is used to match the login
    command to its configuration. For information about using OIDC with
    Microsoft, see `Microsoft Entra ID
    <https://learn.microsoft.com/en-us/power-pages/security/authentication/openid-settings>`_.
    To configure your ``{tenant}`` value in ``server_metadata_url``, see
    `Microsoft identity platform
    <https://learn.microsoft.com/en-us/entra/identity-platform/v2-protocols-oidc#find-your-apps-openid-configuration-document-uri>`_.

    ``.streamlit/secrets.toml``:

    >>> [auth]
    >>> redirect_uri = "http://localhost:8501/oauth2callback"
    >>> cookie_secret = "xxx"
    >>>
    >>> [auth.microsoft]
    >>> client_id = "xxx"
    >>> client_secret = "xxx"
    >>> server_metadata_url = "https://login.microsoftonline.com/{tenant}/v2.0/.well-known/openid-configuration"

    Your app code:

    >>> import streamlit as st
    >>>
    >>> if not st.experimental_user.is_logged_in:
    >>>     st.login("microsoft")
    >>> else:
    >>>     st.write(f"Hello, {st.experimental_user.name}!")

    **Example 3: Use multiple, named providers**

    If you want to give your users a choice of authentication methods,
    configure multiple providers and give them each a unique name. The
    following example lets users choose between Okta and Microsoft to log in.
    Always check with your identity provider to understand the structure of
    their identity tokens because the returned fields may differ. Remember to
    set ``{tenant}`` and ``{subdomain}`` in ``server_metadata_url`` for
    Microsoft and Okta, respectively.

    >>> [auth]
    >>> redirect_uri = "http://localhost:8501/oauth2callback"
    >>> cookie_secret = "xxx"
    >>>
    >>> [auth.microsoft]
    >>> client_id = "xxx"
    >>> client_secret = "xxx"
    >>> server_metadata_url = "https://login.microsoftonline.com/{tenant}/v2.0/.well-known/openid-configuration"
    >>>
    >>> [auth.okta]
    >>> client_id = "xxx"
    >>> client_secret = "xxx"
    >>> server_metadata_url = (
    ...     "https://{subdomain}.okta.com/.well-known/openid-configuration"
    ... )

    Your app code:

    >>> import streamlit as st
    >>>
    >>> if not st.experimental_user.is_logged_in:
    >>>     st.header("Log in:")
    >>>     if st.button("Microsoft"):
    >>>         st.login("microsoft")
    >>>     if st.button("Okta"):
    >>>         st.login("okta")
    >>> else:
    >>>     if st.button("Log out"):
    >>>         st.logout()
    >>>     st.write(f"Hello, {st.experimental_user.name}!")

    **Example 4: Change the default connection settings**

    ``prompt="select_account"`` may be treated differently by some
    providers when a user is already logged into their account. If a user is
    logged into their Google or Microsoft account from a previous session, the
    provider will prompt them to select the account they want to use, even if
    it's the only one. However, if the user is logged into their Okta or Auth0
    account from a previous session, the account will automatically be
    selected. ``st.logout()`` does not clear a user's related cookies. To force
    users to log in every time, use ``prompt="login"`` as described in Auth0's
    `Customize Signup and Login Prompts
    <https://auth0.com/docs/customize/login-pages/universal-login/customize-signup-and-login-prompts>`_.

    ``.streamlit/secrets.toml``:

    >>> [auth]
    >>> redirect_uri = "http://localhost:8501/oauth2callback"
    >>> cookie_secret = "xxx"
    >>>
    >>> [auth.auth0]
    >>> client_id = "xxx"
    >>> client_secret = "xxx"
    >>> server_metadata_url = (
    ...     "https://{account}.{region}.auth0.com/.well-known/openid-configuration"
    ... )
    >>> client_kwargs = { "prompt" = "login" }

    Your app code:

    >>> import streamlit as st
    >>> if st.button("Log in"):
    >>>     st.login("auth0")
    >>> if st.experimental_user.is_logged_in:
    >>>     if st.button("Log out"):
    >>>         st.logout()
    >>>     st.write(f"Hello, {st.experimental_user.name}!)

    NdefaultzcTo use authentication features, you need to install Authlib>=1.3.2, e.g. via `pip install Authlib`.)	_get_script_run_ctxr   r   r   r   generate_login_redirect_urlauth_redirecturlenqueue)r   contextfwd_msg r%   O/var/www/html/chatdoc2/venv/lib/python3.10/site-packages/streamlit/user_info.pyr   0   s    clogoutc                  C  sh   t  } | dur2| j  | j}t rt }|| t	d}t
 }t|t|j_| | dS dS )aS  Logout the current user.

    This command removes the user's information from ``st.experimental_user``,
    deletes their identity cookie, and redirects them back to your app's home
    page. This creates a new session.

    If the user has multiple sessions open in the same browser,
    ``st.experimental_user`` will not be cleared in any other session.
    ``st.experimental_user`` only reads from the identity cookie at the start
    of a session. After a session is running, you must call ``st.login()`` or
    ``st.logout()`` within that session to update ``st.experimental_user``.

    .. Note::
        This does not log the user out of their underlying account from the
        identity provider.

    Example
    -------
    ``.streamlit/secrets.toml``:

    >>> [auth]
    >>> redirect_uri = "http://localhost:8501/oauth2callback"
    >>> cookie_secret = "xxx"
    >>> client_id = "xxx"
    >>> client_secret = "xxx"
    >>> server_metadata_url = (
    ...     "https://accounts.google.com/.well-known/openid-configuration"
    ... )

    Your app code:

    >>> import streamlit as st
    >>>
    >>> if not st.experimental_user.is_logged_in:
    >>>     if st.button("Log in"):
    >>>         st.login()
    >>> else:
    >>>     if st.button("Log out"):
    >>>         st.logout()
    >>>     st.write(f"Hello, {st.experimental_user.name}!")
    Nserver.baseUrlPath)r   	user_infoclear
session_idr
   existsget_instanceclear_user_info_for_sessionr	   
get_optionr   r   r   r    r!   r"   )r#   r+   instance	base_pathr$   r%   r%   r&   r'   "  s   +


strc                 C  s*   t | }td}t|t}| d| S )z7Generate the login redirect URL for the given provider.r(   z
?provider=)r   r	   r/   r   r   )r   provider_tokenr1   
login_pathr%   r%   r&   r   ]  s   

r   r   c                  C  s:   t  } | d u r	i S | j }t }d|vr|rd|d< |S )Nis_logged_inF)r   r)   copyr   )ctxcontext_user_infoauth_section_existsr%   r%   r&   _get_user_infoe  s   
r:   c                   @  sV   e Zd ZdZdddZddd	ZdddZdddZdddZdddZ	d ddZ
dS )!UserInfoProxya  
    A read-only, dict-like object for accessing information about the current    user.

    ``st.experimental_user`` is dependent on the host platform running your
    Streamlit app. If the host platform has not configured the function, it
    will behave as in a locally running app.

    When authentication is configured in ``secrets.toml``, Streamlit will parse
    the OpenID Connect (OIDC) identity token and copy the attributes to
    ``st.experimental_user``. Check your provider's documentation for their
    available attributes (known as claims).

    When authentication is not configured, ``st.experimental_user`` has no
    attributes.

    You can access values via key or attribute notation. For example, use
    ``st.experimental_user["email"]`` or ``st.experimental_user.email`` to
    access the ``email`` attribute.

    .. Important::
        Identity tokens include an issuance and expiration time. Streamlit does
        not implicitly check these. If you want to automatically expire a
        user's authentication, check these values manually and programmatically
        log out your user (``st.logout()``) when needed.

    Attributes
    ----------
    is_logged_in: bool
        Whether a user is logged in. For a locally running app, this attribute
        is only available when authentication (``st.login()``) is configured in
        ``secrets.toml``. Otherwise, it does not exist.

    Examples
    --------
    **Example 1: Google's identity token**

    If you configure a basic Google OIDC connection as shown in Example 1 of
    ``st.login()``, the following data is available in
    ``st.experimental_user``. Streamlit adds the ``is_logged_in`` attribute.
    Additional attributes may be available depending on the configuration of
    the user's Google account. For more information about Google's identity
    tokens, see `Obtain user information from the ID token
    <https://developers.google.com/identity/openid-connect/openid-connect#obtainuserinfo>`_
    in Google's docs.

    Your app code:

    >>> import streamlit as st
    >>>
    >>> if st.experimental_user.is_logged_in:
    >>>     st.write(st.experimental_user)

    Displayed data when a user is logged in:

    >>> {
    >>>     "is_logged_in":true
    >>>     "iss":"https://accounts.google.com"
    >>>     "azp":"{client_id}.apps.googleusercontent.com"
    >>>     "aud":"{client_id}.apps.googleusercontent.com"
    >>>     "sub":"{unique_user_id}"
    >>>     "email":"{user}@gmail.com"
    >>>     "email_verified":true
    >>>     "at_hash":"{access_token_hash}"
    >>>     "nonce":"{nonce_string}"
    >>>     "name":"{full_name}"
    >>>     "picture":"https://lh3.googleusercontent.com/a/{content_path}"
    >>>     "given_name":"{given_name}"
    >>>     "family_name":"{family_name}"
    >>>     "iat":{issued_time}
    >>>     "exp":{expiration_time}
    >>> }

    **Example 2: Microsoft's identity token**

    If you configure a basic Microsoft OIDC connection as shown in Example 2 of
    ``st.login()``, the following data is available in
    ``st.experimental_user``. For more information about Microsoft's identity
    tokens, see `ID token claims reference
    <https://learn.microsoft.com/en-us/entra/identity-platform/id-token-claims-reference>`_
    in Microsoft's docs.

    Your app code:

    >>> import streamlit as st
    >>>
    >>> if st.experimental_user.is_logged_in:
    >>>     st.write(st.experimental_user)

    Displayed data when a user is logged in:

    >>> {
    >>>     "is_logged_in":true
    >>>     "ver":"2.0"
    >>>     "iss":"https://login.microsoftonline.com/{tenant_id}/v2.0"
    >>>     "sub":"{application_user_id}"
    >>>     "aud":"{application_id}"
    >>>     "exp":{expiration_time}
    >>>     "iat":{issued_time}
    >>>     "nbf":{start_time}
    >>>     "name":"{full_name}"
    >>>     "preferred_username":"{username}"
    >>>     "oid":"{user_GUID}"
    >>>     "email":"{email}"
    >>>     "tid":"{tenant_id}"
    >>>     "nonce":"{nonce_string}"
    >>>     "aio":"{opaque_string}"
    >>> }
    keyr2   r   str | bool | Nonec                 C  s,   zt  | W S  ty   td| dw )Nz!st.experimental_user has no key "".)r:   KeyErrorselfr<   r%   r%   r&   __getitem__  
   zUserInfoProxy.__getitem__c                 C  s,   zt  | W S  ty   td| dw )Nz'st.experimental_user has no attribute "r>   )r:   r?   AttributeErrorr@   r%   r%   r&   __getattr__  rC   zUserInfoProxy.__getattr__namevaluer   r   c                 C     t dNz'st.experimental_user cannot be modifiedr   rA   rF   rG   r%   r%   r&   __setattr__     zUserInfoProxy.__setattr__c                 C  rH   rI   rJ   rK   r%   r%   r&   __setitem__  rM   zUserInfoProxy.__setitem__Iterator[str]c                 C  
   t t S N)iterr:   rA   r%   r%   r&   __iter__     
zUserInfoProxy.__iter__intc                 C  rP   rQ   )lenr:   rS   r%   r%   r&   __len__  rU   zUserInfoProxy.__len__r   c                 C  s   t  S )a_  
        Get user info as a dictionary.

        This method primarily exists for internal use and is not needed for
        most cases. ``st.experimental_user`` returns an object that inherits from
        ``dict`` by default.

        Returns
        -------
        Dict[str,str]
            A dictionary of the current user's information.
        )r:   rS   r%   r%   r&   to_dict  s   zUserInfoProxy.to_dictN)r<   r2   r   r=   )rF   r2   rG   r   r   r   )r   rO   )r   rV   r   r   )__name__
__module____qualname____doc__rB   rE   rL   rN   rT   rX   rY   r%   r%   r%   r&   r;   r  s    
n




r;   rQ   )r   r   r   r   )r   r   )r   r2   r   r2   rZ   ))
__future__r   collections.abcr   r   typingr   r   r   r   	streamlitr	   r
   streamlit.auth_utilr   r   r   r   streamlit.errorsr   r   streamlit.proto.ForwardMsg_pb2r   streamlit.runtime.metrics_utilr   7streamlit.runtime.scriptrunner_utils.script_run_contextr   r   streamlit.url_utilr   r   r   __annotations__r   r   r'   r   r:   r2   boolr;   r%   r%   r%   r&   <module>   s,    r
:
&