pyc • Lines: 130�
���j� � � � d Z ddlZddlZddlZddlmZ ddlmZ ddlmZ ddl
m Z ddl m
Z
d� Zd � Zd
� Zd � Zdd �Zd
� Zd� Zd� Zy)u� Per-user context files and HTTP identity resolution.
Identity comes from :mod:`server.auth` — Authentik is the single identity
provider (there is no users.json anymore). This module keeps the names the
chat engine calls: password lookup is gone, context paths are derived from the
username, and ``get_current_user``/``get_current_identity`` resolve the
request's identity through the shared auth layer.
� N)�datetime)�get_current_user)� get_identity)� CONTEXTS_DIR)�Mc �D � t j dd| xs d� }|xs dS )Nz[^A-Za-z0-9_-]�_� �unknown)�re�sub)�user�safes �2/home/palash/git/local-ai/server/features/users.py�_safe_usernamer s$ � �
�6�6�#�S�$�*�"� 5�D� � �9� � c � � | syt j 5 t j � t j | < ddd� y# 1 sw Y yxY w)zIRecord a username as recently active (heartbeat for active-user display).N)r �_user_last_seen_lock�time�_user_last_seen��usernames r �
_mark_seenr s< � � ��
� � � 2�&*�i�i�k����(�#�2� 2� 2�s �&A�A c �2 � t | � }t |� |S �N)�auth_get_current_userr )�headersr s r r r " s � �$�W�-�H��x�� �Or c �< � t | � }|rt |d � |S )Nr )�auth_get_identityr )r �identitys r �get_current_identityr! ( s! � � ��)�H���8�J�'�(� �Or c � � �� t j � �t j 5 t �� fd�t j j � D � � }ddd� |rGt j 5 t t j � }ddd� D �cg c] }|vs�|�� c}S S # 1 sw Y �TxY w# 1 sw Y �,xY wc c}w )zESorted usernames seen within the window, optionally excluding agents.c 3 �8 �K � | ] \ }}�|z
�k r|�� � y �wr � )�.0�u�ts�now�window_secondss ��r � <genexpr>z active_users.<locals>.<genexpr>3 s) �� �� �
��!�R��R�x�>�)�
�
�s �N) r r r �sortedr �items� _tokens_lock�set� _agent_users)r) �exclude_agents�users� agent_usersr& r( s ` @r � active_usersr3 / s� �� �
�)�)�+�C�
� � �
��
� �,�,�2�2�4�
�
��
�
�
�^�^� .� �a�n�n�-�K� .� �9�a�A�[�$8��9�9� �L�
�
�� .� .��9s# �2B,�3B8� C�#C�,B5�8Cc �b � t j j t t | � dz � S )a Derive the context file path from the username.
users.json previously stored an arbitrary per-user ``context_file`` path;
with Authentik as the sole identity store that field no longer exists, so
every user's context lives at ``CONTEXTS_DIR/<username>.txt``.
z.txt)�os�path�joinr r r s r �get_user_context_pathr8 > s# � � �7�7�<�<� �n�X�&>��&G� H�Hr c � � t | � }t d|d| � |r^t j j |� r? t d|� t |� 5 }|j
� }t |� |cd d d � S y# 1 sw Y yxY w# Y yxY w)Nz Context path�for�Readingr
)r8 �printr5 r6 �exists�open�read)r r6 �f�contexts r �read_user_contextrB H s� � � �� *�D� �.�$��x�0� ������t�$� � �)�T� "��d��
�q��&�&�(���g�� �
�
� �
� �� ��s) �B � A;�0 B �;B� B �B �B c � � t | � }|r�t j t j j |� d�� t | � }t
j � j d� }d|� d|� �}|j � r|j � dz |z n|}t |d� 5 }|j |� d d d � y y # 1 sw Y y xY w)NT)�exist_okz%Y-%m-%d %H:%M�[z] z
�w) r8 r5 �makedirsr6 �dirnamerB r r( �strftime�stripr> �write)r �contentr6 �existing� timestamp�entry� new_contentr@ s r �write_user_contextrQ W s� � � �� *�D� �
� � �B�G�G�O�O�D�)�D�9�$�X�.�� �L�L�N�+�+�,<�=� ��I�;�b�� �*��=E�^�^�=M�x�~�~�'�&�0�5�8�SX� �
�$��_� !��
�G�G�K� � !� !�
� !� !�s �,C�C)�x T)�__doc__r5 r r r � server.authr r r r �
server.configr �server.features.stater r r r! r3 r8 rB rQ r$ r r �<module>rW sM ���
� � � � A� 9� &� #� �
2�� � � I� � !r