package mujica

import (
	"fmt"

	"github.com/egoist/mygo/ui"

	"mujicaui/icons"
	"mujicaui/theme"
)

// SearchStage is one step of a web search, as "Planning" or "Reading".
type SearchStage struct {
	Label string
	Done  bool
}

// SearchProgressOptions describe the caller's search.
type SearchProgressOptions struct {
	// Query is the keyword searched; required.
	Query string
	// Read and Total count the sources read so far and found.
	Read, Total int
	Stages      []SearchStage
	// Done folds the block to a summary, which the user can open again.
	Done bool
}

// searchProgressState is the fold, open while searching until the search ends.
type searchProgressState struct {
	open, done bool
}

// SearchProgress shows a search's keyword, sources read and stages; once Done it folds to one line.
func SearchProgress(c *ui.Context, opts SearchProgressOptions) ChoiceResult {
	if opts.Query == "" || opts.Read < 0 || opts.Total < 0 || opts.Read > opts.Total {
		panic("mujica: SearchProgress needs a Query and 0 <= Read <= Total")
	}
	r := ChoiceResult{Element: ui.Column(c).MinWidth(0)}
	st := ui.Local(r.Element, "search", func() searchProgressState { return searchProgressState{open: true} })
	if opts.Done && !st.done {
		// Finishing folds it once; the user may open it again.
		st.open = false
	}
	st.done = opts.Done
	title := fmt.Sprintf(msg(c, "search_progress.searching"), opts.Query)
	if opts.Done {
		title = fmt.Sprintf(msg(c, "search_progress.done"), opts.Query, opts.Read)
	}
	r.Element.Children(func() { r.changed = searchProgressBody(c, st, title, opts) })
	return r
}

// searchProgressBody is the collapsible header and the stages.
func searchProgressBody(c *ui.Context, st *searchProgressState, title string, opts SearchProgressOptions) bool {
	k := Tokens(c)
	u := Density(c).Unit()
	small := FontSize(c, theme.CaptionSize)
	return Collapsible(c, &st.open, CollapsibleOptions{Title: title, Trigger: func() {
		ui.Row(c).Gap(u * 2).AlignItems(ui.Center).TextColor(k.TextMuted).Children(func() {
			if opts.Done {
				ui.Icon(c, icons.Must("search")).Size(14, 14)
			} else {
				Spinner(c, SpinnerOptions{Size: 14, Label: title})
			}
			ui.Text(c, title).FontSize(small).SingleLine()
		})
	}}, func() {
		ui.Text(c, fmt.Sprintf(msg(c, "search_progress.read"), opts.Read, opts.Total)).FontSize(small).FontFeatures("tnum").TextColor(k.TextMuted)
		ui.Column(c).Gap(u).Role(ui.RoleList).Children(func() {
			for _, s := range opts.Stages {
				ui.Row(c).Gap(u * 2).AlignItems(ui.Center).Role(ui.RoleListItem).Children(func() {
					if s.Done {
						ui.Icon(c, icons.Must("check")).Size(14, 14).TextColor(k.Success)
					} else {
						ui.Icon(c, icons.Must("circle")).Size(14, 14).TextColor(k.TextMuted)
					}
					t := ui.Text(c, s.Label).FontSize(small)
					if !s.Done {
						t.TextColor(k.TextMuted)
					}
				})
			}
		})
	}).Changed()
}