fr0 add recurring series recurrence edit

1 files changed, 10 insertions, 2 deletions

diff --git a/internal/ui/handlers.go b/internal/ui/handlers.go
index 7832070..9e5fcec 100644
--- a/internal/ui/handlers.go
+++ b/internal/ui/handlers.go
@@ -209,8 +209,14 @@ func (m *Model) handleRecurrenceMode(msg tea.KeyPressMsg) (tea.Model, tea.Cmd) {
 		}
 		ctx, cancel := m.taskOperationContext()
 		defer cancel()
-		if err := m.taskwarriorClient().SetRecurrenceContext(ctx, m.recurID, value); err != nil {
-			return err
+		if m.recurSeries {
+			if err := m.taskwarriorClient().SetRecurringSeriesRecurrenceContext(ctx, m.recurRoot, value); err != nil {
+				return err
+			}
+		} else {
+			if err := m.taskwarriorClient().SetRecurrenceContext(ctx, m.recurID, value); err != nil {
+				return err
+			}
 		}
 		if err := m.reload(); err != nil {
 			return fmt.Errorf("reloading tasks: %w", err)
@@ -220,6 +226,8 @@ func (m *Model) handleRecurrenceMode(msg tea.KeyPressMsg) (tea.Model, tea.Cmd) {
 
 	onExit := func() {
 		m.recurEditing = false
+		m.recurSeries = false
+		m.recurRoot = ""
 	}
 
 	model, cmd := m.handleTextInput(msg, &m.recurInput, onEnter, onExit)