server.cpython-312.pyc•5.73 kB
�
Yg�
� �& � d dl mZ d dlZd dlmZ d dlZd dlmZ d dlm Z d dl
m Z m Z m
Z
mZ d dlmZ d dlmZ d dlZ ej( ej* d �
� ej, e� Z G d � d ee� Z G d
� de� Z G d� d� Z dd�Z y)� )�EnumN)�Sequence)�Server)� stdio_server)�Tool� TextContent� ImageContent�EmbeddedResource)�McpError)� BaseModelz)%(asctime)s - %(levelname)s - %(message)s)�level�formatc � � e Zd ZdZy)�
ShellTools�execute_commandN)�__name__�
__module__� __qualname__�EXECUTE_COMMAND� � �I/Users/tengjizhang/projects/mcp/cli-server/src/mcp_server_shell/server.pyr r s � �'�Or r c �, � e Zd ZU eed< eed< eed< y)�
CommandResult�command�output� return_codeN)r r r �str�__annotations__�intr r r r r s � �
�L� �K��r r c � � e Zd Zdedefd�Zy)� ShellServerr �returnc � � t j d|� �� t j |ddd�� }t j d|j
|j z � �� t ||j
|j z |j �� S # t $ r=}t j dt |� � �� t dt |� � �� �d}~ww xY w)z-Execute a shell command and return the resultzExecuting command: T)�shell�capture_output�textzCommand output: )r r r zCommand execution failed: N)
�logger�info�
subprocess�run�debug�stdout�stderrr �
returncode� Exception�errorr r )�selfr �result�es r r zShellServer.execute_command s� � �� � �)�'��3�4� B� �^�^��� #�� �F�
�L�L�+�F�M�M�F�M�M�,I�+J�K� L� � � �}�}�v�}�}�4� "� -� -��
��
� B� �L�L�5�c�!�f�X�>� ?�� 7��A��x�@�A� A�� B�s �A6B � C�8C�CN)r r r r r r r r r r" r" s � �B�s� B�}� Br r" c � �V �K � t j d� t d� } t � �| j � dt
t fd�� }| j � dt dt dt t t z t z f�fd� � }t j d� | j � }t � 4 �d { ��� \ }}t j d � | j! |||� � d { ��� d d d � �d { ��� y 7 �H7 �7 � # 1 �d { ���7 sw Y y xY w�w)
Nz Starting MCP shell server...z mcp-shellr# c
� � K � t j d� t t j j
ddddddd�idgd ��
� gS �w) zList available shell tools.zListing available toolszExecute Shell Commandz-Execute a shell command and return its output�objectr �stringzShell command to execute)�type� description)r9 �
properties�required)�name� display_namer: � inputSchema)r( r, r r r �valuer r r �
list_toolszserve.<locals>.list_tools4 s^ � �� � � � �.�/� � �/�/�5�5� 4� K� $�!�$,�+E�$�#� "+� � �
�
�
�s �AA
r= � argumentsc � � �K � t j d| � d|� �� | t j j k( r`|j d� }|s t
d� ��j |� }t dt j |j � d�� �� gS t
d | � �� �# t $ r }t
d
t |� � �� �d }~ww xY w�w) z.Handle tool calls for shell command execution.zTool call received: z with arguments: r z"Missing required argument: commandr' � )�indent)r9 r' zUnknown tool: z Error executing shell command: N)r( r) r r r@ �get�
ValueErrorr r �json�dumps�
model_dumpr0 r )r= rB r r3 r4 � shell_servers �r � call_toolzserve.<locals>.call_toolJ s� �� �� �
� � �*�4�&�0A�)��M�N� I��z�1�1�7�7�7�#�-�-� �2�� �$�%I�J�J�%�5�5�g�>�� �V�$�*�*�V�=N�=N�=P�YZ�2[�\�� � !�>�$��!8�9�9��� I� � >�s�1�v�h� G�H� H�� I�s/ � C�A<B+ � C� B+ �+ C�4C � C�CzInitializing server...z+Server started and ready to handle requests)r( r) r r" rA �listr rL r �dictr r r r
� create_initialization_optionsr r+ )�serverrA rL �options� read_stream� write_streamrK s @r �serverT / s �� �� �
�K�K�.�/�
�K�
�F� �=�L� ����
�d�4�j�
� �
�* ����I��I� "�I� �+� �,�/?�?� @�I� �I�* �K�K�(�)��2�2�4�G��~� =� =�!<�+�|�� � �A�B��j�j��l�G�<�<�<�=� =� =��<��=�� =� =� =�sZ �CD)�D�D)� /D�8D�9D�= D)�D� D)�D�D)�D&�D �D&�"D))r# N) �enumr rH �typingr r* �
mcp.serverr �mcp.server.stdior � mcp.typesr r r r
�mcp.shared.exceptionsr �pydanticr �logging� basicConfig�INFO� getLoggerr r( r r r r" rT r r r �<module>r` s �� � � � � � )� G� G� *� � � �� � �
�,�,� 6��
�� � �8� $��(��d� (��I� �
B� B�*5=r