#!/usr/bin/python3

#
# Copyright (C) 2022 Nethesis S.r.l.
# SPDX-License-Identifier: GPL-2.0-only
#

import os
import nsmigration

(u, data, nmap) = nsmigration.init("passwd.json")

if data['root']:
    nsmigration.vprint(f"Setting root password")
    tmp = open('/etc/shadow.migrated', 'w')
    with open('/etc/shadow', 'r') as fp:
        for line in fp:
            if line.startswith('root'):
                tmp.write(f'root:{data["root"]}:19114:0:99999:7:::\n')
            else:
                tmp.write(line)
    os.rename('/etc/shadow.migrated', '/etc/shadow')
