Ben Davis
60 lines
2.0 KiB
Swift
60 lines
2.0 KiB
Swift
//
|
|
// Tracker.swift
|
|
// BitTorrent
|
|
//
|
|
// Created by Ben Davis on 27/06/2017.
|
|
// Copyright © 2017 Ben Davis. All rights reserved.
|
|
//
|
|
|
|
import Foundation
|
|
|
|
class TorrentHTTPTracker: TorrentTracker {
|
|
|
|
let announceURL: URL
|
|
let connection: BasicHTTPConnection
|
|
|
|
weak var delegate: TorrentTrackerDelegate?
|
|
|
|
init(announceURL: URL, connection: BasicHTTPConnection = HTTPConnection()) {
|
|
self.announceURL = announceURL
|
|
self.connection = connection
|
|
}
|
|
|
|
func announceClient(with peerId: String,
|
|
port: UInt16,
|
|
event: TorrentTrackerEvent = .started,
|
|
infoHash: Data,
|
|
numberOfBytesRemaining: Int,
|
|
numberOfBytesUploaded: Int,
|
|
numberOfBytesDownloaded: Int,
|
|
numberOfPeersToFetch: Int) {
|
|
|
|
let urlParameters = [
|
|
"info_hash" : String(urlEncodingData: infoHash),
|
|
"peer_id" : "\(peerId)",
|
|
"port" : "\(port)",
|
|
"uploaded" : "\(numberOfBytesUploaded)",
|
|
"downloaded" : "\(numberOfBytesDownloaded)",
|
|
"left" : "\(numberOfBytesRemaining)",
|
|
"compact" : "1",
|
|
"event" : event.name,
|
|
"numwant" : "\(numberOfPeersToFetch)"
|
|
]
|
|
|
|
connection.makeRequest(url: announceURL, urlParameters: urlParameters) { [weak self] response in
|
|
|
|
guard self != nil else {
|
|
return
|
|
}
|
|
|
|
if let data = response.responseData {
|
|
if let result = TorrentTrackerResponse(bencode: data) {
|
|
self!.delegate?.torrentTracker(self!, receivedResponse: result)
|
|
} else if let errorMessage = TorrentTrackerResponse.errorMessage(fromResponseData: data) {
|
|
self!.delegate?.torrentTracker(self!, receivedErrorMessage: errorMessage)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|