Add shortU16 number format (#218)

metaplex-foundation · Apr 17, 2024 · 69d2698 · 69d2698
1 parent 2b36e8b
commit 69d2698
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 2 deletions.
diff --git a/.changeset/cold-suns-fail.md b/.changeset/cold-suns-fail.md
@@ -0,0 +1,5 @@
+---
+"@metaplex-foundation/kinobi": patch
+---
+
+Add shortU16 number format
diff --git a/src/idl/IdlType.ts b/src/idl/IdlType.ts
@@ -77,6 +77,7 @@ export const IDL_TYPE_LEAVES = [
   'i128',
   'f32',
   'f64',
+  'shortU16',
 ] as const;
 export type IdlTypeLeaf = (typeof IDL_TYPE_LEAVES)[number];
 

diff --git a/src/nodes/typeNodes/NumberTypeNode.ts b/src/nodes/typeNodes/NumberTypeNode.ts
@@ -10,7 +10,8 @@ export type NumberFormat =
   | 'i64'
   | 'i128'
   | 'f32'
-  | 'f64';
+  | 'f64'
+  | 'shortU16';
 
 export interface NumberTypeNode<TFormat extends NumberFormat = NumberFormat> {
   readonly kind: 'numberTypeNode';
@@ -32,7 +33,7 @@ export function isSignedInteger(node: NumberTypeNode): boolean {
 }
 
 export function isUnsignedInteger(node: NumberTypeNode): boolean {
-  return node.format.startsWith('u');
+  return node.format.startsWith('u') || node.format === 'shortU16';
 }
 
 export function isInteger(node: NumberTypeNode): boolean {

diff --git a/src/renderers/rust/getTypeManifestVisitor.ts b/src/renderers/rust/getTypeManifestVisitor.ts
@@ -389,6 +389,12 @@ export function getTypeManifestVisitor() {
         },
 
         visitNumberType(numberType) {
+          if (numberType.format === 'shortU16') {
+            throw new Error(
+              'shortU16 numbers are not supported by the Rust renderer'
+            );
+          }
+
           if (numberType.endian === 'le') {
             return {
               type: numberType.format,