"""
Overlay registry — company alias lookup for entity resolution.
"""

from sqlalchemy import Column, String, Float, Boolean, Index
from sqlalchemy.dialects.postgresql import UUID, TIMESTAMP
from datetime import datetime, timezone
import uuid

from app.core.database import Base


class CompanyAlias(Base):
    __tablename__ = "company_aliases"

    id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
    symbol = Column(String(10), nullable=False, index=True)
    alias_type = Column(String(20), nullable=False)  # canonical / short / alias / wiki
    alias_value = Column(String(255), nullable=False)
    confidence = Column(Float, default=1.0)
    active = Column(Boolean, default=True)
    created_at = Column(
        TIMESTAMP(timezone=True), default=lambda: datetime.now(timezone.utc)
    )

    __table_args__ = (
        Index("idx_company_aliases_symbol", "symbol"),
        Index("idx_company_aliases_value", "alias_value"),
    )