#!/usr/bin/env python3
import logging

from os import path

import pyrad.packet

from pyrad import server
from pyrad.dictionary import Dictionary

logging.basicConfig(filename='pyrad.log', level='DEBUG',
                    format='%(asctime)s [%(levelname)-8s] %(message)s')


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


class FakeServer(server.Server):
    def HandleAuthPacket(self, packet):
        print('Received an authentication request')
        print_attributes(packet)

        reply = self.CreateReplyPacket(packet, **{
            'Service-Type': 'Framed-User',
            'Framed-IP-Address': '192.168.0.1',
            'Framed-IPv6-Prefix': 'fc66::/64'
        })

        reply.code = pyrad.packet.AccessAccept
        self.SendReplyPacket(packet.fd, reply)

    def HandleAcctPacket(self, packet):
        print('Received an accounting request')
        print_attributes(packet)

        reply = self.CreateReplyPacket(packet)
        self.SendReplyPacket(packet.fd, reply)

    def HandleCoaPacket(self, packet):
        print('Received an coa request')
        print_attributes(packet)

        reply = self.CreateReplyPacket(packet)
        self.SendReplyPacket(packet.fd, reply)

    def HandleDisconnectPacket(self, packet):
        print('Received an disconnect request')
        print_attributes(packet)

        reply = self.CreateReplyPacket(packet)
        # COA NAK
        reply.code = 45
        self.SendReplyPacket(packet.fd, reply)


def main(path_to_dictionary):
    # create server and read dictionary
    srv = FakeServer(dict=Dictionary(path_to_dictionary),
                     coa_enabled=True)

    # add clients (address, secret, name)
    srv.hosts['127.0.0.1'] = server.RemoteHost(
            '127.0.0.1',
            b'Kah3choteereethiejeimaeziecumi',
            'localhost')
    srv.BindToAddress('0.0.0.0')

    # start server
    srv.Run()


if __name__ == '__main__':
    dictionary = path.join(path.dirname(path.abspath(__file__)), 'dictionary')
    main(dictionary)