# -*- coding: utf-8 -*-

import tornado
import tornadoredis
from tornadio2 import SocketConnection
from tornadio2.conn import event
import django
from django.utils.importlib import import_module
from django.conf import settings
from django.utils import simplejson

# start of kmike's sources
_engine = import_module(settings.SESSION_ENGINE)


def get_session(session_key):
    return _engine.SessionStore(session_key)


def get_user(session):
    class Dummy(object):
        pass

    django_request = Dummy()
    django_request.session = session

    return django.contrib.auth.get_user(django_request)
# end of kmike's sources


# конфиг для подключения к redis можно хранить в настройках django
ORDERS_REDIS_HOST = getattr(settings, 'ORDERS_REDIS_HOST', 'localhost')
ORDERS_REDIS_PORT = getattr(settings, 'ORDERS_REDIS_PORT', 6379)
ORDERS_REDIS_PASSWORD = getattr(settings, 'ORDERS_REDIS_PASSWORD', None)
ORDERS_REDIS_DB = getattr(settings, 'ORDERS_REDIS_DB', None)

# немного удобства
unjson = simplejson.loads
json = simplejson.dumps


class Connection(SocketConnection):
    def __init__(self, *args, **kwargs):
        super(Connection, self).__init__(*args, **kwargs)
        self.listen_redis()

    @tornado.gen.engine
    def listen_redis(self):
        """
        Вешаем подписчиков на каналы сообщений.
        """
        self.redis_client = tornadoredis.Client(
                host=ORDERS_REDIS_HOST,
                port=ORDERS_REDIS_PORT,
                password=ORDERS_REDIS_PASSWORD,
                selected_db=ORDERS_REDIS_DB
            )
        self.redis_client.connect()

        yield tornado.gen.Task(self.redis_client.subscribe, [
            'order_lock',
            'order_done'
        ])
        self.redis_client.listen(self.on_redis_queue)  # при получении сообщения
                           #  вызываем self.on_redis_queue

    def on_open(self, info):
        """
        Определяем сессию django.
        """
        self.django_session = get_session(info.get_cookie('sessionid').value)

    @event  # событие, произошедшее в браузере
    def login(self):
        """
        Определение пользователя и его возможностей
        """
        # это просто для примера входящей команды, определять юзера можно и в on_open
        self.user = get_user(self.django_session)
        self.is_client = self.user.has_perm('order.lock')
        self.is_moder = self.user.has_perm('order.delete')

    def on_message(self):
        """
        Обязательный метод.
        """
        pass

    def on_redis_queue(self, message):
        """
        Обновление в списке заказов
        """
        if message.kind == 'message':  # сообщения у редиса бывают разного типа,
                           # много сервисных, нам нужны только эти
            message_body = unjson(message.body)  # разворачиваем сабж, как вы
                                   #  поняли я передаю данные в JSON

            # в зависимости от канала получения распределяем сообщения
            if message.channel == 'order_lock':
                self.on_lock(message_body)

            if message.channel == 'order_done:
                self.on_done(message_body)

    def on_lock(self, message):
        """
        Заказ закреплён
        """
        if message['user'] != self.user.pk:  # юзеру-источнику действия сообщать о нём не надо
            self.emit('lock', message)

    def on_done(self, message):
        """
        Заказ выполнен
        """
        if message['user'] != self.user.pk:
            if self.is_client:
                message['action'] = 'hide'
            else:
                message['action'] = 'highlight'

            self.emit('done', message)

    def on_close(self):
        """
        При закрытии соединения отписываемся от сообщений
        """
        self.redis_client.unsubscribe([
            'order_lock',
            'order_done'
        ])
        self.redis_client.disconnect()