ClientConnect.java

import java.io.*;
import java.net.*;

class ClientConnect extends Thread
{
    protected Socket clientSocket;
    protected BufferedReader in = null;
    protected PrintWriter out = null;
    protected ObjectOutputStream objectOut;

    protected StockMarket mySMRef;

    protected boolean isRegistered = false;
    protected String[] tokens;

    public ClientConnect(Socket aSocket, StockMarket aSM)
    {
        clientSocket = aSocket;
        mySMRef = aSM;
        start();
    }

    public void run()
    {
        System.out.println("New client has connected, new thread started.");
        System.out.println("Client IP is: " + clientSocket.getRemoteSocketAddress() + "\n\n");
            
        try
        {
            in = new BufferedReader(new InputStreamReader(clientSocket.getInputStream()));
            out = new PrintWriter(clientSocket.getOutputStream(), true);
            objectOut = new ObjectOutputStream(clientSocket.getOutputStream());

                
            String inputText;
                
            while((inputText = in.readLine()) != null)
            {
                System.out.println("Client: " + clientSocket.getLocalSocketAddress() + " : " + inputText);
                    
                if(inputText.equals("HELO"))
                {
                    System.out.println("ACK:" + clientSocket.getLocalSocketAddress() + ":" + clientSocket.getRemoteSocketAddress());
                    out.println("ACK:" + clientSocket.getLocalSocketAddress() + ":" + clientSocket.getRemoteSocketAddress());
                    out.println("");
                }
                else if(inputText.equals("EXIT"))
                {
                    System.out.println("ACK:EXIT:Goodbye!");
                    out.println("ACK:EXIT:Connection Closed.");
                    out.println("");
                    break;
                }
                else if(inputText.equals("REGI"))
                {
                    System.out.println("ACK:REGI:"+clientSocket.getLocalSocketAddress());
                    
                    String ID = "" + clientSocket.getRemoteSocketAddress();
                    tokens = ID.split(":");
                    ID = tokens[1];

                    out.println("REGI:SUCCESS:"+ID);
                    out.println("");
                    mySMRef.registerUser(ID);
                    isRegistered = false;
                }
                else if(inputText.equals("DISP"))
                {   // Display Stock Market
                    if(mySMRef.checkID(tokens[3]))
                    {
                        String [][] aStock = mySMRef.getStockMarketState();
                        //objectOut.writeObject(mySMRef.getStockMarketState());
                        for(int i = 0; i < aStock.length; i++)
                        {                            
                            out.println("STK:"+aStock[i][0]+":"+aStock[i][1]+":"+aStock[i][3]);
                        }
                        out.println("END:EOF");
                        out.println("");
                    }
                    else
                    {
                        out.println("ERR:Not Registered");
                        out.println("");
                    }
                }
                else if(inputText.startsWith("BUY"))
                {
                    tokens = inputText.split(":");
                    
                    if(mySMRef.checkID(tokens[3]))
                    {
                        out.println("ACK:BOUGHT:"+ tokens[2] + " shares:In " + tokens[1] + ":@" + mySMRef.checkSharePrice(tokens[1]));
                    }
                    else
                    {
                        out.println("ERR:Not Registered");
                    }
                    out.println("");
                }
                else if(inputText.equals("SELL"))
                {
                    if(isRegistered)
                    {
                        out.println("ACK:SELL:Not implemented yet!");
                    }
                    else
                    {
                        out.println("ERR:Not Registered");
                    }
                    out.println("");
                }
                else if(inputText.equals("HELP"))
                {
                    out.println("Commands:");
                    out.println("REGI:");
                    out.println("BUY:");
                    out.println("SELL:");
                    out.println("EXIT:");
                    out.println("DISP:");
                    out.println("");
                }
                else
                {
                    System.out.println("DEBUG:"+inputText+":");
                }
            }
            out.close();
            in.close();

            clientSocket.close();
            isRegistered = false;
        }
        catch(IOException e)
        {
            System.out.println("Problem with socket: " + e);
        }
    }

}