from PLC.Faults import *
from PLC.Method import Method
from PLC.Parameter import Parameter, Mixed
from PLC.Auth import Auth
from PLC.Peers import Peer, Peers

can_update = lambda field_value: field_value[0] in \
             ['peername', 'peer_url', 'key', 'cacert', 'shortname', 'hrn_root']

class UpdatePeer(Method):
    """
    Updates a peer. Only the fields specified in peer_fields are
    updated, all other fields are left untouched.

    Returns 1 if successful, faults otherwise.
    """

    roles = ['admin']

    peer_fields = dict(list(filter(can_update, list(Peer.fields.items()))))

    accepts = [
        Auth(),
        Mixed(Peer.fields['peer_id'],
              Peer.fields['peername']),
        peer_fields
        ]

    returns = Parameter(int, "1 if successful")

    def call(self, auth, peer_id_or_name, peer_fields):
        peer_fields = dict(list(filter(can_update, list(peer_fields.items()))))

        # Get account information
        peers = Peers(self.api, [peer_id_or_name])
        if not peers:
            raise PLCInvalidArgument("No such peer")
        peer = peers[0]

        if isinstance(self.caller, Peer):
            if self.caller['peer_id'] != peer['peer_id']:
                raise PLCPermissionDenied("Not allowed to update specified peer")

        peer.update(peer_fields)
        peer.sync()

        # Log affected objects
        self.event_objects = {'Peer': [peer['peer_id']]}

        return 1