125 lines
2.3 KiB
Go
125 lines
2.3 KiB
Go
package service
|
|
|
|
import (
|
|
"donniemarko/internal/note"
|
|
"donniemarko/internal/storage"
|
|
"sort"
|
|
"strings"
|
|
)
|
|
|
|
type NotesService struct {
|
|
storage storage.Storage
|
|
}
|
|
|
|
type SortOption func([]*note.Note)
|
|
|
|
type QueryOptions struct {
|
|
SearchTerm string
|
|
SortBy string
|
|
}
|
|
|
|
func NewNoteService() *NotesService {
|
|
return &NotesService{}
|
|
}
|
|
|
|
func (s *NotesService) SetStorage(storage storage.Storage) {
|
|
s.storage = storage
|
|
}
|
|
|
|
func SortByDate(notes []*note.Note) {
|
|
sort.Slice(notes, func(i, j int) bool {
|
|
return notes[i].UpdatedAt.After(notes[j].UpdatedAt)
|
|
})
|
|
}
|
|
|
|
func SortByDateAsc(notes []*note.Note) {
|
|
sort.Slice(notes, func(i, j int) bool {
|
|
return notes[i].UpdatedAt.Before(notes[j].UpdatedAt)
|
|
})
|
|
}
|
|
|
|
func SortByTitle(notes []*note.Note) {
|
|
sort.Slice(notes, func(i, j int) bool {
|
|
return notes[i].Title < notes[j].Title
|
|
})
|
|
}
|
|
|
|
func SortByTitleAsc(notes []*note.Note) {
|
|
sort.Slice(notes, func(i, j int) bool {
|
|
return notes[i].Title > notes[j].Title
|
|
})
|
|
}
|
|
|
|
func (s *NotesService) GetNotesWithSort(sortBy SortOption) ([]*note.Note, error) {
|
|
notes := s.storage.GetAll()
|
|
|
|
if sortBy != nil {
|
|
sortBy(notes)
|
|
}
|
|
|
|
return notes, nil
|
|
}
|
|
|
|
func (s *NotesService) QueryNotes(opts QueryOptions) ([]*note.Note, error) {
|
|
var notes []*note.Note
|
|
|
|
// Search or get all
|
|
if opts.SearchTerm != "" {
|
|
notes = s.storage.Search(opts.SearchTerm)
|
|
} else {
|
|
notes = s.storage.GetAll()
|
|
}
|
|
|
|
// Apply sorting
|
|
switch opts.SortBy {
|
|
case "recent":
|
|
SortByDate(notes)
|
|
case "alpha":
|
|
SortByTitle(notes)
|
|
case "oldest":
|
|
SortByDateAsc(notes)
|
|
default:
|
|
SortByDate(notes) // Default sort
|
|
}
|
|
|
|
return notes, nil
|
|
}
|
|
|
|
func (s *NotesService) GetNoteByHash(hash string) (*note.Note, error) {
|
|
return s.storage.Get(hash)
|
|
}
|
|
|
|
func (s *NotesService) GetNotes() []*note.Note {
|
|
return s.storage.GetAll()
|
|
}
|
|
|
|
func (s *NotesService) AddTag(noteID, tag string) error {
|
|
tag = normalizeTag(tag)
|
|
if tag == "" {
|
|
return nil
|
|
}
|
|
|
|
if _, err := s.storage.Get(noteID); err != nil {
|
|
return err
|
|
}
|
|
|
|
return s.storage.AddTag(noteID, tag)
|
|
}
|
|
|
|
func (s *NotesService) RemoveTag(noteID, tag string) error {
|
|
tag = normalizeTag(tag)
|
|
if tag == "" {
|
|
return nil
|
|
}
|
|
|
|
if _, err := s.storage.Get(noteID); err != nil {
|
|
return err
|
|
}
|
|
|
|
return s.storage.RemoveTag(noteID, tag)
|
|
}
|
|
|
|
func normalizeTag(tag string) string {
|
|
return strings.ToLower(strings.TrimSpace(tag))
|
|
}
|