Add all files

Author: guibescos

src/PythConnection.ts

@@ -105,7 +105,12 @@ export class PythConnection {
   /** Create a PythConnection that reads its data from an underlying solana web3 connection.
    *  pythProgramKey is the public key of the Pyth program running on the chosen solana cluster.
    */
-  constructor(connection: Connection, pythProgramKey: PublicKey, commitment: Commitment = 'finalized', feedIds?: PublicKey[]) {
+  constructor(
+    connection: Connection,
+    pythProgramKey: PublicKey,
+    commitment: Commitment = 'finalized',
+    feedIds?: PublicKey[],
+  ) {
     this.connection = connection
     this.pythProgramKey = pythProgramKey
     this.commitment = commitment
@@ -124,21 +129,20 @@ export class PythConnection {
       this.handleAccount(account.pubkey, account.account, true, currentSlot)
     }
 
-    if(this.feedIds) {
+    if (this.feedIds) {
       // Filter down to only the feeds we want
       const rawIDs = this.feedIds.map((feed) => feed.toString())
       accounts = accounts.filter((feed) => rawIDs.includes(feed.pubkey.toString()))
-      for (const account of accounts){
+      for (const account of accounts) {
         this.connection.onAccountChange(
-        account.pubkey,
+          account.pubkey,
 
-        (accountInfo, context) => {
-          this.handleAccount(account.pubkey, accountInfo, false, context.slot)
-        },
-        this.commitment,
-          )
+          (accountInfo, context) => {
+            this.handleAccount(account.pubkey, accountInfo, false, context.slot)
+          },
+          this.commitment,
+        )
       }
-
     } else {
       this.connection.onProgramAccountChange(
         this.pythProgramKey,

src/__tests__/Anchor.test.ts

@@ -31,6 +31,28 @@ test('Anchor', (done) => {
       expect(decoded?.name).toBe('addMapping')
       expect(decoded?.data).toStrictEqual({})
     })
+  pythOracle.methods
+    .addProduct({
+      asset_type: 'Crypto',
+      base: 'ETH',
+      description: 'ETH/USD',
+      quote_currency: 'USD',
+      symbol: 'Crypto.ETH/USD',
+      generic_symbol: 'ETHUSD',
+    })
+    .accounts({ fundingAccount: PublicKey.unique(), productAccount: PublicKey.unique(), tailMappingAccount: PublicKey.unique() })
+    .instruction()
+    .then((instruction) => {
+      const decoded = pythOracleCoder().instruction.decode(instruction.data)
+      expect(decoded?.name).toBe('addProduct')
+      expect(decoded?.data.asset_type).toBe('Crypto')
+      expect(decoded?.data.base).toBe('ETH')
+      expect(decoded?.data.description).toBe('ETH/USD')
+      expect(decoded?.data.quote_currency).toBe('USD')
+      expect(decoded?.data.symbol).toBe('Crypto.ETH/USD')
+      expect(decoded?.data.generic_symbol).toBe('ETHUSD')
+    })
+
   pythOracle.methods
     .updProduct({
       asset_type: 'Crypto',

src/anchor/coder/instructions.ts

@@ -13,6 +13,8 @@ import { IdlCoder } from './idl'
 import { InstructionCoder } from '@coral-xyz/anchor'
 import { Product } from '../..'
 
+const MAX_METADATA_SIZE: number = 464
+
 export type PythIdlInstruction = {
   name: string
   docs?: string[]
@@ -88,18 +90,18 @@ export class PythOracleInstructionCoder implements InstructionCoder {
     }
 
     /// updProduct has its own format
-    if (methodName === 'updProduct') {
+    if (methodName === 'updProduct' || methodName === 'addProduct') {
       let offset = 0
       for (const key of Object.keys(ix.productMetadata)) {
         offset += buffer.subarray(offset).writeInt8(key.length)
         offset += buffer.subarray(offset).write(key)
         offset += buffer.subarray(offset).writeInt8(ix.productMetadata[key].length)
         offset += buffer.subarray(offset).write(ix.productMetadata[key])
       }
-      if (offset > 464) {
+      if (offset > MAX_METADATA_SIZE) {
         throw new Error('The metadata is too long')
       }
-      const data = buffer.subarray(0, offset)
+      const data = buffer.subarray(0, MAX_METADATA_SIZE)
       return Buffer.concat([discriminator, data])
     } else {
       const len = layout.encode(ix, buffer)
@@ -135,7 +137,7 @@ export class PythOracleInstructionCoder implements InstructionCoder {
     }
 
     /// updProduct has its own format
-    if (decoder.name === 'updProduct') {
+    if (decoder.name === 'updProduct' || decoder.name === 'addProduct') {
       const product: Product = {}
       let idx = 0
       while (idx < data.length) {

src/anchor/idl.json

@@ -103,7 +103,14 @@
           }
         }
       ],
-      "args": []
+      "args": [
+        {
+          "name": "productMetadata",
+          "type": {
+            "defined": "ProductMetadata"
+          }
+        }
+      ]
     },
     {
       "name": "updProduct",

src/anchor/program.ts

@@ -123,7 +123,14 @@ export type PythOracle = {
           }
         },
       ]
-      args: []
+      args: [
+        {
+          name: 'productMetadata'
+          type: {
+            defined: 'ProductMetadata'
+          }
+        },
+      ]
     },
     {
       name: 'updProduct'

src/example_ws_single_feed.ts

@@ -9,7 +9,7 @@ const pythPublicKey = getPythProgramKeyForCluster(PYTHNET_CLUSTER_NAME)
 // This example shows Crypto.SOL/USD
 const feeds = [new PublicKey('H6ARHf6YXhGYeQfUzQNGk6rDNnLBQKrenN712K4AQJEG')]
 
-const pythConnection = new PythConnection(connection, pythPublicKey, "confirmed", feeds)
+const pythConnection = new PythConnection(connection, pythPublicKey, 'confirmed', feeds)
 pythConnection.onPriceChangeVerbose((productAccount, priceAccount) => {
   // The arguments to the callback include solana account information / the update slot if you need it.
   const product = productAccount.accountInfo.data.product