onasty: internal/models/note.go (5290b5fe6a6bba7e2328582aae3c4a359ec60e01)

onasty/internal/models/note.go (view raw)

Smirnov Oleksandr feat: notes manipulations for the note authors (#117)..., 1 year ago

package models

import (
	"errors"
	"time"

	"github.com/gofrs/uuid/v5"
)

var (
	ErrNoteContentIsEmpty     = errors.New("note: content is empty")
	ErrNoteSlugIsAlreadyInUse = errors.New("note: slug is already in use")
	ErrNoteExpired            = errors.New("note: expired")
	ErrNoteNotFound           = errors.New("note: not found")
)

type Note struct {
	ID                   uuid.UUID
	Content              string
	Slug                 string
	Password             string
	BurnBeforeExpiration bool
	ReadAt               time.Time
	CreatedAt            time.Time
	ExpiresAt            time.Time
}

func (n Note) Validate() error {
	if n.Content == "" {
		return ErrNoteContentIsEmpty
	}

	if n.IsExpired() {
		return ErrNoteExpired
	}

	return nil
}

func (n Note) IsExpired() bool {
	return !n.ExpiresAt.IsZero() &&
		n.ExpiresAt.Before(time.Now())
}

func (n Note) ShouldBeBurnt() bool {
	return !n.ExpiresAt.IsZero() &&
		n.BurnBeforeExpiration
}

func (n Note) IsRead() bool {
	return !n.ReadAt.IsZero()
}

1	package models
2
3	import (
4	"errors"
5	"time"
6
7	"github.com/gofrs/uuid/v5"
8	)
9
10	var (
11	ErrNoteContentIsEmpty = errors.New("note: content is empty")
12	ErrNoteSlugIsAlreadyInUse = errors.New("note: slug is already in use")
13	ErrNoteExpired = errors.New("note: expired")
14	ErrNoteNotFound = errors.New("note: not found")
15	)
16
17	type Note struct {
18	ID uuid.UUID
19	Content string
20	Slug string
21	Password string
22	BurnBeforeExpiration bool
23	ReadAt time.Time
24	CreatedAt time.Time
25	ExpiresAt time.Time
26	}
27
28	func (n Note) Validate() error {
29	if n.Content == "" {
30	return ErrNoteContentIsEmpty
31	}
32
33	if n.IsExpired() {
34	return ErrNoteExpired
35	}
36
37	return nil
38	}
39
40	func (n Note) IsExpired() bool {
41	return !n.ExpiresAt.IsZero() &&
42	n.ExpiresAt.Before(time.Now())
43	}
44
45	func (n Note) ShouldBeBurnt() bool {
46	return !n.ExpiresAt.IsZero() &&
47	n.BurnBeforeExpiration
48	}
49
50	func (n Note) IsRead() bool {
51	return !n.ReadAt.IsZero()
52	}