From 9d16b2970cbf83757e9724c9db7c7022fa4daa80 Mon Sep 17 00:00:00 2001
From: filifa <filifa.breath652@8alias.com>
Date: Wed, 4 Sep 2024 21:31:52 -0500
Subject: [PATCH] refactor main loop

---
 cmd/root.go | 30 ++++++++++++++++++------------
 1 file changed, 18 insertions(+), 12 deletions(-)

diff --git a/cmd/root.go b/cmd/root.go
index 13a9816..9739d54 100644
--- a/cmd/root.go
+++ b/cmd/root.go
@@ -87,6 +87,23 @@ func handleEvent(event sdl.Event, m *milkDropWindow) bool {
 	return true
 }
 
+/*
+ * update handles events and renders new frames of the visualization. It
+ * returns a bool indicating whether the program should keep running and an
+ * error, if any.
+ */
+func update(m *milkDropWindow) (bool, error) {
+	for event := sdl.PollEvent(); event != nil; event = sdl.PollEvent() {
+		keepRunning := handleEvent(event, m)
+		if !keepRunning {
+			return false, nil
+		}
+	}
+
+	keepRunning, err := m.render()
+	return keepRunning, err
+}
+
 /*
  * milkbucket sets up the program and starts a rendering loop.
  */
@@ -119,18 +136,7 @@ func milkbucket(cmd *cobra.Command, args []string) {
 
 	running := true
 	for running {
-		for event := sdl.PollEvent(); event != nil; event = sdl.PollEvent() {
-			running = handleEvent(event, m)
-			if !running {
-				break
-			}
-		}
-
-		if !running {
-			break
-		}
-
-		running, err = m.render()
+		running, err = update(m)
 		if err != nil {
 			panic(err)
 		}