Add selection wrapping (-w).

3 files changed, 53 insertions, 2 deletions

diff --git a/lib/bemenu.h b/lib/bemenu.h
index 172c6ed..5457cca 100644
--- a/lib/bemenu.h
+++ b/lib/bemenu.h
@@ -203,6 +203,22 @@ void bmMenuSetFilterMode(bmMenu *menu, bmFilterMode mode);
 bmFilterMode bmMenuGetFilterMode(const bmMenu *menu);
 
 /**
+ * Set selection wrapping on/off.
+ *
+ * @param menu bmMenu instance where to toggle selection wrapping.
+ * @param int 1 == on, 0 == off.
+ */
+void bmMenuSetWrap(bmMenu *menu, int wrap);
+
+/**
+ * Get selection wrapping state.
+ *
+ * @param menu bmMenu instance where to get selection wrapping state.
+ * @return int for wrap state.
+ */
+int bmMenuGetWrap(const bmMenu *menu);
+
+/**
  * Set title to bmMenu instance.
  *
  * @param menu bmMenu instance where to set title.
diff --git a/lib/internal.h b/lib/internal.h
index 4f6afbe..2fc9393 100644
--- a/lib/internal.h
+++ b/lib/internal.h
@@ -141,6 +141,11 @@ struct _bmMenu {
      * Drawing mode used in menu instance.
      */
     bmDrawMode drawMode;
+
+    /**
+     * Should selection be wrapped?
+     */
+    char wrap;
 };
 
 /* draw/curses.c */
diff --git a/lib/menu.c b/lib/menu.c
index 5d2b37b..d949430 100644
--- a/lib/menu.c
+++ b/lib/menu.c
@@ -173,6 +173,30 @@ bmFilterMode bmMenuGetFilterMode(const bmMenu *menu)
 }
 
 /**
+ * Set selection wrapping on/off.
+ *
+ * @param menu bmMenu instance where to toggle selection wrapping.
+ * @param int 1 == on, 0 == off.
+ */
+void bmMenuSetWrap(bmMenu *menu, int wrap)
+{
+    assert(menu);
+    menu->wrap = (wrap ? 1 : 0);
+}
+
+/**
+ * Get selection wrapping state.
+ *
+ * @param menu bmMenu instance where to get selection wrapping state.
+ * @return int for wrap state.
+ */
+int bmMenuGetWrap(const bmMenu *menu)
+{
+    assert(menu);
+    return menu->wrap;
+}
+
+/**
  * Set title to bmMenu instance.
  *
  * @param menu bmMenu instance where to set title.
@@ -572,13 +596,19 @@ bmRunResult bmMenuRunWithKey(bmMenu *menu, bmKey key, unsigned int unicode)
             break;
 
         case BM_KEY_UP:
-            if (menu->index > 0)
+            if (menu->index > 0) {
                 menu->index--;
+            } else if (menu->wrap) {
+                menu->index = itemsCount - 1;
+            }
             break;
 
         case BM_KEY_DOWN:
-            if (menu->index < itemsCount - 1)
+            if (menu->index < itemsCount - 1) {
                 menu->index++;
+            } else if (menu->wrap) {
+                menu->index = 0;
+            }
             break;
 
         case BM_KEY_PAGE_UP: