src/main/java/net/pterodactylus/sone/core/SoneRescuer.java

   1 /*
   2  * Sone - SoneRescuer.java - Copyright © 2011–2012 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 }