@burtthecoder/mcp-virustotal/build/handlers/collection.js

MCP server for VirusTotal API integration

import { queryVirusTotal, queryVirusTotalWithRelationships } from '../utils/api.js';
import { formatCollectionResults } from '../formatters/index.js';
import { logToFile } from '../utils/logging.js';
export async function handleGetCollection(args) {
    const { id, relationships } = args;
    logToFile(`Getting collection ${id}${relationships ? ` with relationships: ${relationships.join(',')}` : ''}`);
    const result = relationships && relationships.length > 0
        ? await queryVirusTotalWithRelationships(`/collections/${id}`, relationships)
        : await queryVirusTotal(`/collections/${id}`);
    return {
        content: [formatCollectionResults(result.data)],
    };
}