diff --git a/internal/repository/notes.go b/internal/repository/notes.go index a73a36a..12ce97b 100644 --- a/internal/repository/notes.go +++ b/internal/repository/notes.go @@ -20,7 +20,7 @@ func (r *NoteRepository) CreateNote(content string) (models.Note, error) { return models.Note{}, fmt.Errorf("content cannot be empty") } note := models.Note{ - Content: content, + Content: content, LastUpdate: time.Now(), } return r.store.SaveNote(note) @@ -35,5 +35,14 @@ func (r *NoteRepository) ListNotes() ([]models.Note, error) { } func (r *NoteRepository) UpdateNote(id int, content string) (models.Note, error) { - return models.Note{}, fmt.Errorf("not implemented") + if content == "" { + return models.Note{}, fmt.Errorf("content cannot be empty") + } + note, err := r.store.GetNoteByID(id) + if err != nil { + return models.Note{}, err + } + note.Content = content + note.LastUpdate = time.Now() + return r.store.SaveNote(note) }