1// SPDX-FileCopyrightText: Amolith <amolith@secluded.site>
 2//
 3// SPDX-License-Identifier: AGPL-3.0-or-later
 4
 5package tools
 6
 7import (
 8	"context"
 9	"fmt"
10	"strings"
11
12	"git.sr.ht/~amolith/lunatask-mcp-server/lunatask"
13	"github.com/mark3labs/mcp-go/mcp"
14)
15
16// HandleListHabitsAndActivities handles the list_habits_and_activities tool call.
17func (h *Handlers) HandleListHabitsAndActivities(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
18	var b strings.Builder
19	for _, habit := range h.config.Habits {
20		fmt.Fprintf(&b, "- %s: %s\n", habit.GetName(), habit.GetID())
21	}
22	return &mcp.CallToolResult{
23		Content: []mcp.Content{
24			mcp.TextContent{
25				Type: "text",
26				Text: b.String(),
27			},
28		},
29	}, nil
30}
31
32// HandleTrackHabitActivity handles the track_habit_activity tool call.
33func (h *Handlers) HandleTrackHabitActivity(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
34	habitID, ok := request.Params.Arguments["habit_id"].(string)
35	if !ok || habitID == "" {
36		return reportMCPError("Missing or invalid required argument: habit_id")
37	}
38
39	performedOn, ok := request.Params.Arguments["performed_on"].(string)
40	if !ok || performedOn == "" {
41		return reportMCPError("Missing or invalid required argument: performed_on")
42	}
43
44	client := lunatask.NewClient(h.config.AccessToken)
45	habitRequest := &lunatask.TrackHabitActivityRequest{
46		PerformedOn: performedOn,
47	}
48
49	resp, err := client.TrackHabitActivity(ctx, habitID, habitRequest)
50	if err != nil {
51		return reportMCPError(fmt.Sprintf("Failed to track habit activity: %v", err))
52	}
53
54	return &mcp.CallToolResult{
55		Content: []mcp.Content{
56			mcp.TextContent{
57				Type: "text",
58				Text: fmt.Sprintf("Habit activity tracked successfully. Status: %s, Message: %s", resp.Status, resp.Message),
59			},
60		},
61	}, nil
62}