aboutsummaryrefslogtreecommitdiffstats
path: root/models/item/item.go
blob: 882ccc11e68f8662cbf1903273f2e86fdc7ce4f2 (plain) (blame)
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
package item

import (
	"fmt"
	"log"
	"github.com/adammathes/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
		}
		i.Description = p.Sanitize(i.Description)
		// TODO: sanitize other fields
		items = append(items, i)
	}
	if err = rows.Err(); err != nil {
		return nil, err
	}
	return items, nil
}