enums branch complete

jwoolbright23 · jwoolbright23 · commit 8abc9201a306 · 2023-07-31T13:47:19.000-05:00
diff --git a/src/main/java/org/launchcode/codingevents/controllers/EventController.java b/src/main/java/org/launchcode/codingevents/controllers/EventController.java
@@ -3,6 +3,7 @@
 import jakarta.validation.Valid;
 import org.launchcode.codingevents.data.EventData;
 import org.launchcode.codingevents.models.Event;
+import org.launchcode.codingevents.models.EventType;
 import org.springframework.stereotype.Controller;
 import org.springframework.web.bind.annotation.*;
 import org.springframework.ui.Model;
@@ -29,6 +30,7 @@ public String displayAllEvents(Model model) {
     public String displayCreateEventForm(Model model) {
         model.addAttribute("title", "Create Event");
         model.addAttribute(new Event());
+        model.addAttribute("types", EventType.values());
         return "events/create";
     }
 
diff --git a/src/main/java/org/launchcode/codingevents/models/Event.java b/src/main/java/org/launchcode/codingevents/models/Event.java
@@ -25,11 +25,14 @@ public class Event {
     @Email(message = "Invalid email. Try again.")
     private String contactEmail;
 
-    public Event(String name, String description, String contactEmail) {
+    private EventType type;
+
+    public Event(String name, String description, String contactEmail, EventType type) {
         this();
         this.name = name;
         this.description = description;
         this.contactEmail = contactEmail;
+        this.type = type;
     }
 
     public Event() {
@@ -61,6 +64,10 @@ public void setContactEmail(String contactEmail) {
         this.contactEmail = contactEmail;
     }
 
+    public EventType getType() { return type; }
+
+    public void setType(EventType type) { this.type = type; }
+
     public int getId() {
         return id;
     }
diff --git a/src/main/java/org/launchcode/codingevents/models/EventType.java b/src/main/java/org/launchcode/codingevents/models/EventType.java
@@ -0,0 +1,22 @@
+package org.launchcode.codingevents.models;
+
+/**
+ * Created by Chris Bay
+ */
+public enum EventType {
+
+    CONFERENCE("Conference"),
+    MEETUP("Meetup"),
+    WORKSHOP("Workshop"),
+    SOCIAL("Social");
+
+    private final String displayName;
+
+    EventType(String displayName) {
+        this.displayName = displayName;
+    }
+
+    public String getDisplayName() {
+        return displayName;
+    }
+}
diff --git a/src/main/resources/templates/events/create.html b/src/main/resources/templates/events/create.html
@@ -26,6 +26,16 @@
         </label>
         <p class="error" th:errors="${event.contactEmail}"></p>
     </div>
+    <div class="form-group">
+        <label>Type
+            <select th:field="${event.type}">
+                <option th:each="type : ${types}"
+                        th:value="${type}"
+                        th:text="${type.displayName}"
+                ></option>
+            </select>
+        </label>
+    </div>
     <div class="form-group">
         <input type="submit" value="Create" class="btn btn-success">
     </div>
diff --git a/src/main/resources/templates/events/index.html b/src/main/resources/templates/events/index.html
@@ -12,13 +12,15 @@
         <th>Name</th>
         <th>Description</th>
         <th>Contact Email</th>
+        <th>Type</th>
     </tr>
     </thead>
     <tr th:each="event : ${events}">
         <td th:text="${event.id}"></td>
         <td th:text="${event.name}"></td>
         <td th:text="${event.description}"></td>
         <td th:text="${event.contactEmail}"></td>
+        <td th:text="${event.type.displayName}"></td>
     </tr>
 </table>
 
