aboutsummaryrefslogtreecommitdiff
path: root/internal/feed/feed.go
blob: 9ed44dfc666758a5c59845b6e556931691ac5260 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
package feed

import (
	"time"

	"github.com/mmcdole/gofeed"

	"github.com/Necoro/feed2imap-go/pkg/config"
	"github.com/Necoro/feed2imap-go/pkg/log"
	"github.com/Necoro/feed2imap-go/pkg/util"
)

type Feed struct {
	*config.Feed
	feed   *gofeed.Feed
	items  []feeditem
	cached CachedFeed
	Global config.GlobalOptions
}

type feedDescriptor struct {
	Name string
	Url  string
}

type feedImage struct {
	image []byte
	mime  string
}

type feeditem struct {
	*gofeed.Feed
	*gofeed.Item
	Body       string
	updateOnly bool
	reasons    []string
	images     []feedImage
	itemId     string
}

// Creator returns the name of the creating author.
// MUST NOT have `*feeditem` has the receiver, because the template breaks then.
func (item feeditem) Creator() string {
	if item.Item.Author != nil {
		return item.Item.Author.Name
	}
	return ""
}

func (item *feeditem) addReason(reason string) {
	if !util.StrContains(item.reasons, reason) {
		item.reasons = append(item.reasons, reason)
	}
}

func (item *feeditem) addImage(img []byte, mime string) int {
	i := feedImage{img, mime}
	item.images = append(item.images, i)
	return len(item.images)
}

func (item *feeditem) clearImages() {
	item.images = []feedImage{}
}

func (feed *Feed) descriptor() feedDescriptor {
	return feedDescriptor{
		Name: feed.Name,
		Url:  feed.Url,
	}
}

func (feed *Feed) NeedsUpdate(updateTime time.Time) bool {
	if feed.MinFreq == 0 { // shortcut
		return true
	}
	if !updateTime.IsZero() && int(time.Since(updateTime).Hours()) < feed.MinFreq {
		log.Printf("Feed '%s' does not need updating, skipping.", feed.Name)
		return false
	}
	return true
}

func (feed *Feed) FetchSuccessful() bool {
	return feed.feed != nil
}

func (feed *Feed) MarkSuccess() {
	if feed.cached != nil {
		feed.cached.Commit()
	}
}