#!/usr/bin/env python3

# Unclutter RADIUS configuration
#
# Move radius-server top level tag nodes to a regular node which allows us
# to specify additional general features for the RADIUS client.

import sys
from vyos.configtree import ConfigTree

if (len(sys.argv) < 1):
    print("Must specify file name!")
    sys.exit(1)

file_name = sys.argv[1]

with open(file_name, 'r') as f:
    config_file = f.read()

config = ConfigTree(config_file)
cfg_base = ['system', 'login']
if not (config.exists(cfg_base + ['radius-server']) or config.exists(cfg_base + ['radius-source-address'])):
    # Nothing to do
    sys.exit(0)
else:
    #
    # Migrate "system login radius-source-address" to "system login radius"
    #
    if config.exists(cfg_base + ['radius-source-address']):
        address = config.return_value(cfg_base + ['radius-source-address'])
        # delete old configuration node
        config.delete(cfg_base + ['radius-source-address'])
        # write new configuration node
        config.set(cfg_base + ['radius', 'source-address'], value=address)

    #
    # Migrate "system login radius-server" tag node to new
    # "system login radius server" tag node and also rename the "secret" node to "key"
    #
    if config.exists(cfg_base + ['radius-server']):
        for server in config.list_nodes(cfg_base + ['radius-server']):
            base_server = cfg_base + ['radius-server', server]
            # "key" node is mandatory
            key = config.return_value(base_server + ['secret'])
            config.set(cfg_base + ['radius', 'server', server, 'key'], value=key)

            # "port" is optional
            if config.exists(base_server + ['port']):
                port = config.return_value(base_server + ['port'])
                config.set(cfg_base + ['radius', 'server', server, 'port'], value=port)

            # "timeout is optional"
            if config.exists(base_server + ['timeout']):
                timeout = config.return_value(base_server + ['timeout'])
                config.set(cfg_base + ['radius', 'server', server, 'timeout'], value=timeout)

            # format as tag node
            config.set_tag(cfg_base + ['radius', 'server'])

            # delete old configuration node
            config.delete(base_server)

        # delete top level tag node
        if config.exists(cfg_base + ['radius-server']):
            config.delete(cfg_base + ['radius-server'])

    try:
        with open(file_name, 'w') as f:
            f.write(config.to_string())
    except OSError as e:
        print("Failed to save the modified config: {}".format(e))
        sys.exit(1)