use build-in tera filters

This commit is contained in:
yggverse 2025-08-20 14:37:59 +03:00
parent c1ae25f04b
commit d95cee6c09
7 changed files with 51 additions and 53 deletions

View file

@ -70,7 +70,10 @@ impl Feed {
self.buffer.push_str("<description>");
self.buffer
.push_str(&format!("{}\n{}", torrent.size(), torrent.files()));
.push_str(&format!("size: {}", size(torrent.size)));
if let Some(f) = torrent.files() {
self.buffer.push_str(&format!(" / files: {f}"));
}
self.buffer.push_str("</description>");
self.buffer.push_str("<pubDate>");
@ -87,6 +90,9 @@ impl Feed {
}
}
// @TODO use tera filters?
// https://keats.github.io/tera/docs/#built-in-filters
fn escape(value: &str) -> String {
value
.replace('&', "&amp;")
@ -95,3 +101,21 @@ fn escape(value: &str) -> String {
.replace('"', "&quot;")
.replace("'", "&apos;")
}
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)
}
}

View file

@ -11,7 +11,6 @@ mod torrent;
use config::Config;
use feed::Feed;
use meta::Meta;
use plurify::Plurify;
use public::{Order, Public, Sort};
use rocket::{State, http::Status, response::content::RawXml, serde::Serialize};
use rocket_dyn_templates::{Template, context};
@ -31,11 +30,11 @@ fn index(
#[serde(crate = "rocket::serde")]
struct R {
created: Option<String>,
files: String,
files: Option<usize>,
indexed: String,
magnet: String,
scrape: Option<Scrape>,
size: String,
size: u64,
torrent: Torrent,
}
let (total, torrents) = public
@ -85,7 +84,7 @@ fn index(
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(),
size: torrent.size,
torrent
}),
Err(e) => {
@ -94,12 +93,9 @@ fn index(
}
})
.collect::<Vec<R>>(),
pagination_totals: if total > 0 { Some(format!(
"Page {} / {} ({total} {} total)",
page.unwrap_or(1),
(total as f64 / public.default_limit as f64).ceil(),
total.plurify(&["torrent", "torrents", "torrents"])
)) } else { None },
page: page.unwrap_or(1),
pages: (total as f64 / public.default_limit as f64).ceil(),
total,
search
},
))
@ -119,7 +115,7 @@ fn info(
struct F {
href: Option<String>,
path: String,
size: String,
size: u64,
}
let torrent = Torrent::from_public(&t.bytes, t.time).map_err(|e| {
error!("Torrent parse error: `{e}`");
@ -147,7 +143,7 @@ fn info(
F {
href: public.href(&torrent.info_hash, &p),
path: p,
size: f.size(),
size: f.length,
}
})
.collect::<Vec<F>>()
@ -155,7 +151,7 @@ fn info(
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(),
size: torrent.size,
torrent
},
))

View file

@ -71,16 +71,8 @@ impl Torrent {
// Format getters
pub fn files(&self) -> String {
use plurify::Plurify;
self.files.as_ref().map_or("1 file".into(), |f| {
let l = f.len();
format!("{l} {}", l.plurify(&["file", "files", "files"]))
})
}
pub fn size(&self) -> String {
size(self.size)
pub fn files(&self) -> Option<usize> {
self.files.as_ref().map(|f| f.len())
}
pub fn magnet(&self, trackers: Option<&Vec<url::Url>>) -> String {
@ -98,21 +90,3 @@ 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)
}
}

View file

@ -12,7 +12,4 @@ impl File {
.map(|p| p.to_string_lossy().into())
.unwrap_or("?".into())
}
pub fn size(&self) -> String {
super::size(self.length)
}
}