donniemarko/internal/service/note.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))
}