From e55c5688a82541edf629a1f62df40c54d4cfef64 Mon Sep 17 00:00:00 2001
From: Jeff <dev@jeffa.io>
Date: Thu, 24 Aug 2023 03:50:23 -0400
Subject: [PATCH] Add type checks

---
 src/tools/random.rs | 18 +++++++++++-------
 1 file changed, 11 insertions(+), 7 deletions(-)

diff --git a/src/tools/random.rs b/src/tools/random.rs
index d3e942d..9fe4535 100644
--- a/src/tools/random.rs
+++ b/src/tools/random.rs
@@ -2,7 +2,7 @@ use std::convert::TryInto;
 
 use rand::{random, thread_rng, Rng};
 
-use crate::{Error, Result, Tool, ToolInfo, Value};
+use crate::{Error, Result, Tool, ToolInfo, Value, ValueType};
 
 pub struct RandomBoolean;
 
@@ -12,7 +12,7 @@ impl Tool for RandomBoolean {
             identifier: "random_boolean",
             description: "Create a random boolean.",
             group: "random",
-            inputs: vec![],
+            inputs: vec![ValueType::Empty],
         }
     }
 
@@ -33,7 +33,11 @@ impl Tool for RandomInteger {
             identifier: "random_integer",
             description: "Create a random integer.",
             group: "random",
-            inputs: vec![],
+            inputs: vec![
+                ValueType::Empty,
+                ValueType::Integer,
+                ValueType::ListExact(vec![ValueType::Integer, ValueType::Integer]),
+            ],
         }
     }
 
@@ -47,14 +51,14 @@ impl Tool for RandomInteger {
             Value::List(min_max) => {
                 Error::expect_function_argument_amount(self.info().identifier, min_max.len(), 2)?;
 
-                let min = min_max.get(0).unwrap().as_int()?;
-                let max = min_max.get(1).unwrap().as_int()? + 1;
-                let integer = rand::thread_rng().gen_range(min..max);
+                let min = min_max[0].as_int()?;
+                let max = min_max[1].as_int()?;
+                let integer = rand::thread_rng().gen_range(min..=max);
 
                 Ok(Value::Integer(integer))
             }
             Value::Empty => Ok(crate::Value::Integer(random())),
-            _ => todo!(),
+            value => Err(Error::expected_empty(value.clone())),
         }
     }
 }