MarketRun.java

import javax.swing.*;
import java.io.*;
import java.net.*;
import java.util.*;
import java.lang.Runnable;
import java.text.*;

public class MarketRun extends Thread implements Runnable {
    private DataInputStream socketReader;
    private DataOutputStream socketWriter;
    private Socket newSocket;
    private MarketServer server;
    private volatile boolean running = true;
    public static HashMap <String, Sellers> sellers = new HashMap<String, Sellers>();
    public static HashMap <String, Customers> customers = new HashMap<String, Customers>();

    public MarketRun(Socket newSocket, DataInputStream socketReader, DataOutputStream socketWriter, MarketServer server) throws IOException {
        this.newSocket = newSocket;
        this.socketReader = socketReader;
        this.socketWriter = socketWriter;
        this.server = server;
    }

    public void run() {
        try {
            populateHashMaps();
        } catch (IOException e) {
            e.printStackTrace();
        }
        while (running) {
            try {
                String user = "";
                final boolean[] validUser = {false};
                do {
                    int userNum = 0;
                    if (socketReader.available() > 0) {
                        user = socketReader.readUTF();
                        userNum = 0;
                        if (user.equals("Customer")) {
                            userNum = 1;
                        } else if (user.equals("Seller")) {
                            userNum = 2;
                        } else if (user.equals("Create Account")) {
                            userNum = 3;
                        }
                    }
        
                    if (userNum == 1) {
                        boolean loggedIn = false;
                        do {
                            BufferedReader br = new BufferedReader(new FileReader("users.txt"));
                            String username = socketReader.readUTF();
                            String password = socketReader.readUTF();
        
                            //User authentication (goes through users.txt file, validates password) PROGRAM QUITS AFTER AUTHENTICATION
                            String line;
                            try {
                                line = br.readLine();
                                while (line != null) {
                                    String[] lineSplit = line.split(";");
                                    if (lineSplit[1].equals(username)) {
                                        if (lineSplit[2].equals(password)) {
                                            loggedIn = true;
                                            validUser[0] = true;
                                            socketWriter.writeUTF("loggedIn");
                                        }
                                    }
                                    line = br.readLine();
                                }
                            } catch (IOException e) {
                                e.printStackTrace();
                            }
        
                            //Customer homepage
                            if (loggedIn) {
                                int selectionNum = 0;
                                do {
                                    String selectionInfo = socketReader.readUTF();
                                    String[] selectionSplit = selectionInfo.split(". ");
                                    selectionNum = Integer.parseInt(selectionSplit[0]);
        
                                    switch (selectionNum) {
                                        //View marketplace QUITS WHEN VIEW MARKETPLACE
                                        case 1:
                                            ArrayList<String> market = new ArrayList<String>();
                                            market = printMarket(); //problem, why?   
                                            String marketplace = "";
        
                                            for (int i = 0; i < market.size(); i++) {
                                                marketplace += market.get(i) + "\n";
                                            }
                                            if (marketplace.equals("")) {
                                                marketplace = "There is nothing currently in the marketplace.";
                                            }

                                            System.out.println(marketplace);
        
                                            socketWriter.writeUTF(marketplace);
        
                                            int sort = Integer.parseInt(socketReader.readUTF());
        
                                            if (sort == 0) {
                                                int sortingNum = Integer.parseInt(socketReader.readUTF());
        
                                                if (sortingNum == 1) {
                                                    ArrayList<String> sortedMarket = (customers.get(username)).sortByPrice(market);
                                                    marketplace = null;
                                                    for (int i = 0; i < sortedMarket.size(); i++) {
                                                        marketplace += sortedMarket.get(i) + "\n";
                                                    }
                                                    if (marketplace == null) {
                                                        marketplace = "There is nothing currently in the marketplace.";
                                                    }
                                                    socketWriter.writeUTF(marketplace);
        
                                                } else if (sortingNum == 2) {
                                                    ArrayList<String> sortedMarket = (customers.get(username)).sortByQuantity(market);
                                                    marketplace = null;
                                                    for (int i = 0; i < sortedMarket.size(); i++) {
                                                        marketplace += sortedMarket.get(i) + "\n";
                                                    }
                                                    if (marketplace == null) {
                                                        marketplace = "There is nothing currently in the marketplace.";
                                                    }
                                                    socketWriter.writeUTF(marketplace);
        
                                                }
                                            } else {
        
                                            }
        
                                            if (marketplace != null) {
                                                String selectedProduct = socketReader.readUTF();
                                                String viewProduct = viewProduct(selectedProduct);
                                                socketWriter.writeUTF(viewProduct);
                                                int purchasing = 0;
            
                                                if (!viewProduct.isEmpty()) {
                                                    purchasing = Integer.parseInt(socketReader.readUTF());
            
                                                    if (purchasing == 0) {
                                                        int availableQuantity = checkQuantity(selectedProduct);
                                                        int quantity = Integer.parseInt(socketReader.readUTF());
                                                        String available = "no";
            
                                                        if (availableQuantity - quantity >= 0) {
                                                            productPurchased(selectedProduct, username, quantity);
                                                            double price = checkPrice(selectedProduct);
                                                            String store = checkStore(selectedProduct);
                                                            String description = checkDescription(selectedProduct);
                                                            (customers.get(username)).updatePurchaseHistory(selectedProduct, store, description, quantity, price);
                                                            available = "yes";
                                                            socketWriter.writeUTF(available);
                                                        } else {
                                                            socketWriter.writeUTF(available);
                                                        }
                                                    }
                                                }
                                            }
        
                                            break;
                                        //Search for product
                                        case 2:
                                            ArrayList<String> newMarket = new ArrayList<String>();
                                            newMarket = printMarket();
                                            int choiceNum = Integer.parseInt(socketReader.readUTF());
        
                                            if (choiceNum == 1) {
                                                String product = socketReader.readUTF();
                                                (customers.get(username)).searchProductName(newMarket, product);
                                                //writeUTF it back to client
                                            } else if (choiceNum == 2) {
                                                String store = socketReader.readUTF();
                                                (customers.get(username)).searchProductStore(newMarket, store);
                                            } else if (choiceNum == 3) {
                                                String description = socketReader.readUTF();
                                                (customers.get(username)).searchProductDescription(newMarket, description);
                                            }
                                            break;
                                        case 3:
                                            String purchaseHistory = (customers.get(username)).getPurchaseHistory();
                                            socketWriter.writeUTF(purchaseHistory);
                                            break;
                                        //Logout
                                        case 4:
                                        break;
                                        default:
                                    }
                                } while (selectionNum != 4);
                            } else {
                            }
                        } while (!loggedIn);
        
                        //Seller section LOGIN SECTION NEEDS TO BE I/O
                    } else if (userNum == 2) {
                        boolean loggedIn = false;
                        do {
                            BufferedReader br = new BufferedReader(new FileReader("users.txt"));
                            String username = socketReader.readUTF();
                            String password = socketReader.readUTF();
        
                            //User authentication (goes through users.txt file, validates password) PROGRAM QUITS AFTER AUTHENTICATION
                            String line;
                            try {
                                line = br.readLine();
                                while (line != null) {
                                    String[] lineSplit = line.split(";");
                                    if (lineSplit[1].equals(username)) {
                                        if (lineSplit[2].equals(password)) {
                                            loggedIn = true;
                                            validUser[0] = true;
                                            socketWriter.writeUTF("loggedIn");
                                        }
                                    }
                                    line = br.readLine();
                                }
                            } catch (IOException e) {
                                e.printStackTrace();
                            }
        
                            if (loggedIn) {
                                String selection;
                                int selectionNum = 0;
                                //Seller homepage
                                do {
                                    selection = socketReader.readUTF();
                                    selectionNum = Integer.parseInt(selection);
        
                                    switch (selectionNum) {
                                        //Create product
                                        case 1:
                                            String product = socketReader.readUTF();
        
                                            String[] storesArray = (sellers.get(username).getStores()).split(", ");
    
                                            String allStores = "";
                                            for (String store : storesArray) {
                                                allStores = allStores + store + ",";
                                            }
                                            socketWriter.writeUTF(allStores);
        
                                            String productStore = socketReader.readUTF();
                                            boolean storeExists = storeExists(productStore);
                                            if (storeExists) {
                                                socketWriter.writeUTF("yes");
                                            } else {
                                                socketWriter.writeUTF("no");
                                            }
        
                                            if (storeExists) {
                                                //Get all info for product, add product
                                                String description = socketReader.readUTF();
                                                int quantity = Integer.parseInt(socketReader.readUTF());
                                                double price = Double.parseDouble(socketReader.readUTF());
        
                                                sellers.get(username).addProduct(product, productStore, description, quantity, price);
                                            }
                                            break;
                                        //Edit/Delete product
                                        case 2:
                                            String choice = socketReader.readUTF();
                                            String[] choiceSplit = choice.split(". ");
                                            int choiceNum = Integer.parseInt(choiceSplit[0]);
                                            
                                            // Edit product (deletes, then adds with changes)
                                            if (choiceNum == 1) {
                                                String[] productsArray = (sellers.get(username).getProducts()).split(", ");
                                            
                                                String allProducts = "";
                                                for (String products : productsArray) {
                                                    allProducts = allProducts + products + ", ";
                                                }
                                                socketWriter.writeUTF(allProducts);
                                            
                                                String editProduct = socketReader.readUTF();
                                                String newProduct = socketReader.readUTF();
                                            
                                                String[] storesArrays = (sellers.get(username).getStores()).split(", ");
                                            
                                                String newAllStores = "";
                                                for (String store : storesArrays) {
                                                    newAllStores = newAllStores + store + ", ";
                                                }
                                                socketWriter.writeUTF(newAllStores);
                                            
                                                String newProductStore = socketReader.readUTF();
                                                boolean newStoreExists = storeExists(newProductStore);
                                                if (newStoreExists) {
                                                    socketWriter.writeUTF("yes");
                                                } else {
                                                    socketWriter.writeUTF("no");
                                                }
                                            
                                                if (newStoreExists) {
                                                    // Get all info for product, add product
                                                    String newDescription = socketReader.readUTF();
                                                    int newQuantity = Integer.parseInt(socketReader.readUTF());
                                                    double newPrice = Double.parseDouble(socketReader.readUTF());
                                            
                                                    sellers.get(username).deleteProduct(editProduct);
                                                    sellers.get(username).addProduct(newProduct, newProductStore, newDescription, newQuantity, newPrice);
                                                } else {
                                                    socketWriter.writeUTF("That store doesn't exist!");
                                                }
                                                // Deletes product
                                            } else if (choiceNum == 2) {
                                                String[] productsArray = (sellers.get(username).getProducts()).split(", ");
                                                String allProducts = "";
                                            
                                                for (String products : productsArray) {
                                                    allProducts = allProducts + products + ", ";
                                                }
                                                socketWriter.writeUTF(allProducts);
                                            
                                                String unwantedProduct = socketReader.readUTF();
                                                sellers.get(username).deleteProduct(unwantedProduct);
                                            }
                                            break;
                                        
                                        //Create store
                                        case 3:
                                            String storeName = socketReader.readUTF();
                                            (sellers.get(username)).addStore(storeName);
                                            break;
                                        //View Stores
                                        case 4:
                                            String everyStore = (sellers.get(username)).getStores();
                                            System.out.println(everyStore);
                                            socketWriter.writeUTF(everyStore);
                                            break;
                                        //View Seller Dashboard NEEDS TO BE CONVERTED TO I/O
                                        case 5:
                                            String sellerStores = (sellers.get(username)).getStores();
                                            String[] stores = sellerStores.split(", ");
                                            SellersDashboard sellersDashboard = new SellersDashboard(stores);
                                            sellersDashboard.viewDashboard();
                                            break;
                                        //Logout
                                        case 6:
                                        break;
                                        default:
                                    }
                                } while (selectionNum != 6);
        
                            } else {
                                JOptionPane.showMessageDialog(null, "Login failed!", "Error",
                                        JOptionPane.ERROR_MESSAGE);
                            }
                        } while (!loggedIn);
        
                        //User creation system
                        //Add new user authentication?
                    } else if (userNum == 3) {
                        boolean userTaken;
                        String username;
                        do {
                            username = JOptionPane.showInputDialog(null, "Please enter a username:",
                                    "Create Account", JOptionPane.QUESTION_MESSAGE);
        
                            BufferedReader br = new BufferedReader(new FileReader("users.txt"));
                            String line = br.readLine();
                            userTaken = false;
        
                            //Makes sure username is not taken
                            while (line != null) {
                                String[] splitLine = line.split(";");
                                if (splitLine[1].equals(username)) {
                                    userTaken = true;
                                    JOptionPane.showMessageDialog(null, "This username is taken!", "Error",
                                            JOptionPane.ERROR_MESSAGE);
                                }
                                line = br.readLine();
                            }
                        } while (userTaken);
        
                        String password = JOptionPane.showInputDialog(null, "Please enter a password:",
                                "Create Account", JOptionPane.QUESTION_MESSAGE);
                        String[] userTypeArray = {"1. Customer", "2. Seller"};
                        String userType = (String) JOptionPane.showInputDialog(null, "\"Are you a...",
                                "Create Account", JOptionPane.QUESTION_MESSAGE, null, userTypeArray, userTypeArray[0]);
                        int userTypeNum = Integer.parseInt(String.valueOf(userType.charAt(0)));
                        if (userTypeNum == 1) {
                            try {
                                addUser("Customer", username, password);
                            } catch (IOException e) {
                                e.printStackTrace();
                            }
                        } else if (userTypeNum == 2) {
                            try {
                                addUser("Seller", username, password);
                            } catch (IOException e) {
                                e.printStackTrace();
                            }
                        } else {
                        }
                    } else {
                        validUser[0] = false;
                    }
                } while (!validUser[0]);
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
            try {
                if (socketReader != null) {
                    socketReader.close();
                }
                if (socketWriter != null) {
                    socketWriter.close();
                }
                if (newSocket != null) {
                    newSocket.close();
                }
                server.stopThread(Thread.currentThread());
            } catch (IOException e) {
                e.printStackTrace();
            }
        }

        public void stopThread() {
            running = false;
        }

            //This will eventually print the marketplace for customers
            public static ArrayList<String> printMarket() throws IOException {
                ArrayList<String> market = new ArrayList<String>();
                BufferedReader br = new BufferedReader(new FileReader("products.txt"));
                String line = br.readLine();
        
                while (line != null) {
                    String[] splitLine = line.split(";");
        
                    if (Integer.parseInt(splitLine[4]) > 0) {
                        market.add(splitLine[2] + " | $" + splitLine[5] + " | " + splitLine[1]);
                    }
                    line = br.readLine();
                }
                return market;
            }
    
            public static String viewProduct(String product) throws IOException {
                String productView = "";
                BufferedReader br = new BufferedReader(new FileReader("products.txt"));
                String line = br.readLine();
        
                while (line != null) {
                    String[] splitLine = line.split(";");
                    if (splitLine[2].equals(product)) {
                        productView = splitLine[2] + "\nPrice: $" + splitLine[5] + "\nQuantity Available: " + splitLine[4] + "\nDescription: " + splitLine[3] + "\nStore: " + splitLine[1];
                    }
                    line = br.readLine();
                }
                return productView;
            }
    
            public static int checkQuantity(String product) throws IOException {
                int quantity = 0;
                BufferedReader br = new BufferedReader(new FileReader("products.txt"));
                String line = br.readLine();
        
                while (line != null) {
                    String[] splitLine = line.split(";");
                    if (splitLine[2].equals(product)) {
                        quantity = Integer.parseInt(splitLine[4]);
                    }
                    line = br.readLine();
                }
                return quantity;
            }
    
            public static void productPurchased(String product, String username, int quantity) throws IOException {
                BufferedReader br = new BufferedReader(new FileReader("products.txt"));
                String line = br.readLine();
                int newQuantity = 0;
        
                while (line != null) {
                    String[] splitLine = line.split(";");
                    if (splitLine[2].equals(product)) {
                        newQuantity = Integer.parseInt(splitLine[4]) - quantity;
                    }
                    if (newQuantity > 0 && !product.isEmpty()) {
                        (sellers.get(splitLine[0])).deleteProduct(product);
                        (sellers.get(splitLine[0])).addProduct(splitLine[2], splitLine[1], splitLine[3], newQuantity, Double.parseDouble(splitLine[5]));
                    } else if (newQuantity == 0) {
                        (sellers.get(splitLine[0])).deleteProduct(product);
                    }
                    line = br.readLine();
                }
            }
    
            public static String checkDescription(String product) throws IOException{
                String description = "";
                BufferedReader br = new BufferedReader(new FileReader("products.txt"));
                String line = br.readLine();
        
                while (line != null) {
                    String[] splitLine = line.split(";");
                    if (splitLine[2].equals(product)) {
                        description = splitLine[3];
                    }
                    line = br.readLine();
                }
                return description;
            }
    
            public static double checkPrice(String product) throws IOException{
                double price = 0;
                BufferedReader br = new BufferedReader(new FileReader("products.txt"));
                String line = br.readLine();
        
                while (line != null) {
                    String[] splitLine = line.split(";");
                    if (splitLine[2].equals(product)) {
                        price = Double.parseDouble(splitLine[5]);
                    }
                    line = br.readLine();
                }
                return price;
            }
    
            public static String checkStore(String product) throws IOException{
                String store = "";
                BufferedReader br = new BufferedReader(new FileReader("products.txt"));
                String line = br.readLine();
        
                while (line != null) {
                    String[] splitLine = line.split(";");
                    if (splitLine[2].equals(product)) {
                        store = splitLine[1];
                    }
                    line = br.readLine();
                }
                return store;
            }
    
            public static boolean storeExists(String store) throws IOException{
                boolean storeExists = false;
                BufferedReader br = new BufferedReader(new FileReader("stores.txt"));
                String line = br.readLine();
        
                while (line != null) {
                    String[] splitLine = line.split(";");
                    if (splitLine[1].equals(store)) {
                        storeExists = true;
                    }
                    line = br.readLine();
                }
                return storeExists;
            }
        
            //Creates new users and adds them to hashmaps
            //EVENTUALLY get rid of file system, authenticate with hashmaps instead
            public static void addUser(String userType, String username, String password) throws IOException {
                PrintWriter pw = new PrintWriter(new FileWriter("users.txt", true), true);
                pw.write(userType + ";" + username + ";" + password);
                pw.println();
                pw.close();
        
                //"put" adds them to the hashmaps, with the specified username,
                //and the specified user profile
                if (userType.equals("Seller")) {
                    Sellers newSeller = new Sellers(username, password);
                    sellers.put(username, newSeller);
                } else {
                    Customers newCustomer = new Customers(username, password);
                    customers.put(username, newCustomer);
                }
            }

            public static void populateHashMaps() throws IOException {
                BufferedReader br = new BufferedReader(new FileReader("users.txt"));
                String line = br.readLine();
        
                while (line != null) {
                    String[] splitLine = line.split(";");
        
                    if (splitLine[0].equals("Seller")) {
                        sellers.put(splitLine[1], new Sellers(splitLine[1], splitLine[2]));
                    } else {
                        customers.put(splitLine[1], new Customers(splitLine[1], splitLine[2]));
                    }
                    line = br.readLine();
                }
            }
}