1
0
Fork 0
neonmodem/ui/windowmanager/windowmanager.go

222 lines
4.3 KiB
Go

package windowmanager
import (
tea "github.com/charmbracelet/bubbletea"
"github.com/mrusme/neonmodem/ui/cmd"
"github.com/mrusme/neonmodem/ui/ctx"
"github.com/mrusme/neonmodem/ui/helpers"
"github.com/mrusme/neonmodem/ui/windows"
)
type StackItem struct {
ID string
Win windows.Window
XYWH [4]int
}
type WM struct {
ctx *ctx.Ctx
stack []StackItem
}
func New(c *ctx.Ctx) *WM {
wm := new(WM)
wm.ctx = c
return wm
}
func (wm *WM) Open(id string, win windows.Window, xywh [4]int, command *cmd.Command) []tea.Cmd {
var tcmds []tea.Cmd
if wm.IsOpen(id) {
if wm.IsFocused(id) {
return tcmds
}
return wm.Focus(id)
}
item := new(StackItem)
item.ID = id
item.Win = win
item.XYWH = xywh
wm.stack = append(wm.stack, *item)
command.AddArg("xywh", item.XYWH)
tcmds = append(tcmds, wm.Update(id, *command))
tcmds = append(tcmds, wm.Resize(id, wm.ctx.Content[0], wm.ctx.Content[1])...)
// tcmds = append(tcmds, wm.Update(id, *cmd.New(
// cmd.WinRefreshData,
// id,
// )))
fcmds := wm.Focus(id)
tcmds = append(tcmds, fcmds...)
return tcmds
}
func (wm *WM) CloseFocused() (bool, []tea.Cmd) {
return wm.Close(wm.Focused())
}
func (wm *WM) Close(id string, args ...cmd.Arg) (bool, []tea.Cmd) {
var tcmds []tea.Cmd
for i := len(wm.stack) - 1; i >= 0; i-- {
if wm.stack[i].ID == id {
wm.stack = append(wm.stack[:i], wm.stack[i+1:]...)
tcmds = append(tcmds, cmd.New(cmd.WinClose, id, args...).Tea())
wm.ctx.Loading = false
nrOpen := wm.GetNumberOpen()
if nrOpen == 0 {
tcmds = append(tcmds, cmd.New(cmd.ViewFocus, "*").Tea())
} else {
tcmds = append(tcmds, wm.Focus(wm.stack[(nrOpen-1)].ID)...)
}
return true, tcmds
}
}
return false, tcmds
}
func (wm *WM) Focus(id string) []tea.Cmd {
var tcmds []tea.Cmd
for i := 0; i < len(wm.stack); i++ {
var tcmd tea.Cmd
if wm.stack[i].ID == id {
wm.stack[i].Win, tcmd = wm.stack[i].Win.Update(*cmd.New(cmd.WinFocus, wm.stack[i].ID))
} else {
wm.stack[i].Win, tcmd = wm.stack[i].Win.Update(*cmd.New(cmd.WinBlur, wm.stack[i].ID))
}
tcmds = append(tcmds, tcmd)
}
tcmds = append(tcmds, cmd.New(cmd.ViewBlur, "*").Tea())
return tcmds
}
func (wm *WM) Focused() string {
l := len(wm.stack) - 1
if l < 0 {
return ""
}
return wm.stack[l].ID
}
func (wm *WM) IsOpen(id string) bool {
for _, win := range wm.stack {
if win.ID == id {
return true
}
}
return false
}
func (wm *WM) IsFocused(id string) bool {
return id == wm.Focused()
}
func (wm *WM) GetNumberOpen() int {
return len(wm.stack)
}
func (wm *WM) Update(id string, msg tea.Msg) tea.Cmd {
var teaCmd tea.Cmd
for i := 0; i < len(wm.stack); i++ {
if wm.stack[i].ID == id {
wm.stack[i].Win, teaCmd = wm.stack[i].Win.Update(msg)
}
}
return teaCmd
}
func (wm *WM) UpdateAll(msg tea.Msg) []tea.Cmd {
var tcmd tea.Cmd
var tcmds []tea.Cmd
for i := 0; i < len(wm.stack); i++ {
wm.stack[i].Win, tcmd = wm.stack[i].Win.Update(msg)
tcmds = append(tcmds, tcmd)
}
return tcmds
}
func (wm *WM) UpdateFocused(msg tea.Msg) []tea.Cmd {
var tcmd tea.Cmd
var tcmds []tea.Cmd
i := len(wm.stack) - 1
if i < 0 {
return tcmds
}
wm.stack[i].Win, tcmd = wm.stack[i].Win.Update(msg)
tcmds = append(tcmds, tcmd)
return tcmds
}
func (wm *WM) Resize(id string, w int, h int) []tea.Cmd {
var tcmd tea.Cmd
var tcmds []tea.Cmd
for i := 0; i < len(wm.stack); i++ {
if wm.stack[i].ID == id {
wm.stack[i].Win, tcmd = wm.stack[i].Win.Update(tea.WindowSizeMsg{
Width: w - wm.stack[i].XYWH[0] - wm.stack[i].XYWH[2],
Height: h - wm.stack[i].XYWH[1] - wm.stack[i].XYWH[3],
})
tcmds = append(tcmds, tcmd)
}
}
return tcmds
}
func (wm *WM) ResizeAll(w int, h int) []tea.Cmd {
var tcmd tea.Cmd
var tcmds []tea.Cmd
for i := 0; i < len(wm.stack); i++ {
wm.stack[i].Win, tcmd = wm.stack[i].Win.Update(tea.WindowSizeMsg{
Width: w - wm.stack[i].XYWH[0] - wm.stack[i].XYWH[2],
Height: h - wm.stack[i].XYWH[1] - wm.stack[i].XYWH[3],
})
tcmds = append(tcmds, tcmd)
}
return tcmds
}
func (wm *WM) View(view string, onlyFocused bool) string {
var v string = view
var j int = 0
if onlyFocused {
j = len(wm.stack) - 1
if j < 0 {
j = 0
}
}
for i := j; i < len(wm.stack); i++ {
v = helpers.PlaceOverlay(
wm.stack[i].XYWH[0],
wm.stack[i].XYWH[1]+(wm.ctx.Screen[1]-wm.ctx.Content[1]),
wm.stack[i].Win.View(),
v,
wm.ctx.Config.RenderShadows,
)
}
return v
}