amcs/internal/mcpserver/server.go

package mcpserver

import (
	"net/http"
	"time"

	"github.com/modelcontextprotocol/go-sdk/mcp"

	"git.warky.dev/wdevs/amcs/internal/config"
	"git.warky.dev/wdevs/amcs/internal/tools"
)

type ToolSet struct {
	Capture   *tools.CaptureTool
	Search    *tools.SearchTool
	List      *tools.ListTool
	Stats     *tools.StatsTool
	Get       *tools.GetTool
	Update    *tools.UpdateTool
	Delete    *tools.DeleteTool
	Archive   *tools.ArchiveTool
	Projects  *tools.ProjectsTool
	Context   *tools.ContextTool
	Recall    *tools.RecallTool
	Summarize *tools.SummarizeTool
	Links     *tools.LinksTool
	Backfill    *tools.BackfillTool
	Reparse     *tools.ReparseMetadataTool
	Household   *tools.HouseholdTool
	Maintenance *tools.MaintenanceTool
	Calendar    *tools.CalendarTool
	Meals       *tools.MealsTool
	CRM         *tools.CRMTool
}

func New(cfg config.MCPConfig, toolSet ToolSet) http.Handler {
	server := mcp.NewServer(&mcp.Implementation{
		Name:    cfg.ServerName,
		Version: cfg.Version,
	}, nil)

	addTool(server, &mcp.Tool{
		Name:        "capture_thought",
		Description: "Store a thought with generated embeddings and extracted metadata.",
	}, toolSet.Capture.Handle)

	addTool(server, &mcp.Tool{
		Name:        "search_thoughts",
		Description: "Search stored thoughts by semantic similarity.",
	}, toolSet.Search.Handle)

	addTool(server, &mcp.Tool{
		Name:        "list_thoughts",
		Description: "List recent thoughts with optional metadata filters.",
	}, toolSet.List.Handle)

	addTool(server, &mcp.Tool{
		Name:        "thought_stats",
		Description: "Get counts and top metadata buckets across stored thoughts.",
	}, toolSet.Stats.Handle)

	addTool(server, &mcp.Tool{
		Name:        "get_thought",
		Description: "Retrieve a full thought by id.",
	}, toolSet.Get.Handle)

	addTool(server, &mcp.Tool{
		Name:        "update_thought",
		Description: "Update thought content or merge metadata.",
	}, toolSet.Update.Handle)

	addTool(server, &mcp.Tool{
		Name:        "delete_thought",
		Description: "Hard-delete a thought by id.",
	}, toolSet.Delete.Handle)

	addTool(server, &mcp.Tool{
		Name:        "archive_thought",
		Description: "Archive a thought so it is hidden from default search and listing.",
	}, toolSet.Archive.Handle)

	addTool(server, &mcp.Tool{
		Name:        "create_project",
		Description: "Create a named project container for thoughts.",
	}, toolSet.Projects.Create)

	addTool(server, &mcp.Tool{
		Name:        "list_projects",
		Description: "List projects and their current thought counts.",
	}, toolSet.Projects.List)

	addTool(server, &mcp.Tool{
		Name:        "set_active_project",
		Description: "Set the active project for the current MCP session.",
	}, toolSet.Projects.SetActive)

	addTool(server, &mcp.Tool{
		Name:        "get_active_project",
		Description: "Return the active project for the current MCP session.",
	}, toolSet.Projects.GetActive)

	addTool(server, &mcp.Tool{
		Name:        "get_project_context",
		Description: "Get recent and semantic context for a project.",
	}, toolSet.Context.Handle)

	addTool(server, &mcp.Tool{
		Name:        "recall_context",
		Description: "Recall semantically relevant and recent context.",
	}, toolSet.Recall.Handle)

	addTool(server, &mcp.Tool{
		Name:        "summarize_thoughts",
		Description: "Summarize a filtered or searched set of thoughts.",
	}, toolSet.Summarize.Handle)

	addTool(server, &mcp.Tool{
		Name:        "link_thoughts",
		Description: "Create a typed relationship between two thoughts.",
	}, toolSet.Links.Link)

	addTool(server, &mcp.Tool{
		Name:        "related_thoughts",
		Description: "Retrieve explicit links and semantic neighbors for a thought.",
	}, toolSet.Links.Related)

	addTool(server, &mcp.Tool{
		Name:        "backfill_embeddings",
		Description: "Generate missing embeddings for stored thoughts using the active embedding model.",
	}, toolSet.Backfill.Handle)

	addTool(server, &mcp.Tool{
		Name:        "reparse_thought_metadata",
		Description: "Re-extract and normalize metadata for stored thoughts from their content.",
	}, toolSet.Reparse.Handle)

	// Household Knowledge
	addTool(server, &mcp.Tool{
		Name:        "add_household_item",
		Description: "Store a household fact (paint color, appliance details, measurement, document, etc.).",
	}, toolSet.Household.AddItem)

	addTool(server, &mcp.Tool{
		Name:        "search_household_items",
		Description: "Search household items by name, category, or location.",
	}, toolSet.Household.SearchItems)

	addTool(server, &mcp.Tool{
		Name:        "get_household_item",
		Description: "Retrieve a household item by id.",
	}, toolSet.Household.GetItem)

	addTool(server, &mcp.Tool{
		Name:        "add_vendor",
		Description: "Add a service provider (plumber, electrician, landscaper, etc.).",
	}, toolSet.Household.AddVendor)

	addTool(server, &mcp.Tool{
		Name:        "list_vendors",
		Description: "List household service vendors, optionally filtered by service type.",
	}, toolSet.Household.ListVendors)

	// Home Maintenance
	addTool(server, &mcp.Tool{
		Name:        "add_maintenance_task",
		Description: "Create a recurring or one-time home maintenance task.",
	}, toolSet.Maintenance.AddTask)

	addTool(server, &mcp.Tool{
		Name:        "log_maintenance",
		Description: "Log completed maintenance work; automatically updates the task's next due date.",
	}, toolSet.Maintenance.LogWork)

	addTool(server, &mcp.Tool{
		Name:        "get_upcoming_maintenance",
		Description: "List maintenance tasks due within the next N days.",
	}, toolSet.Maintenance.GetUpcoming)

	addTool(server, &mcp.Tool{
		Name:        "search_maintenance_history",
		Description: "Search the maintenance log by task name, category, or date range.",
	}, toolSet.Maintenance.SearchHistory)

	// Family Calendar
	addTool(server, &mcp.Tool{
		Name:        "add_family_member",
		Description: "Add a family member to the household.",
	}, toolSet.Calendar.AddMember)

	addTool(server, &mcp.Tool{
		Name:        "list_family_members",
		Description: "List all family members.",
	}, toolSet.Calendar.ListMembers)

	addTool(server, &mcp.Tool{
		Name:        "add_activity",
		Description: "Schedule a one-time or recurring family activity.",
	}, toolSet.Calendar.AddActivity)

	addTool(server, &mcp.Tool{
		Name:        "get_week_schedule",
		Description: "Get all activities scheduled for a given week.",
	}, toolSet.Calendar.GetWeekSchedule)

	addTool(server, &mcp.Tool{
		Name:        "search_activities",
		Description: "Search activities by title, type, or family member.",
	}, toolSet.Calendar.SearchActivities)

	addTool(server, &mcp.Tool{
		Name:        "add_important_date",
		Description: "Track a birthday, anniversary, deadline, or other important date.",
	}, toolSet.Calendar.AddImportantDate)

	addTool(server, &mcp.Tool{
		Name:        "get_upcoming_dates",
		Description: "Get important dates coming up in the next N days.",
	}, toolSet.Calendar.GetUpcomingDates)

	// Meal Planning
	addTool(server, &mcp.Tool{
		Name:        "add_recipe",
		Description: "Save a recipe with ingredients and instructions.",
	}, toolSet.Meals.AddRecipe)

	addTool(server, &mcp.Tool{
		Name:        "search_recipes",
		Description: "Search recipes by name, cuisine, tags, or ingredient.",
	}, toolSet.Meals.SearchRecipes)

	addTool(server, &mcp.Tool{
		Name:        "update_recipe",
		Description: "Update an existing recipe.",
	}, toolSet.Meals.UpdateRecipe)

	addTool(server, &mcp.Tool{
		Name:        "create_meal_plan",
		Description: "Set the meal plan for a week; replaces any existing plan for that week.",
	}, toolSet.Meals.CreateMealPlan)

	addTool(server, &mcp.Tool{
		Name:        "get_meal_plan",
		Description: "Get the meal plan for a given week.",
	}, toolSet.Meals.GetMealPlan)

	addTool(server, &mcp.Tool{
		Name:        "generate_shopping_list",
		Description: "Auto-generate a shopping list from the meal plan for a given week.",
	}, toolSet.Meals.GenerateShoppingList)

	// Professional CRM
	addTool(server, &mcp.Tool{
		Name:        "add_professional_contact",
		Description: "Add a professional contact to the CRM.",
	}, toolSet.CRM.AddContact)

	addTool(server, &mcp.Tool{
		Name:        "search_contacts",
		Description: "Search professional contacts by name, company, title, notes, or tags.",
	}, toolSet.CRM.SearchContacts)

	addTool(server, &mcp.Tool{
		Name:        "log_interaction",
		Description: "Log an interaction with a professional contact.",
	}, toolSet.CRM.LogInteraction)

	addTool(server, &mcp.Tool{
		Name:        "get_contact_history",
		Description: "Get full history (interactions and opportunities) for a contact.",
	}, toolSet.CRM.GetHistory)

	addTool(server, &mcp.Tool{
		Name:        "create_opportunity",
		Description: "Create a deal, project, or opportunity linked to a contact.",
	}, toolSet.CRM.CreateOpportunity)

	addTool(server, &mcp.Tool{
		Name:        "get_follow_ups_due",
		Description: "List contacts with a follow-up date due within the next N days.",
	}, toolSet.CRM.GetFollowUpsDue)

	addTool(server, &mcp.Tool{
		Name:        "link_thought_to_contact",
		Description: "Append a stored thought to a contact's notes.",
	}, toolSet.CRM.LinkThought)

	return mcp.NewStreamableHTTPHandler(func(*http.Request) *mcp.Server {
		return server
	}, &mcp.StreamableHTTPOptions{
		JSONResponse:   true,
		SessionTimeout: 10 * time.Minute,
	})
}