o
�c�gT* � @ s� d Z ddlZddlZddlZddlmZmZmZmZ ddl m
Z
m Z m Z m
Z
ddlmZmZ ddlmZ ddlmZ ddlmZ ddlZddlZe
d d
�Zejd ed d
�d d� edd
�ZG dd� de�Z G dd� de�Z G dd� de�Z G dd� de�Z dZ ej!ej"�#e �dd� ej$de d �d d � �Z%ej&de d �d e fd d!��Z'e�&d"�d#e fd$d%��Z(e�)d&�d'e*fd(d)��Z+e�&d*�d+d,� �Z,e�&d-�d.e-fd/d0��Z.e�/d1�d2e fd3d4��Z0ej$d5ed6�d2e fd7d8��Z1d=d;d<�Z2dS )>zW
HTTP-MCP Server Configuration Tool UI
Provides visual interface for API configuration
� N)�Dict�List�Optional�Union)�FastAPI�
HTTPException�Request�Response)� HTMLResponse� JSONResponse)� StaticFiles)�Jinja2Templates)� BaseModelzHTTP-MCP Configuration Tool)�titlez/staticZstatic)Z directory��namezstatic/templatesc @ s: e Zd ZU eed< eed< dZeed< dZee ed< dS )�
ApiParamModel�type�descF�requiredN�default) �__name__�
__module__� __qualname__�str�__annotations__r �boolr r � r r �QD:\Data\Money\2025\ai-fund-assistant\mcp-servers\http-for-mcp-server\config_ui.pyr s
r c @ s& e Zd ZU eed< dZee ed< dS )�ApiResponseFieldModel�pathNr )r r r r r r r r r r r r % s
r c @ sv e Zd ZU eed< eed< eed< dZeed< i Zeeef ed< i Z eeef ed< i Z
eee ee f f ed< d S )
� ApiToolModelr � description�url�GET�method�params�headers�responseN)
r r r r r r% r& r r r' r( r r r r r r r! ) s
r! c @ s e Zd ZU ee ed< dS )�ApiConfigModel�toolsN)r r r r r! r r r r r r) 2 s
r) zconfig/apis.jsonT��exist_okz /api/config)Zresponse_modelc
� s� �z,t j�t�s
dg iW S ttddd��} t�| �}W d � |W S 1 s&w Y |W S tyT } zdt|�� dt �
� � �}t |� t dd t|�� �d
��d}~ww ) z Get current API configurationr* �r�utf-8��encodingNz Failed to read configuration: �
�� u 读取配置失败: �� status_code�detail)
�osr �exists�API_CONFIG_PATH�open�json�load� Exceptionr � traceback�
format_exc�printr )�f�config�e� error_detailr r r �
get_config< s �
����rD rA c
� s� �z2t jt j�t�dd� ttddd��}tj| �� |ddd� W d � | W S 1 s,w Y | W S t yZ } zd
t
|�� d t � � � �}t
|� td d
t
|�� �d��d }~ww )zSave API configurationTr+ �wr. r/ F� �Z ensure_ascii�indentNz Failed to save configuration: r1 r2 u 保存配置失败: r3 )r6 �makedirsr �dirnamer8 r9 r: �dump�dictr< r r= r> r? r )rA r@ rB rC r r r � save_configL s �
����rM z /api/tool�toolc
� s� �z�t jt j�t�dd� t j�t�rIttddd��&}zt�|�}t di |��}W n tj
y8 t g d�}Y nw W d � n1 sCw Y nt g d�}|j D ]}|j | j krdt
dd | j � d
�d ��qQ|j �| � ttd dd��}tj|�� |d
dd� W d � n1 s�w Y dd | j � d�d�W S t
y� � ty� } zdt|�� dt�� � �}t|� t
ddt|�� �d ��d}~ww )z Add new toolTr+ r- r. r/ )r* N� � 工具 'u ' 已存在r3 rE FrF rG �successu ' 已添加��status�messagezFailed to add tool: r1 r2 u 添加工具失败: r )r6 rI r rJ r8 r7 r9 r: r; r) ZJSONDecodeErrorr* r r �appendrK rL r< r r= r> r? )rN r@ � config_data�current_configZ
existing_toolrB rC r r r �add_tool] s@ �
�� ��
� � ��rX z/api/tool/{tool_name}� tool_namec
� sX �z|t j�t�stddd��ttddd��}t�|�}tdi |��}W d � n1 s,w Y t |j
�}� fdd �|j
D �|_
t |j
�|krRtdd
� � d �d��ttd dd��}tj |� � |d
dd� W d � n1 sow Y dd
� � d�d�W S ty� � t
y� } zdt|�� dt�� � �}t|� tddt|�� �d��d}~ww )z Delete tooli� u 配置文件不存在r3 r- r. r/ Nc s g | ] }|j � kr|�qS r r )�.0rN �rY r r �
<listcomp>� s z delete_tool.<locals>.<listcomp>rP u ' 不存在rE FrF rG rQ u ' 已删除rR zFailed to delete tool: r1 r2 u 删除工具失败: r )r6 r r7 r8 r r9 r: r; r) �lenr* rK rL r< r r= r> r? )rY r@ rV rW Zoriginal_lengthrB rC r r[ r � delete_tool� s0 �
�
� ��r^ z /api/reloadc � s �zgt jdd�4 I dH �O} z | �d�I dH }|�� |�� W W d �I dH W S t jyC } z
t|jjd|jj � �d��d}~w t j
yZ } z tddt |�� �d��d}~ww 1 I dH saw Y W dS t y� } zd t |�� d
t
�� � �}t|� tdd t |�� �d��d}~ww ) z%Notify server to reload configuration� $@��timeoutNz"http://localhost:8000/admin/reloadu 服务器返回错误: r3 r2 u 无法连接到MCP服务器: z Failed to reload configuration: r1 u 重载配置失败: )�httpx� AsyncClient�getZraise_for_statusr: �HTTPStatusErrorr r( r4 �text� RequestErrorr r< r= r> r? )�clientr( rB rC r r r �
reload_config� s4 �� �� ��� � ��ri z/api/test_request� request_datac � st ��z| � d�}| � dd�}| � di �}| � di �}| � di �}|s(tddd ��|||d
�}|dkr3n
|d v r=|r=||d < tjd
d�4 I d H ��}zZtd|� d|� �� td|� �� |rctd|� �� |jd i |��I d H }z|�� } d|j| d�W W W d �I d H W S d|j|jdd�d� Y W W d �I d H W S tj y� }
z t
ddd|
j j� �d�d�W Y d }
~
W d �I d H W S d }
~
w tj y� }
z t
dddt
|
�� �d�d�W Y d }
~
W d �I d H W S d }
~
ww 1 I d H �sw Y W d S t�y9 }
z$d t
|
�� d t�� � �} t| � t
ddd t
|
�� �d�d�W Y d }
~
S d }
~
ww ) Nr# r% r$ r' r& �bodyrO u URL不能为空r3 )r% r# r' )ZPOSTZPUTZPATCHr: r_ r` u 发送API测试请求: � u 请求头: u 请求体: T)rQ r4 �data)�contentZ raw_responser2 Fu HTTP错误: 状态码 )rQ �error�r4 rn u 请求错误: zTest request failed: r1 u 测试请求失败: r )rd r rb rc r? �requestr: r4 rf re r r( rg r r< r= r> ) rj r# r% r' r& rk Zrequest_kwargsrh r( �resultrB rC r r r � test_request� s� �
����
����� ��� �&���rs Zhttprq c
� s� �z|| �I d H W S t yM } z8dt|�� dt�� � �}t|� t|t�r6t|jd|j id�W Y d }~S tdddt|�� �id�W Y d }~S d }~ww )NzUnhandled exception: r1 r5 rp r2 u 服务器内部错误: )
r< r r= r> r? �
isinstancer r r4 r5 )rq Z call_nextrB rC r r r �error_handling_middleware s" �
����ru �/)Zresponse_classc � s �t �dd| i�S )zDConfiguration tool homepage with Apple Design + Pixel style + emojisz
index.htmlrq )� templatesZTemplateResponse)rq r r r � read_root s �rx �0.0.0.0�A c C s ddl }|jt| |d� dS )z Run configuration tool UI serverr N)�host�port)�uvicorn�run�app)r{ r| r} r r r �
run_config_ui! s r� )ry rz )3�__doc__r: r6 r= �typingr r r r Zfastapir r r r Zfastapi.responsesr
r Zfastapi.staticfilesr Zfastapi.templatingr
Zpydanticr rb Zasyncior Zmountrw r r r! r) r8 rI r rJ rd rD ZpostrM rX �deleter r^ ri rL rs Z
middlewareru rx r� r r r r �<module> sL
'
K