from datetime import datetime, timezone
from app.extensions import db
from sqlalchemy.sql import func

# Inside Stripe_Product ===== : {'product_id': 'prod_SvB4RlFx6qXK6v', 'price_id': 'price_1RzKi4LjZLojw6IZA246JDAa'}

class SubscriptionOptions(db.Model):
    __tablename__ = 'subscription_options'

    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    uid = db.Column(db.String(150), nullable=True)
    display_name = db.Column(db.String(25), nullable=False)
    option_name = db.Column(db.String(100), nullable=False)
    monthly = db.Column(db.Integer, nullable=True, default=0)
    status = db.Column(db.Integer, nullable=True, default=1)
    added = db.Column(db.DateTime(timezone=False), server_default=func.now())
    updated = db.Column(db.DateTime(timezone=False), server_default=func.now(), onupdate=func.now())
    stripe_product_id = db.Column(db.String(100), nullable=True)
    stripe_price_id = db.Column(db.String(100), nullable=True)
    country_code = db.Column(db.String(3), nullable=False)


    @classmethod
    def get_subscription_options(cls, status):
        sub_options = cls.query.filter_by(status=status).all()
        if not sub_options:
            return None
        return sub_options

    @classmethod
    def get_subscription_options_by_option_name(cls, option_name):
        sub_options = cls.query.filter_by(option_name=option_name).all()
        if not sub_options:
            return None
        return sub_options


    @classmethod
    def set_stripe_product_id(cls,package_uid, stripe_product_id):
        # Retrieve
        sub_option  = cls.query.filter_by(uid=str(package_uid)).first()

        if not sub_option:
            raise ValueError(f"Reset with ID {package_uid} does not exist.")

        # Update stripe_product_id
        sub_option.stripe_product_id = stripe_product_id

    @classmethod
    def set_stripe_price_id(cls, package_uid, stripe_price_id):
        # Retrieve
        sub_option = cls.query.filter_by(uid=str(package_uid)).first()

        if not sub_option:
            raise ValueError(f"Reset with ID {package_uid} does not exist.")

        # Update stripe_price_id
        sub_option.stripe_price_id = stripe_price_id


    def to_dict(self):
        return {
            "id": self.id,
            "uid": self.uid,
            "display_name": self.display_name,
            "status": self.status,
            "monthly": self.monthly,
            "country_code": self.country_code,
            "stripe_product_id": self.stripe_product_id,
            "stripe_price_id": self.stripe_price_id,
            "added": self.added.isoformat() if self.added else None,
            "updated": self.updated.isoformat() if self.updated else None
        }

    def __repr__(self):
        return f'<SubscriptionOptions {self.id} - {self.amount}>'