from tortoise import BaseDBAsyncClient


async def upgrade(db: BaseDBAsyncClient) -> str:
    return """
        CREATE TABLE IF NOT EXISTS "suno_tasks" (
    "created_at" TIMESTAMPTZ NOT NULL  DEFAULT CURRENT_TIMESTAMP,
    "updated_at" TIMESTAMPTZ NOT NULL  DEFAULT CURRENT_TIMESTAMP,
    "id" UUID NOT NULL  PRIMARY KEY,
    "task_id" VARCHAR(255) NOT NULL UNIQUE,
    "status" VARCHAR(50) NOT NULL  DEFAULT 'PENDING',
    "audio_url" VARCHAR(500),
    "image_url" VARCHAR(500),
    "audio_url2" VARCHAR(500),
    "image_url2" VARCHAR(500),
    "chatbot_id" INT NOT NULL REFERENCES "chatbots" ("id") ON DELETE CASCADE,
    "user_id" UUID NOT NULL REFERENCES "users" ("id") ON DELETE CASCADE
);"""


async def downgrade(db: BaseDBAsyncClient) -> str:
    return """
        DROP TABLE IF EXISTS "suno_tasks";"""