a
��g�! � @ s d dl Z d dlZd dlZd dlmZmZmZmZmZ d dl Z d dl
Z
d dl m Z m
Z
d dlmZ d dlmZmZ d dlmZ d dlmZmZ ejejdd � e�d
�ZG d d � d e�Z G d
d� de�Z G dd� de�Z G dd� de�Z dd� Z dd� Z!d dd�Z"dd� Z#G d d � d �Z$dS ) � N)�Any�Dict�List�Optional�Union)�FastMCP�Server)�SseServerTransport)� BaseModel�Field)� Starlette)�Route�Mountz4%(asctime)s - %(name)s - %(levelname)s - %(message)s)�level�format�
mcp_serverc @ s6 e Zd ZU eed< eed< dZeed< dZeed< dS )�ApiParam�type�descF�requiredN�default) �__name__�
__module__� __qualname__�str�__annotations__r �boolr r � r r �+E:\data\mcp-server\http-4-mcp\mcp_server.pyr s
r c @ s& e Zd ZU eed< dZee ed< dS )�ApiResponseField�pathNr )r r r r r r r r r r r r s
r c @ s� e Zd ZU eed< eed< eed< dZeed< eeef ed< ee d�Z
eeef ed< ee d�Z eee ee
f f ed < d
Zeeeef ed < d
Zeeeef ed < d
S )
�ApiTool�name� description�url�GET�method�params)�default_factory�headers�responseN�response_transform�auth)r r r r r r& r r r �dictr) r* r r r+ r r r, r r r r r! # s
"r! c @ s e Zd ZU ee ed< dS )� ApiConfig�toolsN)r r r r r! r r r r r r. . s
r. c C s> |� d�}| }|D ]&}t|t�r2||v r2|| }q d S q|S )N�.)�split�
isinstancer- )�datar �keys�result�keyr r r �get_nested_value2 s
r7 c
C s� zHt | ddd��}t�|�}W d � n1 s00 Y tf i |��W S ty~ } z t�d|� �� � W Y d }~n
d }~0 0 d S )N�r�utf-8��encodingz"Failed to load API configuration: )�open�json�loadr. � Exception�logger�error)� config_path�fZ config_data�er r r �load_api_config< s (rE � c
� s@ i }| j �� D ]H\}}||v r,|| ||< q|jd urB|j||< q|jrtd|� ���q| j�� }| jr�| j�d�}|dkr�d| j�d�� �|d< nF|dkr�| j�dd�}| j�d d
�} | j�d �d kr�| ||< n| ||< t �
� 4 I d H ��8}
z�| j � � d
k�r |
j| j
|||d�I d H } n�| j � � dk�rH|
j| j
|||d�I d H } nh| j � � dk�rt|
j| j
|||d�I d H } n<| j � � dk�r�|
j| j
|||d�I d H } ntd| j � ���| �� | �� W W d �I d H S t�y } z t�d| � �� � W Y d } ~ n
d } ~ 0 0 W d �I d H �q<1 I d H �s20 Y d S )Nz Missing required parameter: r ZbearerzBearer ZtokenZ
AuthorizationZapi_key�key_name� key_value� �in�headerr% )r' r) �timeoutZPOST)r= r) rL ZPUTZDELETEzUnsupported HTTP method: zHTTP request error: )r' �itemsr r �
ValueErrorr) �copyr, �get�httpxZ AsyncClientr& �upperr$ ZpostZput�deleteZraise_for_statusr= r? r@ rA )
�toolr' rL Zprocessed_paramsr" Z param_configr) Z auth_typerG rH Zclientr* rD r r r �make_http_requestE sn
���� rU c C sJ |s| S i }|� � D ]0\}}t|t�r,|}n
|�d�}t| |�||< q|S )Nr )rM r2 r rP r7 )r3 �mappingr5 Z
output_key�sourcer r r r �transform_response� s
rX c @ s@ e Zd Zddd�Zdd� Zdd� Zd d
� Zd d � Zddd�Zd
S )�
HttpMcpServer�config/config.json�config/apis.jsonc C s@ || _ || _| �� | _t|�| _td�| _| �� | � � | _
d S )NZhttp_mcp_server) rB � apis_path� _load_config�configrE �
api_configr �mcp�_register_tools�_create_starlette_app�app)�selfrB r\ r r r �__init__� s
zHttpMcpServer.__init__c
C s� z@t | jddd�� }t�|�W d � W S 1 s40 Y W nF ty� } z.t�d|� �� ddddd d
d �W Y d }~S d }~0 0 d S ) Nr8 r9 r: zFailed to load config: �0.0.0.0�@ F�info�
rF )�host�port�debug� log_levelZ max_workersrL )r< rB r= r> r? r@ rA )rd rC rD r r r r] � s .�zHttpMcpServer._load_configc C s | j jD ]}| �|� qd S �N)r_ r/ �_register_tool�rd � tool_configr r r ra � s z HttpMcpServer._register_toolsc s< � �fdd�}� j j�j�jd�|� t�d�j� �� d S )Nc
� s� z<t �| � j�dd�d�I d H }�jr8t|�j�}|W S |W S ty� } z0t�d�j� d|� �� dt |�iW Y d }~S d }~0 0 d S )NrL rF )rL zTool z execution failed: rA )
rU r^ rP r* rX r? r@ rA r" r )�kwargsZ
response_dataZtransformed_datarD rp r r � tool_handler� s � z2HttpMcpServer._register_tool.<locals>.tool_handler)r" r# zRegistered API tool: )r` rT r" r# r@ rh )rd rq rs r rp r ro � s ��z HttpMcpServer._register_toolc sd �j j� td��� �fdd�}�fdd�}t�j�dd�td|d �td�jd
�td |d �gd �}|S )
Nz
/messages/c � sh �� | j| j| j�4 I d H �2\}}� �||� �� �I d H W d �I d H qd1 I d H sZ0 Y d S rn )Z connect_sseZscopeZreceiveZ_send�runZ create_initialization_options)�requestZ read_streamZ write_stream)r �sser r �
handle_sse� s � �z7HttpMcpServer._create_starlette_app.<locals>.handle_ssec
� sh z t � j�� _� �� ddd�W S tyb } z*t�d|� �� dt|�d�W Y d }~S d }~0 0 d S )NZsuccessz#Configuration reloaded successfully)�status�messagezFailed to reload config: rA )rE r\ r_ ra r? r@ rA r )ru rD )rd r r �
reload_config� s z:HttpMcpServer._create_starlette_app.<locals>.reload_configrl Fz/mcp/sse)Zendpoint)rc z
/admin/reload)rl Zroutes) r` Z _mcp_serverr r r^ rP r
r Zhandle_post_message)rd rw rz rc r )r rd rv r rb � s
�� z#HttpMcpServer._create_starlette_appNc C s\ |p| j �dd�}|p"| j �dd�}t�d|� d|� �� tj| j||| j �dd�d � d S )
Nrj rf rk rg z.Starting HTTP-MCP middleware server at http://�:rm rh )rj rk rm )r^ rP r@ rh �uvicornrt rc )rd rj rk r r r rt � s �zHttpMcpServer.run)rZ r[ )NN) r r r re r] ra ro rb rt r r r r rY � s
$rY )rF )%r= Zlogging�os�typingr r r r r rQ r| Z
mcp.serverr r Zmcp.server.sser Zpydanticr
r Zstarlette.applicationsr Zstarlette.routingr
r Z basicConfig�INFOZ getLoggerr@ r r r! r. r7 rE rU rX rY r r r r �<module> s0 �
A