Add TomlTableWithSource.getLong()

1 files changed, 23 insertions, 0 deletions

diff --git a/src/main/java/org/shedlang/compiler/config/toml/TomlTableWithSource.java b/src/main/java/org/shedlang/compiler/config/toml/TomlTableWithSource.java
index 4dab6e35..725d510a 100644
--- a/src/main/java/org/shedlang/compiler/config/toml/TomlTableWithSource.java
+++ b/src/main/java/org/shedlang/compiler/config/toml/TomlTableWithSource.java
@@ -148,4 +148,27 @@ public record TomlTableWithSource(
             TomlSource.sourceAtTomlPosition(fullSource, valuePosition)
         );
     }
+
+    public long getLong(String key) {
+        var value = table.get(key);
+        var valuePosition = table.inputPositionOf(key);
+
+        if (value == null) {
+            throw new ConfigError(
+                MessageFormat.format("{0} is missing {1}", path, key),
+                TomlSource.sourceAtTomlPosition(fullSource, position)
+            );
+        }
+
+        if (value instanceof Long valueLong) {
+            return valueLong;
+        }
+
+        throw new UnexpectedTypeConfigError(
+            path.plusKey(key),
+            "a long",
+            value.getClass(),
+            TomlSource.sourceAtTomlPosition(fullSource, valuePosition)
+        );
+    }
 }