onasty: internal/models/notes.go (0afb97fab040e0d661ed9118394a39bcb9d87720)

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

Smirnov Oleksandr feat: add password support to notes (#41)..., 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
	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
}

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	CreatedAt time.Time
24	ExpiresAt time.Time
25	}
26
27	func (n Note) Validate() error {
28	if n.Content == "" {
29	return ErrNoteContentIsEmpty
30	}
31
32	if n.IsExpired() {
33	return ErrNoteExpired
34	}
35
36	return nil
37	}
38
39	func (n Note) IsExpired() bool {
40	return !n.ExpiresAt.IsZero() &&
41	n.ExpiresAt.Before(time.Now())
42	}
43
44	func (n Note) ShouldBeBurnt() bool {
45	return !n.ExpiresAt.IsZero() &&
46	n.BurnBeforeExpiration
47	}