mirror of
https://github.com/truewhile/MeBox.git
synced 2026-10-01 03:56:38 +08:00
truewhile
fa41936095
后端： - /api/reader/media 签名媒体代理：HMAC-SHA256(jwtSecret, bookID|url) 防开放 中转滥用；免 JWT（audio/img 元素带不了头）但需有效签名+书籍存在 - 携书源级请求头与 Referer（防盗链）、Range 透传（音频拖动进度 206）、 m3u8 播放列表分片/密钥地址改写为签名代理（hls.js 无感续播） - 音频/图片地址按最终页面 URL 绝对化，imageStyle 透传； GetContentForBook 将 tracks/images 统一改写为代理地址 前端： - ReaderAudioPanel（仿 ReadAloudDialog）：hls.js 播 m3u8 / <audio> 直链， 上一章/播放暂停/下一章、进度条拖动、倍速 0.75–2x、按秒记忆进度、 播完自动下一章、点背景呼出菜单 - ReaderComic：上下滚动 / 左右单页双模式（复用翻页/滚动切换）、 图片懒加载+失败占位、滚动模式左右点击滚屏/中间呼菜单、 单页模式点击区翻页、菜单进度条按图片序号、按图片序号记忆进度 - 菜单进度条按内容类型适配（文本页/音频章节/漫画图片序号）
398 lines
13 KiB
Go
398 lines
13 KiB
Go
// Package handler — 阅读（legado 书源兼容）子系统路由。
|
|
package handler
|
|
|
|
import (
|
|
"io"
|
|
"net/http"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"github.com/truewhile/MeBox/internal/middleware"
|
|
"github.com/truewhile/MeBox/internal/service"
|
|
"github.com/truewhile/MeBox/internal/service/reader"
|
|
)
|
|
|
|
func registerReaderRoutes(authed *gin.RouterGroup, svc *service.Container) {
|
|
g := authed.Group("/reader")
|
|
|
|
// 书源管理
|
|
g.GET("/sources", readerListSourcesHandler(svc))
|
|
g.POST("/sources/import", readerImportSourcesHandler(svc))
|
|
g.PATCH("/sources/:id", readerUpdateSourceHandler(svc))
|
|
g.DELETE("/sources/:id", readerDeleteSourceHandler(svc))
|
|
g.POST("/sources/:id/debug", readerDebugSourceHandler(svc))
|
|
|
|
// 搜索（多源聚合）
|
|
g.POST("/search", readerSearchHandler(svc))
|
|
|
|
// 详情 / 目录 / 正文
|
|
g.GET("/book-info", readerBookInfoHandler(svc))
|
|
g.GET("/toc", readerTocHandler(svc))
|
|
g.GET("/content", readerContentHandler(svc))
|
|
|
|
// 书架
|
|
g.GET("/books", readerListBooksHandler(svc))
|
|
g.POST("/books", readerAddBookHandler(svc))
|
|
g.DELETE("/books/:id", readerRemoveBookHandler(svc))
|
|
g.PUT("/books/:id/progress", readerSaveProgressHandler(svc))
|
|
g.GET("/books/:id/chapters", readerListChaptersHandler(svc))
|
|
g.POST("/books/:id/chapters", readerReplaceChaptersHandler(svc))
|
|
g.GET("/books/:id/content", readerBookContentHandler(svc))
|
|
|
|
// 替换净化规则
|
|
g.GET("/replace-rules", readerListReplaceRulesHandler(svc))
|
|
g.POST("/replace-rules", readerCreateReplaceRuleHandler(svc))
|
|
g.PATCH("/replace-rules/:id", readerUpdateReplaceRuleHandler(svc))
|
|
g.DELETE("/replace-rules/:id", readerDeleteReplaceRuleHandler(svc))
|
|
}
|
|
|
|
func readerListSourcesHandler(svc *service.Container) gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
sources, err := svc.Reader.ListSources(c.Request.Context())
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, gin.H{"sources": sources})
|
|
}
|
|
}
|
|
|
|
func readerImportSourcesHandler(svc *service.Container) gin.HandlerFunc {
|
|
var body struct {
|
|
Text string `json:"text" binding:"required"`
|
|
}
|
|
return func(c *gin.Context) {
|
|
if err := c.ShouldBindJSON(&body); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
imported, err := svc.Reader.ImportSources(c.Request.Context(), body.Text)
|
|
if err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, gin.H{"imported": imported})
|
|
}
|
|
}
|
|
|
|
func readerUpdateSourceHandler(svc *service.Container) gin.HandlerFunc {
|
|
var body struct {
|
|
Enabled *bool `json:"enabled"`
|
|
}
|
|
return func(c *gin.Context) {
|
|
if err := c.ShouldBindJSON(&body); err != nil || body.Enabled == nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "enabled 字段必填"})
|
|
return
|
|
}
|
|
if err := svc.Reader.UpdateSourceEnabled(c.Request.Context(), c.Param("id"), *body.Enabled); err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, gin.H{"ok": true})
|
|
}
|
|
}
|
|
|
|
func readerDeleteSourceHandler(svc *service.Container) gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
if err := svc.Reader.DeleteSource(c.Request.Context(), c.Param("id")); err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, gin.H{"ok": true})
|
|
}
|
|
}
|
|
|
|
func readerDebugSourceHandler(svc *service.Container) gin.HandlerFunc {
|
|
var body struct {
|
|
Key string `json:"key" binding:"required"`
|
|
}
|
|
return func(c *gin.Context) {
|
|
if err := c.ShouldBindJSON(&body); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
logs, err := svc.Reader.Debug(c.Request.Context(), c.Param("id"), body.Key)
|
|
if err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, gin.H{"logs": logs})
|
|
}
|
|
}
|
|
|
|
func readerSearchHandler(svc *service.Container) gin.HandlerFunc {
|
|
var body struct {
|
|
Key string `json:"key" binding:"required"`
|
|
}
|
|
return func(c *gin.Context) {
|
|
if err := c.ShouldBindJSON(&body); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
books, skipped, err := svc.Reader.Search(c.Request.Context(), body.Key)
|
|
if err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, gin.H{"books": books, "skipped": skipped})
|
|
}
|
|
}
|
|
|
|
func readerBookInfoHandler(svc *service.Container) gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
info, err := svc.Reader.GetBookInfo(
|
|
c.Request.Context(),
|
|
c.Query("source_id"), c.Query("source_url"), c.Query("book_url"),
|
|
)
|
|
if err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, info)
|
|
}
|
|
}
|
|
|
|
func readerTocHandler(svc *service.Container) gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
chapters, err := svc.Reader.GetToc(
|
|
c.Request.Context(),
|
|
c.Query("source_id"), c.Query("source_url"), c.Query("book_url"), c.Query("toc_url"),
|
|
)
|
|
if err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, gin.H{"chapters": chapters})
|
|
}
|
|
}
|
|
|
|
func readerContentHandler(svc *service.Container) gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
content, err := svc.Reader.GetContent(
|
|
c.Request.Context(),
|
|
c.Query("source_id"), c.Query("source_url"), c.Query("book_url"), c.Query("chapter_url"),
|
|
)
|
|
if err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, content)
|
|
}
|
|
}
|
|
|
|
func readerListBooksHandler(svc *service.Container) gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
userID := c.GetString(middleware.CtxUserID)
|
|
books, err := svc.Reader.ListBooks(c.Request.Context(), userID)
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, gin.H{"books": books})
|
|
}
|
|
}
|
|
|
|
func readerAddBookHandler(svc *service.Container) gin.HandlerFunc {
|
|
var body struct {
|
|
Origin reader.SearchOrigin `json:"origin" binding:"required"`
|
|
Name string `json:"name" binding:"required"`
|
|
Author string `json:"author"`
|
|
CoverURL string `json:"cover_url"`
|
|
}
|
|
return func(c *gin.Context) {
|
|
if err := c.ShouldBindJSON(&body); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
userID := c.GetString(middleware.CtxUserID)
|
|
book, err := svc.Reader.AddBook(c.Request.Context(), userID, body.Origin, body.Name, body.Author, body.CoverURL)
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, book)
|
|
}
|
|
}
|
|
|
|
func readerRemoveBookHandler(svc *service.Container) gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
userID := c.GetString(middleware.CtxUserID)
|
|
if err := svc.Reader.RemoveBook(c.Request.Context(), userID, c.Param("id")); err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, gin.H{"ok": true})
|
|
}
|
|
}
|
|
|
|
func readerSaveProgressHandler(svc *service.Container) gin.HandlerFunc {
|
|
var body struct {
|
|
ChapterIndex int `json:"chapter_index"`
|
|
Pos int `json:"pos"`
|
|
ChapterTitle string `json:"chapter_title"`
|
|
}
|
|
return func(c *gin.Context) {
|
|
if err := c.ShouldBindJSON(&body); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
userID := c.GetString(middleware.CtxUserID)
|
|
if err := svc.Reader.SaveProgress(c.Request.Context(), userID, c.Param("id"), body.ChapterIndex, body.Pos, body.ChapterTitle); err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, gin.H{"ok": true})
|
|
}
|
|
}
|
|
|
|
func readerListChaptersHandler(svc *service.Container) gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
chapters, err := svc.Reader.ListChapters(c.Request.Context(), c.Param("id"))
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, gin.H{"chapters": chapters})
|
|
}
|
|
}
|
|
|
|
func readerReplaceChaptersHandler(svc *service.Container) gin.HandlerFunc {
|
|
var body struct {
|
|
Chapters []reader.ChapterInput `json:"chapters"`
|
|
}
|
|
return func(c *gin.Context) {
|
|
if err := c.ShouldBindJSON(&body); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
if err := svc.Reader.SaveChapters(c.Request.Context(), c.Param("id"), body.Chapters); err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, gin.H{"ok": true})
|
|
}
|
|
}
|
|
|
|
func readerListReplaceRulesHandler(svc *service.Container) gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
userID := c.GetString(middleware.CtxUserID)
|
|
rules, err := svc.Reader.ListReplaceRules(c.Request.Context(), userID)
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, gin.H{"rules": rules})
|
|
}
|
|
}
|
|
|
|
func readerCreateReplaceRuleHandler(svc *service.Container) gin.HandlerFunc {
|
|
var body reader.ReplaceRuleInput
|
|
return func(c *gin.Context) {
|
|
if err := c.ShouldBindJSON(&body); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
userID := c.GetString(middleware.CtxUserID)
|
|
rule, err := svc.Reader.CreateReplaceRule(c.Request.Context(), userID, body)
|
|
if err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, rule)
|
|
}
|
|
}
|
|
|
|
func readerUpdateReplaceRuleHandler(svc *service.Container) gin.HandlerFunc {
|
|
var body reader.ReplaceRuleInput
|
|
return func(c *gin.Context) {
|
|
if err := c.ShouldBindJSON(&body); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
userID := c.GetString(middleware.CtxUserID)
|
|
if err := svc.Reader.UpdateReplaceRule(c.Request.Context(), userID, c.Param("id"), body); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, gin.H{"ok": true})
|
|
}
|
|
}
|
|
|
|
func readerDeleteReplaceRuleHandler(svc *service.Container) gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
userID := c.GetString(middleware.CtxUserID)
|
|
if err := svc.Reader.DeleteReplaceRule(c.Request.Context(), userID, c.Param("id")); err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, gin.H{"ok": true})
|
|
}
|
|
}
|
|
|
|
// readerMediaProxyHandler 音频流/漫画图片签名代理：
|
|
// 校验 HMAC 签名 → 携书源防盗链头拉取 → Range 透传（音频拖动）/ m3u8 重写。
|
|
func readerMediaProxyHandler(svc *service.Container) gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
bookID := c.Query("b")
|
|
rawURL, err := svc.Reader.VerifyProxyURL(bookID, c.Query("u"), c.Query("s"))
|
|
if err != nil {
|
|
c.JSON(http.StatusForbidden, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
if !strings.HasPrefix(rawURL, "http://") && !strings.HasPrefix(rawURL, "https://") {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "仅支持 http(s) 媒体地址"})
|
|
return
|
|
}
|
|
book, err := svc.Reader.GetBook(c.Request.Context(), bookID)
|
|
if err != nil {
|
|
c.JSON(http.StatusNotFound, gin.H{"error": "书籍不存在"})
|
|
return
|
|
}
|
|
resp, err := svc.Reader.FetchMedia(c.Request.Context(), book, rawURL, c.GetHeader("Range"))
|
|
if err != nil {
|
|
c.JSON(http.StatusBadGateway, gin.H{"error": "媒体拉取失败: " + err.Error()})
|
|
return
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
ct := resp.Header.Get("Content-Type")
|
|
isPlaylist := strings.Contains(ct, "mpegurl") || strings.Contains(ct, "m3u8") ||
|
|
strings.HasSuffix(strings.ToLower(rawURL), ".m3u8")
|
|
if isPlaylist {
|
|
// m3u8：改写分片/密钥地址为签名代理后返回，hls.js 无感续播
|
|
data, _ := io.ReadAll(io.LimitReader(resp.Body, 4<<20))
|
|
rewritten := svc.Reader.RewritePlaylist(bookID, rawURL, string(data))
|
|
c.Data(http.StatusOK, "application/vnd.apple.mpegurl", []byte(rewritten))
|
|
return
|
|
}
|
|
// 流式透传（含 206 Partial Content，支持音频拖动进度）
|
|
for _, h := range []string{"Content-Type", "Content-Length", "Content-Range", "Accept-Ranges"} {
|
|
if v := resp.Header.Get(h); v != "" {
|
|
c.Header(h, v)
|
|
}
|
|
}
|
|
c.Status(resp.StatusCode)
|
|
_, _ = io.Copy(c.Writer, resp.Body)
|
|
}
|
|
}
|
|
|
|
// readerBookContentHandler 书架维度正文（服务端应用替换净化规则）。
|
|
func readerBookContentHandler(svc *service.Container) gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
chapter, err := strconv.Atoi(c.DefaultQuery("chapter", "0"))
|
|
if err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "chapter 参数需为整数"})
|
|
return
|
|
}
|
|
userID := c.GetString(middleware.CtxUserID)
|
|
content, err := svc.Reader.GetContentForBook(c.Request.Context(), userID, c.Param("id"), chapter)
|
|
if err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, content)
|
|
}
|
|
}
|