notion-mcp-server 1.0.0 → 2.4.2

package/build/schema/tools.js

@@ -1,17 +0,0 @@
-// import { z } from "zod";
-// import { blockSchema } from "./blocks.js";
-// import { parentSchema } from "./page.js";
-export {};
-// export const createPageSchema = {
-//   parent: parentSchema,
-//   properties: z.object({
-//     title: z.object({
-//       title: z.array(z.any()), // Rich text array
-//     }),
-//   }),
-//   children: z.array(blockSchema).optional(),
-// };
-// export const appendBlocksSchema = {
-//   block_id: z.string(),
-//   children: z.array(blockSchema),
-// };

package/build/schema/users.js

@@ -1,13 +0,0 @@
-import { z } from "zod";
-// Schema for listing users with pagination
-export const LIST_USERS_SCHEMA = {
-    start_cursor: z.string().optional().describe("Pagination cursor"),
-    page_size: z
-        .number()
-        .optional()
-        .describe("Number of users to return per page"),
-};
-// Schema for getting a single user
-export const GET_USER_SCHEMA = {
-    user_id: z.string().describe("The ID of the user to retrieve"),
-};

package/build/services/replicate.js

@@ -1,23 +0,0 @@
-import Replicate from "replicate";
-import { CONFIG } from "../config/index.js";
-export function getReplicateApiToken() {
-    const token = process.env.REPLICATE_API_TOKEN;
-    if (!token) {
-        console.error("Error: REPLICATE_API_TOKEN environment variable is required");
-        process.exit(1);
-    }
-    return token;
-}
-export const replicate = new Replicate({
-    auth: getReplicateApiToken(),
-});
-export async function pollForCompletion(predictionId) {
-    for (let i = 0; i < CONFIG.pollingAttempts; i++) {
-        const latest = await replicate.predictions.get(predictionId);
-        if (latest.status !== "starting" && latest.status !== "processing") {
-            return latest;
-        }
-        await new Promise((resolve) => setTimeout(resolve, CONFIG.pollingInterval));
-    }
-    return null;
-}

package/build/tools/appendBlockChildren.js

@@ -1,25 +0,0 @@
-import { notion } from "../services/notion.js";
-import { handleNotionError } from "../utils/error.js";
-export const appendBlockChildren = async (params) => {
-    try {
-        const response = await notion.blocks.children.append({
-            block_id: params.blockId,
-            children: params.children,
-        });
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: `Successfully appended ${params.children.length} block(s) to ${params.blockId}`,
-                },
-                {
-                    type: "text",
-                    text: `Block ID: ${JSON.stringify(response, null, 2)}`,
-                },
-            ],
-        };
-    }
-    catch (error) {
-        return handleNotionError(error);
-    }
-};

package/build/tools/batchAppendBlockChildren.js

@@ -1,33 +0,0 @@
-import { notion } from "../services/notion.js";
-import { handleNotionError } from "../utils/error.js";
-export const batchAppendBlockChildren = async (params) => {
-    try {
-        const results = [];
-        for (const operation of params.operations) {
-            const response = await notion.blocks.children.append({
-                block_id: operation.blockId,
-                children: operation.children,
-            });
-            results.push({
-                blockId: operation.blockId,
-                success: true,
-                response,
-            });
-        }
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: `Successfully completed ${params.operations.length} append operations`,
-                },
-                {
-                    type: "text",
-                    text: JSON.stringify(results, null, 2),
-                },
-            ],
-        };
-    }
-    catch (error) {
-        return handleNotionError(error);
-    }
-};

package/build/tools/batchDeleteBlocks.js

@@ -1,32 +0,0 @@
-import { notion } from "../services/notion.js";
-import { handleNotionError } from "../utils/error.js";
-export const batchDeleteBlocks = async (params) => {
-    try {
-        const results = [];
-        for (const blockId of params.blockIds) {
-            const response = await notion.blocks.delete({
-                block_id: blockId,
-            });
-            results.push({
-                blockId,
-                success: true,
-                response,
-            });
-        }
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: `Successfully deleted ${params.blockIds.length} blocks (moved to trash)`,
-                },
-                {
-                    type: "text",
-                    text: JSON.stringify(results, null, 2),
-                },
-            ],
-        };
-    }
-    catch (error) {
-        return handleNotionError(error);
-    }
-};

package/build/tools/batchMixedOperations.js

@@ -1,58 +0,0 @@
-import { notion } from "../services/notion.js";
-import { handleNotionError } from "../utils/error.js";
-export const batchMixedOperations = async (params) => {
-    try {
-        const results = [];
-        const operationCounts = {
-            append: 0,
-            update: 0,
-            delete: 0,
-        };
-        for (const op of params.operations) {
-            let response;
-            switch (op.operation) {
-                case "append":
-                    response = await notion.blocks.children.append({
-                        block_id: op.blockId,
-                        children: op.children,
-                    });
-                    operationCounts.append++;
-                    break;
-                case "update":
-                    response = await notion.blocks.update({
-                        block_id: op.blockId,
-                        ...op.data,
-                    });
-                    operationCounts.update++;
-                    break;
-                case "delete":
-                    response = await notion.blocks.delete({
-                        block_id: op.blockId,
-                    });
-                    operationCounts.delete++;
-                    break;
-            }
-            results.push({
-                operation: op.operation,
-                blockId: op.blockId,
-                success: true,
-                response,
-            });
-        }
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: `Successfully performed ${params.operations.length} operations (${operationCounts.append} append, ${operationCounts.update} update, ${operationCounts.delete} delete)`,
-                },
-                {
-                    type: "text",
-                    text: JSON.stringify(results, null, 2),
-                },
-            ],
-        };
-    }
-    catch (error) {
-        return handleNotionError(error);
-    }
-};

package/build/tools/batchUpdateBlocks.js

@@ -1,33 +0,0 @@
-import { notion } from "../services/notion.js";
-import { handleNotionError } from "../utils/error.js";
-export const batchUpdateBlocks = async (params) => {
-    try {
-        const results = [];
-        for (const operation of params.operations) {
-            const response = await notion.blocks.update({
-                block_id: operation.blockId,
-                ...operation.data,
-            });
-            results.push({
-                blockId: operation.blockId,
-                success: true,
-                response,
-            });
-        }
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: `Successfully updated ${params.operations.length} blocks`,
-                },
-                {
-                    type: "text",
-                    text: JSON.stringify(results, null, 2),
-                },
-            ],
-        };
-    }
-    catch (error) {
-        return handleNotionError(error);
-    }
-};

package/build/tools/comments.js

@@ -1,62 +0,0 @@
-import { notion } from "../services/notion.js";
-import { handleNotionError } from "../utils/error.js";
-export const registerGetCommentsTool = async (params) => {
-    try {
-        const response = await notion.comments.list(params);
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: `Comments retrieved successfully: ${response.results.length}`,
-                },
-                {
-                    type: "text",
-                    text: JSON.stringify(response, null, 2),
-                },
-            ],
-        };
-    }
-    catch (error) {
-        return handleNotionError(error);
-    }
-};
-export const registerAddPageCommentTool = async (params) => {
-    try {
-        const response = await notion.comments.create(params);
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: `Comment created successfully: ${response.id}`,
-                },
-                {
-                    type: "text",
-                    text: JSON.stringify(response, null, 2),
-                },
-            ],
-        };
-    }
-    catch (error) {
-        return handleNotionError(error);
-    }
-};
-export const registerAddDiscussionCommentTool = async (params) => {
-    try {
-        const response = await notion.comments.create(params);
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: `Comment created successfully: ${response.id}`,
-                },
-                {
-                    type: "text",
-                    text: JSON.stringify(response, null, 2),
-                },
-            ],
-        };
-    }
-    catch (error) {
-        return handleNotionError(error);
-    }
-};

package/build/tools/createDatabase.js

@@ -1,18 +0,0 @@
-import { notion } from "../services/notion.js";
-import { handleNotionError } from "../utils/error.js";
-export const createDatabase = async (params) => {
-    try {
-        const response = await notion.databases.create(params);
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: `Database created successfully: ${response.id}`,
-                },
-            ],
-        };
-    }
-    catch (error) {
-        return handleNotionError(error);
-    }
-};

package/build/tools/createPage.js

@@ -1,18 +0,0 @@
-import { notion } from "../services/notion.js";
-import { handleNotionError } from "../utils/error.js";
-export const registerCreatePageTool = async (params) => {
-    try {
-        const response = await notion.pages.create(params);
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: `Page created successfully: ${response.id}`,
-                },
-            ],
-        };
-    }
-    catch (error) {
-        return handleNotionError(error);
-    }
-};

package/build/tools/createPrediction.js

@@ -1,28 +0,0 @@
-// import { CreatePredictionParams } from "../types/index.js";
-// import { pollForCompletion, replicate } from "../services/replicate.js";
-// import { handleError } from "../utils/error.js";
-// import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
-// import { CONFIG } from "../config/index.js";
-export {};
-// export const registerCreatePredictionTool = async (
-//   input: CreatePredictionParams
-// ): Promise<CallToolResult> => {
-//   try {
-//     const prediction = await replicate.predictions.create({
-//       model: CONFIG.imageModelId,
-//       input,
-//     });
-//     await replicate.predictions.get(prediction.id);
-//     const completed = await pollForCompletion(prediction.id);
-//     return {
-//       content: [
-//         {
-//           type: "text",
-//           text: JSON.stringify(completed || "Processing timed out", null, 2),
-//         },
-//       ],
-//     };
-//   } catch (error) {
-//     handleError(error);
-//   }
-// };

package/build/tools/deleteBlock.js

@@ -1,24 +0,0 @@
-import { notion } from "../services/notion.js";
-import { handleNotionError } from "../utils/error.js";
-export const deleteBlock = async (params) => {
-    try {
-        const response = await notion.blocks.delete({
-            block_id: params.blockId,
-        });
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: `Block ${params.blockId} deleted (moved to trash) successfully`,
-                },
-                {
-                    type: "text",
-                    text: JSON.stringify(response, null, 2),
-                },
-            ],
-        };
-    }
-    catch (error) {
-        return handleNotionError(error);
-    }
-};

package/build/tools/formatRichText.js

@@ -1,83 +0,0 @@
-import { z } from "zod";
-import { handleError } from "../utils/error.js";
-import { boldText, italicText, codeText, colorText, createTextRichText, } from "../utils/richText.js";
-// Schema for the formatRichText tool as a record of ZodTypes
-export const formatRichTextSchema = {
-    text: z.string().describe("Text content to format"),
-    include_bold: z
-        .boolean()
-        .optional()
-        .describe("Whether to include bold formatting"),
-    include_italic: z
-        .boolean()
-        .optional()
-        .describe("Whether to include italic formatting"),
-    include_code: z
-        .boolean()
-        .optional()
-        .describe("Whether to include code formatting"),
-    include_color: z
-        .boolean()
-        .optional()
-        .describe("Whether to include color formatting"),
-    include_link: z.boolean().optional().describe("Whether to include a link"),
-    link_url: z
-        .string()
-        .url()
-        .optional()
-        .describe("URL to link to if include_link is true"),
-    color: z
-        .string()
-        .optional()
-        .describe("Color to use if include_color is true"),
-};
-// Object schema for type inference
-const formatRichTextObjectSchema = z.object(formatRichTextSchema);
-/**
- * Tool that demonstrates rich text formatting in Notion
- * @param params The parameters for formatting rich text
- * @returns A formatted JSON representation of the rich text
- */
-export const registerFormatRichTextTool = async (params) => {
-    try {
-        const { text, include_bold = false, include_italic = false, include_code = false, include_color = false, include_link = false, link_url = "https://example.com", color = "blue", } = params;
-        // Split the text into parts to demonstrate formatting each part differently
-        const parts = text.split(" ");
-        const formattedParts = [];
-        // Format each part based on the parameters
-        for (let i = 0; i < parts.length; i++) {
-            const part = parts[i];
-            if (i % 5 === 0 && include_bold) {
-                formattedParts.push(boldText(part + " "));
-            }
-            else if (i % 5 === 1 && include_italic) {
-                formattedParts.push(italicText(part + " "));
-            }
-            else if (i % 5 === 2 && include_code) {
-                formattedParts.push(codeText(part + " "));
-            }
-            else if (i % 5 === 3 && include_color) {
-                formattedParts.push(colorText(part + " ", color));
-            }
-            else if (i % 5 === 4 && include_link) {
-                formattedParts.push(createTextRichText(part + " ", undefined, link_url));
-            }
-            else {
-                formattedParts.push(createTextRichText(part + " "));
-            }
-        }
-        // The resulting rich text array
-        const richText = formattedParts;
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: JSON.stringify(richText, null, 2),
-                },
-            ],
-        };
-    }
-    catch (error) {
-        return handleError(error);
-    }
-};

package/build/tools/generateImage.js

@@ -1,48 +0,0 @@
-import { replicate } from "../services/replicate.js";
-import { handleError } from "../utils/error.js";
-import { outputToBase64 } from "../utils/image.js";
-import { CONFIG } from "../config/index.js";
-export const registerGenerateImageTool = async (input) => {
-    const { support_image_mcp_response_type, ...predictionInput } = input;
-    try {
-        const [output] = (await replicate.run(CONFIG.imageModelId, {
-            input: predictionInput,
-        }));
-        const imageUrl = output.url();
-        if (support_image_mcp_response_type) {
-            const imageBase64 = await outputToBase64(output);
-            return {
-                content: [
-                    {
-                        type: "text",
-                        text: `This is a generated image link: ${imageUrl}`,
-                    },
-                    {
-                        type: "image",
-                        data: imageBase64,
-                        mimeType: "image/png",
-                    },
-                    {
-                        type: "text",
-                        text: `The image above is generated by the Flux model and prompt: ${input.prompt}`,
-                    },
-                ],
-            };
-        }
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: `This is a generated image link: ${imageUrl}`,
-                },
-                {
-                    type: "text",
-                    text: `The image above is generated by the Flux model and prompt: ${input.prompt}`,
-                },
-            ],
-        };
-    }
-    catch (error) {
-        handleError(error);
-    }
-};

package/build/tools/generateImageVariants.js

@@ -1,105 +0,0 @@
-import { replicate } from "../services/replicate.js";
-import { handleError } from "../utils/error.js";
-import { outputToBase64 } from "../utils/image.js";
-import { CONFIG } from "../config/index.js";
-export const registerGenerateImageVariantsTool = async (input) => {
-    const { prompt, num_variants, seed, support_image_mcp_response_type, prompt_variations, variation_mode, ...commonParams } = input;
-    try {
-        let effectiveVariants = num_variants;
-        let usingPromptVariations = false;
-        // Decide if we're using prompt variations
-        if (prompt_variations && prompt_variations.length > 0) {
-            usingPromptVariations = true;
-            // If using prompt variations, number of variants is limited by available variations
-            effectiveVariants = Math.min(num_variants, prompt_variations.length);
-        }
-        // Process all variants in parallel
-        const generationPromises = Array.from({ length: effectiveVariants }, (_, index) => {
-            // If seed is provided, create deterministic variants by adding the index
-            const variantSeed = seed !== undefined ? seed + index : undefined;
-            // Determine which prompt to use for this variant
-            let variantPrompt = prompt;
-            if (usingPromptVariations) {
-                const variation = prompt_variations[index];
-                if (variation_mode === "append") {
-                    variantPrompt = `${prompt} ${variation}`;
-                }
-                else {
-                    // 'replace' mode
-                    variantPrompt = variation;
-                }
-            }
-            return replicate
-                .run(CONFIG.imageModelId, {
-                input: {
-                    prompt: variantPrompt,
-                    seed: variantSeed,
-                    ...commonParams,
-                },
-            })
-                .then((outputs) => {
-                const [output] = outputs;
-                const imageUrl = output.url();
-                if (support_image_mcp_response_type) {
-                    return outputToBase64(output).then((imageBase64) => ({
-                        variantIndex: index + 1,
-                        imageUrl,
-                        imageBase64,
-                        usedPrompt: variantPrompt,
-                    }));
-                }
-                return {
-                    variantIndex: index + 1,
-                    imageUrl,
-                    usedPrompt: variantPrompt,
-                };
-            });
-        });
-        // Wait for all variant generation to complete
-        const results = (await Promise.all(generationPromises));
-        // Build response content
-        const responseContent = [];
-        // Add intro text - different based on whether we're using prompt variations
-        if (usingPromptVariations) {
-            responseContent.push({
-                type: "text",
-                text: `Generated ${results.length} variants of "${prompt}" using custom prompt variations (${variation_mode} mode)`,
-            });
-        }
-        else {
-            responseContent.push({
-                type: "text",
-                text: `Generated ${results.length} variants of: "${prompt}" using seed variations`,
-            });
-        }
-        // Add each variant with its index and prompt info
-        for (const result of results) {
-            // Build an appropriate description based on variant type
-            let variantDescription = `Variant #${result.variantIndex}`;
-            if (usingPromptVariations) {
-                variantDescription += `\nPrompt: "${result.usedPrompt}"`;
-            }
-            else if (seed !== undefined) {
-                variantDescription += ` (seed: ${seed + (result.variantIndex - 1)})`;
-            }
-            variantDescription += `\nImage URL: ${result.imageUrl}`;
-            responseContent.push({
-                type: "text",
-                text: `\n\n${variantDescription}`,
-            });
-            if (support_image_mcp_response_type && result.imageBase64) {
-                responseContent.push({
-                    type: "image",
-                    data: result.imageBase64,
-                    mimeType: `image/${input.output_format === "jpg" ? "jpeg" : input.output_format}`,
-                });
-            }
-        }
-        return {
-            content: responseContent,
-        };
-    }
-    catch (error) {
-        handleError(error);
-    }
-};