@agent-shield/mcp 0.1.0

package/dist/tools/withdraw.js

@@ -0,0 +1,34 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.withdrawTool = exports.withdrawSchema = void 0;
+exports.withdraw = withdraw;
+const zod_1 = require("zod");
+const utils_1 = require("../utils");
+const errors_1 = require("../errors");
+exports.withdrawSchema = zod_1.z.object({
+    vault: zod_1.z.string().describe("Vault PDA address (base58)"),
+    mint: zod_1.z.string().describe("Token mint address (base58)"),
+    amount: zod_1.z.string().describe("Amount in token base units (e.g. lamports)"),
+});
+async function withdraw(client, input) {
+    try {
+        const sig = await client.withdraw((0, utils_1.toPublicKey)(input.vault), (0, utils_1.toPublicKey)(input.mint), (0, utils_1.toBN)(input.amount));
+        return [
+            "## Withdrawal Successful",
+            `- **Vault:** ${input.vault}`,
+            `- **Token:** ${input.mint}`,
+            `- **Amount:** ${input.amount}`,
+            `- **Transaction:** ${sig}`,
+        ].join("\n");
+    }
+    catch (error) {
+        return (0, errors_1.formatError)(error);
+    }
+}
+exports.withdrawTool = {
+    name: "shield_withdraw",
+    description: "Withdraw tokens from an AgentShield vault. Owner-only operation.",
+    schema: exports.withdrawSchema,
+    handler: withdraw,
+};
+//# sourceMappingURL=withdraw.js.map

package/dist/tools/withdraw.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"withdraw.js","sourceRoot":"","sources":["../../src/tools/withdraw.ts"],"names":[],"mappings":";;;AAaA,4BAqBC;AAlCD,6BAAwB;AAExB,oCAA6C;AAC7C,sCAAwC;AAE3B,QAAA,cAAc,GAAG,OAAC,CAAC,MAAM,CAAC;IACrC,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,4BAA4B,CAAC;IACxD,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,6BAA6B,CAAC;IACxD,MAAM,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,4CAA4C,CAAC;CAC1E,CAAC,CAAC;AAII,KAAK,UAAU,QAAQ,CAC5B,MAAyB,EACzB,KAAoB;IAEpB,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,QAAQ,CAC/B,IAAA,mBAAW,EAAC,KAAK,CAAC,KAAK,CAAC,EACxB,IAAA,mBAAW,EAAC,KAAK,CAAC,IAAI,CAAC,EACvB,IAAA,YAAI,EAAC,KAAK,CAAC,MAAM,CAAC,CACnB,CAAC;QAEF,OAAO;YACL,0BAA0B;YAC1B,gBAAgB,KAAK,CAAC,KAAK,EAAE;YAC7B,gBAAgB,KAAK,CAAC,IAAI,EAAE;YAC5B,iBAAiB,KAAK,CAAC,MAAM,EAAE;YAC/B,sBAAsB,GAAG,EAAE;SAC5B,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACf,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,IAAA,oBAAW,EAAC,KAAK,CAAC,CAAC;IAC5B,CAAC;AACH,CAAC;AAEY,QAAA,YAAY,GAAG;IAC1B,IAAI,EAAE,iBAAiB;IACvB,WAAW,EACT,kEAAkE;IACpE,MAAM,EAAE,sBAAc;IACtB,OAAO,EAAE,QAAQ;CAClB,CAAC"}

package/dist/utils.d.ts

@@ -0,0 +1,11 @@
+import { PublicKey } from "@solana/web3.js";
+import { BN } from "@coral-xyz/anchor";
+import type { VaultStatus, ActionType } from "@agent-shield/sdk";
+export declare function toPublicKey(value: string): PublicKey;
+export declare function toBN(value: string): BN;
+export declare function formatVaultStatus(status: VaultStatus): string;
+export declare function formatActionType(action: ActionType): string;
+export declare function formatBN(value: BN): string;
+export declare function formatTimestamp(ts: BN): string;
+export declare function formatLamports(lamports: BN, decimals?: number): string;
+//# sourceMappingURL=utils.d.ts.map

package/dist/utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../src/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC5C,OAAO,EAAE,EAAE,EAAE,MAAM,mBAAmB,CAAC;AACvC,OAAO,KAAK,EACV,WAAW,EACX,UAAU,EACX,MAAM,mBAAmB,CAAC;AAE3B,wBAAgB,WAAW,CAAC,KAAK,EAAE,MAAM,GAAG,SAAS,CAMpD;AAED,wBAAgB,IAAI,CAAC,KAAK,EAAE,MAAM,GAAG,EAAE,CAQtC;AAED,wBAAgB,iBAAiB,CAAC,MAAM,EAAE,WAAW,GAAG,MAAM,CAK7D;AAED,wBAAgB,gBAAgB,CAAC,MAAM,EAAE,UAAU,GAAG,MAAM,CAS3D;AAED,wBAAgB,QAAQ,CAAC,KAAK,EAAE,EAAE,GAAG,MAAM,CAE1C;AAED,wBAAgB,eAAe,CAAC,EAAE,EAAE,EAAE,GAAG,MAAM,CAG9C;AAED,wBAAgB,cAAc,CAAC,QAAQ,EAAE,EAAE,EAAE,QAAQ,GAAE,MAAU,GAAG,MAAM,CAOzE"}

package/dist/utils.js

@@ -0,0 +1,69 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.toPublicKey = toPublicKey;
+exports.toBN = toBN;
+exports.formatVaultStatus = formatVaultStatus;
+exports.formatActionType = formatActionType;
+exports.formatBN = formatBN;
+exports.formatTimestamp = formatTimestamp;
+exports.formatLamports = formatLamports;
+const web3_js_1 = require("@solana/web3.js");
+const anchor_1 = require("@coral-xyz/anchor");
+function toPublicKey(value) {
+    try {
+        return new web3_js_1.PublicKey(value);
+    }
+    catch {
+        throw new Error(`Invalid public key: "${value}"`);
+    }
+}
+function toBN(value) {
+    try {
+        return new anchor_1.BN(value);
+    }
+    catch {
+        throw new Error(`Invalid numeric value: "${value}". Provide a decimal integer string.`);
+    }
+}
+function formatVaultStatus(status) {
+    if ("active" in status)
+        return "Active";
+    if ("frozen" in status)
+        return "Frozen";
+    if ("closed" in status)
+        return "Closed";
+    return "Unknown";
+}
+function formatActionType(action) {
+    if ("swap" in action)
+        return "Swap";
+    if ("openPosition" in action)
+        return "Open Position";
+    if ("closePosition" in action)
+        return "Close Position";
+    if ("increasePosition" in action)
+        return "Increase Position";
+    if ("decreasePosition" in action)
+        return "Decrease Position";
+    if ("deposit" in action)
+        return "Deposit";
+    if ("withdraw" in action)
+        return "Withdraw";
+    return "Unknown";
+}
+function formatBN(value) {
+    return value.toString();
+}
+function formatTimestamp(ts) {
+    const ms = ts.toNumber() * 1000;
+    return new Date(ms).toISOString();
+}
+function formatLamports(lamports, decimals = 9) {
+    const str = lamports.toString().padStart(decimals + 1, "0");
+    const whole = str.slice(0, str.length - decimals) || "0";
+    const frac = str.slice(str.length - decimals);
+    // Trim trailing zeros
+    const trimmed = frac.replace(/0+$/, "");
+    return trimmed ? `${whole}.${trimmed}` : whole;
+}
+//# sourceMappingURL=utils.js.map

package/dist/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../src/utils.ts"],"names":[],"mappings":";;AAOA,kCAMC;AAED,oBAQC;AAED,8CAKC;AAED,4CASC;AAED,4BAEC;AAED,0CAGC;AAED,wCAOC;AA3DD,6CAA4C;AAC5C,8CAAuC;AAMvC,SAAgB,WAAW,CAAC,KAAa;IACvC,IAAI,CAAC;QACH,OAAO,IAAI,mBAAS,CAAC,KAAK,CAAC,CAAC;IAC9B,CAAC;IAAC,MAAM,CAAC;QACP,MAAM,IAAI,KAAK,CAAC,wBAAwB,KAAK,GAAG,CAAC,CAAC;IACpD,CAAC;AACH,CAAC;AAED,SAAgB,IAAI,CAAC,KAAa;IAChC,IAAI,CAAC;QACH,OAAO,IAAI,WAAE,CAAC,KAAK,CAAC,CAAC;IACvB,CAAC;IAAC,MAAM,CAAC;QACP,MAAM,IAAI,KAAK,CACb,2BAA2B,KAAK,sCAAsC,CACvE,CAAC;IACJ,CAAC;AACH,CAAC;AAED,SAAgB,iBAAiB,CAAC,MAAmB;IACnD,IAAI,QAAQ,IAAI,MAAM;QAAE,OAAO,QAAQ,CAAC;IACxC,IAAI,QAAQ,IAAI,MAAM;QAAE,OAAO,QAAQ,CAAC;IACxC,IAAI,QAAQ,IAAI,MAAM;QAAE,OAAO,QAAQ,CAAC;IACxC,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAgB,gBAAgB,CAAC,MAAkB;IACjD,IAAI,MAAM,IAAI,MAAM;QAAE,OAAO,MAAM,CAAC;IACpC,IAAI,cAAc,IAAI,MAAM;QAAE,OAAO,eAAe,CAAC;IACrD,IAAI,eAAe,IAAI,MAAM;QAAE,OAAO,gBAAgB,CAAC;IACvD,IAAI,kBAAkB,IAAI,MAAM;QAAE,OAAO,mBAAmB,CAAC;IAC7D,IAAI,kBAAkB,IAAI,MAAM;QAAE,OAAO,mBAAmB,CAAC;IAC7D,IAAI,SAAS,IAAI,MAAM;QAAE,OAAO,SAAS,CAAC;IAC1C,IAAI,UAAU,IAAI,MAAM;QAAE,OAAO,UAAU,CAAC;IAC5C,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAgB,QAAQ,CAAC,KAAS;IAChC,OAAO,KAAK,CAAC,QAAQ,EAAE,CAAC;AAC1B,CAAC;AAED,SAAgB,eAAe,CAAC,EAAM;IACpC,MAAM,EAAE,GAAG,EAAE,CAAC,QAAQ,EAAE,GAAG,IAAI,CAAC;IAChC,OAAO,IAAI,IAAI,CAAC,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;AACpC,CAAC;AAED,SAAgB,cAAc,CAAC,QAAY,EAAE,WAAmB,CAAC;IAC/D,MAAM,GAAG,GAAG,QAAQ,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,QAAQ,GAAG,CAAC,EAAE,GAAG,CAAC,CAAC;IAC5D,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,MAAM,GAAG,QAAQ,CAAC,IAAI,GAAG,CAAC;IACzD,MAAM,IAAI,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,GAAG,QAAQ,CAAC,CAAC;IAC9C,sBAAsB;IACtB,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IACxC,OAAO,OAAO,CAAC,CAAC,CAAC,GAAG,KAAK,IAAI,OAAO,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC;AACjD,CAAC"}

package/package.json

@@ -0,0 +1,51 @@
+{
+  "name": "@agent-shield/mcp",
+  "version": "0.1.0",
+  "description": "MCP server for AgentShield — manage on-chain vaults and enforce policies via AI tools",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "bin": {
+    "agent-shield-mcp": "dist/index.js"
+  },
+  "files": [
+    "dist/**/*",
+    "README.md"
+  ],
+  "engines": {
+    "node": ">=18.0.0"
+  },
+  "dependencies": {
+    "@modelcontextprotocol/sdk": "^1.12.1"
+  },
+  "peerDependencies": {
+    "@agent-shield/sdk": ">=0.1.5",
+    "@coral-xyz/anchor": ">=0.32.1",
+    "@solana/web3.js": ">=1.95.0",
+    "zod": ">=3.25.0"
+  },
+  "devDependencies": {
+    "@coral-xyz/anchor": "^0.32.1",
+    "@solana/web3.js": "^1.95.0",
+    "@types/chai": "^4.3.11",
+    "@types/mocha": "^10.0.6",
+    "@types/sinon": "^17.0.3",
+    "zod": "^3.25.0",
+    "chai": "^4.4.1",
+    "mocha": "^10.3.0",
+    "sinon": "^17.0.1",
+    "ts-node": "^10.9.2",
+    "typescript": "^5.3.3",
+    "@agent-shield/sdk": "0.1.6"
+  },
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/agent-shield/agent-shield",
+    "directory": "packages/mcp"
+  },
+  "scripts": {
+    "build": "tsc",
+    "clean": "rm -rf dist",
+    "test": "mocha --require ts-node/register 'tests/*.test.ts' 'tests/**/*.test.ts' --timeout 30000"
+  }
+}