"""add cmdb_object_storage table

Revision ID: 011
Revises: 010
Create Date: 2026-04-14
"""
from alembic import op
import sqlalchemy as sa

revision = '011'
down_revision = '010'
branch_labels = None
depends_on = None


def _table_exists(conn, table_name):
    result = conn.execute(sa.text(
        "SELECT COUNT(*) FROM information_schema.tables "
        "WHERE table_schema = DATABASE() AND table_name = :t"
    ), {"t": table_name})
    return result.scalar() > 0


def upgrade():
    conn = op.get_bind()
    if _table_exists(conn, 'cmdb_object_storage'):
        return
    op.create_table(
        'cmdb_object_storage',
        sa.Column('id', sa.String(36), primary_key=True, server_default=sa.text("(UUID())")),
        sa.Column('provider_account_id', sa.String(36), sa.ForeignKey('cmdb_provider_accounts.id'), nullable=False),
        sa.Column('bucket_name', sa.String(256), nullable=False),
        sa.Column('region', sa.String(64)),
        sa.Column('storage_class', sa.String(64)),
        sa.Column('access_type', sa.String(32)),  # public-read, private, etc.
        sa.Column('size_bytes', sa.BigInteger),
        sa.Column('object_count', sa.Integer),
        sa.Column('status', sa.String(32), nullable=False, server_default='active'),
        sa.Column('owner_raw', sa.String(255)),
        sa.Column('raw_data', sa.JSON),
        sa.Column('last_seen_sync_job_id', sa.String(36)),
        sa.Column('first_seen_at', sa.DateTime(timezone=True), server_default=sa.text('NOW()')),
        sa.Column('last_synced_at', sa.DateTime(timezone=True), server_default=sa.text('NOW()')),
        sa.Column('deleted_at', sa.DateTime(timezone=True)),
        sa.UniqueConstraint('provider_account_id', 'bucket_name', name='uq_oss_account_bucket'),
    )


def downgrade():
    op.drop_table('cmdb_object_storage')