123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199 |
- package v1
- import (
- "context"
- "encoding/json"
- "net/http"
- "strconv"
- "strings"
- "time"
- "github.com/gorilla/feeds"
- "github.com/labstack/echo/v4"
- "github.com/usememos/memos/internal/util"
- "github.com/usememos/memos/plugin/gomark/ast"
- "github.com/usememos/memos/plugin/gomark/parser"
- "github.com/usememos/memos/plugin/gomark/parser/tokenizer"
- "github.com/usememos/memos/plugin/gomark/renderer"
- "github.com/usememos/memos/store"
- )
- const (
- maxRSSItemCount = 100
- maxRSSItemTitleLength = 128
- )
- func (s *APIV1Service) registerRSSRoutes(g *echo.Group) {
- g.GET("/explore/rss.xml", s.GetExploreRSS)
- g.GET("/u/:id/rss.xml", s.GetUserRSS)
- }
- // GetExploreRSS godoc
- //
- // @Summary Get RSS
- // @Tags rss
- // @Produce xml
- // @Success 200 {object} nil "RSS"
- // @Failure 500 {object} nil "Failed to get system customized profile | Failed to find memo list | Failed to generate rss"
- // @Router /explore/rss.xml [GET]
- func (s *APIV1Service) GetExploreRSS(c echo.Context) error {
- ctx := c.Request().Context()
- systemCustomizedProfile, err := s.getSystemCustomizedProfile(ctx)
- if err != nil {
- return echo.NewHTTPError(http.StatusInternalServerError, "Failed to get system customized profile").SetInternal(err)
- }
- normalStatus := store.Normal
- memoFind := store.FindMemo{
- RowStatus: &normalStatus,
- VisibilityList: []store.Visibility{store.Public},
- }
- memoList, err := s.Store.ListMemos(ctx, &memoFind)
- if err != nil {
- return echo.NewHTTPError(http.StatusInternalServerError, "Failed to find memo list").SetInternal(err)
- }
- baseURL := c.Scheme() + "://" + c.Request().Host
- rss, err := s.generateRSSFromMemoList(ctx, memoList, baseURL, systemCustomizedProfile)
- if err != nil {
- return echo.NewHTTPError(http.StatusInternalServerError, "Failed to generate rss").SetInternal(err)
- }
- c.Response().Header().Set(echo.HeaderContentType, echo.MIMEApplicationXMLCharsetUTF8)
- return c.String(http.StatusOK, rss)
- }
- // GetUserRSS godoc
- //
- // @Summary Get RSS for a user
- // @Tags rss
- // @Produce xml
- // @Param id path int true "User ID"
- // @Success 200 {object} nil "RSS"
- // @Failure 400 {object} nil "User id is not a number"
- // @Failure 500 {object} nil "Failed to get system customized profile | Failed to find memo list | Failed to generate rss"
- // @Router /u/{id}/rss.xml [GET]
- func (s *APIV1Service) GetUserRSS(c echo.Context) error {
- ctx := c.Request().Context()
- id, err := util.ConvertStringToInt32(c.Param("id"))
- if err != nil {
- return echo.NewHTTPError(http.StatusBadRequest, "User id is not a number").SetInternal(err)
- }
- systemCustomizedProfile, err := s.getSystemCustomizedProfile(ctx)
- if err != nil {
- return echo.NewHTTPError(http.StatusInternalServerError, "Failed to get system customized profile").SetInternal(err)
- }
- normalStatus := store.Normal
- memoFind := store.FindMemo{
- CreatorID: &id,
- RowStatus: &normalStatus,
- VisibilityList: []store.Visibility{store.Public},
- }
- memoList, err := s.Store.ListMemos(ctx, &memoFind)
- if err != nil {
- return echo.NewHTTPError(http.StatusInternalServerError, "Failed to find memo list").SetInternal(err)
- }
- baseURL := c.Scheme() + "://" + c.Request().Host
- rss, err := s.generateRSSFromMemoList(ctx, memoList, baseURL, systemCustomizedProfile)
- if err != nil {
- return echo.NewHTTPError(http.StatusInternalServerError, "Failed to generate rss").SetInternal(err)
- }
- c.Response().Header().Set(echo.HeaderContentType, echo.MIMEApplicationXMLCharsetUTF8)
- return c.String(http.StatusOK, rss)
- }
- func (s *APIV1Service) generateRSSFromMemoList(ctx context.Context, memoList []*store.Memo, baseURL string, profile *CustomizedProfile) (string, error) {
- feed := &feeds.Feed{
- Title: profile.Name,
- Link: &feeds.Link{Href: baseURL},
- Description: profile.Description,
- Created: time.Now(),
- }
- var itemCountLimit = util.Min(len(memoList), maxRSSItemCount)
- feed.Items = make([]*feeds.Item, itemCountLimit)
- for i := 0; i < itemCountLimit; i++ {
- memoMessage, err := s.convertMemoFromStore(ctx, memoList[i])
- if err != nil {
- return "", err
- }
- description, err := getRSSItemDescription(memoMessage.Content)
- if err != nil {
- return "", err
- }
- feed.Items[i] = &feeds.Item{
- Title: getRSSItemTitle(memoMessage.Content),
- Link: &feeds.Link{Href: baseURL + "/m/" + memoMessage.Name},
- Description: description,
- Created: time.Unix(memoMessage.CreatedTs, 0),
- }
- if len(memoMessage.ResourceList) > 0 {
- resource := memoMessage.ResourceList[0]
- enclosure := feeds.Enclosure{}
- if resource.ExternalLink != "" {
- enclosure.Url = resource.ExternalLink
- } else {
- enclosure.Url = baseURL + "/o/r/" + resource.Name
- }
- enclosure.Length = strconv.Itoa(int(resource.Size))
- enclosure.Type = resource.Type
- feed.Items[i].Enclosure = &enclosure
- }
- }
- rss, err := feed.ToRss()
- if err != nil {
- return "", err
- }
- return rss, nil
- }
- func (s *APIV1Service) getSystemCustomizedProfile(ctx context.Context) (*CustomizedProfile, error) {
- systemSetting, err := s.Store.GetSystemSetting(ctx, &store.FindSystemSetting{
- Name: SystemSettingCustomizedProfileName.String(),
- })
- if err != nil {
- return nil, err
- }
- customizedProfile := &CustomizedProfile{
- Name: "Memos",
- Locale: "en",
- Appearance: "system",
- }
- if systemSetting != nil {
- if err := json.Unmarshal([]byte(systemSetting.Value), customizedProfile); err != nil {
- return nil, err
- }
- }
- return customizedProfile, nil
- }
- func getRSSItemTitle(content string) string {
- tokens := tokenizer.Tokenize(content)
- nodes, _ := parser.Parse(tokens)
- if len(nodes) > 0 {
- firstNode := nodes[0]
- title := renderer.NewStringRenderer().Render([]ast.Node{firstNode})
- return title
- }
- title := strings.Split(content, "\n")[0]
- var titleLengthLimit = util.Min(len(title), maxRSSItemTitleLength)
- if titleLengthLimit < len(title) {
- title = title[:titleLengthLimit] + "..."
- }
- return title
- }
- func getRSSItemDescription(content string) (string, error) {
- tokens := tokenizer.Tokenize(content)
- nodes, err := parser.Parse(tokens)
- if err != nil {
- return "", err
- }
- result := renderer.NewHTMLRenderer().Render(nodes)
- return result, nil
- }
|