"""Capability Model Phase 2: organization (company/account) records, distinct from customer.

Revision ID: 151
Revises: 150
"""
from uuid import uuid4

import sqlalchemy as sa
from alembic import op
from sqlalchemy.dialects.postgresql import JSONB, UUID, insert

revision = "151"
down_revision = "150"
branch_labels = None
depends_on = None

feature_packs = sa.table("feature_packs",
    sa.column("id", UUID(as_uuid=True)), sa.column("slug", sa.String),
    sa.column("name", sa.String), sa.column("description", sa.Text),
    sa.column("version", sa.String),
    sa.column("tags", JSONB),
    sa.column("status", sa.String),
    sa.column("composition", JSONB),
    sa.column("input_schema", JSONB),
    sa.column("access_policy", JSONB),
)


def organization_composition() -> dict:
    from src.domain.feature_packs.crm_authorization import staff_policy
    from src.domain.feature_packs.normalize import stamp_composition

    return stamp_composition("organization-records", {
        "function_slugs": [], "connector_actions": [], "requires": [],
        "entities": [{
            "entity_key": "organization", "name": "Organization", "plural_name": "Organizations",
            "rls_enabled": False, "owner_managed": False,
            "schema": {
                "fields": [
                    {"key": "name", "name": "Name", "type": "text", "required": True, "indexed": True},
                    {"key": "website", "name": "Website", "type": "text"},
                    {"key": "industry", "name": "Industry", "type": "text"},
                    {"key": "phone", "name": "Phone", "type": "text"},
                    {"key": "notes", "name": "Notes", "type": "text"},
                    {"key": "status", "name": "Status", "type": "select", "required": True,
                     "indexed": True, "options": ["active", "inactive"]},
                ],
                "access_policy": staff_policy(),
            },
        }],
    }, implementation_version="1.0.0")


def upgrade() -> None:
    op.get_bind().execute(insert(feature_packs).values(
        id=uuid4(), slug="organization-records", name="Organization Records", version="1.0.0",
        description=(
            "Staff-managed company/account records, distinct from crm.customer contacts."
        ),
        tags=["crm", "organizations"], status="active", composition=organization_composition(),
        input_schema={"type": "object", "properties": {}}, access_policy={"minimum_plan": "free"},
    ).on_conflict_do_nothing(index_elements=["slug"]))


def downgrade() -> None:
    pass