feat: added contract utils

Author: hanzel98

packages/delegation-core/test/caveats/nativeTokenPeriodTransfer.test.ts

@@ -1,3 +1,4 @@
+import { isStrictHexString } from '@metamask/utils';
 import { describe, it, expect } from 'vitest';
 
 import { createNativeTokenPeriodTransferTerms } from '../../src/caveats/nativeTokenPeriodTransfer';
@@ -53,7 +54,8 @@ describe('createNativeTokenPeriodTransferTerms', () => {
     });
 
     expect(result).toHaveLength(194);
-    expect(result).toMatch(/^0x[0-9a-f]{192}$/u);
+    expect(isStrictHexString(result)).toBe(true);
+    expect(result.length).toBe(194); // Additional length validation
   });
 
   it('creates valid terms for maximum safe values', () => {
@@ -68,7 +70,8 @@ describe('createNativeTokenPeriodTransferTerms', () => {
     });
 
     expect(result).toHaveLength(194);
-    expect(result).toMatch(/^0x[0-9a-f]{192}$/u);
+    expect(isStrictHexString(result)).toBe(true);
+    expect(result.length).toBe(194); // Additional length validation
   });
 
   it('throws an error for zero period amount', () => {
@@ -225,8 +228,9 @@ describe('createNativeTokenPeriodTransferTerms', () => {
       startDate,
     });
 
-    expect(result).toMatch(/^0x[0-9a-f]{192}$/u);
+    expect(isStrictHexString(result)).toBe(true);
     expect(result).toHaveLength(194);
+    expect(result.length).toBe(194); // Additional length validation
   });
 
   // Tests for bytes return type

packages/delegation-toolkit/src/DelegationFramework/DelegationManager/methods/disabledDelegations.ts

@@ -0,0 +1,21 @@
+import { DelegationManager } from '@metamask/delegation-abis';
+import type { Address, Client, Hex } from 'viem';
+import { readContract } from 'viem/actions';
+
+export type ReadDisabledDelegationsParameters = {
+  client: Client;
+  contractAddress: Address;
+  delegationHash: Hex;
+};
+
+export const read = async ({
+  client,
+  contractAddress,
+  delegationHash,
+}: ReadDisabledDelegationsParameters) =>
+  await readContract(client, {
+    address: contractAddress,
+    abi: DelegationManager.abi,
+    functionName: 'disabledDelegations',
+    args: [delegationHash],
+  });

packages/delegation-toolkit/src/DelegationFramework/DelegationManager/read.ts

@@ -1,4 +1,5 @@
+import { read as disabledDelegations } from './methods/disabledDelegations';
 import { read as getAnyDelegate } from './methods/getAnyDelegate';
 import { read as getRootAuthority } from './methods/getRootAuthority';
 
-export { getAnyDelegate, getRootAuthority };
+export { getAnyDelegate, getRootAuthority, disabledDelegations };

packages/delegation-toolkit/src/DelegationFramework/ERC20PeriodTransferEnforcer/index.ts

@@ -0,0 +1,3 @@
+import * as read from './read';
+
+export { read };

packages/delegation-toolkit/src/DelegationFramework/ERC20PeriodTransferEnforcer/methods/getAvailableAmount.ts

@@ -0,0 +1,35 @@
+import { ERC20PeriodTransferEnforcer } from '@metamask/delegation-abis';
+import type { Address, Client, Hex } from 'viem';
+import { readContract } from 'viem/actions';
+
+export type ReadGetAvailableAmountParameters = {
+  client: Client;
+  contractAddress: Address;
+  delegationHash: Hex;
+  delegationManager: Address;
+  terms: Hex;
+};
+
+export const read = async ({
+  client,
+  contractAddress,
+  delegationHash,
+  delegationManager,
+  terms,
+}: ReadGetAvailableAmountParameters) => {
+  const [availableAmount, isNewPeriod, currentPeriod] = await readContract(
+    client,
+    {
+      address: contractAddress,
+      abi: ERC20PeriodTransferEnforcer.abi,
+      functionName: 'getAvailableAmount',
+      args: [delegationHash, delegationManager, terms],
+    },
+  );
+
+  return {
+    availableAmount,
+    isNewPeriod,
+    currentPeriod,
+  };
+};

packages/delegation-toolkit/src/DelegationFramework/ERC20PeriodTransferEnforcer/read.ts

@@ -0,0 +1,3 @@
+import { read as getAvailableAmount } from './methods/getAvailableAmount';
+
+export { getAvailableAmount };

packages/delegation-toolkit/src/DelegationFramework/ERC20StreamingEnforcer/index.ts

@@ -0,0 +1,3 @@
+import * as read from './read';
+
+export { read };

packages/delegation-toolkit/src/DelegationFramework/ERC20StreamingEnforcer/methods/getAvailableAmount.ts

@@ -0,0 +1,126 @@
+import { ERC20StreamingEnforcer } from '@metamask/delegation-abis';
+import type { Address, Client, Hex } from 'viem';
+import { readContract, getBlock } from 'viem/actions';
+
+export type ReadGetAvailableAmountParameters = {
+  client: Client;
+  contractAddress: Address;
+  delegationManager: Address;
+  delegationHash: Hex;
+  terms: Hex;
+};
+
+export const read = async ({
+  client,
+  contractAddress,
+  delegationManager,
+  delegationHash,
+  terms,
+}: ReadGetAvailableAmountParameters) => {
+  // Get current block timestamp from blockchain
+  const currentBlock = await getBlock(client);
+  const currentTimestamp = currentBlock.timestamp;
+
+  // First, get the current state from the contract
+  const allowanceState = await readContract(client, {
+    address: contractAddress,
+    abi: ERC20StreamingEnforcer.abi,
+    functionName: 'streamingAllowances',
+    args: [delegationManager, delegationHash],
+  });
+
+  const [initialAmount, maxAmount, amountPerSecond, startTime, spent] =
+    allowanceState;
+
+  // Check if state exists (startTime != 0)
+  if (startTime !== 0n) {
+    // State exists, calculate available amount using the stored state
+    const availableAmount = getAvailableAmount({
+      initialAmount,
+      maxAmount,
+      amountPerSecond,
+      startTime,
+      spent,
+      currentTimestamp,
+    });
+
+    return {
+      availableAmount,
+    };
+  }
+
+  // State doesn't exist, decode terms and simulate with spent = 0
+  const decodedTerms = await readContract(client, {
+    address: contractAddress,
+    abi: ERC20StreamingEnforcer.abi,
+    functionName: 'getTermsInfo',
+    args: [terms],
+  });
+
+  const [
+    ,
+    decodedInitialAmount,
+    decodedMaxAmount,
+    decodedAmountPerSecond,
+    decodedStartTime,
+  ] = decodedTerms;
+
+  // Simulate using decoded terms with spent = 0
+  const availableAmount = getAvailableAmount({
+    initialAmount: decodedInitialAmount,
+    maxAmount: decodedMaxAmount,
+    amountPerSecond: decodedAmountPerSecond,
+    startTime: decodedStartTime,
+    spent: 0n,
+    currentTimestamp,
+  });
+
+  return {
+    availableAmount,
+  };
+};
+
+/**
+ * Replicates the internal _getAvailableAmount logic from the smart contract.
+ *
+ * @param allowance - The allowance object containing all parameters.
+ * @param allowance.initialAmount - The initial amount available.
+ * @param allowance.maxAmount - The maximum amount allowed.
+ * @param allowance.amountPerSecond - The amount streamed per second.
+ * @param allowance.startTime - The start time of the streaming.
+ * @param allowance.spent - The amount already spent.
+ * @param allowance.currentTimestamp - The current timestamp.
+ * @returns The available amount that can be spent.
+ */
+function getAvailableAmount(allowance: {
+  initialAmount: bigint;
+  maxAmount: bigint;
+  amountPerSecond: bigint;
+  startTime: bigint;
+  spent: bigint;
+  currentTimestamp: bigint;
+}): bigint {
+  // If current time is before start time, nothing is available
+  if (allowance.currentTimestamp < allowance.startTime) {
+    return 0n;
+  }
+
+  // Calculate elapsed time since start
+  const elapsed = allowance.currentTimestamp - allowance.startTime;
+
+  // Calculate total unlocked amount
+  let unlocked = allowance.initialAmount + allowance.amountPerSecond * elapsed;
+
+  // Cap by max amount
+  if (unlocked > allowance.maxAmount) {
+    unlocked = allowance.maxAmount;
+  }
+
+  // If spent >= unlocked, nothing available
+  if (allowance.spent >= unlocked) {
+    return 0n;
+  }
+
+  // Return available amount
+  return unlocked - allowance.spent;
+}

packages/delegation-toolkit/src/DelegationFramework/ERC20StreamingEnforcer/read.ts

@@ -0,0 +1,3 @@
+import { read as getAvailableAmount } from './methods/getAvailableAmount';
+
+export { getAvailableAmount };

packages/delegation-toolkit/src/DelegationFramework/MultiTokenPeriodEnforcer/index.ts

@@ -0,0 +1,3 @@
+import * as read from './read';
+
+export { read };