from plugins.cp.main import send_main_menu
from pyrogram import Client, filters
from pyrogram.types import CallbackQuery
from utils import filters as myfilters
from utils.client import MyClient
from utils.db import Depends, User, get_user_query, inject
from utils.settings import upset_setting


@Client.on_callback_query(filters.regex("cp_maintenance_mode") & myfilters.is_admin)
@inject
async def maintenance_mode_change(
    client: MyClient, query: CallbackQuery, user: User = Depends(get_user_query)
) -> None:
    mode = await upset_setting("MAINTENANCE_MODE")
    mode = await upset_setting("MAINTENANCE_MODE", not mode)
    await query.answer("Maintenance Mode Changed", show_alert=True)
    await send_main_menu(client, query, user)