Party-Parrots-At-Sea/src/main/java/seng302/controllers/StartScreenController.java

package seng302.controllers;

import java.net.Inet4Address;
import java.net.NetworkInterface;
import java.util.Enumeration;
import javafx.fxml.FXML;
import javafx.fxml.FXMLLoader;
import javafx.scene.control.Alert;
import javafx.scene.control.Alert.AlertType;
import javafx.scene.control.TextField;
import javafx.scene.layout.AnchorPane;
import javafx.scene.layout.GridPane;
import javafx.scene.layout.Pane;
import seng302.client.ClientState;
import seng302.client.ClientToServerThread;
import seng302.gameServer.GameState;
import seng302.gameServer.MainServerThread;

import java.io.IOException;
import java.net.InetAddress;
import java.net.UnknownHostException;

/**
 * A Class describing the actions of the start screen controller
 * Created by wmu16 on 10/07/17.
 */
public class StartScreenController {

    @FXML
    private TextField ipTextField;
    @FXML
    private TextField portTextField;
    @FXML
    private GridPane startScreen2;

    private Controller controller;

    /**
     * Loads the fxml content into the parent pane
     * @param jfxUrl
     * @return the controller of the fxml
     */
    private Object setContentPane(String jfxUrl) {
        try {
            AnchorPane contentPane = (AnchorPane) startScreen2.getParent();
            contentPane.getChildren().removeAll();
            contentPane.getChildren().clear();
            contentPane.getStylesheets().add(getClass().getResource("/css/master.css").toString());
            FXMLLoader fxmlLoader = new FXMLLoader(getClass().getResource(jfxUrl));
            contentPane.getChildren().addAll((Pane) fxmlLoader.load());

            return fxmlLoader.getController();
        } catch (javafx.fxml.LoadException e) {
            System.out.println("[Controller] FXML load exception");
        } catch (IOException e) {
            System.out.println("[Controller] IO exception");
        }
        return null;
    }


    /**
     * ATTEMPTS TO:
     * Sets up a new game state with your IP address as designated as the host.
     * Starts a thread to listen for incoming connections.
     * Starts a client to server thread and connects to own ip.
     * Switches to the lobby screen
     */
    @FXML
    public void hostButtonPressed() {
        try {
            String ipAddress = InetAddress.getLocalHost().getHostAddress();
            // get the lobby controller so that we can pass the game server thread to it
            new GameState(getLocalHostIp());
            MainServerThread mainServerThread = new MainServerThread();
            ClientState.setHost(true);
            // host will connect and handshake to itself after setting up the server
            // TODO: 24/07/17 wmu16 - Make port number some static global type constant?
            ClientToServerThread clientToServerThread = new ClientToServerThread(ClientState.getHostIp(), 4942);
            ClientState.setConnectedToHost(true);
            controller.setClientToServerThread(clientToServerThread);
            LobbyController lobbyController = (LobbyController) setContentPane("/views/LobbyView.fxml");
            lobbyController.setMainServerThread(mainServerThread);
            lobbyController.setController(controller);
        } catch (Exception e) {
            Alert alert = new Alert(AlertType.ERROR);
            alert.setHeaderText("Cannot host");
            alert.setContentText("Oops, failed to host, try to restart.");
            alert.showAndWait();
        }


    }

    /**
     * ATTEMPTS TO:
     * Connect to an ip address and port using the ip and port specified on start screen.
     * Starts a Client To Server Thread to maintain connection to host.
     * Switch view to lobby view.
     */
    @FXML
    public void connectButtonPressed() {
        // TODO: 10/07/17 wmu16 - Finish function
        try {
            String ipAddress = ipTextField.getText().trim().toLowerCase();
            Integer port = Integer.valueOf(portTextField.getText().trim());

            ClientToServerThread clientToServerThread = new ClientToServerThread(ipAddress, port);
            ClientState.setHost(false);
            ClientState.setConnectedToHost(true);

            ClientState.setHostIp(ipAddress);
            controller.setClientToServerThread(clientToServerThread);
            LobbyController lobbyController = (LobbyController) setContentPane("/views/LobbyView.fxml");
            lobbyController.setController(controller);
        } catch (Exception e) {
            Alert alert = new Alert(AlertType.ERROR);
            alert.setHeaderText("Cannot reach the host");
            alert.setContentText("Please check your host IP address.");
            alert.showAndWait();
        }
    }

    public void setController(Controller controller) {
        this.controller = controller;
    }

    /**
     * Gets the local host ip address and sets this ip to ClientState.
     * Only runs by the host.
     *
     * @return the localhost ip address
     */
    private String getLocalHostIp() {
        String ipAddress = null;
        try {
            Enumeration<NetworkInterface> e = NetworkInterface.getNetworkInterfaces();
            while (e.hasMoreElements()) {
                NetworkInterface ni = e.nextElement();
                if (ni.isLoopback())
                    continue;
                if(ni.isPointToPoint())
                    continue;
                if(ni.isVirtual())
                    continue;

                Enumeration<InetAddress> addresses = ni.getInetAddresses();
                while(addresses.hasMoreElements()) {
                    InetAddress address = addresses.nextElement();
                    if(address instanceof Inet4Address) {    // skip all ipv6
                        ipAddress = address.getHostAddress();
                    }
                }
            }
        } catch (Exception e) {
            System.out.println("[StartScreenController] Exception");
        }
        if (ipAddress == null) {
            System.out.println("[HOST] Cannot obtain local host ip address.");
        }
        ClientState.setHostIp(ipAddress);
        return ipAddress;
    }
}