Update service_exec.go

ewilliams0305 · Dec 21, 2023 · 70d4eb4 · 70d4eb4
1 parent d6e04aa
commit 70d4eb4
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/pkg/tui/service_exec.go b/pkg/tui/service_exec.go
@@ -115,16 +115,17 @@ func (m VirtualControlServiceModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 			return m, nil
 		}
 
+	case progress.Model:
+		m.progress = msg
+
 	case progress.FrameMsg:
 		progressModel, cmd := m.progress.Update(msg)
 		m.progress = progressModel.(progress.Model)
 		return m, cmd
 
 	case progressTick:
 		if m.progress.Percent() == 1.0 {
-			prog := progress.New(progress.WithDefaultGradient())
-			prog.Width = app.width
-			m.progress = prog
+			m.progress.SetPercent(0)
 			return m, tea.Batch(openJournal(), tea.EnterAltScreen)
 		}
 		return m, tea.Batch(systemTickCmd(), m.progress.IncrPercent(0.20))
@@ -159,7 +160,6 @@ func systemTickCmd() tea.Cmd {
 		return progressTick(t)
 	})
 }
-
 func openJournal() tea.Cmd {
 
 	c := exec.Command("journalctl", "-u", "virtualcontrol.service", "-f")