fix(functions-tools): forward needsJsonData when wrapping composite type in opt

Author: peterpeterparker

packages/functions-tools/src/converters/zod-to-rust.ts

@@ -103,7 +103,7 @@ const schemaToRustType = ({
       const inner = schemaToRustType({schema: schema.inner, structName});
       const fieldType = `Option<${inner.fieldType}>`;
       return inner.kind === 'composite'
-        ? composite({fieldType, structs: inner.structs})
+        ? composite({fieldType, structs: inner.structs, needsJsonData: inner.needsJsonData})
         : primitive({fieldType});
     }
 

packages/functions-tools/src/tests/converters/zod-to-rust.spec.ts

@@ -87,6 +87,15 @@ describe('object with enum field', () => {
       '#[derive(CandidType, Serialize, Deserialize, Clone, JsonData)]\npub struct MyFunctionArgs {\n    pub status: MyFunctionArgsStatus,\n}'
     ].join('\n\n')
   );
+
+  rust(
+    'myFunction',
+    z.object({role: z.enum(['admin', 'user']).optional()}),
+    [
+      '#[derive(CandidType, Serialize, Deserialize, Clone)]\npub enum MyFunctionArgsRole {\n    #[serde(rename = "admin")]\n    Admin,\n    #[serde(rename = "user")]\n    User,\n}',
+      '#[derive(CandidType, Serialize, Deserialize, Clone, JsonData)]\npub struct MyFunctionArgs {\n    pub role: Option<MyFunctionArgsRole>,\n}'
+    ].join('\n\n')
+  );
 });
 
 // ─── Arrays ───────────────────────────────────────────────────────────────────