#!/usr/bin/python3

import os
import sys
import subprocess

def main():
    if (len(sys.argv) < 3):
        print("No users/groups to configure.")
        return

    groups = sys.argv[1].split(';')
    users = sys.argv[2].split(';')

    # group,groupid
    for group in groups:
        elements = group.split(',')
        if (len(elements) != 2):
            print("Skipping invalid group config string \"" + group + "\"")
            continue
        subprocess.check_call("addgroup -g " + elements[1] + " " + elements[0], shell=True)
        print("Added group " + elements[0] + " with gid " + elements[1])
        

    # username,uid,password,[group]
    for user in users:
        elements = user.split(',')
        if (len(elements) != 3 and len(elements) != 4):
            print("Skipping invalid user config string \"" + user + "\"")
            continue
        subprocess.check_call("adduser -D -H -u " + elements[1] + " " + elements[0], shell=True)
        print("Added user " + elements[0] + " with uid " + elements[1])
        if (len(elements) == 4):
            subprocess.check_call("addgroup " + elements[0] + " " + elements[3], shell=True)
            print("Added user " + elements[0] + " to group " + elements[3])
        # set passwd
        subprocess.check_call("echo -ne \"" + elements[2] + "\n" + elements[2] + "\n" + "\" | smbpasswd -a -U " + elements[0], shell=True)
        print("Set user " + elements[0] + " password")

main()