mirror of
https://github.com/truewhile/MeBox.git
synced 2026-09-29 03:26:37 +08:00
truewhile
f20c9c3234
不再把 STRM/云盘媒体锁死为直连，解析远端源后允许 ffmpeg 转码播放。 Co-authored-by: Cursor <cursoragent@cursor.com>
298 lines
9.0 KiB
Go
298 lines
9.0 KiB
Go
// Package handler — playback metadata endpoints expected by the Vue UI:
|
|
//
|
|
// GET /playback/:id/info
|
|
// POST /playback/:id/progress
|
|
// GET /playback/:id/external-players
|
|
// GET /playback/:id/external-url
|
|
// GET /playback/transcode/:job_id/status
|
|
package handler
|
|
|
|
import (
|
|
"net/http"
|
|
"net/url"
|
|
"strings"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"github.com/truewhile/MeBox/internal/middleware"
|
|
"github.com/truewhile/MeBox/internal/model"
|
|
"github.com/truewhile/MeBox/internal/service"
|
|
)
|
|
|
|
func findMediaForPlaybackEndpoint(c *gin.Context, svc *service.Container, id string) (*model.Media, error) {
|
|
ctx := c.Request.Context()
|
|
if svc.EmbyRemote != nil && service.IsEmbyRemoteID(id) {
|
|
mountID, remoteID, _ := service.DecodeEmbyRemoteID(id)
|
|
mount, acct, _ := svc.EmbyRemote.ResolveMount(ctx, mountID)
|
|
if mount == nil || acct == nil {
|
|
return nil, nil
|
|
}
|
|
return svc.EmbyRemote.RemoteMediaDetail(ctx, mount, acct, remoteID)
|
|
}
|
|
return svc.Repo.Media.FindByID(ctx, id)
|
|
}
|
|
|
|
// playbackInfoHandler returns the media row + a `stream_url` the React
|
|
// player can hit. Mirrors the Python project's surface.
|
|
func playbackInfoHandler(svc *service.Container) gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
id := c.Param("id")
|
|
m, err := findMediaForPlaybackEndpoint(c, svc, id)
|
|
if err != nil || m == nil || !mediaVisibleForRequest(c, svc, m) {
|
|
c.JSON(http.StatusNotFound, gin.H{"error": "media not found"})
|
|
return
|
|
}
|
|
token := externalPlaybackToken(c, svc, m.ID, m.DurationSec)
|
|
profileQuery := externalProfileQuery(c)
|
|
hlsURL := "/api/hls/" + m.ID + "/index.m3u8?token=" + url.QueryEscape(token) + profileQuery
|
|
if service.IsEmbyRemoteID(m.ID) {
|
|
// 远程 Emby 挂载没有本地文件，不能提供转码地址。STRM 默认直连，直连失败时可走 HLS。
|
|
hlsURL = ""
|
|
}
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"media": m,
|
|
"stream_url": "/api/stream/" + m.ID + "?token=" + url.QueryEscape(token) + profileQuery,
|
|
"hls_url": hlsURL,
|
|
})
|
|
}
|
|
}
|
|
|
|
type playbackProgressReq struct {
|
|
PositionMs int64 `json:"position_ms"`
|
|
DurationMs int64 `json:"duration_ms"`
|
|
Completed bool `json:"completed"`
|
|
}
|
|
|
|
func playbackProgressHandler(svc *service.Container) gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
var req playbackProgressReq
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
uid, _ := c.Get(middleware.CtxUserID)
|
|
if err := svc.Playback.RecordProgress(
|
|
c.Request.Context(), toString(uid), c.Param("id"),
|
|
req.PositionMs, req.DurationMs,
|
|
); err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
c.Status(http.StatusNoContent)
|
|
}
|
|
}
|
|
|
|
func playbackResumeHandler(svc *service.Container) gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
uid, _ := c.Get(middleware.CtxUserID)
|
|
row, err := svc.Playback.GetProgress(c.Request.Context(), toString(uid), c.Param("id"))
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
if row == nil {
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"position_ms": 0,
|
|
"duration_ms": 0,
|
|
"completed": false,
|
|
})
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"position_ms": row.PositionMs,
|
|
"duration_ms": row.DurationMs,
|
|
"completed": row.Completed,
|
|
})
|
|
}
|
|
}
|
|
|
|
// externalPlayersHandler returns the list of external player URI
|
|
// schemes the UI can offer the user. We lookup the media row to
|
|
// produce the per-player launch URL.
|
|
func externalPlayersHandler(svc *service.Container) gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
id := c.Param("id")
|
|
m, err := findMediaForPlaybackEndpoint(c, svc, id)
|
|
if err != nil || m == nil || !mediaVisibleForRequest(c, svc, m) {
|
|
c.JSON(http.StatusNotFound, gin.H{"error": "media not found"})
|
|
return
|
|
}
|
|
token := externalPlaybackToken(c, svc, m.ID, m.DurationSec)
|
|
streamURL := externalPlaybackURL(c, svc, "/api/stream/"+m.ID+"?token="+url.QueryEscape(token)+externalProfileQuery(c))
|
|
escapedStream := url.QueryEscape(streamURL)
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"url": streamURL,
|
|
"players": []gin.H{
|
|
{"name": "VLC", "scheme": "vlc://", "url": "vlc://" + streamURL},
|
|
{"name": "PotPlayer", "scheme": "potplayer://", "url": "potplayer://" + streamURL},
|
|
{"name": "MX Player", "scheme": "intent://", "url": "intent://" + streamURL + "#Intent;package=com.mxtech.videoplayer.ad;end"},
|
|
{"name": "IINA", "scheme": "iina://", "url": "iina://weblink?url=" + escapedStream},
|
|
{"name": "nPlayer", "scheme": "nplayer-", "url": "nplayer-" + streamURL},
|
|
},
|
|
})
|
|
}
|
|
}
|
|
|
|
// externalURLHandler returns just the raw stream URL plus the auth
|
|
// token query string the external player needs.
|
|
func externalURLHandler(svc *service.Container) gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
id := c.Param("id")
|
|
m, err := findMediaForPlaybackEndpoint(c, svc, id)
|
|
if err != nil || m == nil || !mediaVisibleForRequest(c, svc, m) {
|
|
c.JSON(http.StatusNotFound, gin.H{"error": "media not found"})
|
|
return
|
|
}
|
|
token := externalPlaybackToken(c, svc, m.ID, m.DurationSec)
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"url": externalPlaybackURL(c, svc, "/api/stream/"+m.ID+"?token="+url.QueryEscape(token)+externalProfileQuery(c)),
|
|
})
|
|
}
|
|
}
|
|
|
|
func externalProfileQuery(c *gin.Context) string {
|
|
profileID := strings.TrimSpace(c.GetHeader("X-Play-Profile-ID"))
|
|
if profileID == "" {
|
|
profileID = strings.TrimSpace(c.Query("profile_id"))
|
|
}
|
|
if profileID == "" {
|
|
return ""
|
|
}
|
|
query := "&profile_id=" + url.QueryEscape(profileID)
|
|
pinToken := strings.TrimSpace(c.GetHeader("X-Play-Profile-PIN-Token"))
|
|
if pinToken == "" {
|
|
pinToken = strings.TrimSpace(c.Query("profile_pin_token"))
|
|
}
|
|
if pinToken != "" {
|
|
query += "&profile_pin_token=" + url.QueryEscape(pinToken)
|
|
}
|
|
return query
|
|
}
|
|
|
|
func externalPlaybackToken(c *gin.Context, svc *service.Container, mediaID string, durationSec int) string {
|
|
uid, _ := c.Get(middleware.CtxUserID)
|
|
u, err := svc.Repo.User.FindByID(c.Request.Context(), toString(uid))
|
|
if err != nil || u == nil {
|
|
return ""
|
|
}
|
|
token, err := svc.Auth.IssueExternalPlaybackToken(u, mediaID, durationSec)
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
return token
|
|
}
|
|
|
|
func externalPlaybackURL(c *gin.Context, svc *service.Container, path string) string {
|
|
if strings.HasPrefix(path, "http://") || strings.HasPrefix(path, "https://") {
|
|
return path
|
|
}
|
|
headerOrigin := sanitizedPublicOrigin(c.GetHeader("X-MeBox-Public-Origin"))
|
|
if headerOrigin != "" && !isLocalPublicOrigin(headerOrigin) {
|
|
return joinOriginPath(headerOrigin, path)
|
|
}
|
|
if svc != nil {
|
|
if origin := sanitizedPublicOrigin(service.PublicServerURL(c.Request.Context(), svc.Repo, svc.Cfg)); origin != "" {
|
|
return joinOriginPath(origin, path)
|
|
}
|
|
}
|
|
if headerOrigin != "" {
|
|
return joinOriginPath(headerOrigin, path)
|
|
}
|
|
return absoluteRequestURL(c, path)
|
|
}
|
|
|
|
func isLocalPublicOrigin(origin string) bool {
|
|
u, err := url.Parse(origin)
|
|
if err != nil || u == nil {
|
|
return false
|
|
}
|
|
host := strings.ToLower(strings.Trim(u.Hostname(), "[]"))
|
|
switch host {
|
|
case "localhost", "127.0.0.1", "::1":
|
|
return true
|
|
default:
|
|
return strings.HasPrefix(host, "127.")
|
|
}
|
|
}
|
|
|
|
func sanitizedPublicOrigin(raw string) string {
|
|
raw = strings.TrimSpace(strings.Split(raw, ",")[0])
|
|
if raw == "" {
|
|
return ""
|
|
}
|
|
u, err := url.Parse(raw)
|
|
if err != nil || u == nil {
|
|
return ""
|
|
}
|
|
scheme := strings.ToLower(strings.TrimSpace(u.Scheme))
|
|
if scheme != "http" && scheme != "https" {
|
|
return ""
|
|
}
|
|
if strings.TrimSpace(u.Host) == "" {
|
|
return ""
|
|
}
|
|
u.Scheme = scheme
|
|
u.User = nil
|
|
u.Path = ""
|
|
u.RawPath = ""
|
|
u.RawQuery = ""
|
|
u.Fragment = ""
|
|
return strings.TrimRight(u.String(), "/")
|
|
}
|
|
|
|
func joinOriginPath(origin, path string) string {
|
|
if !strings.HasPrefix(path, "/") {
|
|
path = "/" + path
|
|
}
|
|
return strings.TrimRight(origin, "/") + path
|
|
}
|
|
|
|
func absoluteRequestURL(c *gin.Context, path string) string {
|
|
if strings.HasPrefix(path, "http://") || strings.HasPrefix(path, "https://") {
|
|
return path
|
|
}
|
|
scheme := strings.TrimSpace(c.GetHeader("X-Forwarded-Proto"))
|
|
if scheme == "" {
|
|
scheme = "http"
|
|
if c.Request.TLS != nil {
|
|
scheme = "https"
|
|
}
|
|
}
|
|
host := strings.TrimSpace(c.GetHeader("X-Forwarded-Host"))
|
|
if host == "" {
|
|
host = c.Request.Host
|
|
}
|
|
if !strings.HasPrefix(path, "/") {
|
|
path = "/" + path
|
|
}
|
|
return scheme + "://" + host + path
|
|
}
|
|
|
|
func setRedirectNoStoreHeaders(c *gin.Context) {
|
|
if c == nil {
|
|
return
|
|
}
|
|
c.Header("Cache-Control", "no-store, no-cache, must-revalidate, max-age=0")
|
|
c.Header("Pragma", "no-cache")
|
|
c.Header("Expires", "0")
|
|
}
|
|
|
|
// transcodeStatusHandler reports the live status of one transcode job.
|
|
// We surface the active jobs the transcoder knows about.
|
|
func transcodeStatusHandler(svc *service.Container) gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
jobID := c.Param("job_id")
|
|
for _, j := range svc.Transcoder.Active() {
|
|
if j.MediaID == jobID {
|
|
c.JSON(http.StatusOK, gin.H{"job_id": jobID, "status": "running", "job": j})
|
|
return
|
|
}
|
|
}
|
|
c.JSON(http.StatusOK, gin.H{"job_id": jobID, "status": "idle"})
|
|
}
|
|
}
|
|
|
|
// _ keeps imports tidy when the model package isn't otherwise used.
|
|
var _ = model.Media{}
|
|
var _ = service.Container{}
|