import motor.motor_asyncio
from .config import config


client = motor.motor_asyncio.AsyncIOMotorClient(
    config.MONGO_URL,
    maxPoolSize=100,
    minPoolSize=5,
    maxIdleTimeMS=60_000,
    waitQueueTimeoutMS=10_000,
    serverSelectionTimeoutMS=5_000,
)
database = client[config.KITTYKAT_AGENT_CLUSTER]

kittykat_agent_brands_collection = database["brands"]
kittykat_agent_users_collection = database["users"]
kittykat_agent_roles_collection = database["roles"]
kittykat_agent_gallery_collection = database["gallery"]
kittykat_agent_queue_collection = database["queue"]
kittykat_agent_product_categories_collection = database["product_categories"]
kittykat_agent_notification_queue_collection = database["notification_queue"]
kittykat_agent_notification_collection = database["notification"]
ask_kittykat_tasklists_collection = database["ask_kittykat_tasklists"]

models_collection = database["models"]
config_collection = database["config"]
credit_transactions_collection = database["credit_transactions"]
kittykat_agent_audit_logs_collection = database["audit_logs"]
kittykat_agent_teams_collection = database["teams"]
kittykat_agent_product_sku_collection = database["product_sku"]
kittykat_agent_campaign_concepts_collection = database["campaign_concepts"]
kittykat_agent_brand_prompts_collection = database["brand_prompts"]
kittykat_agent_shot_types_collection = database["shot_types"]
kittykat_agent_generation_parameters_collection = database["a2i_generation_parameters"]
a2i_generations_collection = database["generations"]