SmsClient.py

import logging
from typing import override

from fastapi import HTTPException
from twilio.rest import Client as TwilioClient

from Client import Client
from config import Settings, settings
from constants import IGNORE_MESSAGE, PROD_ENV
from database import Database

logger = logging.getLogger(__name__)


class SmsClient(Client):
    def __init__(self, settings: Settings, db: Database | None = None):
        super().__init__(settings, db)
        self.twilio_client = TwilioClient(
            settings.twilio_account_sid, settings.twilio_auth_token
        )
        self.reroute_next_msg_users: set[str] = set()

    @override
    def send_message(self, to: str, text: str) -> None:
        self.twilio_client.messages.create(
            to=to, from_=settings.twilio_phone_number, body=text
        )

    @override
    def receive_message(self, from_: str, text: str) -> None:
        print(from_, text)

    @override
    def handle_beta_message(self, from_: str, text: str) -> str:
        if text == self.settings.beta_code:
            logger.info("handle_beta_reroute_next_msg")
            self.reroute_next_msg_users.add(from_)
            return IGNORE_MESSAGE
        elif self.settings.beta_code in text:
            prompt_text = " ".join(text.split(" ")[1:])
            logger.info("handle_beta_message %s", prompt_text)
            self.reroute_to_beta(from_)
            return prompt_text
        elif from_ in self.reroute_next_msg_users:
            logger.info("handle_beta_image")
            self.reroute_next_msg_users.remove(from_)
            self.reroute_to_beta(from_)
            return text
        return text

    def reroute_to_beta(self, from_: str):
        if self.settings.environment == PROD_ENV:
            logger.info("text routed to beta env")
            raise HTTPException(status_code=501, detail="beta code detected.")
        if from_ not in self.settings.beta_allowlist.split(","):
            logger.info(
                "%s attempted to use BETA environment but was not allowlisted.", from_
            )
            raise HTTPException(status_code=401, detail="Number not allowlisted.")