import socket
import Data
LISTEN_PORT = 1457
WELCOME_MESSAGE = "Welcome to the pink floyd server"
SERVER_KEYS = {"#CODE&1":"got code 1","#CODE&2":"got code 2","#CODE&3":"got code 3","#CODE&4":"got code 4","#CODE&5":"got code 5","#CODE&6":"got code 6","#CODE&7":"got code 7","#CODE&8":"got code 8",}

def main():

    listening_sock = make_listening_socket(LISTEN_PORT)
    client_msg = ""
    while listening_sock:

        client_soc, client_address = listening_sock.accept()
        try:
            client_soc.sendall(WELCOME_MESSAGE.encode())
            client_msg = (client_soc.recv(1024)).decode()
        except Exception as e:
            print("client disconnected")
            print("error: ",e)
            client_soc.close()
            continue

        while True:
            print(client_msg)
            msg = server_answer(client_msg)
            print(msg)

            try:
                client_soc.sendall(msg.encode())
                if client_msg == "#CODE&8#DATA&EXIT":
                    client_soc.close()
                    break
                client_msg = (client_soc.recv(1024)).decode()
            except Exception as e:
                print("client disconnected")
                print("error: ",e)
                client_soc.close()
                break


    if listening_sock is not False:
        listening_sock.close()

def make_listening_socket(port):
    listening_sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    server_address = ('', port)
    try:
        listening_sock.bind(server_address)
        listening_sock.listen(1)
    except Exception as e:
        print("port is already taken.")
        print("error: ",e)
        return False


    return listening_sock

def server_answer(msg):
    answer = msg[:13]
    if msg[:7] == SERVER_KEYS[msg[:7]]:
        answer = answer + album_names()
    print(answer)

if __name__ == '__main__':
    main()