Refactor pyth client (#274)

* Bump

* Refactor pyth client

* Bump. refactor client

staking/app/PythClient.ts

@@ -1,30 +1,3 @@
-import { AnchorProvider, Wallet } from "@project-serum/anchor";
-import { PublicKey } from "@solana/web3.js";
-import { STAKING_ADDRESS } from "./constants";
 import { StakeConnection } from "./StakeConnection";
 
-export class PythClient {
-  program: { programId: PublicKey };
-  cluster: string;
-  constructor(public stakeConnection: StakeConnection, cluster: string) {
-    this.cluster = cluster;
-    this.program = {
-      programId: STAKING_ADDRESS,
-    };
-  }
-  static async connect(
-    provider: AnchorProvider,
-    cluster: string
-  ): Promise<PythClient> {
-    // only supports localnet and devnet for now
-    // TODO: update this to support mainnet when program is deployed
-    return new PythClient(
-      await StakeConnection.createStakeConnection(
-        provider.connection,
-        provider.wallet as unknown as Wallet,
-        STAKING_ADDRESS
-      ),
-      cluster
-    );
-  }
-}
+export const PythClient = StakeConnection;

staking/app/StakeConnection.ts

@@ -34,7 +34,7 @@ import {
   PROGRAM_VERSION_V2,
   withCreateTokenOwnerRecord,
 } from "@solana/spl-governance";
-import { GOVERNANCE_ADDRESS } from "./constants";
+import { GOVERNANCE_ADDRESS, STAKING_ADDRESS } from "./constants";
 import assert from "assert";
 import { PositionAccountJs } from "./PositionAccountJs";
 let wasm = wasm2;
@@ -75,6 +75,17 @@ export class StakeConnection {
     this.governanceAddress = GOVERNANCE_ADDRESS();
   }
 
+  public static async connect(
+    connection: Connection,
+    wallet: Wallet
+  ): Promise<StakeConnection> {
+    return await StakeConnection.createStakeConnection(
+      connection,
+      wallet,
+      STAKING_ADDRESS
+    );
+  }
+
   // creates a program connection and loads the staking config
   // the constructor cannot be async so we use a static method
   public static async createStakeConnection(

staking/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pythnetwork/staking",
-  "version": "1.3.5",
+  "version": "2.0.0",
   "description": "Pyth Network Staking SDK",
   "main": "lib/app/index.js",
   "types": "lib/app/index.d.ts",