-
Notifications
You must be signed in to change notification settings - Fork 44
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
refactor: [#661] move HTTP Tracker Client mod
- Loading branch information
1 parent
0960ff2
commit b96c2c3
Showing
4 changed files
with
105 additions
and
92 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,96 +1,7 @@ | ||
//! HTTP Tracker client: | ||
//! | ||
//! Examples: | ||
//! | ||
//! `Announce` request: | ||
//! | ||
//! ```text | ||
//! cargo run --bin http_tracker_client announce http://127.0.0.1:7070 9c38422213e30bff212b30c360d26f9a02136422 | jq | ||
//! ``` | ||
//! | ||
//! `Scrape` request: | ||
//! | ||
//! ```text | ||
//! cargo run --bin http_tracker_client scrape http://127.0.0.1:7070 9c38422213e30bff212b30c360d26f9a02136422 | jq | ||
//! ``` | ||
use std::str::FromStr; | ||
|
||
use anyhow::Context; | ||
use clap::{Parser, Subcommand}; | ||
use reqwest::Url; | ||
use torrust_tracker::shared::bit_torrent::info_hash::InfoHash; | ||
use torrust_tracker::shared::bit_torrent::tracker::http::client::requests::announce::QueryBuilder; | ||
use torrust_tracker::shared::bit_torrent::tracker::http::client::responses::announce::Announce; | ||
use torrust_tracker::shared::bit_torrent::tracker::http::client::responses::scrape; | ||
use torrust_tracker::shared::bit_torrent::tracker::http::client::{requests, Client}; | ||
|
||
#[derive(Parser, Debug)] | ||
#[command(author, version, about, long_about = None)] | ||
struct Args { | ||
#[command(subcommand)] | ||
command: Command, | ||
} | ||
|
||
#[derive(Subcommand, Debug)] | ||
enum Command { | ||
Announce { tracker_url: String, info_hash: String }, | ||
Scrape { tracker_url: String, info_hashes: Vec<String> }, | ||
} | ||
//! Program to make request to HTTP trackers. | ||
use torrust_tracker::console::clients::http::app; | ||
|
||
#[tokio::main] | ||
async fn main() -> anyhow::Result<()> { | ||
let args = Args::parse(); | ||
|
||
match args.command { | ||
Command::Announce { tracker_url, info_hash } => { | ||
announce_command(tracker_url, info_hash).await?; | ||
} | ||
Command::Scrape { | ||
tracker_url, | ||
info_hashes, | ||
} => { | ||
scrape_command(&tracker_url, &info_hashes).await?; | ||
} | ||
} | ||
Ok(()) | ||
} | ||
|
||
async fn announce_command(tracker_url: String, info_hash: String) -> anyhow::Result<()> { | ||
let base_url = Url::parse(&tracker_url).context("failed to parse HTTP tracker base URL")?; | ||
let info_hash = | ||
InfoHash::from_str(&info_hash).expect("Invalid infohash. Example infohash: `9c38422213e30bff212b30c360d26f9a02136422`"); | ||
|
||
let response = Client::new(base_url) | ||
.announce(&QueryBuilder::with_default_values().with_info_hash(&info_hash).query()) | ||
.await; | ||
|
||
let body = response.bytes().await.unwrap(); | ||
|
||
let announce_response: Announce = serde_bencode::from_bytes(&body) | ||
.unwrap_or_else(|_| panic!("response body should be a valid announce response, got: \"{:#?}\"", &body)); | ||
|
||
let json = serde_json::to_string(&announce_response).context("failed to serialize scrape response into JSON")?; | ||
|
||
println!("{json}"); | ||
|
||
Ok(()) | ||
} | ||
|
||
async fn scrape_command(tracker_url: &str, info_hashes: &[String]) -> anyhow::Result<()> { | ||
let base_url = Url::parse(tracker_url).context("failed to parse HTTP tracker base URL")?; | ||
|
||
let query = requests::scrape::Query::try_from(info_hashes).context("failed to parse infohashes")?; | ||
|
||
let response = Client::new(base_url).scrape(&query).await; | ||
|
||
let body = response.bytes().await.unwrap(); | ||
|
||
let scrape_response = scrape::Response::try_from_bencoded(&body) | ||
.unwrap_or_else(|_| panic!("response body should be a valid scrape response, got: \"{:#?}\"", &body)); | ||
|
||
let json = serde_json::to_string(&scrape_response).context("failed to serialize scrape response into JSON")?; | ||
|
||
println!("{json}"); | ||
|
||
Ok(()) | ||
app::run().await | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,100 @@ | ||
//! HTTP Tracker client: | ||
//! | ||
//! Examples: | ||
//! | ||
//! `Announce` request: | ||
//! | ||
//! ```text | ||
//! cargo run --bin http_tracker_client announce http://127.0.0.1:7070 9c38422213e30bff212b30c360d26f9a02136422 | jq | ||
//! ``` | ||
//! | ||
//! `Scrape` request: | ||
//! | ||
//! ```text | ||
//! cargo run --bin http_tracker_client scrape http://127.0.0.1:7070 9c38422213e30bff212b30c360d26f9a02136422 | jq | ||
//! ``` | ||
use std::str::FromStr; | ||
|
||
use anyhow::Context; | ||
use clap::{Parser, Subcommand}; | ||
use reqwest::Url; | ||
|
||
use crate::shared::bit_torrent::info_hash::InfoHash; | ||
use crate::shared::bit_torrent::tracker::http::client::requests::announce::QueryBuilder; | ||
use crate::shared::bit_torrent::tracker::http::client::responses::announce::Announce; | ||
use crate::shared::bit_torrent::tracker::http::client::responses::scrape; | ||
use crate::shared::bit_torrent::tracker::http::client::{requests, Client}; | ||
|
||
#[derive(Parser, Debug)] | ||
#[command(author, version, about, long_about = None)] | ||
struct Args { | ||
#[command(subcommand)] | ||
command: Command, | ||
} | ||
|
||
#[derive(Subcommand, Debug)] | ||
enum Command { | ||
Announce { tracker_url: String, info_hash: String }, | ||
Scrape { tracker_url: String, info_hashes: Vec<String> }, | ||
} | ||
|
||
/// # Errors | ||
/// | ||
/// Will return an error if the command fails. | ||
pub async fn run() -> anyhow::Result<()> { | ||
let args = Args::parse(); | ||
|
||
match args.command { | ||
Command::Announce { tracker_url, info_hash } => { | ||
announce_command(tracker_url, info_hash).await?; | ||
} | ||
Command::Scrape { | ||
tracker_url, | ||
info_hashes, | ||
} => { | ||
scrape_command(&tracker_url, &info_hashes).await?; | ||
} | ||
} | ||
|
||
Ok(()) | ||
} | ||
|
||
async fn announce_command(tracker_url: String, info_hash: String) -> anyhow::Result<()> { | ||
let base_url = Url::parse(&tracker_url).context("failed to parse HTTP tracker base URL")?; | ||
let info_hash = | ||
InfoHash::from_str(&info_hash).expect("Invalid infohash. Example infohash: `9c38422213e30bff212b30c360d26f9a02136422`"); | ||
|
||
let response = Client::new(base_url) | ||
.announce(&QueryBuilder::with_default_values().with_info_hash(&info_hash).query()) | ||
.await; | ||
|
||
let body = response.bytes().await.unwrap(); | ||
|
||
let announce_response: Announce = serde_bencode::from_bytes(&body) | ||
.unwrap_or_else(|_| panic!("response body should be a valid announce response, got: \"{:#?}\"", &body)); | ||
|
||
let json = serde_json::to_string(&announce_response).context("failed to serialize scrape response into JSON")?; | ||
|
||
println!("{json}"); | ||
|
||
Ok(()) | ||
} | ||
|
||
async fn scrape_command(tracker_url: &str, info_hashes: &[String]) -> anyhow::Result<()> { | ||
let base_url = Url::parse(tracker_url).context("failed to parse HTTP tracker base URL")?; | ||
|
||
let query = requests::scrape::Query::try_from(info_hashes).context("failed to parse infohashes")?; | ||
|
||
let response = Client::new(base_url).scrape(&query).await; | ||
|
||
let body = response.bytes().await.unwrap(); | ||
|
||
let scrape_response = scrape::Response::try_from_bencoded(&body) | ||
.unwrap_or_else(|_| panic!("response body should be a valid scrape response, got: \"{:#?}\"", &body)); | ||
|
||
let json = serde_json::to_string(&scrape_response).context("failed to serialize scrape response into JSON")?; | ||
|
||
println!("{json}"); | ||
|
||
Ok(()) | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1 @@ | ||
pub mod app; |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,2 +1,3 @@ | ||
//! Console clients. | ||
pub mod checker; | ||
pub mod http; |