🎨 Add “imageUrl” attribute to Movie

[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 42480f9..5e831d8 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,20 +16,32 @@ import java.util.stream.Collectors;
  */
 public class Movie {
 
+       @JsonProperty
        private final String name;
+
+       @JsonProperty
+       private final String imageUrl;
+
+       @JsonProperty
        private final List<Performance> performances = new ArrayList<>();
 
-       public static final Predicate<Movie> withPerformances = movie -> !movie.getPerformances().isEmpty();
-       public static final Comparator<Movie> byName = (leftMovie, rightMovie) -> leftMovie.getName().compareToIgnoreCase(rightMovie.getName());
+       public Movie() {
+               this("", "");
+       }
 
-       public Movie(String name) {
+       public Movie(String name, String imageUrl) {
                this.name = name;
+               this.imageUrl = imageUrl;
        }
 
        public String getName() {
                return name;
        }
 
+       public String getImageUrl() {
+               return imageUrl;
+       }
+
        public List<Performance> getPerformances() {
                return performances;
        }
@@ -39,7 +52,20 @@ public class Movie {
 
        @Override
        public String toString() {
-               return format("%s (%s)", name, performances.stream().map(link -> String.format("%s: %s", link.getTime(), link.getLink())).collect(Collectors.joining(", ")));
+               return format("%s (%s, %s)", name, imageUrl, performances.stream().map(link -> String.format("%s: %s", link.getTime(), link.getLink())).collect(Collectors.joining(", ")));
+       }
+
+       @Override
+       public int hashCode() {
+               return Objects.hash(name, imageUrl, 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(imageUrl, movie.imageUrl) && Objects.equals(performances, movie.performances);
        }
 
 }