Eric Kok
12 years ago
5 changed files with 311 additions and 18 deletions
Binary file not shown.
@ -0,0 +1,276 @@
@@ -0,0 +1,276 @@
|
||||
/* |
||||
* This file is part of Transdroid <http://www.transdroid.org>
|
||||
* |
||||
* Transdroid is free software: you can redistribute it and/or modify |
||||
* it under the terms of the GNU General Public License as published by |
||||
* the Free Software Foundation, either version 3 of the License, or |
||||
* (at your option) any later version. |
||||
* |
||||
* Transdroid is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License |
||||
* along with Transdroid. If not, see <http://www.gnu.org/licenses/>.
|
||||
* |
||||
*/ |
||||
package org.transdroid.daemon; |
||||
|
||||
import java.io.File; |
||||
import java.net.URI; |
||||
import java.util.ArrayList; |
||||
import java.util.Arrays; |
||||
import java.util.Date; |
||||
import java.util.List; |
||||
import java.util.Random; |
||||
|
||||
import org.transdroid.daemon.DaemonException.ExceptionType; |
||||
import org.transdroid.daemon.task.AddByFileTask; |
||||
import org.transdroid.daemon.task.AddByMagnetUrlTask; |
||||
import org.transdroid.daemon.task.AddByUrlTask; |
||||
import org.transdroid.daemon.task.DaemonTask; |
||||
import org.transdroid.daemon.task.DaemonTaskFailureResult; |
||||
import org.transdroid.daemon.task.DaemonTaskResult; |
||||
import org.transdroid.daemon.task.DaemonTaskSuccessResult; |
||||
import org.transdroid.daemon.task.GetFileListTask; |
||||
import org.transdroid.daemon.task.GetFileListTaskSuccessResult; |
||||
import org.transdroid.daemon.task.GetTorrentDetailsTask; |
||||
import org.transdroid.daemon.task.GetTorrentDetailsTaskSuccessResult; |
||||
import org.transdroid.daemon.task.RetrieveTask; |
||||
import org.transdroid.daemon.task.RetrieveTaskSuccessResult; |
||||
import org.transdroid.daemon.task.SetFilePriorityTask; |
||||
import org.transdroid.daemon.task.SetLabelTask; |
||||
import org.transdroid.daemon.util.DLog; |
||||
|
||||
import android.net.Uri; |
||||
|
||||
/** |
||||
* A dummy adapter that does not communicate with some server, but maintains a local list of dummy data (reset every |
||||
* time it is recreated) to simplify testing. |
||||
* @author erickok |
||||
*/ |
||||
public class DummyAdapter implements IDaemonAdapter { |
||||
|
||||
private static final String LOG_NAME = "Dummy daemon"; |
||||
|
||||
private DaemonSettings settings; |
||||
private List<Torrent> dummyTorrents; |
||||
private List<Label> dummyLabels; |
||||
|
||||
/** |
||||
* Initialises a dummy adapter with some dummy data that may be manipulated. |
||||
*/ |
||||
public DummyAdapter(DaemonSettings settings) { |
||||
this.settings = settings; |
||||
this.dummyTorrents = new ArrayList<Torrent>(); |
||||
this.dummyLabels = new ArrayList<Label>(); |
||||
String[] names = new String[] { "Documentary ", "Book ", "CD Image ", "Mix tape ", "App " }; |
||||
String[] labels = new String[] { "docs", "books", "isos", "music", "software" }; |
||||
TorrentStatus[] statuses = new TorrentStatus[] { TorrentStatus.Seeding, TorrentStatus.Downloading, TorrentStatus.Paused, TorrentStatus.Queued, TorrentStatus.Downloading, TorrentStatus.Seeding, TorrentStatus.Error }; |
||||
Random random = new Random(); |
||||
for (int i = 0; i < 25; i++) { |
||||
String name = names[i % names.length] + Integer.toString(i); |
||||
TorrentStatus status = statuses[i % statuses.length]; |
||||
int peersGetting = status == TorrentStatus.Downloading ? i * random.nextInt(16) : 0; |
||||
int peersSending = status == TorrentStatus.Downloading ? i * random.nextInt(16) : 0; |
||||
long size = (long) (1024D * 1024D * 1024D * i * random.nextDouble()); |
||||
long left = status == TorrentStatus.Downloading ? (long) ((double)(size) / random.nextDouble()) : 0; |
||||
int rateDownload = status == TorrentStatus.Downloading ? (int) (1024D * 1024D * i * random.nextDouble()) |
||||
: 0; |
||||
int rateUpload = status == TorrentStatus.Downloading || status == TorrentStatus.Seeding ? (int) (1024D * 1024D * i * random |
||||
.nextDouble()) : 0; |
||||
this.dummyTorrents.add( |
||||
new Torrent( |
||||
i, |
||||
"torrent_" + i, |
||||
name, |
||||
status, |
||||
"/downloads/" + name.replace(" ", "_"), |
||||
rateDownload, |
||||
rateUpload, |
||||
peersGetting, |
||||
peersSending, |
||||
peersGetting + peersSending, // Total connections
|
||||
(peersGetting + peersSending) * 2, // Twice the total connections
|
||||
(int) (status == TorrentStatus.Downloading? |
||||
left / rateDownload: 0), // Eta
|
||||
size - left, |
||||
(long)((double)(size - left * 2) * random.nextDouble()), // Up to twice the amount downloaded
|
||||
size, |
||||
(float)(size - left) / size, // Part done
|
||||
1F, // Always 100% available
|
||||
labels[i % labels.length], |
||||
new Date(System.currentTimeMillis() - (7 * 24 * 60 * 60 * 1000)), // Last week
|
||||
null, |
||||
status == TorrentStatus.Error? |
||||
"Dummy error": null, |
||||
settings.getType())); |
||||
} |
||||
for (String label : labels) { |
||||
// TODO: // Add count?
|
||||
dummyLabels.add(new Label(label, 0)); |
||||
} |
||||
} |
||||
|
||||
@Override |
||||
public DaemonTaskResult executeTask(DaemonTask task) { |
||||
|
||||
try { |
||||
switch (task.getMethod()) { |
||||
case Retrieve: |
||||
|
||||
return new RetrieveTaskSuccessResult((RetrieveTask) task, dummyTorrents, dummyLabels); |
||||
|
||||
case GetTorrentDetails: |
||||
|
||||
return new GetTorrentDetailsTaskSuccessResult((GetTorrentDetailsTask) task, new TorrentDetails( |
||||
Arrays.asList("udp://tracker.com/announce:80", "https://torrents.org/announce:443"), |
||||
task.getTargetTorrent().getStatusCode() == TorrentStatus.Error ? |
||||
Arrays.asList("Trackers not working.", "Files not available.") : null)); |
||||
|
||||
case GetFileList: |
||||
|
||||
Torrent t = task.getTargetTorrent(); |
||||
List<TorrentFile> dummyFiles = new ArrayList<TorrentFile>(); |
||||
Priority priorities[] = new Priority[] { Priority.Normal, Priority.High, Priority.High, Priority.Low, Priority.Normal }; |
||||
for (int i = 0; i < 25; i++) { |
||||
String fileName = "file_" + i + ".ext"; |
||||
// Every file has equal part in the total size
|
||||
long size = t.getTotalSize() / 25; |
||||
long done = t.getDownloadedEver() / 25; |
||||
Priority priority = priorities[i % priorities.length]; |
||||
dummyFiles.add(new TorrentFile("file_" + i, t.getName() + " file " + i, fileName, t |
||||
.getLocationDir() + "/" + fileName, size, done, priority)); |
||||
} |
||||
return new GetFileListTaskSuccessResult((GetFileListTask) task, dummyFiles); |
||||
|
||||
case AddByFile: |
||||
|
||||
String file = ((AddByFileTask) task).getFile(); |
||||
DLog.d(LOG_NAME, "Adding torrent " + file); |
||||
File upload = new File(URI.create(file)); |
||||
dummyTorrents.add(new Torrent(0, "torrent_file", upload.getName(), TorrentStatus.Queued, "/downloads/" |
||||
+ file, 0, 0, 0, 0, 0, 0, -1, 0, 0, 1024 * 1024 * 1000, 0, 1F, "isos", new Date(), null, null, |
||||
settings.getType())); |
||||
return new DaemonTaskSuccessResult(task); |
||||
|
||||
case AddByUrl: |
||||
|
||||
String url = ((AddByUrlTask) task).getUrl(); |
||||
DLog.d(LOG_NAME, "Adding torrent " + url); |
||||
if (url == null || url.equals("")) |
||||
throw new DaemonException(DaemonException.ExceptionType.ParsingFailed, "No url specified"); |
||||
Uri uri = Uri.parse(url); |
||||
dummyTorrents.add(new Torrent(0, "torrent_byurl", uri.getLastPathSegment(), TorrentStatus.Queued, |
||||
"/downloads/" + uri.getLastPathSegment(), 0, 0, 0, 0, 0, 0, -1, 0, 0, 1024 * 1024 * 1000, 0, |
||||
1F, "music", new Date(), null, null, settings.getType())); |
||||
return new DaemonTaskSuccessResult(task); |
||||
|
||||
case AddByMagnetUrl: |
||||
|
||||
String magnet = ((AddByMagnetUrlTask) task).getUrl(); |
||||
DLog.d(LOG_NAME, "Adding torrent " + magnet); |
||||
Uri magnetUri = Uri.parse(magnet); |
||||
dummyTorrents.add(new Torrent(0, "torrent_magnet", magnetUri.getLastPathSegment(), |
||||
TorrentStatus.Queued, "/downloads/" + magnetUri.getLastPathSegment(), 0, 0, 0, 0, 0, 0, -1, 0, |
||||
0, 1024 * 1024 * 1000, 0, 1F, "books", new Date(), null, null, settings.getType())); |
||||
return new DaemonTaskSuccessResult(task); |
||||
|
||||
case Remove: |
||||
|
||||
dummyTorrents.remove(task.getTargetTorrent()); |
||||
return new DaemonTaskSuccessResult(task); |
||||
|
||||
case Pause: |
||||
|
||||
task.getTargetTorrent().mimicPause(); |
||||
return new DaemonTaskSuccessResult(task); |
||||
|
||||
case PauseAll: |
||||
|
||||
for (Torrent torrent: dummyTorrents) { |
||||
torrent.mimicPause(); |
||||
} |
||||
return new DaemonTaskSuccessResult(task); |
||||
|
||||
case Resume: |
||||
|
||||
task.getTargetTorrent().mimicPause(); |
||||
return new DaemonTaskSuccessResult(task); |
||||
|
||||
case ResumeAll: |
||||
|
||||
for (Torrent torrent: dummyTorrents) { |
||||
torrent.mimicResume(); |
||||
} |
||||
return new DaemonTaskSuccessResult(task); |
||||
|
||||
case Stop: |
||||
|
||||
task.getTargetTorrent().mimicStop(); |
||||
return new DaemonTaskSuccessResult(task); |
||||
|
||||
case StopAll: |
||||
|
||||
for (Torrent torrent: dummyTorrents) { |
||||
torrent.mimicStop(); |
||||
} |
||||
return new DaemonTaskSuccessResult(task); |
||||
|
||||
case Start: |
||||
|
||||
task.getTargetTorrent().mimicStart(); |
||||
return new DaemonTaskSuccessResult(task); |
||||
|
||||
case StartAll: |
||||
|
||||
for (Torrent torrent: dummyTorrents) { |
||||
torrent.mimicStart(); |
||||
} |
||||
return new DaemonTaskSuccessResult(task); |
||||
|
||||
case SetFilePriorities: |
||||
|
||||
SetFilePriorityTask prioTask = (SetFilePriorityTask) task; |
||||
for (TorrentFile forFile : prioTask.getForFiles()) { |
||||
forFile.mimicPriority(prioTask.getNewPriority()); |
||||
} |
||||
return new DaemonTaskSuccessResult(task); |
||||
|
||||
case SetTransferRates: |
||||
|
||||
// No action, as the result in not visible anyway
|
||||
return new DaemonTaskSuccessResult(task); |
||||
|
||||
case SetLabel: |
||||
|
||||
SetLabelTask labelTask = (SetLabelTask) task; |
||||
task.getTargetTorrent().mimicNewLabel(labelTask.getNewLabel()); |
||||
return new DaemonTaskSuccessResult(task); |
||||
|
||||
case SetTrackers: |
||||
|
||||
// TODO: Override the trackers?
|
||||
return new DaemonTaskSuccessResult(task); |
||||
|
||||
default: |
||||
return new DaemonTaskFailureResult(task, new DaemonException(ExceptionType.MethodUnsupported, |
||||
task.getMethod() + " is not supported by " + getType())); |
||||
} |
||||
} catch (DaemonException e) { |
||||
return new DaemonTaskFailureResult(task, e); |
||||
} |
||||
} |
||||
|
||||
@Override |
||||
public Daemon getType() { |
||||
return settings.getType(); |
||||
} |
||||
|
||||
@Override |
||||
public DaemonSettings getSettings() { |
||||
return this.settings; |
||||
} |
||||
|
||||
} |
||||
Loading…
Reference in new issue