92 lines
1.6 KiB
Go
92 lines
1.6 KiB
Go
package internal
|
|
|
|
import (
|
|
"fmt"
|
|
"freed/internal/database"
|
|
"net/url"
|
|
"strconv"
|
|
|
|
"github.com/mmcdole/gofeed"
|
|
"github.com/pterm/pterm"
|
|
)
|
|
|
|
func AddFeed(feedUrl string) (string, int, error) {
|
|
if _, err := url.ParseRequestURI(feedUrl); err != nil {
|
|
return "", 0, fmt.Errorf("The given URL does not seem to be valid: %s", err)
|
|
}
|
|
|
|
feed, err := parseByUrl(feedUrl)
|
|
|
|
if err != nil {
|
|
return "", 0, err
|
|
}
|
|
|
|
f := database.Feed{
|
|
Name: feed.Title,
|
|
Url: feedUrl,
|
|
}
|
|
|
|
feedId, err := f.Insert()
|
|
if err != nil {
|
|
return "", 0, err
|
|
}
|
|
|
|
articles := make([]database.Article, 0, 10)
|
|
|
|
for i, v := range feed.Items {
|
|
if i < 10 {
|
|
a := database.Article{
|
|
Name: v.Title,
|
|
Url: v.Link,
|
|
FeedId: feedId,
|
|
}
|
|
|
|
articles = append(articles, a)
|
|
}
|
|
}
|
|
|
|
if err := database.InsertMultipleArticles(articles); err != nil {
|
|
return feed.Title, 0, nil
|
|
}
|
|
|
|
return feed.Title, len(articles), nil
|
|
}
|
|
|
|
func GetAllFeedsAsTable() (pterm.TableData, error) {
|
|
feeds, err := database.FindAllFeedsWithArticleCount()
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
tableData := pterm.TableData{}
|
|
headerRow := []string{"ID", "Name", "Url", "Item Count", "Created At", "Last Synced At"}
|
|
|
|
tableData = append(tableData, headerRow)
|
|
|
|
for _, feed := range *feeds {
|
|
rowData := []string{
|
|
fmt.Sprintf("[%d]", feed.ID),
|
|
feed.Name,
|
|
feed.Url,
|
|
strconv.Itoa(feed.ArticleCount),
|
|
pterm.Gray(feed.CreatedAt),
|
|
pterm.Gray(feed.LastSyncedAt),
|
|
}
|
|
|
|
tableData = append(tableData, rowData)
|
|
}
|
|
|
|
return tableData, nil
|
|
}
|
|
|
|
func parseByUrl(u string) (*gofeed.Feed, error) {
|
|
fp := gofeed.NewParser()
|
|
|
|
feed, err := fp.ParseURL(u)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return feed, nil
|
|
}
|