import person

def logIn(friends_list):
	user = " "
	x = False
	while x == False:
		user_input = input("Enter username: ")
		user_passw = input("Password: ")
		for i in range(len(friends_list)):
			if user_input == friends_list[i].getUser() and user_passw == friends_list[i].getPassword():
				print("Username Found!\nWelcome to the CS network!")
				user = friends_list[i]
				x = True
				break
			elif i == len(friends_list) - 1:
				print("User not found! Please Try again.\n")
	return user

def menu():
	print("1. Print all my friends")
	print("2. Print all my messages/status updates")
	print("3. Post a message/status update")
	print("4. Print all my friends' messages/status updates")
	print("5. Add a friend")
	print("6. Logout (Change user)")
	print("7. Exit")

def printFriends(user):
	my_friends = user.getFriendList()

	print("Friends:", len(my_friends))

	for i in range(len(my_friends)):
		print(my_friends[i])
	print("\n")

def PrintMessage(user):
	my_messages = user.getStatUpdate()

	for i in range(len(my_messages)):
		print(my_messages[i])
	print("\n")

def postMessage(user):
	message = input("What is on your mind?\n")
	user.setStatUpdate(message)

def printFMessage(user, friends_list):
	my_friends = user.getFriendList()

	for i in my_friends:
		for x in friends_list:
			if i == x.getUser():
				print(i,"'s messages:")
				other_user = x.getStatUpdate()
				for y in range(len(other_user)):
					print(other_user[y], "\n")

def addFriend(user):
	new_friend = input("Who would you like to add?\n")
	user.setFriendList(new_friend)

def main():
	#read in data file; network.txt
		#pass the list to the readIn fucntion & open the file and read it and return the list.
	friends_list = []

	file = open("network.txt", "r")
	myfile = file.readlines()
	for line in myfile:
		line = line.split(", ")
		line[-1] = line[-1].strip(",\n")
		messages = []
		friends = []
		username = line[0]
		password = line[1]
		count = 3
		while line[count] != "friends":
			messages.append(line[count])
			count += 1
		count += 1
		while count < len(line):
			friends.append(line[count])
			count += 1

		user = person.Person(username, password, messages, friends)
		friends_list.append(user)

	#Log in
	user = logIn(friends_list)
	#Diplay Menu
	while True:
		menu()
		#Input Option
		option = int(input("Please enter a choice: \n"))
		#Carry out option selected
		if option == 1:
			printFriends(user)
		elif option == 2:
			PrintMessage(user)
		elif option == 3:
			postMessage(user)
		elif option == 4:
			printFMessage(user, friends_list)
		elif option == 5:
			addFriend(user)
		elif option == 6:
			user = " "
			user = logIn(friends_list)
		elif option == 7:
			new_file = open("network.txt", 'w')
			for i in range(len(friends_list)):
				user = []
				user.append(friends_list[i].getUser())
				user.append(friends_list[i].getPassword())
				user.append(friends_list[i].getStatUpdate())
				user.append(friends_list[i].getFriendList())
				print(user)
				all_users = ', '.join(user)
				new_file.write(all_users + '\n')



		else:
			print("That is not an option please try again!")
			break


	#If option was to exit the program, write data out to the file network.txt

main()