#!/usr/bin/env python3
import sys

from os import path

import pyrad.packet

from pyrad.client import Client
from pyrad.dictionary import Dictionary


def main(path_to_dictionary, coa_type, nas_identifier):
    # create coa client
    client = Client(server='127.0.0.1',
                    secret=b'Kah3choteereethiejeimaeziecumi',
                    dict=Dictionary(path_to_dictionary))

    # set coa timeout
    client.timeout = 30

    # create coa request packet
    attributes = {
        'Acct-Session-Id': '1337',
        'NAS-Identifier': nas_identifier,
    }

    if coa_type == 'coa':
        # create coa request
        request = client.CreateCoAPacket(**attributes)
    elif coa_type == 'dis':
        # create disconnect request
        request = client.CreateCoAPacket(
            code=pyrad.packet.DisconnectRequest,
            **attributes)
    else:
        sys.exit(1)

    # send request
    result = client.SendPacket(request)
    print(result)
    print(result.code)


if __name__ == '__main__':
    if len(sys.argv) != 3:
        print('usage: coa.py {coa|dis} daemon-1234')
        sys.exit(1)

    dictionary = path.join(path.dirname(path.abspath(__file__)), 'dictionary')

    main(dictionary, sys.argv[1], sys.argv[2])