implement pagination

src/config.rs

@@ -14,7 +14,7 @@ pub struct Config {
 
     /// Default listing limit
     #[arg(long, default_value_t = 50)]
-    pub limit: usize,
+    pub list_limit: usize,
 
     /// Default capacity (estimated torrents in `storage`)
     #[arg(long, default_value_t = 1000)]

src/main.rs

@@ -30,8 +30,12 @@ pub struct Meta {
     pub trackers: Option<HashSet<Url>>,
 }
 
-#[get("/")]
+#[get("/?<page>")]
-fn index(storage: &State<Storage>, meta: &State<Meta>) -> Result<Template, Custom<String>> {
+fn index(
+    page: Option<usize>,
+    storage: &State<Storage>,
+    meta: &State<Meta>,
+) -> Result<Template, Custom<String>> {
     #[derive(Serialize)]
     #[serde(crate = "rocket::serde")]
     struct Row {
@@ -41,25 +45,32 @@ fn index(storage: &State<Storage>, meta: &State<Meta>) -> Result<Template, Custo
         size: String,
         torrent: Torrent,
     }
-    Ok(Template::render(
+    let rows = storage
-        "index",
-        context! {
-            meta: meta.inner(),
-            rows: storage
         .torrents(
             Some((Sort::Modified, Order::Asc)),
+            page.map(|p| if p > 0 { p - 1 } else { p } * storage.default_limit),
             Some(storage.default_limit),
         )
         .map_err(|e| Custom(Status::InternalServerError, e.to_string()))?
         .into_iter()
         .map(|torrent| Row {
-                created: torrent.creation_date.map(|t|t.format(&meta.format_time).to_string()),
+            created: torrent
+                .creation_date
+                .map(|t| t.format(&meta.format_time).to_string()),
             indexed: torrent.time.format(&meta.format_time).to_string(),
             magnet: format::magnet(&torrent.info_hash, meta.trackers.as_ref()),
             size: format::bytes(torrent.size),
             torrent,
         })
-            .collect::<Vec<Row>>()
+        .collect::<Vec<Row>>();
+    Ok(Template::render(
+        "index",
+        context! {
+            meta: meta.inner(),
+            back: page.map(|p| uri!(index(if p > 2 { Some(p - 1) } else { None }))),
+            next: if rows.len() < storage.default_limit { None }
+                    else { Some(uri!(index(Some(page.map_or(2, |p| p + 1))))) },
+            rows
         },
     ))
 }
@@ -70,6 +81,7 @@ fn rss(feed: &State<Feed>, storage: &State<Storage>) -> Result<RawXml<String>, C
     for torrent in storage
         .torrents(
             Some((Sort::Modified, Order::Asc)),
+            None,
             Some(storage.default_limit),
         )
         .map_err(|e| Custom(Status::InternalServerError, e.to_string()))?
@@ -89,7 +101,7 @@ fn rocket() -> _ {
         config.link.clone(),
         config.tracker.clone().map(|u| u.into_iter().collect()),
     );
-    let storage = Storage::init(config.storage, config.limit, config.capacity).unwrap(); // @TODO handle
+    let storage = Storage::init(config.storage, config.list_limit, config.capacity).unwrap(); // @TODO handle
     rocket::build()
         .attach(Template::fairing())
         .configure(rocket::Config {

src/storage.rs

@@ -74,12 +74,13 @@ impl Storage {
     pub fn torrents(
         &self,
         sort_order: Option<(Sort, Order)>,
+        start: Option<usize>,
         limit: Option<usize>,
     ) -> Result<Vec<Torrent>, String> {
         let f = self.files(sort_order)?;
         let l = limit.unwrap_or(f.len());
         let mut b = Vec::with_capacity(l);
-        for file in f.into_iter().take(l) {
+        for file in f.into_iter().skip(start.unwrap_or_default()).take(l) {
             if file
                 .path()
                 .extension()

templates/index.html.tera

@@ -1,6 +1,7 @@
 {% extends "layout/default" %}
 {% block content %}
-{% for row in rows %}
+{% if rows %}
+    {% for row in rows %}
         <div>
             <a name="{{ row.torrent.info_hash }}"></a>
             <h2>{{ row.torrent.name }}</h2>
@@ -18,5 +19,12 @@
                 </a>
             </div>
         </div>
-{% endfor %}
+    {% endfor %}
+{% else %}
+    <div>
+        Nothing.
+    </div>
+{% endif %}
+{% if next %}<a href="{{ next }}">Next</a>{% endif %}
+{% if back %}<a href="{{ back }}">Back</a>{% endif %}
 {% endblock content %}

templates/layout/default.html.tera

@@ -86,6 +86,16 @@
                 margin: 0 auto;
             }
 
+            /* pagination */
+            main > a {
+                background: #34384f;
+                border-radius: 3px;
+                float: right;
+                margin-left: 8px;
+                opacity: .96;
+                padding: 8px;
+            }
+
             /* item row */
             main > div {
                 background-color: #34384f;