Add discriminator field to enum variant nodes (#209)

.changeset/poor-waves-matter.md

@@ -0,0 +1,5 @@
+---
+"@metaplex-foundation/kinobi": minor
+---
+
+Add discriminator field to enum variant nodes

src/nodes/typeNodes/EnumEmptyVariantTypeNode.ts

@@ -6,17 +6,23 @@ export interface EnumEmptyVariantTypeNode {
 
   // Data.
   readonly name: MainCaseString;
+  readonly discriminator?: number;
 }
 
 export function enumEmptyVariantTypeNode(
-  name: string
+  name: string,
+  discriminator?: number
 ): EnumEmptyVariantTypeNode {
   if (!name) {
     throw new InvalidKinobiTreeError(
       'EnumEmptyVariantTypeNode must have a name.'
     );
   }
-  return { kind: 'enumEmptyVariantTypeNode', name: mainCase(name) };
+  return {
+    kind: 'enumEmptyVariantTypeNode',
+    name: mainCase(name),
+    discriminator,
+  };
 }
 
 export function enumEmptyVariantTypeNodeFromIdl(

src/nodes/typeNodes/EnumStructVariantTypeNode.ts

@@ -14,17 +14,27 @@ export interface EnumStructVariantTypeNode<
 
   // Data.
   readonly name: MainCaseString;
+  readonly discriminator?: number;
 }
 
 export function enumStructVariantTypeNode<
   TStruct extends NestedTypeNode<StructTypeNode>,
->(name: string, struct: TStruct): EnumStructVariantTypeNode<TStruct> {
+>(
+  name: string,
+  struct: TStruct,
+  discriminator?: number
+): EnumStructVariantTypeNode<TStruct> {
   if (!name) {
     throw new InvalidKinobiTreeError(
       'EnumStructVariantTypeNode must have a name.'
     );
   }
-  return { kind: 'enumStructVariantTypeNode', name: mainCase(name), struct };
+  return {
+    kind: 'enumStructVariantTypeNode',
+    name: mainCase(name),
+    struct,
+    discriminator,
+  };
 }
 
 export function enumStructVariantTypeNodeFromIdl(

src/nodes/typeNodes/EnumTupleVariantTypeNode.ts

@@ -13,17 +13,27 @@ export interface EnumTupleVariantTypeNode<
 
   // Data.
   readonly name: MainCaseString;
+  readonly discriminator?: number;
 }
 
 export function enumTupleVariantTypeNode<
   TTuple extends NestedTypeNode<TupleTypeNode>,
->(name: string, tuple: TTuple): EnumTupleVariantTypeNode<TTuple> {
+>(
+  name: string,
+  tuple: TTuple,
+  discriminator?: number
+): EnumTupleVariantTypeNode<TTuple> {
   if (!name) {
     throw new InvalidKinobiTreeError(
       'EnumTupleVariantTypeNode must have a name.'
     );
   }
-  return { kind: 'enumTupleVariantTypeNode', name: mainCase(name), tuple };
+  return {
+    kind: 'enumTupleVariantTypeNode',
+    name: mainCase(name),
+    tuple,
+    discriminator,
+  };
 }
 
 export function enumTupleVariantTypeNodeFromIdl(