o
?+h\% � @ sZ d dl Z d dlZd dlZd dlmZ d dlmZmZmZm Z m
Z
m Z m Z d dl
mZ d dlmZ d dlmZ d dlmZ d dlmZ d d lmZ d d
lmZ d d l m Z m Z d d l m Z d d
l!m"Z"m#Z# d dl$m%Z% e� e j&e j'ej(d� e �)e*�Z+ed�Z,dZ-dZ.dZ/G dd� d�Z0edddd�Z1e�2dd�Z3e0e�2dd�d� e d�fd e4fd d �Z5e d �Z6d efd!d"�Z7e1j8j9�:e#d#e7d$gd%�� e1j8j9�:e"d e6j;d&�� e1j<d'e e5�gd(�d)d*� �Z=e1j>d+e e5�gd(�e d,�e
d,�fd-e4d.e?fd/d0��Z@e1�<d1�d2d3� �ZAd4d5� ZBe*d6k�r+d7ejCv �r&d dlDZDeDjEd8d9eFe�2d:d;��d<d=� dS eB� dS dS )>� N)� load_dotenv)�FastAPI�Request�
HTTPException�Depends�Body�Path�Header)� APIKeyHeader)�BaseHTTPMiddleware)r )�Response)�FastMCP)� TonClient)�
PromptManager)� ToolManager�base64url_to_hex)�SseServerTransport)�Mount�Route)�Any)�level�streamzTON MCP ServerZ0EQAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAM9cZ0UQBXbfJhkqlCpDXPn_x5uXDR_cqC7xfjx3jhwx5DOO1DWqZnZ0EQAvlWFDxGF2lXm67y4yzC17wYKD9A0guwPkMs1gOsM__NOTc @ s0 e Zd Zd dedefdd�Zdd� Zdd � Zd
S ) � TonMcpServer� https://tonapi.io�api_key�base_urlc C sF t �d|� || _t||�| _t� | _t| j�| _| � � | �
� d S )Nz6Initializing TonMcpServer with API key and base_url=%s) �logger�debugr r Z
ton_clientr �prompt_managerr � tool_manager�_register_tools�_register_prompts)�selfr r � r# �G/Users/devon/ton-mcp/ton-blockchain-mcp-remote/src/tonmcp/mcp_server.py�__init__! s zTonMcpServer.__init__c s� t �d� tjdd�d!dtdtdtf� fdd �
�}tjd
d�d tdtf� fd d
� �}tjdd�d"dtdtdtf� fdd�
�}tjdd�d#dtdtdtf� fdd�
�}tjdd�d$dtdtf� fd d �
�}tjd d�d$dtdtdtf� fd d �
�}d S )%NzRegistering tools...z�Analyze a TON address for its balance, jetton holdings, NFTs, and recent activity. Optionally performs deep forensic analysis if deep_analysis is True. Use for questions about account overview, holdings, or activity.)� descriptionF�address�
deep_analysis�returnc � sB �t �d| � d|� �� � jj| |d�I dH }t �d|� �� |S )z�Example: analyze_address(address='UQ...youraddress...', deep_analysis=True)
Analyze a TON address including balance, transactions, and patterns.
z$analyze_address called with address=z, deep_analysis=)r' r( Nzanalyze_address result: )r r r �analyze_address)r' r( �result�r" r# r$ r* , s
�z5TonMcpServer._register_tools.<locals>.analyze_addressz�Get details and analysis for a specific TON blockchain transaction by its hash. Use for questions about a particular transaction, its participants, value, or type.�tx_hashc � � �� j j| d�I dH S )z�Example: get_transaction_details(tx_hash='14069dfe829c040e81dd983a0022eec38b45505591a538c47c032364fa0bccb9')
Get details for a transaction hash.
�r- N)r �get_transaction_detailsr/ r, r# r$ r0 8 s �z=TonMcpServer._register_tools.<locals>.get_transaction_detailsz�Find trending tokens, pools, or accounts on the TON blockchain for a given timeframe and category. Use for questions about what's hot, trending, or popular on TON.�1h�tokens� timeframe�categoryc � � �� j j| |d�I dH S )zlExample: find_hot_trends(timeframe='1h', category='tokens')
Find hot trends on TON.
�r3 r4 N)r �find_hot_trendsr6 r, r# r$ r7 A s �z5TonMcpServer._register_tools.<locals>.find_hot_trendsz�Analyze trading patterns for a TON address over a specified timeframe. Use for questions about trading activity, frequency, jetton transfers, or DEX swaps for an account.�24hc � r5 )zzExample: analyze_trading_patterns(address='UQ...youraddress...', timeframe='24h')
Analyze trading patterns for an address.�r' r3 N)r �analyze_trading_patternsr9 r, r# r$ r: J � �z>TonMcpServer._register_tools.<locals>.analyze_trading_patternszfGet the current real-time TON price in the specified currency (default: USD) and recent price changes.�usd�currencyc � r. )z�Example: get_ton_price(currency='usd')
Get the current real-time TON price in the specified currency (default: USD) and recent price changes.�r= N)r �
get_ton_pricer> r, r# r$ r? Q s �z3TonMcpServer._register_tools.<locals>.get_ton_pricez�Get the current price and recent changes for specified jetton tokens (not TON) in the given currency. Provide a list of jetton master addresses as tokens.c � r5 )z�Example: get_jetton_price(tokens=['0:dfbbf59e9b306b86194a2441f4f80b51bbd68253290549bd76a7165c0082ae80'], currency='usd')
Get the current price and recent changes for specified jetton tokens (not TON) in the given currency.�r2 r= N)r �get_jetton_pricer@ r, r# r$ rA X r; z6TonMcpServer._register_tools.<locals>.get_jetton_price)F)r1 r2 )r8 )r< )r r �tmcp�tool�str�boolr �list)r" r* r0 r7 r: r? rA r# r, r$ r * s2
� �� � ��"z TonMcpServer._register_toolsc s\ t �d� t�� dtf� fdd� �}t�� dtf� fdd� �}t�� dtf� fdd� �}d S ) NzRegistering prompts...r) c � � �� j jdi | ��I d H S �Nr# )r Zget_trading_analysis_prompt��kwargsr, r# r$ �trading_analysisa � �z8TonMcpServer._register_prompts.<locals>.trading_analysisc � rG rH )r Zget_forensics_promptrI r, r# r$ �forensics_investigatione rL z?TonMcpServer._register_prompts.<locals>.forensics_investigationc � rG rH )r Zget_trend_analysis_promptrI r, r# r$ �trend_analysisi rL z6TonMcpServer._register_prompts.<locals>.trend_analysis)r r rB �promptrD )r" rK rM rN r# r, r$ r! _ s
z TonMcpServer._register_promptsN)r )�__name__�
__module__� __qualname__rD r% r r! r# r# r# r$ r s 5r zTON MCP Remote Server)�titleZdocs_urlZ redoc_url�API_KEYZchangeme� TON_API_KEY�r �
authorizationc C sB | r| � d�s
tddd��| �dd�d }|tkr tddd��|S )NzBearer i� Z Forbidden�Z status_codeZdetail� � )�
startswithr �splitrT )rW Ztokenr# r# r$ � get_api_keyt s r] z
/messages/�requestc � sx �t �| j| j| j�4 I d H � }tj�|d |d tj�� �I d H W d �I d H t � S 1 I d H s4w Y t � S )Nr rZ )
�
sse_transportZ connect_sseZscopeZreceiveZ_sendrB Z _mcp_server�runZ create_initialization_optionsr )r^ Zstreamsr# r# r$ � sse_endpoint� s � ���ra z/sseZGET)Zendpoint�methods)�appz/tools)Z dependenciesc � sr �g } t j�� D ],}d }|jjr%|jj�� �d�d }d|�� v r%|�� }| �|j |j |j
|j |d�� qd| iS )N�
r zexample:)�id�namer& �
parameters�
usage_example�tools) rB �
_tool_manager�
list_tools�fn�__doc__�stripr\ �lower�appendrf r& rg )ri rC rh Z
first_liner# r# r$ rk � s �
�rk z/tools/{tool_id}/call.�tool_id�bodyc
� s� �t � fdd�tj�� D �d �}|stddd��z
|�|�I d H }d|iW S tyA } zt�d� � �� tdt |�d��d }~ww ) Nc 3 s � | ]
}|j � kr|V qd S )N)rf )�.0�t�rq r# r$ � <genexpr>� s � z call_tool.<locals>.<genexpr>i� zTool not foundrX r+ zError calling tool i� )
�nextrB rj rk r r` � Exceptionr Z exceptionrD )rq rr rC r+ �er# ru r$ � call_tool� s �
��rz z/healthzc C s ddiS )NZstatus�okr# r# r# r# r$ �health� s r| c C sH t �d�} t�d| � �� | std��t| � t�d� tjdd� d S )NrU z Starting main with TON_API_KEY=z,TON_API_KEY environment variable is requiredz"Running FastMCP server on STDIO...Zstdio)Z transport)�os�getenvr r �
ValueErrorr rB r` rV r# r# r$ �main� s
r� �__main__Z runserverztonmcp.mcp_server:appz0.0.0.0ZPORTi@ T)ZhostZport�reload)GZloggingr} �sysZdotenvr Zfastapir r r r r r r Zfastapi.securityr
Zstarlette.middleware.baser Zstarlette.requestsZStarletteRequestZstarlette.responsesr ZStarletteResponseZmcp.server.fastmcpr
Ztonmcp.ton_clientr Ztonmcp.promptsr Z tonmcp.toolsr r Zmcp.server.sser Zstarlette.routingr r �typingr Z basicConfigZDEBUG�stderrZ getLoggerrP r rB ZTON_COIN_ADDRESSZTEST_WALLET_ADDRESSZTEST_JETTON_ADDRESSr rc r~ rT rD r] r_ ra ZrouterZroutesrp Zhandle_post_message�getrk Zpost�dictrz r| r� �argvZuvicornr` �intr# r# r# r$ �<module> sZ $
M
"
"
�
