move packages to src

src/storage/feed.go

@@ -0,0 +1,172 @@
+package storage
+
+import (
+	"html"
+	"net/url"
+)
+
+type Feed struct {
+	Id          int64   `json:"id"`
+	FolderId    *int64  `json:"folder_id"`
+	Title       string  `json:"title"`
+	Description string  `json:"description"`
+	Link        string  `json:"link"`
+	FeedLink    string  `json:"feed_link"`
+	Icon        *[]byte `json:"icon,omitempty"`
+	HasIcon     bool    `json:"has_icon"`
+}
+
+func (s *Storage) CreateFeed(title, description, link, feedLink string, folderId *int64) *Feed {
+	title = html.UnescapeString(title)
+	// WILD: fallback to `feed.link` -> `feed.feed_link` -> "<???>" if title is missing
+	if title == "" {
+		title = link
+		// use domain if possible
+		linkUrl, err := url.Parse(link)
+		if err == nil && linkUrl.Host != "" && len(linkUrl.Path) <= 1 {
+			title = linkUrl.Host
+		}
+	}
+	if title == "" {
+		title = feedLink
+	}
+	if title == "" {
+		title = "<???>"
+	}
+	result, err := s.db.Exec(`
+		insert into feeds (title, description, link, feed_link, folder_id) 
+		values (?, ?, ?, ?, ?)
+		on conflict (feed_link) do update set folder_id=?`,
+		title, description, link, feedLink, folderId,
+		folderId,
+	)
+	if err != nil {
+		return nil
+	}
+	id, idErr := result.LastInsertId()
+	if idErr != nil {
+		return nil
+	}
+	return &Feed{
+		Id:          id,
+		Title:       title,
+		Description: description,
+		Link:        link,
+		FeedLink:    feedLink,
+		FolderId:    folderId,
+	}
+}
+
+func (s *Storage) DeleteFeed(feedId int64) bool {
+	_, err1 := s.db.Exec(`delete from items where feed_id = ?`, feedId)
+	_, err2 := s.db.Exec(`delete from feeds where id = ?`, feedId)
+	return err1 == nil && err2 == nil
+}
+
+func (s *Storage) RenameFeed(feedId int64, newTitle string) bool {
+	_, err := s.db.Exec(`update feeds set title = ? where id = ?`, newTitle, feedId)
+	return err == nil
+}
+
+func (s *Storage) UpdateFeedFolder(feedId int64, newFolderId *int64) bool {
+	_, err := s.db.Exec(`update feeds set folder_id = ? where id = ?`, newFolderId, feedId)
+	return err == nil
+}
+
+func (s *Storage) UpdateFeedIcon(feedId int64, icon *[]byte) bool {
+	_, err := s.db.Exec(`update feeds set icon = ? where id = ?`, icon, feedId)
+	return err == nil
+}
+
+func (s *Storage) ListFeeds() []Feed {
+	result := make([]Feed, 0, 0)
+	rows, err := s.db.Query(`
+		select id, folder_id, title, description, link, feed_link,
+		       ifnull(icon, '') != '' as has_icon
+		from feeds
+		order by title collate nocase
+	`)
+	if err != nil {
+		s.log.Print(err)
+		return result
+	}
+	for rows.Next() {
+		var f Feed
+		err = rows.Scan(
+			&f.Id,
+			&f.FolderId,
+			&f.Title,
+			&f.Description,
+			&f.Link,
+			&f.FeedLink,
+			&f.HasIcon,
+		)
+		if err != nil {
+			s.log.Print(err)
+			return result
+		}
+		result = append(result, f)
+	}
+	return result
+}
+
+func (s *Storage) GetFeed(id int64) *Feed {
+	row := s.db.QueryRow(`
+		select id, folder_id, title, description, link, feed_link, icon,
+		       ifnull(icon, '') != '' as has_icon
+		from feeds where id = ?
+	`, id)
+	if row != nil {
+		var f Feed
+		row.Scan(
+			&f.Id,
+			&f.FolderId,
+			&f.Title,
+			&f.Description,
+			&f.Link,
+			&f.FeedLink,
+			&f.Icon,
+			&f.HasIcon,
+		)
+		return &f
+	}
+	return nil
+}
+
+func (s *Storage) ResetFeedErrors() {
+	if _, err := s.db.Exec(`delete from feed_errors`); err != nil {
+		s.log.Print(err)
+	}
+}
+
+func (s *Storage) SetFeedError(feedID int64, lastError error) {
+	_, err := s.db.Exec(`
+		insert into feed_errors (feed_id, error)
+		values (?, ?)
+		on conflict (feed_id) do update set error = excluded.error`,
+		feedID, lastError.Error(),
+	)
+	if err != nil {
+		s.log.Print(err)
+	}
+}
+
+func (s *Storage) GetFeedErrors() map[int64]string {
+	errors := make(map[int64]string)
+
+	rows, err := s.db.Query(`select feed_id, error from feed_errors`)
+	if err != nil {
+		s.log.Print(err)
+		return errors
+	}
+
+	for rows.Next() {
+		var id int64
+		var error string
+		if err = rows.Scan(&id, &error); err != nil {
+			s.log.Print(err)
+		}
+		errors[id] = error
+	}
+	return errors
+}