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
}