@quartz-labs/sdk 0.0.1

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Quartz Labs
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,17 @@
+<div align="center">
+  <img width="2500" alt="Quartz" src="https://cdn.prod.website-files.com/65707af0f4af991289bbd432/670e37661cdb2314fe8ba469_logo-glow-banner.jpg" />
+
+  <h1 style="margin-top:20px;">Quartz SDK</h1>
+</div>
+
+Typescript SDK for interacting with the Quartz Protocol. 
+
+## Links
+
+Website and waitlist: [quartzpay.io](https://quartzpay.io/)
+
+Docs: [docs.quartzpay.io](https://docs.quartzpay.io/)
+
+X: [@quartzpay](https://x.com/quartzpay)
+
+Contact: [diego@quartzpay.io](mailto:diego@quartzpay.io)

package/package.json

@@ -0,0 +1,44 @@
+{
+  "name": "@quartz-labs/sdk",
+  "version": "0.0.1",
+  "description": "SDK for interacting with the Quartz Protocol",
+  "main": "index.js",
+  "types": "index.d.ts",
+  "typesVersions": {
+    ">=4.7": { "*": ["ts/*"] }
+  },
+  "scripts": {
+    "build": "tsc",
+    "test": "jest"
+  },
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/quartz-labs/sdk.git"
+  },
+  "keywords": [
+    "quartz",
+    "solana",
+    "sdk",
+    "crypto",
+    "protocol"
+  ],
+  "author": "Quartz Labs",
+  "license": "MIT",
+  "bugs": {
+    "email": "iarla@quartzpay.io"
+  },
+  "homepage": "https://quartzpay.io",
+  "devDependencies": {
+    "@types/jest": "^29.5.14",
+    "@types/node": "^22.10.1",
+    "jest": "^29.7.0",
+    "ts-jest": "^29.2.5",
+    "typescript": "^5.7.2"
+  },
+  "dependencies": {
+    "@coral-xyz/anchor": "^0.29.0",
+    "@drift-labs/sdk": "^2.103.0-beta.9",
+    "@pythnetwork/pyth-solana-receiver": "^0.8.2",
+    "@solana/web3.js": "^1.95.8"
+  }
+}

package/src/client.ts

@@ -0,0 +1,117 @@
+import { DriftClient, fetchUserAccountsUsingKeys as fetchDriftAccountsUsingKeys } from "@drift-labs/sdk";
+import { QUARTZ_ADDRESS_TABLE, QUARTZ_PROGRAM_ID, SUPPORTED_DRIFT_MARKETS } from "./config/constants";
+import quartzIdl from "./idl/quartz.json";
+import { Quartz } from "./types/quartz.js";
+import { AnchorProvider, Idl, Program, setProvider, Wallet } from "@coral-xyz/anchor";
+import { Connection, AddressLookupTableAccount } from "@solana/web3.js";
+import { PythSolanaReceiver } from "@pythnetwork/pyth-solana-receiver";
+import { PublicKey } from "@solana/web3.js";
+import { QuartzUser } from "./user";
+import { getDriftUser, getVaultPubkey } from "./helpers";
+
+export class QuartzClient {
+    private connection: Connection;
+    private wallet: Wallet;
+    private program: Program<Quartz>;
+    private quartzAddressTable: AddressLookupTableAccount;
+
+    private driftClient: DriftClient;
+    private oracles: Map<string, PublicKey>;
+
+    constructor(
+        connection: Connection,
+        wallet: Wallet,
+        program: Program<Quartz>,
+        quartzAddressTable: AddressLookupTableAccount,
+        driftClient: DriftClient,
+        oracles: Map<string, PublicKey>
+    ) {
+        this.connection = connection;
+        this.wallet = wallet;
+        this.program = program;
+        this.quartzAddressTable = quartzAddressTable;
+        this.driftClient = driftClient;
+        this.oracles = oracles;
+    }
+
+    static async initialize(
+        connection: Connection, 
+        wallet: Wallet
+    ) {
+        const provider = new AnchorProvider(connection, wallet, { commitment: "confirmed" });
+        setProvider(provider);
+        const program = new Program(quartzIdl as Idl, QUARTZ_PROGRAM_ID, provider) as unknown as Program<Quartz>;
+
+        const quartzLookupTable = await connection.getAddressLookupTable(QUARTZ_ADDRESS_TABLE).then((res) => res.value);
+        if (!quartzLookupTable) throw Error("Address Lookup Table account not found");
+
+        const driftClient = new DriftClient({
+            connection,
+            wallet,
+            env: 'mainnet-beta',
+            userStats: false,
+            perpMarketIndexes: [],
+            spotMarketIndexes: SUPPORTED_DRIFT_MARKETS,
+            accountSubscription: {
+                type: 'websocket',
+                commitment: "confirmed"
+            }
+        });
+        await driftClient.subscribe();
+
+        const pythSolanaReceiver = new PythSolanaReceiver({ connection, wallet });
+        const oracles = new Map<string, PublicKey>([
+            ["SOL/USD", pythSolanaReceiver.getPriceFeedAccountAddress(0, 
+                "0xef0d8b6fda2ceba41da15d4095d1da392a0d2f8ed0c6c7bc0f4cfac8c280b56d")],
+            ["USDC/USD", pythSolanaReceiver.getPriceFeedAccountAddress(0, 
+                "0xeaa020c61cc479712813461ce153894a96a6c00b21ed0cfc2798d1f9a9e9c94a")]
+        ]);
+
+        return new QuartzClient(
+            connection, 
+            wallet, 
+            program, 
+            quartzLookupTable,
+            driftClient,
+            oracles
+        );
+    }
+
+    public async getAllAccountPubkeys(): Promise<PublicKey[]> {
+        return (
+            await this.program.account.vault.all()
+        ).map((vault) => vault.publicKey);
+    }
+
+    public async getQuartzAccount(pubkey: PublicKey): Promise<QuartzUser> {
+        const vault = getVaultPubkey(pubkey);
+        await this.program.account.vault.fetch(vault); // Check account exists
+        return new QuartzUser(pubkey, this.connection, this.driftClient);
+    }
+
+    public async getMultipleQuartzAccounts(pubkeys: PublicKey[]): Promise<QuartzUser[]> {
+        if (pubkeys.length === 0) return [];
+
+        const vaults = pubkeys.map((pubkey) => getVaultPubkey(pubkey));
+        const accounts = await this.program.account.vault.fetchMultiple(vaults);
+
+        accounts.forEach((account, index) => {
+            if (account === null) throw Error(`Account not found for pubkey: ${pubkeys[index].toBase58()}`)
+        });
+
+        const driftUsers = await fetchDriftAccountsUsingKeys(
+            this.connection,
+            this.driftClient.program,
+            vaults.map((vault) => getDriftUser(vault))
+        )
+
+        const undefinedIndex = driftUsers.findIndex(user => !user);
+        if (undefinedIndex !== -1) {
+            throw new Error(`[${this.wallet?.publicKey}] Failed to fetch drift user for vault ${vaults[undefinedIndex].toBase58()}`);
+        }
+
+        return driftUsers.map((driftUser, index) => {
+            return new QuartzUser(pubkeys[index], this.connection, this.driftClient, driftUser)
+        });
+    }
+}

package/src/config/constants.ts

@@ -0,0 +1,10 @@
+import { PublicKey } from "@solana/web3.js";
+
+export const QUARTZ_PROGRAM_ID = new PublicKey("6JjHXLheGSNvvexgzMthEcgjkcirDrGduc3HAKB2P1v2");
+export const QUARTZ_ADDRESS_TABLE = new PublicKey("96BmeKKVGX3LKYSKo3FCEom1YpNY11kCnGscKq6ouxLx");
+
+export const QUARTZ_HEALTH_BUFFER = 0.1;
+
+export const DRIFT_MARKET_INDEX_USDC = 0;
+export const DRIFT_MARKET_INDEX_SOL = 1;
+export const SUPPORTED_DRIFT_MARKETS = [DRIFT_MARKET_INDEX_USDC, DRIFT_MARKET_INDEX_SOL];

package/src/helpers.ts

@@ -0,0 +1,33 @@
+import { PublicKey } from "@solana/web3.js";
+import { QUARTZ_PROGRAM_ID } from "./config/constants";
+import { BN } from "@coral-xyz/anchor";
+import { DRIFT_PROGRAM_ID } from "@drift-labs/sdk";
+
+export const getVaultPubkey = (owner: PublicKey) => {
+    const [vaultPda] = PublicKey.findProgramAddressSync(
+        [Buffer.from("vault"), owner.toBuffer()],
+        QUARTZ_PROGRAM_ID
+    )
+    return vaultPda;
+}
+
+export const getVaultSplPubkey = (owner: PublicKey, mint: PublicKey) => {
+    const vaultPda = getVaultPubkey(owner);
+    const [vaultSplPda] = PublicKey.findProgramAddressSync(
+        [vaultPda.toBuffer(), mint.toBuffer()],
+        QUARTZ_PROGRAM_ID
+    );
+    return vaultSplPda;
+}
+
+export const getDriftUser = (vaultPda: PublicKey) => {
+    const [userPda] = PublicKey.findProgramAddressSync(
+        [
+			Buffer.from("user"),
+			vaultPda.toBuffer(),
+			new BN(0).toArrayLike(Buffer, 'le', 2),
+		],
+		new PublicKey(DRIFT_PROGRAM_ID)
+    );
+    return userPda;
+}