From 4f8ddd1c226f67fca15b4192c37d86399542d318 Mon Sep 17 00:00:00 2001
From: Azalea Gui <22280294+hykilpikonna@users.noreply.github.com>
Date: Sat, 21 Dec 2024 21:40:06 -0500
Subject: [PATCH] [O] Shorten code for #372

---
 crates/hyfetch/src/bin/hyfetch.rs | 17 +++--------------
 1 file changed, 3 insertions(+), 14 deletions(-)

diff --git a/crates/hyfetch/src/bin/hyfetch.rs b/crates/hyfetch/src/bin/hyfetch.rs
index ae4724b9..5001ae39 100644
--- a/crates/hyfetch/src/bin/hyfetch.rs
+++ b/crates/hyfetch/src/bin/hyfetch.rs
@@ -579,10 +579,7 @@ fn create_config(
         print_flag_page(&pages[usize::from(page)], page).context("failed to print flag page")?;
 
         let mut opts: Vec<&str> = <Preset as VariantNames>::VARIANTS.into();
-        opts.push("next");
-        opts.push("n");
-        opts.push("prev");
-        opts.push("p");
+        opts.extend(["next", "n", "prev", "p"]);
 
         writeln!(
             io::stdout(),
@@ -602,17 +599,9 @@ fn create_config(
         .context("failed to ask for choice input")
         .context("failed to select preset")?;
         if selection == "next" || selection == "n" {
-            if page == num_pages.checked_sub(1).unwrap() {
-                page = 0
-            } else {
-                page = page.checked_add(1).unwrap();
-            }
+            page = (page + 1) % num_pages;
         } else if selection == "prev" || selection == "p" {
-            if page == 0 {
-                page = num_pages.checked_sub(1).unwrap();
-            } else {
-                page = page.checked_sub(1).unwrap();
-            }
+            page = (page + num_pages - 1) % num_pages;
         } else {
             preset = selection.parse().expect("selected preset should be valid");
             debug!(?preset, "selected preset");