ElectronicStore.java

import java.util.*;
//MODEL
public class ElectronicStore {
    private final String name;
    private final ArrayList<Product> electronicProduct;
    private int soldCount = 0;
    private double cartPrice = 0;
    private final HashMap<Product, Integer> cartList;

    public ElectronicStore(String InitName){
        name = InitName;
        electronicProduct = new ArrayList<>();
        cartList = new HashMap<>();
    }
    //creating necessary get method
    public String getName() {
        return name;
    }
    public int getSoldCount(){return soldCount;}
    public double getCartPrice(){return cartPrice;}

    public void addProduct(Product p) {
        electronicProduct.add(p);
    }
    //sell product also updates the cart and keeps track of number of sales
    public void sellProducts(){
        for(Product p: cartList.keySet()){
            p.sellUnits(cartList.get(p));
            p.setCartQuantity(0);
        }
        cartList.clear();
        cartPrice = 0;
        soldCount++;
    }

    public double getRevenue(){
        double totalRevenue = 0.0;
        for(Product p: electronicProduct){
            totalRevenue += p.getTotalRevenue();
        }
        return totalRevenue;
    }

    public static ElectronicStore createStore() {
        ElectronicStore store1 = new ElectronicStore("Watts Up Electronics");
        Desktop d1 = new Desktop(100, 10, 3.0, 16, false, 250, "Compact");
        Desktop d2 = new Desktop(200, 10, 4.0, 32, true, 500, "Server");
        Laptop l1 = new Laptop(150, 10, 2.5, 16, true, 250, 15);
        Laptop l2 = new Laptop(250, 10, 3.5, 24, true, 500, 16);
        Fridge f1 = new Fridge(500, 10, 250, "White", "Sub Zero", false);
        Fridge f2 = new Fridge(750, 10, 125, "Stainless Steel", "Sub Zero", true);
        ToasterOven t1 = new ToasterOven(25, 10, 50, "Black", "Dany", false);
        ToasterOven t2 = new ToasterOven(75, 10, 50, "Silver", "Toasty", true);
        store1.addProduct(d1);
        store1.addProduct(d2);
        store1.addProduct(l1);
        store1.addProduct(l2);
        store1.addProduct(f1);
        store1.addProduct(f2);
        store1.addProduct(t1);
        store1.addProduct(t2);
        return store1;
    }
    //returns a list based on the amount of product in the cart and the actual amount of product
    public ArrayList<Product> getStockList() {
        ArrayList<Product>list = new ArrayList<>();
        for (Product p: electronicProduct){
            if (p.getCartQuantity() < p.getStockQuantity()) {
                list.add(p);
            }
        }
        return list;
    }
    //cart list contains product and amount of products
    public void CartList(Product p) {
        if(p.getCartQuantity() < p.getStockQuantity()){
            if(!cartList.containsKey(p)){
                cartList.put(p, 1);
                p.setCartQuantity(1);
            }else{
            cartList.put(p, cartList.get(p)+1);
            p.setCartQuantity(p.getCartQuantity()+1);
            }
            cartPrice += p.getPrice();
        }
    }
    //method creates what is displayed in the cart based on the actual product
    public String CartOutput(Product p){
        return "" + cartList.get(p) + "X" + p;
    }
    //returns a list of items to display on the cart list
    public ArrayList<String> CartArray(){
        ArrayList<String> cartString = new ArrayList<>();
        for(Product p:cartList.keySet()){
            cartString.add(CartOutput(p));
        }
        return cartString;
    }

    public void RemoveItem(String pInput){
        for(Product p: cartList.keySet()){
            if(pInput.equals(CartOutput(p))){
                cartList.put(p, cartList.get(p)-1);
                p.setCartQuantity(p.getCartQuantity()-1);
                cartPrice -= p.getPrice();
                if(cartList.get(p) == 0){
                    cartList.remove(p,cartList.get(p));
                }
            }
        }
    }

    public double getAverage() {
        double rev = getRevenue();
        return rev / soldCount;
    }
    //sorts electronic products based on amount sold and returns an array list of the top 3
    public ArrayList<Product> getPopularProducts() {
        electronicProduct.sort((p1, p2) -> p2.getSoldQuantity() - p1.getSoldQuantity());

        int i =0;
        ArrayList<Product> popular = new ArrayList<>();
        for(Product p: electronicProduct){
            if(i<3){
                popular.add(p);
                i++;
            }
        }

        return popular;
    }
}