If the website URL is empty, assign the feed URL

1 files changed, 8 insertions, 0 deletions

diff --git a/reader/feed/handler.go b/reader/feed/handler.go
index 83cbc6f..35ad388 100644
--- a/reader/feed/handler.go
+++ b/reader/feed/handler.go
@@ -75,6 +75,10 @@ func (h *Handler) CreateFeed(userID, categoryID int64, url string, crawler bool)
 	subscription.UserID = userID
 	subscription.Crawler = crawler
 
+	if subscription.SiteURL == "" {
+		subscription.SiteURL = subscription.FeedURL
+	}
+
 	err = h.store.CreateFeed(subscription)
 	if err != nil {
 		return nil, err
@@ -171,6 +175,10 @@ func (h *Handler) RefreshFeed(userID, feedID int64) error {
 	originalFeed.ParsingErrorCount = 0
 	originalFeed.ParsingErrorMsg = ""
 
+	if originalFeed.SiteURL == "" {
+		originalFeed.SiteURL = originalFeed.FeedURL
+	}
+
 	return h.store.UpdateFeed(originalFeed)
 }