robosats/api/messages.py

from decouple import config
from secrets import token_urlsafe
from api.models import Order
from api.utils import get_session
import time

class Telegram():
    ''' Simple telegram messages by requesting to API'''

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

    def get_context(user):
        """returns context needed to enable TG notifications"""
        context = {}
        if user.profile.telegram_enabled :
            context['tg_enabled'] = True
        else:
            context['tg_enabled'] = False
        
        if user.profile.telegram_token == None:
            user.profile.telegram_token = token_urlsafe(15)
            user.profile.save()

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

        return context

    def send_message(self, user, text):
        """ sends a message to a user with telegram notifications enabled"""

        bot_token=config('TELEGRAM_TOKEN')

        chat_id = user.profile.telegram_chat_id
        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:
                pass
    
    def welcome(self, user):
        ''' User enabled Telegram Notifications'''
        lang = user.profile.telegram_lang_code

        if lang == 'es':
            text = f'Hola {user.username}, te enviaré notificaciones sobre tus órdenes en RoboSats.'
        else:
            text = f"Hey {user.username}, I will send you notifications about your RoboSats orders."
        self.send_message(user, text)
        user.profile.telegram_welcomed = True
        user.profile.save()
        return

    # def welcome(self, user):
    #     lang = user.profile.telegram_lang_code

    #     # In weird cases the order cannot be found (e.g. it is cancelled)
    #     queryset = Order.objects.filter(maker=user)
    #     order = queryset.last()

    #     print(str(order.id))
    #     if lang == 'es':
    #         text = f'Hola {user.username}, te enviaré un mensaje cuando tu orden con ID {str(order.id)} haya sido tomada.'
    #     else:
    #         text = f"Hey {user.username}, I will send you a message when someone takes your order with ID {str(order.id)}."
    #     self.send_message(user, text)
    #     user.profile.telegram_welcomed = True
    #     user.profile.save()
    #     return

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

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

        return

    def fiat_exchange_starts(self, order):
        for user in [order.maker, order.taker]:
            if user.profile.telegram_enabled:
                lang = user.profile.telegram_lang_code
                if lang == 'es':
                    text = 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. Visita http://{self.site}/order/{order.id} para hablar con tu contraparte.'
                else:
                    text = f'Hey {user.username}, the escrow and invoice have been submitted. The fiat exchange starts now via the platform chat. Visit http://{self.site}/order/{order.id} to talk with your counterpart.'
                self.send_message(user, text)
        return

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

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

    def public_order_cancelled(self, order):
        if order.maker.profile.telegram_enabled:
            lang = order.maker.profile.telegram_lang_code
            if lang == 'es':
                text = f'Hey {order.maker.username}, has cancelado tu orden pública con ID {order.id}.'
            else:
                text = f'Hey {order.maker.username}, you have cancelled your public order with ID {order.id}.'
            self.send_message(order.maker, text)
        return
        
    def collaborative_cancelled(self, order):
        for user in [order.maker, order.taker]:
            if user.profile.telegram_enabled:
                lang = user.profile.telegram_lang_code
                if lang == 'es':
                    text = f'Hey {user.username}, tu orden con ID {str(order.id)} fue cancelada colaborativamente.'
                else:
                    text = f'Hey {user.username}, your order with ID {str(order.id)} has been collaboratively cancelled.'
                self.send_message(user, text)
        return
    
    def dispute_opened(self, order):
        for user in [order.maker, order.taker]:
            if user.profile.telegram_enabled:
                lang = user.profile.telegram_lang_code
                if lang == 'es':
                    text = f'Hey {user.username}, la orden con ID {str(order.id)} ha entrado en disputa.'
                else:
                    text = f'Hey {user.username}, a dispute has been opened on your order with ID {str(order.id)}.'
                self.send_message(user, text)
        return

    def order_published(self, order):
        if order.maker.profile.telegram_enabled:
            lang = order.maker.profile.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':
                text = f'Hey {order.maker.username}, tu orden con ID {str(order.id)} es pública en el libro de ordenes.'
            else:
                text = f"Hey {order.maker.username}, your order with ID {str(order.id)} is public in the order book."
            self.send_message(order.maker, text)
        return