�
"�j � �d � % S SK Jr S SKJr S SKJr S SKJr S SKJ r " S S\
5 r " S S \5 r \" S
S 9 " S S
5 5 r
\" S
S 9 " S S5 5 r\
\- r\
" 5 r\\S' S\\\4 S\4S jrS\\\4 S\4S jrS\S\4S jrS\S\\ 4S jrS\S\\\4 S- S\4S jrg)� )�Mapping)� dataclass)�StrEnum)�Any)�builtinsc � � \ rS rSrSrg)� LayoutError� � N)�__name__�
__module__� __qualname__�__firstlineno__�__static_attributes__r � �D/Users/michael/dev/contexts/ctx/rewrite-it-in-rust/src/ctx/layout.pyr r s � �r r c � � \ rS rSrSrSrSrg)�SplitDirection�
�row�columnr N)r r
r r �ROW�COLUMNr r r r r r
s � �
�C�
�Fr r T)�frozenc �` � \ rS rSr% Sr\S- \S' Sr\S- \S' Sr\S- \S' Sr \
\S' Sr g) �Pane� N�command�builtin�argsF�focusr ) r r
r r r �str�__annotations__r r r! �boolr r r r r r s6 � � �G�S�4�Z� � �G�S�4�Z� ��D�#��*���E�4�r r c �0 � \ rS rSr% \\S' \S \S' Srg)�Split� � direction)�Node.�panesr N)r r
r r r r# �tupler r r r r& r& s � � � � �� � r r& �DEFAULT_LAYOUT�data�returnc �P � [ U 5 n[ U5 S:� a [ S5 eU$ )N� z at most one pane may set focus)� _parse_node� _count_focusr )r- �nodes r � parse_layoutr4 % s* � � �t� �D��D��A� ��:�;�;� �Kr c �6 � SU ; a� U R 5 SS1-
nU( a& [ SSR [ U5 5 35 e [ [ U S 5 5 nU R S5 n[ U[ 5 ( a U( d [ S5 e[ U[ S U 5 5 5 $ U R 5 1 S
k-
nU( a& [ S SR [ U5 5 35 eS U ; a S
U ; a [ S5 eS
U ; aK U S
[ R ; a4 SR [ R 5 n[ SU S
SU S35 eSU ; a S
U ; a [ S5 e[ S U ; a [ U S 5 OS S
U ; a [ U S
5 OS SU ; a [ U S 5 OS [ U R SS5 5 S9$ ! [ a7 nSR S [ 5 5 n[ SU SU S < 35 UeS nAff = f)N�splitr* zunknown split key(s): z, c 3 �8 # � U H oR v � M g 7f�N)�value)�.0�ds r � <genexpr>� _parse_node.<locals>.<genexpr>4 s � � � >�~�!�g�g�~�� �zsplit must be one of z, got z&a split needs a non-empty 'panes' listc 3 �8 # � U H n[ U5 v � M g 7fr8 )r1 �r: �panes r r< r= 9 s � � �%J�E�D�k�$�&7�&7�E�r> > r r! r r zunknown pane key(s): r r z4a pane takes either a command or a builtin, not bothzunknown pane builtin 'z' (supported: �)r zpane args require a builtinr! F)r r r r! )�keysr �join�sortedr r" �
ValueError�get�
isinstance�listr& r+ r �
PANE_BUILTINSr r$ )r- �unknownr( �exc�validr* s r r1 r1 , s� � ��$���)�)�+��'� 2�2�� � � 6�t�y�y��� �7Q�6R� S�T� T� _�&�s�4��=�'9�:�I� ����!���%��&�&�e� � F�G� G��Y��%J�E�%J� J�K�K��i�i�k� C�C�G��� 1�$�)�)�F�7�O�2L�1M�N�O�O��D��Y�$� .��P�Q�Q��D��T�)�_�H�4J�4J� J�� � �(�0�0�1��� 2�4� �?�2C�>�RW�QX�XY�Z�[�[�
��~�)�4�/��7�8�8� �(1�T�(9��D��O�$�t�(1�T�(9��D��O�$�t�"(�D�.�S��f� �
�d��4�8�8�G�U�+�,� � ��# � _��I�I� >�~� >�>�E� � 5�e�W�F�4��=�BS� T�U�[^� ^�� _�s �
G �
H�!2H�Hr3 c � � [ U [ 5 ( a [ U R 5 $ [ S U R
5 5 $ )Nc 3 �8 # � U H n[ U5 v � M g 7fr8 )r2 r@ s r r< � _count_focus.<locals>.<genexpr>O s � � �9�j�d�|�D�!�!�j�r> )rH r �intr! �sumr* �r3 s r r2 r2 L s2 � ��$�� � ��4�:�:�� � �9�d�j�j�9� 9�9r c �� � [ U [ 5 ( a; [ R R U R
=( d S[
5 5 $ [
5 R " S U R 5 6 $ )z@All creation-time value keys the layout's builtin panes consume.� c 3 �8 # � U H n[ U5 v � M g 7fr8 )�
accepted_keysr@ s r r< � accepted_keys.<locals>.<genexpr>V s � � � J�z�t�}�T� 2� 2�z�r> ) rH r r � BUILTIN_KEYSrG r � frozenset�unionr* rS s r rW rW R sP � ��$�� � ��$�$�(�(�� � �);��Y�[�I�I� �;� � � J�t�z�z� J� K�Kr �valuesNc �0 ^� [ U [ 5 ( aN U R c U $ [ R " U R U R
T5 n[ X R S9$ [ U R [ U4S jU R 5 5 5 $ )z�Concretise builtin panes into command panes.
`values` carries creation-time key=value data for the builtins; None
means the session is being recreated for an existing context.
)r! c 3 �< ># � U H n[ UT5 v � M g 7fr8 )�resolve_layout)r: rA r\ s �r r< �!resolve_layout.<locals>.<genexpr>d s �� � �&[�PZ��~�d�F�'C�'C�PZ�s � ) rH r r r �builtin_commandr r! r& r( r+ r* )r3 r\ r s ` r r_ r_ Y so �� � �$�� � � �<�<� ��K��*�*�4�<�<����F�K���G�:�:�.�.� �� � ��&[�PT�PZ�PZ�&[�![� \�\r )�collections.abcr � dataclassesr �enumr �typingr �ctxr � Exceptionr r r r& r) r, r# �dictr" r4 r1 rQ r2 rZ rW r_ r r r �<module>ri s �� #� !� � � � �)� ��W� �
�$��� � �� �$�� � � � �
�e�|���v��� ��t�C��H�~� �$� � �d�3��8�n� �� �@:�t� :�� :� L�� L��3� � L� ]�� ]�w�s�C�x�'8�4�'?� ]�D� ]r