cli.cpython-311.pyc•3.34 kB
�
\C�g� �
�D � d dl Z d dlZd dlmZ d dlmZ ddlmZ d dlZd dl Z ddl
m Z e j � � d� � � Z
e
� � � e j d e j d �
� � � � � e j d e j � � � � � e j d
� � ded ed
efd�� � � � � � � � Ze
� � � e j d e j d �
� � � � � defd�� � � � Zedk r e
� � dS dS )� N)�Path)�Optional� )� ABIAnalyzer)� MCPGeneratorc � � dS )zMCP Server Generator CLIN� r � �//Users/arjun/repos/sparkmango/mcp_server/cli.py�clir
s � � �Dr
�abi_fileT)�exists)�type�
output_dir�
contract_namec �� � t j d� � }|st j dd�� � dS t | � � 5 }t j |� � }ddd� � n # 1 swxY w Y t |� � }|� � � }t |t |� � ||�� � }t j |�
� � � � t j d|� �� � dS )z+Generate an MCP server from a contract ABI.�OPENAI_API_KEYz5Error: OPENAI_API_KEY environment variable is not setT)�errN)�analysisr r �openai_api_keyzMCP server generated in )�os�getenv�click�echo�open�json�loadr �analyzer r �asyncio�run�generate) r
r r r �f�abi�analyzerr � generators r r! r! s, � � �Y� /�0�0�N� � �
�
�J�PT�U�U�U�U���
�h� � � �1��i��l�l��� � � � � � � � � � ���� � � � � �3� � �H�� � �!�!�H� ���
�#�#�#�%� � � �I� �K� �"�"�$�$�%�%�%� �J�6�*�6�6�7�7�7�7�7s �A � A#�&A#� cache_dirc �� � t | � � }|� � � st j d� � dS |� d� � D ]}|� � � �t j d� � dS )z&Clear the method implementation cache.z Cache directory does not existNz*.pyz
Cache cleared)r r r r �glob�unlink)r& �
cache_path�files r � clear_cacher, 0 s{ � � �i� � �J� � � � � � �
�
�3�4�4�4��� ����'�'� � �� � � �
�
�
�
� �J�� � � � � r
�__main__)r r �pathlibr �typingr � abi_analyzerr r r �
mcp_generatorr �groupr �command�argument�strr! r, �__name__r r
r �<module>r7 s� �� � � � � � � � � � � � � � � � � � � � � %� %� %� %� %� %� � � � � ���� '� '� '� '� '� '� ��� � � � � � � � ��� � ��� �
����4�!8�!8�!8�9�9�9��� � �:�5�:�<�<�0�0�0��� �� � �8�s� 8�� 8�C� 8� 8� 8� !� � 1�0� :�9� � �8�: ��� � ��� � �*�%�*�D�"9�"9�"9�:�:�:� �3� � � � ;�:� � � � �z����C�E�E�E�E�E� �r