ts: Update types to work for vec and array (coral-xyz#862)

nazreen · Oct 10, 2021 · ee19059 · ee19059
1 parent a845af2
commit ee19059
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/ts/src/program/namespace/types.ts b/ts/src/program/namespace/types.ts
@@ -93,8 +93,10 @@ export type DecodeType<T extends IdlType, Defined> = T extends keyof TypeMap
   ? Defined[T["defined"]]
   : T extends { option: { defined: keyof Defined } }
   ? Defined[T["option"]["defined"]] | null
-  : T extends { vec: { defined: keyof Defined } }
-  ? Defined[T["vec"]["defined"]][]
+  : T extends { vec: keyof TypeMap }
+  ? TypeMap[T["vec"]][]
+  : T extends { array: [defined: keyof TypeMap, size: number] }
+  ? TypeMap[T["array"][0]][]
   : unknown;
 
 /**