Done edit feature

Author: DanWlker

internal/app/edit/edit.go

@@ -4,6 +4,7 @@ import (
 	"fmt"
 	"os"
 	"slices"
+	"strings"
 
 	r_error "github.com/DanWlker/remind/internal/error"
 	"github.com/DanWlker/remind/internal/pkg/data"
@@ -32,7 +33,7 @@ func editTodoAssociatedWith(directory string) error {
 
 	dataFileFullPath := dataFolder + string(os.PathSeparator) + currentDirectoryRecord.DataFileName
 
-	prettyPrintedString, errSPrettyPrintDataFile := data.SPrettyPrintDataFile(dataFileFullPath, "")
+	prettyPrintedString, errSPrettyPrintDataFile := data.SPrettyPrintDataFile(dataFileFullPath, nil)
 	if errSPrettyPrintDataFile != nil {
 		return fmt.Errorf("data.SPrettyPrintDataFile: %w", errSPrettyPrintDataFile)
 	}
@@ -42,7 +43,17 @@ func editTodoAssociatedWith(directory string) error {
 		return fmt.Errorf("shared.OpenDefaultEditor: %w", errOpenDefaultEditor)
 	}
 
-	fmt.Println(result)
+	var todoList []data.TodoEntity
+	for _, item := range strings.Split(string(result), "\n") {
+		if item != "" {
+			todoList = append(todoList, data.TodoEntity{Text: item})
+		}
+	}
+
+	errWriteTodoToFile := data.WriteTodoToFile(dataFileFullPath, todoList)
+	if errWriteTodoToFile != nil {
+		return fmt.Errorf("helper.WriteTodoToFile: %w", errWriteTodoToFile)
+	}
 
 	return nil
 }

internal/app/list/list.go

@@ -31,7 +31,11 @@ func listOne(pathToFind string) error {
 		return fmt.Errorf("helper.GetDataFolder: %w", errGetDataFolder)
 	}
 
-	if errPrettyPrintFile := data.PrettyPrintDataFile(dataFolder+string(os.PathSeparator)+projectRecordEntity.DataFileName, "  "); errPrettyPrintFile != nil {
+	if errPrettyPrintFile := data.PrettyPrintDataFile(
+		dataFolder+string(os.PathSeparator)+projectRecordEntity.DataFileName,
+		func(todo string, index int) string {
+			return fmt.Sprintf("\t%v. %v", index, todo)
+		}); errPrettyPrintFile != nil {
 		return fmt.Errorf("helper.PrettyPrintDataFile: %w", errPrettyPrintFile)
 	}
 	return nil
@@ -48,7 +52,12 @@ func listConcurrently(item record.RecordEntity, dataFolder string) (chan string,
 	}
 
 	go func() {
-		result, errPrettyPrintDataFile := data.SPrettyPrintDataFile(dataFolder+string(os.PathSeparator)+item.DataFileName, "  ")
+		result, errPrettyPrintDataFile := data.SPrettyPrintDataFile(
+			dataFolder+string(os.PathSeparator)+item.DataFileName,
+			func(todo string, index int) string {
+				return fmt.Sprintf("\t%v. %v", index, todo)
+			},
+		)
 		if errPrettyPrintDataFile != nil {
 			c <- fmt.Sprintf("Error: Something went wrong: data.SPrettyPrintDataFile: %v", errPrettyPrintDataFile)
 		}

internal/pkg/data/data.go

@@ -56,23 +56,27 @@ func GetDataFolder() (string, error) {
 	return dataFolder, nil
 }
 
-// TODO: Change prefix to be a func
-func SPrettyPrintDataFile(dataFileFullPath string, prefix string) (string, error) {
+func SPrettyPrintDataFile(dataFileFullPath string, editText func(todo string, index int) string) (string, error) {
 	var b bytes.Buffer
 	todoList, errGetTodoFromDataFile := GetTodoFromDataFile(dataFileFullPath)
 	if errGetTodoFromDataFile != nil {
 		return "", fmt.Errorf("GetTodoFromDataFile: %w", errGetTodoFromDataFile)
 	}
 
 	for i, todo := range todoList {
-		b.WriteString(fmt.Sprintf("%v%v. %v\n", prefix, i, todo.Text))
+		if editText == nil {
+			b.WriteString(todo.Text)
+		} else {
+			b.WriteString(editText(todo.Text, i))
+		}
+		b.WriteString("\n")
 	}
 
 	return b.String(), nil
 }
 
-func PrettyPrintDataFile(dataFileFullPath string, prefix string) error {
-	result, err := SPrettyPrintDataFile(dataFileFullPath, prefix)
+func PrettyPrintDataFile(dataFileFullPath string, editText func(todo string, index int) string) error {
+	result, err := SPrettyPrintDataFile(dataFileFullPath, editText)
 	if err != nil {
 		return fmt.Errorf("SPrettyPrintDataFile: %w", err)
 	}