Adding derive for FromPrimitive to enums. (#199)

metaplex-foundation · Apr 10, 2024 · dc85f39 · dc85f39
1 parent 75f5f91
commit dc85f39
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/.changeset/grumpy-suits-tap.md b/.changeset/grumpy-suits-tap.md
@@ -0,0 +1,5 @@
+---
+"@metaplex-foundation/kinobi": patch
+---
+
+Adding derive for FromPrimitive to enums.
diff --git a/src/renderers/rust/getTypeManifestVisitor.ts b/src/renderers/rust/getTypeManifestVisitor.ts
@@ -80,7 +80,8 @@ export function getTypeManifestVisitor() {
             isNode(definedType.type, 'enumTypeNode') &&
             isScalarEnum(definedType.type)
           ) {
-            traits.push('PartialOrd', 'Hash');
+            traits.push('PartialOrd', 'Hash', 'FromPrimitive');
+            manifest.imports.add(['num_derive::FromPrimitive']);
           }
           return {
             ...manifest,