feat: add transfer restrictions management functionality

Author: F-OBrien

src/context/AssetContext/constants.ts

@@ -14,6 +14,7 @@ import {
   SecurityIdentifier,
   TickerReservation,
   TickerReservationDetails,
+  TransferRestrictionExemption,
   TransferRestrictionStatValues,
   Venue,
 } from '@polymeshassociation/polymesh-sdk/types';
@@ -56,6 +57,7 @@ export interface IDetails {
   compliancePaused: boolean;
   transferRestrictions?: ActiveTransferRestrictions;
   trackedStatistics?: TransferRestrictionStatValues[];
+  exemptions?: TransferRestrictionExemption[];
 }
 export interface IAssetDetails {
   assetId: string;

src/context/AssetContext/provider.tsx

@@ -103,6 +103,7 @@ const AssetProvider = ({ children }: IProviderProps) => {
           compliancePaused,
           transferRestrictions,
           trackedStatistics,
+          exemptions,
         ] = await Promise.all([
           asset.details(),
           asset instanceof NftCollection ? asset.getCollectionId() : undefined,
@@ -127,6 +128,10 @@ const AssetProvider = ({ children }: IProviderProps) => {
           asset instanceof FungibleAsset
             ? asset.transferRestrictions.getValues()
             : undefined,
+
+          asset instanceof FungibleAsset
+            ? asset.transferRestrictions.getExemptions()
+            : undefined,
         ]);
 
         // Fetch all metadata entries with their details and values (if set)
@@ -186,6 +191,7 @@ const AssetProvider = ({ children }: IProviderProps) => {
             compliancePaused,
             transferRestrictions,
             trackedStatistics,
+            exemptions,
           },
           docs: docs.data,
         };

src/hooks/polymesh/useAssetActions.tsx

@@ -7,6 +7,11 @@ import type {
   InputCondition,
   NftMetadataInput,
   Requirement,
+  TransferRestrictionClaimCountInput,
+  TransferRestrictionExemptionParams,
+  TransferRestrictionInputClaimPercentage,
+  TransferRestrictionInputCount,
+  TransferRestrictionInputPercentage,
   TrustedFor,
 } from '@polymeshassociation/polymesh-sdk/types';
 import {
@@ -1057,6 +1062,104 @@ const useAssetActions = (
     }
   };
 
+  const setTransferRestrictions = async ({
+    restrictions,
+    onTransactionRunning,
+  }: {
+    restrictions: (
+      | TransferRestrictionInputCount
+      | TransferRestrictionInputPercentage
+      | TransferRestrictionClaimCountInput
+      | TransferRestrictionInputClaimPercentage
+    )[];
+    onTransactionRunning?: () => void | Promise<void>;
+  }) => {
+    if (!asset) {
+      notifyError('Asset not available');
+      return;
+    }
+
+    // Type narrow: only FungibleAsset supports transfer restrictions
+    if (!('issuance' in asset)) {
+      notifyError(
+        'Transfer restrictions are only supported for fungible assets, not NFT collections',
+      );
+      return;
+    }
+
+    try {
+      await executeTransaction(
+        asset.transferRestrictions.setRestrictions({ restrictions }),
+        createOptions(onTransactionRunning),
+      );
+    } catch (error) {
+      // Error is already handled by the transaction context and notified to the user
+      // This catch block prevents unhandled promise rejection
+    }
+  };
+
+  const addExemptions = async ({
+    exemptions,
+    onTransactionRunning,
+  }: {
+    exemptions: TransferRestrictionExemptionParams;
+    onTransactionRunning?: () => void | Promise<void>;
+  }) => {
+    if (!asset) {
+      notifyError('Asset not available');
+      return;
+    }
+
+    // Type narrow: only FungibleAsset supports transfer restrictions
+    if (!('issuance' in asset)) {
+      notifyError(
+        'Transfer restrictions are only supported for fungible assets, not NFT collections',
+      );
+      return;
+    }
+
+    try {
+      await executeTransaction(
+        asset.transferRestrictions.addExemptions(exemptions),
+        createOptions(onTransactionRunning),
+      );
+    } catch (error) {
+      // Error is already handled by the transaction context and notified to the user
+      // This catch block prevents unhandled promise rejection
+    }
+  };
+
+  const removeExemptions = async ({
+    exemptions,
+    onTransactionRunning,
+  }: {
+    exemptions: TransferRestrictionExemptionParams;
+    onTransactionRunning?: () => void | Promise<void>;
+  }) => {
+    if (!asset) {
+      notifyError('Asset not available');
+      return;
+    }
+
+    // Type narrow: only FungibleAsset supports transfer restrictions
+    if (!('issuance' in asset)) {
+      notifyError(
+        'Transfer restrictions are only supported for fungible assets, not NFT collections',
+      );
+      return;
+    }
+
+    try {
+      await executeTransaction(
+        asset.transferRestrictions.removeExemptions(exemptions),
+        createOptions(onTransactionRunning),
+      );
+    } catch (error) {
+      // Error is already handled by the transaction context and notified to the user
+      // This catch block prevents unhandled promise rejection
+    }
+  };
+
   return {
     issueTokens,
     redeemTokens,
@@ -1097,6 +1200,9 @@ const useAssetActions = (
     modifyComplianceRule,
     removeComplianceRule,
     setTransferRestrictionStats,
+    setTransferRestrictions,
+    addExemptions,
+    removeExemptions,
     transactionInProcess,
   };
 };