import argparse
from src.email.functions import connect, check_inbox


def main(args):
    print('Connecting')
    imap_server = connect(imap_url=args.url, port=args.port, username=args.username, password=args.password,
                          ssl=args.ssl)
    print('Checking emails')
    emails = check_inbox(imap_server)
    print(len(emails))
    [print(e) for e in emails]
    imap_server.close()


if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('-u', '--url')
    parser.add_argument('-p', '--port')
    parser.add_argument('-un', '--username')
    parser.add_argument('-up', '--password')
    parser.add_argument('--ssl')
    _args = parser.parse_args()
    main(_args)