Merge pull request #48 from pyth-network/guibescos/add-permission-account

Add permission account

Author: guibescos

src/PythConnection.ts

@@ -93,6 +93,7 @@ export class PythConnection {
           }
           break
         case AccountType.Test:
+        case AccountType.Permission:
           break
         default:
           throw new Error(`Unknown account type: ${base.type}. Try upgrading pyth-client.`)

src/PythHttpClient.ts

@@ -1,5 +1,14 @@
 import { Commitment, Connection, PublicKey } from '@solana/web3.js'
-import { Product, PriceData, parseProductData, parsePriceData, parseBaseData, AccountType } from '.'
+import {
+  Product,
+  PriceData,
+  parseProductData,
+  parsePriceData,
+  parseBaseData,
+  AccountType,
+  parsePermissionData,
+  PermissionData,
+} from '.'
 
 export interface PythHttpClientResult {
   assetTypes: string[]
@@ -11,6 +20,7 @@ export interface PythHttpClientResult {
   /** The current price of each product. */
   productPrice: Map<string, PriceData>
   prices: PriceData[]
+  permissionData: PermissionData | undefined
 }
 
 /**
@@ -48,6 +58,9 @@ export class PythHttpClient {
     const productAccountKeyToProduct = new Map<string, Product>()
     const currentSlot = await this.connection.getSlot(this.commitment)
 
+    // Initialize permission field as undefined 
+    let permissionData;
+
     accountList.forEach((singleAccount) => {
       const base = parseBaseData(singleAccount.account.data)
       if (base) {
@@ -70,6 +83,10 @@ export class PythHttpClient {
             break
           case AccountType.Test:
             break
+          case AccountType.Permission:
+            permissionData = parsePermissionData(singleAccount.account.data)
+            break
+
           default:
             throw new Error(`Unknown account type: ${base.type}. Try upgrading pyth-client.`)
         }
@@ -92,6 +109,7 @@ export class PythHttpClient {
       productFromSymbol,
       productPrice,
       prices,
+      permissionData,
     }
 
     return result

src/__tests__/Permission.test.ts

@@ -0,0 +1,24 @@
+import { PublicKey } from '@solana/web3.js'
+import { AccountType, Magic, parsePermissionData, Version } from '../index'
+
+test('Parse permission account', (done) => {
+  jest.setTimeout(60000)
+
+  const data = Buffer.from(
+    '1MOyoQIAAAAFAAAAcAAAAImry8+SSJfg5VrdIcyH2oRAsza674wsCWWahgc9QNtMMagMxzMOceCCHTfo6+fDdV+hjbfjevgN0nX2y65Rbw59wrxBgcqYAgQ5yOFn8V1OQYvW2z+qvPSyoctqLEwjMg==',
+    'base64',
+  )
+  const permission = parsePermissionData(data)
+  expect(permission.magic).toBe(Magic)
+  expect(permission.version).toBe(Version)
+  expect(permission.type).toBe(AccountType.Permission)
+  expect(permission.size).toBe(16 + 32 * 3)
+  expect(permission.masterAuthority.equals(new PublicKey('AGQnwBFJ3WwWXqjkaeDKYkU3tJFi8cMkCCevQeLUYSf5'))).toBeTruthy()
+  expect(
+    permission.dataCurationAuthority.equals(new PublicKey('4Lqbt4kQYjc5TxQiyRiLKi9rvizU4VRiXUpzEBkZLbid')),
+  ).toBeTruthy()
+  expect(
+    permission.securityAuthority.equals(new PublicKey('9TvBred1G7eFo8E16MEK4LxHP11XEMqxkdmZ84WtaUau')),
+  ).toBeTruthy()
+  done()
+})

src/index.ts

@@ -37,6 +37,7 @@ export enum AccountType {
   Product,
   Price,
   Test,
+  Permission,
 }
 
 const empty32Buffer = Buffer.alloc(32)
@@ -126,6 +127,12 @@ export interface PriceData extends Base {
   status: PriceStatus
 }
 
+export interface PermissionData extends Base {
+  masterAuthority: PublicKey
+  dataCurationAuthority: PublicKey
+  securityAuthority: PublicKey
+}
+
 /** Parse data as a generic Pyth account. Use this method if you don't know the account type. */
 export function parseBaseData(data: Buffer): Base | undefined {
   // data is too short to have the magic number.
@@ -367,6 +374,29 @@ export const parsePriceData = (data: Buffer, currentSlot?: number): PriceData =>
   }
 }
 
+export const parsePermissionData = (data: Buffer): PermissionData => {
+  // pyth magic number
+  const magic = data.readUInt32LE(0)
+  // program version
+  const version = data.readUInt32LE(4)
+  // account type
+  const type = data.readUInt32LE(8)
+  // price account size
+  const size = data.readUInt32LE(12)
+  const masterAuthority = new PublicKey(data.slice(16, 48))
+  const dataCurationAuthority = new PublicKey(data.slice(48, 80))
+  const securityAuthority = new PublicKey(data.slice(80, 112))
+  return {
+    magic,
+    version,
+    type,
+    size,
+    masterAuthority,
+    dataCurationAuthority,
+    securityAuthority,
+  }
+}
+
 export { PythConnection } from './PythConnection'
 export { PythHttpClient } from './PythHttpClient'
 export { getPythProgramKeyForCluster } from './cluster'