# Copyright 2021-2024 VyOS maintainers and contributors <maintainers@vyos.io>
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 2.1 of the License, or (at your option) any later version.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with this library.  If not, see <http://www.gnu.org/licenses/>.

# - T3344: migrate routing options from "protocols vrf" to "vrf <name> protocols"

from vyos.configtree import ConfigTree

base = ['protocols', 'vrf']

def migrate(config: ConfigTree) -> None:
    if not config.exists(base):
        # Nothing to do
        return

    vrf_base = ['vrf', 'name']
    config.set(vrf_base)
    config.set_tag(vrf_base)

    # Copy all existing static routes to the new base node under "vrf name <name> protocols static"
    for vrf in config.list_nodes(base):
        static_base = base + [vrf, 'static']
        if not config.exists(static_base):
            continue

        new_static_base = vrf_base + [vrf, 'protocols']
        config.set(new_static_base)
        config.copy(static_base, new_static_base + ['static'])
        config.set_tag(new_static_base + ['static', 'route'])

    # Now delete the old configuration
    config.delete(base)