import { Server } from "@modelcontextprotocol/sdk/server/index.js"; import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js"; import { CallToolRequestSchema, ListToolsRequestSchema, } from "@modelcontextprotocol/sdk/types.js"; import axios from "axios"; import * as FormData from "form-data"; const BACKEND_URL = "http://localhost:3051/api"; const ADMIN_USERNAME = process.env.ADMIN_USERNAME || "admin"; const ADMIN_PASSWORD = process.env.ADMIN_PASSWORD || "homaroma123"; // เปลี่ยนรหัสผ่านตรงนี้ให้ตรงกับ Production ด้วยนะครับ let authToken = ""; export const api = axios.create({ baseURL: BACKEND_URL, }); api.interceptors.request.use(async (config) => { if (!authToken && config.url !== "/auth/login") { console.error("[MCP] No token found. Attempting auto-login..."); await performLogin(); } if (authToken) { config.headers.Authorization = `Bearer ${authToken}`; } return config; }); async function performLogin() { try { const res = await axios.post(`${BACKEND_URL}/auth/login`, { username: ADMIN_USERNAME, password: ADMIN_PASSWORD, }); if (res.data && res.data.token) { authToken = res.data.token; console.error("[MCP] Successfully logged in to Backend API."); } } catch (err: any) { console.error("[MCP] Auto-login failed:", err.message); } } const server = new Server( { name: "homaroma-mcp", version: "1.0.0", }, { capabilities: { tools: {}, }, } ); server.setRequestHandler(ListToolsRequestSchema, async () => { return { tools: [ { name: "ping_backend", description: "Ping the backend to verify connection and authentication", inputSchema: { type: "object", properties: {} }, }, { name: "search_products", description: "Search for products and get their basic info", inputSchema: { type: "object", properties: { search: { type: "string", description: "Search term" }, limit: { type: "number", description: "Max results" }, lang: { type: "string", description: "Language code (e.g. th, en, all)" } } } }, { name: "get_product_detail", description: "Get full product details including variants, images, and category", inputSchema: { type: "object", properties: { id: { type: "number", description: "Product ID" }, lang: { type: "string", description: "Language code (e.g. th, en, all)" } }, required: ["id"] } }, { name: "update_category_translation", description: "Add or update translations for a specific category. Do NOT use this tool for products.", inputSchema: { type: "object", properties: { id: { type: "number", description: "Category ID" }, lang: { type: "string", description: "Language code (th, en, ja, zh, vi, ko)" }, name: { type: "string", description: "Translated category name" } }, required: ["id", "lang", "name"] } }, { name: "update_scent_translation", description: "Add or update translations for a specific scent.", inputSchema: { type: "object", properties: { id: { type: "number", description: "Scent ID" }, lang: { type: "string", description: "Language code (th, en, ja, zh, vi, ko)" }, name: { type: "string", description: "Translated scent name" } }, required: ["id", "lang", "name"] } }, { name: "update_page_translation", description: "Add or update translations for a specific page.", inputSchema: { type: "object", properties: { id: { type: "number", description: "Page ID" }, lang: { type: "string", description: "Language code (th, en, ja, zh, vi, ko)" }, title: { type: "string", description: "Translated page title" } }, required: ["id", "lang", "title"] } }, { name: "update_hero_translation", description: "Add or update translations for a specific hero.", inputSchema: { type: "object", properties: { id: { type: "number", description: "Hero ID" }, lang: { type: "string", description: "Language code" }, tagline: { type: "string" }, title: { type: "string" }, description: { type: "string" }, buttonText: { type: "string" } }, required: ["id", "lang"] } }, { name: "update_menu_item_translation", description: "Add or update translations for a specific menu item.", inputSchema: { type: "object", properties: { id: { type: "number", description: "Menu Item ID" }, lang: { type: "string", description: "Language code" }, label: { type: "string", description: "Translated menu item label" } }, required: ["id", "lang", "label"] } }, { name: "update_store_settings_translation", description: "Add or update translations for store settings.", inputSchema: { type: "object", properties: { lang: { type: "string", description: "Language code" }, footerDescription: { type: "string" }, newsletterTitle: { type: "string" }, newsletterDescription: { type: "string" }, newsletterPlaceholder: { type: "string" }, newsletterButtonText: { type: "string" } }, required: ["lang"] } }, { name: "update_article_translation", description: "Add or update translations for a specific article.", inputSchema: { type: "object", properties: { id: { type: "number", description: "Article ID" }, lang: { type: "string", description: "Language code" }, title: { type: "string" }, content: { type: "string" } }, required: ["id", "lang"] } }, { name: "update_product_content", description: "Update product basic content (name, description, details)", inputSchema: { type: "object", properties: { id: { type: "number" }, name: { type: "string" }, description: { type: "string" }, details: { type: "string" }, }, required: ["id"] } }, { name: "update_product_translation", description: "Update product translation for a specific language. Use this to add missing translations.", inputSchema: { type: "object", properties: { id: { type: "number", description: "Product ID" }, lang: { type: "string", description: "Language code (e.g., en, ja, zh, vi, ko)" }, name: { type: "string", description: "Translated product name" }, description: { type: "string", description: "Translated description" }, details: { type: "string", description: "Translated details (HTML or Markdown allowed)" } }, required: ["id", "lang"] } }, { name: "upload_product_image_from_url", description: "Download image from URL, upload to backend, and append to product images", inputSchema: { type: "object", properties: { productId: { type: "number" }, imageUrl: { type: "string", description: "URL of the image to download" } }, required: ["productId", "imageUrl"] } }, { name: "generate_mock_review", description: "Create a dummy review for a product", inputSchema: { type: "object", properties: { productId: { type: "number" }, rating: { type: "number", description: "1-5" }, comment: { type: "string" }, customerName: { type: "string", description: "Name of the mock reviewer" } }, required: ["productId", "rating", "customerName"] } }, { name: "get_order_status", description: "Get the status and details of a specific order", inputSchema: { type: "object", properties: { orderId: { type: "number" } }, required: ["orderId"] } }, { name: "list_pending_orders", description: "List orders that are pending payment or processing", inputSchema: { type: "object", properties: { limit: { type: "number", description: "Max results" } } } }, { name: "update_order_tracking", description: "Update the tracking number of an order and optionally its status", inputSchema: { type: "object", properties: { orderId: { type: "number" }, trackingNo: { type: "string" }, status: { type: "string", description: "Optional status to update (e.g. SHIPPED)" } }, required: ["orderId", "trackingNo"] } }, { name: "search_customer", description: "Search for customers by name, email, phone, or line ID", inputSchema: { type: "object", properties: { search: { type: "string", description: "Search term" } } } }, { name: "adjust_customer_points", description: "Add or deduct points for a customer", inputSchema: { type: "object", properties: { id: { type: "number", description: "Customer ID" }, pointsDelta: { type: "number", description: "Amount of points to add (positive) or deduct (negative)" }, reason: { type: "string", description: "Reason for the adjustment" } }, required: ["id", "pointsDelta"] } }, { name: "generate_special_coupon", description: "Generate a special discount coupon", inputSchema: { type: "object", properties: { code: { type: "string", description: "Coupon code (e.g. VIP20)" }, type: { type: "string", description: "PERCENTAGE, FIXED_AMOUNT, or FREE_SHIPPING" }, value: { type: "number", description: "Discount value" }, minPurchase: { type: "number", description: "Minimum purchase amount" }, maxDiscount: { type: "number", description: "Maximum discount amount for PERCENTAGE type" }, usageLimit: { type: "number", description: "Maximum number of times this coupon can be used" }, expiresAt: { type: "string", description: "ISO date string for expiration" } }, required: ["code", "type", "value"] } }, { name: "list_dynamic_pages", description: "List all dynamic pages and layouts managed by the CMS", inputSchema: { type: "object", properties: {} } }, { name: "create_dynamic_page", description: "Create or update a dynamic page or layout", inputSchema: { type: "object", properties: { title: { type: "string", description: "Title of the page" }, slug: { type: "string", description: "Unique slug identifier (e.g. about-us)" }, isPublished: { type: "boolean", description: "Publish status" }, status: { type: "string", description: "Publish status (e.g. 'published' or 'draft')" }, metadata: { type: "object", description: "SEO metadata" }, id: { type: "number", description: "Numeric page ID (if updating existing page)" } }, required: ["title", "slug"] } }, { name: "add_page_section", description: "Add a block section to a dynamic page layout", inputSchema: { type: "object", properties: { pageId: { type: "number", description: "The numeric ID of the page" }, type: { type: "string", description: "Block type (e.g. HERO, ANNOUNCEMENT_BAR, NEWSLETTER, EDITORIAL_TILE, EDITORIAL_CAROUSEL, PRODUCTS, CATEGORIES, TEXT, BANNER, HERO_ARTICLE, RICH_TEXT, SCENT_NOTES_BREAKDOWN, QUOTE_CALLOUT, PRODUCT_CAROUSEL)" }, order: { type: "number", description: "Display order sequence (starts at 1)" }, payload: { type: "object", description: "The JSON payload matching the section type rules" }, isActive: { type: "boolean", description: "Initial active state" } }, required: ["pageId", "type", "payload"] } }, { name: "update_page_section", description: "Update an existing block section payload, order or active state", inputSchema: { type: "object", properties: { sectionId: { type: "number", description: "The numeric ID of the section to update" }, type: { type: "string", description: "Optional new block type" }, order: { type: "number", description: "Optional new display order sequence" }, payload: { type: "object", description: "Optional new JSON payload matching the section type rules" }, isActive: { type: "boolean", description: "Optional new active state" } }, required: ["sectionId"] } }, { name: "delete_page_section", description: "Delete a block section from a dynamic page layout", inputSchema: { type: "object", properties: { sectionId: { type: "number", description: "The numeric ID of the section to delete" } }, required: ["sectionId"] } }, { name: "update_footer_content", description: "Update the content and links of the 4 Footer Blocks (Brand, Explore, Collections, Newsletter)", inputSchema: { type: "object", properties: { brandDescription: { type: "string", description: "Text for Block 1 (Brand tagline)" }, socialLinks: { type: "object", description: "URLs for facebook, instagram, tiktok (e.g. { facebookUrl: '...' })", properties: { facebookUrl: { type: "string" }, instagramUrl: { type: "string" }, tiktokUrl: { type: "string" } } }, exploreLinks: { type: "array", description: "Array of { label, url, order } for Block 2 (Explore)", items: { type: "object", properties: { label: { type: "string" }, url: { type: "string" }, order: { type: "number" } } } }, collectionLinks: { type: "array", description: "Array of { label, url, order } for Block 3 (Collections)", items: { type: "object", properties: { label: { type: "string" }, url: { type: "string" }, order: { type: "number" } } } }, supportLinks: { type: "array", description: "Array of { label, url, order } for Block 4 (Support/Legal)", items: { type: "object", properties: { label: { type: "string" }, url: { type: "string" }, order: { type: "number" } } } } } } }, { name: "audit_missing_pages", description: "Audit all menus and find links that do not have a corresponding dynamic page or hardcoded route", inputSchema: { type: "object", properties: {} } } ], }; }); server.setRequestHandler(CallToolRequestSchema, async (request) => { const params = request.params.arguments as any; try { switch (request.params.name) { case "ping_backend": { const res = await api.get("/products?limit=1"); return { content: [{ type: "text", text: `Connection successful!` }] }; } case "search_products": { const search = params.search ? `&search=${encodeURIComponent(params.search)}` : ""; const limit = params.limit ? `&limit=${params.limit}` : "&limit=10"; const lang = params.lang ? `&lang=${params.lang}` : ""; const res = await api.get(`/products?${search}${limit}${lang}`); return { content: [{ type: "text", text: JSON.stringify(res.data, null, 2) }] }; } case "get_product_detail": { const lang = params.lang ? `?lang=${params.lang}` : ""; const res = await api.get(`/products/${params.id}${lang}`); return { content: [{ type: "text", text: JSON.stringify(res.data, null, 2) }] }; } case "update_category_translation": { const { id, lang, name } = params as any; const res = await api.patch(`/categories/${id}/translations`, { name: { [lang]: name } }); return { content: [{ type: "text", text: JSON.stringify(res.data, null, 2) }] }; } case "update_scent_translation": { const { id, lang, name } = params as any; const res = await api.patch(`/products/scents/${id}/translations`, { name: { [lang]: name } }); return { content: [{ type: "text", text: JSON.stringify(res.data, null, 2) }] }; } case "update_page_translation": { const { id, lang, title } = params as any; const res = await api.patch(`/pages/${id}/translations`, { title: { [lang]: title } }); return { content: [{ type: "text", text: JSON.stringify(res.data, null, 2) }] }; } case "update_hero_translation": { const { id, lang, tagline, title, description, buttonText } = params as any; const payload: any = {}; if (tagline) payload.tagline = { [lang]: tagline }; if (title) payload.title = { [lang]: title }; if (description) payload.description = { [lang]: description }; if (buttonText) payload.buttonText = { [lang]: buttonText }; const res = await api.patch(`/hero/${id}/translations`, payload); return { content: [{ type: "text", text: JSON.stringify(res.data, null, 2) }] }; } case "update_menu_item_translation": { const { id, lang, label } = params as any; const res = await api.patch(`/menus/items/${id}/translations`, { label: { [lang]: label } }); return { content: [{ type: "text", text: JSON.stringify(res.data, null, 2) }] }; } case "update_store_settings_translation": { const { lang, footerDescription, newsletterTitle, newsletterDescription, newsletterPlaceholder, newsletterButtonText } = params as any; const payload: any = {}; if (footerDescription) payload.footerDescription = { [lang]: footerDescription }; if (newsletterTitle) payload.newsletterTitle = { [lang]: newsletterTitle }; if (newsletterDescription) payload.newsletterDescription = { [lang]: newsletterDescription }; if (newsletterPlaceholder) payload.newsletterPlaceholder = { [lang]: newsletterPlaceholder }; if (newsletterButtonText) payload.newsletterButtonText = { [lang]: newsletterButtonText }; const res = await api.patch(`/settings/translations`, payload); return { content: [{ type: "text", text: JSON.stringify(res.data, null, 2) }] }; } case "update_article_translation": { const { id, lang, title, content } = params as any; const payload: any = {}; if (title) payload.title = { [lang]: title }; if (content) payload.content = { [lang]: content }; const res = await api.patch(`/articles/${id}/translations`, payload); return { content: [{ type: "text", text: JSON.stringify(res.data, null, 2) }] }; } case "update_product_content": { const payload: any = {}; if (params.name) payload.name = params.name; if (params.description) payload.description = params.description; if (params.details) payload.details = params.details; const res = await api.patch(`/products/${params.id}`, payload); return { content: [{ type: "text", text: JSON.stringify(res.data, null, 2) }] }; } case "update_product_translation": { const payload: any = {}; if (params.name) payload.name = { [params.lang]: params.name }; if (params.description) payload.description = { [params.lang]: params.description }; if (params.details) payload.details = { [params.lang]: params.details }; const res = await api.patch(`/products/${params.id}`, payload); return { content: [{ type: "text", text: JSON.stringify(res.data, null, 2) }] }; } case "upload_product_image_from_url": { // 1. Download image const imgRes = await axios.get(params.imageUrl, { responseType: 'arraybuffer' }); const buffer = Buffer.from(imgRes.data, 'binary'); const ext = params.imageUrl.split('.').pop()?.split('?')[0] || 'jpg'; // 2. Upload to backend const formData = new FormData(); formData.append('file', buffer, { filename: `download.${ext}` }); const uploadRes = await api.post('/upload', formData, { headers: formData.getHeaders() }); const uploadedUrl = uploadRes.data.url; // 3. Get current product to append image const prodRes = await api.get(`/products/${params.productId}`); const product = prodRes.data; const newImages = [...product.images, { url: uploadedUrl, order: product.images.length }]; // 4. Update product (using PUT to sync variants and images) // Ensure we pass required fields const putRes = await api.put(`/products/${params.productId}`, { categoryId: product.categoryId, description: product.description, details: product.details, variants: product.variants, images: newImages }); return { content: [{ type: "text", text: `Image uploaded and product updated. New image URL: ${uploadedUrl}` }] }; } case "generate_mock_review": { const res = await api.post("/reviews", { productId: params.productId, rating: params.rating, comment: params.comment, customerName: params.customerName }); return { content: [{ type: "text", text: JSON.stringify(res.data, null, 2) }] }; } case "get_order_status": { const res = await api.get(`/orders/${params.orderId}`); return { content: [{ type: "text", text: JSON.stringify(res.data, null, 2) }] }; } case "list_pending_orders": { const res = await api.get(`/orders`); const pending = res.data.filter((o: any) => o.status === "PENDING_PAYMENT" || o.status === "PENDING" || o.status === "PROCESSING"); if (params.limit) { pending.splice(params.limit); } return { content: [{ type: "text", text: JSON.stringify(pending, null, 2) }] }; } case "update_order_tracking": { const payload: any = { trackingNo: params.trackingNo }; const res1 = await api.patch(`/orders/${params.orderId}/tracking`, payload); let result = res1.data; if (params.status) { const res2 = await api.patch(`/orders/${params.orderId}/status`, { status: params.status }); result = res2.data; } return { content: [{ type: "text", text: JSON.stringify(result, null, 2) }] }; } case "search_customer": { const search = params.search ? `?search=${encodeURIComponent(params.search)}` : ""; const res = await api.get(`/customers${search}`); return { content: [{ type: "text", text: JSON.stringify(res.data, null, 2) }] }; } case "adjust_customer_points": { const payload = { pointsDelta: params.pointsDelta, reason: params.reason }; const res = await api.patch(`/customers/${params.id}/points`, payload); return { content: [{ type: "text", text: JSON.stringify(res.data, null, 2) }] }; } case "generate_special_coupon": { const payload = { code: params.code, type: params.type, value: params.value, minPurchase: params.minPurchase, maxDiscount: params.maxDiscount, usageLimit: params.usageLimit, expiresAt: params.expiresAt }; const res = await api.post(`/coupons`, payload); return { content: [{ type: "text", text: JSON.stringify(res.data, null, 2) }] }; } case "list_dynamic_pages": { const res = await api.get("/pages"); return { content: [{ type: "text", text: JSON.stringify(res.data, null, 2) }] }; } case "create_dynamic_page": { const payload: any = { title: params.title, slug: params.slug, }; if (params.isPublished !== undefined) payload.isPublished = params.isPublished; if (params.status !== undefined) payload.status = params.status; if (params.metadata !== undefined) payload.metadata = params.metadata; if (params.id) { payload.id = params.id; } const res = await api.post("/pages", payload); return { content: [{ type: "text", text: JSON.stringify(res.data, null, 2) }] }; } case "add_page_section": { const payload: any = { type: params.type, payload: params.payload, isActive: params.isActive !== false }; if (params.order !== undefined) payload.order = params.order; const res = await api.post(`/pages/${params.pageId}/sections`, payload); return { content: [{ type: "text", text: JSON.stringify(res.data, null, 2) }] }; } case "update_page_section": { const payload: any = {}; if (params.type) payload.type = params.type; if (params.order !== undefined) payload.order = params.order; if (params.payload) payload.payload = params.payload; if (params.isActive !== undefined) payload.isActive = params.isActive; const res = await api.put(`/pages/sections/${params.sectionId}`, payload); return { content: [{ type: "text", text: JSON.stringify(res.data, null, 2) }] }; } case "delete_page_section": { const res = await api.delete(`/pages/sections/${params.sectionId}`); return { content: [{ type: "text", text: JSON.stringify(res.data, null, 2) }] }; } case "update_footer_content": { const results = []; if (params.brandDescription !== undefined || params.socialLinks) { const settingsPayload: any = {}; if (params.brandDescription !== undefined) settingsPayload.footerDescription = params.brandDescription; if (params.socialLinks?.facebookUrl !== undefined) settingsPayload.facebookUrl = params.socialLinks.facebookUrl; if (params.socialLinks?.instagramUrl !== undefined) settingsPayload.instagramUrl = params.socialLinks.instagramUrl; if (params.socialLinks?.tiktokUrl !== undefined) settingsPayload.tiktokUrl = params.socialLinks.tiktokUrl; if (Object.keys(settingsPayload).length > 0) { // First fetch existing settings because the backend requires full payload validation const currentSettingsRes = await api.get("/settings"); const currentSettings = currentSettingsRes.data || {}; const mergedSettings = { ...currentSettings, ...settingsPayload }; // Remove null values so Elysia validation doesn't fail on t.Optional(t.String()) for (const key in mergedSettings) { if (mergedSettings[key] === null) { delete mergedSettings[key]; } } await api.put("/settings", mergedSettings); results.push({ type: "settings", status: "success" }); } } if (params.exploreLinks) { await api.put("/menus/footer-explore", { items: params.exploreLinks }); results.push({ type: "explore_links", status: "success" }); } if (params.collectionLinks) { await api.put("/menus/footer-collections", { items: params.collectionLinks }); results.push({ type: "collection_links", status: "success" }); } if (params.supportLinks) { await api.put("/menus/footer-support", { items: params.supportLinks }); results.push({ type: "support_links", status: "success" }); } return { content: [{ type: "text", text: JSON.stringify(results, null, 2) }] }; } case "audit_missing_pages": { // 1. Fetch all menus const menusRes = await api.get("/menus"); const menus = menusRes.data || []; // 2. Fetch all dynamic pages const pagesRes = await api.get("/pages"); const pages = pagesRes.data || []; const createdSlugs = pages.map((p: any) => `/${p.slug}`); // 3. Known SPA Hardcoded Routes const hardcodedRoutes = [ "/", "/shop", "/login", "/cart", "/checkout", "/account", "/search", "/contact" ]; const missingLinks: any[] = []; // 4. Audit menus.forEach((menu: any) => { if (menu.items && Array.isArray(menu.items)) { menu.items.forEach((item: any) => { const url = item.url; if (!url) return; // Skip external links if (url.startsWith("http") || url.startsWith("mailto:") || url.startsWith("tel:")) return; // Skip SPA Hardcoded routes if (hardcodedRoutes.includes(url)) return; // Skip dynamic collection routes natively handled by SPA (e.g. /collections/...) if (url.startsWith("/collections/")) return; // Check if it exists in created dynamic pages if (!createdSlugs.includes(url)) { missingLinks.push({ menuName: menu.name, label: item.label, url: url, suggestedSlug: url.startsWith("/") ? url.substring(1) : url }); } }); } }); return { content: [{ type: "text", text: JSON.stringify({ missingPages: missingLinks }, null, 2) }] }; } default: throw new Error(`Unknown tool: ${request.params.name}`); } } catch (err: any) { const errorMsg = err.response?.data ? JSON.stringify(err.response.data) : err.message; return { content: [{ type: "text", text: `Error executing tool: ${errorMsg}` }], isError: true, }; } }); async function main() { await performLogin(); const transport = new StdioServerTransport(); await server.connect(transport); console.error("HomAroma MCP Server running on stdio"); } main().catch(console.error);