from contacts import contact, common
from messenger.messages import *
import os
from contacts.contact_menu import *


class Friend(contact.Contact):
    """
    Friend in list of friends.
    """

    def __init__(self, profile_manager, message_getter, number, name, status_message, widget, tox_id):
        super().__init__(profile_manager, message_getter, number, name, status_message, widget, tox_id)
        self._receipts = 0
        self._typing_notification_handler = common.FriendTypingNotificationHandler(number)

    # -----------------------------------------------------------------------------------------------------------------
    # File transfers support
    # -----------------------------------------------------------------------------------------------------------------

    def update_transfer_data(self, file_number, inline):  # TODO: rewrite
        """
        Update status of active transfer and load inline if needed
        """
        try:
            tr = list(filter(lambda x: x.get_type() == MESSAGE_TYPE['FILE_TRANSFER'] and x.is_active(file_number),
                             self._corr))[0]
            i = self._corr.index(tr)
            if inline:  # inline was loaded
                self._corr.insert(i, inline)
            return i - len(self._corr)
        except:
            pass

    def get_unsent_files(self):
        messages = filter(lambda x: type(x) is UnsentFile, self._corr)
        return messages

    def clear_unsent_files(self):
        self._corr = list(filter(lambda x: type(x) is not UnsentFile, self._corr))

    def remove_invalid_unsent_files(self):
        def is_valid(message):
            if type(message) is not UnsentFile:
                return True
            if message.get_data()[1] is not None:
                return True
            return os.path.exists(message.get_data()[0])

        self._corr = list(filter(is_valid, self._corr))

    def delete_one_unsent_file(self, message_id):
        self._corr = list(filter(lambda m: not (type(m) is UnsentFile and m.message_id == message_id), self._corr))

    # -----------------------------------------------------------------------------------------------------------------
    # History support
    # -----------------------------------------------------------------------------------------------------------------

    def get_receipts(self):
        return self._receipts

    receipts = property(get_receipts)  # read receipts

    def inc_receipts(self):
        self._receipts += 1

    def dec_receipt(self):
        if self._receipts:
            self._receipts -= 1
            self.mark_as_sent()

    # -----------------------------------------------------------------------------------------------------------------
    # Full status
    # -----------------------------------------------------------------------------------------------------------------

    def get_full_status(self):
        return self._status_message

    # -----------------------------------------------------------------------------------------------------------------
    # Typing notifications
    # -----------------------------------------------------------------------------------------------------------------

    def get_typing_notification_handler(self):
        return self._typing_notification_handler

    # -----------------------------------------------------------------------------------------------------------------
    # Context menu support
    # -----------------------------------------------------------------------------------------------------------------

    def get_context_menu_generator(self):
        return FriendMenuGenerator(self)