import socket
import json
import server
import func
import sys, errno
import _thread
import os

Buffer_size = 1024


def login(s):
    print("Welcome")
    user = input("Username:")
    password = input("Password:")
    password=password[::-1]
    s.send(user.encode())
    s.send(password.encode())
    if(s.recv((Buffer_size)).decode()=='0'):
        print("Login errado")
        return(None,0)
    else:
        print("Login correto")
        return(user,1)


def client():
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    port = int(sys.argv[1])
    host = 'localhost'
    dest = (host, port)
    sock.connect(dest)  # connect socket to server
    a=0
    #while(not a):
    #    user,a=login(sock)
    #print("Bem vindo ", user)

    #menu(sock)

def menu(sock):
    ans = True
    while ans:
        if ans != 8:
            func.recebe_nlidas(sock)
        print("""
            1.Listar Mensagens por ler
            *****************************
            2.Listar todos os clientes autorizados
            *****************************
            3.Enviar uma mensagem
            *****************************
            4.Listar todas as mensagens já lidas
            *****************************
            5.Apagar mensagens
            *****************************
            6.Alterar password
            *****************************
            7.Obter privilégios de operador
            *****************************
            8.Criar Utilizador
            *****************************
            9.Sair do sistema
            *****************************
            """)
        ans = input("Escolha a sua opção:")
        if ans == "1":
            print("\nListar Mensagens por ler")
            sock.send(("1").encode())
            func.le_mesagens_client(sock)
            if(func.return_menu()=='0'):
                sock.close()
                exit()

        elif ans == "2":
            print("\nListar todos os clientes autorizados")
            sock.send(("2").encode())
            func.lista_users_client(sock)
            if (func.return_menu() == '0'):
                sock.close()
                exit()
        elif ans == "3":
            print("\nEnviar uma mensagem")
            sock.send(("3").encode())
            a = 1
            while a:
                a = func.verifica_user_client(sock)
            msg = func.pedeMensagem(sock)

            if (func.return_menu() == '0'):
                sock.close()
                exit()
        elif ans == "4":
            print("\nListar todas as mensagens já lidas")
            sock.send(("4").encode())
            func.le_mesagens_client(sock)
            if (func.return_menu() == '0'):
                sock.close()
                exit()
        elif ans == "5":
            print("\nApagar mensagens")
            sock.send(("5").encode())
            func.le_mes_client(sock)
            if (func.return_menu() == '0'):
                sock.close()
                exit()
        elif ans == "6":
            print("\nAlterar password")
            sock.send(("6").encode())
            func.alteraPass(sock)
            if (func.return_menu() == '0'):
                sock.close()
                exit()
        elif ans == "7":
            print("\nObter privilegios de operador")
            sock.send(("7").encode())
            b = (sock.recv(Buffer_size)).decode()
            if b == "1":
                user_oper = input((sock.recv(Buffer_size)).decode() + "\n")
                sock.send(user_oper.encode())
                print((sock.recv(Buffer_size).decode()))

            else:
                print("Erro, impossivel dar permissao pois este user nao e operador")
            if (func.return_menu() == '0'):
                sock.close()
                exit()
        elif ans == "8":
            print("\nCria utilizador")
            sock.send(("8").encode())
            b = (sock.recv(Buffer_size)).decode()
            if b == "1":
                func.pedeNovoUtilizador(sock)
                print ((sock.recv(Buffer_size)).decode())
            else:
                print ((sock.recv(Buffer_size)).decode())

            if (func.return_menu() == '0'):
                sock.close()
                exit()
        elif ans == "9":
            print("\nSair do sistema")
            sock.send(("9").encode())
            sock.close()
            exit()
        else:
            print("Escolha inválida")
            ans = 8

if __name__ == "__main__":
    client()