126 lines
3.6 KiB
Go
126 lines
3.6 KiB
Go
package mujica
|
|
|
|
import (
|
|
"fmt"
|
|
"log/slog"
|
|
"time"
|
|
|
|
"github.com/egoist/mygo/ui"
|
|
|
|
"mujicaui/icons"
|
|
"mujicaui/internal"
|
|
"mujicaui/theme"
|
|
)
|
|
|
|
// ScreenshotFrame is one caller-supplied frame of the agent's screen.
|
|
type ScreenshotFrame struct {
|
|
Image ui.ImageSource
|
|
Caption string
|
|
// At is the time since the run started.
|
|
At time.Duration
|
|
}
|
|
|
|
// ScreenshotStreamState is the shown frame; unpaused, it follows the newest.
|
|
type ScreenshotStreamState struct {
|
|
Index int
|
|
Paused bool
|
|
}
|
|
|
|
// ScreenshotStreamOptions configure a ScreenshotStream.
|
|
type ScreenshotStreamOptions struct {
|
|
// Height of the viewport; 280 DIPs when zero.
|
|
Height float32
|
|
Disabled bool
|
|
}
|
|
|
|
// ScreenshotStream shows screen frames live, paused or stepped.
|
|
func ScreenshotStream(c *ui.Context, st *ScreenshotStreamState, frames []ScreenshotFrame, opts ScreenshotStreamOptions) ChoiceResult {
|
|
if st == nil {
|
|
panic("mujica: ScreenshotStream needs a state")
|
|
}
|
|
if opts.Height < 0 {
|
|
panic("mujica: ScreenshotStream Height is negative")
|
|
}
|
|
for i, f := range frames {
|
|
if f.Image == nil {
|
|
panic(fmt.Sprintf("mujica: ScreenshotStream frame %d has no Image", i))
|
|
}
|
|
}
|
|
h := opts.Height
|
|
if h == 0 {
|
|
h = 280
|
|
}
|
|
k := Tokens(c)
|
|
u := Density(c).Unit()
|
|
n := len(frames)
|
|
if st.Index < 0 {
|
|
panic(fmt.Sprintf("mujica: ScreenshotStream Index %d is negative", st.Index))
|
|
}
|
|
if !st.Paused || st.Index >= n { // frames may be dropped while paused
|
|
|
|
st.Index = max(0, n-1)
|
|
}
|
|
var r ChoiceResult
|
|
r.Element = agentCard(c).Disabled(opts.Disabled).Label(msg(c, "screenshot_stream.label"))
|
|
locked := r.Element.IsDisabled()
|
|
step := func(d int) {
|
|
if n == 0 || locked {
|
|
return
|
|
}
|
|
st.Index = max(0, min(n-1, st.Index+d))
|
|
st.Paused, r.changed = true, true
|
|
}
|
|
r.Element.Children(func() {
|
|
view := ui.Box(c).FillWidth().Height(h).Center().Clip().Radius(theme.ControlRadius).
|
|
Background(k.Background).Border(theme.BorderWidth, k.Border).Focusable().
|
|
Label(msg(c, "screenshot_stream.label"))
|
|
internal.Ring(view, k.Focus, theme.ControlRadius)
|
|
view.Children(func() {
|
|
if n == 0 {
|
|
agentCaption(c, msg(c, "screenshot_stream.waiting"))
|
|
return
|
|
}
|
|
ui.Image(c, frames[st.Index].Image).Fill().Fit(ui.Contain).Label(frames[st.Index].Caption)
|
|
})
|
|
if !locked {
|
|
switch {
|
|
case view.Shortcut(0, ui.KeyLeft):
|
|
step(-1)
|
|
case view.Shortcut(0, ui.KeyRight):
|
|
step(1)
|
|
case view.Shortcut(0, ui.KeySpace):
|
|
st.Paused, r.changed = !st.Paused, true
|
|
}
|
|
}
|
|
ui.Row(c).Gap(u * 2).AlignItems(ui.Center).Children(func() {
|
|
if iconAction(c, icons.Must("chevron-left"), msg(c, "screenshot_stream.prev")).Disabled(locked || st.Index == 0).Clicked() {
|
|
step(-1)
|
|
}
|
|
pause := msg(c, "screenshot_stream.pause")
|
|
if st.Paused {
|
|
pause = msg(c, "screenshot_stream.live")
|
|
}
|
|
if Button(c, pause, ButtonOptions{Variant: Secondary, Disabled: locked || n == 0}).Clicked() && !locked {
|
|
st.Paused, r.changed = !st.Paused, true
|
|
}
|
|
if iconAction(c, icons.Must("chevron-right"), msg(c, "screenshot_stream.next")).Disabled(locked || st.Index >= n-1).Clicked() {
|
|
step(1)
|
|
}
|
|
counter := fmt.Sprintf(msg(c, "screenshot_stream.counter"), min(st.Index+1, n), n)
|
|
agentCaption(c, counter).Role(ui.RoleStatus)
|
|
if n > 0 {
|
|
f := frames[st.Index]
|
|
ui.Text(c, f.Caption).SingleLine().Grow(1).Shrink(1)
|
|
agentCaption(c, agentDuration(f.At))
|
|
}
|
|
if !st.Paused && n > 0 {
|
|
AgentStatus(c, AgentRunning, AgentStatusOptions{Label: msg(c, "screenshot_stream.following"), Still: true})
|
|
}
|
|
})
|
|
})
|
|
if r.changed {
|
|
slog.Debug("mujica: screenshot_stream move", "index", st.Index, "paused", st.Paused)
|
|
}
|
|
return r
|
|
}
|