import pycurl
import sys, getopt
from StringIO import StringIO
import json
import copy

#####################################################################################################################################################################################
#																																		       									 	#
# Import all users from a given file																										   										#
# 																																			   										#
# activation :																																   										#
#       python importUsers.py  [-s <scheme> | --scheme=<scheme> ] [-i <be host> | --ip=<be host>] [-p <be port> | --port=<be port> ] [-f <input file> | --ifile=<input file> ]   	#
#																																		  	   										#
# shortest activation (be host = localhost, be port = 8080): 																				   										#
#		python importUsers.py [-f <input file> | --ifile=<input file> ]												 				           										#
#																																		       										#
#####################################################################################################################################################################################


def importUsers(scheme, beHost, bePort, users, adminUser):
	
	result = []	

	for user in users:
			
		#print("Going to add user " + user['userId'])
	
		getRes = getUser(scheme, beHost, bePort, user)	
		userId = getRes[0]
		error = getRes[1]
		#print error
		if ( error != None and error == 404 ):
			res = createUser(scheme, beHost, bePort, user ,adminUser)			
			result.append(res)			
		else:
			if ( error == 200 ):
				curResult = (userId, 409)
				result.append(curResult)	
			else:
				result.append(getRes)

	return result				


def convertUsersToCreationObject(users):

	cloneUsers  = copy.deepcopy(users)
	for user in cloneUsers:
		#print user	
		if (user.get('fullName') != None):
			del user['fullName']
		#user['userId'] = user['userId'] + '1'
		#print user

	return cloneUsers

def getUser(scheme, beHost, bePort, user):

	userId = user['userId']
	try:
		buffer = StringIO()
		c = pycurl.Curl()

		#print type(userId)
		url = scheme + '://' + beHost + ':' + bePort + '/sdc2/rest/v1/user/' + str(userId)
		c.setopt(c.URL, url)

		#adminHeader = 'USER_ID: ' + adminUser
		c.setopt(pycurl.HTTPHEADER, ['Content-Type: application/json', 'Accept: application/json'])
		c.setopt(c.WRITEFUNCTION, lambda x: None)

		if scheme == 'https':
			c.setopt(pycurl.SSL_VERIFYPEER, 0)
			c.setopt(pycurl.SSL_VERIFYHOST, 0)

		res = c.perform()
					
		#print("Before get response code")	
		httpRes = c.getinfo(c.RESPONSE_CODE)
		#print("After get response code")	
		responseCode = c.getinfo(c.RESPONSE_CODE)
		
		#print('Status: ' + str(responseCode))

		c.close()

		return (userId, httpRes)

	except Exception as inst:
		print(inst)
		return (userId, None)				

		

def createUser(scheme, beHost, bePort, user, adminUser):

	userId = user['userId']
	try:
		buffer = StringIO()
		c = pycurl.Curl()

		url = scheme + '://' + beHost + ':' + bePort + '/sdc2/rest/v1/user'
		c.setopt(c.URL, url)
		c.setopt(c.POST, 1)		

		adminHeader = 'USER_ID: ' + adminUser
		c.setopt(pycurl.HTTPHEADER, ['Content-Type: application/json', 'Accept: application/json', adminHeader])

		data = json.dumps(user)
		c.setopt(c.POSTFIELDS, data)	

		c.setopt(c.WRITEFUNCTION, lambda x: None)

		if scheme == 'https':
			c.setopt(pycurl.SSL_VERIFYPEER, 0)
			c.setopt(pycurl.SSL_VERIFYHOST, 0)

		#print("before perform")	
		res = c.perform()
        #print(res)
	
		#print("Before get response code")	
		httpRes = c.getinfo(c.RESPONSE_CODE)
		#print("After get response code")	
		responseCode = c.getinfo(c.RESPONSE_CODE)
		
		#print('Status: ' + str(responseCode))

		c.close()

		return (userId, httpRes)

	except Exception as inst:
		print(inst)
		return (userId, None)				


def errorAndExit(errorCode, errorDesc):
	if ( errorCode > 0 ):
		print("status=" + str(errorCode) + ". " + errorDesc) 
	else:
		print("status=" + str(errorCode))
	sys.exit(errorCode)
	
def usage():
	print sys.argv[0], '[optional -s <scheme> | --scheme=<scheme>, default http] [-i <be host> | --ip=<be host>] [-p <be port> | --port=<be port> ] [-f <input file> | --ifile=<input file> ]'

def main(argv):
	print 'Number of arguments:', len(sys.argv), 'arguments.'

	beHost = 'localhost' 
	bePort = '8080'
	inputfile = None 
	scheme = 'http'
	adminUser = 'jh0003'

	try:
		opts, args = getopt.getopt(argv,"i:p:f:h:s:",["ip=","port=","ifile=","scheme="])
	except getopt.GetoptError:
		usage()
		errorAndExit(2, 'Invalid input')
		 
	for opt, arg in opts:
	#print opt, arg
		if opt == '-h':
			usage()                        
			sys.exit(3)
		elif opt in ("-i", "--ip"):
			beHost = arg
		elif opt in ("-p", "--port"):
			bePort = arg
		elif opt in ("-f", "--ifile"):
			inputfile = arg
		elif opt in ("-s", "--scheme"):
			scheme = arg

	print 'scheme =',scheme,', be host =',beHost,', be port =', bePort,', users file =',inputfile
	
	if ( inputfile == None ):
		usage()
		sys.exit(3)

	print 'Input file is ', inputfile

	usersFile = open(inputfile)

	json_data = json.load(usersFile)

	#print json_data

	cloneUsers = convertUsersToCreationObject(json_data)

	activeUsers = filter(lambda x: x['status'] == 'ACTIVE', cloneUsers)

	#print activeUsers

	resultTable = importUsers(scheme, beHost, bePort, activeUsers, adminUser)

	g = lambda x: x[1] != 201 and x[1] != 409

	result = filter(g, resultTable)

	if ( len(result) > 0 ):
		#print("ERROR: Failed to load the users " + ', '.join(map(lambda x: x[0],result)))
		errorAndExit(3, "Failed to load the users " + ', '.join(map(lambda x: x[0],result)))	

	g = lambda x: x[1] == 409
	result = filter(g, resultTable)

	print("-------------------------------------------")
	print("Existing users: " + ', '.join(map(lambda x: x[0],result)))

	result = filter(lambda x: x[1] == 201, resultTable)
	if ( len(result) == 0 ):
		print("-------------------------------------------")
		print("No NEW user was loaded. All users are already exist")
		print("-------------------------------------------")
	else:
		print("-------------------------------------------")
		print("Loaded users: " + ', '.join(map(lambda x: x[0],result)))
		print("-------------------------------------------")

	errorAndExit(0, None)


if __name__ == "__main__":
        main(sys.argv[1:])