import { StarknetTool } from '@snakagent/core';
import { getSpecVersion } from '../actions/getSpecVersion.js';
import { getBlockWithTxHashes } from '../actions/getBlockWithTxHashes.js';
import { getBlockWithReceipts } from '../actions/getBlockWithReceipts.js';
import { getTransactionStatus } from '../actions/getTransactionStatus.js';
import { getClass } from '../actions/getClass.js';
import { getChainId } from '../actions/getChainId.js';
import { getSyncingStats } from '../actions/getSyncingStats.js';
import { getBlockNumber } from '../actions/getBlockNumber.js';
import { getBlockTransactionCount } from '../actions/getBlockTransactionCount.js';
import { getStorageAt } from '../actions/getStorageAt.js';
import { getClassAt } from '../actions/getClassAt.js';
import { getClassHashAt } from '../actions/getClassHash.js';
import {
getStorageAtSchema,
blockIdSchema,
blockIdAndContractAddressSchema,
getClassAtSchema,
getClassHashAtSchema,
transactionHashSchema,
emptyInputSchema,
} from '../schema/index.js';
export const registerTools = (SnakToolRegistry: StarknetTool[]) => {
SnakToolRegistry.push({
name: 'get_chain_id',
plugins: 'rpc',
description:
'Retrieve the unique identifier (chain ID) of the Starknet network',
schema: emptyInputSchema,
execute: getChainId,
});
SnakToolRegistry.push({
name: 'get_syncing_status',
plugins: 'rpc',
description: 'Retrieve the syncing status of the Starknet node',
schema: emptyInputSchema,
execute: getSyncingStats,
});
// Add remaining tools from createTools2
SnakToolRegistry.push({
name: 'get_class_hash',
plugins: 'rpc',
description:
'Retrieve the unique class hash for a contract at a specific address',
schema: getClassHashAtSchema,
execute: getClassHashAt,
});
SnakToolRegistry.push({
name: 'get_spec_version',
plugins: 'rpc',
description: 'Get the current spec version from the Starknet RPC provider',
schema: emptyInputSchema,
execute: getSpecVersion,
});
// SnakToolRegistry.push({
// name: 'get_block_with_tx_hashes',
// plugins: 'rpc',
// description:
// 'Retrieve the details of a block, including transaction hashes',
// schema: blockIdSchema,
// execute: getBlockWithTxHashes,
// });
// SnakToolRegistry.push({
// name: 'get_block_with_receipts',
// plugins: 'rpc',
// description: 'Fetch block details with transaction receipts',
// schema: blockIdSchema,
// execute: getBlockWithReceipts,
// });
SnakToolRegistry.push({
name: 'get_transaction_status',
plugins: 'rpc',
description: 'Fetch transaction status by hash',
schema: transactionHashSchema,
execute: getTransactionStatus,
});
// Register blockchain query tools
SnakToolRegistry.push({
name: 'get_block_number',
plugins: 'rpc',
description: 'Get the current block number from the Starknet network',
schema: emptyInputSchema,
execute: getBlockNumber,
});
SnakToolRegistry.push({
name: 'get_block_transaction_count',
plugins: 'rpc',
description: 'Get the number of transactions in a specific block',
schema: blockIdSchema,
execute: getBlockTransactionCount,
});
// SnakToolRegistry.push({
// name: 'get_storage_at',
// plugins: 'rpc',
// description: 'Get the storage value at a specific slot for a contract',
// schema: getStorageAtSchema,
// execute: getStorageAt,
// });
// Register contract-related tools
SnakToolRegistry.push({
name: 'get_class',
plugins: 'rpc',
description:
'Retrieve the complete class definition of a contract at a specified address and block',
schema: blockIdAndContractAddressSchema,
execute: getClass,
});
SnakToolRegistry.push({
name: 'get_class_at',
plugins: 'rpc',
description:
'Fetch the class definition of a contract at a specific address in the latest state',
schema: getClassAtSchema,
execute: getClassAt,
});
};
