From 6155820b9436acaac824cb02015225ba5de26921 Mon Sep 17 00:00:00 2001
From: Kamen Mladenov <kamen@syndamia.com>
Date: Mon, 3 Feb 2025 12:52:50 +0200
Subject: feat: Move some public input parsing logic to parse_fn

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

(limited to 'guests_macro/src')

diff --git a/guests_macro/src/parse_fn.rs b/guests_macro/src/parse_fn.rs
index 00be250..0e57879 100644
--- a/guests_macro/src/parse_fn.rs
+++ b/guests_macro/src/parse_fn.rs
@@ -129,6 +129,17 @@ pub fn args_divide(item: &TokenStream) -> (Vec<TokenStream>, Vec<TokenStream>) {
     (patterns, types)
 }
 
+/// Input:  "(p1 : t1, p2: t2, ...)"
+/// Output: vec![p1, p2, ...], vec![t1, t2, ...]
+pub fn args_divide_public(item: &TokenStream, public: &Vec<&String>) -> (Vec<TokenStream>, Vec<TokenStream>) {
+    let (patterns, types) = args_divide(item);
+    patterns
+        .into_iter()
+        .zip(types.into_iter())
+        .filter(|(p, _)| public.iter().any(|x| p.to_string() == **x))
+        .unzip()
+}
+
 /// Input:  "(p1 : t1, p2: t2, ...)"
 /// Output: "(p1, p2, ...)", "(t1, t2, ...)"
 pub fn args_divide_grouped(item: &TokenStream) -> (TokenStream, TokenStream) {
-- 
cgit v1.2.3