from PySide import QtCore from notifications import * from settings import Settings from profile import Profile from toxcore_enums_and_consts import * from tox import bin_to_string class InvokeEvent(QtCore.QEvent): EVENT_TYPE = QtCore.QEvent.Type(QtCore.QEvent.registerEventType()) def __init__(self, fn, *args, **kwargs): QtCore.QEvent.__init__(self, InvokeEvent.EVENT_TYPE) self.fn = fn self.args = args self.kwargs = kwargs class Invoker(QtCore.QObject): def event(self, event): event.fn(*event.args, **event.kwargs) return True _invoker = Invoker() def invoke_in_main_thread(fn, *args, **kwargs): QtCore.QCoreApplication.postEvent(_invoker, InvokeEvent(fn, *args, **kwargs)) # ----------------------------------------------------------------------------------------------------------------- # Callbacks - current user # ----------------------------------------------------------------------------------------------------------------- def self_connection_status(tox_link): """ :param tox_link: tox instance :return: function for tox.callback_self_connection_status """ def wrapped(tox, connection, user_data): print 'Connection status: ', str(connection) profile = Profile.get_instance() if profile.status is None: status = tox_link.self_get_status() invoke_in_main_thread(profile.set_status, status) elif connection == TOX_CONNECTION['NONE']: invoke_in_main_thread(profile.set_status, None) return wrapped # ----------------------------------------------------------------------------------------------------------------- # Callbacks - friends # ----------------------------------------------------------------------------------------------------------------- def friend_status(tox, friend_num, new_status, user_data): """ Check friend's status (none, busy, away) """ print "Friend's #{} status changed! New status: {}".format(friend_num, new_status) profile = Profile.get_instance() friend = profile.get_friend_by_number(friend_num) invoke_in_main_thread(friend.set_status, new_status) def friend_connection_status(tox, friend_num, new_status, user_data): """ Check friend's connection status (offline, udp, tcp) """ print "Friend #{} connected! Friend's status: {}".format(friend_num, new_status) profile = Profile.get_instance() friend = profile.get_friend_by_number(friend_num) if new_status == TOX_CONNECTION['NONE']: invoke_in_main_thread(friend.set_status, None) invoke_in_main_thread(profile.update_filtration) def friend_name(tox, friend_num, name, size, user_data): profile = Profile.get_instance() friend = profile.get_friend_by_number(friend_num) print 'New name: ', str(friend_num), str(name) invoke_in_main_thread(friend.set_name, name) if profile.get_active_number() == friend_num: invoke_in_main_thread(profile.set_active) def friend_status_message(tox, friend_num, status_message, size, user_data): """ :return: function for callback friend_status_message. It updates friend's status message and calls window repaint """ profile = Profile.get_instance() friend = profile.get_friend_by_number(friend_num) invoke_in_main_thread(friend.set_status_message, status_message) print 'User #{} has new status: {}'.format(friend_num, status_message) if profile.get_active_number() == friend_num: invoke_in_main_thread(profile.set_active) def friend_message(window, tray): """ :param window: main window :param tray: tray :return: function for tox.callback_friend_message. Adds new message to list """ def wrapped(tox, friend_number, message_type, message, size, user_data): profile = Profile.get_instance() settings = Settings.get_instance() invoke_in_main_thread(profile.new_message, friend_number, message_type, message) if not window.isActiveWindow(): friend = profile.get_friend_by_number(friend_number) if settings['notifications']: invoke_in_main_thread(tray_notification, friend.name, message.decode('utf8'), tray) if settings['sound_notifications']: sound_notification(SOUND_NOTIFICATION['MESSAGE']) return wrapped def friend_request(tox, public_key, message, message_size, user_data): """ Called when user get new friend request """ profile = Profile.get_instance() tox_id = bin_to_string(public_key, TOX_PUBLIC_KEY_SIZE) invoke_in_main_thread(profile.process_friend_request, tox_id, message.decode('utf-8')) # ----------------------------------------------------------------------------------------------------------------- # Callbacks - file transfers # ----------------------------------------------------------------------------------------------------------------- def tox_file_recv(window, tray): def wrapped(tox, friend_number, file_number, file_type, size, file_name, file_name_size, user_data): profile = Profile.get_instance() settings = Settings.get_instance() if file_type == TOX_FILE_KIND['DATA']: print 'file' file_name = file_name[:file_name_size].decode('utf8') invoke_in_main_thread(profile.incoming_file_transfer, friend_number, file_number, size, file_name) if not window.isActiveWindow(): friend = profile.get_friend_by_number(friend_number) if settings['notifications']: invoke_in_main_thread(tray_notification, 'File from ' + friend.name, file_name, tray) if settings['sound_notifications']: sound_notification(SOUND_NOTIFICATION['FILE_TRANSFER']) else: # AVATAR print 'Avatar' invoke_in_main_thread(profile.incoming_avatar, friend_number, file_number, size) return wrapped def file_recv_chunk(tox, friend_number, file_number, position, chunk, length, user_data): Profile.get_instance().incoming_chunk(friend_number, file_number, position, chunk if length else None) # ----------------------------------------------------------------------------------------------------------------- # Callbacks - initialization # ----------------------------------------------------------------------------------------------------------------- def init_callbacks(tox, window, tray): """ Initialization of all callbacks. :param tox: tox instance :param window: main window :param tray: tray (for notifications) """ tox.callback_self_connection_status(self_connection_status(tox), 0) tox.callback_friend_status(friend_status, 0) tox.callback_friend_message(friend_message(window, tray), 0) tox.callback_friend_connection_status(friend_connection_status, 0) tox.callback_friend_name(friend_name, 0) tox.callback_friend_status_message(friend_status_message, 0) tox.callback_friend_request(friend_request, 0) tox.callback_file_recv(tox_file_recv(window, tray), 0) tox.callback_file_recv_chunk(file_recv_chunk, 0)