1
0
Fork 0
miniflux/reader/opml/handler.go

100 lines
2.3 KiB
Go
Raw Normal View History

2017-11-20 00:10:04 -05:00
// Copyright 2017 Frédéric Guillot. All rights reserved.
// Use of this source code is governed by the Apache 2.0
// license that can be found in the LICENSE file.
2018-08-25 00:51:50 -04:00
package opml // import "miniflux.app/reader/opml"
2017-11-20 00:10:04 -05:00
import (
"errors"
"fmt"
"io"
2018-08-25 00:51:50 -04:00
"miniflux.app/logger"
"miniflux.app/model"
"miniflux.app/storage"
2017-11-20 00:10:04 -05:00
)
// Handler handles the logic for OPML import/export.
type Handler struct {
2017-11-20 00:10:04 -05:00
store *storage.Storage
}
// Export exports user feeds to OPML.
func (h *Handler) Export(userID int64) (string, error) {
2017-11-28 00:30:04 -05:00
feeds, err := h.store.Feeds(userID)
2017-11-20 00:10:04 -05:00
if err != nil {
2018-04-29 21:56:40 -04:00
return "", err
2017-11-20 00:10:04 -05:00
}
var subscriptions SubcriptionList
for _, feed := range feeds {
subscriptions = append(subscriptions, &Subcription{
Title: feed.Title,
FeedURL: feed.FeedURL,
SiteURL: feed.SiteURL,
CategoryName: feed.Category.Title,
})
}
return Serialize(subscriptions), nil
}
// Import parses and create feeds from an OPML import.
2018-03-14 21:50:06 -04:00
func (h *Handler) Import(userID int64, data io.Reader) error {
2017-11-20 00:10:04 -05:00
subscriptions, err := Parse(data)
if err != nil {
return err
}
for _, subscription := range subscriptions {
if !h.store.FeedURLExists(userID, subscription.FeedURL) {
2017-11-20 00:10:04 -05:00
var category *model.Category
2018-03-14 21:50:06 -04:00
var err error
2017-11-20 00:10:04 -05:00
if subscription.CategoryName == "" {
2017-11-28 00:30:04 -05:00
category, err = h.store.FirstCategory(userID)
2017-11-20 00:10:04 -05:00
if err != nil {
2017-12-15 21:55:57 -05:00
logger.Error("[OPML:Import] %v", err)
return errors.New("unable to find first category")
2017-11-20 00:10:04 -05:00
}
} else {
2017-11-28 00:30:04 -05:00
category, err = h.store.CategoryByTitle(userID, subscription.CategoryName)
2017-11-20 00:10:04 -05:00
if err != nil {
2017-12-15 21:55:57 -05:00
logger.Error("[OPML:Import] %v", err)
return errors.New("unable to search category by title")
2017-11-20 00:10:04 -05:00
}
if category == nil {
category = &model.Category{
UserID: userID,
Title: subscription.CategoryName,
}
err := h.store.CreateCategory(category)
2017-11-20 00:10:04 -05:00
if err != nil {
2017-12-15 21:55:57 -05:00
logger.Error("[OPML:Import] %v", err)
2018-04-29 21:56:40 -04:00
return fmt.Errorf(`unable to create this category: %q`, subscription.CategoryName)
2017-11-20 00:10:04 -05:00
}
}
}
feed := &model.Feed{
UserID: userID,
Title: subscription.Title,
FeedURL: subscription.FeedURL,
SiteURL: subscription.SiteURL,
Category: category,
}
h.store.CreateFeed(feed)
2017-11-20 00:10:04 -05:00
}
}
return nil
}
// NewHandler creates a new handler for OPML files.
func NewHandler(store *storage.Storage) *Handler {
return &Handler{store: store}
2017-11-20 00:10:04 -05:00
}