From 0e2db3d20ccae2090f4a01d172e451c596da0c30 Mon Sep 17 00:00:00 2001
From: Kamen Mladenov <kamen@syndamia.com>
Date: Mon, 3 Feb 2025 14:52:56 +0200
Subject: feat(zkvms_host_io): Add public_inputs and private_inputs attributes
 to RunWith

---
 guests_macro/src/parse_fn.rs | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'guests_macro/src')

diff --git a/guests_macro/src/parse_fn.rs b/guests_macro/src/parse_fn.rs
index 0e57879..e5558ad 100644
--- a/guests_macro/src/parse_fn.rs
+++ b/guests_macro/src/parse_fn.rs
@@ -78,6 +78,15 @@ pub fn args_split(item: &TokenStream) -> Vec<TokenStream> {
     args
 }
 
+/// Input:  "(p1 : t1, p2: t2, ...)"
+/// Output: vec!["p1 : t1", "p2: t2", ...]
+pub fn args_split_public(item: &TokenStream, public: &Vec<&String>) -> Vec<TokenStream> {
+    args_split(item)
+        .into_iter()
+        .filter(|a| public.iter().any(|x| a.to_string().starts_with(*x)))
+        .collect()
+}
+
 /// Input:  "(p1 : t1, p2: t2, ...)"
 /// Output: vec![p1, p2, ...], vec![t1, t2, ...]
 pub fn args_divide(item: &TokenStream) -> (Vec<TokenStream>, Vec<TokenStream>) {
-- 
cgit v1.2.3