BotLino/Lino

from rasa_core_sdk import Action


class ActionButtonsNotificationTypes(Action):
    """
    Send to the user the operation types over notifications.
    'Adicionar' -> The user can register to an type of notification
    'Retirar' -> The user can remove an type of notification
    'Visualizar' -> The user can see the notification types registered so far
    """
    def name(self):
        return "action_show_notifications_types"

    def run(self, dispatcher, tracker, domain):
        options = [
            ('Cadastrar', 'Cadastrar'),
            ('Remover', 'Remover'),
            ('Visualizar', 'Visualizar')
        ]

        buttons_facebook = None
        buttons_telegram = None

        try:
            message = 'Qual das opções você deseja?'
            buttons_telegram = self.build_buttons(options)
            dispatcher.utter_button_message(message,
                                            buttons_telegram,
                                            button_type="custom")
        except Exception as exception:
            print(exception)
            buttons_telegram = None

        if not buttons_telegram:
            try:
                buttons_facebook = self.build_buttons(options, 'facebook')
                element = {
                    'title': 'Qual das opções você deseja?',
                    'buttons': buttons_facebook
                }
                dispatcher.utter_custom_message(element)
            except Exception as exception:
                print(exception)
                buttons_facebook = None

        if not buttons_telegram and not buttons_facebook:
            dispatcher.utter_message(('Tive alguns problemas pra encontrar'
                                      ' os tipos de notificações... :('))
            dispatcher.utter_message(('Vou tentar arrumar rapidão aqui pra te '
                                      'mandar as que eu tinha antes, blz?!'))
            dispatcher.utter_message(('Você precisa de mais algo?'))

        return []

    def build_buttons(self, button_values, channel='telegram'):
        button = {}
        buttons = []

        if channel is 'telegram':
            for item in button_values:
                button['title'] = item[0]
                button['payload'] = item[1]
                buttons.append(button.copy())

            return buttons

        elif channel is 'facebook':
            for item in button_values:
                button['title'] = item[0]
                button['payload'] = item[1]
                button['type'] = 'postback'
                buttons.append(button.copy())

            return buttons


class ActionButtonsNotification(Action):
    def name(self):
        return "action_buttons_notification"

    def run(self, dispatcher, tracker, domain):
        buttons_facebook = None
        buttons_telegram = None

        try:
            message = 'Qual das opções você deseja?'
            buttons_telegram = self.build_buttons_telegram()
            dispatcher.utter_button_message(message,
                                            buttons_telegram,
                                            button_type="custom")
        except Exception as exception:
            print(exception)
            buttons_telegram = None

        if not buttons_telegram:
            try:
                buttons_facebook = self.build_buttons_facebook()
                elements = self.build_facebook_elements(buttons_facebook)
                dispatcher.utter_custom_message(*elements)
            except Exception as exception:
                print(exception)
                buttons_facebook = None

        if not buttons_telegram and not buttons_facebook:
            dispatcher.utter_message(('Tive alguns problemas aqui em encontrar'
                                     ' os tipos de notificações :(...'))
            dispatcher.utter_message(('Vou tentar arrumar rapidão aqui pra te '
                                     'mandar as que eu tinha antes, beleza?'))
            dispatcher.utter_message(('Você precisa de mais alguma outra coisa'
                                     ' além disso? Só pedir :)'))

        return []

    def build_button_dict(self):
        return [
            ('Alerta da Comunidade',
             ('/notifications{"register_notification":'
              '" alerta da comunidade "}')),
            ('Cardápio do Dia',
             '/notifications{"register_notification": " dia "}'),
            ('Cardápio da Semana',
             '/notifications{"register_notification": " semana "}'),
            ('Café da Manhã',
             '/notifications{"register_notification": " café "}'),
            ('Almoço',
             '/notifications{"register_notification": " almoço "}'),
            ('Jantar',
             '/notifications{"register_notification": " jantar "}')
        ]

    def build_facebook_elements(self, buttons):
        message = 'Qual das opções você deseja?'
        elements = []
        for value in buttons:
            print('A'*100)
            print(value)
            element = self.build_element(value, message)
            elements.append(element)
        return elements

    def build_element(self, list_buttons, message):
        return {
            'title': message,
            'buttons': list_buttons
        }

    def build_buttons_facebook(self):
        values = self.build_button_dict()
        lista = [values[x:x+3]for x in range(0, len(values), 3)]
        print(lista)
        for value in lista:
            for i in range(0, len(value)):
                value[i] = {
                    'type': 'postback',
                    'title': value[i][0],
                    'payload': value[i][1]
                }
        return lista

    def build_buttons_telegram(self):
        values = self.build_button_dict()
        lista = [values[x:x+3]for x in range(0, len(values), 3)]
        for value in lista:
            for i in range(0, len(value)):
                value[i] = {
                    'title': value[i][0],
                    'payload': value[i][1]
                }
        return lista