Add rich menu apis

src/index.ts

@@ -20,6 +20,8 @@ import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 import * as line from "@line/bot-sdk";
 import { z } from "zod";
+import pkg from "../package.json" with { type: "json" };
+import fs from "fs";
 import { LINE_BOT_MCP_SERVER_VERSION, USER_AGENT } from "./version.js";
 
 const NO_USER_ID_ERROR =
@@ -40,6 +42,13 @@ const messagingApiClient = new line.messagingApi.MessagingApiClient({
   },
 });
 
+const lineBlobClient = new line.messagingApi.MessagingApiBlobClient({
+  channelAccessToken: channelAccessToken,
+  defaultHeaders: {
+    "User-Agent": `${pkg.name}/${pkg.version}`,
+  },
+});
+
 function createErrorResponse(message: string) {
   return {
     isError: true,
@@ -229,6 +238,90 @@ server.tool(
   },
 );
 
+server.tool(
+  "get_rich_menu_list",
+  "Get the list of rich menus associated with your LINE Official Account.",
+  {},
+  async () => {
+    try {
+      const response = await messagingApiClient.getRichMenuList();
+      return createSuccessResponse(response);
+    } catch (error) {
+      return createErrorResponse(
+        `Failed to broadcast message: ${error.message}`,
+      );
+    }
+  },
+);
+
+server.tool(
+  "delete_rich_menu",
+  "Delete a rich menu from your LINE Official Account.",
+  {
+    richMenuId: z.string().describe("The ID of the rich menu to delete."),
+  },
+  async ({ richMenuId }) => {
+    try {
+      const response = await messagingApiClient.deleteRichMenu(richMenuId);
+      return createSuccessResponse(response);
+    } catch (error) {
+      return createErrorResponse(
+        `Failed to delete rich menu: ${error.message}`,
+      );
+    }
+  },
+);
+
+server.tool(
+  "set_rich_menu_image",
+  "Update a rich menu associated with your LINE Official Account.",
+  {
+    richMenuId: z.string().describe("The ID of the rich menu to update."),
+    imagePath: z.string().describe("The path of the image to update."),
+  },
+  async ({ richMenuId, imagePath }) => {
+    try {
+      const imageBuffer = fs.readFileSync(imagePath);
+      const imageType = "image/png";
+      const imageBlob = new Blob([imageBuffer], { type: imageType });
+
+      const response = await lineBlobClient.setRichMenuImage(
+        richMenuId,
+        imageBlob,
+      );
+      return createSuccessResponse(response);
+    } catch (error) {
+      return createErrorResponse(
+        `Failed to update rich menu: ${error.message}`,
+      );
+    }
+  },
+);
+
+server.tool(
+  "set_rich_menu_default",
+  "Set a rich menu as the default rich menu.",
+  {
+    richMenuId: z
+      .string()
+      .describe("The ID of the rich menu to set as default."),
+  },
+  async ({ richMenuId }) => {
+    const response = await messagingApiClient.setDefaultRichMenu(richMenuId);
+    return createSuccessResponse(response);
+  },
+);
+
+server.tool(
+  "cancel_rich_menu_default",
+  "Cancel the default rich menu.",
+  {},
+  async () => {
+    const response = await messagingApiClient.cancelDefaultRichMenu();
+    return createSuccessResponse(response);
+  },
+);
+
 async function main() {
   if (!process.env.CHANNEL_ACCESS_TOKEN) {
     console.error("Please set CHANNEL_ACCESS_TOKEN");