code

Author: DanAlberts

.idea/workspace.xml

@@ -28,4 +28,19 @@
     <maven.compiler.target>${java.version}</maven.compiler.target>
   </properties>
 
+  <build>
+    <plugins>
+      <plugin>
+        <groupId>org.flywaydb</groupId>
+        <artifactId>flyway-maven-plugin</artifactId>
+        <version>5.2.4</version>
+        <configuration>
+          <url>jdbc:postgresql://localhost:5432/wishlist</url>
+          <user>postgres</user>
+          <password>password</password>
+        </configuration>
+      </plugin>
+    </plugins>
+  </build>
+
 </project>

pom.xml

@@ -0,0 +1,50 @@
+import java.util.ArrayList;
+import java.util.List;
+import javax.persistence.Column;
+import javax.persistence.Entity;
+import javax.persistence.GeneratedValue;
+import javax.persistence.GenerationType;
+import javax.persistence.Id;
+import javax.persistence.OneToMany;
+import javax.persistence.SequenceGenerator;
+import javax.persistence.Table;
+
+@Entity
+@Table(name = "categories")
+public class Category {
+  @Id
+  @SequenceGenerator(name = "category_generator", sequenceName = "categories_id_seq", allocationSize = 1)
+  @GeneratedValue(strategy = GenerationType.SEQUENCE, generator = "category_generator")
+  @Column(name = "id", nullable = false, unique = true)
+  private Long id;
+
+  @Column(name = "name", nullable = false, unique = true)
+  private String name;
+
+  @OneToMany(mappedBy = "category")
+  private List<Product> products = new ArrayList<>();
+
+  public Long getId() {
+    return id;
+  }
+
+  public void setId(Long id) {
+    this.id = id;
+  }
+
+  public String getName() {
+    return name;
+  }
+
+  public void setName(String name) {
+    this.name = name;
+  }
+
+  public List<Product> getProducts() {
+    return products;
+  }
+
+  public void setProducts(List<Product> products) {
+    this.products = products;
+  }
+}

src/main/java/Category.java

@@ -4,29 +4,31 @@
 import javax.persistence.EntityManager;
 import javax.persistence.EntityManagerFactory;
 import javax.persistence.Persistence;
+import javax.persistence.TypedQuery;
 
 
 public class MainMenu {
+
   private static Scanner input = new Scanner(System.in);
   private static EntityManagerFactory emf;
   private static EntityManager em;
 
   public static void main(String[] args) throws IOException {
     menu();
   }
-
   //Create a menu with options
   public static void menu(){
     emf = Persistence.createEntityManagerFactory("com.launchacademy.wishlist");
     em = emf.createEntityManager();
-    String[] options = {". Add a new product", ". View wishlist", ". Exit"};
+    String[] options = {". Add a new item", ". View wishlist", ". Exit"};
     while(true){
-      for (int i = 0; i < options.length; i++){
+      for (int i = 0 ; i < options.length; i++){
         System.out.println((i + 1) + options[i]);
       }
-      System.out.println("CHOOSE a number HUMAN");
+      System.out.println("CHOOSE A NUMBER HUMAN!");
       int choice = input.nextInt();
-      if( choice == options.length){
+      if (choice == options.length){
+        System.out.println("Goodbye");
         em.close();
         emf.close();
         System.exit(0);
@@ -40,28 +42,78 @@ else if (choice == 2){
     }
   }
 
+
   //Add a product
-  private static void addProduct(){
+  public static void addProduct() {
+    String name = "";
+    String price = "";
+    String url = "";
+    String category = "";
+
+    System.out.println("Name of product");
+    Scanner userInputName = new Scanner(System.in);
+    name = userInputName.nextLine();
+
+    System.out.println("Price of product");
+    Scanner userInputPrice = new Scanner(System.in);
+    price = userInputPrice.nextLine();
+    Float floatPrice = Float.parseFloat(price);
+
+    System.out.println("Url to product");
+    Scanner userInputUrl = new Scanner(System.in);
+    url = userInputUrl.nextLine();
+
+    System.out.println("Would you like to add a category? (type 'none' for no");
+    Scanner inputCategory = new Scanner(System.in);
+    category = inputCategory.nextLine();
+
+    Product product = new Product();
+    Category newCategory = new Category();
+
+    TypedQuery<Category> categoryQuery = em.createQuery("SELECT c FROM Category c", Category.class);
+    List<Category> categories = categoryQuery.getResultList();
+
     try{
-      Product newProduct = new Product();
-      System.out.println("What product would you like to add?");
-      newProduct.setName(input.next());
-      System.out.println("What is the product's price?");
-      newProduct.setPrice(input.nextFloat());
-      System.out.println("What is the product's url?");
-      newProduct.setUrl(input.next());
+
+      newCategory.setName(category);
+      em.getTransaction().begin();
+      em.persist(newCategory);
+      em.getTransaction().commit();
+
+      em.getTransaction().begin();
+      product.setName(name);
+      product.setPrice(floatPrice);
+      product.setUrl(url);
+      product.setCategory(newCategory);
+
+      em.persist(product);
+      em.getTransaction().commit();
+
+    } catch (Exception ex){
+      System.out.println("Duplicate Category Name Error");
       em.getTransaction().begin();
-      em.persist(newProduct);
+      product.setName(name);
+      product.setPrice(floatPrice);
+      product.setUrl(url);
+      for (Category eachCategory : categories){
+        if(category.equals(eachCategory.getName())){
+          product.setCategory(eachCategory);
+        }
+      }
+      em.persist(product);
       em.getTransaction().commit();
-    } finally {
     }
   }
 
   //List all products in wishlist
   private static void listWishes(){
-    List<Product> wishes = em.createQuery("SELECT p FROM Product p", Product.class).getResultList();
-    for (Product product : wishes){
-      System.out.println(product.getName() + " " + product.getPrice() + " " + product.getUrl());
+    TypedQuery<Product> query = em.createQuery("SELECT p FROM Product p", Product.class);
+    List<Product> products = query.getResultList();
+
+    for (Product product : products){
+      Category eachCategory = product.getCategory();
+      System.out.println("Product: " + product.getName() + "\n" + "Price: " + product.getPrice()
+      + "\n" + "Url: " + product.getUrl() + "\n" + " Category: " + eachCategory.getName() + "\n");
     }
   }
 }