package agent

import (
	"fmt"
	"time"

	"github.com/egoist/mygo/ui"

	"git.zacharyzhang.com/ZacharyZhang-NY/MujicaUI/core"
	"git.zacharyzhang.com/ZacharyZhang-NY/MujicaUI/data"
	"git.zacharyzhang.com/ZacharyZhang-NY/MujicaUI/theme"
)

// ToolCallGroupOptions configure a ToolCallGroup.
type ToolCallGroupOptions struct {
	// Title replaces the built-in "N tool calls".
	Title string
	Still bool
}

// ToolCallGroup folds tool calls under a count, time, failures summary.
func ToolCallGroup(c *ui.Context, open *bool, calls []ToolCall, opts ToolCallGroupOptions) core.ChoiceResult {
	if open == nil {
		panic("mujica: ToolCallGroup needs open")
	}
	seen := make(map[string]bool, len(calls))
	var total time.Duration
	failed, running := 0, false
	for _, call := range calls {
		if call.ID == "" || seen[call.ID] {
			panic(fmt.Sprintf("mujica: ToolCallGroup call IDs must be unique and set, got %q", call.ID))
		}
		seen[call.ID] = true
		total += call.Duration
		failed += agentB2I(call.State == AgentFailed)
		running = running || call.State == AgentRunning
	}
	k := core.Tokens(c)
	u := core.Density(c).Unit()
	title := opts.Title
	if title == "" {
		title = fmt.Sprintf(core.Msg(c, "tool_call_group.count"), len(calls))
	}
	state := AgentDone
	switch {
	case running:
		state = AgentRunning
	case failed > 0:
		state = AgentFailed
	}
	summary := title + " · " + AgentDuration(total)
	if failed > 0 {
		summary += " · " + fmt.Sprintf(core.Msg(c, "tool_call_group.failed"), failed)
	}
	return data.Collapsible(c, open, data.CollapsibleOptions{Title: summary, Trigger: func() {
		ui.Row(c).Gap(u * 2).AlignItems(ui.Center).Children(func() {
			agentMark(c, state, 16, opts.Still)
			ui.Text(c, title).FontWeight(600).SingleLine()
			AgentCaption(c, AgentDuration(total))
			if failed > 0 {
				ui.Text(c, fmt.Sprintf(core.Msg(c, "tool_call_group.failed"), failed)).TextColor(k.Danger).
					FontSize(core.FontSize(c, theme.CaptionSize+1)).SingleLine()
			}
		})
	}}, func() {
		for _, call := range calls {
			ui.Column(c.Key(call.ID)).Children(func() { ToolCallCard(c, call, ToolCallCardOptions{Still: opts.Still}) })
		}
	})
}

func agentB2I(b bool) int {
	if b {
		return 1
	}
	return 0
}