mirror of
https://github.com/R0m1k3/FlowReader.git
synced 2026-10-18 12:48:07 +02:00
fix(backend): harden security and speed up feeds and article API
Security: - WebSocket events are routed to their owner only (no cross-user leak); hub close is idempotent (fixes double-close panic), adds ping/pong and write deadlines. - Session tokens stored as SHA-256 (migration 008 keeps sessions valid); single-query auth middleware puts the user in the request context. - Client IP only trusts X-Forwarded-For from TRUSTED_PROXIES; rate limiter map is bounded; per-user limit on AI summaries. - Argon2id at OWASP minimum with a concurrency cap; constant-time login for unknown emails; atomic first-admin bootstrap; REGISTRATION_ENABLED. - CSP/HSTS/COOP headers, same-origin guard on mutations, body size limits, wider SSRF denylist, bounded feed/page/AI response reads, generic errors. - Upgrade chi, pgx, x/net, x/text, x/crypto (known CVEs); commit go.sum. Performance: - List endpoints return a plain-text excerpt and reading time instead of full HTML; content is sanitized once at ingest (legacy rows backfilled). - Keyset pagination on (sort_at, id) with matching partial indexes; redundant indexes dropped (migration 007). - Fetcher: bounded worker pool, conditional GET (ETag/Last-Modified), exponential backoff, dedupe before insert, column-safe truncation, retention-aware ingest, per-user refresh coalescing. - Read/favorite/read-all are single ownership-scoped statements. - gzip compression, immutable caching for hashed assets, path-safe SPA handler, server timeouts; expired sessions purged. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
This commit is contained in:
1 parent
d037e2be34
commit
03e57e4308
40 files changed
+2231
-1898
No files matched your search
+193
-377
@@ -1,9 +1,10 @@
|
||||
package handler
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net/http"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/go-chi/chi/v5"
|
||||
"github.com/google/uuid"
|
||||
@@ -13,11 +14,10 @@ import (
|
||||
"github.com/michael/flowreader/internal/ws"
|
||||
)
|
||||
|
||||
// ArticleHandler handles article-related HTTP requests.
|
||||
// ArticleHandler handles article-related HTTP requests. All routes run
|
||||
// behind RequireAuth; ownership is enforced inside each SQL statement.
|
||||
type ArticleHandler struct {
|
||||
articleRepo domain.ArticleRepository
|
||||
feedService *service.FeedService
|
||||
authService *service.AuthService
|
||||
aiService *service.AIService
|
||||
sanitizer *utils.ContentSanitizer
|
||||
extractor *utils.ContentExtractor
|
||||
@@ -25,11 +25,9 @@ type ArticleHandler struct {
|
||||
}
|
||||
|
||||
// NewArticleHandler creates a new article handler.
|
||||
func NewArticleHandler(articleRepo domain.ArticleRepository, feedService *service.FeedService, authService *service.AuthService, aiService *service.AIService, hub *ws.Hub) *ArticleHandler {
|
||||
func NewArticleHandler(articleRepo domain.ArticleRepository, aiService *service.AIService, hub *ws.Hub) *ArticleHandler {
|
||||
return &ArticleHandler{
|
||||
articleRepo: articleRepo,
|
||||
feedService: feedService,
|
||||
authService: authService,
|
||||
aiService: aiService,
|
||||
sanitizer: utils.NewContentSanitizer(),
|
||||
extractor: utils.NewContentExtractor(),
|
||||
@@ -37,409 +35,232 @@ func NewArticleHandler(articleRepo domain.ArticleRepository, feedService *servic
|
||||
}
|
||||
}
|
||||
|
||||
// sanitizeArticle cleans the user-facing HTML fields of a single article to
|
||||
// prevent stored XSS from malicious feeds. AISummary is rendered as plain text
|
||||
// by the client, so only Content and Summary need sanitization.
|
||||
func (h *ArticleHandler) sanitizeArticle(a *domain.Article) {
|
||||
if a == nil {
|
||||
// notify pushes an event to the acting user's other tabs/devices only.
|
||||
func (h *ArticleHandler) notify(userID uuid.UUID, eventType string, payload any) {
|
||||
if h.hub != nil {
|
||||
h.hub.SendToUser(userID, eventType, payload)
|
||||
}
|
||||
}
|
||||
|
||||
func parseLimit(r *http.Request) int {
|
||||
limit, _ := strconv.Atoi(r.URL.Query().Get("limit"))
|
||||
if limit <= 0 || limit > 100 {
|
||||
limit = 30
|
||||
}
|
||||
return limit
|
||||
}
|
||||
|
||||
// parseCursor reads ?cursor=<RFC3339 sort_at>,<uuid> (the last item of the
|
||||
// previous page). A missing cursor means the first page.
|
||||
func parseCursor(r *http.Request) (*domain.ArticleCursor, bool) {
|
||||
raw := r.URL.Query().Get("cursor")
|
||||
if raw == "" {
|
||||
return nil, true
|
||||
}
|
||||
ts, id, ok := strings.Cut(raw, ",")
|
||||
if !ok {
|
||||
return nil, false
|
||||
}
|
||||
sortAt, err := time.Parse(time.RFC3339Nano, ts)
|
||||
if err != nil {
|
||||
return nil, false
|
||||
}
|
||||
uid, err := uuid.Parse(id)
|
||||
if err != nil {
|
||||
return nil, false
|
||||
}
|
||||
return &domain.ArticleCursor{SortAt: sortAt, ID: uid}, true
|
||||
}
|
||||
|
||||
func (h *ArticleHandler) list(w http.ResponseWriter, r *http.Request, f domain.ArticleFilter) {
|
||||
cursor, ok := parseCursor(r)
|
||||
if !ok {
|
||||
respondError(w, http.StatusBadRequest, "Invalid cursor")
|
||||
return
|
||||
}
|
||||
if a.Content != "" {
|
||||
a.Content = h.sanitizer.Sanitize(a.Content)
|
||||
}
|
||||
if a.Summary != "" {
|
||||
a.Summary = h.sanitizer.Sanitize(a.Summary)
|
||||
}
|
||||
}
|
||||
f.UserID = currentUser(r).ID
|
||||
f.Cursor = cursor
|
||||
f.Limit = parseLimit(r)
|
||||
f.UnreadOnly = f.UnreadOnly || r.URL.Query().Get("unread") == "true"
|
||||
|
||||
// sanitizeArticles cleans a slice of articles in place.
|
||||
func (h *ArticleHandler) sanitizeArticles(articles []*domain.Article) {
|
||||
for _, a := range articles {
|
||||
h.sanitizeArticle(a)
|
||||
}
|
||||
}
|
||||
|
||||
// getUserFromRequest extracts the authenticated user from the request.
|
||||
func (h *ArticleHandler) getUserFromRequest(r *http.Request) (uuid.UUID, error) {
|
||||
cookie, err := r.Cookie("session_id")
|
||||
articles, err := h.articleRepo.List(r.Context(), f)
|
||||
if err != nil {
|
||||
return uuid.Nil, err
|
||||
respondError(w, http.StatusInternalServerError, "Failed to get articles")
|
||||
return
|
||||
}
|
||||
|
||||
user, err := h.authService.GetUserByToken(cookie.Value)
|
||||
if err != nil || user == nil {
|
||||
return uuid.Nil, err
|
||||
}
|
||||
|
||||
return user.ID, nil
|
||||
respondJSON(w, http.StatusOK, articles)
|
||||
}
|
||||
|
||||
// List handles GET /api/v1/articles
|
||||
func (h *ArticleHandler) List(w http.ResponseWriter, r *http.Request) {
|
||||
userID, err := h.getUserFromRequest(r)
|
||||
if err != nil {
|
||||
respondError(w, http.StatusUnauthorized, "Not authenticated")
|
||||
return
|
||||
}
|
||||
|
||||
// Parse query parameters
|
||||
limit, _ := strconv.Atoi(r.URL.Query().Get("limit"))
|
||||
if limit <= 0 || limit > 100 {
|
||||
limit = 50
|
||||
}
|
||||
|
||||
offset, _ := strconv.Atoi(r.URL.Query().Get("offset"))
|
||||
if offset < 0 {
|
||||
offset = 0
|
||||
}
|
||||
|
||||
unreadOnly := r.URL.Query().Get("unread") == "true"
|
||||
|
||||
articles, err := h.articleRepo.GetByUserID(userID, limit, offset, unreadOnly)
|
||||
if err != nil {
|
||||
respondError(w, http.StatusInternalServerError, "Failed to get articles")
|
||||
return
|
||||
}
|
||||
|
||||
h.sanitizeArticles(articles)
|
||||
respondJSON(w, http.StatusOK, articles)
|
||||
h.list(w, r, domain.ArticleFilter{})
|
||||
}
|
||||
|
||||
// ListByFeed handles GET /api/v1/feeds/{id}/articles
|
||||
func (h *ArticleHandler) ListByFeed(w http.ResponseWriter, r *http.Request) {
|
||||
userID, err := h.getUserFromRequest(r)
|
||||
if err != nil {
|
||||
respondError(w, http.StatusUnauthorized, "Not authenticated")
|
||||
return
|
||||
}
|
||||
|
||||
feedID, err := uuid.Parse(chi.URLParam(r, "id"))
|
||||
if err != nil {
|
||||
respondError(w, http.StatusBadRequest, "Invalid feed ID")
|
||||
return
|
||||
}
|
||||
|
||||
// Verify feed ownership
|
||||
_, err = h.feedService.GetFeed(feedID, userID)
|
||||
if err != nil {
|
||||
respondError(w, http.StatusForbidden, "Access denied")
|
||||
return
|
||||
}
|
||||
|
||||
// Parse query parameters
|
||||
limit, _ := strconv.Atoi(r.URL.Query().Get("limit"))
|
||||
if limit <= 0 || limit > 100 {
|
||||
limit = 50
|
||||
}
|
||||
|
||||
offset, _ := strconv.Atoi(r.URL.Query().Get("offset"))
|
||||
if offset < 0 {
|
||||
offset = 0
|
||||
}
|
||||
|
||||
articles, err := h.articleRepo.GetByFeedID(feedID, limit, offset)
|
||||
if err != nil {
|
||||
respondError(w, http.StatusInternalServerError, "Failed to get articles")
|
||||
return
|
||||
}
|
||||
|
||||
h.sanitizeArticles(articles)
|
||||
respondJSON(w, http.StatusOK, articles)
|
||||
}
|
||||
|
||||
// Get handles GET /api/v1/articles/{id}
|
||||
func (h *ArticleHandler) Get(w http.ResponseWriter, r *http.Request) {
|
||||
userID, err := h.getUserFromRequest(r)
|
||||
if err != nil {
|
||||
respondError(w, http.StatusUnauthorized, "Not authenticated")
|
||||
return
|
||||
}
|
||||
|
||||
articleID, err := uuid.Parse(chi.URLParam(r, "id"))
|
||||
if err != nil {
|
||||
respondError(w, http.StatusBadRequest, "Invalid article ID")
|
||||
return
|
||||
}
|
||||
|
||||
article, err := h.articleRepo.GetByID(articleID)
|
||||
if err != nil || article == nil {
|
||||
respondError(w, http.StatusNotFound, "Article not found")
|
||||
return
|
||||
}
|
||||
|
||||
// Verify feed ownership
|
||||
_, err = h.feedService.GetFeed(article.FeedID, userID)
|
||||
if err != nil {
|
||||
respondError(w, http.StatusForbidden, "Access denied")
|
||||
return
|
||||
}
|
||||
|
||||
// Sanitize user-facing HTML content (defense against stored XSS).
|
||||
h.sanitizeArticle(article)
|
||||
|
||||
respondJSON(w, http.StatusOK, article)
|
||||
}
|
||||
|
||||
// MarkRead handles POST /api/v1/articles/{id}/read
|
||||
func (h *ArticleHandler) MarkRead(w http.ResponseWriter, r *http.Request) {
|
||||
userID, err := h.getUserFromRequest(r)
|
||||
if err != nil {
|
||||
respondError(w, http.StatusUnauthorized, "Not authenticated")
|
||||
return
|
||||
}
|
||||
|
||||
articleID, err := uuid.Parse(chi.URLParam(r, "id"))
|
||||
if err != nil {
|
||||
respondError(w, http.StatusBadRequest, "Invalid article ID")
|
||||
return
|
||||
}
|
||||
|
||||
article, err := h.articleRepo.GetByID(articleID)
|
||||
if err != nil || article == nil {
|
||||
respondError(w, http.StatusNotFound, "Article not found")
|
||||
return
|
||||
}
|
||||
|
||||
// Verify feed ownership
|
||||
_, err = h.feedService.GetFeed(article.FeedID, userID)
|
||||
if err != nil {
|
||||
respondError(w, http.StatusForbidden, "Access denied")
|
||||
return
|
||||
}
|
||||
|
||||
if err := h.articleRepo.MarkAsRead(articleID); err != nil {
|
||||
respondError(w, http.StatusInternalServerError, "Failed to mark as read")
|
||||
return
|
||||
}
|
||||
|
||||
// Broadcast update
|
||||
if h.hub != nil {
|
||||
h.hub.Broadcast("article_updated", map[string]interface{}{
|
||||
"id": articleID,
|
||||
"is_read": true,
|
||||
})
|
||||
}
|
||||
|
||||
respondJSON(w, http.StatusOK, map[string]bool{"is_read": true})
|
||||
}
|
||||
|
||||
// MarkUnread handles DELETE /api/v1/articles/{id}/read
|
||||
func (h *ArticleHandler) MarkUnread(w http.ResponseWriter, r *http.Request) {
|
||||
userID, err := h.getUserFromRequest(r)
|
||||
if err != nil {
|
||||
respondError(w, http.StatusUnauthorized, "Not authenticated")
|
||||
return
|
||||
}
|
||||
|
||||
articleID, err := uuid.Parse(chi.URLParam(r, "id"))
|
||||
if err != nil {
|
||||
respondError(w, http.StatusBadRequest, "Invalid article ID")
|
||||
return
|
||||
}
|
||||
|
||||
article, err := h.articleRepo.GetByID(articleID)
|
||||
if err != nil || article == nil {
|
||||
respondError(w, http.StatusNotFound, "Article not found")
|
||||
return
|
||||
}
|
||||
|
||||
// Verify feed ownership
|
||||
_, err = h.feedService.GetFeed(article.FeedID, userID)
|
||||
if err != nil {
|
||||
respondError(w, http.StatusForbidden, "Access denied")
|
||||
return
|
||||
}
|
||||
|
||||
if err := h.articleRepo.MarkAsUnread(articleID); err != nil {
|
||||
respondError(w, http.StatusInternalServerError, "Failed to mark as unread")
|
||||
return
|
||||
}
|
||||
|
||||
// Broadcast update
|
||||
if h.hub != nil {
|
||||
h.hub.Broadcast("article_updated", map[string]interface{}{
|
||||
"id": articleID,
|
||||
"is_read": false,
|
||||
})
|
||||
}
|
||||
|
||||
respondJSON(w, http.StatusOK, map[string]bool{"is_read": false})
|
||||
}
|
||||
|
||||
// ToggleFavorite handles POST /api/v1/articles/{id}/favorite
|
||||
func (h *ArticleHandler) ToggleFavorite(w http.ResponseWriter, r *http.Request) {
|
||||
userID, err := h.getUserFromRequest(r)
|
||||
if err != nil {
|
||||
respondError(w, http.StatusUnauthorized, "Not authenticated")
|
||||
return
|
||||
}
|
||||
|
||||
articleID, err := uuid.Parse(chi.URLParam(r, "id"))
|
||||
if err != nil {
|
||||
respondError(w, http.StatusBadRequest, "Invalid article ID")
|
||||
return
|
||||
}
|
||||
|
||||
article, err := h.articleRepo.GetByID(articleID)
|
||||
if err != nil || article == nil {
|
||||
respondError(w, http.StatusNotFound, "Article not found")
|
||||
return
|
||||
}
|
||||
|
||||
// Verify feed ownership
|
||||
_, err = h.feedService.GetFeed(article.FeedID, userID)
|
||||
if err != nil {
|
||||
respondError(w, http.StatusForbidden, "Access denied")
|
||||
return
|
||||
}
|
||||
|
||||
if err := h.articleRepo.ToggleFavorite(articleID); err != nil {
|
||||
respondError(w, http.StatusInternalServerError, "Failed to toggle favorite")
|
||||
return
|
||||
}
|
||||
|
||||
// Broadcast update
|
||||
if h.hub != nil {
|
||||
h.hub.Broadcast("article_updated", map[string]interface{}{
|
||||
"id": articleID,
|
||||
"is_favorite": !article.IsFavorite,
|
||||
})
|
||||
}
|
||||
|
||||
respondJSON(w, http.StatusOK, map[string]bool{"is_favorite": !article.IsFavorite})
|
||||
}
|
||||
|
||||
// MarkAllRead handles POST /api/v1/feeds/{id}/read-all
|
||||
func (h *ArticleHandler) MarkAllRead(w http.ResponseWriter, r *http.Request) {
|
||||
userID, err := h.getUserFromRequest(r)
|
||||
if err != nil {
|
||||
respondError(w, http.StatusUnauthorized, "Not authenticated")
|
||||
return
|
||||
}
|
||||
|
||||
feedID, err := uuid.Parse(chi.URLParam(r, "id"))
|
||||
if err != nil {
|
||||
respondError(w, http.StatusBadRequest, "Invalid feed ID")
|
||||
return
|
||||
}
|
||||
|
||||
// Verify feed ownership
|
||||
_, err = h.feedService.GetFeed(feedID, userID)
|
||||
if err != nil {
|
||||
respondError(w, http.StatusForbidden, "Access denied")
|
||||
return
|
||||
}
|
||||
|
||||
if err := h.articleRepo.MarkAllAsRead(feedID); err != nil {
|
||||
respondError(w, http.StatusInternalServerError, "Failed to mark all as read")
|
||||
return
|
||||
}
|
||||
|
||||
respondJSON(w, http.StatusOK, map[string]string{"message": "All articles marked as read"})
|
||||
}
|
||||
|
||||
// MarkAllReadGlobal handles POST /api/v1/articles/read-all
|
||||
func (h *ArticleHandler) MarkAllReadGlobal(w http.ResponseWriter, r *http.Request) {
|
||||
userID, err := h.getUserFromRequest(r)
|
||||
if err != nil {
|
||||
respondError(w, http.StatusUnauthorized, "Not authenticated")
|
||||
return
|
||||
}
|
||||
|
||||
if err := h.articleRepo.MarkAllAsReadGlobal(userID); err != nil {
|
||||
respondError(w, http.StatusInternalServerError, "Failed to mark all as read")
|
||||
return
|
||||
}
|
||||
|
||||
respondJSON(w, http.StatusOK, map[string]string{"message": "All articles marked as read"})
|
||||
h.list(w, r, domain.ArticleFilter{FeedID: &feedID})
|
||||
}
|
||||
|
||||
// GetFavorites handles GET /api/v1/articles/favorites
|
||||
func (h *ArticleHandler) GetFavorites(w http.ResponseWriter, r *http.Request) {
|
||||
userID, err := h.getUserFromRequest(r)
|
||||
if err != nil {
|
||||
respondError(w, http.StatusUnauthorized, "Not authenticated")
|
||||
return
|
||||
}
|
||||
|
||||
limit, _ := strconv.Atoi(r.URL.Query().Get("limit"))
|
||||
if limit <= 0 || limit > 100 {
|
||||
limit = 50
|
||||
}
|
||||
|
||||
offset, _ := strconv.Atoi(r.URL.Query().Get("offset"))
|
||||
if offset < 0 {
|
||||
offset = 0
|
||||
}
|
||||
|
||||
articles, err := h.articleRepo.GetFavorites(userID, limit, offset)
|
||||
if err != nil {
|
||||
respondError(w, http.StatusInternalServerError, "Failed to get favorites")
|
||||
return
|
||||
}
|
||||
|
||||
h.sanitizeArticles(articles)
|
||||
respondJSON(w, http.StatusOK, articles)
|
||||
h.list(w, r, domain.ArticleFilter{FavoritesOnly: true})
|
||||
}
|
||||
|
||||
// Search handles GET /api/v1/articles/search
|
||||
func (h *ArticleHandler) Search(w http.ResponseWriter, r *http.Request) {
|
||||
userID, err := h.getUserFromRequest(r)
|
||||
if err != nil {
|
||||
respondError(w, http.StatusUnauthorized, "Not authenticated")
|
||||
return
|
||||
}
|
||||
|
||||
query := r.URL.Query().Get("q")
|
||||
query := strings.TrimSpace(r.URL.Query().Get("q"))
|
||||
if query == "" {
|
||||
respondJSON(w, http.StatusOK, []*domain.Article{})
|
||||
return
|
||||
}
|
||||
|
||||
limit, _ := strconv.Atoi(r.URL.Query().Get("limit"))
|
||||
if limit <= 0 || limit > 100 {
|
||||
limit = 50
|
||||
}
|
||||
query = utils.TruncateRunes(query, 200)
|
||||
|
||||
offset, _ := strconv.Atoi(r.URL.Query().Get("offset"))
|
||||
if offset < 0 {
|
||||
if offset < 0 || offset > 10000 {
|
||||
offset = 0
|
||||
}
|
||||
|
||||
articles, err := h.articleRepo.Search(userID, query, limit, offset)
|
||||
articles, err := h.articleRepo.Search(r.Context(), currentUser(r).ID, query, parseLimit(r), offset)
|
||||
if err != nil {
|
||||
respondError(w, http.StatusInternalServerError, "Failed to search articles")
|
||||
return
|
||||
}
|
||||
|
||||
h.sanitizeArticles(articles)
|
||||
respondJSON(w, http.StatusOK, articles)
|
||||
}
|
||||
|
||||
// Summarize handles POST /api/v1/articles/{id}/summarize
|
||||
func (h *ArticleHandler) Summarize(w http.ResponseWriter, r *http.Request) {
|
||||
userID, err := h.getUserFromRequest(r)
|
||||
if err != nil {
|
||||
respondError(w, http.StatusUnauthorized, "Not authenticated")
|
||||
return
|
||||
}
|
||||
|
||||
// Get handles GET /api/v1/articles/{id} and returns the full content.
|
||||
func (h *ArticleHandler) Get(w http.ResponseWriter, r *http.Request) {
|
||||
articleID, err := uuid.Parse(chi.URLParam(r, "id"))
|
||||
if err != nil {
|
||||
respondError(w, http.StatusBadRequest, "Invalid article ID")
|
||||
return
|
||||
}
|
||||
|
||||
article, err := h.articleRepo.GetByID(articleID)
|
||||
if err != nil || article == nil {
|
||||
article, err := h.articleRepo.GetForUser(r.Context(), articleID, currentUser(r).ID)
|
||||
if err != nil {
|
||||
respondError(w, http.StatusInternalServerError, "Failed to get article")
|
||||
return
|
||||
}
|
||||
if article == nil {
|
||||
respondError(w, http.StatusNotFound, "Article not found")
|
||||
return
|
||||
}
|
||||
|
||||
// Verify feed ownership
|
||||
_, err = h.feedService.GetFeed(article.FeedID, userID)
|
||||
// Content is sanitized at ingest; sanitizing again here is cheap for a
|
||||
// single article and keeps defence in depth for rows not yet backfilled.
|
||||
article.Content = h.sanitizer.Sanitize(article.Content)
|
||||
article.Summary = h.sanitizer.Sanitize(article.Summary)
|
||||
|
||||
w.Header().Set("Cache-Control", "private, no-cache")
|
||||
respondJSON(w, http.StatusOK, article)
|
||||
}
|
||||
|
||||
func (h *ArticleHandler) setRead(w http.ResponseWriter, r *http.Request, read bool) {
|
||||
articleID, err := uuid.Parse(chi.URLParam(r, "id"))
|
||||
if err != nil {
|
||||
respondError(w, http.StatusForbidden, "Access denied")
|
||||
respondError(w, http.StatusBadRequest, "Invalid article ID")
|
||||
return
|
||||
}
|
||||
userID := currentUser(r).ID
|
||||
|
||||
found, err := h.articleRepo.SetRead(r.Context(), articleID, userID, read)
|
||||
if err != nil {
|
||||
respondError(w, http.StatusInternalServerError, "Failed to update article")
|
||||
return
|
||||
}
|
||||
if !found {
|
||||
respondError(w, http.StatusNotFound, "Article not found")
|
||||
return
|
||||
}
|
||||
|
||||
h.notify(userID, "article_updated", map[string]any{"id": articleID, "is_read": read})
|
||||
respondJSON(w, http.StatusOK, map[string]bool{"is_read": read})
|
||||
}
|
||||
|
||||
// MarkRead handles POST /api/v1/articles/{id}/read
|
||||
func (h *ArticleHandler) MarkRead(w http.ResponseWriter, r *http.Request) { h.setRead(w, r, true) }
|
||||
|
||||
// MarkUnread handles DELETE /api/v1/articles/{id}/read
|
||||
func (h *ArticleHandler) MarkUnread(w http.ResponseWriter, r *http.Request) { h.setRead(w, r, false) }
|
||||
|
||||
// ToggleFavorite handles POST /api/v1/articles/{id}/favorite
|
||||
func (h *ArticleHandler) ToggleFavorite(w http.ResponseWriter, r *http.Request) {
|
||||
articleID, err := uuid.Parse(chi.URLParam(r, "id"))
|
||||
if err != nil {
|
||||
respondError(w, http.StatusBadRequest, "Invalid article ID")
|
||||
return
|
||||
}
|
||||
userID := currentUser(r).ID
|
||||
|
||||
fav, found, err := h.articleRepo.ToggleFavorite(r.Context(), articleID, userID)
|
||||
if err != nil {
|
||||
respondError(w, http.StatusInternalServerError, "Failed to toggle favorite")
|
||||
return
|
||||
}
|
||||
if !found {
|
||||
respondError(w, http.StatusNotFound, "Article not found")
|
||||
return
|
||||
}
|
||||
|
||||
h.notify(userID, "article_updated", map[string]any{"id": articleID, "is_favorite": fav})
|
||||
respondJSON(w, http.StatusOK, map[string]bool{"is_favorite": fav})
|
||||
}
|
||||
|
||||
// MarkAllRead handles POST /api/v1/feeds/{id}/read-all
|
||||
func (h *ArticleHandler) MarkAllRead(w http.ResponseWriter, r *http.Request) {
|
||||
feedID, err := uuid.Parse(chi.URLParam(r, "id"))
|
||||
if err != nil {
|
||||
respondError(w, http.StatusBadRequest, "Invalid feed ID")
|
||||
return
|
||||
}
|
||||
userID := currentUser(r).ID
|
||||
|
||||
n, err := h.articleRepo.MarkFeedRead(r.Context(), feedID, userID)
|
||||
if err != nil {
|
||||
respondError(w, http.StatusInternalServerError, "Failed to mark all as read")
|
||||
return
|
||||
}
|
||||
if n > 0 {
|
||||
h.notify(userID, "articles_bulk_read", map[string]any{"feed_id": feedID})
|
||||
}
|
||||
respondJSON(w, http.StatusOK, map[string]any{"message": "All articles marked as read", "count": n})
|
||||
}
|
||||
|
||||
// MarkAllReadGlobal handles POST /api/v1/articles/read-all
|
||||
func (h *ArticleHandler) MarkAllReadGlobal(w http.ResponseWriter, r *http.Request) {
|
||||
userID := currentUser(r).ID
|
||||
n, err := h.articleRepo.MarkAllRead(r.Context(), userID)
|
||||
if err != nil {
|
||||
respondError(w, http.StatusInternalServerError, "Failed to mark all as read")
|
||||
return
|
||||
}
|
||||
if n > 0 {
|
||||
h.notify(userID, "articles_bulk_read", map[string]any{})
|
||||
}
|
||||
respondJSON(w, http.StatusOK, map[string]any{"message": "All articles marked as read", "count": n})
|
||||
}
|
||||
|
||||
// Summarize handles POST /api/v1/articles/{id}/summarize
|
||||
func (h *ArticleHandler) Summarize(w http.ResponseWriter, r *http.Request) {
|
||||
articleID, err := uuid.Parse(chi.URLParam(r, "id"))
|
||||
if err != nil {
|
||||
respondError(w, http.StatusBadRequest, "Invalid article ID")
|
||||
return
|
||||
}
|
||||
userID := currentUser(r).ID
|
||||
|
||||
article, err := h.articleRepo.GetForUser(r.Context(), articleID, userID)
|
||||
if err != nil {
|
||||
respondError(w, http.StatusInternalServerError, "Failed to get article")
|
||||
return
|
||||
}
|
||||
if article == nil {
|
||||
respondError(w, http.StatusNotFound, "Article not found")
|
||||
return
|
||||
}
|
||||
|
||||
@@ -448,44 +269,39 @@ func (h *ArticleHandler) Summarize(w http.ResponseWriter, r *http.Request) {
|
||||
respondJSON(w, http.StatusOK, map[string]string{"summary": article.AISummary})
|
||||
return
|
||||
}
|
||||
if !h.aiService.Enabled() {
|
||||
respondError(w, http.StatusServiceUnavailable, "AI summaries are not configured")
|
||||
return
|
||||
}
|
||||
|
||||
// Context for AI is title + content (or summary if content empty)
|
||||
content := article.Content
|
||||
// Summaries take an extractor fetch plus an LLM call: allow more than the
|
||||
// server's default write deadline for this route only.
|
||||
_ = http.NewResponseController(w).SetWriteDeadline(time.Now().Add(90 * time.Second))
|
||||
|
||||
content := utils.PlainText(article.Content)
|
||||
if content == "" {
|
||||
content = article.Summary
|
||||
content = utils.PlainText(article.Summary)
|
||||
}
|
||||
|
||||
// Try to extract full content from URL if available
|
||||
if article.URL != "" {
|
||||
fullContent, err := h.extractor.Extract(r.Context(), article.URL)
|
||||
if err == nil && len(fullContent) > len(content) {
|
||||
content = "--- CONTENU COMPLET EXTRAIT DU SITE WEB ---\n" + fullContent
|
||||
content = fullContent
|
||||
}
|
||||
}
|
||||
|
||||
aiInput := fmt.Sprintf("Titre: %s\n\nContenu: %s", article.Title, content)
|
||||
|
||||
// Summary generation (can be slow, but for this demo/small app we do it synchronously
|
||||
// or we could use WS to notify when done. Here we follow the simple POST -> String pattern).
|
||||
summary, err := h.aiService.Summarize(r.Context(), aiInput)
|
||||
summary, err := h.aiService.Summarize(r.Context(), "Titre : "+article.Title+"\n\n"+content)
|
||||
if err != nil {
|
||||
respondError(w, http.StatusInternalServerError, "Failed to generate summary: "+err.Error())
|
||||
respondError(w, http.StatusBadGateway, "Failed to generate summary")
|
||||
return
|
||||
}
|
||||
|
||||
// Save to DB
|
||||
if err := h.articleRepo.UpdateAISummary(articleID, summary); err != nil {
|
||||
if err := h.articleRepo.UpdateAISummary(r.Context(), articleID, summary); err != nil {
|
||||
respondError(w, http.StatusInternalServerError, "Failed to persist summary")
|
||||
return
|
||||
}
|
||||
|
||||
// Broadcast update via WebSocket
|
||||
if h.hub != nil {
|
||||
h.hub.Broadcast("article_updated", map[string]interface{}{
|
||||
"id": articleID,
|
||||
"ai_summary": summary,
|
||||
})
|
||||
}
|
||||
|
||||
h.notify(userID, "article_updated", map[string]any{"id": articleID, "ai_summary": summary})
|
||||
respondJSON(w, http.StatusOK, map[string]string{"summary": summary})
|
||||
}
|
||||
Reference in new issue
Block a user