add new method for parsing all the order token keys

crates/settings/src/order.rs

@@ -2,7 +2,7 @@ use crate::{yaml::FieldErrorKind, *};
 use alloy::primitives::{private::rand, U256};
 use serde::{Deserialize, Serialize};
 use std::{
-    collections::HashMap,
+    collections::{BTreeSet, HashMap},
     str::FromStr,
     sync::{Arc, RwLock},
 };
@@ -434,6 +434,41 @@ impl OrderCfg {
 
         Ok(vault_ids)
     }
+
+    pub fn parse_io_token_keys(
+        documents: Vec<Arc<RwLock<StrictYaml>>>,
+        order_key: &str,
+    ) -> Result<Vec<String>, YamlError> {
+        let mut token_keys = BTreeSet::new();
+
+        for document in documents {
+            let document_read = document.read().map_err(|_| YamlError::ReadLockError)?;
+
+            if let Ok(orders_hash) = require_hash(&document_read, Some("orders"), None) {
+                if let Some(order_yaml) =
+                    orders_hash.get(&StrictYaml::String(order_key.to_string()))
+                {
+                    let location = format!("order '{}'", order_key);
+
+                    let inputs = require_vec(order_yaml, "inputs", Some(location.clone()))?;
+                    let outputs = require_vec(order_yaml, "outputs", Some(location.clone()))?;
+
+                    for input in inputs {
+                        let token_key =
+                            require_string(input, Some("token"), Some(location.clone()))?;
+                        token_keys.insert(token_key);
+                    }
+                    for output in outputs {
+                        let token_key =
+                            require_string(output, Some("token"), Some(location.clone()))?;
+                        token_keys.insert(token_key);
+                    }
+                }
+            }
+        }
+
+        Ok(token_keys.into_iter().collect())
+    }
 }
 
 impl YamlParsableHash for OrderCfg {

crates/settings/src/yaml/dotrain.rs

@@ -320,6 +320,11 @@ mod tests {
             OrderCfg::parse_vault_ids(dotrain_yaml.documents.clone(), &order.key, false).unwrap();
         assert_eq!(output_vault_ids.len(), 1);
         assert_eq!(output_vault_ids[0], Some("2".to_string()));
+        let io_token_keys =
+            OrderCfg::parse_io_token_keys(dotrain_yaml.documents.clone(), &order.key).unwrap();
+        assert_eq!(io_token_keys.len(), 2);
+        assert_eq!(io_token_keys[0], "token1");
+        assert_eq!(io_token_keys[1], "token2");
 
         let scenario_keys = dotrain_yaml.get_scenario_keys().unwrap();
         assert_eq!(scenario_keys.len(), 3);