mirror of
https://github.com/truewhile/MeBox.git
synced 2026-09-29 03:26:37 +08:00
feat: add STRM repair workflow
This commit is contained in:
ShukeBta
@@ -55,6 +55,7 @@ func registerAuthedSTRMRoutes(authed *gin.RouterGroup, svc *service.Container) {
|
||||
authed.POST("/strm/import", middleware.AdminRequired(), importSTRMHandler(svc))
|
||||
authed.POST("/strm/generate", middleware.AdminRequired(), generateSTRMHandler(svc))
|
||||
authed.POST("/strm/generate-from-tree", middleware.AdminRequired(), generateSTRMFromTreeHandler(svc))
|
||||
authed.POST("/strm/repair", middleware.AdminRequired(), repairSTRMHandler(svc))
|
||||
}
|
||||
|
||||
func registerAuthedDuplicateRoutes(authed *gin.RouterGroup, svc *service.Container) {
|
||||
|
||||
@@ -123,6 +123,12 @@ type generateSTRMTreeReq struct {
|
||||
TransferSubtitles bool `json:"transfer_subtitles"`
|
||||
}
|
||||
|
||||
type repairSTRMReq struct {
|
||||
OutputDir string `json:"output_dir" binding:"required"`
|
||||
BaseURL string `json:"base_url"`
|
||||
DryRun bool `json:"dry_run"`
|
||||
}
|
||||
|
||||
func generateSTRMHandler(svc *service.Container) gin.HandlerFunc {
|
||||
return func(c *gin.Context) {
|
||||
var req generateSTRMReq
|
||||
@@ -204,6 +210,34 @@ func generateSTRMFromTreeHandler(svc *service.Container) gin.HandlerFunc {
|
||||
}
|
||||
}
|
||||
|
||||
func repairSTRMHandler(svc *service.Container) gin.HandlerFunc {
|
||||
return func(c *gin.Context) {
|
||||
var req repairSTRMReq
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
strmSvc := svc.STRM
|
||||
if strmSvc == nil {
|
||||
strmSvc = service.NewSTRMService(svc.Log, svc.Repo, svc.Cfg)
|
||||
}
|
||||
baseURL := strings.TrimRight(strings.TrimSpace(req.BaseURL), "/")
|
||||
if baseURL == "" {
|
||||
baseURL = strings.TrimRight(absoluteRequestURL(c, "/"), "/")
|
||||
}
|
||||
res, err := strmSvc.RepairFiles(c.Request.Context(), service.RepairSTRMOptions{
|
||||
OutputDir: req.OutputDir,
|
||||
BaseURL: baseURL,
|
||||
DryRun: req.DryRun,
|
||||
})
|
||||
if err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
c.JSON(http.StatusOK, res)
|
||||
}
|
||||
}
|
||||
|
||||
func strmPlaybackTokenForRequest(c *gin.Context, svc *service.Container) string {
|
||||
if svc == nil || svc.Auth == nil || svc.Repo == nil || svc.Repo.User == nil {
|
||||
return ""
|
||||
|
||||
@@ -0,0 +1,144 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"io/fs"
|
||||
"net/url"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type RepairSTRMOptions struct {
|
||||
OutputDir string `json:"output_dir"`
|
||||
BaseURL string `json:"base_url,omitempty"`
|
||||
DryRun bool `json:"dry_run,omitempty"`
|
||||
}
|
||||
|
||||
type RepairSTRMResult struct {
|
||||
OutputDir string `json:"output_dir"`
|
||||
Repaired int `json:"repaired"`
|
||||
Previewed int `json:"previewed,omitempty"`
|
||||
Skipped int `json:"skipped"`
|
||||
Errors []string `json:"errors,omitempty"`
|
||||
Items []RepairSTRMItem `json:"items,omitempty"`
|
||||
}
|
||||
|
||||
type RepairSTRMItem struct {
|
||||
FilePath string `json:"file_path"`
|
||||
Before string `json:"before,omitempty"`
|
||||
After string `json:"after,omitempty"`
|
||||
Action string `json:"action"`
|
||||
Reason string `json:"reason,omitempty"`
|
||||
}
|
||||
|
||||
func (s *STRMService) RepairFiles(ctx context.Context, opts RepairSTRMOptions) (*RepairSTRMResult, error) {
|
||||
outputDir := resolveMappedDestinationPath(strings.TrimSpace(opts.OutputDir))
|
||||
if outputDir == "" || outputDir == "." {
|
||||
return nil, errors.New("output_dir required")
|
||||
}
|
||||
info, err := os.Stat(outputDir)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if !info.IsDir() {
|
||||
return nil, errors.New("output_dir must be a directory")
|
||||
}
|
||||
result := &RepairSTRMResult{OutputDir: outputDir}
|
||||
baseURL := strings.TrimRight(strings.TrimSpace(opts.BaseURL), "/")
|
||||
err = filepath.WalkDir(outputDir, func(filePath string, entry fs.DirEntry, walkErr error) error {
|
||||
if walkErr != nil {
|
||||
result.Errors = append(result.Errors, filePath+": "+walkErr.Error())
|
||||
return nil
|
||||
}
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
return ctx.Err()
|
||||
default:
|
||||
}
|
||||
if entry.IsDir() || !strings.EqualFold(filepath.Ext(entry.Name()), ".strm") {
|
||||
return nil
|
||||
}
|
||||
result.addRepairItem(repairSTRMFile(filePath, baseURL, opts.DryRun))
|
||||
return nil
|
||||
})
|
||||
if err != nil {
|
||||
return result, err
|
||||
}
|
||||
return result, nil
|
||||
}
|
||||
|
||||
func repairSTRMFile(filePath, baseURL string, dryRun bool) RepairSTRMItem {
|
||||
item := RepairSTRMItem{FilePath: filePath}
|
||||
body, err := os.ReadFile(filePath) // #nosec G304 -- admin-selected STRM repair output directory.
|
||||
if err != nil {
|
||||
item.Action = "error"
|
||||
item.Reason = err.Error()
|
||||
return item
|
||||
}
|
||||
before := strings.TrimSpace(string(body))
|
||||
after, ok := repairedSTRMPlaybackURL(before, baseURL)
|
||||
if !ok {
|
||||
item.Action = "skipped"
|
||||
item.Reason = "unsupported strm target"
|
||||
return item
|
||||
}
|
||||
if after == before {
|
||||
item.Action = "skipped"
|
||||
item.Reason = "already current"
|
||||
return item
|
||||
}
|
||||
item.Before = before
|
||||
item.After = after
|
||||
if dryRun {
|
||||
item.Action = "preview"
|
||||
item.Reason = "repaired"
|
||||
return item
|
||||
}
|
||||
if err := os.WriteFile(filePath, []byte(after+"\n"), 0o644); err != nil { // #nosec G306 -- STRM files are player-readable sidecars.
|
||||
item.Action = "error"
|
||||
item.Reason = err.Error()
|
||||
return item
|
||||
}
|
||||
item.Action = "repaired"
|
||||
return item
|
||||
}
|
||||
|
||||
func repairedSTRMPlaybackURL(raw, baseURL string) (string, bool) {
|
||||
raw = strings.TrimSpace(raw)
|
||||
if raw == "" || strings.HasPrefix(raw, "//") {
|
||||
return "", false
|
||||
}
|
||||
parsed, err := url.Parse(raw)
|
||||
if err != nil || !strmRepairOwnsAPIPath(parsed.Path) {
|
||||
return "", false
|
||||
}
|
||||
apiPath := parsed.EscapedPath()
|
||||
if apiPath == "" {
|
||||
apiPath = parsed.Path
|
||||
}
|
||||
if parsed.RawQuery != "" {
|
||||
apiPath += "?" + parsed.RawQuery
|
||||
}
|
||||
return buildAbsoluteSTRMAPIURL(baseURL, apiPath, nil), true
|
||||
}
|
||||
|
||||
func strmRepairOwnsAPIPath(apiPath string) bool {
|
||||
value := strings.ToLower(strings.TrimSpace(apiPath))
|
||||
return strings.HasPrefix(value, "/api/stream/") || strings.HasPrefix(value, "/api/cloud/play/")
|
||||
}
|
||||
|
||||
func (r *RepairSTRMResult) addRepairItem(item RepairSTRMItem) {
|
||||
r.Items = append(r.Items, item)
|
||||
switch item.Action {
|
||||
case "repaired":
|
||||
r.Repaired++
|
||||
case "preview":
|
||||
r.Previewed++
|
||||
case "skipped":
|
||||
r.Skipped++
|
||||
case "error":
|
||||
r.Errors = append(r.Errors, item.FilePath+": "+item.Reason)
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,77 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
"go.uber.org/zap"
|
||||
)
|
||||
|
||||
func TestRepairSTRMFilesRewritesOwnedPlaybackURLs(t *testing.T) {
|
||||
outDir := t.TempDir()
|
||||
cloudPath := filepath.Join(outDir, "Movies", "A.strm")
|
||||
streamPath := filepath.Join(outDir, "Shows", "S01E01.strm")
|
||||
externalPath := filepath.Join(outDir, "External.strm")
|
||||
for _, path := range []string{cloudPath, streamPath, externalPath} {
|
||||
if err := os.MkdirAll(filepath.Dir(path), 0o755); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
if err := os.WriteFile(cloudPath, []byte("http://old.local/api/cloud/play/openlist?ref=%2FMovies%2FA.mkv\n"), 0o644); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if err := os.WriteFile(streamPath, []byte("/api/stream/media-1?token=old\n"), 0o644); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if err := os.WriteFile(externalPath, []byte("https://cdn.example.com/video.m3u8\n"), 0o644); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
svc := NewSTRMService(zap.NewNop(), nil, nil)
|
||||
res, err := svc.RepairFiles(t.Context(), RepairSTRMOptions{
|
||||
OutputDir: outDir,
|
||||
BaseURL: "https://media.example.com",
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if res.Repaired != 2 || res.Skipped != 1 || len(res.Errors) != 0 {
|
||||
t.Fatalf("result = %#v, want two repaired and one skipped", res)
|
||||
}
|
||||
if got := readSTRM(t, cloudPath); got != "https://media.example.com/api/cloud/play/openlist?ref=%2FMovies%2FA.mkv" {
|
||||
t.Fatalf("cloud strm = %q", got)
|
||||
}
|
||||
if got := readSTRM(t, streamPath); got != "https://media.example.com/api/stream/media-1?token=old" {
|
||||
t.Fatalf("stream strm = %q", got)
|
||||
}
|
||||
if got := readSTRM(t, externalPath); got != "https://cdn.example.com/video.m3u8" {
|
||||
t.Fatalf("external strm should not change: %q", got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestRepairSTRMFilesDryRunDoesNotWrite(t *testing.T) {
|
||||
outDir := t.TempDir()
|
||||
filePath := filepath.Join(outDir, "Movie.strm")
|
||||
original := "http://old.local/api/cloud/play/openlist?ref=%2FMovie.mkv\n"
|
||||
if err := os.WriteFile(filePath, []byte(original), 0o644); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
svc := NewSTRMService(zap.NewNop(), nil, nil)
|
||||
|
||||
res, err := svc.RepairFiles(t.Context(), RepairSTRMOptions{
|
||||
OutputDir: outDir,
|
||||
BaseURL: "https://media.example.com",
|
||||
DryRun: true,
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if res.Previewed != 1 || res.Repaired != 0 || len(res.Items) != 1 {
|
||||
t.Fatalf("result = %#v, want one repair preview", res)
|
||||
}
|
||||
if got := readSTRM(t, filePath); got != strings.TrimSpace(original) {
|
||||
t.Fatalf("dry run changed file: %q", got)
|
||||
}
|
||||
}
|
||||
@@ -49,6 +49,27 @@ export type GenerateSTRMTreeInput = {
|
||||
transfer_subtitles?: boolean
|
||||
}
|
||||
|
||||
export type RepairSTRMInput = {
|
||||
output_dir: string
|
||||
base_url?: string
|
||||
dry_run?: boolean
|
||||
}
|
||||
|
||||
export type RepairSTRMResult = {
|
||||
output_dir: string
|
||||
repaired: number
|
||||
previewed?: number
|
||||
skipped: number
|
||||
errors?: string[]
|
||||
items?: Array<{
|
||||
file_path: string
|
||||
before?: string
|
||||
after?: string
|
||||
action: string
|
||||
reason?: string
|
||||
}>
|
||||
}
|
||||
|
||||
export const strmAPI = {
|
||||
set: (mediaID: string, url: string) =>
|
||||
api.put(`/media/${mediaID}/strm`, { url }).then((r) => r.data),
|
||||
@@ -63,4 +84,8 @@ export const strmAPI = {
|
||||
api
|
||||
.post<GenerateSTRMResult>('/strm/generate-from-tree', input, { timeout: BATCH_REQUEST_TIMEOUT })
|
||||
.then((r) => r.data),
|
||||
repair: (input: RepairSTRMInput) =>
|
||||
api
|
||||
.post<RepairSTRMResult>('/strm/repair', input, { timeout: BATCH_REQUEST_TIMEOUT })
|
||||
.then((r) => r.data),
|
||||
}
|
||||
|
||||
@@ -2,6 +2,7 @@ import { StrmAttachSection } from './StrmAttachSection'
|
||||
import { StrmGenerateSection } from './StrmGenerateSection'
|
||||
import { StrmImportSection } from './StrmImportSection'
|
||||
import { StrmPageHeader } from './StrmPageHeader'
|
||||
import { StrmRepairSection } from './StrmRepairSection'
|
||||
import { StrmTreeGenerateSection } from './StrmTreeGenerateSection'
|
||||
import { useStrmPage } from './useStrmPage'
|
||||
|
||||
@@ -50,6 +51,8 @@ export function StrmPage() {
|
||||
|
||||
<StrmTreeGenerateSection {...strm.treeGenerate} />
|
||||
|
||||
<StrmRepairSection {...strm.repair} />
|
||||
|
||||
<StrmImportSection
|
||||
libraries={strm.libraries}
|
||||
libraryID={strm.importForm.libraryID}
|
||||
|
||||
@@ -0,0 +1,76 @@
|
||||
import { Loader2, RotateCcw, Wrench } from 'lucide-react'
|
||||
|
||||
import { currentOrigin } from './strmPageModel'
|
||||
import type { useStrmRepairForm } from './useStrmRepairForm'
|
||||
|
||||
type StrmRepairSectionProps = ReturnType<typeof useStrmRepairForm>
|
||||
|
||||
export function StrmRepairSection({
|
||||
baseURL,
|
||||
outputDir,
|
||||
repairing,
|
||||
result,
|
||||
runningMode,
|
||||
onPreview,
|
||||
onRepair,
|
||||
setBaseURL,
|
||||
setOutputDir,
|
||||
}: StrmRepairSectionProps) {
|
||||
return (
|
||||
<section className="glass-panel space-y-4">
|
||||
<div>
|
||||
<h2 className="font-display text-lg font-semibold text-ink-600">STRM 修复</h2>
|
||||
<p className="text-sm text-ink-50">批量修复已存在 .strm 文件中的本服务播放地址。</p>
|
||||
</div>
|
||||
<form onSubmit={onRepair} className="grid gap-3 md:grid-cols-4">
|
||||
<input
|
||||
required
|
||||
className="input-base md:col-span-2"
|
||||
placeholder="STRM 输出目录"
|
||||
value={outputDir}
|
||||
onChange={(e) => setOutputDir(e.target.value)}
|
||||
/>
|
||||
<input
|
||||
className="input-base"
|
||||
placeholder="http://NAS-IP:18080 或 https://media.example.com"
|
||||
value={baseURL}
|
||||
onChange={(e) => setBaseURL(e.target.value)}
|
||||
/>
|
||||
<button
|
||||
type="button"
|
||||
className="rounded-2xl border border-primary-400/40 px-3 py-2 text-sm text-brand-500 transition hover:bg-primary-400/10"
|
||||
onClick={() => setBaseURL(currentOrigin())}
|
||||
>
|
||||
使用当前访问地址
|
||||
</button>
|
||||
<button
|
||||
type="button"
|
||||
disabled={repairing || !outputDir.trim()}
|
||||
className="inline-flex min-h-10 items-center justify-center gap-2 rounded-2xl border border-primary-400/40 px-3 py-2 text-sm font-medium text-brand-500 transition hover:bg-primary-400/10 disabled:cursor-not-allowed disabled:opacity-50 md:col-span-2"
|
||||
onClick={onPreview}
|
||||
>
|
||||
{runningMode === 'preview' ? <Loader2 size={16} className="animate-spin" /> : <RotateCcw size={16} />}
|
||||
{runningMode === 'preview' ? '预检中...' : '预检修复'}
|
||||
</button>
|
||||
<button type="submit" disabled={repairing || !outputDir.trim()} className="neon-button md:col-span-2">
|
||||
{runningMode === 'repair' ? <Loader2 size={16} className="animate-spin" /> : <Wrench size={16} />}
|
||||
{runningMode === 'repair' ? '修复中...' : '修复 STRM'}
|
||||
</button>
|
||||
</form>
|
||||
{result && (
|
||||
<div className="rounded-2xl border border-gray-200 bg-gray-50 p-4 text-sm text-ink-50">
|
||||
<div className="font-semibold text-ink-600">输出目录：{result.output_dir}</div>
|
||||
<div className="mt-1">
|
||||
{result.previewed ? `预检 ${result.previewed} · ` : ''}
|
||||
修复 {result.repaired} · 跳过 {result.skipped}
|
||||
</div>
|
||||
{result.errors && result.errors.length > 0 && (
|
||||
<div className="mt-2 text-red-500">
|
||||
失败 {result.errors.length} 条：{result.errors.slice(0, 3).join('；')}
|
||||
</div>
|
||||
)}
|
||||
</div>
|
||||
)}
|
||||
</section>
|
||||
)
|
||||
}
|
||||
@@ -5,12 +5,14 @@ import type { Library } from '../types'
|
||||
import { useStrmAttachForm } from './useStrmAttachForm'
|
||||
import { useStrmGenerateForm } from './useStrmGenerateForm'
|
||||
import { useStrmImportForm } from './useStrmImportForm'
|
||||
import { useStrmRepairForm } from './useStrmRepairForm'
|
||||
import { useStrmTreeGenerateForm } from './useStrmTreeGenerateForm'
|
||||
|
||||
export function useStrmPage() {
|
||||
const [libraries, setLibraries] = useState<Library[]>([])
|
||||
const generate = useStrmGenerateForm(libraries)
|
||||
const treeGenerate = useStrmTreeGenerateForm()
|
||||
const repair = useStrmRepairForm()
|
||||
const importForm = useStrmImportForm(libraries)
|
||||
const attach = useStrmAttachForm()
|
||||
|
||||
@@ -23,6 +25,7 @@ export function useStrmPage() {
|
||||
generate,
|
||||
importForm,
|
||||
libraries,
|
||||
repair,
|
||||
treeGenerate,
|
||||
}
|
||||
}
|
||||
|
||||
@@ -0,0 +1,65 @@
|
||||
import type { FormEvent } from 'react'
|
||||
import { useState } from 'react'
|
||||
import toast from 'react-hot-toast'
|
||||
|
||||
import { strmAPI, type RepairSTRMResult } from '../api/strm'
|
||||
import { currentOrigin } from './strmPageModel'
|
||||
import { apiErrorMessage, isHTTPURL } from './strmPageUtils'
|
||||
|
||||
export function useStrmRepairForm() {
|
||||
const [baseURL, setBaseURL] = useState(currentOrigin())
|
||||
const [outputDir, setOutputDir] = useState('data/strm/tree')
|
||||
const [runningMode, setRunningMode] = useState<'repair' | 'preview' | null>(null)
|
||||
const [result, setResult] = useState<RepairSTRMResult | null>(null)
|
||||
|
||||
const runRepair = async (dryRun: boolean) => {
|
||||
const trimmedBaseURL = baseURL.trim()
|
||||
if (!outputDir.trim()) {
|
||||
toast.error('请填写 STRM 输出目录')
|
||||
return
|
||||
}
|
||||
if (trimmedBaseURL && !isHTTPURL(trimmedBaseURL)) {
|
||||
toast.error('域名必须以 http:// 或 https:// 开头')
|
||||
return
|
||||
}
|
||||
setRunningMode(dryRun ? 'preview' : 'repair')
|
||||
try {
|
||||
const next = await strmAPI.repair({
|
||||
output_dir: outputDir.trim(),
|
||||
base_url: trimmedBaseURL.replace(/\/+$/, '') || undefined,
|
||||
dry_run: dryRun,
|
||||
})
|
||||
setResult(next)
|
||||
if (dryRun) {
|
||||
toast.success(`预检完成：可修复 ${next.previewed ?? 0} 个 · 跳过 ${next.skipped}`)
|
||||
} else {
|
||||
toast.success(`修复完成：修复 ${next.repaired} 个 · 跳过 ${next.skipped}`)
|
||||
}
|
||||
} catch (err: unknown) {
|
||||
toast.error(apiErrorMessage(err, dryRun ? 'STRM 修复预检失败' : 'STRM 修复失败'))
|
||||
} finally {
|
||||
setRunningMode(null)
|
||||
}
|
||||
}
|
||||
|
||||
const onRepair = async (event: FormEvent) => {
|
||||
event.preventDefault()
|
||||
await runRepair(false)
|
||||
}
|
||||
|
||||
const onPreview = async () => {
|
||||
await runRepair(true)
|
||||
}
|
||||
|
||||
return {
|
||||
baseURL,
|
||||
outputDir,
|
||||
repairing: runningMode !== null,
|
||||
result,
|
||||
runningMode,
|
||||
onPreview,
|
||||
onRepair,
|
||||
setBaseURL,
|
||||
setOutputDir,
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user