@lit-protocol/vincent-policy-spending-limit 1.0.0-1

package/dist/src/lib/vincent-policy.js

@@ -0,0 +1,116 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.vincentPolicy = void 0;
+const vincent_tool_sdk_1 = require("@lit-protocol/vincent-tool-sdk");
+const send_spend_tx_1 = require("./policy-helpers/send-spend-tx");
+const check_spending_limit_1 = require("./policy-helpers/check-spending-limit");
+const schemas_1 = require("./schemas");
+exports.vincentPolicy = (0, vincent_tool_sdk_1.createVincentPolicy)({
+    packageName: '@lit-protocol/vincent-policy-spending-limit',
+    toolParamsSchema: schemas_1.toolParamsSchema,
+    userParamsSchema: schemas_1.userParamsSchema,
+    commitParamsSchema: schemas_1.commitParamsSchema,
+    precheckAllowResultSchema: schemas_1.precheckAllowResultSchema,
+    precheckDenyResultSchema: schemas_1.precheckDenyResultSchema,
+    evalAllowResultSchema: schemas_1.evalAllowResultSchema,
+    evalDenyResultSchema: schemas_1.evalDenyResultSchema,
+    commitAllowResultSchema: schemas_1.commitAllowResultSchema,
+    commitDenyResultSchema: schemas_1.commitDenyResultSchema,
+    precheck: async ({ toolParams, userParams }, { allow, deny, appId, delegation: { delegatorPkpInfo } }) => {
+        console.log('Prechecking spending limit policy', { toolParams, userParams });
+        const { ethAddress } = delegatorPkpInfo;
+        const { buyAmount, ethRpcUrl, rpcUrlForUniswap, chainIdForUniswap, tokenAddress, tokenDecimals, } = toolParams;
+        const { maxDailySpendingLimitInUsdCents } = userParams;
+        const { buyAmountAllowed, buyAmountInUsd, adjustedMaxDailySpendingLimit } = await (0, check_spending_limit_1.checkIfBuyAmountAllowed)({
+            ethRpcUrl,
+            rpcUrlForUniswap,
+            chainIdForUniswap,
+            tokenAddress: tokenAddress,
+            tokenDecimals,
+            buyAmount,
+            maxDailySpendingLimitInUsdCents,
+            pkpEthAddress: ethAddress,
+            appId,
+        });
+        return buyAmountAllowed
+            ? allow({
+                maxSpendingLimitInUsd: Number(adjustedMaxDailySpendingLimit),
+                buyAmountInUsd: Number(buyAmountInUsd),
+            })
+            : deny({
+                reason: 'Attempted buy amount exceeds daily limit',
+                maxSpendingLimitInUsd: Number(adjustedMaxDailySpendingLimit),
+                buyAmountInUsd: Number(buyAmountInUsd),
+            });
+    },
+    evaluate: async ({ toolParams, userParams }, { allow, deny, appId, delegation: { delegatorPkpInfo } }) => {
+        const { ethAddress } = delegatorPkpInfo;
+        console.log('Evaluating spending limit policy', JSON.stringify(toolParams));
+        const { buyAmount, ethRpcUrl, rpcUrlForUniswap, chainIdForUniswap, tokenAddress, tokenDecimals, } = toolParams;
+        const { maxDailySpendingLimitInUsdCents } = userParams;
+        const checkBuyAmountResponse = await Lit.Actions.runOnce({ waitForResponse: true, name: 'checkBuyAmount' }, async () => {
+            try {
+                const { buyAmountAllowed, buyAmountInUsd, adjustedMaxDailySpendingLimit } = await (0, check_spending_limit_1.checkIfBuyAmountAllowed)({
+                    ethRpcUrl,
+                    rpcUrlForUniswap,
+                    chainIdForUniswap,
+                    tokenAddress: tokenAddress,
+                    tokenDecimals,
+                    buyAmount,
+                    maxDailySpendingLimitInUsdCents,
+                    pkpEthAddress: ethAddress,
+                    appId,
+                });
+                return JSON.stringify({
+                    status: 'success',
+                    buyAmountAllowed,
+                    buyAmountInUsd: buyAmountInUsd.toString(),
+                    adjustedMaxDailySpendingLimit: adjustedMaxDailySpendingLimit.toString(),
+                });
+            }
+            catch (error) {
+                return JSON.stringify({
+                    status: 'error',
+                    error: error instanceof Error ? error.message : String(error),
+                });
+            }
+        });
+        const parsedCheckBuyAmountResponse = JSON.parse(checkBuyAmountResponse);
+        if (parsedCheckBuyAmountResponse.status === 'error') {
+            return deny({
+                reason: `Error checking buy amount: ${parsedCheckBuyAmountResponse.error} (evaluate)`,
+            });
+        }
+        const { buyAmountAllowed, buyAmountInUsd, adjustedMaxDailySpendingLimit } = parsedCheckBuyAmountResponse;
+        console.log('Evaluated spending limit policy', {
+            buyAmountAllowed,
+            buyAmountInUsd,
+            adjustedMaxDailySpendingLimit,
+        });
+        return buyAmountAllowed
+            ? allow({
+                maxSpendingLimitInUsd: Number(adjustedMaxDailySpendingLimit),
+                buyAmountInUsd: Number(buyAmountInUsd),
+            })
+            : deny({
+                reason: 'Attempted buy amount exceeds daily limit',
+                maxSpendingLimitInUsd: Number(adjustedMaxDailySpendingLimit),
+                buyAmountInUsd: Number(buyAmountInUsd),
+            });
+    },
+    commit: async (params, { allow, appId, delegation: { delegatorPkpInfo } }) => {
+        const { ethAddress, publicKey } = delegatorPkpInfo;
+        const { amountSpentUsd, maxSpendingLimitInUsd } = params;
+        const spendTxHash = await (0, send_spend_tx_1.sendSpendTx)({
+            appId,
+            amountSpentUsd,
+            maxSpendingLimitInUsd,
+            spendingLimitDuration: 86400, // number of seconds in a day
+            pkpEthAddress: ethAddress,
+            pkpPubKey: publicKey,
+        });
+        return allow({
+            spendTxHash,
+        });
+    },
+});

package/package.json

@@ -0,0 +1,26 @@
+{
+  "name": "@lit-protocol/vincent-policy-spending-limit",
+  "version": "1.0.0-1",
+  "publishConfig": {
+    "access": "public"
+  },
+  "dependencies": {
+    "@uniswap/sdk-core": "^7.7.2",
+    "ethers": "^5.8.0",
+    "tslib": "2.8.1",
+    "zod": "^3.25.64",
+    "@lit-protocol/vincent-tool-sdk": "1.0.0-1"
+  },
+  "devDependencies": {
+    "esbuild": "^0.19.12",
+    "esbuild-plugin-polyfill-node": "^0.3.0",
+    "ipfs-only-hash": "^4.0.0"
+  },
+  "main": "./dist/src/index.js",
+  "module": "./dist/src/index.js",
+  "types": "./dist/src/index.d.ts",
+  "files": [
+    "dist/**",
+    "*.md"
+  ]
+}