borsh: Add discriminant param for rust enum

project-serum · Jan 28, 2022 · 1ff0fc7 · 1ff0fc7
1 parent 4cdea04
commit 1ff0fc7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/packages/borsh/package.json b/packages/borsh/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@project-serum/borsh",
-  "version": "0.2.2",
+  "version": "0.2.4",
   "description": "Serum Borsh",
   "main": "dist/lib/index.js",
   "types": "dist/lib/index.d.ts",

diff --git a/packages/borsh/src/index.ts b/packages/borsh/src/index.ts
@@ -258,8 +258,9 @@ export interface EnumLayout<T> extends Layout<T> {
 export function rustEnum<T>(
   variants: Layout<any>[],
   property?: string,
+  discriminant?: Layout<any>,
 ): EnumLayout<T> {
-  const unionLayout = union(u8(), property);
+  const unionLayout = union(discriminant ?? u8(), property);
   variants.forEach((variant, index) =>
     unionLayout.addVariant(index, variant, variant.property),
   );