From f174f364a4058091f42ca6e5e1bea598c64edd80 Mon Sep 17 00:00:00 2001
From: John Doty <john@d0ty.me>
Date: Tue, 21 Mar 2023 19:45:17 -0700
Subject: [PATCH] Re-add assertions and simplify wrapping

---
 src/client/ui.rs | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/src/client/ui.rs b/src/client/ui.rs
index 610baff..63bc79b 100644
--- a/src/client/ui.rs
+++ b/src/client/ui.rs
@@ -386,6 +386,7 @@ impl UI {
                 | KeyEvent { code: KeyCode::Char('k'), .. } => {
                     let index = match self.selection.selected() {
                         Some(i) => {
+                            assert!(self.ports.len() > 0, "We must have ports because we have a selection.");
                             if i == 0 {
                                 self.ports.len() - 1
                             } else {
@@ -400,12 +401,8 @@ impl UI {
                 | KeyEvent { code: KeyCode::Char('j'), .. } => {
                     let index = match self.selection.selected() {
                         Some(i) => {
-                            let max = self.ports.len() - 1;
-                            if i >= max {
-                                0
-                            } else {
-                                i + 1
-                            }
+                            assert!(self.ports.len() > 0, "We must have ports because we have a selection.");
+                            (i + 1) % self.ports.len()
                         }
                         None => 0,
                     };