opml_upload.go 2.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109
  1. // SPDX-FileCopyrightText: Copyright The Miniflux Authors. All rights reserved.
  2. // SPDX-License-Identifier: Apache-2.0
  3. package ui // import "miniflux.app/v2/internal/ui"
  4. import (
  5. "log/slog"
  6. "net/http"
  7. "miniflux.app/v2/internal/config"
  8. "miniflux.app/v2/internal/http/client"
  9. "miniflux.app/v2/internal/http/request"
  10. "miniflux.app/v2/internal/http/response/html"
  11. "miniflux.app/v2/internal/http/route"
  12. "miniflux.app/v2/internal/reader/opml"
  13. "miniflux.app/v2/internal/ui/session"
  14. "miniflux.app/v2/internal/ui/view"
  15. )
  16. func (h *handler) uploadOPML(w http.ResponseWriter, r *http.Request) {
  17. loggedUserID := request.UserID(r)
  18. user, err := h.store.UserByID(loggedUserID)
  19. if err != nil {
  20. html.ServerError(w, r, err)
  21. return
  22. }
  23. file, fileHeader, err := r.FormFile("file")
  24. if err != nil {
  25. slog.Error("OPML file upload error",
  26. slog.Int64("user_id", loggedUserID),
  27. slog.Any("error", err),
  28. )
  29. html.Redirect(w, r, route.Path(h.router, "import"))
  30. return
  31. }
  32. defer file.Close()
  33. slog.Info("OPML file uploaded",
  34. slog.Int64("user_id", loggedUserID),
  35. slog.String("file_name", fileHeader.Filename),
  36. slog.Int64("file_size", fileHeader.Size),
  37. )
  38. sess := session.New(h.store, request.SessionID(r))
  39. view := view.New(h.tpl, r, sess)
  40. view.Set("menu", "feeds")
  41. view.Set("user", user)
  42. view.Set("countUnread", h.store.CountUnreadEntries(user.ID))
  43. view.Set("countErrorFeeds", h.store.CountUserFeedsWithErrors(user.ID))
  44. if fileHeader.Size == 0 {
  45. view.Set("errorMessage", "error.empty_file")
  46. html.OK(w, r, view.Render("import"))
  47. return
  48. }
  49. if impErr := opml.NewHandler(h.store).Import(user.ID, file); impErr != nil {
  50. view.Set("errorMessage", impErr)
  51. html.OK(w, r, view.Render("import"))
  52. return
  53. }
  54. html.Redirect(w, r, route.Path(h.router, "feeds"))
  55. }
  56. func (h *handler) fetchOPML(w http.ResponseWriter, r *http.Request) {
  57. loggedUserID := request.UserID(r)
  58. user, err := h.store.UserByID(loggedUserID)
  59. if err != nil {
  60. html.ServerError(w, r, err)
  61. return
  62. }
  63. url := r.FormValue("url")
  64. if url == "" {
  65. html.Redirect(w, r, route.Path(h.router, "import"))
  66. return
  67. }
  68. slog.Info("Fetching OPML file remotely",
  69. slog.Int64("user_id", loggedUserID),
  70. slog.String("opml_file_url", url),
  71. )
  72. sess := session.New(h.store, request.SessionID(r))
  73. view := view.New(h.tpl, r, sess)
  74. view.Set("menu", "feeds")
  75. view.Set("user", user)
  76. view.Set("countUnread", h.store.CountUnreadEntries(user.ID))
  77. view.Set("countErrorFeeds", h.store.CountUserFeedsWithErrors(user.ID))
  78. clt := client.NewClientWithConfig(url, config.Opts)
  79. resp, err := clt.Get()
  80. if err != nil {
  81. view.Set("errorMessage", err)
  82. html.OK(w, r, view.Render("import"))
  83. return
  84. }
  85. if impErr := opml.NewHandler(h.store).Import(user.ID, resp.Body); impErr != nil {
  86. view.Set("errorMessage", impErr)
  87. html.OK(w, r, view.Render("import"))
  88. return
  89. }
  90. html.Redirect(w, r, route.Path(h.router, "feeds"))
  91. }