Add copy derive to scalar enums (#45)

* Add copy derive

* Add changeset

* Rename helper

Author: febo

.changeset/tall-beds-share.md

@@ -0,0 +1,5 @@
+---
+"@kinobi-so/renderers-rust": patch
+---
+
+Add copy derive to scalar enums in Rust renderer

packages/renderers-rust/e2e/system/src/generated/types/nonce_state.rs

@@ -10,7 +10,16 @@ use borsh::BorshSerialize;
 use num_derive::FromPrimitive;
 
 #[derive(
-    BorshSerialize, BorshDeserialize, Clone, Debug, Eq, PartialEq, PartialOrd, Hash, FromPrimitive,
+    BorshSerialize,
+    BorshDeserialize,
+    Clone,
+    Debug,
+    Eq,
+    PartialEq,
+    Copy,
+    PartialOrd,
+    Hash,
+    FromPrimitive,
 )]
 #[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
 pub enum NonceState {

packages/renderers-rust/e2e/system/src/generated/types/nonce_version.rs

@@ -10,7 +10,16 @@ use borsh::BorshSerialize;
 use num_derive::FromPrimitive;
 
 #[derive(
-    BorshSerialize, BorshDeserialize, Clone, Debug, Eq, PartialEq, PartialOrd, Hash, FromPrimitive,
+    BorshSerialize,
+    BorshDeserialize,
+    Clone,
+    Debug,
+    Eq,
+    PartialEq,
+    Copy,
+    PartialOrd,
+    Hash,
+    FromPrimitive,
 )]
 #[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
 pub enum NonceVersion {

packages/renderers-rust/src/getTypeManifestVisitor.ts

@@ -132,7 +132,7 @@ export function getTypeManifestVisitor(options: { nestedStruct?: boolean; parent
                     manifest.imports.add(['borsh::BorshSerialize', 'borsh::BorshDeserialize']);
                     const traits = ['BorshSerialize', 'BorshDeserialize', 'Clone', 'Debug', 'Eq', 'PartialEq'];
                     if (isNode(definedType.type, 'enumTypeNode') && isScalarEnum(definedType.type)) {
-                        traits.push('PartialOrd', 'Hash', 'FromPrimitive');
+                        traits.push('Copy', 'PartialOrd', 'Hash', 'FromPrimitive');
                         manifest.imports.add(['num_derive::FromPrimitive']);
                     }
                     return {

packages/renderers-rust/test/_setup.ts

@@ -10,3 +10,14 @@ export function codeContains(actual: string, expected: (RegExp | string)[] | Reg
         }
     });
 }
+
+export function codeDoesNotContains(actual: string, expected: (RegExp | string)[] | RegExp | string) {
+    const expectedArray = Array.isArray(expected) ? expected : [expected];
+    expectedArray.forEach(e => {
+        if (typeof e === 'string') {
+            expect(actual).not.toContain(e);
+        } else {
+            expect(actual).not.toMatch(e);
+        }
+    });
+}

packages/renderers-rust/test/definedTypesPage.test.ts

@@ -1,5 +1,8 @@
 import {
     definedTypeNode,
+    enumEmptyVariantTypeNode,
+    enumStructVariantTypeNode,
+    enumTypeNode,
     numberTypeNode,
     programNode,
     sizePrefixTypeNode,
@@ -11,7 +14,7 @@ import { visit } from '@kinobi-so/visitors-core';
 import { test } from 'vitest';
 
 import { getRenderMapVisitor } from '../src';
-import { codeContains } from './_setup';
+import { codeContains, codeDoesNotContains } from './_setup';
 
 test('it renders a prefix string on a defined type', () => {
     // Given the following program with 1 defined type using a prefixed size string.
@@ -40,3 +43,56 @@ test('it renders a prefix string on a defined type', () => {
         `content_type: U8PrefixString,`,
     ]);
 });
+
+test('it renders a scalar enum with Copy derive', () => {
+    // Given the following program with 1 defined type using a prefixed size string.
+    const node = programNode({
+        definedTypes: [
+            definedTypeNode({
+                name: 'tag',
+                type: enumTypeNode([enumEmptyVariantTypeNode('Uninitialized'), enumEmptyVariantTypeNode('Account')]),
+            }),
+        ],
+        name: 'splToken',
+        publicKey: 'TokenkegQfeZyiNwAJbNbGKPFXCWuBvf9Ss623VQ5DA',
+    });
+
+    // When we render it.
+    const renderMap = visit(node, getRenderMapVisitor());
+
+    // Then we expect the following use and identifier to be rendered.
+    codeContains(renderMap.get('types/tag.rs'), [`#[derive(`, `Copy`, `pub enum Tag`]);
+});
+
+test('it renders a non-scalar enum without Copy derive', () => {
+    // Given the following program with 1 defined type using a prefixed size string.
+    const node = programNode({
+        definedTypes: [
+            definedTypeNode({
+                name: 'tagWithStruct',
+                type: enumTypeNode([
+                    enumEmptyVariantTypeNode('Uninitialized'),
+                    enumStructVariantTypeNode(
+                        'Account',
+                        structTypeNode([
+                            structFieldTypeNode({
+                                name: 'contentType',
+                                type: sizePrefixTypeNode(stringTypeNode('utf8'), numberTypeNode('u8')),
+                            }),
+                        ]),
+                    ),
+                ]),
+            }),
+        ],
+        name: 'splToken',
+        publicKey: 'TokenkegQfeZyiNwAJbNbGKPFXCWuBvf9Ss623VQ5DA',
+    });
+
+    // When we render it.
+    const renderMap = visit(node, getRenderMapVisitor());
+
+    // Then we expect the following use and identifier to be rendered.
+    codeContains(renderMap.get('types/tag_with_struct.rs'), [`#[derive(`, `pub enum TagWithStruct`]);
+    // And we expect the Copy derive to be missing.
+    codeDoesNotContains(renderMap.get('types/tag_with_struct.rs'), `Copy`);
+});