from sqlalchemy import Column, Integer, String, TIMESTAMP, TEXT, VARCHAR, Numeric, DateTime, Boolean
from datetime import datetime, timezone
from . import Base


class Card(Base):
    __tablename__ = "card_card"

    id = Column(Integer, primary_key=True, index=True)
    date_added = Column(DateTime, default=datetime.utcnow)
    user_id = Column(Integer, nullable=False)

    # This stores the payment profile ID for this specific card from Authorize.Net's CIM.
    auth_net_payment_profile_id = Column(String, nullable=True)

    # Columns to store non-sensitive card info for display purposes
    card_number = Column(String(50), nullable=True)
    last_four_digits = Column(Integer, nullable=False)
    name_on_card = Column(String(500), nullable=True)
    expiration_month = Column(String(20), nullable=False)
    expiration_year = Column(String(20), nullable=False)
    type_of_card = Column(String(50), nullable=True)
    security_number = Column(String(10), nullable=True)
    accepted_or_declined = Column(Integer, nullable=True)
    main_card = Column(Boolean, nullable=True)
    zip_code = Column(String(20), nullable=True)


class Transaction(Base):
    __tablename__ = "transactions"

    id = Column(Integer, primary_key=True, index=True)
    # Recommended change: Use Numeric for precision
    preauthorize_amount = Column(Numeric(10, 2), nullable=True)
    charge_amount = Column(Numeric(10, 2), nullable=True)
    customer_id = Column(Integer)
    transaction_type = Column(Integer)# 0 = charge, 1 = auth, 2 = capture
    status = Column(Integer)
    auth_net_transaction_id = Column(String, unique=True, index=True, nullable=True)
    service_id = Column(Integer, nullable=True)
    delivery_id = Column(Integer, nullable=True)
    auto_id = Column(Integer, nullable=True)
    card_id = Column(Integer, nullable=True)
    payment_gateway = Column(Integer, default=1)
    rejection_reason = Column(String, nullable=True)
    created_at = Column(DateTime, default=datetime.utcnow)