import { z } from "zod";
import type { MCPTool } from "../types/tool.js";
import { VerticaService } from "../services/vertica-service.js";
import { getDatabaseConfig } from "../config/database.js";
import { safeJsonStringify } from "../utils/response-formatter.js";
import { validateSchemaName } from "../utils/table-helpers.js";
interface ListTablesInput {
schemaName?: string;
}
export default class ListTablesTool implements MCPTool {
name = "list_tables";
description = "List all tables in a schema with metadata";
inputSchema = {
type: "object" as const,
properties: {
schemaName: {
type: "string" as const,
description:
"Schema name (optional, defaults to configured default schema)",
},
},
};
async execute(input: Record<string, unknown>): Promise<string> {
// Validate input
const parsed = this.parseInput(input);
let verticaService: VerticaService | null = null;
try {
// Validate schema name if provided
if (parsed.schemaName) {
validateSchemaName(parsed.schemaName);
}
// Create Vertica service instance
const config = getDatabaseConfig();
verticaService = new VerticaService(config);
// List tables
const tables = await verticaService.listTables(parsed.schemaName);
return safeJsonStringify(
{
success: true,
schema: parsed.schemaName || config.defaultSchema || "public",
tableCount: tables.length,
tables: tables.map((table) => ({
schemaName: table.schemaName,
tableName: table.tableName,
owner: table.owner,
comment: table.comment,
rowCount: table.rowCount,
})),
queriedAt: new Date().toISOString(),
},
2
);
} catch (error) {
const errorMessage =
error instanceof Error ? error.message : String(error);
return safeJsonStringify(
{
success: false,
error: errorMessage,
schemaName: parsed.schemaName,
queriedAt: new Date().toISOString(),
},
2
);
} finally {
if (verticaService) {
try {
await verticaService.disconnect();
} catch (error) {
console.error("Warning during service cleanup:", error instanceof Error ? error.message : String(error));
}
}
}
}
private parseInput(input: Record<string, unknown>): ListTablesInput {
const schema = z.object({
schemaName: z.string().optional(),
});
return schema.parse(input);
}
}
