@aresrpg/aresrpg-sdk/src/sui.js

General SDK to interract with AresRPG

import { KioskClient, Network } from '@mysten/kiosk'
import { SuiClient, getFullnodeUrl } from '@mysten/sui/client'
import { LRUCache } from 'lru-cache'
import { SuiGraphQLClient } from '@mysten/sui/graphql'

import { borrow_personal_kiosk_cap } from './sui/write/borrow_personal_kiosk_cap.js'
import { admin_promote } from './sui/write/admin_promote.js'
import { get_user_kiosks } from './sui/read/get_user_kiosks.js'
import { admin_freeze_contract } from './sui/write/admin_freeze_contract.js'
import { add_header } from './sui/write/add_header.js'
import {
  BULLSHARK,
  CAPY,
  SUPPORTED_NFTS,
  VAPOREON,
} from './sui/supported_nfts.js'
import { admin_delete_admin_cap } from './sui/write/admin_delete_admin_cap.js'
import { admin_withdraw_profit } from './sui/write/admin_withdraw_profit.js'
import { get_policies_profit } from './sui/read/get_policies_profit.js'
import { get_royalty_fee } from './sui/read/get_royalty_fee.js'
import { admin_create_recipe } from './sui/write/admin_create_recipe.js'
import { admin_delete_recipe } from './sui/write/admin_delete_recipe.js'
import { get_supported_tokens } from './sui/read/get_supported_tokens.js'
import { HSUI, SUPPORTED_TOKENS } from './sui/supported_tokens.js'
import { ITEM_CATEGORY } from './items.js'
import { create_personal_kiosk } from './sui/write/create_personal_kiosk.js'
import types from './types.json' with { type: 'json' }
import { admin_create_sale } from './sui/write/admin_create_sale.js'
import { admin_delete_sale } from './sui/write/admin_delete_sale.js'
import { get_suifren_stats } from './sui/feedable_suifrens.js'
import { get_vaporeon_stats } from './sui/feedable_vaporeon.js'
import { TRANSFER_POLICIES } from './sui/transfer_policies.js'
import { admin_mint_caps } from './sui/write/admin_mint_caps.js'

// keep fetched balances for 3s to avoid spamming the nodes
const balances_cache = new LRUCache({ max: 100, ttl: 3000 })

export async function SDK({
  rpc_url = getFullnodeUrl('testnet'),
  network = Network.TESTNET,
}) {
  const sui_client = new SuiClient({ url: rpc_url })
  const kiosk_client = new KioskClient({
    client: sui_client,
    network,
    // seems the kiosk sdk is missing the correct rule
    ...(network === Network.MAINNET
      ? {
          packageIds: {
            personalKioskRulePackageId:
              '0x0cb4bcc0560340eb1a1b929cabe56b33fc6449820ec8c1980d69bb98b649b802',
          },
        }
      : {}),
  })
  const backup_sui_client = new SuiClient({
    url:
      network === Network.TESTNET
        ? 'https://sui-testnet-rpc.publicnode.com'
        : 'https://sui-mainnet-rpc.publicnode.com',
  })
  const backup_kiosk_client = new KioskClient({
    client: backup_sui_client,
    network,
  })

  const gql_client = new SuiGraphQLClient({
    url: `https://sui-${network}.mystenlabs.com/graphql`,
  })

  const supported_tokens = SUPPORTED_TOKENS(network)
  const supported_nfts = SUPPORTED_NFTS(network)
  const transfer_policies = TRANSFER_POLICIES(network)

  Object.values(supported_tokens).forEach(async token => {
    Object.assign(token, {
      is_token: true,
      item_set: 'none',
      item_type: token.address,
      item_category: ITEM_CATEGORY.RESOURCE,
      level: 1,
    })
  })

  const context = {
    sui_client,
    kiosk_client,
    types,
    network,
    supported_tokens,
    HSUI: HSUI[network].address,
    VAPOREON: VAPOREON[network],
  }

  return {
    sui_client,
    kiosk_client,
    backup_sui_client,
    backup_kiosk_client,
    gql_client,
    ...types,
    SUPPORTED_TOKENS: supported_tokens,
    SUPPORTED_NFTS: supported_nfts,
    TRANSFER_POLICIES: transfer_policies,
    HSUI: HSUI[network],
    VAPOREON: VAPOREON[network],
    SUIFREN_CAPY: CAPY[network],
    SUIFREN_BULLSHARK: BULLSHARK[network],

    get_policies_profit: get_policies_profit(context),
    get_royalty_fee: get_royalty_fee(context),
    get_supported_tokens: get_supported_tokens(context),
    get_suifren_stats,
    get_vaporeon_stats,

    get_user_kiosks: get_user_kiosks(context),

    borrow_personal_kiosk_cap: borrow_personal_kiosk_cap(context),
    create_personal_kiosk: create_personal_kiosk(context),

    add_header: add_header(context),

    admin_promote: admin_promote(context),
    admin_freeze_contract: admin_freeze_contract(context),
    admin_delete_admin_cap: admin_delete_admin_cap(context),
    admin_withdraw_profit: admin_withdraw_profit(context),
    admin_create_recipe: admin_create_recipe(context),
    admin_delete_recipe: admin_delete_recipe(context),
    admin_create_sale: admin_create_sale(context),
    admin_delete_sale: admin_delete_sale(context),
    admin_mint_caps: admin_mint_caps(context),

    /** @return {Promise<bigint>} balance */
    async get_sui_balance(owner) {
      if (!balances_cache.has(owner)) {
        const { totalBalance } = await sui_client.getBalance({
          owner,
        })
        balances_cache.set(owner, BigInt(totalBalance))
      }

      // @ts-ignore
      return balances_cache.get(owner)
    },
  }
}