Ignore chains and states default directories.

[rhynodge.git] / src / main / java / net / pterodactylus / reactor / states / TorrentState.java

diff --git a/src/main/java/net/pterodactylus/reactor/states/TorrentState.java b/src/main/java/net/pterodactylus/reactor/states/TorrentState.java
index 915559a..f3b3546 100644 (file)
--- a/src/main/java/net/pterodactylus/reactor/states/TorrentState.java
+++ b/src/main/java/net/pterodactylus/reactor/states/TorrentState.java
@@ -17,12 +17,17 @@
 
 package net.pterodactylus.reactor.states;
 
+import java.nio.charset.Charset;
 import java.util.Iterator;
 import java.util.List;
 
 import net.pterodactylus.reactor.State;
 import net.pterodactylus.reactor.states.TorrentState.TorrentFile;
 
+import org.apache.http.NameValuePair;
+import org.apache.http.client.utils.URLEncodedUtils;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
 import com.google.common.collect.Lists;
 
 /**
@@ -34,7 +39,8 @@ import com.google.common.collect.Lists;
 public class TorrentState extends AbstractState implements Iterable<TorrentFile> {
 
        /** The torrent files. */
-       private final List<TorrentFile> files = Lists.newArrayList();
+       @JsonProperty
+       private List<TorrentFile> files = Lists.newArrayList();
 
        //
        // ACCESSORS
@@ -84,17 +90,41 @@ public class TorrentState extends AbstractState implements Iterable<TorrentFile>
        public static class TorrentFile {
 
                /** The name of the file. */
+               @JsonProperty
                private final String name;
 
                /** The size of the file. */
+               @JsonProperty
                private final String size;
 
                /** The magnet URI of the file. */
+               @JsonProperty
                private final String magnetUri;
 
                /** The download URI of the file. */
+               @JsonProperty
                private final String downloadUri;
 
+               /** The number of files in this torrent. */
+               @JsonProperty
+               private final int fileCount;
+
+               /** The number of seeds connected to this torrent. */
+               @JsonProperty
+               private final int seedCount;
+
+               /** The number of leechers connected to this torrent. */
+               @JsonProperty
+               private final int leechCount;
+
+               /**
+                * No-arg constructor for deserialization.
+                */
+               @SuppressWarnings("unused")
+               private TorrentFile() {
+                       this(null, null, null, null, 0, 0, 0);
+               }
+
                /**
                 * Creates a new torrent file.
                 *
@@ -106,12 +136,21 @@ public class TorrentState extends AbstractState implements Iterable<TorrentFile>
                 *            The magnet URI of the file
                 * @param downloadUri
                 *            The download URI of the file
+                * @param fileCount
+                *            The number of files
+                * @param seedCount
+                *            The number of connected seeds
+                * @param leechCount
+                *            The number of connected leechers
                 */
-               public TorrentFile(String name, String size, String magnetUri, String downloadUri) {
+               public TorrentFile(String name, String size, String magnetUri, String downloadUri, int fileCount, int seedCount, int leechCount) {
                        this.name = name;
                        this.size = size;
                        this.magnetUri = magnetUri;
                        this.downloadUri = downloadUri;
+                       this.fileCount = fileCount;
+                       this.seedCount = seedCount;
+                       this.leechCount = leechCount;
                }
 
                //
@@ -155,6 +194,78 @@ public class TorrentState extends AbstractState implements Iterable<TorrentFile>
                        return downloadUri;
                }
 
+               /**
+                * Returns the number of files in this torrent.
+                *
+                * @return The number of files in this torrent
+                */
+               public int fileCount() {
+                       return fileCount;
+               }
+
+               /**
+                * Returns the number of seeds connected to this torrent.
+                *
+                * @return The number of connected seeds
+                */
+               public int seedCount() {
+                       return seedCount;
+               }
+
+               /**
+                * Returns the number of leechers connected to this torrent.
+                *
+                * @return The number of connected leechers
+                */
+               public int leechCount() {
+                       return leechCount;
+               }
+
+               //
+               // PRIVATE METHODS
+               //
+
+               /**
+                * Generates an ID for this file. If a {@link #magnetUri} is set, an ID
+                * is {@link #extractId(String) extracted} from it. Otherwise the magnet
+                * URI is used. If the {@link #magnetUri} is not set, the
+                * {@link #downloadUri} is used. If that is not set either, the name of
+                * the file is returned.
+                *
+                * @return The generated ID
+                */
+               private String generateId() {
+                       if (magnetUri != null) {
+                               String id = extractId(magnetUri);
+                               if (id != null) {
+                                       return id;
+                               }
+                               return magnetUri;
+                       }
+                       return (downloadUri != null) ? downloadUri : name;
+               }
+
+               //
+               // STATIC METHODS
+               //
+
+               /**
+                * Tries to extract the “exact target” of a magnet URI.
+                *
+                * @param magnetUri
+                *            The magnet URI to extract the “xt” from
+                * @return The extracted ID, or {@code null} if no ID could be found
+                */
+               private static String extractId(String magnetUri) {
+                       List<NameValuePair> parameters = URLEncodedUtils.parse(magnetUri.substring("magnet:?".length()), Charset.forName("UTF-8"));
+                       for (NameValuePair parameter : parameters) {
+                               if (parameter.getName().equals("xt")) {
+                                       return parameter.getValue();
+                               }
+                       }
+                       return null;
+               }
+
                //
                // OBJECT METHODS
                //
@@ -163,6 +274,28 @@ public class TorrentState extends AbstractState implements Iterable<TorrentFile>
                 * {@inheritDoc}
                 */
                @Override
+               public int hashCode() {
+                       return (generateId() != null) ? generateId().hashCode() : 0;
+               }
+
+               /**
+                * {@inheritDoc}
+                */
+               @Override
+               public boolean equals(Object object) {
+                       if (!(object instanceof TorrentFile)) {
+                               return false;
+                       }
+                       if (generateId() != null) {
+                               return generateId().equals(((TorrentFile) object).generateId());
+                       }
+                       return false;
+               }
+
+               /**
+                * {@inheritDoc}
+                */
+               @Override
                public String toString() {
                        return String.format("%s(%s,%s,%s)", name(), size(), magnetUri(), downloadUri());
                }