package networking;

import logging.LogType;
import logging.ServerLogger;
import res.TicTacToe_GameRules;

import java.awt.*;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.HashMap;
import java.util.Map;
import java.util.Objects;

public class TicTacToe_Server {

    private ServerSocket serverSocket;
    private HashMap<Integer, Socket> clients;
    private HashMap<Socket, String> clientNames;
    private HashMap<Socket, Boolean> clientMoveAuthorizations;
    private HashMap<Socket, DataOutputStream> outstreams;
    private HashMap<Socket, DataInputStream> instreams;
    private TicTacToe_GameRules ticTacToe_gameRules;
    private ServerLogger serverLogger;
    private int requiredConnections;


    public TicTacToe_Server(int port, int requiredConnections) {
        try {
            serverSocket = new ServerSocket(port);
            clients = new HashMap<>();
            clientNames = new HashMap<>();
            clientMoveAuthorizations = new HashMap<>();
            outstreams = new HashMap<>();
            instreams = new HashMap<>();
            ticTacToe_gameRules = new TicTacToe_GameRules();
            serverLogger = new ServerLogger();
            this.requiredConnections = requiredConnections;

            serverLogger.printLog("Server started successfully", LogType.Log);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    private boolean isSingleServer() {
        return requiredConnections == 1;
    }

    public void connectClients() {
        try {
            int id = 0;
            serverLogger.printLog(String.format("Waiting for %d clients to connect ...", requiredConnections), LogType.Log);
            while (clients.size() < requiredConnections) {
                Socket momentaryClient = serverSocket.accept();
                clients.put(id, momentaryClient);
                outstreams.put(momentaryClient, new DataOutputStream(momentaryClient.getOutputStream()));
                instreams.put(momentaryClient, new DataInputStream(momentaryClient.getInputStream()));
                id++;
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public void handshake() {
        for (Socket client : clients.values()) {
            try {
                int handshakeValue = instreams.get(client).readInt();
                if (handshakeValue == 165313125) {
                    outstreams.get(client).writeInt(200);
                    outstreams.get(client).flush();
                    clientNames.put(client, instreams.get(client).readUTF());
                    serverLogger.printLog(String.format("Client \"%s\" got connected", clientNames.get(client)), LogType.Log);
                } else {
                    outstreams.get(client).writeInt(403);
                    outstreams.get(client).flush();
                }

            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

    public void sendGameState() {
        for (Socket client : clients.values()) {
            sendGameState(client);
        }
    }

    public void sendGameState(Socket client) {
        try {
            String gameState = ticTacToe_gameRules.getGameState();
            outstreams.get(client).writeUTF(gameState);
            outstreams.get(client).flush();
            serverLogger.printLog("Sent gameState", gameState, clientNames.get(client), LogType.Log);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public String handleInput(Socket client){
        String message = null;
        try {
            message = instreams.get(client).readUTF();
            serverLogger.printLog(message, clientNames.get(client), LogType.Message);
            outstreams.get(client).writeInt(200);
            outstreams.get(client).flush();
            serverLogger.printLog("Sent verification code", "200", clientNames.get(client), LogType.Log);
        } catch (IOException e) {
            e.printStackTrace();
        }
        return message;
    }

    public void ticTacToe_gameloop() {
        if (isSingleServer()) {
            Socket client = clients.get(0);
            //SingleServer GameLoop
            while (!client.isClosed()) {
                //Get instruction
                this.gameFlow(handleInput(client), client);
            }
        } else {
            //MultiServer GameLoop
            gameFlow("gameState");
        }
    }

    public void gameFlow(String input){
        this.gameFlow(input, null);
    }

    public void gameFlow(String input, Socket client) {
        switch (input) {
            case "ready":
                if (isSingleServer()){
                    sendGameState();
                } else {
                    sendGameState(client);
                }
                break;

            case "gameState":
                sendGameState();
                break;

            case "clientMove":
                try {
                    //Get position (X|Y)
                    String position = handleInput(client);
                    boolean moveAllowed = ticTacToe_gameRules.makeClientMove(position);
                    if (moveAllowed) {
                        sendGameState();
                    } else {
                        //send " "
                        outstreams.get(client).writeUTF("userInput");
                        outstreams.get(client).flush();
                        serverLogger.printLog("Requested userInput", LogType.Log);
                        serverLogger.printLog(String.format("Move is not allowed!"), clientNames.get(client), LogType.Error);
                    }
                } catch (IOException e) {
                    e.printStackTrace();
                }
                break;

            case "computerMove":
                ticTacToe_gameRules.makeComputerMove();
                sendGameState();
                try {
                    outstreams.get(client).writeUTF("userInput");
                    outstreams.get(client).flush();
                    serverLogger.printLog("Requested userInput", LogType.Log);
                } catch (IOException e) {
                    e.printStackTrace();
                }
                break;

            case "serverType":
                try {
                    outstreams.get(client).writeBoolean(true);
                    outstreams.get(client).flush();
                    serverLogger.printLog("Sent serverType", Boolean.toString(true), clientNames.get(client), LogType.Log);
                } catch (IOException e) {
                    e.printStackTrace();
                }
                break;

            case "isClientOne":
                for (Map.Entry<Integer, Socket> entry : clients.entrySet()) {
                    if (Objects.equals(client, entry.getValue())) {
                        try {
                            boolean isClientOne = entry.getKey() == 0;
                            outstreams.get(client).writeBoolean(isClientOne);
                            outstreams.get(client).flush();
                            serverLogger.printLog("Sent isPlayerOne", Boolean.toString(isClientOne), clientNames.get(client), LogType.Log);
                        } catch (IOException e) {
                            e.printStackTrace();
                        }
                    } else {
                        serverLogger.printLog("Current client not in clients!", LogType.Error);
                    }
                }
                break;

            case "gameEnded":
                try {
                    boolean gameEnded = ticTacToe_gameRules.gameEnded();
                    outstreams.get(client).writeBoolean(gameEnded);
                    if (gameEnded) {
                        //send coordinates
                        String coordinates = "";
                        for (Point point : ticTacToe_gameRules.getWinCoordinates()) {
                            coordinates += point.x + ";" + point.y + ";";
                        }
                        //send winning fields
                        outstreams.get(client).writeUTF(coordinates);
                        serverLogger.printLog("Winning coordinates got sent", coordinates, clientNames.get(client), LogType.Log);
                    }
                } catch (IOException e) {
                    e.printStackTrace();
                }
                break;

            case "exit":
                try {
                    outstreams.get(client).close();
                    instreams.get(client).close();
                    client.close();
                    serverLogger.printLog(String.format("%s closed the connection", clientNames.get(client)), LogType.Log);
                } catch (IOException e) {
                    e.printStackTrace();
                }
                break;

            case "reset":
                ticTacToe_gameRules.resetGameState();
                this.sendGameState();
                break;
        }
    }

    public static void main(String[] args) {
        TicTacToe_Server server = new TicTacToe_Server(2589, Integer.valueOf(args[0]));
        server.connectClients();
        server.handshake();
        server.ticTacToe_gameloop();
    }

}