from tortoise import BaseDBAsyncClient


async def upgrade(db: BaseDBAsyncClient) -> str:
    return """
        DROP INDEX "idx_channel_server__d940e4";
        DROP INDEX "idx_ban_user_id_37a9cf";
        ALTER TABLE "server" ADD "name" TEXT NOT NULL;
        CREATE  INDEX "idx_ban_user_id_37a9cf" ON "ban" ("user_id", "expires_at");;
        CREATE  INDEX "idx_channel_server__d940e4" ON "channel" ("server_id", "is_enabled");;"""


async def downgrade(db: BaseDBAsyncClient) -> str:
    return """
        DROP INDEX "idx_channel_server__d940e4";
        DROP INDEX "idx_ban_user_id_37a9cf";
        ALTER TABLE "server" DROP COLUMN "name";
        CREATE  INDEX "idx_ban_user_id_37a9cf" ON "ban" ("user_id", "expires_at");;
        CREATE  INDEX "idx_channel_server__d940e4" ON "channel" ("server_id", "is_enabled");;"""