"""Remove workspace system

Revision ID: 038
Revises: 037
Create Date: 2026-02-15

Drops workspace tables and columns. Keeps tree_categories.color.
"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects.postgresql import UUID


# revision identifiers, used by Alembic.
revision: str = '038'
down_revision: Union[str, None] = '037'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    # 1. Drop workspace_id FK and column from trees
    op.drop_index('ix_trees_workspace_id', 'trees')
    op.drop_constraint('fk_trees_workspace_id', 'trees', type_='foreignkey')
    op.drop_column('trees', 'workspace_id')

    # 2. Drop workspace_id FK and column from tree_categories
    op.drop_index('ix_tree_categories_workspace_id', 'tree_categories')
    op.drop_constraint('fk_tree_categories_workspace_id', 'tree_categories', type_='foreignkey')
    op.drop_column('tree_categories', 'workspace_id')

    # 3. Drop workspaces table
    op.drop_index('ix_workspaces_account_id', 'workspaces')
    op.drop_index('ix_workspaces_slug', 'workspaces')
    op.drop_table('workspaces')

    # DO NOT drop tree_categories.color — we still use it


def downgrade() -> None:
    # Recreate workspaces table
    op.create_table(
        'workspaces',
        sa.Column('id', UUID(as_uuid=True), primary_key=True, server_default=sa.text('gen_random_uuid()')),
        sa.Column('name', sa.String(100), nullable=False),
        sa.Column('slug', sa.String(100), nullable=False),
        sa.Column('description', sa.Text(), nullable=True),
        sa.Column('icon', sa.String(10), nullable=True),
        sa.Column('accent_color', sa.String(7), nullable=True),
        sa.Column('account_id', UUID(as_uuid=True), sa.ForeignKey('accounts.id', ondelete='CASCADE'), nullable=False),
        sa.Column('is_default', sa.Boolean(), nullable=False, server_default='false'),
        sa.Column('sort_order', sa.Integer(), nullable=False, server_default='0'),
        sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('NOW()')),
        sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.text('NOW()')),
        sa.UniqueConstraint('slug', 'account_id', name='uq_workspaces_slug_account'),
    )
    op.create_index('ix_workspaces_slug', 'workspaces', ['slug'])
    op.create_index('ix_workspaces_account_id', 'workspaces', ['account_id'])

    # Re-add workspace_id columns
    op.add_column('trees', sa.Column('workspace_id', UUID(as_uuid=True), nullable=True))
    op.create_foreign_key('fk_trees_workspace_id', 'trees', 'workspaces', ['workspace_id'], ['id'], ondelete='SET NULL')
    op.create_index('ix_trees_workspace_id', 'trees', ['workspace_id'])

    op.add_column('tree_categories', sa.Column('workspace_id', UUID(as_uuid=True), nullable=True))
    op.create_foreign_key('fk_tree_categories_workspace_id', 'tree_categories', 'workspaces', ['workspace_id'], ['id'], ondelete='SET NULL')
    op.create_index('ix_tree_categories_workspace_id', 'tree_categories', ['workspace_id'])
