-
Notifications
You must be signed in to change notification settings - Fork 51
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add script to backfill expiration to delegations
- Loading branch information
1 parent
0ef5b93
commit 307b01b
Showing
19 changed files
with
145 additions
and
4 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
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
127 changes: 127 additions & 0 deletions
127
packages/helium-admin-cli/src/add-expiration-to-delegations.ts
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,127 @@ | ||
| import * as anchor from "@coral-xyz/anchor"; | ||
| import { daoKey, init as initHsd, subDaoEpochInfoKey } from "@helium/helium-sub-daos-sdk"; | ||
| import { init as initProxy } from "@helium/nft-proxy-sdk"; | ||
| import { batchParallelInstructionsWithPriorityFee, HNT_MINT } from "@helium/spl-utils"; | ||
| import { init as initVsr } from "@helium/voter-stake-registry-sdk"; | ||
| import { AccountInfo, PublicKey, SystemProgram, SYSVAR_CLOCK_PUBKEY, TransactionInstruction } from "@solana/web3.js"; | ||
| import { min } from "bn.js"; | ||
| import os from "os"; | ||
| import yargs from "yargs/yargs"; | ||
| import { loadKeypair } from "./utils"; | ||
|
|
||
| export async function run(args: any = process.argv) { | ||
| const yarg = yargs(args).options({ | ||
| wallet: { | ||
| alias: "k", | ||
| describe: "Anchor wallet keypair", | ||
| default: `${os.homedir()}/.config/solana/id.json`, | ||
| }, | ||
| url: { | ||
| alias: "u", | ||
| default: "http://127.0.0.1:8899", | ||
| describe: "The solana url", | ||
| }, | ||
| hntMint: { | ||
| type: "string", | ||
| describe: "HNT mint of the dao to be updated", | ||
| default: HNT_MINT.toBase58(), | ||
| }, | ||
| }); | ||
| const argv = await yarg.argv; | ||
| process.env.ANCHOR_WALLET = argv.wallet; | ||
| process.env.ANCHOR_PROVIDER_URL = argv.url; | ||
| anchor.setProvider(anchor.AnchorProvider.local(argv.url)); | ||
| const provider = anchor.getProvider() as anchor.AnchorProvider; | ||
| const wallet = new anchor.Wallet(loadKeypair(argv.wallet)); | ||
| const proxyProgram = await initProxy(provider); | ||
| const vsrProgram = await initVsr(provider); | ||
| const hsdProgram = await initHsd(provider); | ||
|
|
||
|
|
||
| const hntMint = new PublicKey(argv.hntMint); | ||
| const dao = daoKey(hntMint)[0]; | ||
| const registrarK = (await hsdProgram.account.daoV0.fetch(dao)).registrar; | ||
| const registrar = await vsrProgram.account.registrar.fetch(registrarK); | ||
| const proxyConfig = await proxyProgram.account.proxyConfigV0.fetch( | ||
| registrar.proxyConfig | ||
| ); | ||
|
|
||
| const instructions: TransactionInstruction[] = []; | ||
| const delegations = await hsdProgram.account.delegatedPositionV0.all() | ||
| const needsMigration = delegations.filter(d => d.account.expirationTs.isZero()); | ||
| const positionKeys = needsMigration.map(d => d.account.position); | ||
| const coder = hsdProgram.coder.accounts | ||
| const positionAccs = (await getMultipleAccounts({ | ||
| connection: provider.connection, | ||
| keys: positionKeys, | ||
| })).map(a => coder.decode("positionV0", a.data)); | ||
|
|
||
| const currTs = await getSolanaUnixTimestamp(provider); | ||
| const currTsBN = new anchor.BN(currTs.toString()); | ||
| const proxyEndTs = proxyConfig.seasons.find(s => currTsBN.gt(s.start))?.end; | ||
| for (const [delegation, position] of zip(needsMigration, positionAccs)) { | ||
| const subDao = delegation.account.subDao; | ||
| instructions.push( | ||
| await hsdProgram.methods | ||
| .addExpirationTs() | ||
| .accountsStrict({ | ||
| payer: wallet.publicKey, | ||
| position: delegation.account.position, | ||
| delegatedPosition: delegation.publicKey, | ||
| registrar: registrarK, | ||
| dao, | ||
| subDao: delegation.account.subDao, | ||
| oldClosingTimeSubDaoEpochInfo: subDaoEpochInfoKey( | ||
| subDao, | ||
| position.lockup.endTs | ||
| )[0], | ||
| closingTimeSubDaoEpochInfo: subDaoEpochInfoKey( | ||
| subDao, | ||
| min(position.lockup.endTs, proxyEndTs!) | ||
| )[0], | ||
| genesisEndSubDaoEpochInfo: subDaoEpochInfoKey( | ||
| subDao, | ||
| position.genesisEnd | ||
| )[0], | ||
| proxyConfig: registrar.proxyConfig, | ||
| systemProgram: SystemProgram.programId, | ||
| }) | ||
| .instruction() | ||
| ); | ||
| } | ||
|
|
||
| await batchParallelInstructionsWithPriorityFee(provider, instructions, { | ||
| onProgress: (status) => { | ||
| console.log(status); | ||
| }, | ||
| }); | ||
| } | ||
|
|
||
| async function getMultipleAccounts({ | ||
| connection, | ||
| keys, | ||
| }): Promise<AccountInfo<Buffer>[]> { | ||
| const batchSize = 100; | ||
| const batches = Math.ceil(keys.length / batchSize); | ||
| const results: AccountInfo<Buffer>[] = []; | ||
|
|
||
| for (let i = 0; i < batches; i++) { | ||
| const batchKeys = keys.slice(i * batchSize, (i + 1) * batchSize); | ||
| const batchResults = await connection.getMultipleAccountsInfo(batchKeys); | ||
| results.push(...batchResults); | ||
| } | ||
|
|
||
| return results; | ||
| } | ||
|
|
||
| function zip<T, U>(a: T[], b: U[]): [T, U][] { | ||
| return a.map((_, i) => [a[i], b[i]]); | ||
| } | ||
|
|
||
| async function getSolanaUnixTimestamp( | ||
| provider: anchor.AnchorProvider | ||
| ): Promise<bigint> { | ||
| const clock = await provider.connection.getAccountInfo(SYSVAR_CLOCK_PUBKEY); | ||
| const unixTime = clock!.data.readBigInt64LE(8 * 4); | ||
| return unixTime; | ||
| } |
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
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
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
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
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
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
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