robosats/api/notifications.py

from secrets import token_urlsafe

from decouple import config
from api.models import (
    Order,
    Notification,
)
from api.utils import get_session


class Notifications:
    """Simple telegram messages using TG's API"""

    session = get_session()
    site = config("HOST_NAME")

    def get_context(user):
        """returns context needed to enable TG notifications"""
        context = {}
        if user.robot.telegram_enabled:
            context["tg_enabled"] = True
        else:
            context["tg_enabled"] = False

        if user.robot.telegram_token is None:
            user.robot.telegram_token = token_urlsafe()[:15]
            user.robot.save(update_fields=["telegram_token"])

        context["tg_token"] = user.robot.telegram_token
        context["tg_bot_name"] = config("TELEGRAM_BOT_NAME")

        return context

    def send_message(self, order, robot, title, description=""):
        """Save a message for a user and sends it to Telegram"""
        self.save_message(order, robot, title, description)
        if robot.telegram_enabled:
            self.send_telegram_message(robot.telegram_chat_id, title, description)

    def save_message(self, order, robot, title, description):
        """Save a message for a user"""
        Notification.objects.create(
            title=title, description=description, robot=robot, order=order
        )

    def send_telegram_message(self, chat_id, title, description):
        """sends a message to a user with telegram notifications enabled"""

        bot_token = config("TELEGRAM_TOKEN")
        text = f"{title} {description}"
        message_url = f"https://api.telegram.org/bot{bot_token}/sendMessage?chat_id={chat_id}&text={text}"
        # if it fails, it should keep trying
        while True:
            try:
                self.session.get(message_url).json()
                return
            except Exception:
                pass

    def welcome(self, user):
        """User enabled Telegram Notifications"""
        lang = user.robot.telegram_lang_code

        if lang == "es":
            title = f"🔔 Hola {user.username}, te enviaré notificaciones sobre tus órdenes en RoboSats."
        else:
            title = f"🔔 Hey {user.username}, I will send you notifications about your RoboSats orders."
        self.send_telegram_message(user.robot.telegram_chat_id, title)
        user.robot.telegram_welcomed = True
        user.robot.save(update_fields=["telegram_welcomed"])
        return

    def order_taken_confirmed(self, order):
        lang = order.maker.robot.telegram_lang_code
        if lang == "es":
            title = f"✅ Hey {order.maker.username} ¡Tu orden con ID {order.id} ha sido tomada por {order.taker.username}!🥳"
            description = f"Visita http://{self.site}/order/{order.id} para continuar."
        else:
            title = f"✅ Hey {order.maker.username}, your order was taken by {order.taker.username}!🥳"
            description = (
                f"Visit http://{self.site}/order/{order.id} to proceed with the trade."
            )
        self.send_message(order, order.maker.robot, title, description)

        lang = order.taker.robot.telegram_lang_code
        if lang == "es":
            title = f"✅ Hey {order.taker.username}, acabas de tomar la orden con ID {order.id}."
        else:
            title = f"✅ Hey {order.taker.username}, you just took the order with ID {order.id}."
        self.send_message(order, order.taker.robot, title)

        return

    def fiat_exchange_starts(self, order):
        for user in [order.maker, order.taker]:
            lang = user.robot.telegram_lang_code
            if lang == "es":
                title = f"✅ Hey {user.username}, el depósito de garantía y el recibo del comprador han sido recibidos. Es hora de enviar el dinero fiat."
                description = f"Visita http://{self.site}/order/{order.id} para hablar con tu contraparte."
            else:
                title = f"✅ Hey {user.username}, the escrow and invoice have been submitted. The fiat exchange starts now via the platform chat."
                description = f"Visit http://{self.site}/order/{order.id} to talk with your counterpart."
            self.send_message(order, user.robot, title, description)
        return

    def order_expired_untaken(self, order):
        lang = order.maker.robot.telegram_lang_code
        if lang == "es":
            title = f"😪 Hey {order.maker.username}, tu orden con ID {order.id} ha expirado sin ser tomada por ningún robot."
            description = f"Visita http://{self.site}/order/{order.id} para renovarla."
        else:
            title = f"😪 Hey {order.maker.username}, your order with ID {order.id} has expired without a taker."
            description = f"Visit http://{self.site}/order/{order.id} to renew it."
        self.send_message(order, order.maker.robot, title, description)
        return

    def trade_successful(self, order):
        for user in [order.maker, order.taker]:
            lang = user.robot.telegram_lang_code
            if lang == "es":
                title = f"🥳 ¡Tu orden con ID {order.id} ha finalizado exitosamente!"
                description = (
                    "⚡ Únete a nosotros en @robosats_es y ayúdanos a mejorar."
                )
            else:
                title = f"🥳 Your order with ID {order.id} has finished successfully!"
                description = "⚡ Join us @robosats and help us improve."
            self.send_message(order, user.robot, title, description)
        return

    def public_order_cancelled(self, order):
        lang = order.maker.robot.telegram_lang_code
        if lang == "es":
            title = f"❌ Hey {order.maker.username}, has cancelado tu orden pública con ID {order.id}."
        else:
            title = f"❌ Hey {order.maker.username}, you have cancelled your public order with ID {order.id}."
        self.send_message(order, order.maker.robot, title)
        return

    def collaborative_cancelled(self, order):
        for user in [order.maker, order.taker]:
            lang = user.robot.telegram_lang_code
            if lang == "es":
                title = f"❌ Hey {user.username}, tu orden con ID {str(order.id)} fue cancelada colaborativamente."
            else:
                title = f"❌ Hey {user.username}, your order with ID {str(order.id)} has been collaboratively cancelled."
            self.send_message(order, user.robot, title)
        return

    def dispute_opened(self, order):
        for user in [order.maker, order.taker]:
            lang = user.robot.telegram_lang_code
            if lang == "es":
                title = f"⚖️ Hey {user.username}, la orden con ID {str(order.id)} ha entrado en disputa."
            else:
                title = f"⚖️ Hey {user.username}, a dispute has been opened on your order with ID {str(order.id)}."
            self.send_message(order, user.robot, title)

        admin_chat_id = config("TELEGRAM_COORDINATOR_CHAT_ID")

        if len(admin_chat_id) == 0:
            return

        coordinator_text = (
            f"There is a new dispute opened for the order with ID {str(order.id)}."
        )
        coordinator_description = f"Visit http://{self.site}/coordinator/api/order/{str(order.id)}/change to proceed."
        self.send_telegram_message(
            admin_chat_id, coordinator_text, coordinator_description
        )

        return

    def order_published(self, order):
        lang = order.maker.robot.telegram_lang_code
        # In weird cases the order cannot be found (e.g. it is cancelled)
        queryset = Order.objects.filter(maker=order.maker)
        if len(queryset) == 0:
            return
        order = queryset.last()
        if lang == "es":
            title = f"✅ Hey {order.maker.username}, tu orden con ID {str(order.id)} es pública en el libro de ordenes."
        else:
            title = f"✅ Hey {order.maker.username}, your order with ID {str(order.id)} is public in the order book."
        self.send_message(order, order.maker.robot, title)
        return

    def new_chat_message(self, order, chat_message):
        """
        Sends a TG notification for a new in-app chat message if
        the last chat was at least CHAT_NOTIFICATION_TIMEGAP minutes ago.
        """
        from datetime import timedelta

        from django.utils import timezone

        from chat.models import Message

        TIMEGAP = config("CHAT_NOTIFICATION_TIMEGAP", cast=int, default=5)
        if chat_message.index > 1:
            previous_message = Message.objects.get(
                chatroom=chat_message.chatroom, index=(chat_message.index - 1)
            )
            notification_reason = f"(You receive this notification only because more than {TIMEGAP} minutes have passed since the last in-chat message)"
            if previous_message.created_at > timezone.now() - timedelta(
                minutes=TIMEGAP
            ):
                return
        else:
            notification_reason = f"(You receive this notification because this was the first in-chat message. You will only be notified again if there is a gap bigger than {TIMEGAP} minutes between messages)"

        user = chat_message.receiver
        title = f"💬 Hey {user.username}, a new chat message in-app was sent to you by {chat_message.sender.username} for order ID {str(order.id)}."
        self.send_message(order, user.robot, title, notification_reason)

        return

    def coordinator_cancelled(self, order):
        title = f"🛠️ Your order with ID {order.id} has been cancelled by the coordinator {config('COORDINATOR_ALIAS', cast=str, default='NoAlias')} for the upcoming maintenance stop."
        self.send_message(order, order.maker.robot, title)
        return