Restructure source files into 'src' subfolderHEADmaster

Signed-off-by: Michele Calgaro <[email protected]>

1 files changed, 183 insertions, 0 deletions

diff --git a/src/libktorrent/torrent/peersourcemanager.h b/src/libktorrent/torrent/peersourcemanager.h
new file mode 100644
index 0000000..e7de4f8
--- /dev/null
+++ b/src/libktorrent/torrent/peersourcemanager.h
@@ -0,0 +1,183 @@
+/***************************************************************************
+ *   Copyright (C) 2005 by Joris Guisson                                   *
+ *   [email protected]                                               *
+ *                                                                         *
+ *   This program 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 2 of the License, or     *
+ *   (at your option) any later version.                                   *
+ *                                                                         *
+ *   This program 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 this program; if not, write to the                         *
+ *   Free Software Foundation, Inc.,                                       *
+ *   51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.          *
+ ***************************************************************************/
+#ifndef BTPEERSOURCEMANAGER_H
+#define BTPEERSOURCEMANAGER_H
+
+#include <tqtimer.h>
+#include <tqdatetime.h>
+#include <tqptrlist.h>
+#include <util/ptrmap.h>
+#include <interfaces/trackerslist.h>
+
+namespace kt
+{
+	class PeerSource;
+}
+
+namespace dht
+{
+	class DHTTrackerBackend;
+}
+
+namespace bt
+{
+	class Tracker;
+	class PeerManager;
+	class Torrent;
+	class TorrentControl;
+
+	/**
+	 * @author Joris Guisson <[email protected]>
+	 * 
+	 * This class manages all PeerSources.
+	*/
+	class PeerSourceManager : public TQObject, public kt::TrackersList
+	{
+		TQ_OBJECT
+  
+				
+		TorrentControl* tor;
+		PeerManager* pman;
+		PtrMap<KURL,Tracker> trackers;
+		TQPtrList<kt::PeerSource> additional;
+		Tracker* curr;
+		dht::DHTTrackerBackend* m_dht;
+		bool started;
+		bool pending;
+		KURL::List custom_trackers;
+		TQDateTime request_time;
+		TQTimer timer;
+		Uint32 failures;
+		bool no_save_custom_trackers;
+	public:
+		PeerSourceManager(TorrentControl* tor,PeerManager* pman);
+		virtual ~PeerSourceManager();
+	
+				
+		/**
+		 * Add a PeerSource, the difference between PeerSource and Tracker
+		 * is that only one Tracker can be used at the same time, 
+		 * PeerSource can always be used.
+		 * @param ps The PeerSource
+		 */
+		void addPeerSource(kt::PeerSource* ps);
+
+		/**
+		 * See if the PeerSourceManager has been started 
+		 */
+		bool isStarted() const {return started;}
+		
+		/**
+		 * Start gathering peers
+		 */
+		void start();
+		
+		/**
+		 * Stop gathering peers
+		 * @param wjob WaitJob to wait at exit for the completion of stopped events to the trackers
+		 */
+		void stop(WaitJob* wjob = 0);
+		
+		/**
+		 * Notify peersources and trackrs that the download is complete.
+		 */
+		void completed();
+		
+		/**
+		 * Do a manual update on all peer sources and trackers.
+		 */
+		void manualUpdate();
+		
+		/**
+		 * Remove a Tracker or PeerSource.
+		 * @param ps 
+		 */
+		void removePeerSource(kt::PeerSource* ps);
+		
+		virtual KURL getTrackerURL() const;
+		virtual KURL::List getTrackerURLs();
+		virtual void addTracker(KURL url, bool custom = true,int tier = 1);
+		virtual bool removeTracker(KURL url);
+		virtual void setTracker(KURL url);
+		virtual void restoreDefault();
+		
+		/**
+		 * Get the time to the next tracker update.
+		 * @return The time in seconds
+		 */
+		Uint32 getTimeToNextUpdate() const;
+		
+		/// Get the number of potential seeders
+		Uint32 getNumSeeders() const;
+		
+		/// Get the number of potential leechers
+		Uint32 getNumLeechers() const;
+		
+		/// Get the number of failures
+		Uint32 getNumFailures() const {return failures;}
+		
+		///Adds DHT as PeerSource for this torrent
+		void addDHT();
+		///Removes DHT from PeerSourceManager for this torrent.
+		void removeDHT();
+		///Checks if DHT is enabled
+		bool dhtStarted();
+		
+	private slots:
+		/**
+		 * The an error happened contacting the tracker.
+		 * @param err The error
+		 */
+		void onTrackerError(const TQString & err);
+		
+		/**
+		 * Tracker update was OK.
+		 * @param 
+		 */
+		void onTrackerOK();
+		
+		/**
+		 * Tracker is doing a request.  
+		 */
+		void onTrackerRequestPending();
+		
+		/**
+		 * Update the current tracker manually
+		 */
+		void updateCurrentManually();
+		
+	signals:
+		/**
+		 * Status has changed of the tracker.
+		 * @param ns The new status
+		 */
+		void statusChanged(const TQString & ns);
+		
+	private:
+		void saveCustomURLs();
+		void loadCustomURLs();
+		void addTracker(Tracker* trk);
+		void switchTracker(Tracker* trk);
+		Tracker* selectTracker();
+	};
+
+}
+
+#endif