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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
package item
import (
"fmt"
"log"
"adammathes.com/neko/models"
"github.com/microcosm-cc/bluemonday"
)
type Item struct {
Id int64 `json:"_id,string,omitempty"`
Title string `json:"title"`
Url string `json:"url"`
Description string `json:"description"`
PublishDate string `json:"publish_date"`
FeedId int64
FeedTitle string `json:"feed_title"`
FeedUrl string `json:"feed_url"`
ReadState bool `json:"read"`
Starred bool `json:"starred"`
}
func (i *Item) Print() {
fmt.Printf("id: %d\n", i.Id)
fmt.Printf("title: %s\n", i.Title)
fmt.Printf("ReadState: %d\n", i.ReadState)
}
func (i *Item) Create() error {
res, err := models.DB.Exec(`INSERT INTO
item(title, url, description, publish_date, feed_id)
VALUES(?, ?, ?, ?, ?)`, i.Title, i.Url, i.Description, i.PublishDate, i.FeedId)
if err != nil {
return err
}
id, _ := res.LastInsertId()
i.Id = id
return nil
}
func (i *Item) Save() {
_, err := models.DB.Exec(`UPDATE item
SET read_state=?, starred=?
WHERE id=?`, i.ReadState, i.Starred, i.Id)
if err != nil {
log.Println(err)
}
}
func (i *Item) FullSave() {
_, err := models.DB.Exec(`UPDATE item
SET title=?, url=?, description=?, feed_id=?
WHERE id=?`, i.Title, i.Url, i.Description, i.FeedId, i.Id)
if err != nil {
log.Println(err)
}
}
func Filter(max_id int64, feed_id int64, unread_only bool, starred_only bool) ([]*Item, error) {
var args []interface{}
query := `SELECT item.id, item.title, item.url, item.description,
item.read_state, item.starred, item.publish_date,
feed.url, feed.title
FROM item,feed
WHERE item.feed_id=feed.id `
if max_id != 0 {
query = query + "AND item.id < ? "
args = append(args, max_id)
}
if feed_id != 0 {
query = query + " AND feed.id=? "
args = append(args, feed_id)
}
if unread_only {
query = query + " AND item.read_state=0 "
}
if starred_only {
query = query + " AND item.starred=1 "
}
query = query + "ORDER BY item.id DESC LIMIT 15"
// log.Println(query)
// log.Println(args...)
rows, err := models.DB.Query(query, args...)
if err != nil {
log.Println(err)
return nil, err
}
defer rows.Close()
p := bluemonday.NewPolicy()
p.AllowElements("blockquote", "a", "img", "p", "h1", "h2", "h3", "h4", "b", "i", "em", "strong")
p.AllowAttrs("href").OnElements("a")
p.AllowAttrs("src", "alt").OnElements("img")
items := make([]*Item, 0)
for rows.Next() {
i := new(Item)
err := rows.Scan(&i.Id, &i.Title, &i.Url, &i.Description, &i.ReadState, &i.Starred, &i.PublishDate, &i.FeedUrl, &i.FeedTitle)
if err != nil {
log.Println(err)
return nil, err
}
// sanitize all fields from external input
// should do this at ingest time, probably, for efficiency
// but still may need to adjust rules
i.Title = p.Sanitize(i.Title)
i.Description = p.Sanitize(i.Description)
i.Url = p.Sanitize(i.Url)
i.FeedTitle = p.Sanitize(i.FeedTitle)
i.FeedUrl = p.Sanitize(i.FeedUrl)
items = append(items, i)
}
if err = rows.Err(); err != nil {
return nil, err
}
return items, nil
}
|
