"""add xinyu lubricant company

Revision ID: b2c3d4e5f6a7
Revises: a1b2c3d4e5f6
Create Date: 2026-03-19
"""
from alembic import op

revision = "b2c3d4e5f6a7"
down_revision = "a1b2c3d4e5f6"
branch_labels = None
depends_on = None

XINYU_COMPANY_ID = "aaaaaaaa-bbbb-cccc-dddd-eeeeeeee0002"


def upgrade() -> None:
    # 1. 插入第二个公司：新宇润滑油
    op.execute(f"""
        INSERT INTO sys_companies (id, name, code, is_active)
        VALUES ('{XINYU_COMPANY_ID}', '新宇润滑油', 'XY-LUB', true)
        ON CONFLICT (id) DO NOTHING
    """)

    # 2. 将所有现有用户关联到新宇润滑油（非默认）
    op.execute(f"""
        INSERT INTO sys_user_companies (id, user_id, company_id, is_default)
        SELECT gen_random_uuid(), id, '{XINYU_COMPANY_ID}'::uuid, false
        FROM sys_users
        WHERE id NOT IN (
            SELECT user_id FROM sys_user_companies
            WHERE company_id = '{XINYU_COMPANY_ID}'::uuid
        )
    """)


def downgrade() -> None:
    op.execute(f"""
        DELETE FROM sys_user_companies WHERE company_id = '{XINYU_COMPANY_ID}'::uuid
    """)
    op.execute(f"""
        DELETE FROM sys_companies WHERE id = '{XINYU_COMPANY_ID}'::uuid
    """)