69 lines
2.1 KiB
TypeScript
69 lines
2.1 KiB
TypeScript
|
import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
|
||
|
|
import { and, asc, eq, isNull } from "../drizzle.js";
|
||
|
|
import { z } from "zod";
|
||
|
|
import { db } from "../db.js";
|
||
|
|
import { objects } from "../schema.js";
|
||
|
|
import { objectTypes } from "../shared-types.js";
|
||
|
|
import { toolCatch, toolOk } from "./tool-result.js";
|
||
|
|
|
||
|
|
const objectTypeSchema = z.enum(objectTypes as unknown as [string, ...string[]]);
|
||
|
|
|
||
|
|
const listObjectsInputSchema = z.object({
|
||
|
|
workspaceId: z.string().uuid(),
|
||
|
|
parentId: z.string().uuid().nullable().optional(),
|
||
|
|
type: objectTypeSchema.optional(),
|
||
|
|
status: z.string().optional(),
|
||
|
|
limit: z.number().int().positive().max(500).optional(),
|
||
|
|
offset: z.number().int().nonnegative().optional(),
|
||
|
|
});
|
||
|
|
|
||
|
|
export function registerListObjectsTool(mcp: McpServer): void {
|
||
|
|
mcp.registerTool(
|
||
|
|
"list_objects",
|
||
|
|
{
|
||
|
|
description:
|
||
|
|
"List objects in a workspace with optional filters (parent, type, status) and pagination.",
|
||
|
|
inputSchema: listObjectsInputSchema,
|
||
|
|
},
|
||
|
|
async (args) => {
|
||
|
|
try {
|
||
|
|
const input = listObjectsInputSchema.parse(args);
|
||
|
|
const limit = input.limit ?? 50;
|
||
|
|
const offset = input.offset ?? 0;
|
||
|
|
|
||
|
|
const conditions = [eq(objects.workspaceId, input.workspaceId), isNull(objects.archivedAt)];
|
||
|
|
|
||
|
|
if (input.parentId === null) {
|
||
|
|
conditions.push(isNull(objects.parentId));
|
||
|
|
} else if (input.parentId !== undefined) {
|
||
|
|
conditions.push(eq(objects.parentId, input.parentId));
|
||
|
|
}
|
||
|
|
|
||
|
|
if (input.type !== undefined) {
|
||
|
|
conditions.push(eq(objects.type, input.type));
|
||
|
|
}
|
||
|
|
if (input.status !== undefined) {
|
||
|
|
conditions.push(eq(objects.status, input.status));
|
||
|
|
}
|
||
|
|
|
||
|
|
const rows = await db
|
||
|
|
.select()
|
||
|
|
.from(objects)
|
||
|
|
.where(and(...conditions))
|
||
|
|
.orderBy(asc(objects.sortOrder), asc(objects.id))
|
||
|
|
.limit(limit)
|
||
|
|
.offset(offset);
|
||
|
|
|
||
|
|
return toolOk({
|
||
|
|
objects: rows,
|
||
|
|
count: rows.length,
|
||
|
|
limit,
|
||
|
|
offset,
|
||
|
|
});
|
||
|
|
} catch (e) {
|
||
|
|
return toolCatch(e);
|
||
|
|
}
|
||
|
|
},
|
||
|
|
);
|
||
|
|
}
|