ruạṛ
a �u:j� � @ s^ d dl Z d dlZd dlZd dlmZmZ d dlmZmZm Z dZ dZ dZ dZ G dd � d �ZdS ) � N)�Optional�Dict)�run_sudo_as_user�read_file_as_owner� is_directoryz # --- BART_MAINTENANCE_START ---z # --- BART_MAINTENANCE_END ---z# created_by=BARTa_ <!doctype html><meta charset='utf-8'><title>Maintenance</title><meta http-equiv='Cache-Control' content='no-store, no-cache, must-revalidate'><meta http-equiv='Pragma' content='no-cache'><meta http-equiv='Expires' content='0'><style>html,body{height:100%;margin:0}body{display:flex;align-items:center;justify-content:center;font:16px/1.4 system-ui}.box{max-width:640px;padding:24px;border:1px solid #ddd;border-radius:12px;text-align:center}h1{margin:0 0 8px 0;font-size:22px}</style><div class='box'><h1>We’ll be right back</h1><p>The site is temporarily unavailable while we restore files.</p></div>c @ s� e Zd Zd$ed�dd�Zeeee d�dd�Zee d�d d �Z eed �d d �Z eeeed�dd�Z eed�dd�Z d%eeeed�dd�Z eed�dd�Zeed�d d �Zeed�d d �Zed �d!d"�Zd#S )&�MaintenanceManagerF)�allow_create_temp_htaccessc C s || _ || _d S )N)�logr )�selfZactivity_loggerr � r �P/opt/bart/bart_venv/lib/python3.9/site-packages/app/utils/maintenance_manager.py�__init__ s zMaintenanceManager.__init__)�username� abs_dest_path�returnc C sf t |�r |n tj�|�}tj�|d�}| j�d|� �� |}t||�}d}|d u r�| jsp| j�d|� d�� d S t � d�}| � |||d� d}| j�d |� �� n| j�d |� �� t |vs�t |v�r|� d�r�|n|d | �� }| � |||d� | j�d |� �� n| j�d |� �� tj�|d �} tj�| ��sX| � | t|d� | j�d| � �� ||||d�S )Nz .htaccesszSearching for .htaccess in FzNo .htaccess in z ; skipping maintenance toggle.� �0644Tz Created temporary .htaccess at zFound .htaccess at zAdded maintenance block to z%Maintenance block already present in �_bart_maintenance.htmlzCreated maintenance page )� target_dir�htaccess�created_temp_htaccessr )r �os�path�dirname�joinr �noter r �BART_CREATED_TAG� _atomic_write�BART_HTACCESS_START�BART_HTACCESS_END�endswith�_maintenance_block�exists�_DEFAULT_MAINTENANCE_HTML) r r r r r � run_as_userZcurrent_htaccess� created_temp� new_content�maintenance_htmlr r r �enable_for_path# s: �z"MaintenanceManager.enable_for_path)�tokenc C s� |sd S |d }|d }|d }|d }t j�|�r�| �|�}| �|�}|rv| �|�rv| �||� | j�d|� �� n"| � |||d� | j�d|� �� t j� |d�}t j� |�r�| �||� | j�d |� �� d S ) Nr r r r z Removed temporary .htaccess at r z Removed maintenance block from r zRemoved ) r r �isfile� _read_text� _strip_block�_is_only_tag_or_empty�_rmr r r r r" ) r r) r r% r r$ �currentr& r' r r r �disable_for_pathO s$ z#MaintenanceManager.disable_for_path)r r c C sZ zt �|�}t�|j�jW S tyT } z"| j�d|� �� |�W Y d }~n d }~0 0 d S )NzError in resolving owner: ) r �stat�pwd�getpwuid�st_uid�pw_name� Exceptionr r )r r �st�er r r �_resolve_ownerg s z!MaintenanceManager._resolve_owner)� target_path�contentr$ � file_modec C s� t j�|�}t j�|dtt�� d �� ��}t|d|g|d� t|d||g� zt|dd||g� W n< ty� zt|dd|g� W n ty� Y n0 � Y n0 d S ) N� .barttmp.�� � /usr/bin/tee�Z input_text� /bin/chmod�/bin/mv�-f�/bin/rm)r r r r �int�timer r6 )r r: r; r$ r< r Ztmp_pathr r r r o s z MaintenanceManager._atomic_write)r: r$ c C s t |dd|g� d S )NrD rC )r )r r: r$ r r r r. ~ s zMaintenanceManager._rmr )r: r; r$ �modec C s` t j�t j�|�dtt�� d �� ��}t|d|g|d� t|d||g� t|dd||g� d S )Nr= r> r? r@ rA rB rC )r r r r rE rF r )r r: r; r$ rG �tmpr r r �atomic_write_local� s (z%MaintenanceManager.atomic_write_local)r r c C sX z>t |dddd��}|�� W d � W S 1 s20 Y W n tyR Y dS 0 d S )N�rzutf-8�ignore)�encoding�errors� )�open�readr6 )r r �fhr r r r+ � s , z MaintenanceManager._read_text)r; r c C sh |� t�}|� t�}|dks,|dks,||k r0|S |tt�7 }|d |� ||d � �� }|rd|d S dS )N���r rN )�findr r �len�strip)r r; �start�endr& r r r r, � s z MaintenanceManager._strip_blockc C s |� � }|dkp|tkS )NrN )rU r )r r; �cleanedr r r r- � s z(MaintenanceManager._is_only_tag_or_empty)r c C s t � dt� d�S )Na0 RewriteEngine On RewriteCond %{REQUEST_URI} !^/_bart_maintenance\.html$ RewriteRule ^ _bart_maintenance.html [L] <IfModule mod_headers.c> Header always set Cache-Control "no-store, no-cache, must-revalidate, max-age=0" Header always set Pragma "no-cache" Header always set Expires "0" </IfModule> r )r r )r r r r r! � s ��z%MaintenanceManager._maintenance_blockN)F)r )�__name__� __module__� __qualname__�boolr �strr r r( r0 r9 r r. rI r+ r, r- r! r r r r r s , r )r r2 rF �typingr r Zapp.utils.utility_functionsr r r r r r r# r r r r r �<module> s �
cải xoăn