import random

from anysystem import Context, Message, Process


class Peer(Process):
    def __init__(self, proc_id: int, proc_count: int, fanout: int):
        self._id = proc_id
        self._proc_count = proc_count
        self._peers = [id for id in range(0, self._proc_count) if id != self._id]
        self._fanout = fanout
        self._info = None

    def on_local_message(self, msg: Message, ctx: Context):
        if msg.type == 'START':
            ctx.set_timer("gossip", 1)
        elif msg.type == 'BROADCAST':
            self.got_info(msg['info'], ctx)

    def on_start(self, ctx: Context):
        pass

    def on_message(self, msg: Message, sender: str, ctx: Context):
        if msg.type == 'GOSSIP' and self._info is None:
            self.got_info(msg['info'], ctx)

    def on_timer(self, timer_name: str, ctx: Context):
        if self._info is not None:
            self.gossip(ctx)
        ctx.set_timer("gossip", 1)

    def got_info(self, info, ctx):
        self._info = info
        ctx.send_local(Message('DELIVER', {'info': self._info}))

    def gossip(self, ctx):
        for peer in random.sample(self._peers, self._fanout):
            ctx.send(Message('GOSSIP', {'info': self._info}), str(peer))