�
Qi�h� �
�� � d dl Z d dlZd dlZd dlZd dlmZmZmZmZm Z d dl
m Z m Z m
Z
d dlmZ e j j# e j j% e� d� Z ej* � Zdeeeeeef f fd�Zdeeeeeef f ddfd �Zdeeeeeef f d
eddfd �Zdeeeeeef f d
ed ede ee eeeef f fd
�Z d
edeeef deeef fd�Z d
ed edeeef fd�Z d
edeeeef fd�Z d
ed eddfd�Z d
ed edeeef deeef fd�Z! e d�� Z"e"jG d� e
d� fd
edeeef fd�� Z$e"jK d� d
ed efd�� Z&e"jK d� d
efd�� Z'e"jQ d� d
ed efd �� Z)e"jU d� e
d� fd
ed edeeef fd �� Z+ ed � � Z,e,j[ d d!d"�#� d
edeeef fd$�� Z.e,j[ d%d&d'�#� d
ed efd(�� Z/e,j[ d)d*d+�#� d
efd,�� Z0e,j[ d-d.d/�#� d
ed efd0�� Z1e,j[ d1d2d3�#� d
ed edeeef fd4�� Z2e,jg � Z4e"jk d5e4� e"jK d6� d7� � Z6e7d8k( rd dl8Z8 e8jr d9d:d;d<�=� yy)>� N)�Any�Dict�List�Optional�Tuple)�FastAPI�
HTTPException�Body)�FastMCPzdb.json�returnc � � t j j t � si S t t dd�� 5 } t j | � cddd� S # t
j $ r i cY cddd� S w xY w# 1 sw Y yxY w)z5Load DB from disk; return {} if file missing/invalid.�r�utf-8��encodingN)�os�path�exists�DB_PATH�open�json�load�JSONDecodeError)�fhs �+/home/osoliman/workspace/mcp-server/main.py�load_dbr su � �
�7�7�>�>�'� "�� �
�g�s�W� -� �� ��9�9�R�=�� �� �#�#� ��I� � � ��� �s( �A<�A�A9�-A<�8A9�9A<�<B�dbc �� � t dz }t |dd�� 5 }t j | |dd�� ddd� t j
|t � y# 1 sw Y �$xY w) zAtomic write to disk.z.tmp�wr r � F)�indent� ensure_asciiN)r r r �dumpr �replace)r �tmpr s r �save_dbr& sO � �
�F�
�C�
�c�3�� )� 8�R� � � �"�b���7�8��J�J�s�G� �8� 8�s �A�A �
collectionc � � || vrg | |< y y �N� )r r' s r �ensure_collectionr+ " s � ������:�� � �item_idc � � t | j |g � � D ]1 \ }}t |j d� � t |� k( s�-||fc S y)N�id)NN)� enumerate�get�str)r r' r- �idx�its r � find_itemr5 & sN � � �R�V�V�J�� 3�4� ���R� �r�v�v�d�|� ��G� � ,���7�N�� r, �payloadc �, � t 5 t � }t || � t t j
� � }d|i}t
|t � r|j |� || j |� t |� |cd d d � S # 1 sw Y y xY w)Nr/ ) �_lockr r+ r2 �uuid�uuid4�
isinstance�dict�update�appendr& )r' r6 r r- �items r � _db_add_itemr@ 0 sv � � � �
�Y���"�j�)��d�j�j�l�#���g��� �g�t� $� �K�K�� �
�:�� � �d�#��� �� � � �s �A9B
�
Bc � � t 5 t � }t || |� \ }}|�
t dd�� �|cd d d � S # 1 sw Y y xY w�Ni� zItem not found)� status_code�detail)r8 r r5 r )r' r- r �_r? s r � _db_get_itemrF < sI � � � �
�Y���B�
�G�4���4� �<� �C�8H�I� I�� � � �s �+<�Ac �r � t 5 t � }|j | g � cd d d � S # 1 sw Y y xY wr) )r8 r r1 )r' r s r �_db_list_itemsrH D s/ � � � &�
�Y���v�v�j�"�%�&� &� &�s � -�6c �� � t 5 t � }t || |� \ }}|�
t dd�� �|| j |� t |� d d d � y # 1 sw Y y xY wrB )r8 r r5 r �popr& )r' r- r r3 r? s r �_db_delete_itemrK I s_ � � � �
�Y�� �b�*�g�6� ��T� �<� �C�8H�I� I�
�:����3� ��� �
� � �s �A A�A"c �� � t 5 t � }t || |� \ }}|�
t dd�� �t |t
� r|j
|� ||| |< t |� |cd d d � S # 1 sw Y y xY wrB )r8 r r5 r r; r< r= r& )r' r- r6 r r3 r? s r �_db_update_itemrM R st � � � �
�Y�� �b�*�g�6� ��T� �<� �C�8H�I� I� �g�t� $� �K�K�� � "��:��s���� �� � � �s �A A0�0A9z JSON DB API)�title�/db/{collection}.c �$ � t | |� }d|d�S �NT)�okr? �r@ )r' r6 r? s r �add_itemrT b s � � �
�G� ,�D��� %�%r, z/db/{collection}/{item_id}c � � t | |� S r) �rF �r' r- s r �get_itemrX g � � � �
�G� ,�,r, c � � t | � S r) �rH �r' s r �
list_itemsr] k �
� � �*� %�%r, c �" � t | |� ddiS �NrR T�rK rW s r � delete_itemrb o � � ��J��(� �$�<�r, c �& � t | ||� }d|d�S rQ �rM )r' r- r6 r? s r � update_itemrf t s � � �:�w�� 8�D��� %�%r, �JsonCollectionsDB)�namez db.add_itemz Add DB itemzAdd item to JSON DB)rh rN � descriptionc � � dt | |� d�S rQ rS )r' r6 s r � mcp_add_itemrk } s � �� �Z�� A� B�Br, z db.get_itemz Get DB itemzGet item by idc � � t | |� S r) rV rW s r � mcp_get_itemrm � rY r, z
db.list_itemsz
List DB itemszList items in a collectionc � � t | � S r) r[ r\ s r �mcp_list_itemsro � r^ r, zdb.delete_itemzDelete DB itemzDelete item by idc �" � t | |� ddiS r` ra rW s r �mcp_delete_itemrq � rc r, zdb.update_itemzUpdate DB itemzUpdate an itemc �" � dt | ||� d�S rQ re )r' r- r6 s r �mcp_update_itemrs � s � ���
�G�W� M� N�Nr, z/mcp�/c � � dddd�S )NTrO z/mcp/)rR �rest�mcp_sser* r* r, r �healthrx � s � �� 2�w� G�Gr, �__main__zmain:appz0.0.0.0i@ T)�host�port�reload):r r r9 � threading�typingr r r r r �fastapir r r
�fastmcpr r �join�dirname�__file__r �Lockr8 r2 r r& r+ �intr5 r@ rF rH rK rM �app�postrT r1 rX r] �deleterb �putrf �mcp�toolrk rm ro rq rs �sse_app� mcp_sse_app�mountrx �__name__�uvicorn�runr* r, r �<module>r� sh ��
� � � � 3� 3� 0� 0� �
�'�'�,�,�r�w�w���x�0�)�
<��� �������c�4��S�#�X��/�/�0� � ��S�$�t�C��H�~�.�.�/� �D� � �$�s�D��c�3�h� �$8� 8�9� �s� �t� �� �S�$�t�C��H�~�&�
&�'��58��CF��
�8�C�=�(�4��S��>�2� 2�3��
�S�
�4��S��>�
�d�3��8�n�
��S� �3� �4��S��>� �&�s� &�t�D��c��N�';� &�
�� �c� �d� �
��
�c�
�D��c��N�
�t�TW�Y\�T\�~�
� �M�"�����
� �8<�S� � &�� &�t�C��H�~� &� �&� ��� %�&�-�� -�s� -� '�-� ��� � �&�3� &� �&� ��� (�)��C� �#� � *�� ��� %�&�IM�c�� &�C� &�#� &��S�#�X�� &� '�&� �&�'�����}�M�?T��U�C�S� C�4��S��>� C� V�C� ���}�M�?O��P�-�S� -�3� -� Q�-� ����o�C_��`�&�s� &� a�&� ��� �'7�EX��Y��� �c� � Z�� ��� �'7�EU��V�O�� O�c� O�D��c��N� O� W�O� �k�k�m� � � � �&�+� ����� �H� �H� �z����G�K�K�
���d�C� r,