import socket
import base64
import ssl
import time
import sys

#Variables:
hostPop,portPop = "pop.gmail.com",995
hostSmtp,portSmtp = "stmp.gmail.com",465
smtpServer = (hostSmtp,portSmtp) #SMTP Server Mail (Send)
popServer = (hostPop,portPop) #POP3 Server Mail (Receive)
username = ""
password = ""

#Var socket of POP Server
sockPop = socket.socket(socket.AF_INET,socket.SOCK_STREAM)

#Var socket of SMTP Server
sockSmtp = socket.socket(socket.AF_INET,socket.SOCK_STREAM)

#Menu:
def menu():
	menu = "\nMENU:"+\
	"\n\t1.Composer & Send Mail."+\
	"\n\t2.Load List By Title."+\
	"\n\t3.Read content of a mail."+\
	"\n\t4.Exit."
	print menu

def connectPop():
	global sockPop
	try:
		sockPop = ssl.wrap_socket(sockPop)
		sockPop.connect(popServer)
		print "Connected POP Gmail Server!"
	except:
		print "Connect Failed!"
		menu()
	recvMessage = sockPop.recv(1024)
	print recvMessage

	#Auth POP Server Mail

	sockPop.send("USER " + username + "\r\n")
	print sockPop.recv(1024)

	sockPop.send("PASS " + password + "\r\n")
	print sockPop.recv(1024)

	#Success return True

def connectStmp():
	global sockSmtp
	try:
		sockSmtp = ssl.wrap_socket(sockSmtp)
		sockSmtp.connect(smtpServer)
		print "Connected SMTP Gmail Server!"
	except:
		print "Connect Failed!"
		menu()
	recvMessage = sockSmtp.recv(1024).decode()
	print recvMessage
	if recvMessage[:3] !="220":
		print "Error Connect! False!"
		return False
	else:
		#Send Hello
		helloCommand ="EHLO Alice\r\n"
		sockSmtp.send(helloCommand.encode())
		recvMessage = sockSmtp.recv(1024).decode()

		if recvMessage[:3] !="250"
			print "Error Hello! False!"
			return False
		else:
			auth_base64 = base64.b64encode("\x00"+username+"\x00"+password)
			authMessage = "AUTH PLAIN".encode() + auth_base64 + "\r\n".encode()
			sockSmtp.send(authMessage)
			recvMessage = sockSmtp.recv(1024).decode()
			print recvMessage


#Load List by Tittle
def ListMail():
	pass

#Main handler:
def main():
	global username,password
	username = raw_input("Nhap username: ")
	password = raw_input("Nhap password: ")

	menu()


if __name__ == '__main__':
	main()