enums addition

dsy05 · dsy05 · commit bb75ecfa9b82 · 2024-02-03T00:35:46.000-06: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.validation.Errors;
 import org.springframework.web.bind.annotation.*;
@@ -27,6 +28,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 an array of the 4 types of enums created
         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
@@ -22,11 +22,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() {
@@ -62,6 +65,14 @@ public void setContactEmail(String contactEmail) {
         this.contactEmail = contactEmail;
     }
 
+    public EventType getType() {
+        return type;
+    }
+
+    public void setType(EventType type) {
+        this.type = type;
+    }
+
     @Override
     public String toString() {
         return name;
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,18 @@
+package org.launchcode.codingevents.models;
+
+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
@@ -24,6 +24,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 Event" class="btn btn-success">
     </div>
diff --git a/src/main/resources/templates/events/index.html b/src/main/resources/templates/events/index.html
@@ -16,13 +16,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>
 
