Main.java

import java.util.*;
import java.io.*;

public class Main {
    public static void main(String[] args) {
        ArrayList<Grocery> storeList = new ArrayList<>();
        storeList.add(readStoreData("Walmart Demo Data.csv"));
        storeList.add(readStoreData("Costco Demo Data.csv"));
        storeList.add(readStoreData("Target Demo Data.csv"));
        User u = new User("Ethan");
        ArrayList<Grocery> closestStores = bestStores(storeList, u);
        TreeMap<String, Double> averageCost = new TreeMap<>();
        averageCost.put("apple", 1.29);
        averageCost.put("yogurt", 2.33);
        averageCost.put("cereal", 3.27);
        averageCost.put("ice cream", 5.52);
        Map<String, Grocery> stores = new TreeMap<>();
        stores.put("Walmart", storeList.get(0));
        stores.put("Costco", storeList.get(1));
        stores.put("Target", storeList.get(2));
        //command line input for array of products to search

        printSuggestions(closestStores, args, u, averageCost);
        makeTransactions("Transactions Demo Data.csv", u, averageCost,
                new Scanner(System.in), stores);
        System.out.println();
        printSuggestions(closestStores, args, u, averageCost);



    }

    public static ArrayList<Grocery> bestStores(ArrayList<Grocery> storeList, User u){
        TreeMap<Double, Grocery> groceryTreeMap = closeStores(storeList, u);
        ArrayList<Grocery> fiveBest = new ArrayList<>();
        int i = 0;
        for(Grocery a : groceryTreeMap.values()){
            if(i>=3) break;
            fiveBest.add(a);
            i++;
        }
        return fiveBest;
    }

    public static Grocery readStoreData(String name) {
        try(Reader r = new FileReader(name); BufferedReader br = new BufferedReader(r)) {
            String line = br.readLine();
            String[] a = line.split(",", Integer.MAX_VALUE);
            Point b = new Point(Double.parseDouble(a[1]), Double.parseDouble(a[2]));
            Grocery store = new Grocery(a[0], b);
            line = br.readLine();
            while (line != null) {
                a = line.split(",", Integer.MAX_VALUE);
                Product product = new Product(a[0], Double.parseDouble(a[1]), a[2], Double.parseDouble(a[3]),
                        Boolean.parseBoolean(a[4]), Boolean.parseBoolean(a[5]));
                store.addProduct(product);
                line = br.readLine();
            }
            return store;
        } catch (FileNotFoundException e) {
            System.err.println("File " + name + " not found");
        } catch (IOException e) {
            System.err.println("Error reading from file " + name);
        }
        return null;
    }

    public static TreeMap<Double, Grocery> closeStores(ArrayList<Grocery> storeList, User u){
        TreeMap<Double, Grocery> groceryTreeMap = new TreeMap<>();
        double x = u.getLoc().getLat();
        double y = u.getLoc().getLon();
        for(Grocery b : storeList){
            double dist = Math.pow(b.getLoc().getLat()-x,2)+Math.pow(b.getLoc().getLon()-y, 2);
            groceryTreeMap.put(Math.sqrt(dist), b);
        }
        return groceryTreeMap;
    }

    public static void printSuggestions(ArrayList<Grocery> storeList, String [] products, User u, TreeMap<String, Double> averageCost){
        TreeMap<Double, Grocery> ranking= new TreeMap<>();
        ArrayList<Double> costs = new ArrayList<>();
        for(String a : products){
            costs.add(averageCost.get(findCategory(a, averageCost)));
        }
        for(Grocery store: storeList){
            Grocery returnedStore = new Grocery(store.getName(), store.getLoc());
            int i = 0;
            for(String product: products){
                returnedStore.addProduct(store.bestProduct(product, u, costs.get(i)));
                i++;
            }
            ranking.put(store.storeScore(products, u, costs), returnedStore);
        }
        for (Map.Entry<Double, Grocery> entry : ranking.entrySet()) {
            System.out.println(entry.getValue().getName()+":");
            for(Product a : entry.getValue().getProducts()){
                System.out.println("\t"+a.getName()+" | "+a.getOrganic());
            }
        }
    }

    //gather relevant products (call to Grocery) -> ArrayList<Product> might not even need this one
    // accept list of search and send to each grocer (for each loop) ->
    //adds confirmed purchase to each user class -> void
    // creates a tree map to sort products by score -> treemap

    public static void makeTransactions(String fileName, User u, Map<String, Double>
            averageCost, Scanner sc, Map<String, Grocery> stores) {
        try (Reader r = new FileReader(fileName); BufferedReader br = new BufferedReader(r)) {
            while (true) {
                System.out.println("\nMake transactions: ");
                String input = sc.nextLine().trim();
                String [] words = input.split(" ");
                if (words.length > 0) {
                    break;
                }
            }
            String line = br.readLine();
            while (line != null) {
                String[] transaction = line.split(",");
                // System.out.println("Bought: " + transaction[0] + " from " + transaction[1]);
                Grocery g = stores.get(transaction[1]);
                List<Product> products = g.search(transaction[0]);
                for (Product p : products) {
                    u.addTransaction(p, averageCost.get(transaction[2]));
                    System.out.println("Bought: " + p.getName() + " from " + transaction[1] +" | "+ p.getOrganic());
                }
                line = br.readLine();
            }
        }
        catch (IOException e) {

        }
    }

    public static String findCategory(String name, TreeMap<String, Double> averageCost){
        for(String a : name.split(" ")){
            if (averageCost.containsKey(a)) return a;
        }
        return "";
    }

}