transdroid/connect/src/main/java/org/transdroid/connect/clients/rtorrent/Rtorrent.java

package org.transdroid.connect.clients.rtorrent;

import com.jakewharton.retrofit2.adapter.rxjava2.RxJava2CallAdapterFactory;

import org.transdroid.connect.Configuration;
import org.transdroid.connect.clients.ClientSpec;
import org.transdroid.connect.clients.Feature;
import org.transdroid.connect.model.Torrent;
import org.transdroid.connect.model.TorrentStatus;
import org.transdroid.connect.util.OkHttpBuilder;
import org.transdroid.connect.util.RxUtil;

import java.util.Date;
import java.util.HashSet;
import java.util.Set;

import io.reactivex.Flowable;
import io.reactivex.functions.Function;
import nl.nl2312.xmlrpc.Nothing;
import nl.nl2312.xmlrpc.XmlRpcConverterFactory;
import retrofit2.Retrofit;

public final class Rtorrent implements ClientSpec {

	public static final Set<Feature> FEATURES = new HashSet<>();

	{
		FEATURES.add(Feature.VERSION);
		FEATURES.add(Feature.STARTING);
		FEATURES.add(Feature.STOPPING);
		FEATURES.add(Feature.RESUMING);
		FEATURES.add(Feature.PAUSING);
	}

	private final Configuration configuration;
	private final Service service;

	public Rtorrent(Configuration configuration) {
		this.configuration = configuration;
		Retrofit retrofit = new Retrofit.Builder()
				.baseUrl(configuration.baseUrl())
				.client(new OkHttpBuilder(configuration).build())
				.addCallAdapterFactory(RxJava2CallAdapterFactory.create())
				.addConverterFactory(XmlRpcConverterFactory.create())
				.build();
		this.service = retrofit.create(Service.class);
	}

	@Override
	public Flowable<String> clientVersion() {
		return service.clientVersion(configuration.endpoint(), Nothing.NOTHING);
	}

	@Override
	public Flowable<Torrent> torrents() {
		return service.torrents(
				configuration.endpoint(),
				"",
				"main",
				"d.hash=",
				"d.name=",
				"d.state=",
				"d.down.rate=",
				"d.up.rate=",
				"d.peers_connected=",
				"d.peers_not_connected=",
				"d.bytes_done=",
				"d.up.total=",
				"d.size_bytes=",
				"d.left_bytes=",
				"d.creation_date=",
				"d.complete=",
				"d.is_active=",
				"d.is_hash_checking=",
				"d.base_path=",
				"d.base_filename=",
				"d.message=",
				"d.custom=addtime",
				"d.custom=seedingtime",
				"d.custom1=",
				"d.peers_complete=",
				"d.peers_accounted=")
				.compose(RxUtil.<TorrentSpec>asList())
				.map(new Function<TorrentSpec, Torrent>() {
					@Override
					public Torrent apply(TorrentSpec torrentSpec) throws Exception {
						return new Torrent(
								torrentSpec.hash.hashCode(),
								torrentSpec.hash,
								torrentSpec.name,
								torrentStatus(torrentSpec.state, torrentSpec.isComplete, torrentSpec.isActive, torrentSpec.isHashChecking),
								torrentSpec.basePath.substring(0, torrentSpec.basePath.indexOf(torrentSpec.baseFilename)),
								(int) torrentSpec.downloadRate,
								(int) torrentSpec.uploadRate,
								(int) torrentSpec.seedersConnected,
								(int) (torrentSpec.peersConnected + torrentSpec.peersNotConnected),
								(int) torrentSpec.leechersConnected,
								(int) (torrentSpec.peersConnected + torrentSpec.peersNotConnected),
								torrentSpec.downloadRate > 0 ? (torrentSpec.bytesleft / torrentSpec.downloadRate) : Torrent.UNKNOWN,
								torrentSpec.bytesDone,
								torrentSpec.bytesUploaded,
								torrentSpec.bytesTotal,
								torrentSpec.bytesDone / torrentSpec.bytesTotal,
								0F,
								torrentSpec.label,
								torrentTimeAdded(torrentSpec.timeAdded, torrentSpec.timeCreated),
								torrentTimeFinished(torrentSpec.timeFinished),
								torrentSpec.errorMessage
						);
					}
				});
	}

	private TorrentStatus torrentStatus(long state, long complete, long active, long checking) {
		if (state == 0) {
			return TorrentStatus.QUEUED;
		} else if (active == 1) {
			if (complete == 1) {
				return TorrentStatus.SEEDING;
			} else {
				return TorrentStatus.DOWNLOADING;
			}
		} else if (checking == 1) {
			return TorrentStatus.CHECKING;
		} else {
			return TorrentStatus.PAUSED;
		}
	}

	private Date torrentTimeAdded(String timeAdded, long timeCreated) {
		if (timeAdded != null || timeAdded.trim().length() != 0) {
			return new Date(Long.parseLong(timeAdded.trim()) * 1000L);
		}
		return new Date(timeCreated * 1000L);
	}

	private Date torrentTimeFinished(String timeFinished) {
		if (timeFinished == null || timeFinished.trim().length() == 0)
			return null;
		return new Date(Long.parseLong(timeFinished.trim()) * 1000L);
	}

}