quivr/backend/core/models/brains_subscription_invitations.py

from typing import Optional
from uuid import UUID

import resend
from logger import get_logger
from models.settings import BrainSettings, CommonsDep, common_dependencies
from pydantic import BaseModel

logger = get_logger(__name__)


class BrainSubscription(BaseModel):
    brain_id: Optional[UUID] = None
    inviter_email: Optional[str]
    email: Optional[str]
    rights: Optional[str]

    class Config:
        arbitrary_types_allowed = True

    @property
    def commons(self) -> CommonsDep:
        return common_dependencies()

    def create_subscription_invitation(self):
        logger.info("Creating subscription invitation")
        response = (
            self.commons["supabase"]
            .table("brain_subscription_invitations")
            .insert(
                {
                    "brain_id": str(self.brain_id),
                    "email": self.email,
                    "rights": self.rights,
                }
            )
            .execute()
        )
        return response.data

    def update_subscription_invitation(self):
        logger.info("Updating subscription invitation")
        response = (
            self.commons["supabase"]
            .table("brain_subscription_invitations")
            .update({"rights": self.rights})
            .eq("brain_id", str(self.brain_id))
            .eq("email", self.email)
            .execute()
        )
        return response.data

    def create_or_update_subscription_invitation(self):
        response = (
            self.commons["supabase"]
            .table("brain_subscription_invitations")
            .select("*")
            .eq("brain_id", str(self.brain_id))
            .eq("email", self.email)
            .execute()
        )

        if response.data:
            response = self.update_subscription_invitation()
        else:
            response = self.create_subscription_invitation()

        return response

    def get_brain_url(self) -> str:
        """Generates the brain URL based on the brain_id."""
        base_url = "https://www.quivr.app/chat"
        return f"{base_url}?brain_subscription_invitation={self.brain_id}"

    def resend_invitation_email(self):
        brains_settings = BrainSettings()  # pyright: ignore reportPrivateUsage=none
        resend.api_key = brains_settings.resend_api_key

        brain_url = self.get_brain_url()

        html_body = f"""
        <p>This brain has been shared with you by {self.inviter_email}.</p>
        <p><a href='{brain_url}'>Click here</a> to access your brain.</p>
        """

        try:
            r = resend.Emails.send(
                {
                    "from": brains_settings.resend_email_address,
                    "to": self.email,
                    "subject": "Quivr - Brain Shared With You",
                    "html": html_body,
                }
            )
            print("Resend response", r)
        except Exception as e:
            logger.error(f"Error sending email: {e}")
            return

        return r