#!/usr/bin/env python3
#
# Copyright 6WIND, 2017
#

import sys

from os import path

import pyrad.packet

from pyrad.dictionary import Dictionary
from pyrad.server import Server, RemoteHost


def print_attributes(packet):
    print("Attributes")
    for key, value in packet.items():
        print(f"{key}: {value}")


class FakeCoA(Server):
    def HandleCoaPacket(self, packet):
        """Accounting packet handler.
        Function that is called when a valid
        accounting packet has been received.

        :param packet: packet to process
        :type  packet: Packet class instance
        """
        print("Received a coa request %d" % packet.code)
        print_attributes(packet)

        reply = self.CreateReplyPacket(packet)
        # try ACK or NACK
        # reply.code = packet.CoANAK
        reply.code = packet.CoAACK
        self.SendReplyPacket(packet.fd, reply)

    def HandleDisconnectPacket(self, packet):
        print("Received a disconnect request %d" % packet.code)
        print_attributes(packet)

        reply = self.CreateReplyPacket(packet)
        # try ACK or NACK
        # reply.code = packet.DisconnectNAK
        reply.code = pyrad.packet.DisconnectACK
        self.SendReplyPacket(packet.fd, reply)


def main(path_to_dictionary, coa_port):
    # create server/coa only and read dictionary
    # bind and listen only on 127.0.0.1:argv[1]
    coa = FakeCoA(
        addresses=["127.0.0.1"],
        dict=Dictionary(path_to_dictionary),
        coaport=coa_port,
        auth_enabled=False,
        acct_enabled=False,
        coa_enabled=True,
    )

    # add peers (address, secret, name)
    coa.hosts["127.0.0.1"] = RemoteHost(
        "127.0.0.1", b"Kah3choteereethiejeimaeziecumi", "localhost"
    )

    # start
    coa.Run()


if __name__ == "__main__":
    if len(sys.argv) != 2:
        print("usage: client-coa.py {portnumber}")
        sys.exit(1)

    dictionary = path.join(path.dirname(path.abspath(__file__)), "dictionary")
    main(dictionary, int(sys.argv[1]))