Show new files in green and bold.

[rhynodge.git] / src / main / java / net / pterodactylus / rhynodge / filters / EpisodeFilter.java

diff --git a/src/main/java/net/pterodactylus/rhynodge/filters/EpisodeFilter.java b/src/main/java/net/pterodactylus/rhynodge/filters/EpisodeFilter.java
index 91fe098..7709d9e 100644 (file)
--- a/src/main/java/net/pterodactylus/rhynodge/filters/EpisodeFilter.java
+++ b/src/main/java/net/pterodactylus/rhynodge/filters/EpisodeFilter.java
@@ -19,7 +19,8 @@ package net.pterodactylus.rhynodge.filters;
 
 import static com.google.common.base.Preconditions.checkState;
 
-import java.util.LinkedHashMap;
+import java.util.HashMap;
+import java.util.Map;
 import java.util.regex.Matcher;
 import java.util.regex.Pattern;
 
@@ -57,13 +58,15 @@ public class EpisodeFilter implements Filter {
                checkState(state instanceof TorrentState, "state is not a TorrentState but a %s!", state.getClass());
 
                TorrentState torrentState = (TorrentState) state;
-               LinkedHashMap<Episode, Episode> episodes = new LinkedHashMap<Episode, Episode>();
+               Map<Episode, Episode> episodes = new HashMap<Episode, Episode>();
                for (TorrentFile torrentFile : torrentState) {
                        Episode episode = extractEpisode(torrentFile);
                        if (episode == null) {
                                continue;
                        }
-                       episodes.put(episode, episode);
+                       if (!episodes.containsKey(episode)) {
+                               episodes.put(episode, episode);
+                       }
                        episode = episodes.get(episode);
                        episode.addTorrentFile(torrentFile);
                }