�
�d�h� � �T � d dl Z d dlZd dlmZmZ d dlmZ d dlmZ d dlm Z d dl
m Z d dl m
Z
d dlmZ ej e� � Z e � � Z e � � Z e
� � Z e� � Z G d � d
� � Z e� � Z ej d d g�
� � ed� � � � � ZdS )� N)�request�jsonify)�app)�require_api_key)�
DexOperations)�LendingOperations)�YieldFarmingOperations)�PortfolioManagerc �N � e Zd ZdZd� Zd� Zd� Zd� Zd� Zd� Z d� Z
d � Z d
� Z d � Z
d S )
� MCPServerz1Model Context Protocol Server for DeFi operationsc �v � | j | j | j | j | j | j | j | j d�| _ d S )N)z defi.swapz defi.lendz defi.farmzdefi.portfoliozdefi.positionszdefi.transaction_statuszdefi.protocolsz defi.chains) � handle_swap� handle_lend� handle_farm�handle_portfolio�handle_positions�handle_transaction_status�handle_protocols�
handle_chains�methods)�selfs �$/home/runner/workspace/mcp_server.py�__init__zMCPServer.__init__ sD � � �)� �)� �)� "� 3� "� 3�'+�'E� "� 3� �-�
�
�� � � � c � � |� d� � }|� di � � }|� d� � }|| j vr dddd�|d�S | j | |� � }d||d �S # t $ r^}t � d
t |� � � �� � dd d t |� � d
�|� d� � d�cY d}~S d}~ww xY w)zHandle JSON-RPC 2.0 request�method�params�id�2.0i����zMethod not found��code�message��jsonrpc�errorr )r$ �resultr zMCP request failed: 饀���Internal error)r! r" �dataN)�getr � Exception�loggerr% �str)r r) r r �
request_idr& �es r �handle_requestzMCPServer.handle_request" s5 � �! ��X�X�h�'�'�F��X�X�h��+�+�F� ���$� � �J��T�\�)�)� $� &�#5� � � %�
� � � *�T�\�&�)�&�1�1�F� !� � �� �
�� �
�
�
� �L�L�8��A���8�8� 9� 9� 9� � "� /� ��F�F�� �
�h�h�t�n�n�� �
�
�
�
�
�
�����
���s% �AA0 �A0 �0
C�:AC�
C�Cc �� � t � |d |d |d |d |d |� dd� � |� dd � � �
� � S ) zHandle DEX swap operation�
blockchain�wallet_address�token_in� token_out� amount_in�slippageg �?�protocol�uniswap)r2 r3 r4 r5 r6 r7 r8 )�dex_ops� execute_swapr* �r r s r r zMCPServer.handle_swapG sj � ��#�#� �l�+�!�"2�3��J�'� �[�)� �[�)��Z�Z�
�C�0�0��Z�Z�
�I�6�6� $�
�
�
r c �| � t � |d |d |d |d |d �� � S )zHandle lending operationr2 r8 r3 �token�amount)r2 r8 r3 r> r? )� lending_ops�
lend_assetr< s r r zMCPServer.handle_lendS sG � ��%�%� �l�+��J�'�!�"2�3���/��(�#� &�
�
�
r c � � t � |d |d |d |d |d |d |d |d � � � S )
z Handle yield farming operationr2 r8 r3 �pool_id�token_a�token_b�amount_a�amount_b)r2 r8 r3 rC rD rE rF rG )� yield_ops�
add_liquidityr< s r r zMCPServer.handle_farm] s` � ��&�&� �l�+��J�'�!�"2�3��9�%��9�%��9�%��J�'��J�'� '�
�
�
r c �P � t � |d |d � � S )zHandle portfolio queryr3 r2 )�
portfolio_mgr�
get_portfolior< s r r zMCPServer.handle_portfolioj �+ � � �*�*� �#� $� �<� �
�
�
r c �P � t � |d |d � � S )zHandle positions queryr3 r2 )rK �
get_positionsr< s r r zMCPServer.handle_positionsq rM r c � � |d dddd�S )z Handle transaction status query�tx_hash� confirmediNa� �21000)rQ �status� block_number�gas_used� r< s r r z#MCPServer.handle_transaction_statusx s$ � � �i�(�!� $� �
�
�
r c � � dg d�ddgddgd�iS )z Handle supported protocols query� protocols)r9 �compound�aave� quickswapr[ �raydium�orca��ethereum�polygon�solanarW r< s r r zMCPServer.handle_protocols� s4 � �
� ;� ;� ;� '��0�$�f�-�� �
�
r c � � g d�dd�S )z Handle supported chains queryr_ r` )�chains�
default_chainrW r< s r r zMCPServer.handle_chains� s � � 8�7�7� '�
�
�
r N)�__name__�
__module__� __qualname__�__doc__r r0 r r r r r r r r rW r r r r s� � � � � � �;�;�
�
�
�#� #� #�J
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
r r z/mcp�POST)r c � � t j � � } | st dddd�dd�� � dfS t | t � � rEg }| D ]1}t
� |� � }|� |� � �2t |� � S t
� | � � }t |� � S # t $ rL}t �
dt |� � � �� � t dd d
d�dd�� � d fcY d}~S d}~ww xY w) zMCP JSON-RPC endpointr iD���z Parse errorr Nr# i� zMCP endpoint error: r' r( i� ) r �get_jsonr �
isinstance�list�
mcp_serverr0 �appendr+ r, r% r- )r) �results�reqr& r/ s r � mcp_endpointrs � so � �!�� �!�!��� �� � "� ,�� � �
� � � � ��
� �d�D� !� !� #��G��
'�
'��#�2�2�3�7�7�����v�&�&�&�&��7�#�#� #� �.�.�t�4�4�F��6�?�?� "�� � � � �� � �4�C��F�F�4�4�5�5�5�� � �+�� � �
�
� � �
�� � � � � � ����� ���s* �,B2 �AB2 � (B2 �2
D�<AD�=D�D)�json�logging�flaskr r r �routesr �defi_tools.dex_operationsr �defi_tools.lendingr �defi_tools.yield_farmingr �defi_tools.portfolior
� getLoggerrf r, r: r@ rH rK r ro �routers rW r r �<module>r~ sc �� � � � � ���� "� "� "� "� "� "� "� "� � � � � � � "� "� "� "� "� "� 3� 3� 3� 3� 3� 3� 0� 0� 0� 0� 0� 0� ;� ;� ;� ;� ;� ;� 1� 1� 1� 1� 1� 1� �� �8� $� $�� �-�/�/�� � �!�!� � "� "� $� $� � � �"�"�
�}
� }
� }
� }
� }
� }
� }
� }
�@ �Y�[�[�
�
���6�F�8�$�$�$��#� #� � � %�$�#� #� #r