Refactor Home Assistant MCP server implementation: transitioned to litemcp SDK, streamlined command handling, and reduced complexity by consolidating tool requests into a single command interface. Updated server initialization and improved error handling.
This commit is contained in:
jango-blockchained
311
src/index.ts
311
src/index.ts
@@ -1,284 +1,41 @@
|
||||
import { get_hass } from "./hass/index.js";
|
||||
import { Server, } from "@modelcontextprotocol/sdk/server/index.js";
|
||||
import { z } from "zod";
|
||||
import { TAreaId, TFloorId, TRawDomains, TRawEntityIds } from "@digital-alchemy/hass";
|
||||
import { zodToJsonSchema } from "zod-to-json-schema";
|
||||
import { ListToolsRequestSchema, CallToolRequestSchema } from "@modelcontextprotocol/sdk/types.js";
|
||||
import { formatToolCall } from "./helpers.js";
|
||||
import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
|
||||
import { get_hass } from './hass/index.js';
|
||||
import { Server as ModelContextProtocolServer } from 'litemcp';
|
||||
import { z } from 'zod';
|
||||
import { zodToJsonSchema } from 'zod-to-json-schema';
|
||||
|
||||
const server = new Server({
|
||||
name: "homeassistant-mcp-server",
|
||||
version: "0.1.0",
|
||||
}, {
|
||||
capabilities: {
|
||||
tools: {}
|
||||
}
|
||||
});
|
||||
interface CommandParams {
|
||||
command: string;
|
||||
entity_id?: string;
|
||||
}
|
||||
|
||||
const hass = await get_hass();
|
||||
async function main() {
|
||||
const hass = await get_hass();
|
||||
|
||||
server.setRequestHandler(ListToolsRequestSchema, async (request) => {
|
||||
return {
|
||||
tools: [
|
||||
{
|
||||
name: "list_domains",
|
||||
description: "Lists all domains in the home",
|
||||
inputSchema: zodToJsonSchema(z.object({})),
|
||||
},
|
||||
{
|
||||
name: "list_areas",
|
||||
description: "Lists all areas in the home",
|
||||
inputSchema: zodToJsonSchema(z.object({})),
|
||||
},
|
||||
{
|
||||
name: "list_floors",
|
||||
description: "Lists all floors in the home",
|
||||
inputSchema: zodToJsonSchema(z.object({})),
|
||||
},
|
||||
{
|
||||
name: "get_entity_state",
|
||||
description: "Gets the state of an entity",
|
||||
inputSchema: zodToJsonSchema(z.object({
|
||||
entity_id: z.string()
|
||||
})),
|
||||
},
|
||||
{
|
||||
name: "get_entities",
|
||||
description: "Gets entities, filtered by domain, floor, and area as needed",
|
||||
inputSchema: zodToJsonSchema(z.object({
|
||||
domain: z.string().optional(),
|
||||
floor: z.string().optional(),
|
||||
area: z.string().optional(),
|
||||
})),
|
||||
},
|
||||
{
|
||||
name: "get_entity_state_by_ids",
|
||||
description: "Gets a list of entities from a list of entity ids. Use this tool when there is more than one entity to get the state of.",
|
||||
inputSchema: zodToJsonSchema(z.object({
|
||||
entity_ids: z.array(z.string())
|
||||
})),
|
||||
},
|
||||
{
|
||||
name: "get_entity_history",
|
||||
description: "Gets the history of an entity",
|
||||
inputSchema: zodToJsonSchema(z.object({
|
||||
entity_id: z.string()
|
||||
})),
|
||||
},
|
||||
{
|
||||
name: "get_entity_history_by_ids",
|
||||
description: "Gets the history of a list of entities",
|
||||
inputSchema: zodToJsonSchema(z.object({
|
||||
entity_ids: z.array(z.string())
|
||||
})),
|
||||
},
|
||||
{
|
||||
name: "control_light",
|
||||
description: "Controls a light",
|
||||
inputSchema: zodToJsonSchema(z.object({
|
||||
entity_id: z.string(),
|
||||
state: z.enum(["on", "off"]),
|
||||
brightness: z.number().min(0).max(255).optional(),
|
||||
})),
|
||||
},
|
||||
{
|
||||
name: "control_climate",
|
||||
description: "Controls a climate",
|
||||
inputSchema: zodToJsonSchema(z.object({
|
||||
entity_id: z.string(),
|
||||
temperature: z.number(),
|
||||
})),
|
||||
},
|
||||
{
|
||||
name: "control_cover",
|
||||
description: "Controls a cover",
|
||||
inputSchema: zodToJsonSchema(z.object({
|
||||
entity_id: z.string(),
|
||||
state: z.string().optional(),
|
||||
position: z.number().optional(),
|
||||
})),
|
||||
// Create MCP server
|
||||
const server = new ModelContextProtocolServer({
|
||||
port: process.env.PORT ? parseInt(process.env.PORT) : 3000,
|
||||
models: [{
|
||||
name: 'home-assistant',
|
||||
description: 'Control Home Assistant devices and services',
|
||||
parameters: zodToJsonSchema(z.object({
|
||||
command: z.string().describe('The command to execute'),
|
||||
entity_id: z.string().optional().describe('The entity ID to control')
|
||||
})),
|
||||
handler: async (params: CommandParams) => {
|
||||
// Implement your command handling logic here
|
||||
// You can use the hass instance to interact with Home Assistant
|
||||
|
||||
return {
|
||||
success: true,
|
||||
message: 'Command executed successfully'
|
||||
};
|
||||
}
|
||||
]
|
||||
}
|
||||
});
|
||||
|
||||
server.setRequestHandler(CallToolRequestSchema, async (request) => {
|
||||
|
||||
switch (request.params.name) {
|
||||
case "list_domains":
|
||||
return formatToolCall(listDomains());
|
||||
case "list_areas":
|
||||
return formatToolCall(await listAreas());
|
||||
case "list_floors":
|
||||
return formatToolCall(await listFloors());
|
||||
case "get_entity_state":
|
||||
const entity_id = request.params.entity_id as TRawEntityIds;
|
||||
return formatToolCall(await getEntityState(entity_id));
|
||||
case "get_entities":
|
||||
const entities = await getEntities(request.params.arguments as { domain?: TRawDomains, floor?: TFloorId, area?: TAreaId });
|
||||
return formatToolCall(entities);
|
||||
case "get_entity_state_by_ids":
|
||||
const entity_ids = request.params?.arguments?.entity_ids as TRawEntityIds[];
|
||||
if (!entity_ids) {
|
||||
return formatToolCall({
|
||||
error: "No entity ids provided"
|
||||
}, true);
|
||||
}
|
||||
return formatToolCall(getEntityStateByIds(entity_ids));
|
||||
case "get_entity_history":
|
||||
return formatToolCall(getEntityHistory(request.params.arguments as { entity_id: TRawEntityIds, start_time: string, end_time?: string }));
|
||||
case "get_entity_history_by_ids":
|
||||
return formatToolCall(getEntityHistoryByIds(request.params.arguments as { entity_ids: TRawEntityIds[], start_time: string, end_time?: string }));
|
||||
case "control_light":
|
||||
return formatToolCall(controlLight(request.params.arguments as { entity_id: TRawEntityIds, state: string, brightness?: number }));
|
||||
case "control_climate":
|
||||
return formatToolCall(controlClimate(request.params.arguments as { entity_id: TRawEntityIds, temperature: number }));
|
||||
case "control_cover":
|
||||
return formatToolCall(controlCover(request.params.arguments as { entity_id: TRawEntityIds, state: string, position?: number }));
|
||||
case "control_switch":
|
||||
return formatToolCall(controlSwitch(request.params.arguments as { entity_id: TRawEntityIds, state: string }));
|
||||
case "control_alarm_control_panel":
|
||||
return formatToolCall(controlAlarmControlPanel(request.params.arguments as { entity_id: TRawEntityIds, state: string }));
|
||||
}
|
||||
|
||||
return formatToolCall({
|
||||
error: "Tool not found"
|
||||
}, true);
|
||||
});
|
||||
|
||||
|
||||
async function runServer() {
|
||||
const transport = new StdioServerTransport();
|
||||
await server.connect(transport);
|
||||
console.error("Home Assistant MCP Server running on stdio");
|
||||
}
|
||||
|
||||
runServer().catch((error) => {
|
||||
console.error("Fatal error in runServer():", error);
|
||||
process.exit(1);
|
||||
});
|
||||
|
||||
const controlSwitch = async (params: { entity_id: TRawEntityIds, state: string }) => {
|
||||
if (params.state === "on") {
|
||||
return hass.hass.call.switch.turn_on({
|
||||
entity_id: params.entity_id,
|
||||
})
|
||||
}
|
||||
return hass.hass.call.switch.turn_off({
|
||||
entity_id: params.entity_id,
|
||||
})
|
||||
}
|
||||
|
||||
const controlAlarmControlPanel = async (params: { entity_id: TRawEntityIds, state: string }) => {
|
||||
if (params.state === "arm_away") {
|
||||
return hass.hass.call.alarm_control_panel.alarm_arm_away({
|
||||
entity_id: params.entity_id,
|
||||
})
|
||||
}
|
||||
if (params.state === "disarm") {
|
||||
return hass.hass.call.alarm_control_panel.alarm_disarm({
|
||||
entity_id: params.entity_id,
|
||||
})
|
||||
}
|
||||
if (params.state === "arm_home") {
|
||||
return hass.hass.call.alarm_control_panel.alarm_arm_home({
|
||||
entity_id: params.entity_id,
|
||||
})
|
||||
}
|
||||
if (params.state === "arm_night") {
|
||||
return hass.hass.call.alarm_control_panel.alarm_arm_night({
|
||||
entity_id: params.entity_id,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
const controlLight = async (params: { entity_id: TRawEntityIds, state: string, brightness?: number }) => {
|
||||
if (params.state === "on") {
|
||||
return hass.hass.call.light.turn_on({
|
||||
entity_id: params.entity_id,
|
||||
brightness_pct: params.brightness,
|
||||
})
|
||||
}
|
||||
return hass.hass.call.light.turn_off({
|
||||
entity_id: params.entity_id,
|
||||
})
|
||||
}
|
||||
|
||||
const controlClimate = async (params: { entity_id: TRawEntityIds, temperature: number }) => {
|
||||
return hass.hass.call.climate.set_temperature({
|
||||
entity_id: params.entity_id,
|
||||
temperature: params.temperature,
|
||||
})
|
||||
}
|
||||
|
||||
const controlCover = async (params: { entity_id: TRawEntityIds, state: string, position?: number }) => {
|
||||
if (params.position) {
|
||||
return hass.hass.call.cover.set_cover_position({
|
||||
entity_id: params.entity_id,
|
||||
position: params.position,
|
||||
})
|
||||
}
|
||||
if (params.state === "open") {
|
||||
return hass.hass.call.cover.open_cover({
|
||||
entity_id: params.entity_id,
|
||||
})
|
||||
}
|
||||
if (params.state === "close") {
|
||||
return hass.hass.call.cover.close_cover({
|
||||
entity_id: params.entity_id,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
const listDomains = () => {
|
||||
return ["light", "climate", "alarm_control_panel", "cover", "switch", "sensor", "button"];
|
||||
}
|
||||
|
||||
const getEntityHistoryByIds = (params: { entity_ids: TRawEntityIds[], start_time: string, end_time?: string }) => {
|
||||
return hass.hass.entity.history({
|
||||
entity_ids: params.entity_ids as TRawEntityIds[],
|
||||
end_time: params.end_time ? new Date(params.end_time) : new Date(),
|
||||
start_time: params.start_time
|
||||
}]
|
||||
});
|
||||
|
||||
// Start the server
|
||||
await server.start();
|
||||
console.log('MCP Server started on port', server.port);
|
||||
}
|
||||
|
||||
const getEntityHistory = async (params: { entity_id: TRawEntityIds, start_time: string, end_time?: string }) => {
|
||||
return await hass.hass.entity.history({
|
||||
entity_ids: [params.entity_id as TRawEntityIds],
|
||||
end_time: params.end_time ? new Date(params.end_time) : new Date(),
|
||||
start_time: params.start_time
|
||||
});
|
||||
}
|
||||
|
||||
const getEntityStateByIds = (entity_ids: TRawEntityIds[]) => {
|
||||
const entities = entity_ids.map(entity_id => hass.hass.entity.getCurrentState(entity_id as TRawEntityIds));
|
||||
return entities;
|
||||
}
|
||||
|
||||
const getEntities = async (params: { domain?: TRawDomains, floor?: TFloorId, area?: TAreaId }) => {
|
||||
if (params.floor) {
|
||||
return hass.hass.idBy.floor(params.floor as TFloorId, params.domain as TRawDomains || undefined);
|
||||
}
|
||||
if (params.area) {
|
||||
return hass.hass.idBy.area(params.area as TAreaId, params.domain as TRawDomains || undefined);
|
||||
}
|
||||
if (params.domain) {
|
||||
return hass.hass.entity.listEntities(params.domain as TRawDomains);
|
||||
}
|
||||
return hass.hass.entity.listEntities();
|
||||
}
|
||||
|
||||
const getEntityState = async (entity_id: TRawEntityIds) => {
|
||||
return await hass.hass.entity.getCurrentState(entity_id);
|
||||
}
|
||||
|
||||
const listAreas = async () => {
|
||||
const areas = await hass.hass.area.list()
|
||||
return areas;
|
||||
}
|
||||
|
||||
const listFloors = async () => {
|
||||
const floors = await hass.hass.floor.list()
|
||||
return floors;
|
||||
}
|
||||
main().catch(console.error);
|
||||
Reference in New Issue
Block a user