feat: add import list & calendar

pkg/importlist/plexwatchlist/plex.go

@@ -0,0 +1,96 @@
+package plexwatchlist
+
+import (
+	"encoding/xml"
+	"io"
+	"net/http"
+	"polaris/pkg/importlist"
+	"strings"
+
+	"github.com/pkg/errors"
+)
+
+type Response struct {
+	XMLName xml.Name `xml:"rss"`
+	Text    string   `xml:",chardata"`
+	Atom    string   `xml:"atom,attr"`
+	Media   string   `xml:"media,attr"`
+	Version string   `xml:"version,attr"`
+	Channel struct {
+		Text  string `xml:",chardata"`
+		Title string `xml:"title"`
+		Link  struct {
+			Text string `xml:",chardata"`
+			Href string `xml:"href,attr"`
+			Rel  string `xml:"rel,attr"`
+			Type string `xml:"type,attr"`
+		} `xml:"link"`
+		Description string `xml:"description"`
+		Category    string `xml:"category"`
+		Item        []struct {
+			Text        string `xml:",chardata"`
+			Title       string `xml:"title"`
+			PubDate     string `xml:"pubDate"`
+			Link        string `xml:"link"`
+			Description string `xml:"description"`
+			Category    string `xml:"category"`
+			Credit      []struct {
+				Text string `xml:",chardata"`
+				Role string `xml:"role,attr"`
+			} `xml:"credit"`
+			Thumbnail struct {
+				Text string `xml:",chardata"`
+				URL  string `xml:"url,attr"`
+			} `xml:"thumbnail"`
+			Keywords string `xml:"keywords"`
+			Rating   struct {
+				Text   string `xml:",chardata"`
+				Scheme string `xml:"scheme,attr"`
+			} `xml:"rating"`
+			Guid struct {
+				Text        string `xml:",chardata"`
+				IsPermaLink string `xml:"isPermaLink,attr"`
+			} `xml:"guid"`
+		} `xml:"item"`
+	} `xml:"channel"`
+}
+
+func (r *Response) convert() *importlist.Response {
+	res := &importlist.Response{}
+	for _, im := range r.Channel.Item {
+		item := importlist.Item{
+			Title: im.Title,
+		}
+		id := strings.ToLower(im.Guid.Text)
+		if strings.HasPrefix(id, "tvdb") {
+			tvdbid := strings.TrimPrefix(id, "tvdb://")
+			item.TvdbID = tvdbid
+		} else if strings.HasPrefix(id, "imdb") {
+			imdbid := strings.TrimPrefix(id, "imdb://")
+			item.ImdbID = imdbid
+		} else if strings.HasPrefix(id, "tmdb") {
+			tmdbid := strings.TrimPrefix(id, "tmdb://")
+			item.TmdbID = tmdbid
+		}
+		res.Items = append(res.Items, item)
+	}
+	return res
+}
+
+func ParsePlexWatchlist(url string) (*importlist.Response, error) {
+	resp, err := http.Get(url)
+	if err != nil {
+		return nil, errors.Wrap(err, "http get")
+	}
+	defer resp.Body.Close()
+	data, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return nil, errors.Wrap(err, "read data")
+	}
+	var rrr Response
+	err = xml.Unmarshal(data, &rrr)
+	if err != nil {
+		return nil, errors.Wrap(err, "xml")
+	}
+	return rrr.convert(), nil
+}