"""scs.presence"""

from __future__ import absolute_import, with_statement

import warnings

from collections import defaultdict
from contextlib import contextmanager
from functools import wraps
from random import shuffle
from time import time, sleep

from kombu import Exchange, Queue
from kombu.common import ipublish
from kombu.log import LogMixin
from kombu.mixins import ConsumerMixin
from kombu.pools import producers
from kombu.utils.functional import promise

from .agents import Agent
from .g import spawn, timer
from .utils import cached_property, first_or_raise, shortuuid


class State(LogMixin):
    logger_name = "cl.presence.state"

    def __init__(self, presence):
        self.presence = presence
        self._agents = defaultdict(lambda: {})
        self.heartbeat_expire = self.presence.interval * 2.5
        self.handlers = {"online": self.when_online,
                         "offline": self.when_offline,
                         "heartbeat": self.when_heartbeat,
                         "wakeup": self.when_wakeup}

    def can(self, actor):
        able = set()
        for id, state in self.agents.iteritems():
            if actor in state["actors"]:
                # remove the . from the agent, which means that the
                # agent is a clone of another agent.
                able.add(id.partition(".")[0])
        return able

    def meta_for(self, actor):
        return self._agents["meta"][actor]

    def update_meta_for(self, agent, meta):
        self._agents[agent].update(meta=meta)

    def agents_by_meta(self, predicate, *sections):
        agents = self._agents
        agent_ids = agents.keys()
        # shuffle the agents so we don't get the same agent every time.
        shuffle(agent_ids)
        for agent in agent_ids:
            d = agents[agent]["meta"]
            for i, section in enumerate(sections):
                d = d[section]
            if predicate(d):
                yield agent

    def first_agent_by_meta(self, predicate, *sections):
        for agent in self.agents_by_meta(predicate, *sections):
            return agent
        raise KeyError()

    def on_message(self, body, message):
        event = body["event"]
        self.handlers[event](**body)
        self.debug("agents after event recv: %s", promise(lambda: self.agents))

    def when_online(self, agent=None, **kw):
        self._update_agent(agent, kw)

    def when_wakeup(self, **kw):
        self.presence.send_heartbeat()

    def when_heartbeat(self, agent=None, **kw):
        self._update_agent(agent, kw)

    def when_offline(self, agent=None, **kw):
        self._remove_agent(agent)

    def expire_agents(self):
        expired = set()
        for id, state in self._agents.iteritems():
            if state and state.get("ts"):
                if time() > state["ts"] + self.heartbeat_expire:
                    expired.add(id)

        for id in expired:
            self._remove_agent(id)
        return self._agents

    def update_agent(self, agent=None, **kw):
        return self._update_agent(agent, kw)

    def _update_agent(self, agent, kw):
        kw = dict(kw)
        meta = kw.pop("meta", None)
        if meta:
            self.update_meta_for(agent, meta)
        self._agents[agent].update(kw)

    def _remove_agent(self, agent):
        self._agents[agent].clear()

    def neighbors(self):
        return {"agents": self.agents.keys()}

    @property
    def agents(self):
        return self.expire_agents()


class Event(dict):
    pass


class Presence(ConsumerMixin):
    Event = Event
    State = State

    exchange = Exchange("cl.agents", type="topic", auto_delete=True)
    interval = 10
    _channel = None
    g = None

    def __init__(self, agent, interval=None, on_awake=None):
        self.agent = agent
        self.state = self.State(self)
        self.interval = interval or self.interval
        self.connection = agent.connection
        self.on_awake = on_awake

    def get_queue(self):
        return Queue("cl.agents.%s" % (self.agent.id, ), self.exchange,
                     routing_key="#", auto_delete=True)

    def get_consumers(self, Consumer, channel):
        return [Consumer(self.get_queue(),
                         callbacks=[self.state.on_message], no_ack=True)]

    def create_event(self, type):
        return self.Event(agent=self.agent.id,
                          event=type,
                          actors=[actor.name for actor in self.agent.actors],
                          meta=self.meta(),
                          ts=time(),
                          neighbors=self.state.neighbors())

    def meta(self):
        return dict((actor.name, actor.meta) for actor in self.agent.actors)

    @contextmanager
    def extra_context(self, connection, channel):
        self.send_online()
        self.wakeup()
        sleep(1.0)
        if self.on_awake:
            self.on_awake()
        timer(self.interval, self.send_heartbeat)
        self.agent.on_presence_ready()
        yield
        self.send_offline()

    def _announce(self, event, producer=None):
        producer.publish(event, exchange=self.exchange.name,
                                routing_key=self.agent.id)

    def announce(self, event, **retry_policy):
        return ipublish(producers[self.agent.connection],
                        self._announce, (event, ), **retry_policy)

    def start(self):
        self.g = spawn(self.run)

    def send_online(self):
        return self.announce(self.create_event("online"))

    def send_heartbeat(self):
        return self.announce(self.create_event("heartbeat"))

    def send_offline(self):
        return self.announce(self.create_event("offline"))

    def wakeup(self):
        event = self.create_event("wakeup")
        self.state.update_agent(**event)
        return self.announce(event)

    def can(self, actor):
        return self.state.can(actor)

    @property
    def logger_name(self):
        return "Presence#%s" % (shortuuid(self.agent.id), )

    @property
    def should_stop(self):
        return self.agent.should_stop


class AwareAgent(Agent):

    def on_run(self):
        self.presence.start()

    def get_default_scatter_limit(self, actor):
        able = self.presence.can(actor)
        if not able:
            warnings.warn("Presence running, but no agents available?!?")
        return len(able) if able else None

    def on_awake(self):
        pass

    def on_presence_ready(self):
        pass

    def lookup_agent(self, pred, *sections):
        return self.presence.state.first_agent_by_meta(pred, *sections)

    def lookup_agents(self, pred, *sections):
        return self.presence.state.agents_by_meta(pred, *sections)

    @cached_property
    def presence(self):
        return Presence(self, on_awake=self.on_awake)


class AwareActorMixin(object):
    meta_lookup_section = None

    def lookup(self, value):
        if self.agent:
            return self.agent.lookup_agent(lambda values: value in values,
                                           self.name, self.meta_lookup_section)

    def send_to_able(self, method, args={}, to=None, **kwargs):
        actor = None
        try:
            actor = self.lookup(to)
        except KeyError:
            raise self.NoRouteError(to)

        if actor:
            return self.send(method, args, to=actor, **kwargs)
        r = self.scatter(method, args, propagate=True, **kwargs)
        if r:
            return first_or_raise(r, self.NoRouteError(to))

    def wakeup_all_agents(self):
        if self.agent:
            self.log.info("presence wakeup others")
            self.agent.presence.wakeup()


def announce_after(fun):

    @wraps(fun)
    def _inner(self, *args, **kwargs):
        try:
            return fun(self, *args, **kwargs)
        finally:
            self.actor.wakeup_all_agents()
    return _inner