YGGverse/btracker
1
0
Fork 0
mirror of https://github.com/YGGverse/btracker.git synced 2026-03-31 17:15:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

implement files list

Browse source
This commit is contained in:
yggverse 2025-08-09 18:37:50 +03:00
parent c84e0ffbdb
commit fa748fbd18
7 changed files with 133 additions and 76 deletions
Download patch file Download diff file Expand all files Collapse all files

30
src/format.rs
View file

@ -1,30 +0,0 @@
use crate::{Meta, Scrape, Scraper, Torrent};
use rocket::{State, serde::Serialize};
#[derive(Serialize)]
#[serde(crate = "rocket::serde")]
pub struct Format {
pub created: Option<String>,
pub files: String,
pub indexed: String,
pub magnet: String,
pub scrape: Option<Scrape>,
pub size: String,
pub torrent: Torrent,
}
impl Format {
pub fn from_torrent(torrent: Torrent, scraper: &State<Scraper>, meta: &State<Meta>) -> Self {
Self {
created: torrent
.creation_date
.map(|t| t.format(&meta.format_time).to_string()),
indexed: torrent.time.format(&meta.format_time).to_string(),
magnet: torrent.magnet(meta.trackers.as_ref()),
scrape: scraper.scrape(&torrent.info_hash),
size: torrent.size(),
files: torrent.files(),
torrent,
}
}
}

73
src/main.rs
View file

@ -3,7 +3,6 @@ extern crate rocket;
mod config;
mod feed;
mod format;
mod meta;
mod scraper;
mod storage;
@ -11,13 +10,13 @@ mod torrent;
use config::Config;
use feed::Feed;
use format::Format;
use meta::Meta;
use plurify::Plurify;
use rocket::{
State,
http::Status,
response::{content::RawXml, status::Custom},
serde::Serialize,
};
use rocket_dyn_templates::{Template, context};
use scraper::{Scrape, Scraper};
@ -32,6 +31,17 @@ fn index(
storage: &State<Storage>,
meta: &State<Meta>,
) -> Result<Template, Custom<String>> {
#[derive(Serialize)]
#[serde(crate = "rocket::serde")]
struct R {
created: Option<String>,
files: String,
indexed: String,
magnet: String,
scrape: Option<Scrape>,
size: String,
torrent: Torrent,
}
let (total, torrents) = storage
.torrents(
Some((Sort::Modified, Order::Desc)),
@ -42,7 +52,6 @@ fn index(
error!("Torrents storage read error: `{e}`");
Custom(Status::InternalServerError, E.to_string())
})?;
Ok(Template::render(
"index",
context! {
@ -53,13 +62,21 @@ fn index(
rows: torrents
.into_iter()
.filter_map(|t| match Torrent::from_storage(&t.bytes, t.time) {
Ok(torrent) => Some(Format::from_torrent(torrent, scraper, meta)),
Ok(torrent) => Some(R {
created: torrent.creation_date.map(|t| t.format(&meta.format_time).to_string()),
files: torrent.files(),
indexed: torrent.time.format(&meta.format_time).to_string(),
magnet: torrent.magnet(meta.trackers.as_ref()),
scrape: scraper.scrape(&torrent.info_hash),
size: torrent.size(),
torrent
}),
Err(e) => {
error!("Torrent storage read error: `{e}`");
None
}
})
.collect::<Vec<Format>>(),
.collect::<Vec<R>>(),
pagination_totals: format!(
"Page {} / {} ({total} {} total)",
page.unwrap_or(1),
@ -81,19 +98,39 @@ fn info(
warn!("Torrent info-hash parse error: `{e}`");
Custom(Status::BadRequest, Status::BadRequest.to_string())
})?) {
Some(t) => Ok(Template::render(
"info",
context! {
meta: meta.inner(),
torrent: Format::from_torrent(
Torrent::from_storage(&t.bytes, t.time).map_err(|e| {
error!("Torrent parse error: `{e}`");
Custom(Status::InternalServerError, E.to_string())
})?, scraper, meta
),
info_hash
},
)),
Some(t) => {
#[derive(Serialize)]
#[serde(crate = "rocket::serde")]
struct F {
name: String,
size: String,
}
let torrent = Torrent::from_storage(&t.bytes, t.time).map_err(|e| {
error!("Torrent parse error: `{e}`");
Custom(Status::InternalServerError, E.to_string())
})?;
Ok(Template::render(
"info",
context! {
meta: meta.inner(),
created: torrent.creation_date.map(|t| t.format(&meta.format_time).to_string()),
files_total: torrent.files(),
files_list: torrent.files.as_ref().map(|f| {
f.iter()
.map(|f| F {
name: f.name(),
size: f.size(),
})
.collect::<Vec<F>>()
}),
indexed: torrent.time.format(&meta.format_time).to_string(),
magnet: torrent.magnet(meta.trackers.as_ref()),
scrape: scraper.scrape(info_hash),
size: torrent.size(),
torrent
},
))
}
None => Err(Custom(Status::NotFound, E.to_string())),
}
}

34
src/torrent.rs
View file

@ -88,21 +88,7 @@ impl Torrent {
}
pub fn size(&self) -> String {
const KB: f32 = 1024.0;
const MB: f32 = KB * KB;
const GB: f32 = MB * KB;
let f = self.size as f32;
if f < KB {
format!("{} B", self.size)
} else if f < MB {
format!("{:.2} KB", f / KB)
} else if f < GB {
format!("{:.2} MB", f / MB)
} else {
format!("{:.2} GB", f / GB)
}
size(self.size)
}
pub fn magnet(&self, trackers: Option<&Vec<url::Url>>) -> String {
@ -120,3 +106,21 @@ impl Torrent {
b
}
}
fn size(value: u64) -> String {
const KB: f32 = 1024.0;
const MB: f32 = KB * KB;
const GB: f32 = MB * KB;
let f = value as f32;
if f < KB {
format!("{value} B")
} else if f < MB {
format!("{:.2} KB", f / KB)
} else if f < GB {
format!("{:.2} MB", f / MB)
} else {
format!("{:.2} GB", f / GB)
}
}

9
src/torrent/file.rs
View file

@ -6,3 +6,12 @@ pub struct File {
pub name: Option<String>,
pub length: u64,
}
impl File {
pub fn name(&self) -> String {
self.name.as_deref().unwrap_or("?").into()
}
pub fn size(&self) -> String {
super::size(self.length)
}
}