Add orThrow and orDefault methods to Index

api/src/main/java/net/kyori/adventure/util/Index.java

@@ -29,6 +29,7 @@
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
+import java.util.NoSuchElementException;
 import java.util.Set;
 import java.util.function.Function;
 import java.util.function.IntFunction;
@@ -154,6 +155,22 @@ private Index(final Map<K, V> keyToValue, final Map<V, K> valueToKey) {
     return this.valueToKey.get(value);
   }
 
+  /**
+   * Gets the key for a value or throws an exception.
+   *
+   * @param value the value
+   * @return the key
+   * @throws NoSuchElementException if there is no key for the value
+   * @since 4.11.0
+   */
+  public @NotNull K keyOrThrow(final @NotNull V value) {
+    final K key = this.key(value);
+    if (key == null) {
+      throw new NoSuchElementException("There is no key for value " + value);
+    }
+    return key;
+  }
+
   /**
    * Gets the keys.
    *
@@ -175,6 +192,22 @@ private Index(final Map<K, V> keyToValue, final Map<V, K> valueToKey) {
     return this.keyToValue.get(key);
   }
 
+  /**
+   * Gets a value by its key.
+   *
+   * @param key the key
+   * @return the value
+   * @throws NoSuchElementException if there is no value for the key
+   * @since 4.11.0
+   */
+  public @NotNull V valueOrThrow(final @NotNull K key) {
+    final V value = this.value(key);
+    if (value == null) {
+      throw new NoSuchElementException("There is no value for key " + key);
+    }
+    return value;
+  }
+
   /**
    * Get an unmodifiable mapping of index entries from key to value.
    *

api/src/test/java/net/kyori/adventure/util/IndexTest.java

@@ -23,6 +23,7 @@
  */
 package net.kyori.adventure.util;
 
+import java.util.NoSuchElementException;
 import java.util.UUID;
 import org.junit.jupiter.api.Test;
 
@@ -31,7 +32,7 @@
 import static org.junit.jupiter.api.Assertions.assertThrows;
 
 class IndexTest {
-  private static final Index<String, Thing> THINGS = Index.create(Thing.class, thing -> thing.name);
+  private static final Index<String, Thing> THINGS = Index.create(Thing.class, thing -> thing.name, Thing.ABC, Thing.DEF);
 
   @Test
   void testCreateWithNonUniqueKey() {
@@ -46,17 +47,29 @@ void testCreateWithNonUniqueValue() {
   @Test
   void testKey() {
     for (final Thing thing : Thing.values()) {
+      if (thing == Thing.NOT_PRESENT) continue;
       assertEquals(thing.name, THINGS.key(thing));
     }
   }
 
   @Test
   void testValue() {
     for (final Thing thing : Thing.values()) {
+      if (thing == Thing.NOT_PRESENT) continue;
       assertEquals(thing, THINGS.value(thing.name));
     }
   }
 
+  @Test
+  void testNoValue() {
+    assertThrows(NoSuchElementException.class, () -> THINGS.valueOrThrow("__NO_VALUE__"));
+  }
+
+  @Test
+  void testNoKey() {
+    assertThrows(NoSuchElementException.class, () -> THINGS.keyOrThrow(Thing.NOT_PRESENT));
+  }
+
   @Test
   void testKeys() {
     assertThat(THINGS.keys()).containsExactly("abc", "def");
@@ -69,7 +82,8 @@ void testValues() {
 
   private enum Thing {
     ABC("abc"),
-    DEF("def");
+    DEF("def"),
+    NOT_PRESENT("not_present");
 
     private final String name;