Merge branch 'dev'

Author: VeerChaurasia

bun.lock

@@ -9,7 +9,6 @@ import { privateKeyToAccount } from "viem/accounts";
 import dotenv from "dotenv";
 import { privateKeySchema } from "./tools/hyper-evm/sendFunds/schemas.js";
 import * as hyper from "@nktkas/hyperliquid";
-
 dotenv.config();
 
 export const hyperEvmConfig = defineChain({

package-lock.json

@@ -5,28 +5,36 @@ import {
   ListToolsRequestSchema,
   type CallToolRequest,
 } from "@modelcontextprotocol/sdk/types.js";
+
 import {
   GET_BALANCE_TOOL,
   GET_LATEST_BLOCK_TOOL,
   DEPLOY_CONTRACTS_TOOL,
   SEND_FUNDS_TOOL,
   GET_TRANSACTION_RECEIPT_TOOL,
   GET_TOKEN_BALANCE_TOOL,
+  FETCH_TRANSACTIONS_TOOL,
+  GET_LOGS_TOOL,
+  CALL_CONTRACT_FUNCTION,
   STAKE_TOOL,
   UNSTAKE_TOOL,
-  GET_LOGS_TOOL,
   GET_HISTORICAL_ORDERS_TOOL,
+  TRACK_STAKED_TOKENS,
 } from "./tools/tools.js";
 import { getBalance } from "./tools/hyper-evm/getBalance/index.js";
 import { getLatestBlock } from "./tools/hyper-evm/getBlockNumber/index.js";
 import { deployContracts } from "./tools/hyper-evm/deployContracts/index.js";
 import type { DeployContractsInput } from "./tools/hyper-evm/deployContracts/schemas.js";
+import { callContracts } from "./tools/hyper-evm/callContracts/index.js";
+import { CallContractSchema } from "./tools/hyper-evm/callContracts/schema.js";
 import { sendFunds } from "./tools/hyper-evm/sendFunds/index.js";
 import { sendFundsInputSchema } from "./tools/hyper-evm/sendFunds/schemas.js";
 import { getTransactionReceipt } from "./tools/hyper-evm/getTransactionReceipt/index.js";
 import type { getTransactionReceiptInput } from "./tools/hyper-evm/getTransactionReceipt/schemas.js";
 import { getTokenBalanceInputSchema } from "./tools/hyper-evm/getTokenBalance/schemas.js";
 import { getTokenBalance } from "./tools/hyper-evm/getTokenBalance/index.js";
+import { fetchTransactions } from "./tools/hyper-evm/fetchTransactions/index.js";
+import type { FetchTransactionsInput } from "./tools/hyper-evm/fetchTransactions/schemas.js";
 import {
   performStaking,
   performUnstaking,
@@ -37,9 +45,12 @@ import {
 } from "./tools/hyper-evm/handleStake/schemas.js";
 import { getLogs } from "./tools/hyper-evm/getLogs/index.js";
 import { getHistoricalOrders } from "./tools/hypercore/getHistoricalOrders/index.js";
+import { getStakedtokens } from "./tools/hypercore/trackstakedtokens/index.js";
+import { StakedInputSchema } from "./tools/hypercore/trackstakedtokens/schema.js";
 
 async function main() {
   console.error("Starting Hyperliquid MCP server...");
+
   const server = new Server(
     {
       name: "hyperliquid",
@@ -68,6 +79,54 @@ async function main() {
             return balance;
           }
 
+          case "call_contract_function": {
+            try {
+              const { contractAddress, functionName, abi, functionArgs } =
+                args as {
+                  contractAddress: string;
+                  functionName: string;
+                  abi: any;
+                  functionArgs?: any[];
+                };
+
+              const validatedInput = CallContractSchema.parse({
+                contractAddress,
+                functionName,
+                abi,
+                functionArgs,
+              });
+
+              const result = await callContracts(validatedInput);
+
+              return {
+                content: [
+                  {
+                    type: "text",
+                    text: JSON.stringify(
+                      result,
+                      (_, v) => (typeof v === "bigint" ? v.toString() : v),
+                      2
+                    ),
+                  },
+                ],
+              };
+            } catch (validationError) {
+              console.error("Validation error:", validationError);
+              return {
+                content: [
+                  {
+                    type: "text",
+                    text: `Error: ${
+                      validationError instanceof Error
+                        ? validationError.message
+                        : String(validationError)
+                    }`,
+                  },
+                ],
+              };
+            }
+          }
+
           case "deploy_contracts": {
             const input = args as DeployContractsInput;
             const result = await deployContracts(input);
@@ -110,6 +169,12 @@ async function main() {
             return result;
           }
 
+          case "fetch_transactions": {
+            const input = args as FetchTransactionsInput;
+            const result = await fetchTransactions(input);
+            return result;
+          }
+
           case "stake": {
             const input = args as {
               amountToStake: string;
@@ -151,9 +216,19 @@ async function main() {
             return result;
           }
 
+          case "track_staked_tokens": {
+            const input = args as {
+              userAddress: string;
+              isTestnet: boolean | string;
+            };
+            const validatedInput = StakedInputSchema.parse(input);
+            const result = await getStakedtokens(validatedInput);
+            return result;
+          }
+
           default: {
             throw new Error(
-              `Tool '${name}' not found. Available tools: get_latest_block, get_balance, deploy_contracts, send_funds, get_transaction_receipt, get_token_balance, stake, unstake`
+              `Tool '${name}' not found. Available tools: get_latest_block, get_balance, deploy_contracts, send_funds, get_transaction_receipt, get_token_balance, stake, unstake, get_logs, call_contract_function, track_staked_tokens`
             );
           }
         }
@@ -181,10 +256,13 @@ async function main() {
         SEND_FUNDS_TOOL,
         GET_TRANSACTION_RECEIPT_TOOL,
         GET_TOKEN_BALANCE_TOOL,
+        FETCH_TRANSACTIONS_TOOL,
+        CALL_CONTRACT_FUNCTION,
         STAKE_TOOL,
         UNSTAKE_TOOL,
         GET_LOGS_TOOL,
         GET_HISTORICAL_ORDERS_TOOL,
+        TRACK_STAKED_TOKENS,
       ],
     };
   });

src/config.ts

@@ -0,0 +1,98 @@
+import {
+  encodeFunctionData,
+  type Abi,
+  type AbiFunction,
+  isAddress,
+} from "viem";
+import { createWalletClient, http } from "viem";
+import { publicClient, walletClient } from "../../../config.js";
+import { hyperEvmConfig } from "../../../config.js";
+import type { GetContractDetails } from "./schema.js";
+import { privateKeySchema } from "../sendFunds/schemas.js";
+import { privateKeyToAccount } from "viem/accounts";
+
+export async function callContracts(contractDetails: GetContractDetails) {
+  try {
+    const address = contractDetails.contractAddress;
+    if (!isAddress(address)) {
+      throw new Error(`Invalid HyperEVM address: ${address}`);
+    }
+
+    let abi: Abi;
+    if (typeof contractDetails.abi === "string") {
+      try {
+        abi = JSON.parse(contractDetails.abi) as Abi;
+      } catch (error) {
+        throw new Error(`Invalid ABI string: ${error}`);
+      }
+    } else {
+      abi = contractDetails.abi as Abi;
+    }
+
+    const functionAbi = abi.find(
+      (item): item is AbiFunction =>
+        "type" in item &&
+        item.type === "function" &&
+        "name" in item &&
+        item.name === contractDetails.functionName
+    );
+
+    if (!functionAbi) {
+      throw new Error(
+        `Function ${contractDetails.functionName} not found in ABI`
+      );
+    }
+
+    if (
+      functionAbi.stateMutability === "view" ||
+      functionAbi.stateMutability === "pure"
+    ) {
+      const callParams: any = {
+        address: contractDetails.contractAddress,
+        abi,
+        functionName: contractDetails.functionName,
+      };
+
+      if (functionAbi.inputs && functionAbi.inputs.length > 0) {
+        if (
+          !contractDetails.functionArgs ||
+          contractDetails.functionArgs.length !== functionAbi.inputs.length
+        ) {
+          throw Error(
+            `Function ${contractDetails.functionName} expects ${functionAbi.inputs.length} arguments, ` +
+              `${contractDetails.functionArgs?.length || 0} provided`
+          );
+        }
+        callParams.args = contractDetails.functionArgs;
+      }
+
+      return await publicClient.readContract(callParams);
+    }
+
+    const callParams: any = {
+      address: contractDetails.contractAddress,
+      abi,
+      functionName: contractDetails.functionName,
+    };
+
+    if (functionAbi.inputs && functionAbi.inputs.length > 0) {
+      if (
+        !contractDetails.functionArgs ||
+        contractDetails.functionArgs.length !== functionAbi.inputs.length
+      ) {
+        throw Error(
+          `Function ${contractDetails.functionName} expects ${functionAbi.inputs.length} arguments, ` +
+            `${contractDetails.functionArgs?.length || 0} provided`
+        );
+      }
+      callParams.args = contractDetails.functionArgs;
+    }
+
+    return await walletClient.writeContract(callParams);
+  } catch (error) {
+    console.log(error);
+    throw Error(
+      `Failed to call contract function: ${error instanceof Error ? error.message : String(error)}`
+    );
+  }
+}

src/main.ts

@@ -0,0 +1,22 @@
+import { z } from "zod";
+import { isAddress } from "viem";
+
+export const CallContractSchema = z.object({
+  contractAddress: z.string().refine(address => isAddress(address), {
+    message: "Must be a valid HyperEVM address (0x format, 42 characters)",
+  }),
+  functionName: z.string().min(1, "Function name cannot be empty"),
+  abi: z.union([z.string(), z.array(z.any())]),
+  functionArgs: z.preprocess(val => {
+    if (typeof val === "string") {
+      try {
+        return JSON.parse(val);
+      } catch {
+        return [val];
+      }
+    }
+    return val;
+  }, z.array(z.any()).optional()),
+});
+
+export type GetContractDetails = z.infer<typeof CallContractSchema>;