127 lines
3.1 KiB
Go
127 lines
3.1 KiB
Go
package feeds
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/spf13/viper"
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/clause"
|
|
"gorm.io/gorm/logger"
|
|
)
|
|
|
|
func (fs *FeedStore) getDB() *gorm.DB {
|
|
if fs.db == nil {
|
|
db, err := gorm.Open(sqlite.Open(viper.GetString("DB_PATH")), &gorm.Config{
|
|
Logger: logger.Default.LogMode(logger.Error),
|
|
})
|
|
if err != nil {
|
|
panic("failed to connect database")
|
|
}
|
|
db.AutoMigrate(&Feed{})
|
|
db.AutoMigrate(&Item{})
|
|
|
|
fs.db = db
|
|
}
|
|
|
|
return fs.db
|
|
}
|
|
|
|
func (fs *FeedStore) GetFeed(id string) *Feed {
|
|
feed := &Feed{}
|
|
fs.getDB().Preload("Items").Where("id = ?", id).First(feed)
|
|
return feed
|
|
}
|
|
|
|
func (fs *FeedStore) DeleteFeed(id string) {
|
|
fs.getDB().Delete(Feed{ID: id})
|
|
fs.getDB().Unscoped().Delete(Item{}, "feed_id = ?", id)
|
|
}
|
|
|
|
func (fs *FeedStore) GetItem(id string) *Item {
|
|
item := &Item{}
|
|
fs.getDB().Where("id = ?", id).First(item)
|
|
return item
|
|
}
|
|
|
|
func (fs *FeedStore) GetFeeds() *[]Feed {
|
|
feeds := &[]Feed{}
|
|
fs.getDB().Order("title asc").Find(feeds)
|
|
return feeds
|
|
}
|
|
|
|
func (fs *FeedStore) GetUnread() *[]ItemWithFeed {
|
|
items := &[]ItemWithFeed{}
|
|
fs.getDB().Table("items").
|
|
Where("read = ?", false).
|
|
Select("items.*, feeds.title as feed_title, feeds.homepage_url as feed_homepage_url").
|
|
Order("items.created desc, items.title").
|
|
Joins("left join feeds on feeds.id = items.feed_id").
|
|
Find(items)
|
|
|
|
return items
|
|
}
|
|
|
|
func (fs *FeedStore) GetSaved() *[]ItemWithFeed {
|
|
items := &[]ItemWithFeed{}
|
|
fs.getDB().Table("items").
|
|
Where("save = ?", true).
|
|
Select("items.*, feeds.title as feed_title, feeds.homepage_url as feed_homepage_url").
|
|
Order("items.created desc, items.title").
|
|
Joins("left join feeds on feeds.id = items.feed_id").
|
|
Find(items)
|
|
|
|
return items
|
|
}
|
|
|
|
func (fs *FeedStore) DeleteOldReadItems() {
|
|
t := time.Now()
|
|
threshold := t.Add(-time.Hour * 24 * 7)
|
|
fs.getDB().Table("items").
|
|
Where("save = ? and read = ? and created < ?", false, true, threshold).
|
|
Delete(Item{})
|
|
}
|
|
|
|
func (fs *FeedStore) GetAll() *[]ItemWithFeed {
|
|
items := &[]ItemWithFeed{}
|
|
fs.getDB().Table("items").
|
|
Select("items.*, feeds.title as feed_title, feeds.homepage_url as feed_homepage_url").
|
|
Order("items.created desc, items.title").
|
|
Joins("left join feeds on feeds.id = items.feed_id").
|
|
Find(items)
|
|
|
|
return items
|
|
}
|
|
|
|
func (fs *FeedStore) SaveFeed(feed Feed) {
|
|
fs.getDB().Omit("Items").Clauses(clause.OnConflict{
|
|
Columns: []clause.Column{{Name: "id"}},
|
|
DoUpdates: clause.AssignmentColumns([]string{"title", "description", "homepage_url", "image_url", "last_updated"}),
|
|
}).Create(feed)
|
|
|
|
for _, item := range feed.Items {
|
|
fs.getDB().Clauses(clause.OnConflict{
|
|
Columns: []clause.Column{{Name: "id"}},
|
|
DoUpdates: clause.AssignmentColumns([]string{"title", "description", "content", "url", "image_url", "last_updated"}),
|
|
}).Create(item)
|
|
}
|
|
}
|
|
|
|
func (fs *FeedStore) MarkAsRead(itemID string) {
|
|
item := &Item{}
|
|
fs.getDB().Where("id = ?", itemID).First(item)
|
|
|
|
item.Read = true
|
|
|
|
fs.getDB().Save(*item)
|
|
}
|
|
|
|
func (fs *FeedStore) ToggleSaved(itemID string) {
|
|
item := &Item{}
|
|
fs.getDB().Where("id = ?", itemID).First(item)
|
|
|
|
item.Save = !item.Save
|
|
|
|
fs.getDB().Save(*item)
|
|
}
|