from Christmas.SearchEngine import Search



class Snowball:
    
    async def register_guild(guild_id: int, guild_name: str, region: str):
        search = Search()
        index = await search.connect()
        await index.add_documents([{"guild_id": guild_id, "guild_name": guild_name, "region": region}])
    
    async def query_guild(query: str):
        search = Search()
        index = await search.connect()
        result = await index.search(query)
        return result