�
�F�g5
� �� � d dl mZmZmZmZ d dlZd dlmZ d dlm Z d dl
m Z G d� de � � Z e G d� d � � � � Z
e G d
� d � � � � Z G d � d
� � ZdS )� )�Dict�List�Optional�UnionN)� dataclass)�Enum)�Pathc � � e Zd ZdZdZdZdZdS )� FunctionType�view�pure�
nonpayable�payableN)�__name__�
__module__� __qualname__�VIEW�PURE�
NONPAYABLE�PAYABLE� � �8/Users/arjun/repos/sparkmango/mcp_server/abi_analyzer.pyr r s" � � � � � � �D� �D� �J��G�G�Gr r c �J � e Zd ZU eed< eed< dZeed ed< dS )�FunctionParameter�name�typeN�
components)r r r �str�__annotations__r r r r r r r r
s? � � � � � � �
�I�I�I�
�I�I�I�6:�J��� 1�2�3�:�:�:�:�:r r c �^ � e Zd ZU eed< ee ed< ee ed< eed< dZe ed< dS )�FunctionDefinitionr �inputs�outputs�state_mutabilityF�is_constructorN)
r r r r r r r r r& �boolr r r r" r" s[ � � � � � � �
�I�I�I� �"� #�#�#�#�
�#�
$�$�$�$�"�"�"�"� �N�D� � � � � r r" c � � e Zd Zdeeef fd�Zdefd�Zdee fd�Z
dee fd�Z dee fd�Z de
e fd�Zd edefd
�Zd S ) � ABIAnalyzer� abi_inputc �( � t |t t f� � rZt |d� � 5 }t j |� � }|� d|� � | _ ddd� � dS # 1 swxY w Y dS |� d|� � | _ dS )zNInitialize the ABI analyzer with either a path to an ABI file or a dictionary.�r�abiN)�
isinstancer r �open�json�load�getr- )�selfr* �f�datas r �__init__zABIAnalyzer.__init__ s� � � �i�#�t�� -� -� 7��i��%�%�
1���y��|�|�� �8�8�E�4�0�0���
1�
1�
1�
1�
1�
1�
1�
1�
1�
1�
1�
1����
1�
1�
1�
1�
1�
1� !�}�}�U�I�6�6�D�H�H�Hs �0A*�*A.�1A.�returnc � � | j | � � � | � � � | � � � | � � � d�}|S )z7Analyze the ABI and return a structured representation.)r- � functions�events�state_variables� constructor)r- �_get_functions� _get_events�_get_state_variables�_get_constructor)r3 �analysiss r �analyzezABIAnalyzer.analyze% sX � � �8� �,�,�.�.��&�&�(�(� #� 8� 8� :� :� �0�0�2�2�
�
�� �r c �V � � g }� j D ]�}|d dk r�t |d � fd�|� dg � � D � � � fd�|� dg � � D � � t |� dd � � � � �
� � }|� |� � ��|S ) z#Extract all functions from the ABI.r �functionr c �: �� g | ]}�� |� � ��S r ��_parse_parameter��.0�pr3 s �r �
<listcomp>z.ABIAnalyzer._get_functions.<locals>.<listcomp>7 s' �� �U�U�U��D� 1� 1�!� 4� 4�U�U�Ur r# c �: �� g | ]}�� |� � ��S r rF rH s �r rK z.ABIAnalyzer._get_functions.<locals>.<listcomp>8 s' �� � W� W� W�!�T� 2� 2�1� 5� 5� W� W� Wr r$ �stateMutabilityr )r r# r$ r% )r- r" r2 r �append)r3 r9 �item�funcs ` r r= zABIAnalyzer._get_functions0 s� �� �� ��H� '� '�D��F�|�z�)�)�)� �f� �U�U�U�U�d�h�h�x�QS�>T�>T�U�U�U� W� W� W� W�t�x�x� �SU�?V�?V� W� W� W�%1�$�(�(�;L�l�2[�2[�%\�%\� � � �� � � ��&�&�&���r c �$ � d� | j D � � S )z Extract all events from the ABI.c �* � g | ]}|d dk �|��S )r �eventr )rI rO s r rK z+ABIAnalyzer._get_events.<locals>.<listcomp>@ s& � �E�E�E��T�&�\�W�-D�-D��-D�-D�-Dr �r- )r3 s r r> zABIAnalyzer._get_events> s � �E�E���E�E�E�Er c �D � g }| j D ]�}|d dk r�|� d� � dk rnt |� dg � � � � dk rG|� |d |� d� � r|d d d nd d
�� � ��|S ) z,Extract state variables from view functions.r rD rM r r# r r r$ �unknown)r r )r- r2 �lenrN )r3 �
state_varsrO s r r? z ABIAnalyzer._get_state_variablesB s� � ��
��H� � �D��V� �
�*�*����*�+�+�v�5�5��D�H�H�X�r�*�*�+�+�q�0�0��!�!� ��L�:>�(�(�9�:M�:M� \�D��O�A� .�v� 6� 6�S\�#� #� � � �� �r c �: � | j D ]}|d dk r|c S �dS )z%Extract the constructor from the ABI.r r< NrT )r3 rO s r r@ z ABIAnalyzer._get_constructorO s5 � ��H� � �D��F�|�}�,�,�� � � � -��tr �paramc � � � d}d|v r� fd�|d D � � }t |� dd� � |d |�� � S )z Parse a parameter definition.Nr c �: �� g | ]}�� |� � ��S r rF )rI �cr3 s �r rK z0ABIAnalyzer._parse_parameter.<locals>.<listcomp>Z s' �� �P�P�P�q�$�/�/��2�2�P�P�Pr r � r )r r r )r r2 )r3 rZ r s ` r rG z ABIAnalyzer._parse_parameterV sc �� ��
� �5� � �P�P�P�P�E�,�<O�P�P�P�J� ����6�2�&�&��v� �!�
�
�
�
r N)r r r r r r r6 rB r r" r= r> r? r r@ r rG r r r r) r) s� � � � � � �7�%��T� �"2� 7� 7� 7� 7� �� � � � � ��%7� 8� � � � � F�T�$�Z� F� F� F� F� �d�4�j� � � � ��(�4�.� � � � �
�d�
�/@�
�
�
�
�
�
r r) )�typingr r r r r0 � dataclassesr �enumr �pathlibr r r r" r) r r r �<module>rc s6 �� .� .� .� .� .� .� .� .� .� .� .� .� � � � � !� !� !� !� !� !� � � � � � � � � � � � �� � � � �4� � � � �;� ;� ;� ;� ;� ;� ;� ��;�
�!� !� !� !� !� !� !� ��!�E
� E
� E
� E
� E
� E
� E
� E
� E
� E
r
