tcjlj/MeBox
1
0
Fork 0
mirror of https://github.com/truewhile/MeBox.git synced 2026-09-29 19:36:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: scrape STRM libraries after refresh

Browse Source
This commit is contained in:
ShukeBta
2026-07-03 01:18:27 +08:00
parent a6f2d0208d
commit 9833c60e38
12 changed files with 257 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/handler/strm.go
+100 -10
View File
@@ -108,6 +108,7 @@ type generateSTRMReq struct {
IncludeLocal *bool `json:"include_local"`
PreserveTree bool `json:"preserve_tree"`
Refresh bool `json:"refresh_library"`
ScrapeAfter bool `json:"scrape_after"`
}
type generateSTRMTreeReq struct {
@@ -125,6 +126,7 @@ type generateSTRMTreeReq struct {
TransferSubtitles bool `json:"transfer_subtitles"`
MissingOnly bool `json:"missing_only"`
RefreshLibrary bool `json:"refresh_library"`
ScrapeAfter bool `json:"scrape_after"`
}
type repairSTRMReq struct {
@@ -132,6 +134,7 @@ type repairSTRMReq struct {
BaseURL string `json:"base_url"`
DryRun bool `json:"dry_run"`
RefreshLibrary bool `json:"refresh_library"`
ScrapeAfter bool `json:"scrape_after"`
}
func listSTRMOutputPresetsHandler(svc *service.Container) gin.HandlerFunc {
@@ -186,7 +189,11 @@ func generateSTRMHandler(svc *service.Container) gin.HandlerFunc {
return
}
if req.Refresh {
res.Refresh = queueSTRMRefreshAfterChanges(c.Request.Context(), svc, res.OutputDir, "STRM 生成后刷新媒体库", strmGenerationChanged(res))
res.Refresh = queueSTRMRefreshAfterChanges(c.Request.Context(), svc, res.OutputDir, strmRefreshQueueOptions{
TaskName: "STRM 生成后刷新媒体库",
Changed: strmGenerationChanged(res),
ScrapeAfter: req.ScrapeAfter,
})
}
c.JSON(http.StatusOK, res)
}
@@ -227,7 +234,11 @@ func generateSTRMFromTreeHandler(svc *service.Container) gin.HandlerFunc {
return
}
if req.RefreshLibrary && !req.DryRun {
res.Refresh = queueSTRMRefreshAfterChanges(c.Request.Context(), svc, res.OutputDir, "STRM 目录树生成后刷新媒体库", strmGenerationChanged(res))
res.Refresh = queueSTRMRefreshAfterChanges(c.Request.Context(), svc, res.OutputDir, strmRefreshQueueOptions{
TaskName: "STRM 目录树生成后刷新媒体库",
Changed: strmGenerationChanged(res),
ScrapeAfter: req.ScrapeAfter,
})
}
c.JSON(http.StatusOK, res)
}
@@ -258,7 +269,11 @@ func repairSTRMHandler(svc *service.Container) gin.HandlerFunc {
return
}
if req.RefreshLibrary && !req.DryRun {
res.Refresh = queueSTRMRefreshAfterChanges(c.Request.Context(), svc, res.OutputDir, "STRM 修复后刷新媒体库", res.Repaired > 0)
res.Refresh = queueSTRMRefreshAfterChanges(c.Request.Context(), svc, res.OutputDir, strmRefreshQueueOptions{
TaskName: "STRM 修复后刷新媒体库",
Changed: res.Repaired > 0,
ScrapeAfter: req.ScrapeAfter,
})
}
c.JSON(http.StatusOK, res)
}
@@ -268,26 +283,43 @@ func strmGenerationChanged(res *service.GenerateSTRMResult) bool {
return res != nil && (res.Generated > 0 || res.Updated > 0 || res.Cleaned > 0)
}
func queueSTRMRefreshAfterChanges(ctx context.Context, svc *service.Container, outputDir, taskName string, changed bool) *service.STRMRefreshResult {
refresh := &service.STRMRefreshResult{Requested: true}
if !changed {
type strmRefreshQueueOptions struct {
TaskName string
Changed bool
ScrapeAfter bool
}
type strmRefreshRunOptions struct {
ScrapeAfter bool
}
func queueSTRMRefreshAfterChanges(ctx context.Context, svc *service.Container, outputDir string, options strmRefreshQueueOptions) *service.STRMRefreshResult {
refresh := &service.STRMRefreshResult{Requested: true, ScrapeRequested: options.ScrapeAfter}
if !options.Changed {
refresh.Reason = "no strm changes"
refresh.ScrapeReason = strmRefreshScrapeSkipReason(refresh)
return refresh
}
if svc == nil || svc.Scan == nil {
refresh.Reason = "scanner unavailable"
refresh.ScrapeReason = strmRefreshScrapeSkipReason(refresh)
return refresh
}
targets, err := service.FindSTRMRefreshTargets(ctx, svc.Repo, outputDir)
if err != nil {
refresh.Reason = err.Error()
refresh.ScrapeReason = strmRefreshScrapeSkipReason(refresh)
return refresh
}
if len(targets) == 0 {
refresh.Reason = "no matching local library"
refresh.ScrapeReason = strmRefreshScrapeSkipReason(refresh)
return refresh
}
refresh.Targets = targets
if options.ScrapeAfter && svc.Scraper == nil {
refresh.ScrapeReason = "scraper unavailable"
}
for _, target := range targets {
key := target.LibraryID
if target.RootID != "" {
@@ -298,16 +330,21 @@ func queueSTRMRefreshAfterChanges(ctx context.Context, svc *service.Container, o
continue
}
refresh.Queued = true
task := startScanHTTPTask(svc, taskName, target.Name, target.Path)
go runSTRMRefreshScan(svc, target, task, finishScan)
runOptions := strmRefreshRunOptions{ScrapeAfter: options.ScrapeAfter && svc.Scraper != nil}
if runOptions.ScrapeAfter {
refresh.ScrapeQueued = true
}
task := startScanHTTPTask(svc, options.TaskName, target.Name, target.Path)
go runSTRMRefreshScan(svc, target, task, finishScan, runOptions)
}
if !refresh.Queued {
refresh.Reason = "matching library already scanning"
refresh.ScrapeReason = strmRefreshScrapeSkipReason(refresh)
}
return refresh
}
func runSTRMRefreshScan(svc *service.Container, target service.STRMRefreshTarget, task *service.TaskHandle, finish func()) {
func runSTRMRefreshScan(svc *service.Container, target service.STRMRefreshTarget, task *service.TaskHandle, finish func(), options strmRefreshRunOptions) {
defer finish()
var (
res *service.ScanResult
@@ -322,7 +359,60 @@ func runSTRMRefreshScan(svc *service.Container, target service.STRMRefreshTarget
finishHTTPTask(task, err, "scan", "STRM 刷新媒体库失败", scanTaskMetrics(res), scanTaskDetails(res, 20))
return
}
finishHTTPTask(task, nil, "completed", "STRM 刷新媒体库结束", scanTaskMetrics(res), scanTaskDetails(res, 20))
if !options.ScrapeAfter {
finishHTTPTask(task, nil, "completed", "STRM 刷新媒体库结束", scanTaskMetrics(res), scanTaskDetails(res, 20))
return
}
scrape, reclassified, scrapeErr := runSTRMRefreshScrape(svc, target, task)
metrics := strmRefreshTaskMetrics(res, scrape, reclassified)
if scrapeErr != nil {
finishHTTPTask(task, scrapeErr, "scrape", "STRM 刷新媒体库完成，刮削失败", metrics, scanTaskDetails(res, 20))
return
}
finishHTTPTask(task, nil, "completed", "STRM 刷新媒体库和刮削结束", metrics, scanTaskDetails(res, 20))
}
func runSTRMRefreshScrape(svc *service.Container, target service.STRMRefreshTarget, task *service.TaskHandle) (service.EnrichLibraryResult, int, error) {
if task != nil {
task.Update(service.TaskUpdate{
Stage: "scrape",
SourcePath: target.Path,
Message: "正在刮削 STRM 媒体库",
})
}
result, err := svc.Scraper.EnrichLibraryDetailedWithOptions(context.Background(), target.LibraryID, service.ScrapeOptions{RetryNoMatch: true})
reclassified := 0
if result.Processed > 0 {
reclassified = reclassifyLibraryAfterScrape(context.Background(), svc, target.LibraryID)
}
return result, reclassified, err
}
func strmRefreshTaskMetrics(scan *service.ScanResult, scrape service.EnrichLibraryResult, reclassified int) map[string]int64 {
metrics := scanTaskMetrics(scan)
if metrics == nil {
metrics = map[string]int64{}
}
metrics["scrape_matched"] = int64(scrape.Matched)
metrics["scrape_processed"] = int64(scrape.Processed)
metrics["scrape_candidates"] = int64(scrape.Candidates)
if scrape.Failed > 0 {
metrics["scrape_failed"] = int64(scrape.Failed)
}
if reclassified > 0 {
metrics["scrape_reclassified"] = int64(reclassified)
}
return metrics
}
func strmRefreshScrapeSkipReason(refresh *service.STRMRefreshResult) string {
if refresh == nil || !refresh.ScrapeRequested {
return ""
}
if refresh.Reason != "" {
return "refresh not queued: " + refresh.Reason
}
return "refresh not queued"
}
func strmPlaybackTokenForRequest(c *gin.Context, svc *service.Container) string {
internal/handler/strm_test.go
+47
View File
@@ -0,0 +1,47 @@
package handler
import (
"testing"
"github.com/ShukeBta/MediaStationGo/internal/service"
)
func TestSTRMRefreshTaskMetricsIncludesScanAndScrape(t *testing.T) {
metrics := strmRefreshTaskMetrics(
&service.ScanResult{Visited: 3, Added: 2, Updated: 1, ErrorCount: 1},
service.EnrichLibraryResult{Matched: 4, Processed: 5, Candidates: 6, Failed: 1},
2,
)
want := map[string]int64{
"visited": 3,
"added": 2,
"updated": 1,
"errors": 1,
"scrape_matched": 4,
"scrape_processed": 5,
"scrape_candidates": 6,
"scrape_failed": 1,
"scrape_reclassified": 2,
}
for key, value := range want {
if metrics[key] != value {
t.Fatalf("metrics[%q] = %d, want %d in %#v", key, metrics[key], value, metrics)
}
}
}
func TestSTRMRefreshScrapeSkipReasonRequiresScrapeRequest(t *testing.T) {
if got := strmRefreshScrapeSkipReason(&service.STRMRefreshResult{Requested: true, Reason: "no strm changes"}); got != "" {
t.Fatalf("skip reason without scrape request = %q, want empty", got)
}
got := strmRefreshScrapeSkipReason(&service.STRMRefreshResult{
Requested: true,
ScrapeRequested: true,
Reason: "no matching local library",
})
if got != "refresh not queued: no matching local library" {
t.Fatalf("skip reason = %q", got)
}
}
internal/service/strm_refresh.go
+7 -4
View File
@@ -10,10 +10,13 @@ import (
)
type STRMRefreshResult struct {
Requested bool `json:"requested"`
Queued bool `json:"queued"`
Reason string `json:"reason,omitempty"`
Targets []STRMRefreshTarget `json:"targets,omitempty"`
Requested bool `json:"requested"`
Queued bool `json:"queued"`
Reason string `json:"reason,omitempty"`
ScrapeRequested bool `json:"scrape_requested,omitempty"`
ScrapeQueued bool `json:"scrape_queued,omitempty"`
ScrapeReason string `json:"scrape_reason,omitempty"`
Targets []STRMRefreshTarget `json:"targets,omitempty"`
}
type STRMRefreshTarget struct {
web/src/api/strm.ts
+6
View File
@@ -9,6 +9,7 @@ export type GenerateSTRMInput = {
include_local?: boolean
preserve_tree?: boolean
refresh_library?: boolean
scrape_after?: boolean
}
export type GenerateSTRMResult = {
@@ -51,6 +52,7 @@ export type GenerateSTRMTreeInput = {
transfer_subtitles?: boolean
missing_only?: boolean
refresh_library?: boolean
scrape_after?: boolean
}
export type RepairSTRMInput = {
@@ -58,6 +60,7 @@ export type RepairSTRMInput = {
base_url?: string
dry_run?: boolean
refresh_library?: boolean
scrape_after?: boolean
}
export type RepairSTRMResult = {
@@ -80,6 +83,9 @@ export type STRMRefreshResult = {
requested: boolean
queued: boolean
reason?: string
scrape_requested?: boolean
scrape_queued?: boolean
scrape_reason?: string
targets?: Array<{
library_id: string
root_id?: string
web/src/pages/StrmGenerateSection.tsx
+2
View File
@@ -27,6 +27,7 @@ export type StrmGenerateSectionProps = {
includeLocal: boolean
preserveTree: boolean
refreshLibrary: boolean
scrapeAfter: boolean
generating: boolean
generateResult: GenerateSTRMResult | null
playbackStatus: string
@@ -43,6 +44,7 @@ export type StrmGenerateSectionProps = {
setIncludeLocal: (value: boolean) => void
setPreserveTree: (value: boolean) => void
setRefreshLibrary: (value: boolean) => void
setScrapeAfter: (value: boolean) => void
}
export function StrmGenerateSection(props: StrmGenerateSectionProps) {
web/src/pages/StrmGenerateSectionParts.tsx
+39 -11
View File
@@ -1,6 +1,6 @@
import { Loader2, Save, Wand2 } from 'lucide-react'
import type { GenerateSTRMResult } from '../api/strm'
import type { GenerateSTRMResult, STRMRefreshResult } from '../api/strm'
import type { Library } from '../types'
import { currentOrigin, type CloudPlaybackMode } from './strmPageModel'
import type { StrmGenerateSectionProps } from './StrmGenerateSection'
@@ -147,6 +147,7 @@ type StrmGenerateFormProps = Pick<
| 'includeLocal'
| 'preserveTree'
| 'refreshLibrary'
| 'scrapeAfter'
| 'generating'
| 'onGenerate'
| 'setGenerateLibraryID'
@@ -156,6 +157,7 @@ type StrmGenerateFormProps = Pick<
| 'setIncludeLocal'
| 'setPreserveTree'
| 'setRefreshLibrary'
| 'setScrapeAfter'
>
export function StrmGenerateForm({
@@ -168,6 +170,7 @@ export function StrmGenerateForm({
includeLocal,
preserveTree,
refreshLibrary,
scrapeAfter,
generating,
onGenerate,
setGenerateLibraryID,
@@ -177,6 +180,7 @@ export function StrmGenerateForm({
setIncludeLocal,
setPreserveTree,
setRefreshLibrary,
setScrapeAfter,
}: StrmGenerateFormProps) {
return (
<form onSubmit={onGenerate} className="grid gap-3 md:grid-cols-4">
@@ -195,11 +199,17 @@ export function StrmGenerateForm({
>
使用当前访问地址
</button>
<div className="grid gap-2 md:col-span-4 md:grid-cols-4">
<div className="grid gap-2 md:col-span-4 md:grid-cols-5">
<CompactOption checked={overwrite} label="覆盖已存在" onChange={setOverwrite} />
<CompactOption checked={includeLocal} label="包含本地媒体" onChange={setIncludeLocal} />
<CompactOption checked={preserveTree} label="保留目录树" onChange={setPreserveTree} />
<CompactOption checked={refreshLibrary} label="生成后刷新媒体库" onChange={setRefreshLibrary} />
<CompactOption
checked={refreshLibrary && scrapeAfter}
disabled={!refreshLibrary}
label="刷新后自动刮削"
onChange={setScrapeAfter}
/>
</div>
<StrmOutputDirPicker
className="md:col-span-4"
@@ -218,16 +228,18 @@ export function StrmGenerateForm({
type CompactOptionProps = {
checked: boolean
disabled?: boolean
label: string
onChange: (value: boolean) => void
}
function CompactOption({ checked, label, onChange }: CompactOptionProps) {
function CompactOption({ checked, disabled, label, onChange }: CompactOptionProps) {
return (
<label className="flex min-h-10 items-center gap-2 rounded-2xl border border-gray-200 bg-white/70 px-3 py-2 text-sm text-ink-50">
<label className={`flex min-h-10 items-center gap-2 rounded-2xl border border-gray-200 bg-white/70 px-3 py-2 text-sm text-ink-50 ${disabled ? 'cursor-not-allowed opacity-50' : ''}`}>
<input
type="checkbox"
checked={checked}
disabled={disabled}
onChange={(e) => onChange(e.target.checked)}
/>
{label}
@@ -288,13 +300,7 @@ export function StrmGenerateResultPanel({ result }: { result: GenerateSTRMResult
{result.batch_limited && (
<div className="mt-1 text-amber-600">本批已达到上限，剩余 {result.remaining || 0} 个可继续运行。</div>
)}
{result.refresh && (
<div className={result.refresh.queued ? 'mt-1 text-emerald-600' : 'mt-1 text-amber-600'}>
{result.refresh.queued
? `媒体库刷新已排队：${result.refresh.targets?.map((target) => target.name).join('、') || '已匹配媒体库'}`
: `媒体库未刷新：${result.refresh.reason || '未匹配到可扫描媒体库'}`}
</div>
)}
<StrmRefreshStatus refresh={result.refresh} />
{result.ignored_items && result.ignored_items.length > 0 && (
<div className="mt-1 text-amber-600">
已忽略 {result.ignored || result.ignored_items.length} 个非视频/sidecar：{result.ignored_items.slice(0, 3).join('；')}
@@ -308,3 +314,25 @@ export function StrmGenerateResultPanel({ result }: { result: GenerateSTRMResult
</div>
)
}
export function StrmRefreshStatus({ refresh }: { refresh?: STRMRefreshResult }) {
if (!refresh) {
return null
}
return (
<>
<div className={refresh.queued ? 'mt-1 text-emerald-600' : 'mt-1 text-amber-600'}>
{refresh.queued
? `媒体库刷新已排队：${refresh.targets?.map((target) => target.name).join('、') || '已匹配媒体库'}`
: `媒体库未刷新：${refresh.reason || '未匹配到可扫描媒体库'}`}
</div>
{refresh.scrape_requested && (
<div className={refresh.scrape_queued ? 'mt-1 text-emerald-600' : 'mt-1 text-amber-600'}>
{refresh.scrape_queued
? '扫描完成后将自动刮削'
: `未安排刮削：${refresh.scrape_reason || '媒体库未刷新'}`}
</div>
)}
</>
)
}
web/src/pages/StrmPage.tsx
+2
View File
@@ -34,6 +34,7 @@ export function StrmPage() {
includeLocal={strm.generate.includeLocal}
preserveTree={strm.generate.preserveTree}
refreshLibrary={strm.generate.refreshLibrary}
scrapeAfter={strm.generate.scrapeAfter}
generating={strm.generate.generating}
generateResult={strm.generate.generateResult}
playbackStatus={strm.generate.playbackStatus}
@@ -50,6 +51,7 @@ export function StrmPage() {
setIncludeLocal={strm.generate.setIncludeLocal}
setPreserveTree={strm.generate.setPreserveTree}
setRefreshLibrary={strm.generate.setRefreshLibrary}
setScrapeAfter={strm.generate.setScrapeAfter}
/>
<StrmTreeGenerateSection {...strm.treeGenerate} outputPresets={strm.outputPresets} />
web/src/pages/StrmRepairSection.tsx
+13 -7
View File
@@ -2,6 +2,7 @@ import { Loader2, RotateCcw, Wrench } from 'lucide-react'
import type { STRMOutputPreset } from '../api/strm'
import { currentOrigin } from './strmPageModel'
import { StrmRefreshStatus } from './StrmGenerateSectionParts'
import { StrmOutputDirPicker } from './StrmOutputDirPicker'
import type { useStrmRepairForm } from './useStrmRepairForm'
@@ -17,11 +18,13 @@ export function StrmRepairSection({
refreshLibrary,
result,
runningMode,
scrapeAfter,
onPreview,
onRepair,
setBaseURL,
setOutputDir,
setRefreshLibrary,
setScrapeAfter,
}: StrmRepairSectionProps) {
return (
<section className="glass-panel space-y-4">
@@ -55,6 +58,15 @@ export function StrmRepairSection({
<input type="checkbox" checked={refreshLibrary} onChange={(e) => setRefreshLibrary(e.target.checked)} />
修复后刷新媒体库
</label>
<label className={`flex min-h-10 items-center gap-2 rounded-2xl border border-gray-200 bg-white/70 px-3 py-2 text-sm text-ink-50 md:col-span-4 ${!refreshLibrary ? 'cursor-not-allowed opacity-50' : ''}`}>
<input
type="checkbox"
checked={refreshLibrary && scrapeAfter}
disabled={!refreshLibrary}
onChange={(e) => setScrapeAfter(e.target.checked)}
/>
刷新后自动刮削
</label>
<button
type="button"
disabled={repairing || !outputDir.trim()}
@@ -76,13 +88,7 @@ export function StrmRepairSection({
{result.previewed ? `预检 ${result.previewed} · ` : ''}
修复 {result.repaired} · 跳过 {result.skipped}
</div>
{result.refresh && (
<div className={result.refresh.queued ? 'mt-1 text-emerald-600' : 'mt-1 text-amber-600'}>
{result.refresh.queued
? `媒体库刷新已排队：${result.refresh.targets?.map((target) => target.name).join('、') || '已匹配媒体库'}`
: `媒体库未刷新：${result.refresh.reason || '未匹配到可扫描媒体库'}`}
</div>
)}
<StrmRefreshStatus refresh={result.refresh} />
{result.errors && result.errors.length > 0 && (
<div className="mt-2 text-red-500">
失败 {result.errors.length} 条：{result.errors.slice(0, 3).join('；')}
web/src/pages/StrmTreeGenerateSection.tsx
+11
View File
@@ -49,6 +49,7 @@ export function StrmTreeGenerateSection({
refreshLibrary,
result,
runningMode,
scrapeAfter,
setBaseURL,
setBatchLimit,
setCleanup,
@@ -59,6 +60,7 @@ export function StrmTreeGenerateSection({
setPathsText,
setProvider,
setRefreshLibrary,
setScrapeAfter,
setSourceRoot,
setTreeText,
setTransferSubtitles,
@@ -203,6 +205,15 @@ export function StrmTreeGenerateSection({
<input type="checkbox" checked={refreshLibrary} onChange={(e) => setRefreshLibrary(e.target.checked)} />
生成后刷新媒体库
</label>
<label className={`flex min-h-10 items-center gap-2 rounded-2xl border border-gray-200 bg-white/70 px-3 py-2 text-sm text-ink-50 md:col-span-4 ${!refreshLibrary ? 'cursor-not-allowed opacity-50' : ''}`}>
<input
type="checkbox"
checked={refreshLibrary && scrapeAfter}
disabled={!refreshLibrary}
onChange={(e) => setScrapeAfter(e.target.checked)}
/>
刷新后自动刮削
</label>
<label className="flex min-h-10 items-center gap-2 rounded-2xl border border-gray-200 bg-white/70 px-3 py-2 text-sm text-ink-50 md:col-span-4">
<input
type="checkbox"
web/src/pages/useStrmGenerateForm.ts
+10 -1
View File
@@ -34,6 +34,7 @@ export function useStrmGenerateForm(libraries: Library[]) {
const [includeLocal, setIncludeLocal] = useState(true)
const [preserveTree, setPreserveTree] = useState(false)
const [refreshLibrary, setRefreshLibrary] = useState(true)
const [scrapeAfter, setScrapeAfter] = useState(false)
const [generating, setGenerating] = useState(false)
const [generateResult, setGenerateResult] = useState<GenerateSTRMResult | null>(null)
@@ -119,6 +120,7 @@ export function useStrmGenerateForm(libraries: Library[]) {
include_local: includeLocal,
preserve_tree: preserveTree,
refresh_library: refreshLibrary,
scrape_after: refreshLibrary && scrapeAfter,
})
const nextOutputDir = result.output_dir || outputDir
setGenerateResult(result)
@@ -135,6 +137,11 @@ export function useStrmGenerateForm(libraries: Library[]) {
}
}
const setRefreshLibraryEnabled = (value: boolean) => {
setRefreshLibrary(value)
if (!value) setScrapeAfter(false)
}
const saveSTRMSettings = async () => {
setSavingSettings(true)
try {
@@ -177,6 +184,7 @@ export function useStrmGenerateForm(libraries: Library[]) {
overwrite,
preserveTree,
refreshLibrary,
scrapeAfter,
playbackStatus: playbackStatusText(strmPlaybackEnabled, redirectProxyEnabled, cloudPlaybackMode),
redirectProxyEnabled,
saveSTRMSettings,
@@ -189,7 +197,8 @@ export function useStrmGenerateForm(libraries: Library[]) {
setOutputDir: onOutputDirChange,
setOverwrite,
setPreserveTree,
setRefreshLibrary,
setRefreshLibrary: setRefreshLibraryEnabled,
setScrapeAfter,
setRedirectProxyEnabled,
setStrmPlaybackEnabled,
strmPlaybackEnabled,
web/src/pages/useStrmRepairForm.ts
+10 -1
View File
@@ -10,6 +10,7 @@ export function useStrmRepairForm() {
const [baseURL, setBaseURL] = useState(currentOrigin())
const [outputDir, setOutputDir] = useState('data/strm/tree')
const [refreshLibrary, setRefreshLibrary] = useState(true)
const [scrapeAfter, setScrapeAfter] = useState(false)
const [runningMode, setRunningMode] = useState<'repair' | 'preview' | null>(null)
const [result, setResult] = useState<RepairSTRMResult | null>(null)
@@ -30,6 +31,7 @@ export function useStrmRepairForm() {
base_url: trimmedBaseURL.replace(/\/+$/, '') || undefined,
dry_run: dryRun,
refresh_library: refreshLibrary,
scrape_after: !dryRun && refreshLibrary && scrapeAfter,
})
setResult(next)
if (dryRun) {
@@ -53,17 +55,24 @@ export function useStrmRepairForm() {
await runRepair(true)
}
const setRefreshLibraryEnabled = (value: boolean) => {
setRefreshLibrary(value)
if (!value) setScrapeAfter(false)
}
return {
baseURL,
outputDir,
repairing: runningMode !== null,
refreshLibrary,
scrapeAfter,
result,
runningMode,
onPreview,
onRepair,
setBaseURL,
setOutputDir,
setRefreshLibrary,
setRefreshLibrary: setRefreshLibraryEnabled,
setScrapeAfter,
}
}
web/src/pages/useStrmTreeGenerateForm.ts
+10 -1
View File
@@ -20,6 +20,7 @@ export function useStrmTreeGenerateForm() {
const [transferSubtitles, setTransferSubtitles] = useState(true)
const [missingOnly, setMissingOnly] = useState(false)
const [refreshLibrary, setRefreshLibrary] = useState(true)
const [scrapeAfter, setScrapeAfter] = useState(false)
const [runningMode, setRunningMode] = useState<'generate' | 'preview' | null>(null)
const [result, setResult] = useState<GenerateSTRMResult | null>(null)
@@ -75,6 +76,7 @@ export function useStrmTreeGenerateForm() {
transfer_subtitles: transferSubtitles,
missing_only: missingOnly,
refresh_library: refreshLibrary,
scrape_after: !dryRun && refreshLibrary && scrapeAfter,
})
setResult(next)
if (dryRun) {
@@ -98,6 +100,11 @@ export function useStrmTreeGenerateForm() {
await runGenerate(true)
}
const setRefreshLibraryEnabled = (value: boolean) => {
setRefreshLibrary(value)
if (!value) setScrapeAfter(false)
}
return {
baseURL,
batchLimit,
@@ -115,6 +122,7 @@ export function useStrmTreeGenerateForm() {
result,
runningMode,
refreshLibrary,
scrapeAfter,
setBaseURL,
setBatchLimit,
setCleanup,
@@ -124,7 +132,8 @@ export function useStrmTreeGenerateForm() {
setOverwrite,
setPathsText,
setProvider,
setRefreshLibrary,
setRefreshLibrary: setRefreshLibraryEnabled,
setScrapeAfter,
setSourceRoot,
setTreeText,
setTransferSubtitles,