package mujica

import (
	"fmt"
	"time"

	"github.com/egoist/mygo/ui"

	"mujicaui/icons"
	"mujicaui/theme"
)

// ThinkingBlockOptions configure a ThinkingBlock.
type ThinkingBlockOptions struct {
	// Thinking shows the spinner and a running timer from Started.
	Thinking bool
	Started  time.Time
	// Elapsed is the final thinking time, shown once Thinking is false.
	Elapsed time.Duration
}

// ThinkingBlock folds a model's reasoning under a header with its time; *open's first value sets the default.
func ThinkingBlock(c *ui.Context, open *bool, opts ThinkingBlockOptions, content func()) ChoiceResult {
	if opts.Elapsed < 0 || opts.Thinking && opts.Started.IsZero() {
		panic("mujica: ThinkingBlock needs Started while Thinking and a non-negative Elapsed")
	}
	k := Tokens(c)
	u := Density(c).Unit()
	secs := int(opts.Elapsed.Round(time.Second).Seconds())
	title := fmt.Sprintf(msg(c, "thinking_block.done"), secs)
	if opts.Thinking {
		run := c.Now().Sub(opts.Started)
		secs = int(max(run, 0) / time.Second)
		title = fmt.Sprintf(msg(c, "thinking_block.thinking"), secs)
		c.After(time.Second - run%time.Second)
	}
	return Collapsible(c, open, CollapsibleOptions{Title: title, Trigger: func() {
		ui.Row(c).Gap(u * 2).AlignItems(ui.Center).TextColor(k.TextMuted).Children(func() {
			if opts.Thinking {
				Spinner(c, SpinnerOptions{Size: 14, Label: title})
			} else {
				ui.Icon(c, icons.Must("brain")).Size(16, 16)
			}
			ui.Text(c, title).FontSize(FontSize(c, theme.CaptionSize)).FontFeatures("tnum").SingleLine()
		})
	}}, func() {
		ui.Column(c).Gap(u*2).Padding(0, 0, 0, u*3).BorderWidth(0, 0, 0, theme.BorderWidth).BorderColor(k.Border).
			TextColor(k.TextMuted).Children(content)
	})
}