"""create fruits table
Revision ID: 0002
Revises: 0001
Create Date: 2024-01-20 10:00:00.000000
"""
from alembic import op
import sqlalchemy as sa

revision = '0002'
down_revision = '0001'
branch_labels = None
depends_on = None

def upgrade():
    op.create_table(
        'fruits',
        sa.Column('id', sa.String(36), primary_key=True),
        sa.Column('name', sa.String(), nullable=False),
        sa.Column('color', sa.String(), nullable=False),
        sa.Column('created_at', sa.DateTime(), server_default=sa.text('CURRENT_TIMESTAMP')),
        sa.Column('updated_at', sa.DateTime(), server_default=sa.text('CURRENT_TIMESTAMP'))
    )
    op.create_index(op.f('ix_fruits_name'), 'fruits', ['name'], unique=False)

    # Create trigger for updated_at
    op.execute('''
        CREATE TRIGGER trig_fruits_updated_at 
        AFTER UPDATE ON fruits
        FOR EACH ROW
        BEGIN
            UPDATE fruits SET updated_at = CURRENT_TIMESTAMP
            WHERE id = NEW.id;
        END;
    ''')

def downgrade():
    op.execute('DROP TRIGGER IF EXISTS trig_fruits_updated_at')
    op.drop_index(op.f('ix_fruits_name'), table_name='fruits')
    op.drop_table('fruits')