@coffeerunhobby/mcp-ha-connect 0.5.0

package/src/tools/getState.ts

@@ -0,0 +1,103 @@
+/**
+ * Get State Tool
+ * Gets the state of a specific entity
+ */
+
+import { Server } from '@modelcontextprotocol/sdk/server/index.js';
+import {
+  CallToolRequestSchema,
+  ListToolsRequestSchema,
+} from '@modelcontextprotocol/sdk/types.js';
+import type { HaClient } from '../haClient/index.js';
+import { logger } from '../utils/logger.js';
+
+export function registerGetStateTool(server: Server, client: HaClient): void {
+  logger.debug('Registering getState tool');
+
+  server.setRequestHandler(ListToolsRequestSchema, async () => {
+    return {
+      tools: [
+        {
+          name: 'getState',
+          description: 'Get the state of a specific entity by entity_id.',
+          inputSchema: {
+            type: 'object',
+            properties: {
+              entity_id: {
+                type: 'string',
+                description: 'The entity ID (e.g., "light.living_room", "sensor.temperature")',
+              },
+            },
+            required: ['entity_id'],
+          },
+        },
+      ],
+    };
+  });
+
+  server.setRequestHandler(CallToolRequestSchema, async (request) => {
+    if (request.params.name !== 'getState') {
+      return {
+        content: [
+          {
+            type: 'text',
+            text: JSON.stringify({ error: 'Unknown tool' }),
+          },
+        ],
+        isError: true,
+      };
+    }
+
+    try {
+      const { entity_id } = request.params.arguments as { entity_id: string };
+
+      if (!entity_id) {
+        throw new Error('entity_id is required');
+      }
+
+      logger.info('Executing getState tool', { entity_id });
+      const state = await client.getState(entity_id);
+
+      if (!state) {
+        return {
+          content: [
+            {
+              type: 'text',
+              text: JSON.stringify({
+                error: 'Entity not found',
+                entity_id,
+              }),
+            },
+          ],
+          isError: true,
+        };
+      }
+
+      return {
+        content: [
+          {
+            type: 'text',
+            text: JSON.stringify(state, null, 2),
+          },
+        ],
+      };
+    } catch (error) {
+      logger.error('Failed to get state', {
+        error: error instanceof Error ? error.message : String(error),
+      });
+
+      return {
+        content: [
+          {
+            type: 'text',
+            text: JSON.stringify({
+              error: 'Failed to get state',
+              message: error instanceof Error ? error.message : String(error),
+            }),
+          },
+        ],
+        isError: true,
+      };
+    }
+  });
+}

package/src/tools/getStates.ts

@@ -0,0 +1,80 @@
+/**
+ * Get States Tool
+ * Gets all entity states from Home Assistant
+ */
+
+import { Server } from '@modelcontextprotocol/sdk/server/index.js';
+import {
+  CallToolRequestSchema,
+  ListToolsRequestSchema,
+} from '@modelcontextprotocol/sdk/types.js';
+import type { HaClient } from '../haClient/index.js';
+import { logger } from '../utils/logger.js';
+
+export function registerGetStatesTool(server: Server, client: HaClient): void {
+  logger.debug('Registering getStates tool');
+
+  server.setRequestHandler(ListToolsRequestSchema, async () => {
+    return {
+      tools: [
+        {
+          name: 'getStates',
+          description: 'Get all entity states from Home Assistant.',
+          inputSchema: {
+            type: 'object',
+            properties: {},
+            required: [],
+          },
+        },
+      ],
+    };
+  });
+
+  server.setRequestHandler(CallToolRequestSchema, async (request) => {
+    if (request.params.name !== 'getStates') {
+      return {
+        content: [
+          {
+            type: 'text',
+            text: JSON.stringify({ error: 'Unknown tool' }),
+          },
+        ],
+        isError: true,
+      };
+    }
+
+    try {
+      logger.info('Executing getStates tool');
+      const states = await client.getStates();
+
+      return {
+        content: [
+          {
+            type: 'text',
+            text: JSON.stringify({
+              count: states.length,
+              entities: states,
+            }, null, 2),
+          },
+        ],
+      };
+    } catch (error) {
+      logger.error('Failed to get states', {
+        error: error instanceof Error ? error.message : String(error),
+      });
+
+      return {
+        content: [
+          {
+            type: 'text',
+            text: JSON.stringify({
+              error: 'Failed to get states',
+              message: error instanceof Error ? error.message : String(error),
+            }),
+          },
+        ],
+        isError: true,
+      };
+    }
+  });
+}

package/src/tools/index.ts

@@ -0,0 +1,9 @@
+/**
+ * Tools index - exports all tool registration functions
+ */
+
+export { registerGetStatesTool } from './getStates.js';
+export { registerGetStateTool } from './getState.js';
+export { registerCallServiceTool } from './callService.js';
+export { registerGetEntitiesByDomainTool } from './getEntitiesByDomain.js';
+export { registerSearchEntitiesTool } from './searchEntities.js';