o
�j� � @ s� d Z ddlmZ ddlZddlZddlZddlZddlZddlZddl Z ddl
m Z ddl m
Z
mZmZmZmZmZ ddlmZmZmZ ejd*d d
��ZG dd� de
�Zd+dd�Z d,d-d d �Zd.d d!�Z d/d"d#�Z d/d$d%�Z d0d(d)�Z dS )1z
The ``relenv create`` command.
� )� annotationsN)�Iterator� )�RelenvException�arches�archived_build�
build_arch�format_shebang�relative_interpreter)�get_default_python_version�python_versions�resolve_python_version�path�str | os.PathLike[str]�return�Iterator[None]c c s8 � t �� }zt �| � dV W t �|� dS t �|� w )z�
Context manager that changes to the specified directory and back.
:param path: The path to temporarily change to
:type path: str
N)�os�getcwd�chdir)r �cwd� r �A/opt/saltstack/salt/lib/python3.10/site-packages/relenv/create.pyr ! s �
r c @ s e Zd ZdZdS )�CreateExceptionzJ
Raised when there is an issue creating a new relenv environment.
N)�__name__�
__module__� __qualname__�__doc__r r r r r 1 s r �
subparsers�3argparse._SubParsersAction[argparse.ArgumentParser]�Nonec C s` | j ddd�}|jtd� |jddd� |jdt� ttj td d
� t � }|jd |td d
� dS )z�
Setup the subparser for the ``create`` command.
:param subparsers: The subparsers object returned from ``add_subparsers``
:type subparsers: argparse._SubParsersAction
�createzrCreate a Relenv environment. This will create a directory of the given name with newly created Relenv environment.)� description)�func�namez#The name of the directory to create)�helpz--archz,The host architecture [default: %(default)s])�default�choices�typer$ z--pythonz>The python version (e.g., 3.10, 3.13.7) [default: %(default)s])r% r' r$ N)
�
add_parser� set_defaults�main� add_argumentr r �sys�platform�strr )r Z subparserZdefault_versionr r r � setup_parser7 s( � �
�r/ r# r. �dest� str | os.PathLike[str] | None�arch�
str | None�versionc
C sf |du rt � }|rt�|�| }nt�| ��� }|du r t� }t�|��� r*td��tj}|dkrB|t | v r>d�
||�}n0td��|dkrV|t | v rRd�
|�}n td��|dkrj|t | v rfd �
|�}ntd��td
��t |� d |� ��}|�� s�td |� d
|� d���t �
|d��}|D ]} |�| |� q�W d � n1 s�w Y t||� t||� dS )ai
Create a relenv environment.
:param name: The name of the environment
:type name: str
:param dest: The path the environment should be created under
:type dest: str
:param arch: The architecture to create the environment for
:type arch: str
:raises CreateException: If there is a problem in creating the relenv environment
Nz"The requested path already exists.�linuxz {}-{}-gnuz Unknown arch�darwinz{}-macos�win32z{}-winzUnknown platform�-zError, build archive for z doesn't exist: z,
You might try relenv fetch to resolve this.zr:xz)r �pathlib�Path�resolver �existsr r, r- r �formatr �tarfile�open�extract�_sync_relenv_package�_repair_script_shebangs)
r# r0 r2 r4 Zwriteto�plat�triplet�tar�fp�fr r r r Y sF �� �
r �root� pathlib.Pathc C sD d� |�d�dd� �}tjdkr| d d S | d d|� � d S ) zL
Return the site-packages directory within the created environment.
�.N� r7 ZLibz
site-packages�lib�python)�join�splitr, r- )rH r4 � major_minorr r r �_site_packages_dir� s
rQ c C s` t | |�}|�� s dS |d }t�t��� j}|�� r t�|� tj ||dt�
ddd�d� dS )zT
Ensure the relenv package within the created environment matches this CLI.
N�relenvT� __pycache__z*.pycz*.pyo)�
dirs_exist_ok�ignore) rQ r< r9 r: �__file__r; �parent�shutil�rmtree�copytree�ignore_patterns)rH r4 Z target_site�target�sourcer r r rA � s
�rA c
C s� t jdkrdS | d }|�� sdS d�|�d�dd� �}|d|� � |d|�d�d � � |d |d g}d}|D ]
}|�� rD|} nq:|du rKdS zt| ||�}W n
ty] Y dS w z
tt t
� d �| ��}W n
t yu Y dS w d
} d }
|�
� D ]P} | �� s�q~z| jd d
�} W n ttfy� Y q~w | �| �s�q~| �|
�}
|
dkr�q~|
t|
� }| |d� }||�d� }z | j|d d
� W q~ ty� Y q~w dS )a�
Update legacy shell-wrapped entry points to the current shebang format.
Older archives shipped scripts that started with the ``"true" ''''`` preamble.
Those files break when executed directly under Python (the parser sees the
unmatched triple-quoted literal). Patch any remaining copies to the new
`format_shebang` layout so fresh installs do not inherit stale loaders.
r7 N�binrJ rK rM r Zpython3�/z#!/bin/sh
"true" ''''
z
'''zutf-8)�encoding����
)r, r- �is_dirrN rO r< r
�
ValueErrorr r. r9 �
PurePosixPath� Exception�iterdir�is_file� read_text�OSError�UnicodeDecodeError�
startswith�find�len�lstrip�
write_text)rH r4 � scripts_dirrP Zinterpreter_candidatesZinterpreter_path� candidateZrel_interpreter�shebangZ
legacy_prefix�markerZscript�text�idxZidy�rest�updatedr r r rB � sj
�� � � �
��rB �args�argparse.Namespacec
C s� | j }| jt� kr
td� zt| j�}W n2 tyF } z&td|� �� t� }d�dd� |D ��}td|� �� t �
d� W Y d}~nd}~ww z t || j|d � W dS t yn } zt|� t �
d� W Y d}~dS d}~ww )
z�
The entrypoint into the ``relenv create`` command.
:param args: The args passed to the command
:type args: argparse.Namespace
zVWarning: Cross compilation support is experimental and is not fully tested or working!zError: rb c S s g | ]}t |��qS r )r. )�.0�_r r r �
<listcomp> s zmain.<locals>.<listcomp>zKnown versions are:
r N)r2 r4 )
r# r2 r �printr
rM � RuntimeErrorr rN r, �exitr r )ry r# Zcreate_version�e�
pyversionsZ strversions�excr r r r* � s, �����r* )r r r r )r r r r )NNN)
r# r. r0 r1 r2 r3 r4 r3 r r )rH rI r4 r. r rI )rH rI r4 r. r r )ry rz r r ) r �
__future__r �argparse�
contextlibr r9 rX r, r> �collections.abcr �commonr r r r r r
r� r r r
�contextmanagerr r r/ r rQ rA rB r* r r r r �<module> s0
$ �
B
A