Update tokens SDK

Author: jakeloo

apps/dashboard/src/app/(app)/team/[team_slug]/[project_slug]/(sidebar)/tokens/create/token/distribution/token-sale.tsx

@@ -5,7 +5,7 @@ import { DollarSignIcon, XIcon } from "lucide-react";
 import { useEffect, useState } from "react";
 import type { ThirdwebClient } from "thirdweb";
 import { defineChain } from "thirdweb";
-import { isRouterEnabled } from "thirdweb/tokens";
+import { isPoolRouterEnabled } from "thirdweb/tokens";
 import { DistributionBarChart } from "@/components/blocks/distribution-chart";
 import { FormFieldSetup } from "@/components/blocks/FormFieldSetup";
 import { Badge } from "@/components/ui/badge";
@@ -36,7 +36,7 @@ export function TokenSaleSection(props: {
   const isRouterEnabledQuery = useQuery({
     queryFn: async () => {
       try {
-        return await isRouterEnabled({
+        return await isPoolRouterEnabled({
           // eslint-disable-next-line no-restricted-syntax
           chain: defineChain(Number(props.chainId)),
           client: props.client,

apps/dashboard/src/app/(app)/team/[team_slug]/[project_slug]/contract/[chainIdOrSlug]/[contractAddress]/rewards/components/claim-rewards-page.tsx

@@ -10,7 +10,7 @@ import {
   toTokens,
 } from "thirdweb";
 import { TokenIcon, TokenProvider, useSendTransaction } from "thirdweb/react";
-import { claimReward } from "thirdweb/tokens";
+import { claimRewards } from "thirdweb/tokens";
 import { DistributionBarChart } from "@/components/blocks/distribution-chart";
 import { WalletAddress } from "@/components/blocks/wallet-address";
 import { Button } from "@/components/ui/button";
@@ -42,7 +42,7 @@ export function ClaimRewardsPage(props: {
   const router = useDashboardRouter();
 
   async function handleClaim() {
-    const claimRewardsTx = claimReward({
+    const claimRewardsTx = claimRewards({
       asset: props.assetContractClient.address,
       contract: props.entrypointContractClient,
     });

apps/dashboard/src/app/(app)/team/[team_slug]/[project_slug]/contract/[chainIdOrSlug]/[contractAddress]/rewards/page.tsx

@@ -2,8 +2,6 @@ import { notFound, redirect } from "next/navigation";
 import { getContract } from "thirdweb";
 import {
   getDeployedEntrypointERC20,
-  getRewardLocker,
-  v3PositionManager as getV3PositionManager,
 } from "thirdweb/tokens";
 import { getProject } from "@/api/projects";
 import { getContractPageParamsInfo } from "../../../../../../../(dashboard)/(chain)/[chain_id]/[contractAddress]/_utils/getContractFromParams";
@@ -44,27 +42,7 @@ export default async function Page(props: {
     entrypointContract: entrypointContractClient,
   });
 
-  const rewardLocker = await getRewardLocker({
-    contract: entrypointContractClient,
-  }).catch(() => null);
-
-  if (!reward || !rewardLocker) {
-    redirect(
-      `/team/${params.team_slug}/${params.project_slug}/contract/${params.chainIdOrSlug}/${params.contractAddress}`,
-    );
-  }
-
-  const rewardLockerContractClient = getContract({
-    address: rewardLocker,
-    chain: assetContractClient.chain,
-    client: assetContractClient.client,
-  });
-
-  const v3PositionManager = await getV3PositionManager({
-    contract: rewardLockerContractClient,
-  }).catch(() => null);
-
-  if (!v3PositionManager || v3PositionManager !== reward.positionManager) {
+  if (!reward) {
     redirect(
       `/team/${params.team_slug}/${params.project_slug}/contract/${params.chainIdOrSlug}/${params.contractAddress}`,
     );
@@ -79,7 +57,7 @@ export default async function Page(props: {
   const unclaimedFees = await getUnclaimedFees({
     positionManager: v3PositionManagerContract,
     reward: {
-      tokenId: reward.tokenId,
+      tokenId: reward.positionId,
       recipient: reward.recipient,
     },
   });

apps/dashboard/src/app/(app)/team/[team_slug]/[project_slug]/contract/[chainIdOrSlug]/[contractAddress]/rewards/utils/rewards.ts

@@ -1,27 +1,22 @@
-import { type ThirdwebContract, ZERO_ADDRESS } from "thirdweb";
-import { getReward } from "thirdweb/tokens";
+import { type ThirdwebContract } from "thirdweb";
+import { getRewards } from "thirdweb/tokens";
 
 export async function getValidReward(params: {
   assetContract: ThirdwebContract;
   entrypointContract: ThirdwebContract;
 }) {
   try {
-    const reward = await getReward({
+    const rewards = await getRewards({
       contract: params.entrypointContract,
       asset: params.assetContract.address,
     });
 
-    if (
-      reward.positionManager === ZERO_ADDRESS ||
-      reward.recipient === ZERO_ADDRESS ||
-      reward.referrer === ZERO_ADDRESS ||
-      reward.referrerBps === 0 ||
-      reward.tokenId === BigInt(0)
-    ) {
+    if (rewards.length === 0) {
       return null;
     }
 
-    return reward;
+    // It's potentially possible to have multiple rewards locked up, but it's not the default use case.
+    return rewards[0];
   } catch {
     return null;
   }