🚧 Make Movie and Performance JSON-serializable

[rhynodge.git] / src / main / java / net / pterodactylus / rhynodge / filters / webpages / savoy / Movie.java

diff --git a/src/main/java/net/pterodactylus/rhynodge/filters/webpages/savoy/Movie.java b/src/main/java/net/pterodactylus/rhynodge/filters/webpages/savoy/Movie.java
index 562c392..0844b7c 100644 (file)
--- a/src/main/java/net/pterodactylus/rhynodge/filters/webpages/savoy/Movie.java
+++ b/src/main/java/net/pterodactylus/rhynodge/filters/webpages/savoy/Movie.java
@@ -1,13 +1,14 @@
 package net.pterodactylus.rhynodge.filters.webpages.savoy;
 
-import static java.lang.String.format;
+import com.fasterxml.jackson.annotation.JsonProperty;
 
 import java.util.ArrayList;
-import java.util.Comparator;
 import java.util.List;
-import java.util.function.Predicate;
+import java.util.Objects;
 import java.util.stream.Collectors;
 
+import static java.lang.String.format;
+
 /**
  * Information about a movie.
  *
@@ -15,11 +16,15 @@ import java.util.stream.Collectors;
  */
 public class Movie {
 
+       @JsonProperty
        private final String name;
-       private final List<TicketLink> ticketLinks = new ArrayList<>();
 
-       public static final Predicate<Movie> withPresentations = movie -> !movie.getTicketLinks().isEmpty();
-       public static final Comparator<Movie> byName = (leftMovie, rightMovie) -> leftMovie.getName().compareToIgnoreCase(rightMovie.getName());
+       @JsonProperty
+       private final List<Performance> performances = new ArrayList<>();
+
+       public Movie() {
+               this("");
+       }
 
        public Movie(String name) {
                this.name = name;
@@ -29,17 +34,30 @@ public class Movie {
                return name;
        }
 
-       public List<TicketLink> getTicketLinks() {
-               return ticketLinks;
+       public List<Performance> getPerformances() {
+               return performances;
        }
 
-       public void addTicketLink(TicketLink ticketLink) {
-               ticketLinks.add(ticketLink);
+       public void addPerformance(Performance performance) {
+               performances.add(performance);
        }
 
        @Override
        public String toString() {
-               return format("%s (%s)", name, ticketLinks.stream().map(link -> String.format("%s: %s", link.getPresentationTime(), link.getLink())).collect(Collectors.joining(", ")));
+               return format("%s (%s)", name, performances.stream().map(link -> String.format("%s: %s", link.getTime(), link.getLink())).collect(Collectors.joining(", ")));
+       }
+
+       @Override
+       public int hashCode() {
+               return Objects.hash(name, performances);
+       }
+
+       @Override
+       public boolean equals(Object o) {
+               if (this == o) return true;
+               if (o == null || getClass() != o.getClass()) return false;
+               Movie movie = (Movie) o;
+               return Objects.equals(name, movie.name) && Objects.equals(performances, movie.performances);
        }
 
 }