From d3da081886d53eeb7d4ff51ca7308813b7d6113e Mon Sep 17 00:00:00 2001
From: alemi <me@alemi.dev>
Date: Sat, 21 Sep 2024 18:08:19 +0200
Subject: [PATCH] fix: handle mut modifier in arguments

---
 macro/macro.rs | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/macro/macro.rs b/macro/macro.rs
index d2862eb..edccc56 100644
--- a/macro/macro.rs
+++ b/macro/macro.rs
@@ -117,7 +117,7 @@ fn generate_jni_wrapper(attrs: TokenStream, input: TokenStream) -> Result<TokenS
 			return Err(syn::Error::new(Span::call_site(), "#[jni] macro doesn't work on methods"));
 		};
 		incoming.append_all(quote::quote!( #ty , ));
-		let pat = ty.pat;
+		let pat = unpack_pat(*ty.pat)?;
 		forwarding.append_all(quote::quote!( #pat , ));
 	}
 
@@ -195,3 +195,16 @@ enum WhatNext {
 	Class,
 	Exception,
 }
+
+fn unpack_pat(pat: syn::Pat) -> Result<TokenStream, syn::Error> {
+	match pat {
+		syn::Pat::Ident(i) => {
+			let ident = i.ident;
+			Ok(quote::quote!( #ident ,))
+		},
+		syn::Pat::Reference(r) => {
+			unpack_pat(*r.pat)
+		},
+		_ => Err(syn::Error::new(Span::call_site(), "unsupported argument type")),
+	}
+}