import json
from app.agents.tools.search_room_tool import search_room


async def add_to_cart(roomList, checkin, checkout, id):
    
    if not id:
        res = search_room(adultCount=2,childCount=0,check_in=checkin,check_out=checkout,sortValue='default',keyValue='default')
        async for token in res:
            yield token
        return
    room = await find_room_by_type_id(rooms_list=roomList, target_type_id=id)
    
    yield json.dumps({"fdGN32r": room})
    # yield "Room added to cart"
    return

async def find_room_by_type_id(rooms_list, target_type_id):
    for room in rooms_list:
        if room['id'] == target_type_id:
            return room
    return None