X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;ds=sidebyside;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Frhynodge%2Fstates%2FEpisodeState.java;h=7ed3dcecd9ec3d725be693e74af20b308c3b0e9e;hb=6ea5acf682dec9e234df216e61a60bb23cdd5d0c;hp=c604634a105d14f79f63544bbd9b6a9b6cd488a1;hpb=992e01bce6d9818d4e94460a30137e2f39033649;p=rhynodge.git

diff --git a/src/main/java/net/pterodactylus/rhynodge/states/EpisodeState.java b/src/main/java/net/pterodactylus/rhynodge/states/EpisodeState.java
index c604634..7ed3dce 100644
--- a/src/main/java/net/pterodactylus/rhynodge/states/EpisodeState.java
+++ b/src/main/java/net/pterodactylus/rhynodge/states/EpisodeState.java
@@ -29,6 +29,7 @@ import net.pterodactylus.rhynodge.states.EpisodeState.Episode;
 import net.pterodactylus.rhynodge.states.TorrentState.TorrentFile;
 
 import com.fasterxml.jackson.annotation.JsonProperty;
+import com.google.common.base.Function;
 
 /**
  * {@link State} implementation that stores episodes of TV shows, parsed via
@@ -100,6 +101,15 @@ public class EpisodeState extends AbstractState implements Iterable<Episode> {
 	 */
 	public static class Episode implements Comparable<Episode>, Iterable<TorrentFile> {
 
+		/** Function to extract the season of an episode. */
+		public static final Function<Episode, Integer> BY_SEASON = new Function<Episode, Integer>() {
+
+			@Override
+			public Integer apply(Episode episode) {
+				return episode.season();
+			}
+		};
+
 		/** The season of the episode. */
 		@JsonProperty
 		private final int season;