Implement hashCode() and equals().
[Sone.git] / src / main / java / net / pterodactylus / sone / core / SoneRescuer.java
1 /*
2  * Sone - SoneRescuer.java - Copyright © 2011–2013 David Roden
3  *
4  * This program is free software: you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation, either version 3 of the License, or
7  * (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
16  */
17
18 package net.pterodactylus.sone.core;
19
20 import net.pterodactylus.sone.data.Sone;
21 import net.pterodactylus.util.service.AbstractService;
22 import freenet.keys.FreenetURI;
23
24 /**
25  * The Sone rescuer downloads older editions of a Sone and updates the currently
26  * stored Sone with it.
27  *
28  * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
29  */
30 public class SoneRescuer extends AbstractService {
31
32         /** The core. */
33         private final Core core;
34
35         /** The Sone downloader. */
36         private final SoneDownloader soneDownloader;
37
38         /** The Sone being rescued. */
39         private final Sone sone;
40
41         /** Whether the rescuer is currently fetching a Sone. */
42         private volatile boolean fetching;
43
44         /** The currently tried edition. */
45         private volatile long currentEdition;
46
47         /** Whether the last fetch was successful. */
48         private volatile boolean lastFetchSuccessful = true;
49
50         /**
51          * Creates a new Sone rescuer.
52          *
53          * @param core
54          *            The core
55          * @param soneDownloader
56          *            The Sone downloader
57          * @param sone
58          *            The Sone to rescue
59          */
60         public SoneRescuer(Core core, SoneDownloader soneDownloader, Sone sone) {
61                 super("Sone Rescuer for " + sone.getName());
62                 this.core = core;
63                 this.soneDownloader = soneDownloader;
64                 this.sone = sone;
65                 currentEdition = sone.getRequestUri().getEdition();
66         }
67
68         //
69         // ACCESSORS
70         //
71
72         /**
73          * Returns whether the Sone rescuer is currently fetching a Sone.
74          *
75          * @return {@code true} if the Sone rescuer is currently fetching a Sone
76          */
77         public boolean isFetching() {
78                 return fetching;
79         }
80
81         /**
82          * Returns the edition that is currently being downloaded.
83          *
84          * @return The edition that is currently being downloaded
85          */
86         public long getCurrentEdition() {
87                 return currentEdition;
88         }
89
90         /**
91          * Returns whether the Sone rescuer can download a next edition.
92          *
93          * @return {@code true} if the Sone rescuer can download a next edition,
94          *         {@code false} if the last edition was already tried
95          */
96         public boolean hasNextEdition() {
97                 return currentEdition > 0;
98         }
99
100         /**
101          * Returns the next edition the Sone rescuer can download.
102          *
103          * @return The next edition the Sone rescuer can download
104          */
105         public long getNextEdition() {
106                 return currentEdition - 1;
107         }
108
109         /**
110          * Sets the edition to rescue.
111          *
112          * @param edition
113          *            The edition to rescue
114          * @return This Sone rescuer
115          */
116         public SoneRescuer setEdition(long edition) {
117                 currentEdition = edition;
118                 return this;
119         }
120
121         /**
122          * Sets whether the last fetch was successful.
123          *
124          * @return {@code true} if the last fetch was successful, {@code false}
125          *         otherwise
126          */
127         public boolean isLastFetchSuccessful() {
128                 return lastFetchSuccessful;
129         }
130
131         //
132         // ACTIONS
133         //
134
135         /**
136          * Starts the next fetch. If you want to fetch a different edition than “the
137          * next older one,” remember to call {@link #setEdition(long)} before
138          * calling this method.
139          */
140         public void startNextFetch() {
141                 fetching = true;
142                 notifySyncObject();
143         }
144
145         //
146         // SERVICE METHODS
147         //
148
149         /**
150          * {@inheritDoc}
151          */
152         @Override
153         protected void serviceRun() {
154                 while (!shouldStop()) {
155                         while (!shouldStop() && !fetching) {
156                                 sleep();
157                         }
158                         if (fetching) {
159                                 core.lockSone(sone);
160                                 FreenetURI soneUri = sone.getRequestUri().setKeyType("SSK").setDocName("Sone-" + currentEdition).setMetaString(new String[] { "sone.xml" });
161                                 System.out.println("URI: " + soneUri);
162                                 Sone fetchedSone = soneDownloader.fetchSone(sone, soneUri, true);
163                                 System.out.println("Sone: " + fetchedSone);
164                                 lastFetchSuccessful = (fetchedSone != null);
165                                 if (lastFetchSuccessful) {
166                                         core.updateSone(fetchedSone, true);
167                                 }
168                                 fetching = false;
169                         }
170                 }
171         }
172
173 }