Convert single-element lists to integers when an integer is required

This allows an integer invocation with arguments to be wrapped in `[]`
square brackets so that it can be passed to another invocation as an
argument.

1 files changed, 5 insertions, 2 deletions

diff --git a/src/stages/intermediate.rs b/src/stages/intermediate.rs
index d630cc2..c5be44f 100644
--- a/src/stages/intermediate.rs
+++ b/src/stages/intermediate.rs
@@ -321,8 +321,11 @@ impl IntermediateParser {
         let result = self.parse_invocation(invocation, source)?;
         match result.value {
             IntermediateValue::Integer(integer) => {
-                let source = integer.source.clone();
-                Some(Tracked::from(integer, source))
+                Some(Tracked::from(integer, source.clone()))
+            }
+            IntermediateValue::List(mut integers) if integers.len() == 1 => {
+                let integer = integers.pop().unwrap();
+                Some(Tracked::from(integer, source.clone()))
             }
             IntermediateValue::Block(_) | IntermediateValue::List(_) => {
                 let error = IntermediateError::ExpectedInteger;