@fatalmistake02/rian 0.0.0-stage → 0.2.1

package/internal/ui/update.go

@@ -0,0 +1,470 @@
+package ui
+
+import (
+	"fmt"
+	"time"
+
+	"charm.land/bubbles/v2/spinner"
+	tea "charm.land/bubbletea/v2"
+
+	"rian/internal/config"
+	"rian/internal/project"
+)
+
+func (m *Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
+	switch msg := msg.(type) {
+	case tea.WindowSizeMsg:
+		m.width, m.height = msg.Width, msg.Height
+		m.resize()
+	case readyMsg:
+		m.available = msg.available
+		if msg.err != nil {
+			return m, m.notify("Unable to load projects", msg.err)
+		}
+		m.cfg = msg.cfg
+		cmds := []tea.Cmd{}
+		// Initial badges are loaded in small parallel batches; subsequent polls
+		// inspect only the active project.
+		for i, p := range m.cfg.Projects {
+			if i >= 8 {
+				break
+			}
+			cmds = append(cmds, m.refresh(p, true))
+		}
+		return m, tea.Batch(cmds...)
+	case gitMsg:
+		m.statuses[msg.id], m.loading[msg.id], m.lastRefresh[msg.id] = msg.status, false, time.Now()
+	case inspectedMsg:
+		m.busy = false
+		if msg.purpose == "add" && m.mode != add || msg.purpose == "edit" && m.mode != edit {
+			return m, nil
+		}
+		if msg.err != nil {
+			m.formError = msg.err.Error()
+			return m, m.notify("Could not inspect directory", msg.err)
+		}
+		m.available = msg.result.Available
+		if msg.purpose == "add" {
+			for _, p := range m.cfg.Projects {
+				if project.SamePath(p.Path, msg.result.Project.Path) {
+					m.formError = "Already saved as " + p.Name
+					return m, nil
+				}
+			}
+			m.pending, m.mode = msg.result, addConfirm
+			m.viewport.SetContent(m.detectionSummary(msg.result))
+			m.viewport.GotoTop()
+		} else if msg.purpose == "edit" {
+			p := m.editProject
+			p.Path, p.AutoCommands, p.Stack = msg.result.Project.Path, msg.result.Project.AutoCommands, msg.result.Project.Stack
+			m.statuses[p.ID] = msg.result.Git
+			return m, m.save(p.ID, "Project updated", func(cfg *config.Config) error { return config.Replace(cfg, p) })
+		} else {
+			m.statuses[msg.id] = msg.result.Git
+			return m, m.save(msg.id, "Project rescanned", func(cfg *config.Config) error {
+				for i := range cfg.Projects {
+					if cfg.Projects[i].ID == msg.id {
+						cfg.Projects[i].AutoCommands = msg.result.Project.AutoCommands
+						cfg.Projects[i].Stack = msg.result.Project.Stack
+						return nil
+					}
+				}
+				return fmt.Errorf("project no longer exists")
+			})
+		}
+	case savedMsg:
+		// A recent-use write can finish while an edit is still saving. Only
+		// the explicit mutation that set busy may release that input guard.
+		if msg.notice != "" {
+			m.busy = false
+		}
+		if msg.err != nil {
+			m.formError = msg.err.Error()
+			return m, m.notify("Could not save project", msg.err)
+		}
+		selectedID := msg.id
+		if msg.notice == "Project added" || msg.notice == "Project updated" {
+			m.search.Reset()
+			m.searching = false
+			m.search.Blur()
+		}
+		if msg.notice == "" {
+			if p, ok := m.current(); ok {
+				selectedID = p.ID
+			}
+		}
+		if msg.cfg.Revision >= m.cfg.Revision {
+			m.cfg = msg.cfg
+		}
+		m.selectID(selectedID)
+		if msg.notice != "" {
+			m.closeModal()
+			return m, m.notify(msg.notice, nil)
+		}
+	case preparedExecMsg:
+		if msg.err != nil {
+			return m, m.notify(msg.label, msg.err)
+		}
+		return m, tea.ExecProcess(msg.cmd, func(err error) tea.Msg {
+			if err != nil {
+				err = fmt.Errorf("%s failed: %w", msg.label, err)
+			}
+			return actionMsg{msg.id, msg.label + " finished", err, msg.refresh}
+		})
+	case actionMsg:
+		cmds := []tea.Cmd{m.notify(msg.notice, msg.err)}
+		if msg.err == nil && msg.id != "" {
+			store := m.store
+			cmds = append(cmds, func() tea.Msg {
+				cfg, err := store.Update(func(cfg *config.Config) error { return config.Touch(cfg, msg.id) })
+				return savedMsg{cfg: cfg, id: msg.id, err: err}
+			})
+		}
+		if msg.refresh {
+			for _, p := range m.cfg.Projects {
+				if p.ID == msg.id {
+					cmds = append(cmds, m.refresh(p, true))
+					break
+				}
+			}
+		}
+		return m, tea.Batch(cmds...)
+	case processMsg:
+		m.busy = false
+		if msg.err != nil {
+			return m, m.notify("Command failed", msg.err)
+		}
+		if m.processID == msg.id {
+			m.processState = msg.state
+		}
+		store := m.store
+		return m, func() tea.Msg {
+			cfg, err := store.Update(func(cfg *config.Config) error { return config.Touch(cfg, msg.id) })
+			return savedMsg{cfg: cfg, id: msg.id, err: err}
+		}
+	case tickMsg:
+		cmds := []tea.Cmd{tick()}
+		if p, ok := m.current(); ok {
+			cmds = append(cmds, m.refresh(p, false))
+		}
+		if m.mode == processView {
+			m.processState = m.manager.Snapshot(m.processID)
+			m.viewport.SetContent(plain(m.processState.Output))
+			if m.followOutput {
+				m.viewport.GotoBottom()
+			}
+		}
+		return m, tea.Batch(cmds...)
+	case spinner.TickMsg:
+		var cmd tea.Cmd
+		m.spin, cmd = m.spin.Update(msg)
+		return m, cmd
+	case toastExpired:
+		if int(msg) == m.toastID {
+			m.toast = ""
+		}
+	case tea.FocusMsg:
+		if p, ok := m.current(); ok {
+			return m, m.refresh(p, true)
+		}
+	case tea.KeyPressMsg:
+		return m, m.key(msg)
+	default:
+		var cmd tea.Cmd
+		if m.mode == edit && len(m.fields) > 0 {
+			m.fields[m.fieldIndex].input, cmd = m.fields[m.fieldIndex].input.Update(msg)
+		} else if m.commandEditing {
+			if m.commandField == 0 {
+				m.commandName, cmd = m.commandName.Update(msg)
+			} else {
+				m.commandValue, cmd = m.commandValue.Update(msg)
+			}
+		} else if m.searching && m.mode == home {
+			m.search, cmd = m.search.Update(msg)
+		} else {
+			m.query, cmd = m.query.Update(msg)
+		}
+		return m, cmd
+	}
+	return m, nil
+}
+
+func (m *Model) key(msg tea.KeyPressMsg) tea.Cmd {
+	key := msg.String()
+	if key == "ctrl+c" {
+		return m.quit()
+	}
+	if m.busy {
+		if key == "esc" {
+			m.closeModal()
+		}
+		return nil
+	}
+	if m.mode == home && m.searching {
+		switch key {
+		case "esc":
+			m.searching = false
+			m.search.Reset()
+			m.search.Blur()
+			m.selected = 0
+			return nil
+		case "enter":
+			m.searching = false
+			m.search.Blur()
+			if p, ok := m.current(); ok {
+				return m.refresh(p, false)
+			}
+			return nil
+		case "up", "down":
+			return m.navigate(key)
+		default:
+			var cmd tea.Cmd
+			m.search, cmd = m.search.Update(msg)
+			m.selected = 0
+			return cmd
+		}
+	}
+	if m.mode != home {
+		return m.modalKey(msg)
+	}
+	switch key {
+	case "q":
+		return m.quit()
+	case "up", "down", "j", "k", "home", "end":
+		return m.navigate(key)
+	case "/":
+		m.searching = true
+		return m.search.Focus()
+	case "esc":
+		m.search.Reset()
+		m.selected = 0
+	case "ctrl+k":
+		return m.openMenu(palette)
+	case "?":
+		m.mode = help
+		m.resize()
+		m.viewport.GotoTop()
+	case "a", "+":
+		m.startAdd()
+		return m.query.Focus()
+	case "enter":
+		if _, ok := m.current(); ok {
+			return m.openMenu(palette)
+		}
+	case "v":
+		return m.execute(Action{ID: "vscode", Label: "VS Code"})
+	case "o":
+		return m.execute(Action{ID: "opencode", Label: "OpenCode"})
+	case "K", "shift+k":
+		return m.execute(Action{ID: "kilocode", Label: "KiloCode"})
+	case "c":
+		return m.execute(Action{ID: "chatgpt", Label: "ChatGPT"})
+	case "t":
+		return m.execute(Action{ID: "terminal", Label: "Terminal"})
+	case "g":
+		return m.execute(Action{ID: "github", Label: "GitHub"})
+	case "G", "shift+g":
+		return m.openMenu(gitMenu)
+	case "r":
+		if p, ok := m.current(); ok && m.manager.Snapshot(p.ID).Running {
+			m.showProcess(p.ID)
+			return nil
+		}
+		return m.openMenu(commands)
+	case "d":
+		return m.openMenu(commands)
+	case "p":
+		if p, ok := m.current(); ok {
+			m.showProcess(p.ID)
+		}
+	case "f":
+		return m.execute(Action{ID: "favorite"})
+	case "e":
+		return m.execute(Action{ID: "edit"})
+	case "delete":
+		return m.execute(Action{ID: "remove"})
+	case "ctrl+r":
+		if _, ok := m.current(); ok {
+			return m.execute(Action{ID: "refresh"})
+		}
+	}
+	return nil
+}
+
+func (m *Model) navigate(key string) tea.Cmd {
+	count := len(m.projects())
+	if count == 0 {
+		return nil
+	}
+	switch key {
+	case "up", "k":
+		m.selected = max(0, m.selected-1)
+	case "down", "j":
+		m.selected = min(count-1, m.selected+1)
+	case "home":
+		m.selected = 0
+	case "end":
+		m.selected = count - 1
+	}
+	p, _ := m.current()
+	return m.refresh(p, false)
+}
+
+func (m *Model) modalKey(msg tea.KeyPressMsg) tea.Cmd {
+	key := msg.String()
+	if key == "esc" {
+		m.closeModal()
+		return nil
+	}
+	switch m.mode {
+	case help:
+		if key == "?" || key == "q" || key == "enter" {
+			m.closeModal()
+			return nil
+		}
+		var cmd tea.Cmd
+		m.viewport, cmd = m.viewport.Update(msg)
+		return cmd
+	case add:
+		if key == "enter" {
+			return m.inspectDirectory("add", m.query.Value(), "")
+		}
+		var cmd tea.Cmd
+		m.query, cmd = m.query.Update(msg)
+		return cmd
+	case addConfirm:
+		if key == "enter" {
+			m.statuses[m.pending.Project.ID] = m.pending.Git
+			return m.confirmAdd()
+		}
+		var cmd tea.Cmd
+		m.viewport, cmd = m.viewport.Update(msg)
+		return cmd
+	case edit:
+		switch key {
+		case "ctrl+s":
+			return m.submitEdit()
+		case "tab", "down", "enter":
+			if key == "enter" && m.fields[m.fieldIndex].toggle {
+				m.fields[m.fieldIndex].checked = !m.fields[m.fieldIndex].checked
+				return nil
+			}
+			return m.focusField(m.fieldIndex + 1)
+		case "shift+tab", "up":
+			return m.focusField(m.fieldIndex - 1)
+		case " ", "space":
+			if m.fields[m.fieldIndex].toggle {
+				m.fields[m.fieldIndex].checked = !m.fields[m.fieldIndex].checked
+				return nil
+			}
+		}
+		var cmd tea.Cmd
+		m.fields[m.fieldIndex].input, cmd = m.fields[m.fieldIndex].input.Update(msg)
+		return cmd
+	case palette, commands, gitMenu:
+		items := m.filteredActions()
+		switch key {
+		case "up", "ctrl+p":
+			m.menuIndex = max(0, m.menuIndex-1)
+		case "down", "ctrl+n":
+			if key == "ctrl+n" && m.mode == commands {
+				m.startCustom("", "")
+				return m.commandName.Focus()
+			}
+			m.menuIndex = min(max(0, len(items)-1), m.menuIndex+1)
+		case "enter":
+			if len(items) > 0 {
+				return m.execute(items[min(m.menuIndex, len(items)-1)])
+			}
+		case "ctrl+e":
+			if m.mode == commands {
+				return m.commandEdit(false)
+			}
+		case "ctrl+d":
+			if m.mode == commands {
+				return m.commandEdit(true)
+			}
+		default:
+			var cmd tea.Cmd
+			m.query, cmd = m.query.Update(msg)
+			m.menuIndex = 0
+			return cmd
+		}
+	case removeConfirm:
+		if key == "enter" || key == "y" {
+			p := m.editProject
+			if m.manager.Snapshot(p.ID).Running {
+				return m.notify("Command running", fmt.Errorf("stop this project's command before removing it"))
+			}
+			return m.save("", "Project removed from Rian", func(cfg *config.Config) error { return config.Remove(cfg, p.ID) })
+		}
+		if key == "n" {
+			m.closeModal()
+		}
+	case quitConfirm:
+		if key == "enter" || key == "y" {
+			manager := m.manager
+			return tea.Sequence(func() tea.Msg { manager.Close(); return nil }, tea.Quit)
+		}
+		if key == "n" {
+			m.closeModal()
+		}
+	case prompt:
+		if m.commandEditing {
+			if key == "ctrl+s" {
+				return m.submitCustom()
+			}
+			if key == "tab" || key == "shift+tab" || key == "enter" {
+				if m.commandField == 0 {
+					m.commandName.Blur()
+					m.commandField = 1
+					return m.commandValue.Focus()
+				}
+				if key == "enter" {
+					return m.submitCustom()
+				}
+				m.commandValue.Blur()
+				m.commandField = 0
+				return m.commandName.Focus()
+			}
+			var cmd tea.Cmd
+			if m.commandField == 0 {
+				m.commandName, cmd = m.commandName.Update(msg)
+			} else {
+				m.commandValue, cmd = m.commandValue.Update(msg)
+			}
+			return cmd
+		}
+		if key == "enter" {
+			return m.submitPrompt()
+		}
+		var cmd tea.Cmd
+		m.query, cmd = m.query.Update(msg)
+		return cmd
+	case processView:
+		if key == "s" && m.processState.Running {
+			manager, id := m.manager, m.processID
+			return func() tea.Msg { err := manager.Stop(id); return actionMsg{notice: "Process stopped", err: err} }
+		}
+		if key == "r" && !m.processState.Running && m.processState.Command != "" {
+			p, ok := m.current()
+			if ok {
+				return m.runCommand(p, "", m.processState.Command)
+			}
+		}
+		if key == "f" {
+			m.followOutput = !m.followOutput
+			if m.followOutput {
+				m.viewport.GotoBottom()
+			}
+			return nil
+		}
+		if key == "up" || key == "k" || key == "pgup" || key == "home" {
+			m.followOutput = false
+		}
+		var cmd tea.Cmd
+		m.viewport, cmd = m.viewport.Update(msg)
+		return cmd
+	}
+	return nil
+}