# Path: src/lifeRaft/states/voter.py
# class Voter(State):
# 
#     def __init__(self, timeout=1.0):
#         super(Voter, self).__init__(timeout=timeout)
#         self._last_vote = None
# 
#     def on_vote_request(self, message):
#         if(self._last_vote is None and
#            message.data["lastLogIndex"] >= self._server._lastLogIndex):
#             self._last_vote = message.sender
#             self._send_vote_response_message(message)
#         else:
#             self._send_vote_response_message(message, yes=False)
# 
#         return self, None
# 
#     def _send_vote_response_message(self, msg, yes=True):
#         voteResponse = RequestVoteResponseMessage(
#             self._server._name,
#             msg.sender,
#             msg.term,
#             {"response": yes})
#         self._server.send_message_response(voteResponse)
#     
#     def on_client_message(self, message):
#         ''' return the message response for the client '''
#         return ClientFollowerResponse(message.sender, self._last_vote)
# 
# Path: src/lifeRaft/messages/append_entries.py
# class AppendEntriesMessage(BaseMessage):
# 
#     _type = BaseMessage.AppendEntries
# 
#     def __init__(self, sender, receiver, term, data):
#         BaseMessage.__init__(self, sender, receiver, term, data)
# 
# Path: src/lifeRaft/messages/client.py
# class ClientMessage(object):
#     # Types
#     Leader = 0
#     Append = 1
# 
#     def __init__(self, sender, receiver_host, receiver_port, command):
#         self._snd_id = sender
#         self._rcv_host = receiver_host
#         self._rcv_port = receiver_port
#         self._command = command
# 
#     @property
#     def command(self):
#         return self._command
# 
#     @property
#     def sender(self):
#         return self._snd_id
# 
#     @property
#     def receiver_host(self):
#         return self._receiver_host
# 
#     @property
#     def receiver_port(self):
#         return self._receiver_port
# 
# class ClientLeaderResponse(ClientResponse):
#     def __init__(self, receiver_id, leader_id, response):
#         super(ClientLeaderResponse, self).__init__(receiver_id, leader_id, response)


from collections import defaultdict
from .voter import Voter
from ..messages.append_entries import AppendEntriesMessage
from ..messages.client import ClientMessage, ClientLeaderResponse
            from .follower import Follower


class Leader(Voter):

    def __init__(self, timeout=1.0):
        super(Leader, self).__init__(timeout=timeout)
        self._nextIndexes = defaultdict(int)
        self._matchIndex = defaultdict(int)
        self._last_heartbeat = 0

    def set_server(self, server):
        self._server = server

        for n in self._server._neighbors:
            self._nextIndexes[n._name] = self._server._lastLogIndex + 1
            self._matchIndex[n._name] = 0

    def on_append_entries(self, message):
        if message.sender == self._server._name:
            # bcast

# Path: src/lifeRaft/states/voter.py
# class Voter(State):
# 
#     def __init__(self, timeout=1.0):
#         super(Voter, self).__init__(timeout=timeout)
#         self._last_vote = None
# 
#     def on_vote_request(self, message):
#         if(self._last_vote is None and
#            message.data["lastLogIndex"] >= self._server._lastLogIndex):
#             self._last_vote = message.sender
#             self._send_vote_response_message(message)
#         else:
#             self._send_vote_response_message(message, yes=False)
# 
#         return self, None
# 
#     def _send_vote_response_message(self, msg, yes=True):
#         voteResponse = RequestVoteResponseMessage(
#             self._server._name,
#             msg.sender,
#             msg.term,
#             {"response": yes})
#         self._server.send_message_response(voteResponse)
#     
#     def on_client_message(self, message):
#         ''' return the message response for the client '''
#         return ClientFollowerResponse(message.sender, self._last_vote)
# 
# Path: src/lifeRaft/messages/append_entries.py
# class AppendEntriesMessage(BaseMessage):
# 
#     _type = BaseMessage.AppendEntries
# 
#     def __init__(self, sender, receiver, term, data):
#         BaseMessage.__init__(self, sender, receiver, term, data)
# 
# Path: src/lifeRaft/messages/client.py
# class ClientMessage(object):
#     # Types
#     Leader = 0
#     Append = 1
# 
#     def __init__(self, sender, receiver_host, receiver_port, command):
#         self._snd_id = sender
#         self._rcv_host = receiver_host
#         self._rcv_port = receiver_port
#         self._command = command
# 
#     @property
#     def command(self):
#         return self._command
# 
#     @property
#     def sender(self):
#         return self._snd_id
# 
#     @property
#     def receiver_host(self):
#         return self._receiver_host
# 
#     @property
#     def receiver_port(self):
#         return self._receiver_port
# 
# class ClientLeaderResponse(ClientResponse):
#     def __init__(self, receiver_id, leader_id, response):
#         super(ClientLeaderResponse, self).__init__(receiver_id, leader_id, response)


# Path: src/lifeRaft/states/leader.py
from collections import defaultdict
from .voter import Voter
from ..messages.append_entries import AppendEntriesMessage
from ..messages.client import ClientMessage, ClientLeaderResponse
            from .follower import Follower



class Leader(Voter):

    def __init__(self, timeout=1.0):
        super(Leader, self).__init__(timeout=timeout)
        self._nextIndexes = defaultdict(int)
        self._matchIndex = defaultdict(int)
        self._last_heartbeat = 0

    def set_server(self, server):
        self._server = server

        for n in self._server._neighbors:
            self._nextIndexes[n._name] = self._server._lastLogIndex + 1
            self._matchIndex[n._name] = 0

    def on_append_entries(self, message):
        if message.sender == self._server._name:
            # bcast

            return self, None