-
Notifications
You must be signed in to change notification settings - Fork 1.3k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Build and link new SPL packages before tests and decode token account…
… for AccountStore
- Loading branch information
1 parent
83fb65a
commit 57f0f7e
Showing
4 changed files
with
171 additions
and
9 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
|
|
@@ -6,4 +6,3 @@ for D in */; | |
| cd $D && yarn init:yarn; cd ..; | ||
| fi | ||
| done | ||
|
|
||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,146 @@ | ||
| import BN from "bn.js"; | ||
| import * as BufferLayout from "buffer-layout"; | ||
| import { Layout } from "buffer-layout"; | ||
| import { PublicKey } from "@solana/web3.js"; | ||
|
|
||
| function uint64(property?: string): Layout<u64 | null> { | ||
| return new WrappedLayout( | ||
| BufferLayout.blob(8), | ||
| (b: Buffer) => u64.fromBuffer(b), | ||
| (n: u64) => n.toBuffer(), | ||
| property | ||
| ); | ||
| } | ||
|
|
||
| function publicKey(property?: string): Layout<PublicKey> { | ||
| return new WrappedLayout( | ||
| BufferLayout.blob(32), | ||
| (b: Buffer) => new PublicKey(b), | ||
| (key: PublicKey) => key.toBuffer(), | ||
| property | ||
| ); | ||
| } | ||
|
|
||
| function coption<T>(layout: Layout<T>, property?: string): Layout<T | null> { | ||
| return new COptionLayout<T>(layout, property); | ||
| } | ||
|
|
||
| class WrappedLayout<T, U> extends Layout<U> { | ||
| layout: Layout<T>; | ||
| decoder: (data: T) => U; | ||
| encoder: (src: U) => T; | ||
|
|
||
| constructor( | ||
| layout: Layout<T>, | ||
| decoder: (data: T) => U, | ||
| encoder: (src: U) => T, | ||
| property?: string | ||
| ) { | ||
| super(layout.span, property); | ||
| this.layout = layout; | ||
| this.decoder = decoder; | ||
| this.encoder = encoder; | ||
| } | ||
|
|
||
| decode(b: Buffer, offset?: number): U { | ||
| return this.decoder(this.layout.decode(b, offset)); | ||
| } | ||
|
|
||
| encode(src: U, b: Buffer, offset?: number): number { | ||
| return this.layout.encode(this.encoder(src), b, offset); | ||
| } | ||
|
|
||
| getSpan(b: Buffer, offset?: number): number { | ||
| return this.layout.getSpan(b, offset); | ||
| } | ||
| } | ||
|
|
||
| class COptionLayout<T> extends Layout<T | null> { | ||
| layout: Layout<T>; | ||
| discriminator: Layout<number>; | ||
|
|
||
| constructor(layout: Layout<T>, property?: string) { | ||
| super(-1, property); | ||
| this.layout = layout; | ||
| this.discriminator = BufferLayout.u32(); | ||
| } | ||
|
|
||
| encode(src: T | null, b: Buffer, offset = 0): number { | ||
| if (src === null || src === undefined) { | ||
| return this.layout.span + this.discriminator.encode(0, b, offset); | ||
| } | ||
| this.discriminator.encode(1, b, offset); | ||
| return this.layout.encode(src, b, offset + 4) + 4; | ||
| } | ||
|
|
||
| decode(b: Buffer, offset = 0): T | null { | ||
| const discriminator = this.discriminator.decode(b, offset); | ||
| if (discriminator === 0) { | ||
| return null; | ||
| } else if (discriminator === 1) { | ||
| return this.layout.decode(b, offset + 4); | ||
| } | ||
| throw new Error("Invalid coption " + this.layout.property); | ||
| } | ||
|
|
||
| getSpan(b: Buffer, offset = 0): number { | ||
| return this.layout.getSpan(b, offset + 4) + 4; | ||
| } | ||
| } | ||
|
|
||
| class u64 extends BN { | ||
| /** | ||
| * Convert to Buffer representation | ||
| */ | ||
| toBuffer(): Buffer { | ||
| const a = super.toArray().reverse(); | ||
| const b = Buffer.from(a); | ||
| if (length === 8) { | ||
| return b; | ||
| } | ||
| if (length >= 8) { | ||
| throw new Error("u64 too large"); | ||
| } | ||
|
|
||
| const zeroPad = Buffer.alloc(8); | ||
| b.copy(zeroPad); | ||
| return zeroPad; | ||
| } | ||
|
|
||
| /** | ||
| * Construct a u64 from Buffer representation | ||
| */ | ||
| static fromBuffer(buffer: Buffer): u64 { | ||
| if (buffer.length !== 8) { | ||
| throw new Error(`Invalid buffer length: ${buffer.length}`); | ||
| } | ||
| return new u64( | ||
| [...buffer] | ||
| .reverse() | ||
| .map((i) => `00${i.toString(16)}`.slice(-2)) | ||
| .join(""), | ||
| 16 | ||
| ); | ||
| } | ||
| } | ||
|
|
||
| const TOKEN_ACCOUNT_LAYOUT = BufferLayout.struct([ | ||
| publicKey("mint"), | ||
| publicKey("owner"), | ||
| uint64("amount"), | ||
| coption(publicKey(), "delegate"), | ||
| ((p: string) => { | ||
| const U = BufferLayout.union(BufferLayout.u8("discriminator"), null, p); | ||
| U.addVariant(0, BufferLayout.struct([]), "uninitialized"); | ||
| U.addVariant(1, BufferLayout.struct([]), "initialized"); | ||
| U.addVariant(2, BufferLayout.struct([]), "frozen"); | ||
| return U; | ||
| })("state"), | ||
| coption(uint64(), "isNative"), | ||
| uint64("delegatedAmount"), | ||
| coption(publicKey(), "closeAuthority"), | ||
| ]); | ||
|
|
||
| export function decodeTokenAccount(b: Buffer) { | ||
| return TOKEN_ACCOUNT_LAYOUT.decode(b); | ||
| } |