Version final

ui/update.go

@@ -0,0 +1,175 @@
+package ui
+
+import (
+	"strings"
+
+	"github.com/charmbracelet/bubbles/textinput"
+	tea "github.com/charmbracelet/bubbletea"
+)
+
+func (m Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
+	// 1. Manejo especial cuando el Registry está caído (StateRegistryDown)
+	if m.state == StateRegistryDown {
+		if keyMsg, ok := msg.(tea.KeyMsg); ok {
+			switch keyMsg.String() {
+			case "enter", "L", "l":
+				m.state = StateLoading
+				m.mode = ModeLocal
+				m.statusMsg = ""
+				return m, fetchLocalImagesCmd(m.client)
+			case "q", "ctrl+c", "CTRL+C":
+				return m, tea.Quit
+			}
+		}
+		return m, nil
+	}
+
+	switch msg := msg.(type) {
+	case tea.KeyMsg:
+		if m.state == StateNewTag {
+			switch msg.String() {
+			case "esc":
+				m.state = StateMainTable
+				m.textInput.Reset()
+				return m, nil
+			case "enter":
+				inputVal := m.textInput.Value()
+				if inputVal != "" {
+					lastColon := strings.LastIndex(inputVal, ":")
+					var destRepo, destTag string
+					if lastColon != -1 {
+						destRepo = inputVal[:lastColon]
+						destTag = inputVal[lastColon+1:]
+					} else {
+						destRepo = inputVal
+						destTag = "latest"
+					}
+
+					if destRepo != "" && destTag != "" {
+						selectedRow := m.rows[m.cursor]
+						m.state = StateLoading
+						m.textInput.Reset()
+						if m.mode == ModeLocal {
+							return m, tagLocalImageCmd(m.client, selectedRow.RepoName, selectedRow.Tag, destRepo, destTag)
+						}
+						return m, copyImageCmd(m.client, selectedRow.RepoName, selectedRow.Tag, destRepo, destTag)
+					}
+				}
+				return m, nil
+			}
+			var cmd tea.Cmd
+			m.textInput, cmd = m.textInput.Update(msg)
+			return m, cmd
+		}
+
+		switch msg.String() {
+		case "ctrl+c", "q":
+			if m.state == StateConfirmDelete {
+				m.state = StateMainTable
+				return m, nil
+			}
+			return m, tea.Quit
+
+		case "up", "k":
+			if m.cursor > 0 {
+				m.cursor--
+			}
+
+		case "down", "j":
+			if m.cursor < len(m.rows)-1 {
+				m.cursor++
+			}
+
+		case "r":
+			if m.state == StateMainTable && m.mode != ModeRegistry {
+				m.state = StateLoading
+				m.cursor = 0
+				m.statusMsg = ""
+				m.mode = ModeRegistry
+				return m, fetchRegistryImagesCmd(m.client)
+			}
+
+		case "l":
+			if m.state == StateMainTable && m.mode != ModeLocal {
+				m.state = StateLoading
+				m.cursor = 0
+				m.statusMsg = ""
+				m.mode = ModeLocal
+				return m, fetchLocalImagesCmd(m.client)
+			}
+
+		case "d":
+			if m.state == StateMainTable && len(m.rows) > 0 {
+				m.state = StateConfirmDelete
+			}
+
+		case "t":
+			if m.state == StateMainTable && len(m.rows) > 0 {
+				m.state = StateNewTag
+				m.textInput.Focus()
+				return m, textinput.Blink
+			}
+			
+		case "p":
+			if m.state == StateMainTable && m.mode == ModeLocal && len(m.rows) > 0 {
+				selectedRow := m.rows[m.cursor]
+				m.state = StateLoading
+				return m, pushImageCmd(m.client, selectedRow.RepoName, selectedRow.Tag)
+			}            
+
+		case "y":
+			if m.state == StateConfirmDelete {
+				selectedRow := m.rows[m.cursor]
+				m.state = StateLoading
+				if m.mode == ModeLocal {
+					return m, deleteLocalImageCmd(m.client, selectedRow.RepoName, selectedRow.Tag)
+				}
+				return m, deleteTagCmd(m.client, selectedRow.RepoName, selectedRow.Tag)
+			}
+
+		case "n":
+			if m.state == StateConfirmDelete {
+				m.state = StateMainTable
+			}
+		}
+
+	case imagesMsg:
+		m.rows = msg.rows
+		if msg.mode == ModeRegistry {
+			m.registryRows = msg.rows
+		} else {
+			m.localRows = msg.rows
+		}
+		m.state = StateMainTable
+		m.cursor = 0
+
+	case successMsg:
+		m.statusMsg = string(msg)
+		m.state = StateLoading
+		if m.mode == ModeRegistry {
+			return m, fetchRegistryImagesCmd(m.client)
+		} else {
+			return m, fetchLocalImagesCmd(m.client)
+		}
+
+	case errorMsg:
+		errVal := error(msg)
+		errMsgStr := errVal.Error()
+		
+		// Comprobamos si el error es de conexión con el registro
+		if strings.Contains(errMsgStr, "connection refused") || 
+		   strings.Contains(errMsgStr, "dial tcp") ||
+		   strings.Contains(errMsgStr, "Get \"http://localhost:5000") {
+			m.state = StateRegistryDown
+			m.statusMsg = errMsgStr
+			return m, nil
+		}
+
+		// Error estándar para otros casos
+		m.err = errVal
+		m.state = StateError
+		return m, nil
+	}
+
+	return m, nil
+}