btracker-scrape 0.1.0

Shared BitTorrent scrape API for the βtracker project components
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36

mod udp;

use std::net::SocketAddr;
use udp::Udp;

#[derive(Default)]
pub struct Result {
    pub leechers: u32,
    pub peers: u32,
    pub seeders: u32,
}

pub struct Scrape {
    udp: Option<Udp>,
    // tcp: @TODO
}

impl Scrape {
    pub fn init(udp: Option<(Vec<SocketAddr>, Vec<SocketAddr>)>) -> Self {
        Self {
            udp: udp.map(|(local, remote)| Udp::init(local, remote)),
        }
    }

    pub fn get(&self, info_hash: [u8; 20]) -> Option<Result> {
        self.udp.as_ref()?;
        let mut t = Result::default();
        if let Some(ref u) = self.udp {
            let r = u.scrape(info_hash).ok()?; // @TODO handle
            t.leechers += r.leechers;
            t.peers += r.peers;
            t.seeders += r.seeders;
        }
        Some(t)
    }
}