refactor user input

main.go

@@ -116,18 +116,31 @@ func (ctx *AppContext) AddNote(note Note) {
 }
 
 func main() {
+	// TODO: Clean up menu prompt
+	// TODO: Write getNote method
+	// TODO: Write option handlers (handleNewNote, handleNewReading)
+
 	var appCtx AppContext
 
-	err := appCtx.LoadRecords(recordsFilename)
-	if err != nil {
-		fmt.Printf("Error loading records: %s\n", err)
+	if err := appCtx.LoadRecords(recordsFilename); err != nil {
+		fmt.Printf("Error loading records from file: %s\n", err)
 	}
 
-	// Display menu
+	for {
 		choice, err := getUserInput("1. Enter new reading\n2. Enter new note\n3. Exit\nYour choice")
+		if err != nil {
+			fmt.Printf("Error reading choice: %s\n", err)
+			continue
+		}
+
+		if choice == "3" {
+			fmt.Println("Exiting...")
+			break
+		}
+
 		switch choice {
 		case "1":
-		// Take a new reading from the user
+			// Get a new Reading from the user
 			reading, err := getReading()
 			if err != nil {
 				fmt.Printf("Error getting reading: %s\n", err)
@@ -149,20 +162,21 @@ func main() {
 				appCtx.AddReading(reading)
 			}
 		case "2":
+			// Get a new Note from the user
 			noteText, err := getUserInput("Enter your note text: ")
 			if err != nil {
-			fmt.Println("Error getting note: %s\n", err)
+				fmt.Printf("Error getting note: %s\n", err)
 			} else {
 				note := NewNote(noteText)
 				appCtx.AddNote(note)
 			}
 		default:
-		fmt.Println("Quitting...")
+			fmt.Printf("Invalid choice, please try again.\n\n")
+		}
 	}
 
 	// Save Records before exiting
-	err = appCtx.SaveRecords(recordsFilename)
-	if err != nil {
+	if err := appCtx.SaveRecords(recordsFilename); err != nil {
 		fmt.Printf("Error saving records to file: %s\n", err)
 	}
 }