#!/usr/bin/env python

########################################################################
#  Copyright (C) 2016 Max Mehl <mail [at] mehl [dot] mx>
########################################################################
#  
#  This program is free software: you can redistribute it and/or modify 
#  it under the terms of the GNU Affero General Public License as 
#  published by the Free Software Foundation, either version 3 of the 
#  License, or (at your option) any later version.
#  
#  This program 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 Affero General Public License for more details.
#  
#  You should have received a copy of the GNU Affero General Public 
#  License along with this program.  If not, see 
#  <http://www.gnu.org/licenses/>.
#  
########################################################################
#  
#  This script is responsible to add a user using vadduser. It is 
#  needed because we need the expect function and using python we can 
#  avoid potential password leaks when other users see the output of ps.
#  
########################################################################

import sys
import pexpect

# Read variables of first and second given argument
user = str(sys.argv[1])
passwordfile = str(sys.argv[2])

# Read content of passwordfile and put as variable "password"
with open (passwordfile, "r") as myfile:
    password=myfile.read().replace('\n', '')

# Define function
def _vadduser(user, password):
    adduser = pexpect.spawn(
        command='vadduser',  # actual shell command which should be started
        args=[user]   # necessary argument
    )

    adduser.expect('password:')
    adduser.sendline(password)

    adduser.expect('verification:')
    adduser.sendline(password)

    adduser.expect(pexpect.EOF)
    adduser.close()
    
    # Error handling
    if adduser.exitstatus != 0:
        last_line = filter(None, adduser.before.split('\r\n'))[-1]
        problem = last_line.split(':')[-1]
        print "ERROR! Code: " + str(adduser.exitstatus)
        print "Last error: " + str(problem)
    else:
        print "SUCCESS!"
        print "New user " + str(user) + " has been created with the password: " + str(password)

# Start function
_vadduser(user, password)