From d4aac2c729d3b038ee851b6fb742d3d765329d08 Mon Sep 17 00:00:00 2001
From: Jeff <dev@jeffa.io>
Date: Tue, 14 Nov 2023 21:24:47 -0500
Subject: [PATCH] Implement parallel find loop

---
 src/abstract_tree/find.rs       | 40 ++++++++++++++++++++++++---------
 src/abstract_tree/identifier.rs |  2 +-
 2 files changed, 30 insertions(+), 12 deletions(-)

diff --git a/src/abstract_tree/find.rs b/src/abstract_tree/find.rs
index 09d3d59..ea4ad5b 100644
--- a/src/abstract_tree/find.rs
+++ b/src/abstract_tree/find.rs
@@ -1,7 +1,8 @@
+use rayon::prelude::*;
 use serde::{Deserialize, Serialize};
 use tree_sitter::Node;
 
-use crate::{AbstractTree, Block, Expression, Identifier, Map, Result, Value};
+use crate::{AbstractTree, Block, Error, Expression, Identifier, Map, Result, Value};
 
 #[derive(Debug, Clone, Serialize, Deserialize, Eq, PartialEq, PartialOrd, Ord)]
 pub struct Find {
@@ -32,20 +33,37 @@ impl AbstractTree for Find {
         let value = self.expression.run(source, context)?;
         let values = value.as_list()?.items();
         let key = self.identifier.inner();
+        let loop_context = Map::clone_from(context)?;
 
-        let mut loop_context = Map::clone_from(context)?;
-        let mut variables = context.variables_mut()?;
+        let find_result = values.par_iter().find_map_first(|value| {
+            loop_context
+                .variables_mut()
+                .unwrap()
+                .insert(key.clone(), (*value).clone());
 
-        for value in values.iter() {
-            variables.insert(key.clone(), value.clone());
+            let run_result = self.item.run(source, &mut loop_context.clone());
 
-            let should_return = self.item.run(source, &mut loop_context)?.as_boolean()?;
-
-            if should_return {
-                return Ok(value.clone());
+            if let Ok(run_result_value) = run_result {
+                if let Ok(should_return) = run_result_value.as_boolean() {
+                    if should_return {
+                        Some(Ok(value.clone()))
+                    } else {
+                        None
+                    }
+                } else {
+                    Some(Err(Error::ExpectedBoolean {
+                        actual: value.clone(),
+                    }))
+                }
+            } else {
+                Some(run_result)
             }
-        }
+        });
 
-        Ok(Value::Empty)
+        if let Some(result) = find_result {
+            result
+        } else {
+            Ok(Value::Empty)
+        }
     }
 }
diff --git a/src/abstract_tree/identifier.rs b/src/abstract_tree/identifier.rs
index d26a675..15689da 100644
--- a/src/abstract_tree/identifier.rs
+++ b/src/abstract_tree/identifier.rs
@@ -22,7 +22,7 @@ impl Identifier {
 
 impl AbstractTree for Identifier {
     fn from_syntax_node(source: &str, node: Node) -> Result<Self> {
-        debug_assert_eq!("identifier", node.kind());
+        Error::expect_syntax_node(source, "identifier", node)?;
 
         let identifier = &source[node.byte_range()];