OcsfPhase2

code/simplechat1/ClientConsole.java

@@ -41,11 +41,11 @@ public class ClientConsole implements ChatIF
    * @param host The host to connect to.
    * @param port The port to connect on.
    */
-  public ClientConsole(String host, int port) 
+  public ClientConsole(String loginid, String host, int port) 
   {
     try 
     {
-      client= new ChatClient(host, port, this);
+      client= new ChatClient(loginid, host, port, this);
     } 
     catch(IOException exception) 
     {
@@ -100,22 +100,45 @@ public void display(String message)
   /**
    * This method is responsible for the creation of the Client UI.
    *
-   * @param args[0] The host to connect to.
+   * @param args[0] The login ID of the client.
+   * @param args[1] The host to connect to.
+   * @param args[2] The port to connect on.
    */
   public static void main(String[] args) 
   {
+	String loginid = "";
     String host = "";
-    int port = 0;  //The port number
+    int port = 0; 
 
     try
     {
-      host = args[0];
+      loginid = args[0];
+    }
+    catch(ArrayIndexOutOfBoundsException e)
+    {
+    	System.out.println("You must provide a valid login ID.");
+    	System.exit(0); 
+    }
+
+    try
+    {
+      host = args[1];
     }
     catch(ArrayIndexOutOfBoundsException e)
     {
       host = "localhost";
     }
-    ClientConsole chat= new ClientConsole(host, DEFAULT_PORT);
+
+    try
+    {
+      port = Integer.parseInt(args[2]);
+    }
+    catch(ArrayIndexOutOfBoundsException e)
+    {
+      port = DEFAULT_PORT;
+    }
+
+    ClientConsole chat= new ClientConsole(loginid, host, port);
     chat.accept();  //Wait for console data
   }
 }

code/simplechat1/EchoServer.java

@@ -3,6 +3,8 @@
 // license found at www.lloseng.com 
 
 import java.io.*;
+
+import common.*;
 import ocsf.server.*;
 
 /**
@@ -17,93 +19,191 @@
  */
 public class EchoServer extends AbstractServer 
 {
-  //Class variables *************************************************
-
-  /**
-   * The default port to listen on.
-   */
-  final public static int DEFAULT_PORT = 5555;
-
-  //Constructors ****************************************************
-
-  /**
-   * Constructs an instance of the echo server.
-   *
-   * @param port The port number to connect on.
-   */
-  public EchoServer(int port) 
-  {
-    super(port);
-  }
-
-
-  //Instance methods ************************************************
-
-  /**
-   * This method handles any messages received from the client.
-   *
-   * @param msg The message received from the client.
-   * @param client The connection from which the message originated.
-   */
-  public void handleMessageFromClient
-    (Object msg, ConnectionToClient client)
-  {
-    System.out.println("Message received: " + msg + " from " + client);
-    this.sendToAllClients(msg);
-  }
-
-  /**
-   * This method overrides the one in the superclass.  Called
-   * when the server starts listening for connections.
-   */
-  protected void serverStarted()
-  {
-    System.out.println
-      ("Server listening for connections on port " + getPort());
-  }
-
-  /**
-   * This method overrides the one in the superclass.  Called
-   * when the server stops listening for connections.
-   */
-  protected void serverStopped()
-  {
-    System.out.println
-      ("Server has stopped listening for connections.");
-  }
-
-  //Class methods ***************************************************
-
-  /**
-   * This method is responsible for the creation of 
-   * the server instance (there is no UI in this phase).
-   *
-   * @param args[0] The port number to listen on.  Defaults to 5555 
-   *          if no argument is entered.
-   */
-  public static void main(String[] args) 
-  {
-    int port = 0; //Port to listen on
-
-    try
-    {
-      port = Integer.parseInt(args[0]); //Get port from command line
-    }
-    catch(Throwable t)
-    {
-      port = DEFAULT_PORT; //Set port to 5555
-    }
-
-    EchoServer sv = new EchoServer(port);
-
-    try 
-    {
-      sv.listen(); //Start listening for connections
-    } 
-    catch (Exception ex) 
-    {
-      System.out.println("ERROR - Could not listen for clients!");
-    }
-  }
+	//Class variables *************************************************
+
+	/**
+	 * The default port to listen on.
+	 */
+	final public static int DEFAULT_PORT = 5555;
+
+	ChatIF server;
+	boolean serverStatus;
+
+	//Constructors ****************************************************
+
+	/**
+	 * Constructs an instance of the echo server.
+	 *
+	 * @param port The port number to connect on.
+	 */
+	public EchoServer(int port) 
+	{
+		super(port);
+	}
+
+	//Instance methods ************************************************
+
+	/**
+	 * This method handles any messages received from the client.
+	 *
+	 * @param message The message received from the client.
+	 * @param client The connection from which the message originated.
+	 */
+	public void handleMessageFromClient
+	(Object message, ConnectionToClient client)
+	{
+		if(((String) message).charAt(0) == '#') {
+			if((boolean)(client.getInfo("connected"))){	
+				String[] input = ((String)message).split(" ", 0);
+				client.setInfo("connected", false);
+
+				//check to see if message is login message
+				if(input[0].equals("#login")) {
+					client.setInfo("login", input[1]);
+				}
+				else {
+					try {
+						client.sendToClient("You must log in to start session.");
+						client.close();
+					}
+					catch(IOException e){
+						System.exit(0);
+					}
+				}
+			}
+			else{
+				if(((String)message).startsWith("#login")){
+					try{
+						client.sendToClient("You are already logged in.");
+					}
+					catch(IOException e){
+						System.exit(0);
+					}
+				}
+			}
+
+			System.out.println("Message received: " + message + " from " + client.getInfo("login"));
+			this.sendToAllClients("["+ client.getInfo("login") + "] " + message);
+		}
+	}
+
+	public void processServerMessage(String message) throws IOException {
+		//create string array to handle setHost and setPort
+		String[] input = message.split(" ", 2);
+
+		if(message.charAt(0) == '#') {
+			try {
+				switch (input[0]){ 
+				case "#quit": System.exit(0);
+				break;	
+
+				case "#stop": stopListening();
+				break;
+
+				case "#close": close();
+				break;
+
+				case "#setport":
+					if(!serverStatus) {
+						setPort(Integer.parseInt(input[1]));
+					}
+					break;
+
+				case "#start":
+					if(!isListening()) {
+						this.listen();
+					}
+					break;
+
+				case "#getport":
+					server.display("Port: "+ getPort());
+					break;
+
+				default:
+					throw new IOException("Invalid Command"); 
+
+				}
+			}
+			catch(IOException e) {
+				e.printStackTrace();
+			}
+		}
+		else {
+			server.display("SERVER MSG>" + message);
+			sendToAllClients("SERVER MSG>" + message);
+		}
+
+	}
+
+	/**
+	 * This method overrides the one in the superclass.  Called
+	 * when the server starts listening for connections.
+	 */
+	protected void serverStarted()
+	{
+		System.out.println
+		("Server listening for connections on port " + getPort());
+		serverStatus = true;
+	}
+
+	/**
+	 * This method overrides the one in the superclass.  Called
+	 * when the server stops listening for connections.
+	 */
+	protected void serverStopped()
+	{
+		System.out.println
+		("Server has stopped listening for connections.");
+		serverStatus = false;
+	}
+
+	//handle client connection and disconnection
+	protected void clientConnected(ConnectionToClient client) {
+		System.out.println("The client, " + client + ", has successfully connected.");
+		client.setInfo("connected", true);
+	}
+
+	protected void clientDisconnected(ConnectionToClient client) {
+		System.out.println("The client, " + client + ", has successfully disconnected.");
+	}
+
+	protected void clientException(ConnectionToClient client, Throwable exception) {
+		clientDisconnected(client);
+	}  
+
+	//Class methods ***************************************************
+
+	/**
+	 * This method is responsible for the creation of 
+	 * the server instance (there is no UI in this phase).
+	 *
+	 * @param args[0] The port number to listen on.  Defaults to 5555 
+	 *          if no argument is entered.
+	 */
+	public static void main(String[] args) 
+	{
+		int port = 0; //Port to listen on
+
+		try
+		{
+			port = Integer.parseInt(args[0]); //Get port from command line
+		}
+		catch(Throwable t)
+		{
+			port = DEFAULT_PORT; //Set port to 5555
+		}
+
+		EchoServer sv = new EchoServer(port);
+
+		try 
+		{
+			sv.listen(); //Start listening for connections
+		} 
+		catch (Exception ex) 
+		{
+			System.out.println("ERROR - Could not listen for clients!");
+		}
+	}
+
 }
 //End of EchoServer class