Edgee MCP Server

import { z } from 'zod'; import * as api from '../api/index.js'; import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js'; /** * Register component-related tools with the MCP server */ export function registerComponentTools(server: McpServer): void { // List Public Components server.tool( 'edgee-listPublicComponents', 'List all public components.', { category: z.enum(['data_collection']).optional(), subcategory: z.enum(['analytics', 'warehouse', 'attribution', 'conversion api']).optional(), }, async ({ category, subcategory }) => { try { const data = await api.listPublicComponents(category, subcategory); if (!data || !data.data) { return { content: [ { type: 'text', text: 'Failed to retrieve public components.', }, ], }; } const components = data.data; // Format components const formattedComponents = components.map((component) => [ `${component.name || 'Unknown'}:`, `ID: ${component.id}`, `Slug: ${component.slug || 'Unknown'}`, `Category: ${component.category || 'Unknown'}`, `Subcategory: ${component.subcategory || 'Unknown'}`, `Description: ${component.description || 'None'}`, `Latest Version: ${component.latest_version || 'None'}`, `Public: ${component.is_public ? 'Yes' : 'No'}`, `Archived: ${component.is_archived ? 'Yes' : 'No'}`, `Created at: ${component.created_at || 'Unknown'}`, `Updated at: ${component.updated_at || 'Unknown'}`, '---', ].join('\n') ); const componentsText = `Public Components:\n\n${formattedComponents.join('\n')}`; return { content: [ { type: 'text', text: componentsText, }, ], }; } catch (error) { console.error('Error in edgee-listPublicComponents:', error); return { content: [ { type: 'text', text: `Error retrieving public components: ${error instanceof Error ? error.message : String(error)}`, }, ], isError: true, }; } } ); // List Organization Components server.tool( 'edgee-listOrganizationComponents', 'List all components for an organization.', { id: z.string(), category: z.enum(['data_collection']).optional(), subcategory: z.enum(['analytics', 'warehouse', 'attribution', 'conversion api']).optional(), }, async ({ id, category, subcategory }) => { try { const data = await api.listOrganizationComponents(id, category, subcategory); if (!data || !data.data) { return { content: [ { type: 'text', text: `Failed to retrieve components for organization with ID: ${id}`, }, ], }; } const components = data.data; // Format components const formattedComponents = components.map((component) => [ `${component.name || 'Unknown'}:`, `ID: ${component.id}`, `Slug: ${component.slug || 'Unknown'}`, `Category: ${component.category || 'Unknown'}`, `Subcategory: ${component.subcategory || 'Unknown'}`, `Description: ${component.description || 'None'}`, `Latest Version: ${component.latest_version || 'None'}`, `Public: ${component.is_public ? 'Yes' : 'No'}`, `Archived: ${component.is_archived ? 'Yes' : 'No'}`, `Created at: ${component.created_at || 'Unknown'}`, `Updated at: ${component.updated_at || 'Unknown'}`, '---', ].join('\n') ); const componentsText = `Components for organization ${id}:\n\n${formattedComponents.join('\n')}`; return { content: [ { type: 'text', text: componentsText, }, ], }; } catch (error) { console.error('Error in edgee-listOrganizationComponents:', error); return { content: [ { type: 'text', text: `Error retrieving organization components: ${error instanceof Error ? error.message : String(error)}`, }, ], isError: true, }; } } ); // Get Component by UUID server.tool( 'edgee-getComponentByUuid', 'Get a component by UUID.', { id: z.string(), }, async ({ id }) => { try { const component = await api.getComponentByUuid(id); if (!component) { return { content: [ { type: 'text', text: `Failed to retrieve component with ID: ${id}`, }, ], }; } // Format component versions let versionsText = ''; if (component.versions && Object.keys(component.versions).length > 0) { const formattedVersions = Object.entries(component.versions).map(([version, versionData]) => [ `Version: ${version}`, `WIT World Version: ${versionData.wit_world_version || 'Unknown'}`, `WASM URL: ${versionData.wasm_url || 'Unknown'}`, `Created at: ${versionData.created_at || 'Unknown'}`, `Changelog: ${versionData.changelog || 'None'}`, `Dynamic Fields: ${versionData.dynamic_fields ? JSON.stringify(versionData.dynamic_fields, null, 2) : 'None'}`, ].join('\n ') ); versionsText = `\nVersions:\n ${formattedVersions.join('\n ---\n ')}`; } const componentText = [ `Component: ${component.name || 'Unknown'}`, `ID: ${component.id}`, `Slug: ${component.slug || 'Unknown'}`, `Category: ${component.category || 'Unknown'}`, `Subcategory: ${component.subcategory || 'Unknown'}`, `Description: ${component.description || 'None'}`, `Latest Version: ${component.latest_version || 'None'}`, `Repository Link: ${component.repo_link || 'None'}`, `Documentation Link: ${component.documentation_link || 'None'}`, `Public: ${component.is_public ? 'Yes' : 'No'}`, `Archived: ${component.is_archived ? 'Yes' : 'No'}`, `Created at: ${component.created_at || 'Unknown'}`, `Updated at: ${component.updated_at || 'Unknown'}`, versionsText, ].join('\n'); return { content: [ { type: 'text', text: componentText, }, ], }; } catch (error) { console.error('Error in edgee-getComponentByUuid:', error); return { content: [ { type: 'text', text: `Error retrieving component: ${error instanceof Error ? error.message : String(error)}`, }, ], isError: true, }; } } ); // Get Component by Slug server.tool( 'edgee-getComponentBySlug', 'Get a component by slug.', { orgSlug: z.string(), componentSlug: z.string(), }, async ({ orgSlug, componentSlug }) => { try { const component = await api.getComponentBySlug(orgSlug, componentSlug); if (!component) { return { content: [ { type: 'text', text: `Failed to retrieve component with slug: ${orgSlug}/${componentSlug}`, }, ], }; } // Format component versions let versionsText = ''; if (component.versions && Object.keys(component.versions).length > 0) { const formattedVersions = Object.entries(component.versions).map(([version, versionData]) => [ `Version: ${version}`, `WIT World Version: ${versionData.wit_world_version || 'Unknown'}`, `WASM URL: ${versionData.wasm_url || 'Unknown'}`, `Created at: ${versionData.created_at || 'Unknown'}`, `Changelog: ${versionData.changelog || 'None'}`, `Dynamic Fields: ${versionData.dynamic_fields ? JSON.stringify(versionData.dynamic_fields, null, 2) : 'None'}`, ].join('\n ') ); versionsText = `\nVersions:\n ${formattedVersions.join('\n ---\n ')}`; } const componentText = [ `Component: ${component.name || 'Unknown'}`, `ID: ${component.id}`, `Slug: ${component.slug || 'Unknown'}`, `Category: ${component.category || 'Unknown'}`, `Subcategory: ${component.subcategory || 'Unknown'}`, `Description: ${component.description || 'None'}`, `Latest Version: ${component.latest_version || 'None'}`, `Repository Link: ${component.repo_link || 'None'}`, `Documentation Link: ${component.documentation_link || 'None'}`, `Public: ${component.is_public ? 'Yes' : 'No'}`, `Archived: ${component.is_archived ? 'Yes' : 'No'}`, `Created at: ${component.created_at || 'Unknown'}`, `Updated at: ${component.updated_at || 'Unknown'}`, versionsText, ].join('\n'); return { content: [ { type: 'text', text: componentText, }, ], }; } catch (error) { console.error('Error in edgee-getComponentBySlug:', error); return { content: [ { type: 'text', text: `Error retrieving component: ${error instanceof Error ? error.message : String(error)}`, }, ], isError: true, }; } } ); // Create Component server.tool( 'edgee-createComponent', 'Create a new component.', { organization_id: z.string(), name: z.string(), slug: z.string(), category: z.enum(['data_collection']), subcategory: z.enum(['analytics', 'warehouse', 'attribution', 'conversion api']), description: z.string().optional(), documentation_link: z.string().optional(), repo_link: z.string().optional(), avatar_url: z.string().optional(), public: z.boolean().optional(), }, async ({ organization_id, name, slug, category, subcategory, description, documentation_link, repo_link, avatar_url, public: isPublic, }) => { try { const component = await api.createComponent({ organization_id, name, slug, category, subcategory, description, documentation_link, repo_link, avatar_url, public: isPublic, }); if (!component) { return { content: [ { type: 'text', text: 'Failed to create component.', }, ], }; } const componentText = [ 'Component created successfully:', `Name: ${component.name || 'Unknown'}`, `ID: ${component.id}`, `Slug: ${component.slug || 'Unknown'}`, `Category: ${component.category || 'Unknown'}`, `Subcategory: ${component.subcategory || 'Unknown'}`, `Description: ${component.description || 'None'}`, `Repository Link: ${component.repo_link || 'None'}`, `Documentation Link: ${component.documentation_link || 'None'}`, `Public: ${component.is_public ? 'Yes' : 'No'}`, `Created at: ${component.created_at || 'Unknown'}`, `Updated at: ${component.updated_at || 'Unknown'}`, ].join('\n'); return { content: [ { type: 'text', text: componentText, }, ], }; } catch (error) { console.error('Error in edgee-createComponent:', error); return { content: [ { type: 'text', text: `Error creating component: ${error instanceof Error ? error.message : String(error)}`, }, ], isError: true, }; } } ); // Create Component Version server.tool( 'edgee-createComponentVersion', 'Create a new component version.', { id: z.string(), version: z.string(), wit_version: z.string(), wasm_url: z.string(), changelog: z.string().optional(), dynamic_fields: z .array( z.object({ name: z.string(), title: z.string(), type: z.enum(['string', 'bool', 'number']), required: z.boolean(), description: z.string().optional(), }) ) .optional(), }, async ({ id, version, wit_version, wasm_url, changelog, dynamic_fields }) => { try { const componentVersion = await api.createComponentVersionByUuid(id, { version, wit_version, wasm_url, changelog, dynamic_fields, }); if (!componentVersion) { return { content: [ { type: 'text', text: `Failed to create version for component with ID: ${id}`, }, ], }; } const versionText = [ 'Component version created successfully:', `Version: ${componentVersion.version}`, `WIT World Version: ${componentVersion.wit_world_version || 'Unknown'}`, `WASM URL: ${componentVersion.wasm_url || 'Unknown'}`, `Created at: ${componentVersion.created_at || 'Unknown'}`, `Changelog: ${componentVersion.changelog || 'None'}`, `Dynamic Fields: ${componentVersion.dynamic_fields ? JSON.stringify(componentVersion.dynamic_fields, null, 2) : 'None'}`, ].join('\n'); return { content: [ { type: 'text', text: versionText, }, ], }; } catch (error) { console.error('Error in edgee-createComponentVersion:', error); return { content: [ { type: 'text', text: `Error creating component version: ${error instanceof Error ? error.message : String(error)}`, }, ], isError: true, }; } } ); }