Refactoring the main code

.gitignore

@@ -0,0 +1,25 @@
+# Binaries
+todo
+todo.exe
+todo-mac
+todo-linux
+main
+
+# OS files
+.DS_Store
+
+# Editor files
+.vscode/
+.idea/
+
+# Dependency directories
+vendor/
+
+# Output files
+*.out
+
+# Data files
+todos.json
+
+# Test cache
+*.test

main.go

@@ -18,88 +18,60 @@ type Todo struct {
 }
 
 func printTodo(todo Todo) {
-	fmt.Println("- ID: ", todo.Id)
+	fmt.Printf("- ID: %d\n- Title: %s\n- Description: %s\n- Completed: %v\n--------\n",
-	fmt.Println("- Title: ", todo.Title)
+		todo.Id, todo.Title, todo.Description, todo.Completed)
-	fmt.Println("- Description: ", todo.Description)
-	fmt.Println("- Completed: ", todo.Completed)
-	fmt.Println("--------")
 }
 
-func listTodos() {
+func listTodos(todos []Todo) {
-	todos := getAllTodos()
 	fmt.Println("TODO: \n------")
 	for _, todo := range todos {
-		if todo.Completed {
+		if !todo.Completed {
-			continue
-		}
 			printTodo(todo)
 		}
+	}
 	fmt.Println("\nCompleted: \n------")
 	for _, todo := range todos {
-		if !todo.Completed {
+		if todo.Completed {
-			continue
-		}
 			printTodo(todo)
 		}
-
-}
-
-func checkFile() {
-	_, err := os.Stat(fileName)
-	if os.IsNotExist(err) {
-		_, err := os.Create(fileName)
-		if err != nil {
-			log.Fatal("\n Error creating file: ", err)
-		}
 	}
 }
 
-func saveTodoList(todos []Todo) error {
+func ensureFile() error {
-	writeBytes, err := json.Marshal(todos)
+	f, err := os.OpenFile(fileName, os.O_CREATE|os.O_RDWR, 0666)
 	if err != nil {
 		return err
 	}
-	return os.WriteFile(fileName, writeBytes, 0666)
+	return f.Close()
 }
 
-func getAllTodos() []Todo {
+func loadTodos() ([]Todo, error) {
-	checkFile()
+	if err := ensureFile(); err != nil {
-	fileData, err := os.ReadFile(fileName)
+		return nil, err
+	}
+	data, err := os.ReadFile(fileName)
 	if err != nil {
-		saveTodoList([]Todo{})
+		return nil, err
 	}
-	todos := []Todo{}
+	if len(data) == 0 {
-	if err := json.Unmarshal(fileData, &todos); err != nil {
+		return []Todo{}, nil
-		saveTodoList([]Todo{})
 	}
-	return todos
+	var todos []Todo
+	if err := json.Unmarshal(data, &todos); err != nil {
+		return []Todo{}, nil // Reset on corrupt file
+	}
+	return todos, nil
 }
 
-func completeTodo(id int) {
+func saveTodos(todos []Todo) error {
-	todos := getAllTodos()
+	data, err := json.MarshalIndent(todos, "", "  ")
-	for i, todo := range todos {
+	if err != nil {
-		if todo.Id == id {
+		return err
-			todo.Completed = true
-			todos[i] = todo
-			break
 	}
-	}
+	return os.WriteFile(fileName, data, 0666)
-	saveTodoList(todos)
 }
 
-func removeTodo(id int) {
+func getNextId(todos []Todo) int {
-	todos := getAllTodos()
-	for i, todo := range todos {
-		if todo.Id == id {
-			todos = append(todos[:i], todos[i+1:]...)
-			break
-		}
-	}
-	saveTodoList(todos)
-}
-
-func getNextId() int {
-	todos := getAllTodos()
 	maxId := 0
 	for _, todo := range todos {
 		if todo.Id > maxId {
@@ -109,56 +81,103 @@ func getNextId() int {
 	return maxId + 1
 }
 
-func addTodo(todo Todo) {
+func addTodo(title, description string) error {
-	todos := getAllTodos()
+	todos, err := loadTodos()
-	todo.Id = getNextId()
-	todo.Completed = false
-	todos = append(todos, todo)
-	writeBytes, err := json.Marshal(todos)
 	if err != nil {
-		log.Fatal(err)
+		return err
 	}
-	if err := os.WriteFile(fileName, writeBytes, 0666); err != nil {
+	todo := Todo{
-		log.Fatal("There was an error writing the todo to the file", err)
+		Id:          getNextId(todos),
+		Title:       title,
+		Description: description,
+		Completed:   false,
 	}
+	todos = append(todos, todo)
+	return saveTodos(todos)
+}
+
+func completeTodo(id int) error {
+	todos, err := loadTodos()
+	if err != nil {
+		return err
+	}
+	for i := range todos {
+		if todos[i].Id == id {
+			todos[i].Completed = true
+			break
+		}
+	}
+	return saveTodos(todos)
+}
+
+func removeTodo(id int) error {
+	todos, err := loadTodos()
+	if err != nil {
+		return err
+	}
+	for i, todo := range todos {
+		if todo.Id == id {
+			todos = append(todos[:i], todos[i+1:]...)
+			break
+		}
+	}
+	return saveTodos(todos)
 }
 
 var (
-	action      *string
-	title       *string
-	description *string
-	id          *int
-)
-
-func init() {
 	action      = flag.String("action", "add", "action (add, remove, list, complete)")
 	title       = flag.String("title", "", "title of the todo")
 	description = flag.String("description", "", "description of the todo")
 	id          = flag.Int("id", 0, "ID of the todo to be actioned upon. Used for remove and complete")
+)
 
-}
 func main() {
 	flag.Parse()
+	switch *action {
+	case "add":
+		handleAdd()
+	case "remove":
+		handleRemove()
+	case "complete":
+		handleComplete()
+	case "list":
+		handleList()
+	default:
+		log.Fatal("Please input a valid action.")
+	}
+}
 
-	if *action == "add" {
+func handleAdd() {
 	if *title == "" || *description == "" {
 		log.Fatal("Title and Description must be provided for adding a new TODO")
 	}
-		addTodo(Todo{Title: *title, Description: *description})
+	if err := addTodo(*title, *description); err != nil {
-	} else if *action == "remove" {
+		log.Fatal(err)
+	}
+}
+
+func handleRemove() {
 	if *id == 0 {
 		log.Fatal("ID must be provided for removing a TODO")
 	}
-		removeTodo(*id)
+	if err := removeTodo(*id); err != nil {
-	} else if *action == "complete" {
+		log.Fatal(err)
+	}
+}
+
+func handleComplete() {
 	if *id == 0 {
 		log.Fatal("ID must be provided for marking a Todo as complete")
 	}
-		completeTodo(*id)
+	if err := completeTodo(*id); err != nil {
-	} else if *action == "list" {
+		log.Fatal(err)
-		listTodos()
-	} else {
-		log.Fatal("Please input a valid action.")
 	}
-
+}
+
+func handleList() {
+	todos, err := loadTodos()
+	if err != nil {
+		log.Fatal(err)
+	}
+	listTodos(todos)
 }